transcode.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660
  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. * @file demuxing, decoding, filtering, encoding and muxing API usage example
  26. * @example transcode.c
  27. *
  28. * Convert input to output file, applying some hard-coded filter-graph on both
  29. * audio and video streams.
  30. */
  31. #include <libavcodec/avcodec.h>
  32. #include <libavformat/avformat.h>
  33. #include <libavfilter/buffersink.h>
  34. #include <libavfilter/buffersrc.h>
  35. #include <libavutil/channel_layout.h>
  36. #include <libavutil/opt.h>
  37. #include <libavutil/pixdesc.h>
  38. static AVFormatContext *ifmt_ctx;
  39. static AVFormatContext *ofmt_ctx;
  40. typedef struct FilteringContext {
  41. AVFilterContext *buffersink_ctx;
  42. AVFilterContext *buffersrc_ctx;
  43. AVFilterGraph *filter_graph;
  44. AVPacket *enc_pkt;
  45. AVFrame *filtered_frame;
  46. } FilteringContext;
  47. static FilteringContext *filter_ctx;
  48. typedef struct StreamContext {
  49. AVCodecContext *dec_ctx;
  50. AVCodecContext *enc_ctx;
  51. AVFrame *dec_frame;
  52. } StreamContext;
  53. static StreamContext *stream_ctx;
  54. static int open_input_file(const char *filename)
  55. {
  56. int ret;
  57. unsigned int i;
  58. ifmt_ctx = NULL;
  59. if ((ret = avformat_open_input(&ifmt_ctx, filename, NULL, NULL)) < 0) {
  60. av_log(NULL, AV_LOG_ERROR, "Cannot open input file\n");
  61. return ret;
  62. }
  63. if ((ret = avformat_find_stream_info(ifmt_ctx, NULL)) < 0) {
  64. av_log(NULL, AV_LOG_ERROR, "Cannot find stream information\n");
  65. return ret;
  66. }
  67. stream_ctx = av_calloc(ifmt_ctx->nb_streams, sizeof(*stream_ctx));
  68. if (!stream_ctx)
  69. return AVERROR(ENOMEM);
  70. for (i = 0; i < ifmt_ctx->nb_streams; i++) {
  71. AVStream *stream = ifmt_ctx->streams[i];
  72. const AVCodec *dec = avcodec_find_decoder(stream->codecpar->codec_id);
  73. AVCodecContext *codec_ctx;
  74. if (!dec) {
  75. av_log(NULL, AV_LOG_ERROR, "Failed to find decoder for stream #%u\n", i);
  76. return AVERROR_DECODER_NOT_FOUND;
  77. }
  78. codec_ctx = avcodec_alloc_context3(dec);
  79. if (!codec_ctx) {
  80. av_log(NULL, AV_LOG_ERROR, "Failed to allocate the decoder context for stream #%u\n", i);
  81. return AVERROR(ENOMEM);
  82. }
  83. ret = avcodec_parameters_to_context(codec_ctx, stream->codecpar);
  84. if (ret < 0) {
  85. av_log(NULL, AV_LOG_ERROR, "Failed to copy decoder parameters to input decoder context "
  86. "for stream #%u\n", i);
  87. return ret;
  88. }
  89. /* Inform the decoder about the timebase for the packet timestamps.
  90. * This is highly recommended, but not mandatory. */
  91. codec_ctx->pkt_timebase = stream->time_base;
  92. /* Reencode video & audio and remux subtitles etc. */
  93. if (codec_ctx->codec_type == AVMEDIA_TYPE_VIDEO
  94. || codec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {
  95. if (codec_ctx->codec_type == AVMEDIA_TYPE_VIDEO)
  96. codec_ctx->framerate = av_guess_frame_rate(ifmt_ctx, stream, NULL);
  97. /* Open decoder */
  98. ret = avcodec_open2(codec_ctx, dec, NULL);
  99. if (ret < 0) {
  100. av_log(NULL, AV_LOG_ERROR, "Failed to open decoder for stream #%u\n", i);
  101. return ret;
  102. }
  103. }
  104. stream_ctx[i].dec_ctx = codec_ctx;
  105. stream_ctx[i].dec_frame = av_frame_alloc();
  106. if (!stream_ctx[i].dec_frame)
  107. return AVERROR(ENOMEM);
  108. }
  109. av_dump_format(ifmt_ctx, 0, filename, 0);
  110. return 0;
  111. }
  112. static int open_output_file(const char *filename)
  113. {
  114. AVStream *out_stream;
  115. AVStream *in_stream;
  116. AVCodecContext *dec_ctx, *enc_ctx;
  117. const AVCodec *encoder;
  118. int ret;
  119. unsigned int i;
  120. ofmt_ctx = NULL;
  121. avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, filename);
  122. if (!ofmt_ctx) {
  123. av_log(NULL, AV_LOG_ERROR, "Could not create output context\n");
  124. return AVERROR_UNKNOWN;
  125. }
  126. for (i = 0; i < ifmt_ctx->nb_streams; i++) {
  127. out_stream = avformat_new_stream(ofmt_ctx, NULL);
  128. if (!out_stream) {
  129. av_log(NULL, AV_LOG_ERROR, "Failed allocating output stream\n");
  130. return AVERROR_UNKNOWN;
  131. }
  132. in_stream = ifmt_ctx->streams[i];
  133. dec_ctx = stream_ctx[i].dec_ctx;
  134. if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO
  135. || dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {
  136. /* in this example, we choose transcoding to same codec */
  137. encoder = avcodec_find_encoder(dec_ctx->codec_id);
  138. if (!encoder) {
  139. av_log(NULL, AV_LOG_FATAL, "Necessary encoder not found\n");
  140. return AVERROR_INVALIDDATA;
  141. }
  142. enc_ctx = avcodec_alloc_context3(encoder);
  143. if (!enc_ctx) {
  144. av_log(NULL, AV_LOG_FATAL, "Failed to allocate the encoder context\n");
  145. return AVERROR(ENOMEM);
  146. }
  147. /* In this example, we transcode to same properties (picture size,
  148. * sample rate etc.). These properties can be changed for output
  149. * streams easily using filters */
  150. if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {
  151. enc_ctx->height = dec_ctx->height;
  152. enc_ctx->width = dec_ctx->width;
  153. enc_ctx->sample_aspect_ratio = dec_ctx->sample_aspect_ratio;
  154. /* take first format from list of supported formats */
  155. if (encoder->pix_fmts)
  156. enc_ctx->pix_fmt = encoder->pix_fmts[0];
  157. else
  158. enc_ctx->pix_fmt = dec_ctx->pix_fmt;
  159. /* video time_base can be set to whatever is handy and supported by encoder */
  160. enc_ctx->time_base = av_inv_q(dec_ctx->framerate);
  161. } else {
  162. enc_ctx->sample_rate = dec_ctx->sample_rate;
  163. ret = av_channel_layout_copy(&enc_ctx->ch_layout, &dec_ctx->ch_layout);
  164. if (ret < 0)
  165. return ret;
  166. /* take first format from list of supported formats */
  167. enc_ctx->sample_fmt = encoder->sample_fmts[0];
  168. enc_ctx->time_base = (AVRational){1, enc_ctx->sample_rate};
  169. }
  170. if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
  171. enc_ctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
  172. /* Third parameter can be used to pass settings to encoder */
  173. ret = avcodec_open2(enc_ctx, encoder, NULL);
  174. if (ret < 0) {
  175. av_log(NULL, AV_LOG_ERROR, "Cannot open video encoder for stream #%u\n", i);
  176. return ret;
  177. }
  178. ret = avcodec_parameters_from_context(out_stream->codecpar, enc_ctx);
  179. if (ret < 0) {
  180. av_log(NULL, AV_LOG_ERROR, "Failed to copy encoder parameters to output stream #%u\n", i);
  181. return ret;
  182. }
  183. out_stream->time_base = enc_ctx->time_base;
  184. stream_ctx[i].enc_ctx = enc_ctx;
  185. } else if (dec_ctx->codec_type == AVMEDIA_TYPE_UNKNOWN) {
  186. av_log(NULL, AV_LOG_FATAL, "Elementary stream #%d is of unknown type, cannot proceed\n", i);
  187. return AVERROR_INVALIDDATA;
  188. } else {
  189. /* if this stream must be remuxed */
  190. ret = avcodec_parameters_copy(out_stream->codecpar, in_stream->codecpar);
  191. if (ret < 0) {
  192. av_log(NULL, AV_LOG_ERROR, "Copying parameters for stream #%u failed\n", i);
  193. return ret;
  194. }
  195. out_stream->time_base = in_stream->time_base;
  196. }
  197. }
  198. av_dump_format(ofmt_ctx, 0, filename, 1);
  199. if (!(ofmt_ctx->oformat->flags & AVFMT_NOFILE)) {
  200. ret = avio_open(&ofmt_ctx->pb, filename, AVIO_FLAG_WRITE);
  201. if (ret < 0) {
  202. av_log(NULL, AV_LOG_ERROR, "Could not open output file '%s'", filename);
  203. return ret;
  204. }
  205. }
  206. /* init muxer, write output file header */
  207. ret = avformat_write_header(ofmt_ctx, NULL);
  208. if (ret < 0) {
  209. av_log(NULL, AV_LOG_ERROR, "Error occurred when opening output file\n");
  210. return ret;
  211. }
  212. return 0;
  213. }
  214. static int init_filter(FilteringContext* fctx, AVCodecContext *dec_ctx,
  215. AVCodecContext *enc_ctx, const char *filter_spec)
  216. {
  217. char args[512];
  218. int ret = 0;
  219. const AVFilter *buffersrc = NULL;
  220. const AVFilter *buffersink = NULL;
  221. AVFilterContext *buffersrc_ctx = NULL;
  222. AVFilterContext *buffersink_ctx = NULL;
  223. AVFilterInOut *outputs = avfilter_inout_alloc();
  224. AVFilterInOut *inputs = avfilter_inout_alloc();
  225. AVFilterGraph *filter_graph = avfilter_graph_alloc();
  226. if (!outputs || !inputs || !filter_graph) {
  227. ret = AVERROR(ENOMEM);
  228. goto end;
  229. }
  230. if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {
  231. buffersrc = avfilter_get_by_name("buffer");
  232. buffersink = avfilter_get_by_name("buffersink");
  233. if (!buffersrc || !buffersink) {
  234. av_log(NULL, AV_LOG_ERROR, "filtering source or sink element not found\n");
  235. ret = AVERROR_UNKNOWN;
  236. goto end;
  237. }
  238. snprintf(args, sizeof(args),
  239. "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d",
  240. dec_ctx->width, dec_ctx->height, dec_ctx->pix_fmt,
  241. dec_ctx->pkt_timebase.num, dec_ctx->pkt_timebase.den,
  242. dec_ctx->sample_aspect_ratio.num,
  243. dec_ctx->sample_aspect_ratio.den);
  244. ret = avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in",
  245. args, NULL, filter_graph);
  246. if (ret < 0) {
  247. av_log(NULL, AV_LOG_ERROR, "Cannot create buffer source\n");
  248. goto end;
  249. }
  250. ret = avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out",
  251. NULL, NULL, filter_graph);
  252. if (ret < 0) {
  253. av_log(NULL, AV_LOG_ERROR, "Cannot create buffer sink\n");
  254. goto end;
  255. }
  256. ret = av_opt_set_bin(buffersink_ctx, "pix_fmts",
  257. (uint8_t*)&enc_ctx->pix_fmt, sizeof(enc_ctx->pix_fmt),
  258. AV_OPT_SEARCH_CHILDREN);
  259. if (ret < 0) {
  260. av_log(NULL, AV_LOG_ERROR, "Cannot set output pixel format\n");
  261. goto end;
  262. }
  263. } else if (dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {
  264. char buf[64];
  265. buffersrc = avfilter_get_by_name("abuffer");
  266. buffersink = avfilter_get_by_name("abuffersink");
  267. if (!buffersrc || !buffersink) {
  268. av_log(NULL, AV_LOG_ERROR, "filtering source or sink element not found\n");
  269. ret = AVERROR_UNKNOWN;
  270. goto end;
  271. }
  272. if (dec_ctx->ch_layout.order == AV_CHANNEL_ORDER_UNSPEC)
  273. av_channel_layout_default(&dec_ctx->ch_layout, dec_ctx->ch_layout.nb_channels);
  274. av_channel_layout_describe(&dec_ctx->ch_layout, buf, sizeof(buf));
  275. snprintf(args, sizeof(args),
  276. "time_base=%d/%d:sample_rate=%d:sample_fmt=%s:channel_layout=%s",
  277. dec_ctx->pkt_timebase.num, dec_ctx->pkt_timebase.den, dec_ctx->sample_rate,
  278. av_get_sample_fmt_name(dec_ctx->sample_fmt),
  279. buf);
  280. ret = avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in",
  281. args, NULL, filter_graph);
  282. if (ret < 0) {
  283. av_log(NULL, AV_LOG_ERROR, "Cannot create audio buffer source\n");
  284. goto end;
  285. }
  286. ret = avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out",
  287. NULL, NULL, filter_graph);
  288. if (ret < 0) {
  289. av_log(NULL, AV_LOG_ERROR, "Cannot create audio buffer sink\n");
  290. goto end;
  291. }
  292. ret = av_opt_set_bin(buffersink_ctx, "sample_fmts",
  293. (uint8_t*)&enc_ctx->sample_fmt, sizeof(enc_ctx->sample_fmt),
  294. AV_OPT_SEARCH_CHILDREN);
  295. if (ret < 0) {
  296. av_log(NULL, AV_LOG_ERROR, "Cannot set output sample format\n");
  297. goto end;
  298. }
  299. av_channel_layout_describe(&enc_ctx->ch_layout, buf, sizeof(buf));
  300. ret = av_opt_set(buffersink_ctx, "ch_layouts",
  301. buf, AV_OPT_SEARCH_CHILDREN);
  302. if (ret < 0) {
  303. av_log(NULL, AV_LOG_ERROR, "Cannot set output channel layout\n");
  304. goto end;
  305. }
  306. ret = av_opt_set_bin(buffersink_ctx, "sample_rates",
  307. (uint8_t*)&enc_ctx->sample_rate, sizeof(enc_ctx->sample_rate),
  308. AV_OPT_SEARCH_CHILDREN);
  309. if (ret < 0) {
  310. av_log(NULL, AV_LOG_ERROR, "Cannot set output sample rate\n");
  311. goto end;
  312. }
  313. } else {
  314. ret = AVERROR_UNKNOWN;
  315. goto end;
  316. }
  317. /* Endpoints for the filter graph. */
  318. outputs->name = av_strdup("in");
  319. outputs->filter_ctx = buffersrc_ctx;
  320. outputs->pad_idx = 0;
  321. outputs->next = NULL;
  322. inputs->name = av_strdup("out");
  323. inputs->filter_ctx = buffersink_ctx;
  324. inputs->pad_idx = 0;
  325. inputs->next = NULL;
  326. if (!outputs->name || !inputs->name) {
  327. ret = AVERROR(ENOMEM);
  328. goto end;
  329. }
  330. if ((ret = avfilter_graph_parse_ptr(filter_graph, filter_spec,
  331. &inputs, &outputs, NULL)) < 0)
  332. goto end;
  333. if ((ret = avfilter_graph_config(filter_graph, NULL)) < 0)
  334. goto end;
  335. /* Fill FilteringContext */
  336. fctx->buffersrc_ctx = buffersrc_ctx;
  337. fctx->buffersink_ctx = buffersink_ctx;
  338. fctx->filter_graph = filter_graph;
  339. end:
  340. avfilter_inout_free(&inputs);
  341. avfilter_inout_free(&outputs);
  342. return ret;
  343. }
  344. static int init_filters(void)
  345. {
  346. const char *filter_spec;
  347. unsigned int i;
  348. int ret;
  349. filter_ctx = av_malloc_array(ifmt_ctx->nb_streams, sizeof(*filter_ctx));
  350. if (!filter_ctx)
  351. return AVERROR(ENOMEM);
  352. for (i = 0; i < ifmt_ctx->nb_streams; i++) {
  353. filter_ctx[i].buffersrc_ctx = NULL;
  354. filter_ctx[i].buffersink_ctx = NULL;
  355. filter_ctx[i].filter_graph = NULL;
  356. if (!(ifmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO
  357. || ifmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO))
  358. continue;
  359. if (ifmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
  360. filter_spec = "null"; /* passthrough (dummy) filter for video */
  361. else
  362. filter_spec = "anull"; /* passthrough (dummy) filter for audio */
  363. ret = init_filter(&filter_ctx[i], stream_ctx[i].dec_ctx,
  364. stream_ctx[i].enc_ctx, filter_spec);
  365. if (ret)
  366. return ret;
  367. filter_ctx[i].enc_pkt = av_packet_alloc();
  368. if (!filter_ctx[i].enc_pkt)
  369. return AVERROR(ENOMEM);
  370. filter_ctx[i].filtered_frame = av_frame_alloc();
  371. if (!filter_ctx[i].filtered_frame)
  372. return AVERROR(ENOMEM);
  373. }
  374. return 0;
  375. }
  376. static int encode_write_frame(unsigned int stream_index, int flush)
  377. {
  378. StreamContext *stream = &stream_ctx[stream_index];
  379. FilteringContext *filter = &filter_ctx[stream_index];
  380. AVFrame *filt_frame = flush ? NULL : filter->filtered_frame;
  381. AVPacket *enc_pkt = filter->enc_pkt;
  382. int ret;
  383. av_log(NULL, AV_LOG_INFO, "Encoding frame\n");
  384. /* encode filtered frame */
  385. av_packet_unref(enc_pkt);
  386. if (filt_frame && filt_frame->pts != AV_NOPTS_VALUE)
  387. filt_frame->pts = av_rescale_q(filt_frame->pts, filt_frame->time_base,
  388. stream->enc_ctx->time_base);
  389. ret = avcodec_send_frame(stream->enc_ctx, filt_frame);
  390. if (ret < 0)
  391. return ret;
  392. while (ret >= 0) {
  393. ret = avcodec_receive_packet(stream->enc_ctx, enc_pkt);
  394. if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
  395. return 0;
  396. /* prepare packet for muxing */
  397. enc_pkt->stream_index = stream_index;
  398. av_packet_rescale_ts(enc_pkt,
  399. stream->enc_ctx->time_base,
  400. ofmt_ctx->streams[stream_index]->time_base);
  401. av_log(NULL, AV_LOG_DEBUG, "Muxing frame\n");
  402. /* mux encoded frame */
  403. ret = av_interleaved_write_frame(ofmt_ctx, enc_pkt);
  404. }
  405. return ret;
  406. }
  407. static int filter_encode_write_frame(AVFrame *frame, unsigned int stream_index)
  408. {
  409. FilteringContext *filter = &filter_ctx[stream_index];
  410. int ret;
  411. av_log(NULL, AV_LOG_INFO, "Pushing decoded frame to filters\n");
  412. /* push the decoded frame into the filtergraph */
  413. ret = av_buffersrc_add_frame_flags(filter->buffersrc_ctx,
  414. frame, 0);
  415. if (ret < 0) {
  416. av_log(NULL, AV_LOG_ERROR, "Error while feeding the filtergraph\n");
  417. return ret;
  418. }
  419. /* pull filtered frames from the filtergraph */
  420. while (1) {
  421. av_log(NULL, AV_LOG_INFO, "Pulling filtered frame from filters\n");
  422. ret = av_buffersink_get_frame(filter->buffersink_ctx,
  423. filter->filtered_frame);
  424. if (ret < 0) {
  425. /* if no more frames for output - returns AVERROR(EAGAIN)
  426. * if flushed and no more frames for output - returns AVERROR_EOF
  427. * rewrite retcode to 0 to show it as normal procedure completion
  428. */
  429. if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
  430. ret = 0;
  431. break;
  432. }
  433. filter->filtered_frame->time_base = av_buffersink_get_time_base(filter->buffersink_ctx);;
  434. filter->filtered_frame->pict_type = AV_PICTURE_TYPE_NONE;
  435. ret = encode_write_frame(stream_index, 0);
  436. av_frame_unref(filter->filtered_frame);
  437. if (ret < 0)
  438. break;
  439. }
  440. return ret;
  441. }
  442. static int flush_encoder(unsigned int stream_index)
  443. {
  444. if (!(stream_ctx[stream_index].enc_ctx->codec->capabilities &
  445. AV_CODEC_CAP_DELAY))
  446. return 0;
  447. av_log(NULL, AV_LOG_INFO, "Flushing stream #%u encoder\n", stream_index);
  448. return encode_write_frame(stream_index, 1);
  449. }
  450. int main(int argc, char **argv)
  451. {
  452. int ret;
  453. AVPacket *packet = NULL;
  454. unsigned int stream_index;
  455. unsigned int i;
  456. if (argc != 3) {
  457. av_log(NULL, AV_LOG_ERROR, "Usage: %s <input file> <output file>\n", argv[0]);
  458. return 1;
  459. }
  460. if ((ret = open_input_file(argv[1])) < 0)
  461. goto end;
  462. if ((ret = open_output_file(argv[2])) < 0)
  463. goto end;
  464. if ((ret = init_filters()) < 0)
  465. goto end;
  466. if (!(packet = av_packet_alloc()))
  467. goto end;
  468. /* read all packets */
  469. while (1) {
  470. if ((ret = av_read_frame(ifmt_ctx, packet)) < 0)
  471. break;
  472. stream_index = packet->stream_index;
  473. av_log(NULL, AV_LOG_DEBUG, "Demuxer gave frame of stream_index %u\n",
  474. stream_index);
  475. if (filter_ctx[stream_index].filter_graph) {
  476. StreamContext *stream = &stream_ctx[stream_index];
  477. av_log(NULL, AV_LOG_DEBUG, "Going to reencode&filter the frame\n");
  478. ret = avcodec_send_packet(stream->dec_ctx, packet);
  479. if (ret < 0) {
  480. av_log(NULL, AV_LOG_ERROR, "Decoding failed\n");
  481. break;
  482. }
  483. while (ret >= 0) {
  484. ret = avcodec_receive_frame(stream->dec_ctx, stream->dec_frame);
  485. if (ret == AVERROR_EOF || ret == AVERROR(EAGAIN))
  486. break;
  487. else if (ret < 0)
  488. goto end;
  489. stream->dec_frame->pts = stream->dec_frame->best_effort_timestamp;
  490. ret = filter_encode_write_frame(stream->dec_frame, stream_index);
  491. if (ret < 0)
  492. goto end;
  493. }
  494. } else {
  495. /* remux this frame without reencoding */
  496. av_packet_rescale_ts(packet,
  497. ifmt_ctx->streams[stream_index]->time_base,
  498. ofmt_ctx->streams[stream_index]->time_base);
  499. ret = av_interleaved_write_frame(ofmt_ctx, packet);
  500. if (ret < 0)
  501. goto end;
  502. }
  503. av_packet_unref(packet);
  504. }
  505. /* flush decoders, filters and encoders */
  506. for (i = 0; i < ifmt_ctx->nb_streams; i++) {
  507. StreamContext *stream;
  508. if (!filter_ctx[i].filter_graph)
  509. continue;
  510. stream = &stream_ctx[i];
  511. av_log(NULL, AV_LOG_INFO, "Flushing stream %u decoder\n", i);
  512. /* flush decoder */
  513. ret = avcodec_send_packet(stream->dec_ctx, NULL);
  514. if (ret < 0) {
  515. av_log(NULL, AV_LOG_ERROR, "Flushing decoding failed\n");
  516. goto end;
  517. }
  518. while (ret >= 0) {
  519. ret = avcodec_receive_frame(stream->dec_ctx, stream->dec_frame);
  520. if (ret == AVERROR_EOF)
  521. break;
  522. else if (ret < 0)
  523. goto end;
  524. stream->dec_frame->pts = stream->dec_frame->best_effort_timestamp;
  525. ret = filter_encode_write_frame(stream->dec_frame, i);
  526. if (ret < 0)
  527. goto end;
  528. }
  529. /* flush filter */
  530. ret = filter_encode_write_frame(NULL, i);
  531. if (ret < 0) {
  532. av_log(NULL, AV_LOG_ERROR, "Flushing filter failed\n");
  533. goto end;
  534. }
  535. /* flush encoder */
  536. ret = flush_encoder(i);
  537. if (ret < 0) {
  538. av_log(NULL, AV_LOG_ERROR, "Flushing encoder failed\n");
  539. goto end;
  540. }
  541. }
  542. av_write_trailer(ofmt_ctx);
  543. end:
  544. av_packet_free(&packet);
  545. for (i = 0; i < ifmt_ctx->nb_streams; i++) {
  546. avcodec_free_context(&stream_ctx[i].dec_ctx);
  547. if (ofmt_ctx && ofmt_ctx->nb_streams > i && ofmt_ctx->streams[i] && stream_ctx[i].enc_ctx)
  548. avcodec_free_context(&stream_ctx[i].enc_ctx);
  549. if (filter_ctx && filter_ctx[i].filter_graph) {
  550. avfilter_graph_free(&filter_ctx[i].filter_graph);
  551. av_packet_free(&filter_ctx[i].enc_pkt);
  552. av_frame_free(&filter_ctx[i].filtered_frame);
  553. }
  554. av_frame_free(&stream_ctx[i].dec_frame);
  555. }
  556. av_free(filter_ctx);
  557. av_free(stream_ctx);
  558. avformat_close_input(&ifmt_ctx);
  559. if (ofmt_ctx && !(ofmt_ctx->oformat->flags & AVFMT_NOFILE))
  560. avio_closep(&ofmt_ctx->pb);
  561. avformat_free_context(ofmt_ctx);
  562. if (ret < 0)
  563. av_log(NULL, AV_LOG_ERROR, "Error occurred: %s\n", av_err2str(ret));
  564. return ret ? 1 : 0;
  565. }