Ticket #8330: decode_test.c

#include <stdio.h>
#include <stdlib.h>

#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
#include <libavutil/avutil.h>
#include <libavutil/mastering_display_metadata.h>

static void decode(const char *url, const char *name) {
    printf("Decode by %s\n", name);

    AVFormatContext *fmt_ctx = NULL;
    int ret;

    if (avformat_open_input(&fmt_ctx, url, NULL, NULL) < 0) {
        fprintf(stderr, "avformat_open_input failed\n");
        exit(1);
    }

    if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
        fprintf(stderr, "avformat_find_stream_info failed\n");
        exit(1);
    }

    AVCodec *codec = avcodec_find_decoder_by_name(name);
    if (!codec) {
        fprintf(stderr, "avcodec_find_decoder_by_name failed\n");
        exit(1);
    }

    AVCodecContext *avctx = avcodec_alloc_context3(codec);
    if (!avctx) {
        fprintf(stderr, "avcodec_alloc_context3 failed\n");
        exit(1);
    }

    if (avcodec_parameters_to_context(avctx, fmt_ctx->streams[0]->codecpar) < 0) {
        fprintf(stderr, "avcodec_parameters_to_context failed\n");
        exit(1);
    }

    avctx->pkt_timebase = fmt_ctx->streams[0]->time_base;

    if (avcodec_open2(avctx, codec, NULL) < 0) {
        fprintf(stderr, "avcodec_open2 failed\n");
        exit(1);
    }

    AVPacket *pkt = av_packet_alloc();
    if (!pkt) {
        fprintf(stderr, "av_packet_alloc failed\n");
        exit(1);
    }

    AVFrame *frame = av_frame_alloc();
    if (!frame) {
        fprintf(stderr, "av_frame_alloc failed\n");
        exit(1);
    }

    while (av_read_frame(fmt_ctx, pkt) >= 0) {
        ret = avcodec_send_packet(avctx, pkt);
        if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) {
            fprintf(stderr, "avcodec_send_packet failed\n");
            exit(1);
        }

        ret = avcodec_receive_frame(avctx, frame);
        if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) {
            fprintf(stderr, "avcodec_receive_frame failed\n");
            exit(1);
        }

        if (ret >= 0) {
            printf("Picture type: %c\n", av_get_picture_type_char(frame->pict_type));

            const AVFrameSideData *side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_MASTERING_DISPLAY_METADATA);
            if (side_data) {
                const AVMasteringDisplayMetadata *mastering_display = (const AVMasteringDisplayMetadata *)side_data->data;
                if (mastering_display->has_luminance) {
                    printf("Min luminance: %f\n", av_q2d(mastering_display->min_luminance));
                    printf("Max luminance: %f\n", av_q2d(mastering_display->max_luminance));
                }
            } else {
                printf("No side data in the frame\n");
            }

            printf("\n");
            break;
        }

        av_packet_unref(pkt);
    }

    avformat_close_input(&fmt_ctx);
    avcodec_free_context(&avctx);
    av_packet_free(&pkt);
    av_frame_free(&frame);
}

int main(int argc, char **argv) {
    decode(argv[1], "hevc");
    decode(argv[1], "hevc_cuvid");

    return 0;
}