[deb_ffmpeg.git] / ffmpeg / doc / examples / demuxing_decoding.c

/*
 * Copyright (c) 2012 Stefano Sabatini
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

/**
 * @file
 * Demuxing and decoding example.
 *
 * Show how to use the libavformat and libavcodec API to demux and
 * decode audio and video data.
 * @example demuxing_decoding.c
 */

#include <libavutil/imgutils.h>
#include <libavutil/samplefmt.h>
#include <libavutil/timestamp.h>
#include <libavformat/avformat.h>

static AVFormatContext *fmt_ctx = NULL;
static AVCodecContext *video_dec_ctx = NULL, *audio_dec_ctx;
static AVStream *video_stream = NULL, *audio_stream = NULL;
static const char *src_filename = NULL;
static const char *video_dst_filename = NULL;
static const char *audio_dst_filename = NULL;
static FILE *video_dst_file = NULL;
static FILE *audio_dst_file = NULL;

static uint8_t *video_dst_data[4] = {NULL};
static int      video_dst_linesize[4];
static int video_dst_bufsize;

static int video_stream_idx = -1, audio_stream_idx = -1;
static AVFrame *frame = NULL;
static AVPacket pkt;
static int video_frame_count = 0;
static int audio_frame_count = 0;

/* The different ways of decoding and managing data memory. You are not
 * supposed to support all the modes in your application but pick the one most
 * appropriate to your needs. Look for the use of api_mode in this example to
 * see what are the differences of API usage between them */
enum {
    API_MODE_OLD                  = 0, /* old method, deprecated */
    API_MODE_NEW_API_REF_COUNT    = 1, /* new method, using the frame reference counting */
    API_MODE_NEW_API_NO_REF_COUNT = 2, /* new method, without reference counting */
};

static int api_mode = API_MODE_OLD;

static int decode_packet(int *got_frame, int cached)
{
    int ret = 0;
    int decoded = pkt.size;

    *got_frame = 0;

    if (pkt.stream_index == video_stream_idx) {
        /* decode video frame */
        ret = avcodec_decode_video2(video_dec_ctx, frame, got_frame, &pkt);
        if (ret < 0) {
            fprintf(stderr, "Error decoding video frame (%s)\n", av_err2str(ret));
            return ret;
        }

        if (*got_frame) {
            printf("video_frame%s n:%d coded_n:%d pts:%s\n",
                   cached ? "(cached)" : "",
                   video_frame_count++, frame->coded_picture_number,
                   av_ts2timestr(frame->pts, &video_dec_ctx->time_base));

            /* copy decoded frame to destination buffer:
             * this is required since rawvideo expects non aligned data */
            av_image_copy(video_dst_data, video_dst_linesize,
                          (const uint8_t **)(frame->data), frame->linesize,
                          video_dec_ctx->pix_fmt, video_dec_ctx->width, video_dec_ctx->height);

            /* write to rawvideo file */
            fwrite(video_dst_data[0], 1, video_dst_bufsize, video_dst_file);
        }
    } else if (pkt.stream_index == audio_stream_idx) {
        /* decode audio frame */
        ret = avcodec_decode_audio4(audio_dec_ctx, frame, got_frame, &pkt);
        if (ret < 0) {
            fprintf(stderr, "Error decoding audio frame (%s)\n", av_err2str(ret));
            return ret;
        }
        /* Some audio decoders decode only part of the packet, and have to be
         * called again with the remainder of the packet data.
         * Sample: fate-suite/lossless-audio/luckynight-partial.shn
         * Also, some decoders might over-read the packet. */
        decoded = FFMIN(ret, pkt.size);

        if (*got_frame) {
            size_t unpadded_linesize = frame->nb_samples * av_get_bytes_per_sample(frame->format);
            printf("audio_frame%s n:%d nb_samples:%d pts:%s\n",
                   cached ? "(cached)" : "",
                   audio_frame_count++, frame->nb_samples,
                   av_ts2timestr(frame->pts, &audio_dec_ctx->time_base));

            /* Write the raw audio data samples of the first plane. This works
             * fine for packed formats (e.g. AV_SAMPLE_FMT_S16). However,
             * most audio decoders output planar audio, which uses a separate
             * plane of audio samples for each channel (e.g. AV_SAMPLE_FMT_S16P).
             * In other words, this code will write only the first audio channel
             * in these cases.
             * You should use libswresample or libavfilter to convert the frame
             * to packed data. */
            fwrite(frame->extended_data[0], 1, unpadded_linesize, audio_dst_file);
        }
    }

    /* If we use the new API with reference counting, we own the data and need
     * to de-reference it when we don't use it anymore */
    if (*got_frame && api_mode == API_MODE_NEW_API_REF_COUNT)
        av_frame_unref(frame);

    return decoded;
}

