1 /*
2 * Copyright (c) 2010 Nicolas George
3 * Copyright (c) 2011 Stefano Sabatini
4 * Copyright (c) 2014 Andrey Utkin
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 * THE SOFTWARE.
23 */
24
25 /**
26 * @file
27 * API example for demuxing, decoding, filtering, encoding and muxing
28 * @example transcoding.c
29 */
30
38
47
53
55 {
56 int ret;
57 unsigned int i;
58
62 return ret;
63 }
64
67 return ret;
68 }
69
71 if (!stream_ctx)
73
78 if (!dec) {
81 }
83 if (!codec_ctx) {
86 }
88 if (ret < 0) {
90 "for stream #%u\n", i);
91 return ret;
92 }
93 /* Reencode video & audio and remux subtitles etc. */
98 /* Open decoder */
100 if (ret < 0) {
102 return ret;
103 }
104 }
105 stream_ctx[i].
dec_ctx = codec_ctx;
106 }
107
109 return 0;
110 }
111
113 {
118 int ret;
119 unsigned int i;
120
123 if (!ofmt_ctx) {
126 }
127
128
131 if (!out_stream) {
134 }
135
136 in_stream = ifmt_ctx->
streams[i];
137 dec_ctx = stream_ctx[i].
dec_ctx;
138
141 /* in this example, we choose transcoding to same codec */
143 if (!encoder) {
146 }
148 if (!enc_ctx) {
151 }
152
153 /* In this example, we transcode to same properties (picture size,
154 * sample rate etc.). These properties can be changed for output
155 * streams easily using filters */
160 /* take first format from list of supported formats */
163 else
165 /* video time_base can be set to whatever is handy and supported by encoder */
167 } else {
171 /* take first format from list of supported formats */
174 }
175
176 /* Third parameter can be used to pass settings to encoder */
178 if (ret < 0) {
180 return ret;
181 }
183 if (ret < 0) {
185 return ret;
186 }
189
191 stream_ctx[i].
enc_ctx = enc_ctx;
195 } else {
196 /* if this stream must be remuxed */
198 if (ret < 0) {
200 return ret;
201 }
203 }
204
205 }
207
210 if (ret < 0) {
212 return ret;
213 }
214 }
215
216 /* init muxer, write output file header */
218 if (ret < 0) {
220 return ret;
221 }
222
223 return 0;
224 }
225
228 {
230 int ret = 0;
238
239 if (!outputs || !inputs || !filter_graph) {
242 }
243
247 if (!buffersrc || !buffersink) {
251 }
252
254 "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d",
259
261 args,
NULL, filter_graph);
262 if (ret < 0) {
265 }
266
269 if (ret < 0) {
272 }
273
277 if (ret < 0) {
280 }
284 if (!buffersrc || !buffersink) {
288 }
289
294 "time_base=%d/%d:sample_rate=%d:sample_fmt=%s:channel_layout=0x%"PRIx64,
299 args,
NULL, filter_graph);
300 if (ret < 0) {
303 }
304
307 if (ret < 0) {
310 }
311
315 if (ret < 0) {
318 }
319
323 if (ret < 0) {
326 }
327
331 if (ret < 0) {
334 }
335 } else {
338 }
339
340 /* Endpoints for the filter graph. */
345
350
351 if (!outputs->
name || !inputs->
name) {
354 }
355
357 &inputs, &outputs,
NULL)) < 0)
359
362
363 /* Fill FilteringContext */
367
371
372 return ret;
373 }
374
376 {
377 const char *filter_spec;
378 unsigned int i;
379 int ret;
381 if (!filter_ctx)
383
390 continue;
391
392
394 filter_spec = "null"; /* passthrough (dummy) filter for video */
395 else
396 filter_spec = "anull"; /* passthrough (dummy) filter for audio */
398 stream_ctx[i].enc_ctx, filter_spec);
399 if (ret)
400 return ret;
401 }
402 return 0;
403 }
404
406 int ret;
407 int got_frame_local;
412
413 if (!got_frame)
414 got_frame = &got_frame_local;
415
417 /* encode filtered frame */
421 ret = enc_func(stream_ctx[stream_index].enc_ctx, &enc_pkt,
422 filt_frame, got_frame);
424 if (ret < 0)
425 return ret;
426 if (!(*got_frame))
427 return 0;
428
429 /* prepare packet for muxing */
432 stream_ctx[stream_index].enc_ctx->
time_base,
434
436 /* mux encoded frame */
438 return ret;
439 }
440
442 {
443 int ret;
445
447 /* push the decoded frame into the filtergraph */
449 frame, 0);
450 if (ret < 0) {
452 return ret;
453 }
454
455 /* pull filtered frames from the filtergraph */
456 while (1) {
458 if (!filt_frame) {
460 break;
461 }
464 filt_frame);
465 if (ret < 0) {
466 /* if no more frames for output - returns AVERROR(EAGAIN)
467 * if flushed and no more frames for output - returns AVERROR_EOF
468 * rewrite retcode to 0 to show it as normal procedure completion
469 */
471 ret = 0;
473 break;
474 }
475
478 if (ret < 0)
479 break;
480 }
481
482 return ret;
483 }
484
486 {
487 int ret;
488 int got_frame;
489
490 if (!(stream_ctx[stream_index].enc_ctx->codec->capabilities &
492 return 0;
493
494 while (1) {
497 if (ret < 0)
498 break;
499 if (!got_frame)
500 return 0;
501 }
502 return ret;
503 }
504
505 int main(
int argc,
char **argv)
506 {
507 int ret;
511 unsigned int stream_index;
512 unsigned int i;
513 int got_frame;
515
516 if (argc != 3) {
518 return 1;
519 }
520
523
530
531 /* read all packets */
532 while (1) {
534 break;
538 stream_index);
539
543 if (!frame) {
545 break;
546 }
552 ret = dec_func(stream_ctx[stream_index].
dec_ctx, frame,
553 &got_frame, &packet);
554 if (ret < 0) {
557 break;
558 }
559
560 if (got_frame) {
564 if (ret < 0)
566 } else {
568 }
569 } else {
570 /* remux this frame without reencoding */
574
576 if (ret < 0)
578 }
580 }
581
582 /* flush filters and encoders */
584 /* flush filter */
586 continue;
588 if (ret < 0) {
591 }
592
593 /* flush encoder */
595 if (ret < 0) {
598 }
599 }
600
611 }
618
619 if (ret < 0)
621
622 return ret ? 1 : 0;
623 }
int avio_open(AVIOContext **s, const char *url, int flags)
Create and initialize a AVIOContext for accessing the resource indicated by url.
AVFilterGraph * filter_graph
AVFilterContext * buffersink_ctx
static int encode_write_frame(AVFrame *filt_frame, unsigned int stream_index, int *got_frame)
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
This structure describes decoded (raw) audio or video data.
AVCodec * avcodec_find_encoder(enum AVCodecID id)
Find a registered encoder with a matching codec ID.
int av_interleaved_write_frame(AVFormatContext *s, AVPacket *pkt)
Write a packet to an output media file ensuring correct interleaving.
AVFilterGraph * avfilter_graph_alloc(void)
Allocate a filter graph.
Memory buffer source API.
AVFilterGraph * filter_graph
int avfilter_graph_config(AVFilterGraph *graphctx, void *log_ctx)
Check validity and configure all the links and formats in the graph.
void avfilter_inout_free(AVFilterInOut **inout)
Free the supplied list of AVFilterInOut and set *inout to NULL.
struct AVFilterInOut * next
next input/input in the list, NULL if this is the last
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
#define AVIO_FLAG_WRITE
write-only
AVRational sample_aspect_ratio
sample aspect ratio (0 if unknown) That is the width of a pixel divided by the height of the pixel...
void avfilter_graph_free(AVFilterGraph **graph)
Free a graph, destroy its links, and set *graph to NULL.
attribute_deprecated int avcodec_encode_audio2(AVCodecContext *avctx, AVPacket *avpkt, const AVFrame *frame, int *got_packet_ptr)
Encode a frame of audio.
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
attribute_deprecated int avcodec_decode_audio4(AVCodecContext *avctx, AVFrame *frame, int *got_frame_ptr, const AVPacket *avpkt)
Decode the audio frame of size avpkt->size from avpkt->data into frame.
int av_opt_set_bin(void *obj, const char *name, const uint8_t *val, int len, int search_flags)
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
int av_get_channel_layout_nb_channels(uint64_t channel_layout)
Return the number of channels in the channel layout.
static int open_input_file(const char *filename)
AVRational av_guess_frame_rate(AVFormatContext *ctx, AVStream *stream, AVFrame *frame)
Guess the frame rate, based on both the container and codec information.
memory buffer sink API for audio and video
#define AV_CODEC_CAP_DELAY
Encoder or decoder requires flushing with NULL input at the end in order to give the complete and cor...
enum AVSampleFormat sample_fmt
audio sample format
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
static av_cold int end(AVCodecContext *avctx)
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
AVStream ** streams
A list of all streams in the file.
int avcodec_parameters_to_context(AVCodecContext *codec, const AVCodecParameters *par)
Fill the codec context based on the values from the supplied codec parameters.
void avfilter_register_all(void)
Initialize the filter system.
int avfilter_graph_create_filter(AVFilterContext **filt_ctx, const AVFilter *filt, const char *name, const char *args, void *opaque, AVFilterGraph *graph_ctx)
Create and add a filter instance into an existing graph.
#define AVERROR_EOF
End of file.
int64_t av_frame_get_best_effort_timestamp(const AVFrame *frame)
Accessors for some AVFrame fields.
struct AVOutputFormat * oformat
The output container format.
void av_dump_format(AVFormatContext *ic, int index, const char *url, int is_output)
Print detailed information about the input or output format, such as duration, bitrate, streams, container, programs, metadata, side data, codec and time base.
int avcodec_parameters_copy(AVCodecParameters *dst, const AVCodecParameters *src)
Copy the contents of src to dst.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static int flush_encoder(unsigned int stream_index)
static AVFormatContext * ifmt_ctx
void av_packet_rescale_ts(AVPacket *pkt, AVRational tb_src, AVRational tb_dst)
Convert valid timing fields (timestamps / durations) in a packet from one timebase to another...
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
enum AVMediaType codec_type
General type of the encoded data.
int flags
AV_CODEC_FLAG_*.
AVFilterContext * buffersrc_ctx
static FilteringContext * filter_ctx
const char * av_get_sample_fmt_name(enum AVSampleFormat sample_fmt)
Return the name of sample_fmt, or NULL if sample_fmt is not recognized.
static void * av_mallocz_array(size_t nmemb, size_t size)
uint64_t channel_layout
Audio channel layout.
const AVFilter * avfilter_get_by_name(const char *name)
Get a filter definition matching the given name.
unsigned int nb_streams
Number of elements in AVFormatContext.streams.
AVFilterContext * buffersrc_ctx
enum AVPixelFormat * pix_fmts
array of supported pixel formats, or NULL if unknown, array is terminated by -1
enum AVPictureType pict_type
Picture type of the frame.
static int init_filter(FilteringContext *fctx, AVCodecContext *dec_ctx, AVCodecContext *enc_ctx, const char *filter_spec)
av_warn_unused_result int avformat_write_header(AVFormatContext *s, AVDictionary **options)
Allocate the stream private data and write the stream header to an output media file.
AVCodecContext * avcodec_alloc_context3(const AVCodec *codec)
Allocate an AVCodecContext and set its fields to default values.
#define AV_OPT_SEARCH_CHILDREN
Search in possible children of the given object first.
int width
picture width / height.
#define av_err2str(errnum)
Convenience macro, the return value should be used only directly in function arguments but never stan...
Usually treated as AVMEDIA_TYPE_DATA.
static const AVFilterPad outputs[]
AVFilterContext * filter_ctx
filter context associated to this input/output
static int init_filters(void)
A linked-list of the inputs/outputs of the filter chain.
#define AV_LOG_INFO
Standard information.
static const AVFilterPad inputs[]
char * av_strdup(const char *s)
Duplicate a string.
Libavcodec external API header.
enum AVMediaType codec_type
attribute_deprecated int avcodec_decode_video2(AVCodecContext *avctx, AVFrame *picture, int *got_picture_ptr, const AVPacket *avpkt)
Decode the video frame of size avpkt->size from avpkt->data into picture.
void avcodec_free_context(AVCodecContext **avctx)
Free the codec context and everything associated with it and write NULL to the provided pointer...
int sample_rate
samples per second
AVIOContext * pb
I/O context.
main external API structure.
AVCodec * avcodec_find_decoder(enum AVCodecID id)
Find a registered decoder with a matching codec ID.
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
static StreamContext * stream_ctx
int pad_idx
index of the filt_ctx pad to use for linking
Rational number (pair of numerator and denominator).
attribute_deprecated int avcodec_encode_video2(AVCodecContext *avctx, AVPacket *avpkt, const AVFrame *frame, int *got_packet_ptr)
Encode a frame of video.
static int open_output_file(const char *filename)
int avcodec_parameters_from_context(AVCodecParameters *par, const AVCodecContext *codec)
Fill the parameters struct based on the values from the supplied codec context.
int avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
Initialize the AVCodecContext to use the given AVCodec.
void avformat_free_context(AVFormatContext *s)
Free an AVFormatContext and all its streams.
int av_read_frame(AVFormatContext *s, AVPacket *pkt)
Return the next frame of a stream.
char * name
unique name for this input/output in the list
#define AV_CODEC_FLAG_GLOBAL_HEADER
Place global headers in extradata instead of every keyframe.
static av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
AVFilterInOut * avfilter_inout_alloc(void)
Allocate a single AVFilterInOut entry.
int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
Read packets of a media file to get stream information.
static AVCodecContext * dec_ctx
void av_init_packet(AVPacket *pkt)
Initialize optional fields of a packet with default values.
static int filter_encode_write_frame(AVFrame *frame, unsigned int stream_index)
int attribute_align_arg av_buffersrc_add_frame_flags(AVFilterContext *ctx, AVFrame *frame, int flags)
Add a frame to the buffer source.
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
void avformat_close_input(AVFormatContext **s)
Close an opened input AVFormatContext.
#define AVERROR_DECODER_NOT_FOUND
Decoder not found.
int channels
number of audio channels
int avformat_open_input(AVFormatContext **ps, const char *url, AVInputFormat *fmt, AVDictionary **options)
Open an input stream and read the header.
int avfilter_graph_parse_ptr(AVFilterGraph *graph, const char *filters, AVFilterInOut **inputs, AVFilterInOut **outputs, void *log_ctx)
Add a graph described by a string to a graph.
int64_t av_get_default_channel_layout(int nb_channels)
Return default channel layout for a given number of channels.
int av_write_trailer(AVFormatContext *s)
Write the stream trailer to an output media file and free the file private data.
int main(int argc, char **argv)
#define AV_LOG_FATAL
Something went wrong and recovery is not possible.
AVCodecParameters * codecpar
#define av_malloc_array(a, b)
enum AVSampleFormat * sample_fmts
array of supported sample formats, or NULL if unknown, array is terminated by -1
AVFilterContext * buffersink_ctx
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
int attribute_align_arg av_buffersink_get_frame(AVFilterContext *ctx, AVFrame *frame)
Get a frame with filtered data from sink and put it in frame.
This structure stores compressed data.
void av_register_all(void)
Initialize libavformat and register all the muxers, demuxers and protocols.
int avio_closep(AVIOContext **s)
Close the resource accessed by the AVIOContext *s, free it and set the pointer pointing to it to NULL...
static AVFormatContext * ofmt_ctx