Merge commit '749a89d1b8bb73b4d4f14c48f33259a1300c1761'
[ffmpeg.git] / doc / examples / transcode_aac.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 /**
20  * @file
21  * simple audio converter
22  *
23  * @example transcode_aac.c
24  * Convert an input audio file to AAC in an MP4 container using FFmpeg.
25  * @author Andreas Unterweger (dustsigns@gmail.com)
26  */
27
28 #include <stdio.h>
29
30 #include "libavformat/avformat.h"
31 #include "libavformat/avio.h"
32
33 #include "libavcodec/avcodec.h"
34
35 #include "libavutil/audio_fifo.h"
36 #include "libavutil/avassert.h"
37 #include "libavutil/avstring.h"
38 #include "libavutil/frame.h"
39 #include "libavutil/opt.h"
40
41 #include "libswresample/swresample.h"
42
43 /** The output bit rate in kbit/s */
44 #define OUTPUT_BIT_RATE 96000
45 /** The number of output channels */
46 #define OUTPUT_CHANNELS 2
47
48 /**
49  * Convert an error code into a text message.
50  * @param error Error code to be converted
51  * @return Corresponding error text (not thread-safe)
52  */
53 static const char *get_error_text(const int error)
54 {
55     static char error_buffer[255];
56     av_strerror(error, error_buffer, sizeof(error_buffer));
57     return error_buffer;
58 }
59
60 /** Open an input file and the required decoder. */
61 static int open_input_file(const char *filename,
62                            AVFormatContext **input_format_context,
63                            AVCodecContext **input_codec_context)
64 {
65     AVCodec *input_codec;
66     int error;
67
68     /** Open the input file to read from it. */
69     if ((error = avformat_open_input(input_format_context, filename, NULL,
70                                      NULL)) < 0) {
71         fprintf(stderr, "Could not open input file '%s' (error '%s')\n",
72                 filename, get_error_text(error));
73         *input_format_context = NULL;
74         return error;
75     }
76
77     /** Get information on the input file (number of streams etc.). */
78     if ((error = avformat_find_stream_info(*input_format_context, NULL)) < 0) {
79         fprintf(stderr, "Could not open find stream info (error '%s')\n",
80                 get_error_text(error));
81         avformat_close_input(input_format_context);
82         return error;
83     }
84
85     /** Make sure that there is only one stream in the input file. */
86     if ((*input_format_context)->nb_streams != 1) {
87         fprintf(stderr, "Expected one audio input stream, but found %d\n",
88                 (*input_format_context)->nb_streams);
89         avformat_close_input(input_format_context);
90         return AVERROR_EXIT;
91     }
92
93     /** Find a decoder for the audio stream. */
94     if (!(input_codec = avcodec_find_decoder((*input_format_context)->streams[0]->codec->codec_id))) {
95         fprintf(stderr, "Could not find input codec\n");
96         avformat_close_input(input_format_context);
97         return AVERROR_EXIT;
98     }
99
100     /** Open the decoder for the audio stream to use it later. */
101     if ((error = avcodec_open2((*input_format_context)->streams[0]->codec,
102                                input_codec, NULL)) < 0) {
103         fprintf(stderr, "Could not open input codec (error '%s')\n",
104                 get_error_text(error));
105         avformat_close_input(input_format_context);
106         return error;
107     }
108
109     /** Save the decoder context for easier access later. */
110     *input_codec_context = (*input_format_context)->streams[0]->codec;
111
112     return 0;
113 }
114
115 /**
116  * Open an output file and the required encoder.
117  * Also set some basic encoder parameters.
118  * Some of these parameters are based on the input file's parameters.
119  */
120 static int open_output_file(const char *filename,
121                             AVCodecContext *input_codec_context,
122                             AVFormatContext **output_format_context,
123                             AVCodecContext **output_codec_context)
124 {
125     AVIOContext *output_io_context = NULL;
126     AVStream *stream               = NULL;
127     AVCodec *output_codec          = NULL;
128     int error;
129
130     /** Open the output file to write to it. */
131     if ((error = avio_open(&output_io_context, filename,
132                            AVIO_FLAG_WRITE)) < 0) {
133         fprintf(stderr, "Could not open output file '%s' (error '%s')\n",
134                 filename, get_error_text(error));
135         return error;
136     }
137
138     /** Create a new format context for the output container format. */
139     if (!(*output_format_context = avformat_alloc_context())) {
140         fprintf(stderr, "Could not allocate output format context\n");
141         return AVERROR(ENOMEM);
142     }
143
144     /** Associate the output file (pointer) with the container format context. */
145     (*output_format_context)->pb = output_io_context;
146
147     /** Guess the desired container format based on the file extension. */
148     if (!((*output_format_context)->oformat = av_guess_format(NULL, filename,
149                                                               NULL))) {
150         fprintf(stderr, "Could not find output file format\n");
151         goto cleanup;
152     }
153
154     av_strlcpy((*output_format_context)->filename, filename,
155                sizeof((*output_format_context)->filename));
156
157     /** Find the encoder to be used by its name. */
158     if (!(output_codec = avcodec_find_encoder(AV_CODEC_ID_AAC))) {
159         fprintf(stderr, "Could not find an AAC encoder.\n");
160         goto cleanup;
161     }
162
163     /** Create a new audio stream in the output file container. */
164     if (!(stream = avformat_new_stream(*output_format_context, output_codec))) {
165         fprintf(stderr, "Could not create new stream\n");
166         error = AVERROR(ENOMEM);
167         goto cleanup;
168     }
169
170     /** Save the encoder context for easiert access later. */
171     *output_codec_context = stream->codec;
172
173     /**
174      * Set the basic encoder parameters.
175      * The input file's sample rate is used to avoid a sample rate conversion.
176      */
177     (*output_codec_context)->channels       = OUTPUT_CHANNELS;
178     (*output_codec_context)->channel_layout = av_get_default_channel_layout(OUTPUT_CHANNELS);
179     (*output_codec_context)->sample_rate    = input_codec_context->sample_rate;
180     (*output_codec_context)->sample_fmt     = output_codec->sample_fmts[0];
181     (*output_codec_context)->bit_rate       = OUTPUT_BIT_RATE;
182
183     /** Allow the use of the experimental AAC encoder */
184     (*output_codec_context)->strict_std_compliance = FF_COMPLIANCE_EXPERIMENTAL;
185
186     /**
187      * Some container formats (like MP4) require global headers to be present
188      * Mark the encoder so that it behaves accordingly.
189      */
190     if ((*output_format_context)->oformat->flags & AVFMT_GLOBALHEADER)
191         (*output_codec_context)->flags |= CODEC_FLAG_GLOBAL_HEADER;
192
193     /** Open the encoder for the audio stream to use it later. */
194     if ((error = avcodec_open2(*output_codec_context, output_codec, NULL)) < 0) {
195         fprintf(stderr, "Could not open output codec (error '%s')\n",
196                 get_error_text(error));
197         goto cleanup;
198     }
199
200     return 0;
201
202 cleanup:
203     avio_closep(&(*output_format_context)->pb);
204     avformat_free_context(*output_format_context);
205     *output_format_context = NULL;
206     return error < 0 ? error : AVERROR_EXIT;
207 }
208
209 /** Initialize one data packet for reading or writing. */
210 static void init_packet(AVPacket *packet)
211 {
212     av_init_packet(packet);
213     /** Set the packet data and size so that it is recognized as being empty. */
214     packet->data = NULL;
215     packet->size = 0;
216 }
217
218 /** Initialize one audio frame for reading from the input file */
219 static int init_input_frame(AVFrame **frame)
220 {
221     if (!(*frame = av_frame_alloc())) {
222         fprintf(stderr, "Could not allocate input frame\n");
223         return AVERROR(ENOMEM);
224     }
225     return 0;
226 }
227
228 /**
229  * Initialize the audio resampler based on the input and output codec settings.
230  * If the input and output sample formats differ, a conversion is required
231  * libswresample takes care of this, but requires initialization.
232  */
233 static int init_resampler(AVCodecContext *input_codec_context,
234                           AVCodecContext *output_codec_context,
235                           SwrContext **resample_context)
236 {
237         int error;
238
239         /**
240          * Create a resampler context for the conversion.
241          * Set the conversion parameters.
242          * Default channel layouts based on the number of channels
243          * are assumed for simplicity (they are sometimes not detected
244          * properly by the demuxer and/or decoder).
245          */
246         *resample_context = swr_alloc_set_opts(NULL,
247                                               av_get_default_channel_layout(output_codec_context->channels),
248                                               output_codec_context->sample_fmt,
249                                               output_codec_context->sample_rate,
250                                               av_get_default_channel_layout(input_codec_context->channels),
251                                               input_codec_context->sample_fmt,
252                                               input_codec_context->sample_rate,
253                                               0, NULL);
254         if (!*resample_context) {
255             fprintf(stderr, "Could not allocate resample context\n");
256             return AVERROR(ENOMEM);
257         }
258         /**
259         * Perform a sanity check so that the number of converted samples is
260         * not greater than the number of samples to be converted.
261         * If the sample rates differ, this case has to be handled differently
262         */
263         av_assert0(output_codec_context->sample_rate == input_codec_context->sample_rate);
264
265         /** Open the resampler with the specified parameters. */
266         if ((error = swr_init(*resample_context)) < 0) {
267             fprintf(stderr, "Could not open resample context\n");
268             swr_free(resample_context);
269             return error;
270         }
271     return 0;
272 }
273
274 /** Initialize a FIFO buffer for the audio samples to be encoded. */
275 static int init_fifo(AVAudioFifo **fifo, AVCodecContext *output_codec_context)
276 {
277     /** Create the FIFO buffer based on the specified output sample format. */
278     if (!(*fifo = av_audio_fifo_alloc(output_codec_context->sample_fmt,
279                                       output_codec_context->channels, 1))) {
280         fprintf(stderr, "Could not allocate FIFO\n");
281         return AVERROR(ENOMEM);
282     }
283     return 0;
284 }
285
286 /** Write the header of the output file container. */
287 static int write_output_file_header(AVFormatContext *output_format_context)
288 {
289     int error;
290     if ((error = avformat_write_header(output_format_context, NULL)) < 0) {
291         fprintf(stderr, "Could not write output file header (error '%s')\n",
292                 get_error_text(error));
293         return error;
294     }
295     return 0;
296 }
297
298 /** Decode one audio frame from the input file. */
299 static int decode_audio_frame(AVFrame *frame,
300                               AVFormatContext *input_format_context,
301                               AVCodecContext *input_codec_context,
302                               int *data_present, int *finished)
303 {
304     /** Packet used for temporary storage. */
305     AVPacket input_packet;
306     int error;
307     init_packet(&input_packet);
308
309     /** Read one audio frame from the input file into a temporary packet. */
310     if ((error = av_read_frame(input_format_context, &input_packet)) < 0) {
311         /** If we are at the end of the file, flush the decoder below. */
312         if (error == AVERROR_EOF)
313             *finished = 1;
314         else {
315             fprintf(stderr, "Could not read frame (error '%s')\n",
316                     get_error_text(error));
317             return error;
318         }
319     }
320
321     /**
322      * Decode the audio frame stored in the temporary packet.
323      * The input audio stream decoder is used to do this.
324      * If we are at the end of the file, pass an empty packet to the decoder
325      * to flush it.
326      */
327     if ((error = avcodec_decode_audio4(input_codec_context, frame,
328                                        data_present, &input_packet)) < 0) {
329         fprintf(stderr, "Could not decode frame (error '%s')\n",
330                 get_error_text(error));
331         av_free_packet(&input_packet);
332         return error;
333     }
334
335     /**
336      * If the decoder has not been flushed completely, we are not finished,
337      * so that this function has to be called again.
338      */
339     if (*finished && *data_present)
340         *finished = 0;
341     av_free_packet(&input_packet);
342     return 0;
343 }
344
345 /**
346  * Initialize a temporary storage for the specified number of audio samples.
347  * The conversion requires temporary storage due to the different format.
348  * The number of audio samples to be allocated is specified in frame_size.
349  */
350 static int init_converted_samples(uint8_t ***converted_input_samples,
351                                   AVCodecContext *output_codec_context,
352                                   int frame_size)
353 {
354     int error;
355
356     /**
357      * Allocate as many pointers as there are audio channels.
358      * Each pointer will later point to the audio samples of the corresponding
359      * channels (although it may be NULL for interleaved formats).
360      */
361     if (!(*converted_input_samples = calloc(output_codec_context->channels,
362                                             sizeof(**converted_input_samples)))) {
363         fprintf(stderr, "Could not allocate converted input sample pointers\n");
364         return AVERROR(ENOMEM);
365     }
366
367     /**
368      * Allocate memory for the samples of all channels in one consecutive
369      * block for convenience.
370      */
371     if ((error = av_samples_alloc(*converted_input_samples, NULL,
372                                   output_codec_context->channels,
373                                   frame_size,
374                                   output_codec_context->sample_fmt, 0)) < 0) {
375         fprintf(stderr,
376                 "Could not allocate converted input samples (error '%s')\n",
377                 get_error_text(error));
378         av_freep(&(*converted_input_samples)[0]);
379         free(*converted_input_samples);
380         return error;
381     }
382     return 0;
383 }
384
385 /**
386  * Convert the input audio samples into the output sample format.
387  * The conversion happens on a per-frame basis, the size of which is specified
388  * by frame_size.
389  */
390 static int convert_samples(const uint8_t **input_data,
391                            uint8_t **converted_data, const int frame_size,
392                            SwrContext *resample_context)
393 {
394     int error;
395
396     /** Convert the samples using the resampler. */
397     if ((error = swr_convert(resample_context,
398                              converted_data, frame_size,
399                              input_data    , frame_size)) < 0) {
400         fprintf(stderr, "Could not convert input samples (error '%s')\n",
401                 get_error_text(error));
402         return error;
403     }
404
405     return 0;
406 }
407
408 /** Add converted input audio samples to the FIFO buffer for later processing. */
409 static int add_samples_to_fifo(AVAudioFifo *fifo,
410                                uint8_t **converted_input_samples,
411                                const int frame_size)
412 {
413     int error;
414
415     /**
416      * Make the FIFO as large as it needs to be to hold both,
417      * the old and the new samples.
418      */
419     if ((error = av_audio_fifo_realloc(fifo, av_audio_fifo_size(fifo) + frame_size)) < 0) {
420         fprintf(stderr, "Could not reallocate FIFO\n");
421         return error;
422     }
423
424     /** Store the new samples in the FIFO buffer. */
425     if (av_audio_fifo_write(fifo, (void **)converted_input_samples,
426                             frame_size) < frame_size) {
427         fprintf(stderr, "Could not write data to FIFO\n");
428         return AVERROR_EXIT;
429     }
430     return 0;
431 }
432
433 /**
434  * Read one audio frame from the input file, decodes, converts and stores
435  * it in the FIFO buffer.
436  */
437 static int read_decode_convert_and_store(AVAudioFifo *fifo,
438                                          AVFormatContext *input_format_context,
439                                          AVCodecContext *input_codec_context,
440                                          AVCodecContext *output_codec_context,
441                                          SwrContext *resampler_context,
442                                          int *finished)
443 {
444     /** Temporary storage of the input samples of the frame read from the file. */
445     AVFrame *input_frame = NULL;
446     /** Temporary storage for the converted input samples. */
447     uint8_t **converted_input_samples = NULL;
448     int data_present;
449     int ret = AVERROR_EXIT;
450
451     /** Initialize temporary storage for one input frame. */
452     if (init_input_frame(&input_frame))
453         goto cleanup;
454     /** Decode one frame worth of audio samples. */
455     if (decode_audio_frame(input_frame, input_format_context,
456                            input_codec_context, &data_present, finished))
457         goto cleanup;
458     /**
459      * If we are at the end of the file and there are no more samples
460      * in the decoder which are delayed, we are actually finished.
461      * This must not be treated as an error.
462      */
463     if (*finished && !data_present) {
464         ret = 0;
465         goto cleanup;
466     }
467     /** If there is decoded data, convert and store it */
468     if (data_present) {
469         /** Initialize the temporary storage for the converted input samples. */
470         if (init_converted_samples(&converted_input_samples, output_codec_context,
471                                    input_frame->nb_samples))
472             goto cleanup;
473
474         /**
475          * Convert the input samples to the desired output sample format.
476          * This requires a temporary storage provided by converted_input_samples.
477          */
478         if (convert_samples((const uint8_t**)input_frame->extended_data, converted_input_samples,
479                             input_frame->nb_samples, resampler_context))
480             goto cleanup;
481
482         /** Add the converted input samples to the FIFO buffer for later processing. */
483         if (add_samples_to_fifo(fifo, converted_input_samples,
484                                 input_frame->nb_samples))
485             goto cleanup;
486         ret = 0;
487     }
488     ret = 0;
489
490 cleanup:
491     if (converted_input_samples) {
492         av_freep(&converted_input_samples[0]);
493         free(converted_input_samples);
494     }
495     av_frame_free(&input_frame);
496
497     return ret;
498 }
499
500 /**
501  * Initialize one input frame for writing to the output file.
502  * The frame will be exactly frame_size samples large.
503  */
504 static int init_output_frame(AVFrame **frame,
505                              AVCodecContext *output_codec_context,
506                              int frame_size)
507 {
508     int error;
509
510     /** Create a new frame to store the audio samples. */
511     if (!(*frame = av_frame_alloc())) {
512         fprintf(stderr, "Could not allocate output frame\n");
513         return AVERROR_EXIT;
514     }
515
516     /**
517      * Set the frame's parameters, especially its size and format.
518      * av_frame_get_buffer needs this to allocate memory for the
519      * audio samples of the frame.
520      * Default channel layouts based on the number of channels
521      * are assumed for simplicity.
522      */
523     (*frame)->nb_samples     = frame_size;
524     (*frame)->channel_layout = output_codec_context->channel_layout;
525     (*frame)->format         = output_codec_context->sample_fmt;
526     (*frame)->sample_rate    = output_codec_context->sample_rate;
527
528     /**
529      * Allocate the samples of the created frame. This call will make
530      * sure that the audio frame can hold as many samples as specified.
531      */
532     if ((error = av_frame_get_buffer(*frame, 0)) < 0) {
533         fprintf(stderr, "Could allocate output frame samples (error '%s')\n",
534                 get_error_text(error));
535         av_frame_free(frame);
536         return error;
537     }
538
539     return 0;
540 }
541
542 /** Encode one frame worth of audio to the output file. */
543 static int encode_audio_frame(AVFrame *frame,
544                               AVFormatContext *output_format_context,
545                               AVCodecContext *output_codec_context,
546                               int *data_present)
547 {
548     /** Packet used for temporary storage. */
549     AVPacket output_packet;
550     int error;
551     init_packet(&output_packet);
552
553     /**
554      * Encode the audio frame and store it in the temporary packet.
555      * The output audio stream encoder is used to do this.
556      */
557     if ((error = avcodec_encode_audio2(output_codec_context, &output_packet,
558                                        frame, data_present)) < 0) {
559         fprintf(stderr, "Could not encode frame (error '%s')\n",
560                 get_error_text(error));
561         av_free_packet(&output_packet);
562         return error;
563     }
564
565     /** Write one audio frame from the temporary packet to the output file. */
566     if (*data_present) {
567         if ((error = av_write_frame(output_format_context, &output_packet)) < 0) {
568             fprintf(stderr, "Could not write frame (error '%s')\n",
569                     get_error_text(error));
570             av_free_packet(&output_packet);
571             return error;
572         }
573
574         av_free_packet(&output_packet);
575     }
576
577     return 0;
578 }
579
580 /**
581  * Load one audio frame from the FIFO buffer, encode and write it to the
582  * output file.
583  */
584 static int load_encode_and_write(AVAudioFifo *fifo,
585                                  AVFormatContext *output_format_context,
586                                  AVCodecContext *output_codec_context)
587 {
588     /** Temporary storage of the output samples of the frame written to the file. */
589     AVFrame *output_frame;
590     /**
591      * Use the maximum number of possible samples per frame.
592      * If there is less than the maximum possible frame size in the FIFO
593      * buffer use this number. Otherwise, use the maximum possible frame size
594      */
595     const int frame_size = FFMIN(av_audio_fifo_size(fifo),
596                                  output_codec_context->frame_size);
597     int data_written;
598
599     /** Initialize temporary storage for one output frame. */
600     if (init_output_frame(&output_frame, output_codec_context, frame_size))
601         return AVERROR_EXIT;
602
603     /**
604      * Read as many samples from the FIFO buffer as required to fill the frame.
605      * The samples are stored in the frame temporarily.
606      */
607     if (av_audio_fifo_read(fifo, (void **)output_frame->data, frame_size) < frame_size) {
608         fprintf(stderr, "Could not read data from FIFO\n");
609         av_frame_free(&output_frame);
610         return AVERROR_EXIT;
611     }
612
613     /** Encode one frame worth of audio samples. */
614     if (encode_audio_frame(output_frame, output_format_context,
615                            output_codec_context, &data_written)) {
616         av_frame_free(&output_frame);
617         return AVERROR_EXIT;
618     }
619     av_frame_free(&output_frame);
620     return 0;
621 }
622
623 /** Write the trailer of the output file container. */
624 static int write_output_file_trailer(AVFormatContext *output_format_context)
625 {
626     int error;
627     if ((error = av_write_trailer(output_format_context)) < 0) {
628         fprintf(stderr, "Could not write output file trailer (error '%s')\n",
629                 get_error_text(error));
630         return error;
631     }
632     return 0;
633 }
634
635 /** Convert an audio file to an AAC file in an MP4 container. */
636 int main(int argc, char **argv)
637 {
638     AVFormatContext *input_format_context = NULL, *output_format_context = NULL;
639     AVCodecContext *input_codec_context = NULL, *output_codec_context = NULL;
640     SwrContext *resample_context = NULL;
641     AVAudioFifo *fifo = NULL;
642     int ret = AVERROR_EXIT;
643
644     if (argc < 3) {
645         fprintf(stderr, "Usage: %s <input file> <output file>\n", argv[0]);
646         exit(1);
647     }
648
649     /** Register all codecs and formats so that they can be used. */
650     av_register_all();
651     /** Open the input file for reading. */
652     if (open_input_file(argv[1], &input_format_context,
653                         &input_codec_context))
654         goto cleanup;
655     /** Open the output file for writing. */
656     if (open_output_file(argv[2], input_codec_context,
657                          &output_format_context, &output_codec_context))
658         goto cleanup;
659     /** Initialize the resampler to be able to convert audio sample formats. */
660     if (init_resampler(input_codec_context, output_codec_context,
661                        &resample_context))
662         goto cleanup;
663     /** Initialize the FIFO buffer to store audio samples to be encoded. */
664     if (init_fifo(&fifo, output_codec_context))
665         goto cleanup;
666     /** Write the header of the output file container. */
667     if (write_output_file_header(output_format_context))
668         goto cleanup;
669
670     /**
671      * Loop as long as we have input samples to read or output samples
672      * to write; abort as soon as we have neither.
673      */
674     while (1) {
675         /** Use the encoder's desired frame size for processing. */
676         const int output_frame_size = output_codec_context->frame_size;
677         int finished                = 0;
678
679         /**
680          * Make sure that there is one frame worth of samples in the FIFO
681          * buffer so that the encoder can do its work.
682          * Since the decoder's and the encoder's frame size may differ, we
683          * need to FIFO buffer to store as many frames worth of input samples
684          * that they make up at least one frame worth of output samples.
685          */
686         while (av_audio_fifo_size(fifo) < output_frame_size) {
687             /**
688              * Decode one frame worth of audio samples, convert it to the
689              * output sample format and put it into the FIFO buffer.
690              */
691             if (read_decode_convert_and_store(fifo, input_format_context,
692                                               input_codec_context,
693                                               output_codec_context,
694                                               resample_context, &finished))
695                 goto cleanup;
696
697             /**
698              * If we are at the end of the input file, we continue
699              * encoding the remaining audio samples to the output file.
700              */
701             if (finished)
702                 break;
703         }
704
705         /**
706          * If we have enough samples for the encoder, we encode them.
707          * At the end of the file, we pass the remaining samples to
708          * the encoder.
709          */
710         while (av_audio_fifo_size(fifo) >= output_frame_size ||
711                (finished && av_audio_fifo_size(fifo) > 0))
712             /**
713              * Take one frame worth of audio samples from the FIFO buffer,
714              * encode it and write it to the output file.
715              */
716             if (load_encode_and_write(fifo, output_format_context,
717                                       output_codec_context))
718                 goto cleanup;
719
720         /**
721          * If we are at the end of the input file and have encoded
722          * all remaining samples, we can exit this loop and finish.
723          */
724         if (finished) {
725             int data_written;
726             /** Flush the encoder as it may have delayed frames. */
727             do {
728                 if (encode_audio_frame(NULL, output_format_context,
729                                        output_codec_context, &data_written))
730                     goto cleanup;
731             } while (data_written);
732             break;
733         }
734     }
735
736     /** Write the trailer of the output file container. */
737     if (write_output_file_trailer(output_format_context))
738         goto cleanup;
739     ret = 0;
740
741 cleanup:
742     if (fifo)
743         av_audio_fifo_free(fifo);
744     swr_free(&resample_context);
745     if (output_codec_context)
746         avcodec_close(output_codec_context);
747     if (output_format_context) {
748         avio_closep(&output_format_context->pb);
749         avformat_free_context(output_format_context);
750     }
751     if (input_codec_context)
752         avcodec_close(input_codec_context);
753     if (input_format_context)
754         avformat_close_input(&input_format_context);
755
756     return ret;
757 }