static int open_codec_context(int *stream_idx,
                              AVFormatContext *fmt_ctx, enum AVMediaType type)
{
    int ret;
    AVStream *st;
    AVCodecContext *dec_ctx = NULL;
    AVCodec *dec = NULL;
    AVDictionary *opts = NULL;

    ret = av_find_best_stream(fmt_ctx, type, -1, -1, NULL, 0);
    if (ret < 0) {
        fprintf(stderr, "Could not find %s stream in input file '%s'\n",
                av_get_media_type_string(type), src_filename);
        return ret;
    } else {
        *stream_idx = ret;
        st = fmt_ctx->streams[*stream_idx];

        /* find decoder for the stream */
        dec_ctx = st->codec;
        dec = avcodec_find_decoder(dec_ctx->codec_id);
        if (!dec) {
            fprintf(stderr, "Failed to find %s codec\n",
                    av_get_media_type_string(type));
            return AVERROR(EINVAL);
        }

        /* Init the decoders, with or without reference counting */
        if (api_mode == API_MODE_NEW_API_REF_COUNT)
            av_dict_set(&opts, "refcounted_frames", "1", 0);
        if ((ret = avcodec_open2(dec_ctx, dec, &opts)) < 0) {
            fprintf(stderr, "Failed to open %s codec\n",
                    av_get_media_type_string(type));
            return ret;
        }
    }

    return 0;
}

static int get_format_from_sample_fmt(const char **fmt,
                                      enum AVSampleFormat sample_fmt)
{
    int i;
    struct sample_fmt_entry {
        enum AVSampleFormat sample_fmt; const char *fmt_be, *fmt_le;
    } sample_fmt_entries[] = {
        { AV_SAMPLE_FMT_U8,  "u8",    "u8"    },
        { AV_SAMPLE_FMT_S16, "s16be", "s16le" },
        { AV_SAMPLE_FMT_S32, "s32be", "s32le" },
        { AV_SAMPLE_FMT_FLT, "f32be", "f32le" },
        { AV_SAMPLE_FMT_DBL, "f64be", "f64le" },
    };
    *fmt = NULL;

    for (i = 0; i < FF_ARRAY_ELEMS(sample_fmt_entries); i++) {
        struct sample_fmt_entry *entry = &sample_fmt_entries[i];
        if (sample_fmt == entry->sample_fmt) {
            *fmt = AV_NE(entry->fmt_be, entry->fmt_le);
            return 0;
        }
    }

    fprintf(stderr,
            "sample format %s is not supported as output format\n",
            av_get_sample_fmt_name(sample_fmt));
    return -1;
}

