FFmpeg - нужна консультация, как я могу узнать что кадр был прочитан? - C/C++ - Ответ 16682967

#include <Windows.h>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <fstream>
#include <cstring>
#include <math.h>
#include <string.h>
 
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>
#include <libavformat/avio.h>
#include "libavutil/imgutils.h"
#include "libavutil/opt.h"
#include <libavutil/imgutils.h>
#include <libavutil/samplefmt.h>
#include <libavutil/timestamp.h>
#include <libavformat/avformat.h>
#include <libswresample/swresample.h>
#include "libavcodec/avcodec.h"
#include "libavcodec/avfft.h"
#include "libavdevice/avdevice.h"
#include "libavfilter/avfilter.h"
#include "libavfilter/buffersink.h"
#include "libavfilter/buffersrc.h"
#include "libavformat/avformat.h"
#include "libavformat/avio.h"
#include "libavutil/opt.h"
#include "libavutil/common.h"
#include "libavutil/channel_layout.h"
#include "libavutil/imgutils.h"
#include "libavutil/mathematics.h"
#include "libavutil/samplefmt.h"
#include "libavutil/time.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/file.h"
#include "libswscale/swscale.h"
#include <libavutil/imgutils.h>
#include <libavutil/samplefmt.h>
#include <libavutil/timestamp.h>
#include <libavformat/avformat.h>
 
}
#define INBUF_SIZE 4096
 
 
static void save_gray_frame(unsigned char* buf, int wrap, int xsize, int ysize, char* filename)
{
    return;
    FILE* f = NULL;
    int i;
    f = fopen(filename, "w");
    if (f == NULL) {
        fprintf(stderr, "ffmpeg: Не удалось создать новый файл\n");
    }
    fprintf(f, "P5\n%d %d\n%d\n", xsize, ysize, 255);
    for (i = 0; i < ysize; i++)
        fwrite(buf + i * wrap, 1, xsize, f);
    fclose(f);
}
 
void pgm_save(unsigned char* buf, int wrap, int xsize, int ysize, FILE* f) {
    fprintf(f, "P5\n%d %d\n%d\n", wrap, xsize, ysize);
    for (int i = 0; i < ysize; i++) {
        fwrite(buf + i * wrap, 1, xsize, f);
    }
}
 
void decode(AVCodecContext* dec_ctx, AVFrame* Frame, AVPacket* pkt, FILE* f)
{
    int ret;
    ret = avcodec_send_packet(dec_ctx, pkt);
    if (ret < 0) {
        fprintf(stderr, "Ошибка отправки пакета в кодирование.\n");
        exit(1);
    }
 
    while (ret >= 0) {
        ret = avcodec_receive_frame(dec_ctx, Frame);
        if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
            return;
        }
        else if (ret < 0) {
            fprintf(stderr, "during decoding.\n");
            exit(1);
        }
        printf("sawing frame %3d\n", dec_ctx->frame_number);
        fflush(stdout);
        pgm_save(Frame->data[0], Frame->linesize[0], Frame->width, Frame->height, f);
    }
}
 
