ffmpegのC言語を用いたSDKによるデスクトップ画像の収集

#include 

extern "C"{
#include 
#include 
#include 
#include 
#include 
#include 
#include 
}
using namespace std;

static AVFrame *alloc_picture(enum AVPixelFormat pix_fmt, int width, int height) {
    AVFrame *picture;
    int ret;

    picture = av_frame_alloc();
    if (!picture)
        return NULL;

    picture->format = pix_fmt;
    picture->width = width;
    picture->height = height;

    ret = av_frame_get_buffer(picture, 0);
    if (ret < 0) {
        fprintf(stderr, "Could not allocate frame data.
");
        return NULL;
    }


    return picture;


}

//      
static int test1(){
    //          
    AVFormatContext *formatCtx = avformat_alloc_context();
    AVInputFormat *ifmt = av_find_input_format("avfoundation");

    //       
    AVDictionary *options = NULL;
//    av_dict_set(&options, "video_size". "1920*1080",0);
    av_dict_set(&options, "framerate", "15", 0);

    //      
    if (avformat_open_input(&formatCtx, "1", ifmt, &options) != 0){
        printf("open input device fail
");
        return -1;
    }
    //              
    if (avformat_find_stream_info(formatCtx, NULL)<0)
    {
        printf("        !
");
        return -1;
    }
    if(formatCtx->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_VIDEO)
    {
        printf("       ！");
    }
    //      
    AVCodec *codec = avcodec_find_decoder(formatCtx->streams[0]->codecpar->codec_id);
    if (codec ==NULL){
        printf("codec not found.
");
        return -1;
    }

    AVCodecContext *ctx = avcodec_alloc_context3(codec);
    //     
    avcodec_parameters_to_context(ctx, formatCtx->streams[0]->codecpar);
    if (avcodec_open2(ctx, codec, NULL)<0){
        printf("       ");
        return -1;
    }

    //      
    AVFrame *frame = alloc_picture(
            (AVPixelFormat)formatCtx->streams[0]->codecpar->format,
            formatCtx->streams[0]->codecpar->width,
            formatCtx->streams[0]->codecpar->height
            );
    //      
    SwsContext *img_convert_ctx = sws_getContext(
            formatCtx->streams[0]->codecpar->width,
            formatCtx->streams[0]->codecpar->height,
            (AVPixelFormat)formatCtx->streams[0]->codecpar->format,
            formatCtx->streams[0]->codecpar->width,
            formatCtx->streams[0]->codecpar->height,
            AV_PIX_FMT_BGR24,
            SWS_BICUBIC,
            NULL,
            NULL,
            NULL
    );
    AVFrame *bgrFrame = alloc_picture(
            AV_PIX_FMT_BGR24,
            formatCtx->streams[0]->codecpar->width,
            formatCtx->streams[0]->codecpar->height
            );

    while (true){


    AVPacket packet = {0};
    av_init_packet(&packet);
    if(av_read_frame(formatCtx, &packet) >= 0){

        //        
        avcodec_send_packet(ctx, &packet);
        if (avcodec_receive_frame(ctx, frame)<0)
        {
            printf("Decode Error
");
        } else{
            printf("     
");
            sws_scale(img_convert_ctx,
                    (const unsigned char * const *)frame->data,
                    frame->linesize,
                    0,
                    frame->height,
                    bgrFrame->data,
                    bgrFrame->linesize
                    );

        }
    }
    av_packet_unref(&packet);
    }

    //     
    av_frame_free(&frame);
    av_frame_free(&bgrFrame);

    avcodec_free_context(&ctx);
    avformat_close_input(&formatCtx);

    return 0;


}

int main(){
    int ver = avcodec_version();

    avdevice_register_all();

    test1();
}