int main (int argc, char **argv)
{
    int ret = 0, got_frame;

    if (argc != 4 && argc != 5) {
        fprintf(stderr, "usage: %s [-refcount=<old|new_norefcount|new_refcount>] "
                "input_file video_output_file audio_output_file\n"
                "API example program to show how to read frames from an input file.\n"
                "This program reads frames from a file, decodes them, and writes decoded\n"
                "video frames to a rawvideo file named video_output_file, and decoded\n"
                "audio frames to a rawaudio file named audio_output_file.\n\n"
                "If the -refcount option is specified, the program use the\n"
                "reference counting frame system which allows keeping a copy of\n"
                "the data for longer than one decode call. If unset, it's using\n"
                "the classic old method.\n"
                "\n", argv[0]);
        exit(1);
    }
    if (argc == 5) {
        const char *mode = argv[1] + strlen("-refcount=");
        if      (!strcmp(mode, "old"))            api_mode = API_MODE_OLD;
        else if (!strcmp(mode, "new_norefcount")) api_mode = API_MODE_NEW_API_NO_REF_COUNT;
        else if (!strcmp(mode, "new_refcount"))   api_mode = API_MODE_NEW_API_REF_COUNT;
        else {
            fprintf(stderr, "unknow mode '%s'\n", mode);
            exit(1);
        }
        argv++;
    }
    src_filename = argv[1];
    video_dst_filename = argv[2];
    audio_dst_filename = argv[3];

    /* register all formats and codecs */
    av_register_all();

    /* open input file, and allocate format context */
    if (avformat_open_input(&fmt_ctx, src_filename, NULL, NULL) < 0) {
        fprintf(stderr, "Could not open source file %s\n", src_filename);
        exit(1);
    }

    /* retrieve stream information */
    if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
        fprintf(stderr, "Could not find stream information\n");
        exit(1);
    }

    if (open_codec_context(&video_stream_idx, fmt_ctx, AVMEDIA_TYPE_VIDEO) >= 0) {
        video_stream = fmt_ctx->streams[video_stream_idx];
        video_dec_ctx = video_stream->codec;

        video_dst_file = fopen(video_dst_filename, "wb");
        if (!video_dst_file) {
            fprintf(stderr, "Could not open destination file %s\n", video_dst_filename);
            ret = 1;
            goto end;
        }

        /* allocate image where the decoded image will be put */
        ret = av_image_alloc(video_dst_data, video_dst_linesize,
                             video_dec_ctx->width, video_dec_ctx->height,
                             video_dec_ctx->pix_fmt, 1);
        if (ret < 0) {
            fprintf(stderr, "Could not allocate raw video buffer\n");
            goto end;
        }
        video_dst_bufsize = ret;
    }

    if (open_codec_context(&audio_stream_idx, fmt_ctx, AVMEDIA_TYPE_AUDIO) >= 0) {
        audio_stream = fmt_ctx->streams[audio_stream_idx];
        audio_dec_ctx = audio_stream->codec;
        audio_dst_file = fopen(audio_dst_filename, "wb");
        if (!audio_dst_file) {
            fprintf(stderr, "Could not open destination file %s\n", audio_dst_filename);
            ret = 1;
            goto end;
        }
    }

    /* dump input information to stderr */
    av_dump_format(fmt_ctx, 0, src_filename, 0);

    if (!audio_stream && !video_stream) {
        fprintf(stderr, "Could not find audio or video stream in the input, aborting\n");
        ret = 1;
        goto end;
    }

    /* When using the new API, you need to use the libavutil/frame.h API, while
     * the classic frame management is available in libavcodec */
    if (api_mode == API_MODE_OLD)
        frame = avcodec_alloc_frame();
    else
        frame = av_frame_alloc();
    if (!frame) {
        fprintf(stderr, "Could not allocate frame\n");
        ret = AVERROR(ENOMEM);
        goto end;
    }

    /* initialize packet, set data to NULL, let the demuxer fill it */
    av_init_packet(&pkt);
    pkt.data = NULL;
    pkt.size = 0;

    if (video_stream)
        printf("Demuxing video from file '%s' into '%s'\n", src_filename, video_dst_filename);
    if (audio_stream)
        printf("Demuxing audio from file '%s' into '%s'\n", src_filename, audio_dst_filename);

    /* read frames from the file */
    while (av_read_frame(fmt_ctx, &pkt) >= 0) {
        AVPacket orig_pkt = pkt;
        do {
            ret = decode_packet(&got_frame, 0);
            if (ret < 0)
                break;
            pkt.data += ret;
            pkt.size -= ret;
        } while (pkt.size > 0);
        av_free_packet(&orig_pkt);
    }

    /* flush cached frames */
    pkt.data = NULL;
    pkt.size = 0;
    do {
        decode_packet(&got_frame, 1);
    } while (got_frame);

    printf("Demuxing succeeded.\n");

    if (video_stream) {
        printf("Play the output video file with the command:\n"
               "ffplay -f rawvideo -pix_fmt %s -video_size %dx%d %s\n",
               av_get_pix_fmt_name(video_dec_ctx->pix_fmt), video_dec_ctx->width, video_dec_ctx->height,
               video_dst_filename);
    }

    if (audio_stream) {
        enum AVSampleFormat sfmt = audio_dec_ctx->sample_fmt;
        int n_channels = audio_dec_ctx->channels;
        const char *fmt;

        if (av_sample_fmt_is_planar(sfmt)) {
            const char *packed = av_get_sample_fmt_name(sfmt);
            printf("Warning: the sample format the decoder produced is planar "
                   "(%s). This example will output the first channel only.\n",
                   packed ? packed : "?");
            sfmt = av_get_packed_sample_fmt(sfmt);
            n_channels = 1;
        }

        if ((ret = get_format_from_sample_fmt(&fmt, sfmt)) < 0)
            goto end;

        printf("Play the output audio file with the command:\n"
               "ffplay -f %s -ac %d -ar %d %s\n",
               fmt, n_channels, audio_dec_ctx->sample_rate,
               audio_dst_filename);
    }

