Ticket #8439: ffmpeg_ticket_8439.c

#include <stdio.h>
#include <stddef.h>
#include <stdint.h>
#include <string.h>
#include <errno.h>
#include <libavcodec/avcodec.h>
#include <libavutil/samplefmt.h>
#include <libavutil/channel_layout.h>
#include <libavutil/avutil.h>
#include <libavutil/frame.h>

int main()
{
    AVCodec *codec = NULL;
    AVCodecContext *ctx = NULL;
    AVFrame frame = {};
    AVPacket packet = {};
    size_t sample_size = 0;
    int64_t in_pts = 0;
    int res, final_result = -1;

    /* First, see if ffmpeg was built with libfdk-aac. It provides better quality. */
    codec = avcodec_find_encoder_by_name("libfdk_aac");
    if (!codec)
    {
        puts("Failed to find encoder");
        return -1;
    }

    ctx = avcodec_alloc_context3(codec);
    if (!ctx)
    {
        puts("Failed to allocate encoder context");
        return -1;
    }

    ctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;

    ctx->sample_fmt = AV_SAMPLE_FMT_S16;
    ctx->sample_rate = 48000;
    ctx->channels = 1;
    ctx->channel_layout = AV_CH_LAYOUT_MONO;

    ctx->time_base.num = 1;
    ctx->time_base.den = ctx->sample_rate;

    ctx->bit_rate = 320000;

    ctx->profile = FF_PROFILE_AAC_LOW;

    res = avcodec_open2(ctx, codec, (AVDictionary **)NULL);
    if (res < 0)
    {
        puts("Failed to open encoder context");
        goto finish;
    }

    av_frame_unref(&frame);
    av_init_packet(&packet);

    frame.format = ctx->sample_fmt;
    frame.sample_rate = ctx->sample_rate;
    frame.channel_layout = ctx->channel_layout;
    frame.channels = ctx->channels;

    sample_size = av_get_bytes_per_sample(ctx->sample_fmt);

    frame.nb_samples = ctx->frame_size;

    res = av_frame_get_buffer(&frame, 0);
    if (res < 0)
    {
        puts("Failed to allocate input frame");
        goto finish;
    }

    memset(frame.data[0], 0, sample_size * frame.nb_samples);

    for (unsigned int i = 0; i < 5; ++i)
    {
        frame.pts = in_pts;
        printf("Input pts: %lld", (long long)in_pts);

        in_pts += frame.nb_samples;

        res = avcodec_send_frame(ctx, &frame);
        if (res < 0)
        {
            puts("Failed to send input frame to the encoder");
            goto finish;
        }

        res = avcodec_receive_packet(ctx, &packet);
        if (res < 0)
        {
            if (res != AVERROR(EAGAIN))
            {
                puts("Failed to receive encoded frame from the encoder");
                goto finish;
            }

            printf(", output pts: -");
        }
        else
        {
            printf(", output pts: %lld", (long long)packet.pts);
        }

        printf("\n");
    }

    final_result = 0;

finish:
    av_frame_unref(&frame);
    av_packet_unref(&packet);
    avcodec_free_context(&ctx);

    return final_result;
}