int main(int argc, char* argv[])
{
    setlocale(LC_ALL, "Rus");
 
    const AVCodec* codec = NULL;
    AVCodecParserContext* parser = NULL;
 
    // содержит данные о конфигурации медиаконтента, такой как битрейт, 
    // частота кадров, частота дискретизации, каналы, высота и многое другое.
    AVCodecContext* AVCodecContext_ = NULL;
    AVCodecParameters* AVCodecParametr_ = NULL;
    FILE* f = NULL;
 
    //Эта структура описывает декодированные (необработанные) аудио- или видеоданные.
    AVFrame* frame = NULL;
    uint8_t inbuf[INBUF_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
    uint8_t* data = NULL;
    size_t   data_size = NULL;
    int ret;
    int eof;
    AVFormatContext* AVfc = NULL;
    int ERRORS;
    //AVCodec* codec;
    char buf[1024];
 
    // Хранит сжатый один кадр.
    AVPacket* pkt = NULL;
 
    //**********************************************************************
    // Ниже переменные которые я задействовал для части кодирования, что бы небыло перехлеста
    // с теми которые используются для декодирования. 
    //**********************************************************************
 
    const AVCodec* AVCodecDE = NULL;
    AVCodecContext* AVCodecContextDE = NULL;
    AVFrame* AVframeDE = NULL;
    AVPacket* AVpktDE;
    FILE* fDE;
 
    //Входящий видео файл AVI
    const char* InputFileName = "c:\\1.avi";
    // Адрес с видеокамеры. 
    const char* InputCam = "rtsp://admin:admin192.168.1.75:554/11";
    //Определение видео файла для сохранения как видео.
    const char* OutFileNameDE = "C:\\Users\\Users\\Desktop\\2.yuv";
    //Определяем имя файла для сохранения картинки.
    const char* FileName1 = "C:\\Users\\Users\\Desktop\\NyFile.PGM";
 
    //**********************************************************************
    // Начало чтения видео с камеры. 
    //**********************************************************************
 
    avdevice_register_all();
 
    // Открываем поток видео (это файл или камера). 
    ERRORS = avformat_open_input(&AVfc, InputCam, NULL, NULL);
    if (ERRORS < 0) {
        fprintf(stderr, "ffmpeg: не смог открыть файл\n");
        return -1;
    }
 
    // После открытия, можем распечатать информацию о видеофайле (iformat = название формата; 
    // duration = длительность). Но так как у меня подключена камера мне написал: Duration: N/A, 
    // start: 0.000000, bitrate: N/A
    printf("Format %s, duration %lld us", AVfc->iformat->long_name, AVfc->duration);
 
    ERRORS = avformat_find_stream_info(AVfc, NULL);
    if (ERRORS < 0) {
        fprintf(stderr, "ffmpeg: Unable to find stream info\n");
        return -1;
    }
 
    int CountStream;
 
    // Узнаем кол-во потоков. 
    CountStream = AVfc->nb_streams;
 
    // Поищим кодек. 
    int video_stream;
    for (video_stream = 0; video_stream < AVfc->nb_streams; ++video_stream) {
        if (AVfc->streams[video_stream]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
 
            // В ДАННОЙ ТОЧКЕ ЧАСТО ОШИБКА ПРИ ИСПОЛЬЗОВАНИЕ IP КАМЕРЫ.
            break; // Тут вылетает ошибка, что типа не обнаружен поток, надо заново пустить. 
        }
    }
 
    if (video_stream == AVfc->nb_streams) {
        fprintf(stderr, "ffmpeg: Unable to find video stream\n");
        return -1;
    }
 
    // Тут определяем вид кодека, для моей камеры он равен как AV_CODEC_ID_HEVC (Это то в чем вещает моя камера)
    codec = avcodec_find_decoder(AVfc->streams[video_stream]->codecpar->codec_id);
 
    // Функции для запроса возможностей libavcodec,
    AVCodecContext_ = avcodec_alloc_context3(codec);
    if (!AVCodecContext_) {
        fprintf(stderr, "Не удалось выделить контекст видеокодека, т.к. он не поддрерживается\n");
        exit(1);
    }
 
    // Эта функция используется для инициализации 
    // AVCodecContext видео и аудио кодека. Объявление avcodec_open2() находится в libavcodecavcodec.h
    // Открываем кодек. 
 
    ERRORS = avcodec_open2(AVCodecContext_, codec, NULL);
    if (ERRORS < 0) {
        fprintf(stderr, "ffmpeg: Не удается открыть кодек\n");
        return -1;
    }
 
    // Это для обработки звука - задел.
    //swr_alloc_set_opts()
    // swr_init(); 
 
    // Вывести всю информацию о видеофайле. 
    av_dump_format(AVfc, 0, argv[1], 0);
 
    //=========================================================================================
    // Далее, получаем фреймы..до этого мы только получили всю инфомрацию о входящем видео.
    //=========================================================================================
 
    // Теперь мы собираемся прочитать пакеты из потока и декодировать их в кадры, но сначала 
    // нам нужно выделить память для обоих компонентов(AVPacket и AVFrame).
    frame = av_frame_alloc();
 
    f = fopen(OutFileNameDE, "w");
    if (!f) {
        fprintf(stderr, "Could not open %s\n", OutFileNameDE);
        exit(1);
    }
 
    if (!frame) {
        fprintf(stderr, "Не удается выделить память для видеокадра\n");
        exit(1);
    }
    // Выделяем память для пакета 
    pkt = av_packet_alloc();
 
    // Чтение данных если они есть. 
    while (av_read_frame(AVfc, pkt) >= 0) {
        // Это пакет из видеопотока? Потомучто есть еще звуковая дорожка.
        if (pkt->stream_index == video_stream)
        {
            int ret;
 
            // Обновление 1, типа сохраняем декодированое видео в файл. 
            decode(AVCodecContext_, frame, pkt, f);
        }
        av_packet_unref(pkt);
    }
 
 
    if (codec)
        av_free(&codec);
    if (parser)
        av_free(&parser);
    if (AVCodecContext_)
        av_free(&AVCodecContext_);
    if (AVCodecParametr_)
        av_free(&AVCodecParametr_);
    if (f)
        av_free(&f);
    if (AVfc)
        av_free(&inbuf);
 
    if (pkt)
        av_free(&pkt);
 
}