end:
    avcodec_close(video_dec_ctx);
    avcodec_close(audio_dec_ctx);
    avformat_close_input(&fmt_ctx);
    if (video_dst_file)
        fclose(video_dst_file);
    if (audio_dst_file)
        fclose(audio_dst_file);
    if (api_mode == API_MODE_OLD)
        avcodec_free_frame(&frame);
    else
        av_frame_free(&frame);
    av_free(video_dst_data[0]);

    return ret < 0;
}
Commit	Line	Data
2ba45a60 DM	1	/*
	2	* Copyright (c) 2012 Stefano Sabatini
	3	*
	4	* Permission is hereby granted, free of charge, to any person obtaining a copy
	5	* of this software and associated documentation files (the "Software"), to deal
	6	* in the Software without restriction, including without limitation the rights
	7	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	8	* copies of the Software, and to permit persons to whom the Software is
	9	* furnished to do so, subject to the following conditions:
	10	*
	11	* The above copyright notice and this permission notice shall be included in
	12	* all copies or substantial portions of the Software.
	13	*
	14	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	15	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	16	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
	17	* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	18	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	19	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
	20	* THE SOFTWARE.
	21	*/
	22
	23	/**
	24	* @file
	25	* Demuxing and decoding example.
	26	*
	27	* Show how to use the libavformat and libavcodec API to demux and
	28	* decode audio and video data.
	29	* @example demuxing_decoding.c
	30	*/
	31
	32	#include <libavutil/imgutils.h>
	33	#include <libavutil/samplefmt.h>
	34	#include <libavutil/timestamp.h>
	35	#include <libavformat/avformat.h>
	36
	37	static AVFormatContext *fmt_ctx = NULL;
	38	static AVCodecContext video_dec_ctx = NULL, audio_dec_ctx;
	39	static AVStream video_stream = NULL, audio_stream = NULL;
	40	static const char *src_filename = NULL;
	41	static const char *video_dst_filename = NULL;
	42	static const char *audio_dst_filename = NULL;
	43	static FILE *video_dst_file = NULL;
	44	static FILE *audio_dst_file = NULL;
	45
	46	static uint8_t *video_dst_data[4] = {NULL};
	47	static int video_dst_linesize[4];
	48	static int video_dst_bufsize;
	49
	50	static int video_stream_idx = -1, audio_stream_idx = -1;
	51	static AVFrame *frame = NULL;
	52	static AVPacket pkt;
	53	static int video_frame_count = 0;
	54	static int audio_frame_count = 0;
	55
	56	/* The different ways of decoding and managing data memory. You are not
	57	* supposed to support all the modes in your application but pick the one most
	58	* appropriate to your needs. Look for the use of api_mode in this example to
	59	* see what are the differences of API usage between them */
	60	enum {
	61	API_MODE_OLD = 0, /* old method, deprecated */
	62	API_MODE_NEW_API_REF_COUNT = 1, /* new method, using the frame reference counting */
	63	API_MODE_NEW_API_NO_REF_COUNT = 2, /* new method, without reference counting */
	64	};
65
66	static int api_mode = API_MODE_OLD;
67
68	static int decode_packet(int *got_frame, int cached)
69	{
70	int ret = 0;
71	int decoded = pkt.size;
72
73	*got_frame = 0;
74
75	if (pkt.stream_index == video_stream_idx) {
76	/* decode video frame */
77	ret = avcodec_decode_video2(video_dec_ctx, frame, got_frame, &pkt);
78	if (ret < 0) {
79	fprintf(stderr, "Error decoding video frame (%s)\n", av_err2str(ret));
80	return ret;
81	}
82
83	if (*got_frame) {
84	printf("video_frame%s n:%d coded_n:%d pts:%s\n",
85	cached ? "(cached)" : "",
86	video_frame_count++, frame->coded_picture_number,
87	av_ts2timestr(frame->pts, &video_dec_ctx->time_base));
88
89	/* copy decoded frame to destination buffer:
90	* this is required since rawvideo expects non aligned data */
91	av_image_copy(video_dst_data, video_dst_linesize,
92	(const uint8_t **)(frame->data), frame->linesize,
93	video_dec_ctx->pix_fmt, video_dec_ctx->width, video_dec_ctx->height);
94
95	/* write to rawvideo file */
96	fwrite(video_dst_data[0], 1, video_dst_bufsize, video_dst_file);
97	}
98	} else if (pkt.stream_index == audio_stream_idx) {
99	/* decode audio frame */
100	ret = avcodec_decode_audio4(audio_dec_ctx, frame, got_frame, &pkt);
101	if (ret < 0) {
102	fprintf(stderr, "Error decoding audio frame (%s)\n", av_err2str(ret));
103	return ret;
104	}
105	/* Some audio decoders decode only part of the packet, and have to be
106	* called again with the remainder of the packet data.
107	* Sample: fate-suite/lossless-audio/luckynight-partial.shn
108	* Also, some decoders might over-read the packet. */
109	decoded = FFMIN(ret, pkt.size);
110
111	if (*got_frame) {
112	size_t unpadded_linesize = frame->nb_samples * av_get_bytes_per_sample(frame->format);
113	printf("audio_frame%s n:%d nb_samples:%d pts:%s\n",
114	cached ? "(cached)" : "",
115	audio_frame_count++, frame->nb_samples,
116	av_ts2timestr(frame->pts, &audio_dec_ctx->time_base));
117
118	/* Write the raw audio data samples of the first plane. This works
119	* fine for packed formats (e.g. AV_SAMPLE_FMT_S16). However,
120	* most audio decoders output planar audio, which uses a separate
121	* plane of audio samples for each channel (e.g. AV_SAMPLE_FMT_S16P).
122	* In other words, this code will write only the first audio channel
123	* in these cases.
124	* You should use libswresample or libavfilter to convert the frame
125	* to packed data. */
126	fwrite(frame->extended_data[0], 1, unpadded_linesize, audio_dst_file);
127	}
128	}
129
130	/* If we use the new API with reference counting, we own the data and need
131	* to de-reference it when we don't use it anymore */
132	if (*got_frame && api_mode == API_MODE_NEW_API_REF_COUNT)
133	av_frame_unref(frame);
134
135	return decoded;
136	}
137
138	static int open_codec_context(int *stream_idx,
139	AVFormatContext *fmt_ctx, enum AVMediaType type)
140	{
141	int ret;
142	AVStream *st;
143	AVCodecContext *dec_ctx = NULL;
144	AVCodec *dec = NULL;
145	AVDictionary *opts = NULL;
146
147	ret = av_find_best_stream(fmt_ctx, type, -1, -1, NULL, 0);
148	if (ret < 0) {
149	fprintf(stderr, "Could not find %s stream in input file '%s'\n",
150	av_get_media_type_string(type), src_filename);
151	return ret;
152	} else {
153	*stream_idx = ret;
154	st = fmt_ctx->streams[*stream_idx];
155
156	/* find decoder for the stream */
157	dec_ctx = st->codec;
158	dec = avcodec_find_decoder(dec_ctx->codec_id);
159	if (!dec) {
160	fprintf(stderr, "Failed to find %s codec\n",
161	av_get_media_type_string(type));
162	return AVERROR(EINVAL);
163	}
164
165	/* Init the decoders, with or without reference counting */
166	if (api_mode == API_MODE_NEW_API_REF_COUNT)
167	av_dict_set(&opts, "refcounted_frames", "1", 0);
168	if ((ret = avcodec_open2(dec_ctx, dec, &opts)) < 0) {
169	fprintf(stderr, "Failed to open %s codec\n",
170	av_get_media_type_string(type));
171	return ret;
172	}
173	}
174
175	return 0;
176	}
177
178	static int get_format_from_sample_fmt(const char **fmt,
179	enum AVSampleFormat sample_fmt)
180	{
181	int i;
182	struct sample_fmt_entry {
183	enum AVSampleFormat sample_fmt; const char fmt_be, fmt_le;
184	} sample_fmt_entries[] = {
185	{ AV_SAMPLE_FMT_U8, "u8", "u8" },
186	{ AV_SAMPLE_FMT_S16, "s16be", "s16le" },
187	{ AV_SAMPLE_FMT_S32, "s32be", "s32le" },
188	{ AV_SAMPLE_FMT_FLT, "f32be", "f32le" },
189	{ AV_SAMPLE_FMT_DBL, "f64be", "f64le" },
190	};
191	*fmt = NULL;
192
193	for (i = 0; i < FF_ARRAY_ELEMS(sample_fmt_entries); i++) {
194	struct sample_fmt_entry *entry = &sample_fmt_entries[i];
195	if (sample_fmt == entry->sample_fmt) {
196	*fmt = AV_NE(entry->fmt_be, entry->fmt_le);
197	return 0;
198	}
199	}
200
201	fprintf(stderr,
202	"sample format %s is not supported as output format\n",
203	av_get_sample_fmt_name(sample_fmt));
204	return -1;
205	}
206
207	int main (int argc, char **argv)
208	{
209	int ret = 0, got_frame;
210
211	if (argc != 4 && argc != 5) {
212	fprintf(stderr, "usage: %s [-refcount=<old\|new_norefcount\|new_refcount>] "
213	"input_file video_output_file audio_output_file\n"
214	"API example program to show how to read frames from an input file.\n"
215	"This program reads frames from a file, decodes them, and writes decoded\n"
216	"video frames to a rawvideo file named video_output_file, and decoded\n"
217	"audio frames to a rawaudio file named audio_output_file.\n\n"
218	"If the -refcount option is specified, the program use the\n"
219	"reference counting frame system which allows keeping a copy of\n"
220	"the data for longer than one decode call. If unset, it's using\n"
221	"the classic old method.\n"
222	"\n", argv[0]);
223	exit(1);
224	}
225	if (argc == 5) {
226	const char *mode = argv[1] + strlen("-refcount=");
227	if (!strcmp(mode, "old")) api_mode = API_MODE_OLD;
228	else if (!strcmp(mode, "new_norefcount")) api_mode = API_MODE_NEW_API_NO_REF_COUNT;
229	else if (!strcmp(mode, "new_refcount")) api_mode = API_MODE_NEW_API_REF_COUNT;
230	else {
231	fprintf(stderr, "unknow mode '%s'\n", mode);
232	exit(1);
233	}
234	argv++;
235	}
236	src_filename = argv[1];
237	video_dst_filename = argv[2];
238	audio_dst_filename = argv[3];
239
240	/* register all formats and codecs */
241	av_register_all();
242
243	/* open input file, and allocate format context */
244	if (avformat_open_input(&fmt_ctx, src_filename, NULL, NULL) < 0) {
245	fprintf(stderr, "Could not open source file %s\n", src_filename);
246	exit(1);
247	}
248
249	/* retrieve stream information */
250	if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
251	fprintf(stderr, "Could not find stream information\n");
252	exit(1);
253	}
254
255	if (open_codec_context(&video_stream_idx, fmt_ctx, AVMEDIA_TYPE_VIDEO) >= 0) {
256	video_stream = fmt_ctx->streams[video_stream_idx];
257	video_dec_ctx = video_stream->codec;
258
259	video_dst_file = fopen(video_dst_filename, "wb");
260	if (!video_dst_file) {
261	fprintf(stderr, "Could not open destination file %s\n", video_dst_filename);
262	ret = 1;
263	goto end;
264	}
265
266	/* allocate image where the decoded image will be put */
267	ret = av_image_alloc(video_dst_data, video_dst_linesize,
268	video_dec_ctx->width, video_dec_ctx->height,
269	video_dec_ctx->pix_fmt, 1);
270	if (ret < 0) {
271	fprintf(stderr, "Could not allocate raw video buffer\n");
272	goto end;
273	}
274	video_dst_bufsize = ret;
275	}
276
277	if (open_codec_context(&audio_stream_idx, fmt_ctx, AVMEDIA_TYPE_AUDIO) >= 0) {
278	audio_stream = fmt_ctx->streams[audio_stream_idx];
279	audio_dec_ctx = audio_stream->codec;
280	audio_dst_file = fopen(audio_dst_filename, "wb");
281	if (!audio_dst_file) {
282	fprintf(stderr, "Could not open destination file %s\n", audio_dst_filename);
283	ret = 1;
284	goto end;
285	}
286	}
287
288	/* dump input information to stderr */
289	av_dump_format(fmt_ctx, 0, src_filename, 0);
290
291	if (!audio_stream && !video_stream) {
292	fprintf(stderr, "Could not find audio or video stream in the input, aborting\n");
293	ret = 1;
294	goto end;
295	}
296
297	/* When using the new API, you need to use the libavutil/frame.h API, while
298	* the classic frame management is available in libavcodec */
299	if (api_mode == API_MODE_OLD)
300	frame = avcodec_alloc_frame();
301	else
302	frame = av_frame_alloc();
303	if (!frame) {
304	fprintf(stderr, "Could not allocate frame\n");
305	ret = AVERROR(ENOMEM);
306	goto end;
307	}
308
309	/* initialize packet, set data to NULL, let the demuxer fill it */
310	av_init_packet(&pkt);
311	pkt.data = NULL;
312	pkt.size = 0;
313
314	if (video_stream)
315	printf("Demuxing video from file '%s' into '%s'\n", src_filename, video_dst_filename);
316	if (audio_stream)
317	printf("Demuxing audio from file '%s' into '%s'\n", src_filename, audio_dst_filename);
318
319	/* read frames from the file */
320	while (av_read_frame(fmt_ctx, &pkt) >= 0) {
321	AVPacket orig_pkt = pkt;
322	do {
323	ret = decode_packet(&got_frame, 0);
324	if (ret < 0)
325	break;
326	pkt.data += ret;
327	pkt.size -= ret;
328	} while (pkt.size > 0);
329	av_free_packet(&orig_pkt);
330	}
331
332	/* flush cached frames */
333	pkt.data = NULL;
334	pkt.size = 0;
335	do {
336	decode_packet(&got_frame, 1);
337	} while (got_frame);
338
339	printf("Demuxing succeeded.\n");
340
341	if (video_stream) {
342	printf("Play the output video file with the command:\n"
343	"ffplay -f rawvideo -pix_fmt %s -video_size %dx%d %s\n",
344	av_get_pix_fmt_name(video_dec_ctx->pix_fmt), video_dec_ctx->width, video_dec_ctx->height,
345	video_dst_filename);
346	}
347
348	if (audio_stream) {
349	enum AVSampleFormat sfmt = audio_dec_ctx->sample_fmt;
350	int n_channels = audio_dec_ctx->channels;
351	const char *fmt;
352
353	if (av_sample_fmt_is_planar(sfmt)) {
354	const char *packed = av_get_sample_fmt_name(sfmt);
355	printf("Warning: the sample format the decoder produced is planar "
356	"(%s). This example will output the first channel only.\n",
357	packed ? packed : "?");
358	sfmt = av_get_packed_sample_fmt(sfmt);
359	n_channels = 1;
360	}
361
362	if ((ret = get_format_from_sample_fmt(&fmt, sfmt)) < 0)
363	goto end;
364
365	printf("Play the output audio file with the command:\n"
366	"ffplay -f %s -ac %d -ar %d %s\n",
367	fmt, n_channels, audio_dec_ctx->sample_rate,
368	audio_dst_filename);
369	}
370
371	end:
372	avcodec_close(video_dec_ctx);
373	avcodec_close(audio_dec_ctx);
374	avformat_close_input(&fmt_ctx);
375	if (video_dst_file)
376	fclose(video_dst_file);
377	if (audio_dst_file)
378	fclose(audio_dst_file);
379	if (api_mode == API_MODE_OLD)
380	avcodec_free_frame(&frame);
381	else
382	av_frame_free(&frame);
383	av_free(video_dst_data[0]);
384
385	return ret < 0;
386	}