2011-07-27 21:56:59 +03:00
/*
* avconv main
* Copyright ( c ) 2000 - 2011 The libav developers .
*
* This file is part of Libav .
*
* Libav is free software ; you can redistribute it and / or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
* version 2.1 of the License , or ( at your option ) any later version .
*
* Libav is distributed in the hope that it will be useful ,
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
*
* You should have received a copy of the GNU Lesser General Public
* License along with Libav ; if not , write to the Free Software
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
*/
# include "config.h"
# include <ctype.h>
# include <string.h>
# include <math.h>
# include <stdlib.h>
# include <errno.h>
# include <signal.h>
# include <limits.h>
# include <unistd.h>
# include "libavformat/avformat.h"
# include "libavdevice/avdevice.h"
# include "libswscale/swscale.h"
2012-04-05 21:06:28 +03:00
# include "libavresample/avresample.h"
2011-07-27 21:56:59 +03:00
# include "libavutil/opt.h"
# include "libavutil/audioconvert.h"
# include "libavutil/parseutils.h"
# include "libavutil/samplefmt.h"
# include "libavutil/colorspace.h"
# include "libavutil/fifo.h"
# include "libavutil/intreadwrite.h"
# include "libavutil/dict.h"
# include "libavutil/mathematics.h"
# include "libavutil/pixdesc.h"
# include "libavutil/avstring.h"
# include "libavutil/libm.h"
2011-12-19 17:44:58 +03:00
# include "libavutil/imgutils.h"
2011-07-27 21:56:59 +03:00
# include "libavformat/os_support.h"
# include "libavfilter / avfilter.h"
# include "libavfilter / avfiltergraph.h"
2011-12-21 23:04:05 +03:00
# include "libavfilter / buffersrc.h"
2011-07-27 21:56:59 +03:00
# include "libavfilter / vsrc_buffer.h"
# if HAVE_SYS_RESOURCE_H
# include <sys/types.h>
# include <sys/time.h>
# include <sys/resource.h>
# elif HAVE_GETPROCESSTIMES
# include <windows.h>
# endif
# if HAVE_GETPROCESSMEMORYINFO
# include <windows.h>
# include <psapi.h>
# endif
# if HAVE_SYS_SELECT_H
# include <sys/select.h>
# endif
# include <time.h>
# include "cmdutils.h"
# include "libavutil/avassert.h"
2011-11-26 13:01:33 +03:00
# define VSYNC_AUTO -1
# define VSYNC_PASSTHROUGH 0
# define VSYNC_CFR 1
# define VSYNC_VFR 2
2011-07-27 21:56:59 +03:00
const char program_name [ ] = " avconv " ;
const int program_birth_year = 2000 ;
/* select an input stream for an output stream */
typedef struct StreamMap {
2011-07-29 16:21:51 +03:00
int disabled ; /** 1 is this mapping is disabled by a negative map */
2011-07-27 21:56:59 +03:00
int file_index ;
int stream_index ;
int sync_file_index ;
int sync_stream_index ;
2012-04-01 16:08:33 +03:00
char * linklabel ; /** name of an output link, for mapping lavfi outputs */
2011-07-27 21:56:59 +03:00
} StreamMap ;
/**
* select an input file for an output file
*/
typedef struct MetadataMap {
2011-08-05 14:15:04 +03:00
int file ; ///< file index
char type ; ///< type of metadata to copy -- (g)lobal, (s)tream, (c)hapter or (p)rogram
int index ; ///< stream/chapter/program number
2011-07-27 21:56:59 +03:00
} MetadataMap ;
static const OptionDef options [ ] ;
static int video_discard = 0 ;
2011-08-09 12:32:16 +03:00
static int same_quant = 0 ;
2011-07-27 21:56:59 +03:00
static int do_deinterlace = 0 ;
static int intra_dc_precision = 8 ;
static int qp_hist = 0 ;
static int file_overwrite = 0 ;
static int do_benchmark = 0 ;
static int do_hex_dump = 0 ;
static int do_pkt_dump = 0 ;
static int do_pass = 0 ;
static char * pass_logfilename_prefix = NULL ;
2011-11-26 13:01:33 +03:00
static int video_sync_method = VSYNC_AUTO ;
2011-12-29 23:28:29 +03:00
static int audio_sync_method = 0 ;
static float audio_drift_threshold = 0.1 ;
static int copy_ts = 0 ;
2011-11-29 13:10:31 +03:00
static int copy_tb = 1 ;
2011-07-27 21:56:59 +03:00
static int opt_shortest = 0 ;
static char * vstats_filename ;
static FILE * vstats_file ;
static int audio_volume = 256 ;
static int exit_on_error = 0 ;
static int using_stdin = 0 ;
static int64_t video_size = 0 ;
static int64_t audio_size = 0 ;
static int64_t extra_size = 0 ;
static int nb_frames_dup = 0 ;
static int nb_frames_drop = 0 ;
static int input_sync ;
static float dts_delta_threshold = 10 ;
2011-10-09 16:57:30 +03:00
static int print_stats = 1 ;
2011-07-27 21:56:59 +03:00
static uint8_t * audio_buf ;
2011-12-31 17:36:25 +03:00
static unsigned int allocated_audio_buf_size ;
2012-03-21 23:36:23 +03:00
static uint8_t * async_buf ;
static unsigned int allocated_async_buf_size ;
2011-07-27 21:56:59 +03:00
# define DEFAULT_PASS_LOGFILENAME_PREFIX "av2pass"
2012-03-29 09:51:17 +03:00
typedef struct InputFilter {
AVFilterContext * filter ;
struct InputStream * ist ;
struct FilterGraph * graph ;
} InputFilter ;
typedef struct OutputFilter {
AVFilterContext * filter ;
struct OutputStream * ost ;
struct FilterGraph * graph ;
2012-04-01 16:08:33 +03:00
/* temporary storage until stream maps are processed */
AVFilterInOut * out_tmp ;
2012-03-29 09:51:17 +03:00
} OutputFilter ;
typedef struct FilterGraph {
2012-04-01 16:08:33 +03:00
int index ;
const char * graph_desc ;
2012-03-29 09:51:17 +03:00
AVFilterGraph * graph ;
InputFilter * * inputs ;
int nb_inputs ;
OutputFilter * * outputs ;
int nb_outputs ;
} FilterGraph ;
2011-12-19 17:44:58 +03:00
typedef struct FrameBuffer {
uint8_t * base [ 4 ] ;
uint8_t * data [ 4 ] ;
int linesize [ 4 ] ;
int h , w ;
enum PixelFormat pix_fmt ;
int refcount ;
struct InputStream * ist ;
struct FrameBuffer * next ;
} FrameBuffer ;
2011-08-17 19:40:40 +03:00
typedef struct InputStream {
int file_index ;
AVStream * st ;
int discard ; /* true if stream data should be discarded */
int decoding_needed ; /* true if the packets must be decoded in 'raw_fifo' */
AVCodec * dec ;
2011-12-05 19:49:38 +03:00
AVFrame * decoded_frame ;
2011-08-17 19:40:40 +03:00
int64_t start ; /* time when read started */
2012-02-03 17:32:51 +03:00
/* predicted dts of the next packet read for this stream or (when there are
* several frames in a packet ) of the next frame in current packet */
int64_t next_dts ;
2012-02-05 23:12:43 +03:00
/* dts of the last packet read for this stream */
int64_t last_dts ;
2011-08-17 19:40:40 +03:00
PtsCorrectionContext pts_ctx ;
double ts_scale ;
int is_start ; /* is 1 at the start and after a discontinuity */
int showed_multi_packet_warning ;
AVDictionary * opts ;
2011-12-19 17:44:58 +03:00
2012-03-29 08:40:26 +03:00
int resample_height ;
int resample_width ;
int resample_pix_fmt ;
2011-12-19 17:44:58 +03:00
/* a pool of free buffers for decoded data */
FrameBuffer * buffer_pool ;
2012-03-29 09:51:17 +03:00
/* decoded data from this stream goes into all those filters
* currently video only */
InputFilter * * filters ;
int nb_filters ;
2011-08-17 19:40:40 +03:00
} InputStream ;
typedef struct InputFile {
AVFormatContext * ctx ;
int eof_reached ; /* true if eof reached */
int ist_index ; /* index of first stream in ist_table */
int buffer_size ; /* current total buffer size */
int64_t ts_offset ;
2011-08-13 13:20:50 +03:00
int nb_streams ; /* number of stream that avconv is aware of; may be different
from ctx . nb_streams if new streams appear during av_read_frame ( ) */
2011-08-29 10:16:42 +03:00
int rate_emu ;
2011-08-17 19:40:40 +03:00
} InputFile ;
2011-07-27 21:56:59 +03:00
typedef struct OutputStream {
int file_index ; /* file index */
int index ; /* stream index in the output file */
int source_index ; /* InputStream index */
AVStream * st ; /* stream in the output file */
int encoding_needed ; /* true if encoding needed for this stream */
int frame_number ;
/* input pts and corresponding output pts
for A / V sync */
2011-12-29 23:28:29 +03:00
// double sync_ipts; /* dts from the AVPacket of the demuxer in second units */
2011-07-27 21:56:59 +03:00
struct InputStream * sync_ist ; /* input stream to sync against */
2011-12-29 23:28:29 +03:00
int64_t sync_opts ; /* output frame counter, could be changed to some true timestamp */ // FIXME look at frame_number
2012-02-05 16:32:10 +03:00
/* pts of the first frame encoded for this stream, used for limiting
* recording time */
int64_t first_pts ;
2011-07-27 21:56:59 +03:00
AVBitStreamFilterContext * bitstream_filters ;
AVCodec * enc ;
2011-08-29 10:16:42 +03:00
int64_t max_frames ;
2011-12-31 17:36:25 +03:00
AVFrame * output_frame ;
2012-03-29 10:00:09 +03:00
AVFrame * filtered_frame ;
2011-07-27 21:56:59 +03:00
/* video only */
AVRational frame_rate ;
2011-09-01 09:20:50 +03:00
int force_fps ;
2011-09-02 19:03:31 +03:00
int top_field_first ;
2011-07-27 21:56:59 +03:00
float frame_aspect_ratio ;
2012-03-29 09:51:17 +03:00
float last_quality ;
2011-07-27 21:56:59 +03:00
/* forced key frames */
int64_t * forced_kf_pts ;
int forced_kf_count ;
int forced_kf_index ;
/* audio only */
int audio_resample ;
2012-04-05 21:06:28 +03:00
AVAudioResampleContext * avr ;
2011-07-27 21:56:59 +03:00
int resample_sample_fmt ;
int resample_channels ;
2012-04-05 21:06:28 +03:00
uint64_t resample_channel_layout ;
2011-07-27 21:56:59 +03:00
int resample_sample_rate ;
AVFifoBuffer * fifo ; /* for compression: one audio fifo per codec */
FILE * logfile ;
2012-03-29 09:51:17 +03:00
OutputFilter * filter ;
2011-07-27 21:56:59 +03:00
char * avfilter ;
2011-11-06 12:27:43 +03:00
int64_t sws_flags ;
AVDictionary * opts ;
int is_past_recording_time ;
int stream_copy ;
const char * attachment_filename ;
int copy_initial_nonkeyframes ;
2012-03-23 17:46:30 +03:00
enum PixelFormat pix_fmts [ 2 ] ;
2011-07-27 21:56:59 +03:00
} OutputStream ;
2011-08-17 10:56:08 +03:00
typedef struct OutputFile {
AVFormatContext * ctx ;
AVDictionary * opts ;
2011-08-17 11:21:37 +03:00
int ost_index ; /* index of the first stream in output_streams */
2011-08-17 12:42:44 +03:00
int64_t recording_time ; /* desired length of the resulting file in microseconds */
2011-08-17 12:42:44 +03:00
int64_t start_time ; /* start time in microseconds */
2011-08-19 18:52:44 +03:00
uint64_t limit_filesize ;
2011-08-17 10:56:08 +03:00
} OutputFile ;
2012-04-15 16:28:30 +03:00
static InputStream * * input_streams = NULL ;
static int nb_input_streams = 0 ;
static InputFile * * input_files = NULL ;
static int nb_input_files = 0 ;
2011-07-27 21:56:59 +03:00
2012-04-15 16:28:30 +03:00
static OutputStream * * output_streams = NULL ;
static int nb_output_streams = 0 ;
static OutputFile * * output_files = NULL ;
static int nb_output_files = 0 ;
2011-08-17 10:56:08 +03:00
2012-03-29 09:51:17 +03:00
static FilterGraph * * filtergraphs ;
int nb_filtergraphs ;
2011-08-28 18:21:56 +03:00
typedef struct OptionsContext {
2011-08-28 19:20:17 +03:00
/* input/output options */
int64_t start_time ;
2011-08-28 19:20:17 +03:00
const char * format ;
2011-08-28 19:20:17 +03:00
2011-08-29 10:16:42 +03:00
SpecifierOpt * codec_names ;
int nb_codec_names ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * audio_channels ;
int nb_audio_channels ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * audio_sample_rate ;
int nb_audio_sample_rate ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * frame_rates ;
int nb_frame_rates ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * frame_sizes ;
int nb_frame_sizes ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * frame_pix_fmts ;
int nb_frame_pix_fmts ;
2011-08-29 10:16:42 +03:00
2011-08-28 19:20:17 +03:00
/* input options */
int64_t input_ts_offset ;
2011-08-29 10:16:42 +03:00
int rate_emu ;
2011-08-28 19:20:17 +03:00
2011-08-29 10:16:42 +03:00
SpecifierOpt * ts_scale ;
int nb_ts_scale ;
2011-11-02 23:50:54 +03:00
SpecifierOpt * dump_attachment ;
int nb_dump_attachment ;
2011-08-29 10:16:42 +03:00
2011-08-28 18:21:56 +03:00
/* output options */
StreamMap * stream_maps ;
int nb_stream_maps ;
2011-08-29 10:16:42 +03:00
/* first item specifies output metadata, second is input */
MetadataMap ( * meta_data_maps ) [ 2 ] ;
int nb_meta_data_maps ;
int metadata_global_manual ;
int metadata_streams_manual ;
int metadata_chapters_manual ;
2011-10-30 09:10:08 +03:00
const char * * attachments ;
int nb_attachments ;
2011-08-28 19:20:17 +03:00
2011-08-29 10:16:42 +03:00
int chapters_input_file ;
2011-08-28 19:20:17 +03:00
int64_t recording_time ;
2011-08-28 19:20:17 +03:00
uint64_t limit_filesize ;
2011-08-31 09:51:15 +03:00
float mux_preload ;
float mux_max_delay ;
2011-08-29 10:16:42 +03:00
2011-08-31 09:51:15 +03:00
int video_disable ;
int audio_disable ;
int subtitle_disable ;
int data_disable ;
2011-09-02 23:07:30 +03:00
/* indexed by output file stream index */
int * streamid_map ;
int nb_streamid_map ;
2011-08-29 10:16:42 +03:00
SpecifierOpt * metadata ;
int nb_metadata ;
2011-08-29 10:16:42 +03:00
SpecifierOpt * max_frames ;
int nb_max_frames ;
2011-08-31 09:45:25 +03:00
SpecifierOpt * bitstream_filters ;
int nb_bitstream_filters ;
2011-08-31 09:51:15 +03:00
SpecifierOpt * codec_tags ;
int nb_codec_tags ;
2011-08-31 09:51:15 +03:00
SpecifierOpt * sample_fmts ;
int nb_sample_fmts ;
2011-09-01 11:04:35 +03:00
SpecifierOpt * qscale ;
int nb_qscale ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * forced_key_frames ;
int nb_forced_key_frames ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * force_fps ;
int nb_force_fps ;
2011-09-01 09:20:50 +03:00
SpecifierOpt * frame_aspect_ratios ;
int nb_frame_aspect_ratios ;
2011-09-02 17:06:18 +03:00
SpecifierOpt * rc_overrides ;
int nb_rc_overrides ;
2011-09-02 18:57:56 +03:00
SpecifierOpt * intra_matrices ;
int nb_intra_matrices ;
SpecifierOpt * inter_matrices ;
int nb_inter_matrices ;
2011-09-02 19:03:31 +03:00
SpecifierOpt * top_field_first ;
int nb_top_field_first ;
2011-12-13 13:23:06 +03:00
SpecifierOpt * metadata_map ;
int nb_metadata_map ;
2011-10-10 13:52:13 +03:00
SpecifierOpt * presets ;
int nb_presets ;
2011-11-06 12:26:56 +03:00
SpecifierOpt * copy_initial_nonkeyframes ;
int nb_copy_initial_nonkeyframes ;
2011-09-02 22:11:00 +03:00
SpecifierOpt * filters ;
int nb_filters ;
2011-08-28 18:21:56 +03:00
} OptionsContext ;
2011-08-29 10:16:42 +03:00
# define MATCH_PER_STREAM_OPT(name, type, outvar, fmtctx, st)\
{ \
int i , ret ; \
for ( i = 0 ; i < o - > nb_ # # name ; i + + ) { \
char * spec = o - > name [ i ] . specifier ; \
if ( ( ret = check_stream_specifier ( fmtctx , st , spec ) ) > 0 ) \
outvar = o - > name [ i ] . u . type ; \
else if ( ret < 0 ) \
exit_program ( 1 ) ; \
} \
}
2011-08-28 18:21:56 +03:00
static void reset_options ( OptionsContext * o )
{
const OptionDef * po = options ;
2012-04-01 16:08:33 +03:00
int i ;
2011-08-28 18:21:56 +03:00
/* all OPT_SPEC and OPT_STRING can be freed in generic way */
while ( po - > name ) {
void * dst = ( uint8_t * ) o + po - > u . off ;
if ( po - > flags & OPT_SPEC ) {
SpecifierOpt * * so = dst ;
int i , * count = ( int * ) ( so + 1 ) ;
for ( i = 0 ; i < * count ; i + + ) {
av_freep ( & ( * so ) [ i ] . specifier ) ;
if ( po - > flags & OPT_STRING )
av_freep ( & ( * so ) [ i ] . u . str ) ;
}
av_freep ( so ) ;
* count = 0 ;
} else if ( po - > flags & OPT_OFFSET & & po - > flags & OPT_STRING )
av_freep ( dst ) ;
po + + ;
}
2012-04-01 16:08:33 +03:00
for ( i = 0 ; i < o - > nb_stream_maps ; i + + )
av_freep ( & o - > stream_maps [ i ] . linklabel ) ;
2011-08-28 18:21:56 +03:00
av_freep ( & o - > stream_maps ) ;
2011-08-29 10:16:42 +03:00
av_freep ( & o - > meta_data_maps ) ;
2011-09-02 23:07:30 +03:00
av_freep ( & o - > streamid_map ) ;
2011-08-28 18:21:56 +03:00
memset ( o , 0 , sizeof ( * o ) ) ;
2011-08-28 19:20:17 +03:00
2011-08-31 09:51:15 +03:00
o - > mux_max_delay = 0.7 ;
2011-08-28 19:20:17 +03:00
o - > recording_time = INT64_MAX ;
2011-08-28 19:20:17 +03:00
o - > limit_filesize = UINT64_MAX ;
2011-08-29 10:16:42 +03:00
o - > chapters_input_file = INT_MAX ;
2011-08-28 19:20:17 +03:00
2011-08-28 18:21:56 +03:00
uninit_opts ( ) ;
init_opts ( ) ;
}
2012-03-24 01:03:03 +03:00
static int alloc_buffer ( InputStream * ist , AVCodecContext * s , FrameBuffer * * pbuf )
2011-12-19 17:44:58 +03:00
{
FrameBuffer * buf = av_mallocz ( sizeof ( * buf ) ) ;
2012-01-29 23:55:10 +03:00
int i , ret ;
2011-12-19 17:44:58 +03:00
const int pixel_size = av_pix_fmt_descriptors [ s - > pix_fmt ] . comp [ 0 ] . step_minus1 + 1 ;
int h_chroma_shift , v_chroma_shift ;
int edge = 32 ; // XXX should be avcodec_get_edge_width(), but that fails on svq1
int w = s - > width , h = s - > height ;
if ( ! buf )
return AVERROR ( ENOMEM ) ;
if ( ! ( s - > flags & CODEC_FLAG_EMU_EDGE ) ) {
w + = 2 * edge ;
h + = 2 * edge ;
}
avcodec_align_dimensions ( s , & w , & h ) ;
if ( ( ret = av_image_alloc ( buf - > base , buf - > linesize , w , h ,
s - > pix_fmt , 32 ) ) < 0 ) {
av_freep ( & buf ) ;
return ret ;
}
/* XXX this shouldn't be needed, but some tests break without this line
* those decoders are buggy and need to be fixed .
* the following tests fail :
2012-01-18 23:45:47 +03:00
* cdgraphics , ansi , aasc , fraps - v1 , qtrle - 1 bit
2011-12-19 17:44:58 +03:00
*/
memset ( buf - > base [ 0 ] , 128 , ret ) ;
avcodec_get_chroma_sub_sample ( s - > pix_fmt , & h_chroma_shift , & v_chroma_shift ) ;
2012-01-29 23:55:10 +03:00
for ( i = 0 ; i < FF_ARRAY_ELEMS ( buf - > data ) ; i + + ) {
2011-12-19 17:44:58 +03:00
const int h_shift = i = = 0 ? 0 : h_chroma_shift ;
const int v_shift = i = = 0 ? 0 : v_chroma_shift ;
if ( s - > flags & CODEC_FLAG_EMU_EDGE )
buf - > data [ i ] = buf - > base [ i ] ;
else
buf - > data [ i ] = buf - > base [ i ] +
FFALIGN ( ( buf - > linesize [ i ] * edge > > v_shift ) +
( pixel_size * edge > > h_shift ) , 32 ) ;
}
buf - > w = s - > width ;
buf - > h = s - > height ;
buf - > pix_fmt = s - > pix_fmt ;
buf - > ist = ist ;
* pbuf = buf ;
return 0 ;
}
static void free_buffer_pool ( InputStream * ist )
{
FrameBuffer * buf = ist - > buffer_pool ;
while ( buf ) {
ist - > buffer_pool = buf - > next ;
av_freep ( & buf - > base [ 0 ] ) ;
av_free ( buf ) ;
buf = ist - > buffer_pool ;
}
}
static void unref_buffer ( InputStream * ist , FrameBuffer * buf )
{
av_assert0 ( buf - > refcount ) ;
buf - > refcount - - ;
if ( ! buf - > refcount ) {
buf - > next = ist - > buffer_pool ;
ist - > buffer_pool = buf ;
}
}
static int codec_get_buffer ( AVCodecContext * s , AVFrame * frame )
{
InputStream * ist = s - > opaque ;
FrameBuffer * buf ;
int ret , i ;
2012-03-24 01:03:03 +03:00
if ( ! ist - > buffer_pool & & ( ret = alloc_buffer ( ist , s , & ist - > buffer_pool ) ) < 0 )
2011-12-19 17:44:58 +03:00
return ret ;
buf = ist - > buffer_pool ;
ist - > buffer_pool = buf - > next ;
buf - > next = NULL ;
if ( buf - > w ! = s - > width | | buf - > h ! = s - > height | | buf - > pix_fmt ! = s - > pix_fmt ) {
av_freep ( & buf - > base [ 0 ] ) ;
av_free ( buf ) ;
2012-03-24 01:03:03 +03:00
if ( ( ret = alloc_buffer ( ist , s , & buf ) ) < 0 )
2011-12-19 17:44:58 +03:00
return ret ;
}
buf - > refcount + + ;
frame - > opaque = buf ;
frame - > type = FF_BUFFER_TYPE_USER ;
frame - > extended_data = frame - > data ;
frame - > pkt_pts = s - > pkt ? s - > pkt - > pts : AV_NOPTS_VALUE ;
2012-03-24 01:03:03 +03:00
frame - > width = buf - > w ;
frame - > height = buf - > h ;
frame - > format = buf - > pix_fmt ;
frame - > sample_aspect_ratio = s - > sample_aspect_ratio ;
2011-12-19 17:44:58 +03:00
for ( i = 0 ; i < FF_ARRAY_ELEMS ( buf - > data ) ; i + + ) {
frame - > base [ i ] = buf - > base [ i ] ; // XXX h264.c uses base though it shouldn't
frame - > data [ i ] = buf - > data [ i ] ;
frame - > linesize [ i ] = buf - > linesize [ i ] ;
}
return 0 ;
}
static void codec_release_buffer ( AVCodecContext * s , AVFrame * frame )
{
InputStream * ist = s - > opaque ;
FrameBuffer * buf = frame - > opaque ;
int i ;
for ( i = 0 ; i < FF_ARRAY_ELEMS ( frame - > data ) ; i + + )
frame - > data [ i ] = NULL ;
unref_buffer ( ist , buf ) ;
}
2011-12-21 23:04:05 +03:00
static void filter_release_buffer ( AVFilterBuffer * fb )
{
FrameBuffer * buf = fb - > priv ;
av_free ( fb ) ;
unref_buffer ( buf - > ist , buf ) ;
}
2012-03-23 17:46:30 +03:00
static const enum PixelFormat * choose_pixel_fmts ( OutputStream * ost )
{
if ( ost - > st - > codec - > pix_fmt ! = PIX_FMT_NONE ) {
ost - > pix_fmts [ 0 ] = ost - > st - > codec - > pix_fmt ;
return ost - > pix_fmts ;
} else if ( ost - > enc - > pix_fmts )
return ost - > enc - > pix_fmts ;
else
return NULL ;
}
2012-03-29 09:51:17 +03:00
static int configure_video_filters ( FilterGraph * fg )
2011-07-27 21:56:59 +03:00
{
2012-03-29 09:51:17 +03:00
InputStream * ist = fg - > inputs [ 0 ] - > ist ;
OutputStream * ost = fg - > outputs [ 0 ] - > ost ;
2011-07-27 21:56:59 +03:00
AVFilterContext * last_filter , * filter ;
AVCodecContext * codec = ost - > st - > codec ;
2012-03-23 17:46:30 +03:00
SinkContext sink_ctx = { . pix_fmts = choose_pixel_fmts ( ost ) } ;
2011-07-27 21:56:59 +03:00
AVRational sample_aspect_ratio ;
char args [ 255 ] ;
int ret ;
2012-03-29 09:51:17 +03:00
avfilter_graph_free ( & fg - > graph ) ;
fg - > graph = avfilter_graph_alloc ( ) ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
if ( ist - > st - > sample_aspect_ratio . num ) {
2011-07-27 21:56:59 +03:00
sample_aspect_ratio = ist - > st - > sample_aspect_ratio ;
2011-12-29 23:28:29 +03:00
} else
2011-07-27 21:56:59 +03:00
sample_aspect_ratio = ist - > st - > codec - > sample_aspect_ratio ;
snprintf ( args , 255 , " %d:%d:%d:%d:%d:%d:%d " , ist - > st - > codec - > width ,
ist - > st - > codec - > height , ist - > st - > codec - > pix_fmt , 1 , AV_TIME_BASE ,
sample_aspect_ratio . num , sample_aspect_ratio . den ) ;
2012-03-29 09:51:17 +03:00
ret = avfilter_graph_create_filter ( & fg - > inputs [ 0 ] - > filter ,
avfilter_get_by_name ( " buffer " ) ,
" src " , args , NULL , fg - > graph ) ;
2011-07-27 21:56:59 +03:00
if ( ret < 0 )
return ret ;
2012-03-29 09:51:17 +03:00
ret = avfilter_graph_create_filter ( & fg - > outputs [ 0 ] - > filter , & sink ,
" out " , NULL , & sink_ctx , fg - > graph ) ;
2011-07-27 21:56:59 +03:00
if ( ret < 0 )
return ret ;
2012-03-29 09:51:17 +03:00
last_filter = fg - > inputs [ 0 ] - > filter ;
2011-07-27 21:56:59 +03:00
2012-03-29 08:48:27 +03:00
if ( codec - > width | | codec - > height ) {
2011-07-27 21:56:59 +03:00
snprintf ( args , 255 , " %d:%d:flags=0x%X " ,
codec - > width ,
codec - > height ,
2011-10-03 22:10:06 +03:00
( unsigned ) ost - > sws_flags ) ;
2011-07-27 21:56:59 +03:00
if ( ( ret = avfilter_graph_create_filter ( & filter , avfilter_get_by_name ( " scale " ) ,
2012-03-29 09:51:17 +03:00
NULL , args , NULL , fg - > graph ) ) < 0 )
2011-07-27 21:56:59 +03:00
return ret ;
if ( ( ret = avfilter_link ( last_filter , 0 , filter , 0 ) ) < 0 )
return ret ;
last_filter = filter ;
}
2011-10-03 22:10:06 +03:00
snprintf ( args , sizeof ( args ) , " flags=0x%X " , ( unsigned ) ost - > sws_flags ) ;
2012-03-29 09:51:17 +03:00
fg - > graph - > scale_sws_opts = av_strdup ( args ) ;
2011-07-27 21:56:59 +03:00
if ( ost - > avfilter ) {
2011-06-11 16:30:46 +03:00
AVFilterInOut * outputs = avfilter_inout_alloc ( ) ;
AVFilterInOut * inputs = avfilter_inout_alloc ( ) ;
2011-07-27 21:56:59 +03:00
outputs - > name = av_strdup ( " in " ) ;
outputs - > filter_ctx = last_filter ;
outputs - > pad_idx = 0 ;
outputs - > next = NULL ;
inputs - > name = av_strdup ( " out " ) ;
2012-03-29 09:51:17 +03:00
inputs - > filter_ctx = fg - > outputs [ 0 ] - > filter ;
2011-07-27 21:56:59 +03:00
inputs - > pad_idx = 0 ;
inputs - > next = NULL ;
2012-03-29 09:51:17 +03:00
if ( ( ret = avfilter_graph_parse ( fg - > graph , ost - > avfilter , inputs , outputs , NULL ) ) < 0 )
2011-07-27 21:56:59 +03:00
return ret ;
} else {
2012-03-29 09:51:17 +03:00
if ( ( ret = avfilter_link ( last_filter , 0 , fg - > outputs [ 0 ] - > filter , 0 ) ) < 0 )
2011-07-27 21:56:59 +03:00
return ret ;
}
2012-03-29 09:51:17 +03:00
if ( ( ret = avfilter_graph_config ( fg - > graph , NULL ) ) < 0 )
2011-07-27 21:56:59 +03:00
return ret ;
2012-03-29 09:51:17 +03:00
ost - > filter = fg - > outputs [ 0 ] ;
2011-07-27 21:56:59 +03:00
return 0 ;
}
2012-03-29 09:51:17 +03:00
static FilterGraph * init_simple_filtergraph ( InputStream * ist , OutputStream * ost )
{
FilterGraph * fg = av_mallocz ( sizeof ( * fg ) ) ;
if ( ! fg )
exit_program ( 1 ) ;
2012-04-01 16:08:33 +03:00
fg - > index = nb_filtergraphs ;
2012-03-29 09:51:17 +03:00
fg - > outputs = grow_array ( fg - > outputs , sizeof ( * fg - > outputs ) , & fg - > nb_outputs ,
fg - > nb_outputs + 1 ) ;
if ( ! ( fg - > outputs [ 0 ] = av_mallocz ( sizeof ( * fg - > outputs [ 0 ] ) ) ) )
exit_program ( 1 ) ;
fg - > outputs [ 0 ] - > ost = ost ;
fg - > outputs [ 0 ] - > graph = fg ;
fg - > inputs = grow_array ( fg - > inputs , sizeof ( * fg - > inputs ) , & fg - > nb_inputs ,
fg - > nb_inputs + 1 ) ;
if ( ! ( fg - > inputs [ 0 ] = av_mallocz ( sizeof ( * fg - > inputs [ 0 ] ) ) ) )
exit_program ( 1 ) ;
fg - > inputs [ 0 ] - > ist = ist ;
fg - > inputs [ 0 ] - > graph = fg ;
ist - > filters = grow_array ( ist - > filters , sizeof ( * ist - > filters ) ,
& ist - > nb_filters , ist - > nb_filters + 1 ) ;
ist - > filters [ ist - > nb_filters - 1 ] = fg - > inputs [ 0 ] ;
filtergraphs = grow_array ( filtergraphs , sizeof ( * filtergraphs ) ,
& nb_filtergraphs , nb_filtergraphs + 1 ) ;
filtergraphs [ nb_filtergraphs - 1 ] = fg ;
return fg ;
}
2012-04-01 16:08:33 +03:00
static void init_input_filter ( FilterGraph * fg , AVFilterInOut * in )
{
InputStream * ist ;
enum AVMediaType type = in - > filter_ctx - > input_pads [ in - > pad_idx ] . type ;
int i ;
// TODO: support other filter types
if ( type ! = AVMEDIA_TYPE_VIDEO ) {
av_log ( NULL , AV_LOG_FATAL , " Only video filters supported currently. \n " ) ;
exit_program ( 1 ) ;
}
if ( in - > name ) {
AVFormatContext * s ;
AVStream * st = NULL ;
char * p ;
int file_idx = strtol ( in - > name , & p , 0 ) ;
2012-04-23 02:37:54 +03:00
if ( file_idx < 0 | | file_idx > = nb_input_files ) {
2012-04-01 16:08:33 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid file index %d in filtegraph description %s. \n " ,
file_idx , fg - > graph_desc ) ;
exit_program ( 1 ) ;
}
s = input_files [ file_idx ] - > ctx ;
for ( i = 0 ; i < s - > nb_streams ; i + + ) {
if ( s - > streams [ i ] - > codec - > codec_type ! = type )
continue ;
if ( check_stream_specifier ( s , s - > streams [ i ] , * p = = ' : ' ? p + 1 : p ) = = 1 ) {
st = s - > streams [ i ] ;
break ;
}
}
if ( ! st ) {
av_log ( NULL , AV_LOG_FATAL , " Stream specifier '%s' in filtergraph description %s "
" matches no streams. \n " , p , fg - > graph_desc ) ;
exit_program ( 1 ) ;
}
ist = input_streams [ input_files [ file_idx ] - > ist_index + st - > index ] ;
} else {
/* find the first unused stream of corresponding type */
for ( i = 0 ; i < nb_input_streams ; i + + ) {
ist = input_streams [ i ] ;
if ( ist - > st - > codec - > codec_type = = type & & ist - > discard )
break ;
}
if ( i = = nb_input_streams ) {
av_log ( NULL , AV_LOG_FATAL , " Cannot find a matching stream for "
" unlabeled input pad %d on filter %s " , in - > pad_idx ,
in - > filter_ctx - > name ) ;
exit_program ( 1 ) ;
}
}
ist - > discard = 0 ;
ist - > decoding_needed = 1 ;
ist - > st - > discard = AVDISCARD_NONE ;
fg - > inputs = grow_array ( fg - > inputs , sizeof ( * fg - > inputs ) ,
& fg - > nb_inputs , fg - > nb_inputs + 1 ) ;
if ( ! ( fg - > inputs [ fg - > nb_inputs - 1 ] = av_mallocz ( sizeof ( * fg - > inputs [ 0 ] ) ) ) )
exit_program ( 1 ) ;
fg - > inputs [ fg - > nb_inputs - 1 ] - > ist = ist ;
fg - > inputs [ fg - > nb_inputs - 1 ] - > graph = fg ;
ist - > filters = grow_array ( ist - > filters , sizeof ( * ist - > filters ) ,
& ist - > nb_filters , ist - > nb_filters + 1 ) ;
ist - > filters [ ist - > nb_filters - 1 ] = fg - > inputs [ fg - > nb_inputs - 1 ] ;
}
static int configure_output_filter ( FilterGraph * fg , OutputFilter * ofilter , AVFilterInOut * out )
{
SinkContext sink_ctx ;
AVCodecContext * codec = ofilter - > ost - > st - > codec ;
AVFilterContext * last_filter = out - > filter_ctx ;
int pad_idx = out - > pad_idx ;
int ret ;
sink_ctx . pix_fmts = choose_pixel_fmts ( ofilter - > ost ) ;
ret = avfilter_graph_create_filter ( & ofilter - > filter , & sink ,
" out " , NULL , & sink_ctx , fg - > graph ) ;
if ( ret < 0 )
return ret ;
if ( codec - > width | | codec - > height ) {
char args [ 255 ] ;
snprintf ( args , sizeof ( args ) , " %d:%d:flags=0x%X " ,
codec - > width ,
codec - > height ,
( unsigned ) ofilter - > ost - > sws_flags ) ;
if ( ( ret = avfilter_graph_create_filter ( & last_filter , avfilter_get_by_name ( " scale " ) ,
NULL , args , NULL , fg - > graph ) ) < 0 )
return ret ;
if ( ( ret = avfilter_link ( out - > filter_ctx , out - > pad_idx , last_filter , 0 ) ) < 0 )
return ret ;
pad_idx = 0 ;
}
if ( ( ret = avfilter_link ( last_filter , pad_idx , ofilter - > filter , 0 ) ) < 0 )
return ret ;
return 0 ;
}
static int configure_complex_filter ( FilterGraph * fg )
{
AVFilterInOut * inputs , * outputs , * cur ;
int ret , i , init = ! fg - > graph ;
avfilter_graph_free ( & fg - > graph ) ;
if ( ! ( fg - > graph = avfilter_graph_alloc ( ) ) )
return AVERROR ( ENOMEM ) ;
if ( ( ret = avfilter_graph_parse2 ( fg - > graph , fg - > graph_desc , & inputs , & outputs ) ) < 0 )
return ret ;
for ( cur = inputs ; init & & cur ; cur = cur - > next )
init_input_filter ( fg , cur ) ;
for ( cur = inputs , i = 0 ; cur ; cur = cur - > next , i + + ) {
InputFilter * ifilter = fg - > inputs [ i ] ;
InputStream * ist = ifilter - > ist ;
AVRational sar ;
char args [ 255 ] ;
sar = ist - > st - > sample_aspect_ratio . num ? ist - > st - > sample_aspect_ratio :
ist - > st - > codec - > sample_aspect_ratio ;
snprintf ( args , sizeof ( args ) , " %d:%d:%d:%d:%d:%d:%d " , ist - > st - > codec - > width ,
ist - > st - > codec - > height , ist - > st - > codec - > pix_fmt , 1 , AV_TIME_BASE ,
sar . num , sar . den ) ;
if ( ( ret = avfilter_graph_create_filter ( & ifilter - > filter ,
avfilter_get_by_name ( " buffer " ) , cur - > name ,
args , NULL , fg - > graph ) ) < 0 )
return ret ;
if ( ( ret = avfilter_link ( ifilter - > filter , 0 ,
cur - > filter_ctx , cur - > pad_idx ) ) < 0 )
return ret ;
}
avfilter_inout_free ( & inputs ) ;
if ( ! init ) {
/* we already know the mappings between lavfi outputs and output streams,
* so we can finish the setup */
for ( cur = outputs , i = 0 ; cur ; cur = cur - > next , i + + )
configure_output_filter ( fg , fg - > outputs [ i ] , cur ) ;
avfilter_inout_free ( & outputs ) ;
if ( ( ret = avfilter_graph_config ( fg - > graph , NULL ) ) < 0 )
return ret ;
} else {
/* wait until output mappings are processed */
for ( cur = outputs ; cur ; ) {
fg - > outputs = grow_array ( fg - > outputs , sizeof ( * fg - > outputs ) ,
& fg - > nb_outputs , fg - > nb_outputs + 1 ) ;
if ( ! ( fg - > outputs [ fg - > nb_outputs - 1 ] = av_mallocz ( sizeof ( * fg - > outputs [ 0 ] ) ) ) )
exit_program ( 1 ) ;
fg - > outputs [ fg - > nb_outputs - 1 ] - > graph = fg ;
fg - > outputs [ fg - > nb_outputs - 1 ] - > out_tmp = cur ;
cur = cur - > next ;
fg - > outputs [ fg - > nb_outputs - 1 ] - > out_tmp - > next = NULL ;
}
}
return 0 ;
}
static int configure_complex_filters ( void )
{
int i , ret = 0 ;
for ( i = 0 ; i < nb_filtergraphs ; i + + )
if ( ! filtergraphs [ i ] - > graph & &
( ret = configure_complex_filter ( filtergraphs [ i ] ) ) < 0 )
return ret ;
return 0 ;
}
static int configure_filtergraph ( FilterGraph * fg )
{
return fg - > graph_desc ? configure_complex_filter ( fg ) : configure_video_filters ( fg ) ;
}
static int ist_in_filtergraph ( FilterGraph * fg , InputStream * ist )
{
int i ;
for ( i = 0 ; i < fg - > nb_inputs ; i + + )
if ( fg - > inputs [ i ] - > ist = = ist )
return 1 ;
return 0 ;
}
2011-07-27 21:56:59 +03:00
static void term_exit ( void )
{
av_log ( NULL , AV_LOG_QUIET , " " ) ;
}
static volatile int received_sigterm = 0 ;
static volatile int received_nb_signals = 0 ;
static void
sigterm_handler ( int sig )
{
received_sigterm = sig ;
received_nb_signals + + ;
term_exit ( ) ;
}
static void term_init ( void )
{
2011-12-29 23:28:29 +03:00
signal ( SIGINT , sigterm_handler ) ; /* Interrupt (ANSI). */
2011-07-27 21:56:59 +03:00
signal ( SIGTERM , sigterm_handler ) ; /* Termination (ANSI). */
# ifdef SIGXCPU
signal ( SIGXCPU , sigterm_handler ) ;
# endif
}
2011-11-07 00:22:04 +03:00
static int decode_interrupt_cb ( void * ctx )
2011-07-27 21:56:59 +03:00
{
return received_nb_signals > 1 ;
}
2011-11-07 00:22:04 +03:00
static const AVIOInterruptCB int_cb = { decode_interrupt_cb , NULL } ;
2011-08-29 08:03:24 +03:00
void exit_program ( int ret )
2011-07-27 21:56:59 +03:00
{
2012-03-29 09:51:17 +03:00
int i , j ;
for ( i = 0 ; i < nb_filtergraphs ; i + + ) {
avfilter_graph_free ( & filtergraphs [ i ] - > graph ) ;
for ( j = 0 ; j < filtergraphs [ i ] - > nb_inputs ; j + + )
av_freep ( & filtergraphs [ i ] - > inputs [ j ] ) ;
av_freep ( & filtergraphs [ i ] - > inputs ) ;
for ( j = 0 ; j < filtergraphs [ i ] - > nb_outputs ; j + + )
av_freep ( & filtergraphs [ i ] - > outputs [ j ] ) ;
av_freep ( & filtergraphs [ i ] - > outputs ) ;
av_freep ( & filtergraphs [ i ] ) ;
}
av_freep ( & filtergraphs ) ;
2011-07-27 21:56:59 +03:00
/* close files */
2011-12-29 23:28:29 +03:00
for ( i = 0 ; i < nb_output_files ; i + + ) {
2012-04-15 16:28:30 +03:00
AVFormatContext * s = output_files [ i ] - > ctx ;
2011-07-27 21:56:59 +03:00
if ( ! ( s - > oformat - > flags & AVFMT_NOFILE ) & & s - > pb )
avio_close ( s - > pb ) ;
avformat_free_context ( s ) ;
2012-04-15 16:28:30 +03:00
av_dict_free ( & output_files [ i ] - > opts ) ;
av_freep ( & output_files [ i ] ) ;
2011-07-27 21:56:59 +03:00
}
2012-01-01 19:56:28 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
AVBitStreamFilterContext * bsfc = output_streams [ i ] - > bitstream_filters ;
2012-01-01 19:56:28 +03:00
while ( bsfc ) {
AVBitStreamFilterContext * next = bsfc - > next ;
av_bitstream_filter_close ( bsfc ) ;
bsfc = next ;
}
2012-04-15 16:28:30 +03:00
output_streams [ i ] - > bitstream_filters = NULL ;
2012-01-02 17:02:46 +03:00
2012-04-15 16:28:30 +03:00
if ( output_streams [ i ] - > output_frame ) {
AVFrame * frame = output_streams [ i ] - > output_frame ;
2011-12-31 17:36:25 +03:00
if ( frame - > extended_data ! = frame - > data )
av_freep ( & frame - > extended_data ) ;
av_freep ( & frame ) ;
}
2012-04-15 16:28:30 +03:00
av_freep ( & output_streams [ i ] - > avfilter ) ;
2012-03-29 10:00:09 +03:00
av_freep ( & output_streams [ i ] - > filtered_frame ) ;
2012-04-15 16:28:30 +03:00
av_freep ( & output_streams [ i ] ) ;
2012-01-01 19:56:28 +03:00
}
2011-12-29 23:28:29 +03:00
for ( i = 0 ; i < nb_input_files ; i + + ) {
2012-04-15 16:28:30 +03:00
avformat_close_input ( & input_files [ i ] - > ctx ) ;
av_freep ( & input_files [ i ] ) ;
2011-07-27 21:56:59 +03:00
}
2011-12-05 19:49:38 +03:00
for ( i = 0 ; i < nb_input_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
av_freep ( & input_streams [ i ] - > decoded_frame ) ;
av_dict_free ( & input_streams [ i ] - > opts ) ;
free_buffer_pool ( input_streams [ i ] ) ;
2012-03-29 09:51:17 +03:00
av_freep ( & input_streams [ i ] - > filters ) ;
2012-04-15 16:28:30 +03:00
av_freep ( & input_streams [ i ] ) ;
2011-12-05 19:49:38 +03:00
}
2011-07-27 21:56:59 +03:00
if ( vstats_file )
fclose ( vstats_file ) ;
av_free ( vstats_filename ) ;
av_freep ( & input_streams ) ;
av_freep ( & input_files ) ;
2011-08-17 11:21:37 +03:00
av_freep ( & output_streams ) ;
2011-08-17 10:56:08 +03:00
av_freep ( & output_files ) ;
2011-07-27 21:56:59 +03:00
uninit_opts ( ) ;
av_free ( audio_buf ) ;
2011-12-31 17:36:25 +03:00
allocated_audio_buf_size = 0 ;
2012-03-21 23:36:23 +03:00
av_free ( async_buf ) ;
allocated_async_buf_size = 0 ;
2011-07-27 21:56:59 +03:00
avfilter_uninit ( ) ;
2011-11-06 03:47:48 +03:00
avformat_network_deinit ( ) ;
2011-07-27 21:56:59 +03:00
if ( received_sigterm ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_INFO , " Received signal %d: terminating. \n " ,
( int ) received_sigterm ) ;
2011-07-27 21:56:59 +03:00
exit ( 255 ) ;
}
2011-12-08 04:17:18 +03:00
exit ( ret ) ;
2011-07-27 21:56:59 +03:00
}
static void assert_avoptions ( AVDictionary * m )
{
AVDictionaryEntry * t ;
if ( ( t = av_dict_get ( m , " " , NULL , AV_DICT_IGNORE_SUFFIX ) ) ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Option %s not found. \n " , t - > key ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
}
static void assert_codec_experimental ( AVCodecContext * c , int encoder )
{
const char * codec_string = encoder ? " encoder " : " decoder " ;
AVCodec * codec ;
if ( c - > codec - > capabilities & CODEC_CAP_EXPERIMENTAL & &
c - > strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " %s '%s' is experimental and might produce bad "
2011-07-27 21:56:59 +03:00
" results. \n Add '-strict experimental' if you want to use it. \n " ,
codec_string , c - > codec - > name ) ;
codec = encoder ? avcodec_find_encoder ( c - > codec - > id ) : avcodec_find_decoder ( c - > codec - > id ) ;
if ( ! ( codec - > capabilities & CODEC_CAP_EXPERIMENTAL ) )
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Or use the non experimental %s '%s'. \n " ,
2011-07-27 21:56:59 +03:00
codec_string , codec - > name ) ;
exit_program ( 1 ) ;
}
}
static void choose_sample_fmt ( AVStream * st , AVCodec * codec )
{
2011-12-29 23:28:29 +03:00
if ( codec & & codec - > sample_fmts ) {
const enum AVSampleFormat * p = codec - > sample_fmts ;
for ( ; * p ! = - 1 ; p + + ) {
if ( * p = = st - > codec - > sample_fmt )
2011-07-27 21:56:59 +03:00
break ;
}
if ( * p = = - 1 ) {
av_log ( NULL , AV_LOG_WARNING ,
" Incompatible sample format '%s' for codec '%s', auto-selecting format '%s' \n " ,
av_get_sample_fmt_name ( st - > codec - > sample_fmt ) ,
codec - > name ,
av_get_sample_fmt_name ( codec - > sample_fmts [ 0 ] ) ) ;
st - > codec - > sample_fmt = codec - > sample_fmts [ 0 ] ;
}
}
}
/**
* Update the requested input sample format based on the output sample format .
* This is currently only used to request float output from decoders which
* support multiple sample formats , one of which is AV_SAMPLE_FMT_FLT .
* Ideally this will be removed in the future when decoders do not do format
* conversion and only output in their native format .
*/
static void update_sample_fmt ( AVCodecContext * dec , AVCodec * dec_codec ,
AVCodecContext * enc )
{
/* if sample formats match or a decoder sample format has already been
requested , just return */
if ( enc - > sample_fmt = = dec - > sample_fmt | |
dec - > request_sample_fmt > AV_SAMPLE_FMT_NONE )
return ;
/* if decoder supports more than one output format */
if ( dec_codec & & dec_codec - > sample_fmts & &
dec_codec - > sample_fmts [ 0 ] ! = AV_SAMPLE_FMT_NONE & &
dec_codec - > sample_fmts [ 1 ] ! = AV_SAMPLE_FMT_NONE ) {
const enum AVSampleFormat * p ;
int min_dec = - 1 , min_inc = - 1 ;
/* find a matching sample format in the encoder */
for ( p = dec_codec - > sample_fmts ; * p ! = AV_SAMPLE_FMT_NONE ; p + + ) {
if ( * p = = enc - > sample_fmt ) {
dec - > request_sample_fmt = * p ;
return ;
} else if ( * p > enc - > sample_fmt ) {
min_inc = FFMIN ( min_inc , * p - enc - > sample_fmt ) ;
} else
min_dec = FFMIN ( min_dec , enc - > sample_fmt - * p ) ;
}
/* if none match, provide the one that matches quality closest */
dec - > request_sample_fmt = min_inc > 0 ? enc - > sample_fmt + min_inc :
enc - > sample_fmt - min_dec ;
}
}
static void choose_sample_rate ( AVStream * st , AVCodec * codec )
{
2011-12-29 23:28:29 +03:00
if ( codec & & codec - > supported_samplerates ) {
const int * p = codec - > supported_samplerates ;
int best = 0 ;
int best_dist = INT_MAX ;
for ( ; * p ; p + + ) {
int dist = abs ( st - > codec - > sample_rate - * p ) ;
if ( dist < best_dist ) {
best_dist = dist ;
best = * p ;
2011-07-27 21:56:59 +03:00
}
}
2011-12-29 23:28:29 +03:00
if ( best_dist ) {
2011-07-27 21:56:59 +03:00
av_log ( st - > codec , AV_LOG_WARNING , " Requested sampling rate unsupported using closest supported (%d) \n " , best ) ;
}
2011-12-29 23:28:29 +03:00
st - > codec - > sample_rate = best ;
2011-07-27 21:56:59 +03:00
}
}
static double
2012-02-05 22:56:24 +03:00
get_sync_ipts ( const OutputStream * ost , int64_t pts )
2011-07-27 21:56:59 +03:00
{
2012-04-15 16:28:30 +03:00
OutputFile * of = output_files [ ost - > file_index ] ;
2012-02-05 22:56:24 +03:00
return ( double ) ( pts - of - > start_time ) / AV_TIME_BASE ;
2011-07-27 21:56:59 +03:00
}
2012-01-01 23:51:26 +03:00
static void write_frame ( AVFormatContext * s , AVPacket * pkt , OutputStream * ost )
2011-12-29 23:28:29 +03:00
{
2012-01-01 23:51:26 +03:00
AVBitStreamFilterContext * bsfc = ost - > bitstream_filters ;
AVCodecContext * avctx = ost - > st - > codec ;
2011-07-27 21:56:59 +03:00
int ret ;
2012-01-02 11:22:41 +03:00
/*
* Audio encoders may split the packets - - # frames in ! = # packets out .
* But there is no reordering , so we can limit the number of output packets
* by simply dropping them here .
* Counting encoded video frames needs to be done separately because of
* reordering , see do_video_out ( )
*/
if ( ! ( avctx - > codec_type = = AVMEDIA_TYPE_VIDEO & & avctx - > codec ) ) {
2012-03-20 22:36:28 +03:00
if ( ost - > frame_number > = ost - > max_frames ) {
av_free_packet ( pkt ) ;
2012-01-02 11:22:41 +03:00
return ;
2012-03-20 22:36:28 +03:00
}
2012-01-02 11:22:41 +03:00
ost - > frame_number + + ;
}
2011-12-29 23:28:29 +03:00
while ( bsfc ) {
AVPacket new_pkt = * pkt ;
int a = av_bitstream_filter_filter ( bsfc , avctx , NULL ,
& new_pkt . data , & new_pkt . size ,
pkt - > data , pkt - > size ,
pkt - > flags & AV_PKT_FLAG_KEY ) ;
if ( a > 0 ) {
2011-07-27 21:56:59 +03:00
av_free_packet ( pkt ) ;
2011-12-29 23:28:29 +03:00
new_pkt . destruct = av_destruct_packet ;
} else if ( a < 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_ERROR , " %s failed for stream %d, codec %s " ,
bsfc - > filter - > name , pkt - > stream_index ,
avctx - > codec ? avctx - > codec - > name : " copy " ) ;
2011-07-27 21:56:59 +03:00
print_error ( " " , a ) ;
if ( exit_on_error )
exit_program ( 1 ) ;
}
2011-12-29 23:28:29 +03:00
* pkt = new_pkt ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
bsfc = bsfc - > next ;
2011-07-27 21:56:59 +03:00
}
2012-02-01 12:23:28 +03:00
pkt - > stream_index = ost - > index ;
2011-12-29 23:28:29 +03:00
ret = av_interleaved_write_frame ( s , pkt ) ;
if ( ret < 0 ) {
2011-07-27 21:56:59 +03:00
print_error ( " av_interleaved_write_frame() " , ret ) ;
exit_program ( 1 ) ;
}
}
2012-02-05 16:32:10 +03:00
static int check_recording_time ( OutputStream * ost )
{
2012-04-15 16:28:30 +03:00
OutputFile * of = output_files [ ost - > file_index ] ;
2012-02-05 16:32:10 +03:00
if ( of - > recording_time ! = INT64_MAX & &
av_compare_ts ( ost - > sync_opts - ost - > first_pts , ost - > st - > codec - > time_base , of - > recording_time ,
AV_TIME_BASE_Q ) > = 0 ) {
ost - > is_past_recording_time = 1 ;
return 0 ;
}
return 1 ;
}
2012-04-17 20:30:00 +03:00
static void get_default_channel_layouts ( OutputStream * ost , InputStream * ist )
{
char layout_name [ 256 ] ;
AVCodecContext * enc = ost - > st - > codec ;
AVCodecContext * dec = ist - > st - > codec ;
if ( dec - > channel_layout & &
av_get_channel_layout_nb_channels ( dec - > channel_layout ) ! = dec - > channels ) {
av_get_channel_layout_string ( layout_name , sizeof ( layout_name ) ,
dec - > channels , dec - > channel_layout ) ;
av_log ( NULL , AV_LOG_ERROR , " New channel layout (%s) is invalid \n " ,
layout_name ) ;
dec - > channel_layout = 0 ;
}
if ( ! dec - > channel_layout ) {
if ( enc - > channel_layout & & dec - > channels = = enc - > channels ) {
dec - > channel_layout = enc - > channel_layout ;
} else {
dec - > channel_layout = av_get_default_channel_layout ( dec - > channels ) ;
if ( ! dec - > channel_layout ) {
av_log ( NULL , AV_LOG_FATAL , " Unable to find default channel "
" layout for Input Stream #%d.%d \n " , ist - > file_index ,
ist - > st - > index ) ;
exit_program ( 1 ) ;
}
}
av_get_channel_layout_string ( layout_name , sizeof ( layout_name ) ,
dec - > channels , dec - > channel_layout ) ;
av_log ( NULL , AV_LOG_WARNING , " Guessed Channel Layout for Input Stream "
" #%d.%d : %s \n " , ist - > file_index , ist - > st - > index , layout_name ) ;
}
if ( ! enc - > channel_layout ) {
if ( dec - > channels = = enc - > channels ) {
enc - > channel_layout = dec - > channel_layout ;
return ;
} else {
enc - > channel_layout = av_get_default_channel_layout ( enc - > channels ) ;
}
if ( ! enc - > channel_layout ) {
av_log ( NULL , AV_LOG_FATAL , " Unable to find default channel layout "
" for Output Stream #%d.%d \n " , ost - > file_index ,
ost - > st - > index ) ;
exit_program ( 1 ) ;
}
av_get_channel_layout_string ( layout_name , sizeof ( layout_name ) ,
enc - > channels , enc - > channel_layout ) ;
av_log ( NULL , AV_LOG_WARNING , " Guessed Channel Layout for Output Stream "
" #%d.%d : %s \n " , ost - > file_index , ost - > st - > index , layout_name ) ;
}
}
2011-11-29 02:19:39 +03:00
static void generate_silence ( uint8_t * buf , enum AVSampleFormat sample_fmt , size_t size )
{
int fill_char = 0x00 ;
if ( sample_fmt = = AV_SAMPLE_FMT_U8 )
fill_char = 0x80 ;
memset ( buf , fill_char , size ) ;
}
2011-12-31 17:36:25 +03:00
static int encode_audio_frame ( AVFormatContext * s , OutputStream * ost ,
const uint8_t * buf , int buf_size )
{
AVCodecContext * enc = ost - > st - > codec ;
AVFrame * frame = NULL ;
AVPacket pkt ;
int ret , got_packet ;
av_init_packet ( & pkt ) ;
pkt . data = NULL ;
pkt . size = 0 ;
if ( buf ) {
if ( ! ost - > output_frame ) {
ost - > output_frame = avcodec_alloc_frame ( ) ;
if ( ! ost - > output_frame ) {
av_log ( NULL , AV_LOG_FATAL , " out-of-memory in encode_audio_frame() \n " ) ;
exit_program ( 1 ) ;
}
}
frame = ost - > output_frame ;
if ( frame - > extended_data ! = frame - > data )
av_freep ( & frame - > extended_data ) ;
avcodec_get_frame_defaults ( frame ) ;
frame - > nb_samples = buf_size /
( enc - > channels * av_get_bytes_per_sample ( enc - > sample_fmt ) ) ;
if ( ( ret = avcodec_fill_audio_frame ( frame , enc - > channels , enc - > sample_fmt ,
buf , buf_size , 1 ) ) < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Audio encoding failed \n " ) ;
exit_program ( 1 ) ;
}
2012-02-05 16:32:10 +03:00
if ( ! check_recording_time ( ost ) )
return 0 ;
2012-03-01 21:02:49 +03:00
frame - > pts = ost - > sync_opts ;
2012-02-05 16:32:10 +03:00
ost - > sync_opts + = frame - > nb_samples ;
2011-12-31 17:36:25 +03:00
}
got_packet = 0 ;
if ( avcodec_encode_audio2 ( enc , & pkt , frame , & got_packet ) < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Audio encoding failed \n " ) ;
exit_program ( 1 ) ;
}
if ( got_packet ) {
if ( pkt . pts ! = AV_NOPTS_VALUE )
pkt . pts = av_rescale_q ( pkt . pts , enc - > time_base , ost - > st - > time_base ) ;
2012-02-07 03:08:32 +03:00
if ( pkt . dts ! = AV_NOPTS_VALUE )
pkt . dts = av_rescale_q ( pkt . dts , enc - > time_base , ost - > st - > time_base ) ;
2011-12-31 17:36:25 +03:00
if ( pkt . duration > 0 )
pkt . duration = av_rescale_q ( pkt . duration , enc - > time_base , ost - > st - > time_base ) ;
write_frame ( s , & pkt , ost ) ;
audio_size + = pkt . size ;
}
return pkt . size ;
}
2012-03-21 23:55:30 +03:00
static int alloc_audio_output_buf ( AVCodecContext * dec , AVCodecContext * enc ,
2012-04-05 21:06:28 +03:00
int nb_samples , int * buf_linesize )
2012-03-21 23:55:30 +03:00
{
int64_t audio_buf_samples ;
int audio_buf_size ;
/* calculate required number of samples to allocate */
audio_buf_samples = ( ( int64_t ) nb_samples * enc - > sample_rate + dec - > sample_rate ) /
dec - > sample_rate ;
2012-03-21 23:56:49 +03:00
audio_buf_samples = 4 * audio_buf_samples + 16 ; // safety factors for resampling
2012-03-21 23:55:30 +03:00
audio_buf_samples = FFMAX ( audio_buf_samples , enc - > frame_size ) ;
if ( audio_buf_samples > INT_MAX )
return AVERROR ( EINVAL ) ;
2012-04-05 21:06:28 +03:00
audio_buf_size = av_samples_get_buffer_size ( buf_linesize , enc - > channels ,
2012-03-21 23:55:30 +03:00
audio_buf_samples ,
2012-04-06 02:09:51 +03:00
enc - > sample_fmt , 0 ) ;
2012-03-21 23:55:30 +03:00
if ( audio_buf_size < 0 )
return audio_buf_size ;
av_fast_malloc ( & audio_buf , & allocated_audio_buf_size , audio_buf_size ) ;
if ( ! audio_buf )
return AVERROR ( ENOMEM ) ;
return 0 ;
}
2011-11-22 01:41:49 +03:00
static void do_audio_out ( AVFormatContext * s , OutputStream * ost ,
InputStream * ist , AVFrame * decoded_frame )
2011-07-27 21:56:59 +03:00
{
uint8_t * buftmp ;
2012-04-05 21:06:28 +03:00
int size_out , frame_bytes , resample_changed , ret ;
2011-12-29 23:28:29 +03:00
AVCodecContext * enc = ost - > st - > codec ;
AVCodecContext * dec = ist - > st - > codec ;
2011-07-27 21:56:59 +03:00
int osize = av_get_bytes_per_sample ( enc - > sample_fmt ) ;
int isize = av_get_bytes_per_sample ( dec - > sample_fmt ) ;
2011-11-22 01:41:49 +03:00
uint8_t * buf = decoded_frame - > data [ 0 ] ;
int size = decoded_frame - > nb_samples * dec - > channels * isize ;
2012-04-05 21:06:28 +03:00
int out_linesize = 0 ;
int buf_linesize = decoded_frame - > linesize [ 0 ] ;
2011-12-29 23:28:29 +03:00
2012-04-17 20:30:00 +03:00
get_default_channel_layouts ( ost , ist ) ;
2012-04-05 21:06:28 +03:00
if ( alloc_audio_output_buf ( dec , enc , decoded_frame - > nb_samples , & out_linesize ) < 0 ) {
2012-03-21 23:55:30 +03:00
av_log ( NULL , AV_LOG_FATAL , " Error allocating audio buffer \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2012-04-05 21:06:28 +03:00
if ( audio_sync_method > 1 | |
enc - > channels ! = dec - > channels | |
enc - > channel_layout ! = dec - > channel_layout | |
enc - > sample_rate ! = dec - > sample_rate | |
dec - > sample_fmt ! = enc - > sample_fmt )
2011-07-27 21:56:59 +03:00
ost - > audio_resample = 1 ;
resample_changed = ost - > resample_sample_fmt ! = dec - > sample_fmt | |
ost - > resample_channels ! = dec - > channels | |
2012-04-05 21:06:28 +03:00
ost - > resample_channel_layout ! = dec - > channel_layout | |
2011-07-27 21:56:59 +03:00
ost - > resample_sample_rate ! = dec - > sample_rate ;
2012-04-05 21:06:28 +03:00
if ( ( ost - > audio_resample & & ! ost - > avr ) | | resample_changed ) {
2011-07-27 21:56:59 +03:00
if ( resample_changed ) {
2012-04-05 21:06:28 +03:00
av_log ( NULL , AV_LOG_INFO , " Input stream #%d:%d frame changed from rate:%d fmt:%s ch:%d chl:0x% " PRIx64 " to rate:%d fmt:%s ch:%d chl:0x% " PRIx64 " \n " ,
2011-07-27 21:56:59 +03:00
ist - > file_index , ist - > st - > index ,
2012-04-05 21:06:28 +03:00
ost - > resample_sample_rate , av_get_sample_fmt_name ( ost - > resample_sample_fmt ) ,
ost - > resample_channels , ost - > resample_channel_layout ,
dec - > sample_rate , av_get_sample_fmt_name ( dec - > sample_fmt ) ,
dec - > channels , dec - > channel_layout ) ;
2011-07-27 21:56:59 +03:00
ost - > resample_sample_fmt = dec - > sample_fmt ;
ost - > resample_channels = dec - > channels ;
2012-04-05 21:06:28 +03:00
ost - > resample_channel_layout = dec - > channel_layout ;
2011-07-27 21:56:59 +03:00
ost - > resample_sample_rate = dec - > sample_rate ;
2012-04-05 21:06:28 +03:00
if ( ost - > avr )
avresample_close ( ost - > avr ) ;
2011-07-27 21:56:59 +03:00
}
/* if audio_sync_method is >1 the resampler is needed for audio drift compensation */
if ( audio_sync_method < = 1 & &
ost - > resample_sample_fmt = = enc - > sample_fmt & &
ost - > resample_channels = = enc - > channels & &
2012-04-05 21:06:28 +03:00
ost - > resample_channel_layout = = enc - > channel_layout & &
2011-07-27 21:56:59 +03:00
ost - > resample_sample_rate = = enc - > sample_rate ) {
ost - > audio_resample = 0 ;
} else if ( ost - > audio_resample ) {
2012-04-05 21:06:28 +03:00
if ( ! ost - > avr ) {
ost - > avr = avresample_alloc_context ( ) ;
if ( ! ost - > avr ) {
av_log ( NULL , AV_LOG_FATAL , " Error allocating context for libavresample \n " ) ;
exit_program ( 1 ) ;
}
2011-07-27 21:56:59 +03:00
}
2012-04-05 21:06:28 +03:00
av_opt_set_int ( ost - > avr , " in_channel_layout " , dec - > channel_layout , 0 ) ;
av_opt_set_int ( ost - > avr , " in_sample_fmt " , dec - > sample_fmt , 0 ) ;
av_opt_set_int ( ost - > avr , " in_sample_rate " , dec - > sample_rate , 0 ) ;
av_opt_set_int ( ost - > avr , " out_channel_layout " , enc - > channel_layout , 0 ) ;
av_opt_set_int ( ost - > avr , " out_sample_fmt " , enc - > sample_fmt , 0 ) ;
av_opt_set_int ( ost - > avr , " out_sample_rate " , enc - > sample_rate , 0 ) ;
if ( audio_sync_method > 1 )
av_opt_set_int ( ost - > avr , " force_resampling " , 1 , 0 ) ;
/* if both the input and output formats are s16 or u8, use s16 as
the internal sample format */
if ( av_get_bytes_per_sample ( dec - > sample_fmt ) < = 2 & &
av_get_bytes_per_sample ( enc - > sample_fmt ) < = 2 ) {
av_opt_set_int ( ost - > avr , " internal_sample_fmt " , AV_SAMPLE_FMT_S16P , 0 ) ;
}
ret = avresample_open ( ost - > avr ) ;
if ( ret < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Error opening libavresample \n " ) ;
exit_program ( 1 ) ;
}
2011-07-27 21:56:59 +03:00
}
}
2012-04-08 22:00:01 +03:00
if ( audio_sync_method > 0 ) {
2012-02-05 23:12:43 +03:00
double delta = get_sync_ipts ( ost , ist - > last_dts ) * enc - > sample_rate - ost - > sync_opts -
2011-12-29 23:28:29 +03:00
av_fifo_size ( ost - > fifo ) / ( enc - > channels * osize ) ;
2011-11-29 02:19:39 +03:00
int idelta = delta * dec - > sample_rate / enc - > sample_rate ;
int byte_delta = idelta * isize * dec - > channels ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
// FIXME resample delay
if ( fabs ( delta ) > 50 ) {
if ( ist - > is_start | | fabs ( delta ) > audio_drift_threshold * enc - > sample_rate ) {
if ( byte_delta < 0 ) {
byte_delta = FFMAX ( byte_delta , - size ) ;
2011-07-27 21:56:59 +03:00
size + = byte_delta ;
buf - = byte_delta ;
2011-11-29 02:19:39 +03:00
av_log ( NULL , AV_LOG_VERBOSE , " discarding %d audio samples \n " ,
- byte_delta / ( isize * dec - > channels ) ) ;
2011-12-29 23:28:29 +03:00
if ( ! size )
2011-07-27 21:56:59 +03:00
return ;
2011-12-29 23:28:29 +03:00
ist - > is_start = 0 ;
} else {
2012-03-21 23:36:23 +03:00
av_fast_malloc ( & async_buf , & allocated_async_buf_size ,
byte_delta + size ) ;
if ( ! async_buf ) {
av_log ( NULL , AV_LOG_FATAL , " Out of memory in do_audio_out \n " ) ;
exit_program ( 1 ) ;
}
2011-07-27 21:56:59 +03:00
2012-04-05 21:06:28 +03:00
if ( alloc_audio_output_buf ( dec , enc , decoded_frame - > nb_samples + idelta , & out_linesize ) < 0 ) {
2012-03-21 23:55:30 +03:00
av_log ( NULL , AV_LOG_FATAL , " Error allocating audio buffer \n " ) ;
exit_program ( 1 ) ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
ist - > is_start = 0 ;
2011-07-27 21:56:59 +03:00
2012-03-21 23:36:23 +03:00
generate_silence ( async_buf , dec - > sample_fmt , byte_delta ) ;
memcpy ( async_buf + byte_delta , buf , size ) ;
buf = async_buf ;
2011-07-27 21:56:59 +03:00
size + = byte_delta ;
2012-04-05 21:06:28 +03:00
buf_linesize = allocated_async_buf_size ;
2011-11-29 02:19:39 +03:00
av_log ( NULL , AV_LOG_VERBOSE , " adding %d audio samples of silence \n " , idelta ) ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
} else if ( audio_sync_method > 1 ) {
int comp = av_clip ( delta , - audio_sync_method , audio_sync_method ) ;
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_VERBOSE , " compensating audio timestamp drift:%f compensation:%d in:%d \n " ,
delta , comp , enc - > sample_rate ) ;
2011-07-27 21:56:59 +03:00
// fprintf(stderr, "drift:%f len:%d opts:%"PRId64" ipts:%"PRId64" fifo:%d\n", delta, -1, ost->sync_opts, (int64_t)(get_sync_ipts(ost) * enc->sample_rate), av_fifo_size(ost->fifo)/(ost->st->codec->channels * 2));
2012-04-05 21:06:28 +03:00
avresample_set_compensation ( ost - > avr , comp , enc - > sample_rate ) ;
2011-07-27 21:56:59 +03:00
}
}
2012-04-08 22:00:01 +03:00
} else if ( audio_sync_method = = 0 )
2012-02-05 23:12:43 +03:00
ost - > sync_opts = lrintf ( get_sync_ipts ( ost , ist - > last_dts ) * enc - > sample_rate ) -
2011-12-29 23:28:29 +03:00
av_fifo_size ( ost - > fifo ) / ( enc - > channels * osize ) ; // FIXME wrong
2011-07-27 21:56:59 +03:00
if ( ost - > audio_resample ) {
buftmp = audio_buf ;
2012-04-05 21:06:28 +03:00
size_out = avresample_convert ( ost - > avr , ( void * * ) & buftmp ,
allocated_audio_buf_size , out_linesize ,
( void * * ) & buf , buf_linesize ,
size / ( dec - > channels * isize ) ) ;
2011-07-27 21:56:59 +03:00
size_out = size_out * enc - > channels * osize ;
} else {
buftmp = buf ;
size_out = size ;
}
/* now encode as many frames as possible */
2011-12-31 17:36:25 +03:00
if ( ! ( enc - > codec - > capabilities & CODEC_CAP_VARIABLE_FRAME_SIZE ) ) {
2011-07-27 21:56:59 +03:00
/* output resampled raw samples */
if ( av_fifo_realloc2 ( ost - > fifo , av_fifo_size ( ost - > fifo ) + size_out ) < 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " av_fifo_realloc2() failed \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
av_fifo_generic_write ( ost - > fifo , buftmp , size_out , NULL ) ;
frame_bytes = enc - > frame_size * osize * enc - > channels ;
while ( av_fifo_size ( ost - > fifo ) > = frame_bytes ) {
av_fifo_generic_read ( ost - > fifo , audio_buf , frame_bytes , NULL ) ;
2011-12-31 17:36:25 +03:00
encode_audio_frame ( s , ost , audio_buf , frame_bytes ) ;
2011-07-27 21:56:59 +03:00
}
} else {
2011-12-31 17:36:25 +03:00
encode_audio_frame ( s , ost , buftmp , size_out ) ;
2011-07-27 21:56:59 +03:00
}
}
static void pre_process_video_frame ( InputStream * ist , AVPicture * picture , void * * bufp )
{
AVCodecContext * dec ;
AVPicture * picture2 ;
AVPicture picture_tmp ;
uint8_t * buf = 0 ;
dec = ist - > st - > codec ;
/* deinterlace : must be done before any resize */
if ( do_deinterlace ) {
int size ;
/* create temporary picture */
size = avpicture_get_size ( dec - > pix_fmt , dec - > width , dec - > height ) ;
2011-12-29 23:28:29 +03:00
buf = av_malloc ( size ) ;
2011-07-27 21:56:59 +03:00
if ( ! buf )
return ;
picture2 = & picture_tmp ;
avpicture_fill ( picture2 , buf , dec - > pix_fmt , dec - > width , dec - > height ) ;
2011-12-29 23:28:29 +03:00
if ( avpicture_deinterlace ( picture2 , picture ,
2011-07-27 21:56:59 +03:00
dec - > pix_fmt , dec - > width , dec - > height ) < 0 ) {
/* if error, do not deinterlace */
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_WARNING , " Deinterlacing failed \n " ) ;
2011-07-27 21:56:59 +03:00
av_free ( buf ) ;
buf = NULL ;
picture2 = picture ;
}
} else {
picture2 = picture ;
}
if ( picture ! = picture2 )
* picture = * picture2 ;
* bufp = buf ;
}
static void do_subtitle_out ( AVFormatContext * s ,
OutputStream * ost ,
InputStream * ist ,
AVSubtitle * sub ,
int64_t pts )
{
static uint8_t * subtitle_out = NULL ;
int subtitle_out_max_size = 1024 * 1024 ;
int subtitle_out_size , nb , i ;
AVCodecContext * enc ;
AVPacket pkt ;
if ( pts = = AV_NOPTS_VALUE ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_ERROR , " Subtitle packets must have a pts \n " ) ;
2011-07-27 21:56:59 +03:00
if ( exit_on_error )
exit_program ( 1 ) ;
return ;
}
enc = ost - > st - > codec ;
if ( ! subtitle_out ) {
subtitle_out = av_malloc ( subtitle_out_max_size ) ;
}
/* Note: DVB subtitle need one packet to draw them and one other
packet to clear them */
/* XXX: signal it in the codec context ? */
if ( enc - > codec_id = = CODEC_ID_DVB_SUBTITLE )
nb = 2 ;
else
nb = 1 ;
2011-12-29 23:28:29 +03:00
for ( i = 0 ; i < nb ; i + + ) {
2012-02-05 16:32:10 +03:00
ost - > sync_opts = av_rescale_q ( pts , ist - > st - > time_base , enc - > time_base ) ;
if ( ! check_recording_time ( ost ) )
return ;
2011-07-27 21:56:59 +03:00
sub - > pts = av_rescale_q ( pts , ist - > st - > time_base , AV_TIME_BASE_Q ) ;
// start_display_time is required to be 0
2011-12-29 23:28:29 +03:00
sub - > pts + = av_rescale_q ( sub - > start_display_time , ( AVRational ) { 1 , 1000 } , AV_TIME_BASE_Q ) ;
sub - > end_display_time - = sub - > start_display_time ;
2011-07-27 21:56:59 +03:00
sub - > start_display_time = 0 ;
subtitle_out_size = avcodec_encode_subtitle ( enc , subtitle_out ,
subtitle_out_max_size , sub ) ;
if ( subtitle_out_size < 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Subtitle encoding failed \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
av_init_packet ( & pkt ) ;
pkt . data = subtitle_out ;
pkt . size = subtitle_out_size ;
2011-12-29 23:28:29 +03:00
pkt . pts = av_rescale_q ( sub - > pts , AV_TIME_BASE_Q , ost - > st - > time_base ) ;
2011-07-27 21:56:59 +03:00
if ( enc - > codec_id = = CODEC_ID_DVB_SUBTITLE ) {
/* XXX: the pts correction is handled here. Maybe handling
it in the codec would be better */
if ( i = = 0 )
pkt . pts + = 90 * sub - > start_display_time ;
else
pkt . pts + = 90 * sub - > end_display_time ;
}
2012-01-01 23:51:26 +03:00
write_frame ( s , & pkt , ost ) ;
2011-07-27 21:56:59 +03:00
}
}
2011-08-17 04:43:27 +03:00
static void do_video_out ( AVFormatContext * s ,
OutputStream * ost ,
AVFrame * in_picture ,
int * frame_size , float quality )
{
2011-08-17 05:06:46 +03:00
int nb_frames , i , ret , format_video_sync ;
2011-08-30 05:28:42 +03:00
AVCodecContext * enc ;
2012-02-06 23:47:41 +03:00
double sync_ipts , delta ;
2011-08-17 04:43:27 +03:00
enc = ost - > st - > codec ;
2012-02-05 22:56:24 +03:00
sync_ipts = get_sync_ipts ( ost , in_picture - > pts ) / av_q2d ( enc - > time_base ) ;
2012-02-06 23:47:41 +03:00
delta = sync_ipts - ost - > sync_opts ;
2011-08-17 04:43:27 +03:00
/* by default, we output a single frame */
nb_frames = 1 ;
* frame_size = 0 ;
2011-08-17 05:06:46 +03:00
format_video_sync = video_sync_method ;
2011-11-26 13:01:33 +03:00
if ( format_video_sync = = VSYNC_AUTO )
format_video_sync = ( s - > oformat - > flags & AVFMT_NOTIMESTAMPS ) ? VSYNC_PASSTHROUGH :
( s - > oformat - > flags & AVFMT_VARIABLE_FPS ) ? VSYNC_VFR : VSYNC_CFR ;
2011-08-17 05:06:46 +03:00
2012-02-06 23:47:41 +03:00
switch ( format_video_sync ) {
case VSYNC_CFR :
2011-12-29 23:28:29 +03:00
// FIXME set to 0.5 after we fix some dts/pts bugs like in avidec.c
2012-02-06 23:47:41 +03:00
if ( delta < - 1.1 )
2011-08-17 04:43:27 +03:00
nb_frames = 0 ;
2012-02-06 23:47:41 +03:00
else if ( delta > 1.1 )
nb_frames = lrintf ( delta ) ;
break ;
case VSYNC_VFR :
if ( delta < = - 0.6 )
nb_frames = 0 ;
else if ( delta > 0.6 )
ost - > sync_opts = lrintf ( sync_ipts ) ;
break ;
case VSYNC_PASSTHROUGH :
2011-12-29 23:28:29 +03:00
ost - > sync_opts = lrintf ( sync_ipts ) ;
2012-02-06 23:47:41 +03:00
break ;
default :
av_assert0 ( 0 ) ;
}
2011-08-17 04:43:27 +03:00
2011-08-29 10:16:42 +03:00
nb_frames = FFMIN ( nb_frames , ost - > max_frames - ost - > frame_number ) ;
2012-02-06 23:47:41 +03:00
if ( nb_frames = = 0 ) {
nb_frames_drop + + ;
av_log ( NULL , AV_LOG_VERBOSE , " *** drop! \n " ) ;
2011-08-17 04:43:27 +03:00
return ;
2012-02-06 23:47:41 +03:00
} else if ( nb_frames > 1 ) {
nb_frames_dup + = nb_frames - 1 ;
av_log ( NULL , AV_LOG_VERBOSE , " *** %d dup! \n " , nb_frames - 1 ) ;
}
2011-08-17 04:43:27 +03:00
2012-02-05 16:32:10 +03:00
if ( ! ost - > frame_number )
ost - > first_pts = ost - > sync_opts ;
2011-07-27 21:56:59 +03:00
/* duplicates frame if needed */
2011-12-29 23:28:29 +03:00
for ( i = 0 ; i < nb_frames ; i + + ) {
2011-07-27 21:56:59 +03:00
AVPacket pkt ;
av_init_packet ( & pkt ) ;
2012-02-01 12:51:36 +03:00
pkt . data = NULL ;
pkt . size = 0 ;
2011-07-27 21:56:59 +03:00
2012-02-05 16:32:10 +03:00
if ( ! check_recording_time ( ost ) )
return ;
2011-11-29 21:38:02 +03:00
if ( s - > oformat - > flags & AVFMT_RAWPICTURE & &
enc - > codec - > id = = CODEC_ID_RAWVIDEO ) {
2011-07-27 21:56:59 +03:00
/* raw pictures are written as AVPicture structure to
2011-08-18 22:09:17 +03:00
avoid any copies . We support temporarily the older
2011-07-27 21:56:59 +03:00
method . */
2011-08-30 05:28:42 +03:00
enc - > coded_frame - > interlaced_frame = in_picture - > interlaced_frame ;
enc - > coded_frame - > top_field_first = in_picture - > top_field_first ;
2012-03-14 11:20:17 +03:00
pkt . data = ( uint8_t * ) in_picture ;
2011-12-29 23:28:29 +03:00
pkt . size = sizeof ( AVPicture ) ;
pkt . pts = av_rescale_q ( ost - > sync_opts , enc - > time_base , ost - > st - > time_base ) ;
2011-07-27 21:56:59 +03:00
pkt . flags | = AV_PKT_FLAG_KEY ;
2012-01-01 23:51:26 +03:00
write_frame ( s , & pkt , ost ) ;
2011-07-27 21:56:59 +03:00
} else {
2012-02-01 12:51:36 +03:00
int got_packet ;
2011-07-27 21:56:59 +03:00
AVFrame big_picture ;
2012-03-14 11:20:17 +03:00
big_picture = * in_picture ;
2011-07-27 21:56:59 +03:00
/* better than nothing: use input picture interlaced
settings */
big_picture . interlaced_frame = in_picture - > interlaced_frame ;
if ( ost - > st - > codec - > flags & ( CODEC_FLAG_INTERLACED_DCT | CODEC_FLAG_INTERLACED_ME ) ) {
2011-09-02 19:03:31 +03:00
if ( ost - > top_field_first = = - 1 )
2011-07-27 21:56:59 +03:00
big_picture . top_field_first = in_picture - > top_field_first ;
else
2011-09-02 19:03:31 +03:00
big_picture . top_field_first = ! ! ost - > top_field_first ;
2011-07-27 21:56:59 +03:00
}
2011-08-09 12:32:16 +03:00
/* handles same_quant here. This is not correct because it may
2011-07-27 21:56:59 +03:00
not be a global option */
big_picture . quality = quality ;
2011-09-02 18:06:30 +03:00
if ( ! enc - > me_threshold )
2011-07-27 21:56:59 +03:00
big_picture . pict_type = 0 ;
2011-12-29 23:28:29 +03:00
big_picture . pts = ost - > sync_opts ;
2011-07-27 21:56:59 +03:00
if ( ost - > forced_kf_index < ost - > forced_kf_count & &
big_picture . pts > = ost - > forced_kf_pts [ ost - > forced_kf_index ] ) {
big_picture . pict_type = AV_PICTURE_TYPE_I ;
ost - > forced_kf_index + + ;
}
2012-02-01 12:51:36 +03:00
ret = avcodec_encode_video2 ( enc , & pkt , & big_picture , & got_packet ) ;
2011-07-27 21:56:59 +03:00
if ( ret < 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Video encoding failed \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2012-02-01 12:51:36 +03:00
if ( got_packet ) {
if ( pkt . pts ! = AV_NOPTS_VALUE )
pkt . pts = av_rescale_q ( pkt . pts , enc - > time_base , ost - > st - > time_base ) ;
if ( pkt . dts ! = AV_NOPTS_VALUE )
pkt . dts = av_rescale_q ( pkt . dts , enc - > time_base , ost - > st - > time_base ) ;
2011-07-27 21:56:59 +03:00
2012-01-01 23:51:26 +03:00
write_frame ( s , & pkt , ost ) ;
2012-03-06 10:57:58 +03:00
* frame_size = pkt . size ;
video_size + = pkt . size ;
2012-02-01 12:25:19 +03:00
2011-07-27 21:56:59 +03:00
/* if two pass, output log */
if ( ost - > logfile & & enc - > stats_out ) {
fprintf ( ost - > logfile , " %s " , enc - > stats_out ) ;
}
}
}
ost - > sync_opts + + ;
2012-01-02 11:22:41 +03:00
/*
* For video , number of frames in = = number of packets out .
* But there may be reordering , so we can ' t throw away frames on encoder
* flush , we need to limit them here , before they go into encoder .
*/
ost - > frame_number + + ;
2011-07-27 21:56:59 +03:00
}
}
2011-12-29 23:28:29 +03:00
static double psnr ( double d )
{
return - 10.0 * log ( d ) / log ( 10.0 ) ;
2011-07-27 21:56:59 +03:00
}
static void do_video_stats ( AVFormatContext * os , OutputStream * ost ,
int frame_size )
{
AVCodecContext * enc ;
int frame_number ;
double ti1 , bitrate , avg_bitrate ;
/* this is executed just the first time do_video_stats is called */
if ( ! vstats_file ) {
vstats_file = fopen ( vstats_filename , " w " ) ;
if ( ! vstats_file ) {
perror ( " fopen " ) ;
exit_program ( 1 ) ;
}
}
enc = ost - > st - > codec ;
if ( enc - > codec_type = = AVMEDIA_TYPE_VIDEO ) {
frame_number = ost - > frame_number ;
2011-12-29 23:28:29 +03:00
fprintf ( vstats_file , " frame= %5d q= %2.1f " , frame_number , enc - > coded_frame - > quality / ( float ) FF_QP2LAMBDA ) ;
2011-07-27 21:56:59 +03:00
if ( enc - > flags & CODEC_FLAG_PSNR )
2011-12-29 23:28:29 +03:00
fprintf ( vstats_file , " PSNR= %6.2f " , psnr ( enc - > coded_frame - > error [ 0 ] / ( enc - > width * enc - > height * 255.0 * 255.0 ) ) ) ;
2011-07-27 21:56:59 +03:00
fprintf ( vstats_file , " f_size= %6d " , frame_size ) ;
/* compute pts value */
ti1 = ost - > sync_opts * av_q2d ( enc - > time_base ) ;
if ( ti1 < 0.01 )
ti1 = 0.01 ;
2011-12-29 23:28:29 +03:00
bitrate = ( frame_size * 8 ) / av_q2d ( enc - > time_base ) / 1000.0 ;
2011-07-27 21:56:59 +03:00
avg_bitrate = ( double ) ( video_size * 8 ) / ti1 / 1000.0 ;
fprintf ( vstats_file , " s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s " ,
2011-12-29 23:28:29 +03:00
( double ) video_size / 1024 , ti1 , bitrate , avg_bitrate ) ;
2011-07-27 21:56:59 +03:00
fprintf ( vstats_file , " type= %c \n " , av_get_picture_type_char ( enc - > coded_frame - > pict_type ) ) ;
}
}
2012-03-29 09:51:17 +03:00
/* check for new output on any of the filtergraphs */
static int poll_filters ( void )
{
AVFilterBufferRef * picref ;
AVFrame * filtered_frame = NULL ;
int i , frame_size , ret ;
for ( i = 0 ; i < nb_output_streams ; i + + ) {
OutputStream * ost = output_streams [ i ] ;
OutputFile * of = output_files [ ost - > file_index ] ;
if ( ! ost - > filter | | ost - > is_past_recording_time )
continue ;
if ( ! ost - > filtered_frame & & ! ( ost - > filtered_frame = avcodec_alloc_frame ( ) ) ) {
return AVERROR ( ENOMEM ) ;
} else
avcodec_get_frame_defaults ( ost - > filtered_frame ) ;
filtered_frame = ost - > filtered_frame ;
while ( avfilter_poll_frame ( ost - > filter - > filter - > inputs [ 0 ] ) ) {
AVRational ist_pts_tb ;
if ( ( ret = get_filtered_video_frame ( ost - > filter - > filter ,
filtered_frame , & picref ,
& ist_pts_tb ) ) < 0 )
return ret ;
filtered_frame - > pts = av_rescale_q ( picref - > pts , ist_pts_tb , AV_TIME_BASE_Q ) ;
if ( of - > start_time & & filtered_frame - > pts < of - > start_time )
return 0 ;
switch ( ost - > filter - > filter - > inputs [ 0 ] - > type ) {
case AVMEDIA_TYPE_VIDEO :
if ( ! ost - > frame_aspect_ratio )
ost - > st - > codec - > sample_aspect_ratio = picref - > video - > pixel_aspect ;
do_video_out ( of - > ctx , ost , filtered_frame , & frame_size ,
same_quant ? ost - > last_quality :
ost - > st - > codec - > global_quality ) ;
if ( vstats_filename & & frame_size )
do_video_stats ( of - > ctx , ost , frame_size ) ;
break ;
default :
// TODO support audio/subtitle filters
av_assert0 ( 0 ) ;
}
avfilter_unref_buffer ( picref ) ;
}
}
return 0 ;
}
2012-04-15 16:03:07 +03:00
static void print_report ( int is_last_report , int64_t timer_start )
2011-07-27 21:56:59 +03:00
{
char buf [ 1024 ] ;
OutputStream * ost ;
AVFormatContext * oc ;
int64_t total_size ;
AVCodecContext * enc ;
int frame_number , vid , i ;
double bitrate , ti1 , pts ;
static int64_t last_time = - 1 ;
static int qp_histogram [ 52 ] ;
2011-10-09 16:57:30 +03:00
if ( ! print_stats & & ! is_last_report )
return ;
2011-07-27 21:56:59 +03:00
if ( ! is_last_report ) {
int64_t cur_time ;
/* display the report every 0.5 seconds */
cur_time = av_gettime ( ) ;
if ( last_time = = - 1 ) {
last_time = cur_time ;
return ;
}
if ( ( cur_time - last_time ) < 500000 )
return ;
last_time = cur_time ;
}
2012-04-15 16:28:30 +03:00
oc = output_files [ 0 ] - > ctx ;
2011-07-27 21:56:59 +03:00
total_size = avio_size ( oc - > pb ) ;
2011-12-29 23:28:29 +03:00
if ( total_size < 0 ) // FIXME improve avio_size() so it works with non seekable output too
total_size = avio_tell ( oc - > pb ) ;
2011-07-27 21:56:59 +03:00
buf [ 0 ] = ' \0 ' ;
ti1 = 1e10 ;
vid = 0 ;
2012-04-15 16:03:07 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2011-07-27 21:56:59 +03:00
float q = - 1 ;
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
2011-07-27 21:56:59 +03:00
enc = ost - > st - > codec ;
2011-10-23 12:22:33 +03:00
if ( ! ost - > stream_copy & & enc - > coded_frame )
2011-12-29 23:28:29 +03:00
q = enc - > coded_frame - > quality / ( float ) FF_QP2LAMBDA ;
2011-07-27 21:56:59 +03:00
if ( vid & & enc - > codec_type = = AVMEDIA_TYPE_VIDEO ) {
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " q=%2.1f " , q ) ;
}
if ( ! vid & & enc - > codec_type = = AVMEDIA_TYPE_VIDEO ) {
2011-12-29 23:28:29 +03:00
float t = ( av_gettime ( ) - timer_start ) / 1000000.0 ;
2011-07-27 21:56:59 +03:00
frame_number = ost - > frame_number ;
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " frame=%5d fps=%3d q=%3.1f " ,
2011-12-29 23:28:29 +03:00
frame_number , ( t > 1 ) ? ( int ) ( frame_number / t + 0.5 ) : 0 , q ) ;
if ( is_last_report )
2011-07-27 21:56:59 +03:00
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " L " ) ;
2011-12-29 23:28:29 +03:00
if ( qp_hist ) {
2011-07-27 21:56:59 +03:00
int j ;
int qp = lrintf ( q ) ;
2011-12-29 23:28:29 +03:00
if ( qp > = 0 & & qp < FF_ARRAY_ELEMS ( qp_histogram ) )
2011-07-27 21:56:59 +03:00
qp_histogram [ qp ] + + ;
2011-12-29 23:28:29 +03:00
for ( j = 0 ; j < 32 ; j + + )
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " %X " , ( int ) lrintf ( log ( qp_histogram [ j ] + 1 ) / log ( 2 ) ) ) ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
if ( enc - > flags & CODEC_FLAG_PSNR ) {
2011-07-27 21:56:59 +03:00
int j ;
2011-12-29 23:28:29 +03:00
double error , error_sum = 0 ;
double scale , scale_sum = 0 ;
char type [ 3 ] = { ' Y ' , ' U ' , ' V ' } ;
2011-07-27 21:56:59 +03:00
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " PSNR= " ) ;
2011-12-29 23:28:29 +03:00
for ( j = 0 ; j < 3 ; j + + ) {
if ( is_last_report ) {
error = enc - > error [ j ] ;
scale = enc - > width * enc - > height * 255.0 * 255.0 * frame_number ;
} else {
error = enc - > coded_frame - > error [ j ] ;
scale = enc - > width * enc - > height * 255.0 * 255.0 ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
if ( j )
scale / = 4 ;
2011-07-27 21:56:59 +03:00
error_sum + = error ;
scale_sum + = scale ;
2011-12-29 23:28:29 +03:00
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " %c:%2.2f " , type [ j ] , psnr ( error / scale ) ) ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " *:%2.2f " , psnr ( error_sum / scale_sum ) ) ;
2011-07-27 21:56:59 +03:00
}
vid = 1 ;
}
/* compute min output value */
pts = ( double ) ost - > st - > pts . val * av_q2d ( ost - > st - > time_base ) ;
if ( ( pts < ti1 ) & & ( pts > 0 ) )
ti1 = pts ;
}
if ( ti1 < 0.01 )
ti1 = 0.01 ;
2011-09-13 08:31:21 +03:00
bitrate = ( double ) ( total_size * 8 ) / ti1 / 1000.0 ;
2011-07-27 21:56:59 +03:00
2011-09-13 08:31:21 +03:00
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) ,
2011-07-27 21:56:59 +03:00
" size=%8.0fkB time=%0.2f bitrate=%6.1fkbits/s " ,
( double ) total_size / 1024 , ti1 , bitrate ) ;
2011-09-13 08:31:21 +03:00
if ( nb_frames_dup | | nb_frames_drop )
snprintf ( buf + strlen ( buf ) , sizeof ( buf ) - strlen ( buf ) , " dup=%d drop=%d " ,
nb_frames_dup , nb_frames_drop ) ;
2011-07-27 21:56:59 +03:00
2011-10-09 17:32:54 +03:00
av_log ( NULL , AV_LOG_INFO , " %s \r " , buf ) ;
2011-07-27 21:56:59 +03:00
2011-09-13 08:31:21 +03:00
fflush ( stderr ) ;
2011-07-27 21:56:59 +03:00
2011-09-13 08:31:21 +03:00
if ( is_last_report ) {
2011-07-27 21:56:59 +03:00
int64_t raw = audio_size + video_size + extra_size ;
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_INFO , " \n " ) ;
av_log ( NULL , AV_LOG_INFO , " video:%1.0fkB audio:%1.0fkB global headers:%1.0fkB muxing overhead %f%% \n " ,
2011-12-29 23:28:29 +03:00
video_size / 1024.0 ,
audio_size / 1024.0 ,
extra_size / 1024.0 ,
100.0 * ( total_size - raw ) / raw
2011-07-27 21:56:59 +03:00
) ;
}
}
2012-04-15 16:03:07 +03:00
static void flush_encoders ( void )
2011-08-18 09:55:29 +03:00
{
int i , ret ;
2012-04-15 16:03:07 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
OutputStream * ost = output_streams [ i ] ;
2011-08-16 12:19:20 +03:00
AVCodecContext * enc = ost - > st - > codec ;
2012-04-15 16:28:30 +03:00
AVFormatContext * os = output_files [ ost - > file_index ] - > ctx ;
2011-12-31 17:36:25 +03:00
int stop_encoding = 0 ;
2011-08-16 12:19:20 +03:00
2011-08-19 18:52:46 +03:00
if ( ! ost - > encoding_needed )
2011-08-16 12:19:20 +03:00
continue ;
2011-12-29 23:28:29 +03:00
if ( ost - > st - > codec - > codec_type = = AVMEDIA_TYPE_AUDIO & & enc - > frame_size < = 1 )
2011-08-16 12:19:20 +03:00
continue ;
2011-11-29 21:38:02 +03:00
if ( ost - > st - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO & & ( os - > oformat - > flags & AVFMT_RAWPICTURE ) & & enc - > codec - > id = = CODEC_ID_RAWVIDEO )
2011-08-16 12:19:20 +03:00
continue ;
2011-12-29 23:28:29 +03:00
for ( ; ; ) {
2011-08-16 12:19:20 +03:00
AVPacket pkt ;
2012-02-01 12:51:36 +03:00
int fifo_bytes , got_packet ;
2011-08-16 12:19:20 +03:00
av_init_packet ( & pkt ) ;
2011-12-31 17:36:25 +03:00
pkt . data = NULL ;
pkt . size = 0 ;
2011-08-16 12:19:20 +03:00
switch ( ost - > st - > codec - > codec_type ) {
case AVMEDIA_TYPE_AUDIO :
fifo_bytes = av_fifo_size ( ost - > fifo ) ;
if ( fifo_bytes > 0 ) {
2011-12-31 17:36:25 +03:00
/* encode any samples remaining in fifo */
int frame_bytes = fifo_bytes ;
2011-08-16 12:19:20 +03:00
av_fifo_generic_read ( ost - > fifo , audio_buf , fifo_bytes , NULL ) ;
2011-12-31 17:36:25 +03:00
/* pad last frame with silence if needed */
if ( ! ( enc - > codec - > capabilities & CODEC_CAP_SMALL_LAST_FRAME ) ) {
frame_bytes = enc - > frame_size * enc - > channels *
av_get_bytes_per_sample ( enc - > sample_fmt ) ;
2011-08-16 12:19:20 +03:00
if ( allocated_audio_buf_size < frame_bytes )
2011-08-18 09:55:29 +03:00
exit_program ( 1 ) ;
2011-08-16 12:19:20 +03:00
generate_silence ( audio_buf + fifo_bytes , enc - > sample_fmt , frame_bytes - fifo_bytes ) ;
2011-08-18 09:55:29 +03:00
}
2011-12-31 17:36:25 +03:00
encode_audio_frame ( os , ost , audio_buf , frame_bytes ) ;
} else {
/* flush encoder with NULL frames until it is done
returning packets */
if ( encode_audio_frame ( os , ost , NULL , 0 ) = = 0 ) {
stop_encoding = 1 ;
break ;
}
2011-08-16 12:19:20 +03:00
}
break ;
case AVMEDIA_TYPE_VIDEO :
2012-02-01 12:51:36 +03:00
ret = avcodec_encode_video2 ( enc , & pkt , NULL , & got_packet ) ;
2011-08-16 12:19:20 +03:00
if ( ret < 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Video encoding failed \n " ) ;
2011-08-16 12:19:20 +03:00
exit_program ( 1 ) ;
}
video_size + = ret ;
if ( ost - > logfile & & enc - > stats_out ) {
fprintf ( ost - > logfile , " %s " , enc - > stats_out ) ;
}
2012-02-01 12:51:36 +03:00
if ( ! got_packet ) {
2011-12-31 17:36:25 +03:00
stop_encoding = 1 ;
break ;
}
2012-02-01 12:51:36 +03:00
if ( pkt . pts ! = AV_NOPTS_VALUE )
pkt . pts = av_rescale_q ( pkt . pts , enc - > time_base , ost - > st - > time_base ) ;
if ( pkt . dts ! = AV_NOPTS_VALUE )
pkt . dts = av_rescale_q ( pkt . dts , enc - > time_base , ost - > st - > time_base ) ;
2011-12-31 17:36:25 +03:00
write_frame ( os , & pkt , ost ) ;
2011-08-16 12:19:20 +03:00
break ;
default :
2011-12-31 17:36:25 +03:00
stop_encoding = 1 ;
2011-08-18 09:55:29 +03:00
}
2011-12-31 17:36:25 +03:00
if ( stop_encoding )
2011-08-16 12:19:20 +03:00
break ;
2011-08-18 09:55:29 +03:00
}
}
}
2011-11-21 15:48:45 +03:00
/*
* Check whether a packet from ist should be written into ost at this time
*/
static int check_output_constraints ( InputStream * ist , OutputStream * ost )
{
2012-04-15 16:28:30 +03:00
OutputFile * of = output_files [ ost - > file_index ] ;
int ist_index = input_files [ ist - > file_index ] - > ist_index + ist - > st - > index ;
2011-11-21 15:48:45 +03:00
if ( ost - > source_index ! = ist_index )
return 0 ;
2012-02-05 23:12:43 +03:00
if ( of - > start_time & & ist - > last_dts < of - > start_time )
2011-11-21 15:48:45 +03:00
return 0 ;
return 1 ;
}
static void do_streamcopy ( InputStream * ist , OutputStream * ost , const AVPacket * pkt )
{
2012-04-15 16:28:30 +03:00
OutputFile * of = output_files [ ost - > file_index ] ;
2011-11-21 15:48:45 +03:00
int64_t ost_tb_start_time = av_rescale_q ( of - > start_time , AV_TIME_BASE_Q , ost - > st - > time_base ) ;
AVPacket opkt ;
av_init_packet ( & opkt ) ;
if ( ( ! ost - > frame_number & & ! ( pkt - > flags & AV_PKT_FLAG_KEY ) ) & &
! ost - > copy_initial_nonkeyframes )
return ;
2012-02-05 16:32:10 +03:00
if ( of - > recording_time ! = INT64_MAX & &
2012-02-05 23:12:43 +03:00
ist - > last_dts > = of - > recording_time + of - > start_time ) {
2012-02-05 16:32:10 +03:00
ost - > is_past_recording_time = 1 ;
return ;
}
2011-11-21 15:48:45 +03:00
/* force the input stream PTS */
if ( ost - > st - > codec - > codec_type = = AVMEDIA_TYPE_AUDIO )
audio_size + = pkt - > size ;
else if ( ost - > st - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO ) {
video_size + = pkt - > size ;
ost - > sync_opts + + ;
}
if ( pkt - > pts ! = AV_NOPTS_VALUE )
opkt . pts = av_rescale_q ( pkt - > pts , ist - > st - > time_base , ost - > st - > time_base ) - ost_tb_start_time ;
else
opkt . pts = AV_NOPTS_VALUE ;
if ( pkt - > dts = = AV_NOPTS_VALUE )
2012-02-05 23:12:43 +03:00
opkt . dts = av_rescale_q ( ist - > last_dts , AV_TIME_BASE_Q , ost - > st - > time_base ) ;
2011-11-21 15:48:45 +03:00
else
opkt . dts = av_rescale_q ( pkt - > dts , ist - > st - > time_base , ost - > st - > time_base ) ;
opkt . dts - = ost_tb_start_time ;
opkt . duration = av_rescale_q ( pkt - > duration , ist - > st - > time_base , ost - > st - > time_base ) ;
opkt . flags = pkt - > flags ;
2011-12-29 23:28:29 +03:00
// FIXME remove the following 2 lines they shall be replaced by the bitstream filters
if ( ost - > st - > codec - > codec_id ! = CODEC_ID_H264
2011-11-21 15:48:45 +03:00
& & ost - > st - > codec - > codec_id ! = CODEC_ID_MPEG1VIDEO
& & ost - > st - > codec - > codec_id ! = CODEC_ID_MPEG2VIDEO
2012-01-23 14:06:07 +03:00
& & ost - > st - > codec - > codec_id ! = CODEC_ID_VC1
2011-11-21 15:48:45 +03:00
) {
if ( av_parser_change ( ist - > st - > parser , ost - > st - > codec , & opkt . data , & opkt . size , pkt - > data , pkt - > size , pkt - > flags & AV_PKT_FLAG_KEY ) )
opkt . destruct = av_destruct_packet ;
} else {
opkt . data = pkt - > data ;
opkt . size = pkt - > size ;
}
2012-01-01 23:51:26 +03:00
write_frame ( of - > ctx , & opkt , ost ) ;
2011-11-21 15:48:45 +03:00
ost - > st - > codec - > frame_number + + ;
av_free_packet ( & opkt ) ;
}
2011-11-21 16:05:38 +03:00
static void rate_emu_sleep ( InputStream * ist )
{
2012-04-15 16:28:30 +03:00
if ( input_files [ ist - > file_index ] - > rate_emu ) {
2012-02-05 23:12:43 +03:00
int64_t pts = av_rescale ( ist - > last_dts , 1000000 , AV_TIME_BASE ) ;
2011-11-21 16:05:38 +03:00
int64_t now = av_gettime ( ) - ist - > start ;
if ( pts > now )
usleep ( pts - now ) ;
}
}
2011-11-21 16:39:22 +03:00
static int transcode_audio ( InputStream * ist , AVPacket * pkt , int * got_output )
{
2011-11-22 01:41:49 +03:00
AVFrame * decoded_frame ;
AVCodecContext * avctx = ist - > st - > codec ;
2011-11-21 16:39:22 +03:00
int bps = av_get_bytes_per_sample ( ist - > st - > codec - > sample_fmt ) ;
int i , ret ;
2011-12-05 19:49:38 +03:00
if ( ! ist - > decoded_frame & & ! ( ist - > decoded_frame = avcodec_alloc_frame ( ) ) )
2011-11-22 01:41:49 +03:00
return AVERROR ( ENOMEM ) ;
2011-12-05 19:49:38 +03:00
else
avcodec_get_frame_defaults ( ist - > decoded_frame ) ;
decoded_frame = ist - > decoded_frame ;
2011-11-21 16:39:22 +03:00
2011-11-22 01:41:49 +03:00
ret = avcodec_decode_audio4 ( avctx , decoded_frame , got_output , pkt ) ;
if ( ret < 0 ) {
2011-11-21 16:39:22 +03:00
return ret ;
2011-11-22 01:41:49 +03:00
}
2011-11-21 16:39:22 +03:00
if ( ! * got_output ) {
/* no audio frame */
2011-11-21 23:30:23 +03:00
return ret ;
2011-11-21 16:39:22 +03:00
}
2011-11-22 01:41:49 +03:00
/* if the decoder provides a pts, use it instead of the last packet pts.
the decoder could be delaying output by a packet or more . */
if ( decoded_frame - > pts ! = AV_NOPTS_VALUE )
2012-02-03 17:32:51 +03:00
ist - > next_dts = decoded_frame - > pts ;
2011-11-22 01:41:49 +03:00
2012-02-03 17:32:51 +03:00
/* increment next_dts to use for the case where the input stream does not
2011-11-22 01:41:49 +03:00
have timestamps or there are multiple frames in the packet */
2012-02-03 17:32:51 +03:00
ist - > next_dts + = ( ( int64_t ) AV_TIME_BASE * decoded_frame - > nb_samples ) /
2011-11-22 01:41:49 +03:00
avctx - > sample_rate ;
2011-11-21 16:39:22 +03:00
// preprocess audio (volume)
if ( audio_volume ! = 256 ) {
2011-11-22 01:41:49 +03:00
int decoded_data_size = decoded_frame - > nb_samples * avctx - > channels * bps ;
void * samples = decoded_frame - > data [ 0 ] ;
switch ( avctx - > sample_fmt ) {
2011-11-21 16:39:22 +03:00
case AV_SAMPLE_FMT_U8 :
{
uint8_t * volp = samples ;
for ( i = 0 ; i < ( decoded_data_size / sizeof ( * volp ) ) ; i + + ) {
int v = ( ( ( * volp - 128 ) * audio_volume + 128 ) > > 8 ) + 128 ;
* volp + + = av_clip_uint8 ( v ) ;
}
break ;
}
case AV_SAMPLE_FMT_S16 :
{
int16_t * volp = samples ;
for ( i = 0 ; i < ( decoded_data_size / sizeof ( * volp ) ) ; i + + ) {
int v = ( ( * volp ) * audio_volume + 128 ) > > 8 ;
* volp + + = av_clip_int16 ( v ) ;
}
break ;
}
case AV_SAMPLE_FMT_S32 :
{
int32_t * volp = samples ;
for ( i = 0 ; i < ( decoded_data_size / sizeof ( * volp ) ) ; i + + ) {
int64_t v = ( ( ( int64_t ) * volp * audio_volume + 128 ) > > 8 ) ;
* volp + + = av_clipl_int32 ( v ) ;
}
break ;
}
case AV_SAMPLE_FMT_FLT :
{
float * volp = samples ;
float scale = audio_volume / 256.f ;
for ( i = 0 ; i < ( decoded_data_size / sizeof ( * volp ) ) ; i + + ) {
* volp + + * = scale ;
}
break ;
}
case AV_SAMPLE_FMT_DBL :
{
double * volp = samples ;
double scale = audio_volume / 256. ;
for ( i = 0 ; i < ( decoded_data_size / sizeof ( * volp ) ) ; i + + ) {
* volp + + * = scale ;
}
break ;
}
default :
av_log ( NULL , AV_LOG_FATAL ,
" Audio volume adjustment on sample format %s is not supported. \n " ,
av_get_sample_fmt_name ( ist - > st - > codec - > sample_fmt ) ) ;
exit_program ( 1 ) ;
}
}
rate_emu_sleep ( ist ) ;
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
OutputStream * ost = output_streams [ i ] ;
2011-11-21 16:39:22 +03:00
if ( ! check_output_constraints ( ist , ost ) | | ! ost - > encoding_needed )
continue ;
2012-04-15 16:28:30 +03:00
do_audio_out ( output_files [ ost - > file_index ] - > ctx , ost , ist , decoded_frame ) ;
2011-11-21 16:39:22 +03:00
}
2011-12-05 18:36:54 +03:00
2011-11-21 23:30:23 +03:00
return ret ;
2011-11-21 16:39:22 +03:00
}
2011-11-21 16:39:22 +03:00
static int transcode_video ( InputStream * ist , AVPacket * pkt , int * got_output , int64_t * pkt_pts )
{
2012-03-29 09:51:17 +03:00
AVFrame * decoded_frame ;
2011-11-21 16:39:22 +03:00
void * buffer_to_free = NULL ;
2012-03-29 08:40:26 +03:00
int i , ret = 0 , resample_changed ;
2011-11-21 16:39:22 +03:00
float quality ;
2011-12-05 19:49:38 +03:00
if ( ! ist - > decoded_frame & & ! ( ist - > decoded_frame = avcodec_alloc_frame ( ) ) )
2011-11-21 16:39:22 +03:00
return AVERROR ( ENOMEM ) ;
2011-12-05 19:49:38 +03:00
else
avcodec_get_frame_defaults ( ist - > decoded_frame ) ;
decoded_frame = ist - > decoded_frame ;
2011-11-21 16:39:22 +03:00
pkt - > pts = * pkt_pts ;
2012-02-05 23:12:43 +03:00
pkt - > dts = ist - > last_dts ;
2011-11-21 16:39:22 +03:00
* pkt_pts = AV_NOPTS_VALUE ;
ret = avcodec_decode_video2 ( ist - > st - > codec ,
decoded_frame , got_output , pkt ) ;
if ( ret < 0 )
2011-12-05 19:49:38 +03:00
return ret ;
2011-11-21 16:39:22 +03:00
quality = same_quant ? decoded_frame - > quality : 0 ;
if ( ! * got_output ) {
/* no picture yet */
2012-04-01 20:01:37 +03:00
if ( ! pkt - > size )
for ( i = 0 ; i < ist - > nb_filters ; i + + )
av_buffersrc_buffer ( ist - > filters [ i ] - > filter , NULL ) ;
2011-11-21 23:30:23 +03:00
return ret ;
2011-11-21 16:39:22 +03:00
}
2012-02-02 14:21:37 +03:00
decoded_frame - > pts = guess_correct_pts ( & ist - > pts_ctx , decoded_frame - > pkt_pts ,
decoded_frame - > pkt_dts ) ;
2011-11-21 16:39:22 +03:00
pkt - > size = 0 ;
pre_process_video_frame ( ist , ( AVPicture * ) decoded_frame , & buffer_to_free ) ;
rate_emu_sleep ( ist ) ;
2012-03-22 11:37:33 +03:00
if ( ist - > st - > sample_aspect_ratio . num )
decoded_frame - > sample_aspect_ratio = ist - > st - > sample_aspect_ratio ;
2012-03-29 08:40:26 +03:00
resample_changed = ist - > resample_width ! = decoded_frame - > width | |
ist - > resample_height ! = decoded_frame - > height | |
ist - > resample_pix_fmt ! = decoded_frame - > format ;
if ( resample_changed ) {
av_log ( NULL , AV_LOG_INFO ,
" Input stream #%d:%d frame changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s \n " ,
ist - > file_index , ist - > st - > index ,
ist - > resample_width , ist - > resample_height , av_get_pix_fmt_name ( ist - > resample_pix_fmt ) ,
decoded_frame - > width , decoded_frame - > height , av_get_pix_fmt_name ( decoded_frame - > format ) ) ;
ist - > resample_width = decoded_frame - > width ;
ist - > resample_height = decoded_frame - > height ;
ist - > resample_pix_fmt = decoded_frame - > format ;
2012-04-01 16:08:33 +03:00
for ( i = 0 ; i < nb_filtergraphs ; i + + )
if ( ist_in_filtergraph ( filtergraphs [ i ] , ist ) & &
configure_filtergraph ( filtergraphs [ i ] ) < 0 ) {
2012-01-02 17:02:46 +03:00
av_log ( NULL , AV_LOG_FATAL , " Error reinitializing filters! \n " ) ;
exit_program ( 1 ) ;
2012-04-01 16:08:33 +03:00
}
}
2012-01-02 17:02:46 +03:00
2012-04-01 16:08:33 +03:00
for ( i = 0 ; i < ist - > nb_filters ; i + + ) {
2012-03-29 09:51:17 +03:00
// XXX what an ugly hack
if ( ist - > filters [ i ] - > graph - > nb_outputs = = 1 )
ist - > filters [ i ] - > graph - > outputs [ 0 ] - > ost - > last_quality = quality ;
2011-12-21 23:04:05 +03:00
if ( ist - > st - > codec - > codec - > capabilities & CODEC_CAP_DR1 ) {
FrameBuffer * buf = decoded_frame - > opaque ;
AVFilterBufferRef * fb = avfilter_get_video_buffer_ref_from_arrays (
decoded_frame - > data , decoded_frame - > linesize ,
AV_PERM_READ | AV_PERM_PRESERVE ,
ist - > st - > codec - > width , ist - > st - > codec - > height ,
ist - > st - > codec - > pix_fmt ) ;
avfilter_copy_frame_props ( fb , decoded_frame ) ;
fb - > buf - > priv = buf ;
fb - > buf - > free = filter_release_buffer ;
buf - > refcount + + ;
2012-03-29 09:51:17 +03:00
av_buffersrc_buffer ( ist - > filters [ i ] - > filter , fb ) ;
2011-12-21 23:04:05 +03:00
} else
2012-03-29 09:51:17 +03:00
av_vsrc_buffer_add_frame ( ist - > filters [ i ] - > filter , decoded_frame ,
2012-02-05 22:56:24 +03:00
decoded_frame - > pts , decoded_frame - > sample_aspect_ratio ) ;
2011-11-21 16:39:22 +03:00
}
av_free ( buffer_to_free ) ;
return ret ;
}
2011-11-21 16:39:22 +03:00
static int transcode_subtitles ( InputStream * ist , AVPacket * pkt , int * got_output )
{
AVSubtitle subtitle ;
int i , ret = avcodec_decode_subtitle2 ( ist - > st - > codec ,
& subtitle , got_output , pkt ) ;
if ( ret < 0 )
return ret ;
if ( ! * got_output )
2011-11-21 23:30:23 +03:00
return ret ;
2011-11-21 16:39:22 +03:00
rate_emu_sleep ( ist ) ;
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
OutputStream * ost = output_streams [ i ] ;
2011-11-21 16:39:22 +03:00
if ( ! check_output_constraints ( ist , ost ) | | ! ost - > encoding_needed )
continue ;
2012-04-15 16:28:30 +03:00
do_subtitle_out ( output_files [ ost - > file_index ] - > ctx , ost , ist , & subtitle , pkt - > pts ) ;
2011-11-21 16:39:22 +03:00
}
avsubtitle_free ( & subtitle ) ;
2011-11-21 23:30:23 +03:00
return ret ;
2011-11-21 16:39:22 +03:00
}
2011-07-27 21:56:59 +03:00
/* pkt = NULL means EOF (needed to flush decoder buffers) */
2012-04-15 16:03:07 +03:00
static int output_packet ( InputStream * ist , const AVPacket * pkt )
2011-07-27 21:56:59 +03:00
{
2011-11-22 09:13:29 +03:00
int i ;
2011-07-27 21:56:59 +03:00
int got_output ;
int64_t pkt_pts = AV_NOPTS_VALUE ;
AVPacket avpkt ;
2012-02-03 17:32:51 +03:00
if ( ist - > next_dts = = AV_NOPTS_VALUE )
2012-02-05 23:12:43 +03:00
ist - > next_dts = ist - > last_dts ;
2011-07-27 21:56:59 +03:00
if ( pkt = = NULL ) {
/* EOF handling */
av_init_packet ( & avpkt ) ;
avpkt . data = NULL ;
avpkt . size = 0 ;
goto handle_eof ;
} else {
avpkt = * pkt ;
}
2011-12-29 23:28:29 +03:00
if ( pkt - > dts ! = AV_NOPTS_VALUE )
2012-02-05 23:12:43 +03:00
ist - > next_dts = ist - > last_dts = av_rescale_q ( pkt - > dts , ist - > st - > time_base , AV_TIME_BASE_Q ) ;
2011-12-29 23:28:29 +03:00
if ( pkt - > pts ! = AV_NOPTS_VALUE )
2011-07-27 21:56:59 +03:00
pkt_pts = av_rescale_q ( pkt - > pts , ist - > st - > time_base , AV_TIME_BASE_Q ) ;
2011-12-29 23:28:29 +03:00
// while we have more to decode or while the decoder did output something on EOF
2011-11-21 16:05:38 +03:00
while ( ist - > decoding_needed & & ( avpkt . size > 0 | | ( ! pkt & & got_output ) ) ) {
2011-11-22 09:13:29 +03:00
int ret = 0 ;
2011-07-27 21:56:59 +03:00
handle_eof :
2012-02-05 23:12:43 +03:00
ist - > last_dts = ist - > next_dts ;
2011-11-22 09:16:23 +03:00
if ( avpkt . size & & avpkt . size ! = pkt - > size ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , ist - > showed_multi_packet_warning ? AV_LOG_VERBOSE : AV_LOG_WARNING ,
" Multiple frames in a packet from stream %d \n " , pkt - > stream_index ) ;
2011-11-22 09:16:23 +03:00
ist - > showed_multi_packet_warning = 1 ;
}
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
switch ( ist - > st - > codec - > codec_type ) {
2011-11-21 17:37:40 +03:00
case AVMEDIA_TYPE_AUDIO :
ret = transcode_audio ( ist , & avpkt , & got_output ) ;
break ;
case AVMEDIA_TYPE_VIDEO :
ret = transcode_video ( ist , & avpkt , & got_output , & pkt_pts ) ;
2012-02-02 14:21:37 +03:00
if ( avpkt . duration )
ist - > next_dts + = av_rescale_q ( avpkt . duration , ist - > st - > time_base , AV_TIME_BASE_Q ) ;
2012-02-02 15:18:49 +03:00
else if ( ist - > st - > r_frame_rate . num )
ist - > next_dts + = av_rescale_q ( 1 , ( AVRational ) { ist - > st - > r_frame_rate . den ,
ist - > st - > r_frame_rate . num } ,
AV_TIME_BASE_Q ) ;
2012-02-02 14:21:37 +03:00
else if ( ist - > st - > codec - > time_base . num ! = 0 ) {
int ticks = ist - > st - > parser ? ist - > st - > parser - > repeat_pict + 1 :
ist - > st - > codec - > ticks_per_frame ;
ist - > next_dts + = av_rescale_q ( ticks , ist - > st - > codec - > time_base , AV_TIME_BASE_Q ) ;
}
2011-11-21 17:37:40 +03:00
break ;
case AVMEDIA_TYPE_SUBTITLE :
ret = transcode_subtitles ( ist , & avpkt , & got_output ) ;
break ;
2011-11-21 16:06:25 +03:00
default :
return - 1 ;
}
2011-07-27 21:56:59 +03:00
2011-09-23 17:17:42 +03:00
if ( ret < 0 )
return ret ;
2011-11-24 02:18:33 +03:00
// touch data and size only if not EOF
if ( pkt ) {
avpkt . data + = ret ;
avpkt . size - = ret ;
}
2011-11-21 17:37:40 +03:00
if ( ! got_output ) {
2011-11-21 23:30:23 +03:00
continue ;
2011-11-21 17:37:40 +03:00
}
2011-07-27 21:56:59 +03:00
}
2011-11-21 15:48:45 +03:00
/* handle stream copy */
2011-11-21 16:05:38 +03:00
if ( ! ist - > decoding_needed ) {
rate_emu_sleep ( ist ) ;
2012-02-05 23:12:43 +03:00
ist - > last_dts = ist - > next_dts ;
2011-11-21 16:05:38 +03:00
switch ( ist - > st - > codec - > codec_type ) {
case AVMEDIA_TYPE_AUDIO :
2012-02-03 17:32:51 +03:00
ist - > next_dts + = ( ( int64_t ) AV_TIME_BASE * ist - > st - > codec - > frame_size ) /
2011-11-21 16:05:38 +03:00
ist - > st - > codec - > sample_rate ;
break ;
case AVMEDIA_TYPE_VIDEO :
if ( ist - > st - > codec - > time_base . num ! = 0 ) {
2011-12-29 23:28:29 +03:00
int ticks = ist - > st - > parser ? ist - > st - > parser - > repeat_pict + 1 : ist - > st - > codec - > ticks_per_frame ;
2012-02-03 17:32:51 +03:00
ist - > next_dts + = ( ( int64_t ) AV_TIME_BASE *
2011-11-21 16:05:38 +03:00
ist - > st - > codec - > time_base . num * ticks ) /
ist - > st - > codec - > time_base . den ;
}
break ;
}
}
2012-04-15 16:03:07 +03:00
for ( i = 0 ; pkt & & i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
OutputStream * ost = output_streams [ i ] ;
2011-11-21 15:48:45 +03:00
if ( ! check_output_constraints ( ist , ost ) | | ost - > encoding_needed )
continue ;
do_streamcopy ( ist , ost , pkt ) ;
}
2011-07-27 21:56:59 +03:00
return 0 ;
}
2012-04-15 16:03:07 +03:00
static void print_sdp ( void )
2011-07-27 21:56:59 +03:00
{
char sdp [ 2048 ] ;
2011-08-17 10:56:08 +03:00
int i ;
2012-04-15 16:03:07 +03:00
AVFormatContext * * avc = av_malloc ( sizeof ( * avc ) * nb_output_files ) ;
2011-08-17 10:56:08 +03:00
if ( ! avc )
exit_program ( 1 ) ;
2012-04-15 16:03:07 +03:00
for ( i = 0 ; i < nb_output_files ; i + + )
2012-04-15 16:28:30 +03:00
avc [ i ] = output_files [ i ] - > ctx ;
2011-07-27 21:56:59 +03:00
2012-04-15 16:03:07 +03:00
av_sdp_create ( avc , nb_output_files , sdp , sizeof ( sdp ) ) ;
2011-07-27 21:56:59 +03:00
printf ( " SDP: \n %s \n " , sdp ) ;
fflush ( stdout ) ;
2011-08-17 10:56:08 +03:00
av_freep ( & avc ) ;
2011-07-27 21:56:59 +03:00
}
2012-04-15 16:03:07 +03:00
static int init_input_stream ( int ist_index , char * error , int error_len )
2011-07-06 09:49:07 +03:00
{
int i ;
2012-04-15 16:28:30 +03:00
InputStream * ist = input_streams [ ist_index ] ;
2011-07-06 09:49:07 +03:00
if ( ist - > decoding_needed ) {
AVCodec * codec = ist - > dec ;
if ( ! codec ) {
2011-11-06 15:00:13 +03:00
snprintf ( error , error_len , " Decoder (codec id %d) not found for input stream #%d:%d " ,
2011-07-06 09:49:07 +03:00
ist - > st - > codec - > codec_id , ist - > file_index , ist - > st - > index ) ;
return AVERROR ( EINVAL ) ;
}
/* update requested sample format for the decoder based on the
corresponding encoder sample format */
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
OutputStream * ost = output_streams [ i ] ;
2011-07-06 09:49:07 +03:00
if ( ost - > source_index = = ist_index ) {
update_sample_fmt ( ist - > st - > codec , codec , ost - > st - > codec ) ;
break ;
}
}
2011-12-19 17:44:58 +03:00
if ( codec - > type = = AVMEDIA_TYPE_VIDEO & & codec - > capabilities & CODEC_CAP_DR1 ) {
ist - > st - > codec - > get_buffer = codec_get_buffer ;
ist - > st - > codec - > release_buffer = codec_release_buffer ;
ist - > st - > codec - > opaque = ist ;
}
2012-01-18 12:53:41 +03:00
if ( ! av_dict_get ( ist - > opts , " threads " , NULL , 0 ) )
av_dict_set ( & ist - > opts , " threads " , " auto " , 0 ) ;
2011-07-06 09:49:07 +03:00
if ( avcodec_open2 ( ist - > st - > codec , codec , & ist - > opts ) < 0 ) {
2011-11-06 15:00:13 +03:00
snprintf ( error , error_len , " Error while opening decoder for input stream #%d:%d " ,
2011-07-06 09:49:07 +03:00
ist - > file_index , ist - > st - > index ) ;
return AVERROR ( EINVAL ) ;
}
assert_codec_experimental ( ist - > st - > codec , 0 ) ;
assert_avoptions ( ist - > opts ) ;
2012-04-06 01:11:28 +03:00
if ( ist - > st - > codec - > codec_type = = AVMEDIA_TYPE_AUDIO ) {
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
OutputStream * ost = output_streams [ i ] ;
2012-04-06 01:11:28 +03:00
if ( ost - > source_index = = ist_index ) {
if ( ! ist - > st - > codec - > channel_layout | | ! ost - > st - > codec - > channel_layout )
get_default_channel_layouts ( ost , ist ) ;
break ;
}
}
}
2011-07-06 09:49:07 +03:00
}
2012-02-05 23:12:43 +03:00
ist - > last_dts = ist - > st - > avg_frame_rate . num ? - ist - > st - > codec - > has_b_frames * AV_TIME_BASE / av_q2d ( ist - > st - > avg_frame_rate ) : 0 ;
2012-02-03 17:32:51 +03:00
ist - > next_dts = AV_NOPTS_VALUE ;
2011-07-06 09:49:07 +03:00
init_pts_correction ( & ist - > pts_ctx ) ;
ist - > is_start = 1 ;
return 0 ;
}
2012-04-01 16:08:33 +03:00
static InputStream * get_input_stream ( OutputStream * ost )
{
if ( ost - > source_index > = 0 )
return input_streams [ ost - > source_index ] ;
if ( ost - > filter ) {
FilterGraph * fg = ost - > filter - > graph ;
int i ;
for ( i = 0 ; i < fg - > nb_inputs ; i + + )
if ( fg - > inputs [ i ] - > ist - > st - > codec - > codec_type = = ost - > st - > codec - > codec_type )
return fg - > inputs [ i ] - > ist ;
}
return NULL ;
}
2012-04-15 16:03:07 +03:00
static int transcode_init ( void )
2011-07-27 21:56:59 +03:00
{
2011-08-31 23:24:06 +03:00
int ret = 0 , i , j , k ;
2011-11-20 14:23:55 +03:00
AVFormatContext * oc ;
2011-07-27 21:56:59 +03:00
AVCodecContext * codec , * icodec ;
2011-08-17 11:21:37 +03:00
OutputStream * ost ;
2011-07-27 21:56:59 +03:00
InputStream * ist ;
char error [ 1024 ] ;
int want_sdp = 1 ;
2011-08-17 11:27:17 +03:00
2011-08-29 10:16:42 +03:00
/* init framerate emulation */
for ( i = 0 ; i < nb_input_files ; i + + ) {
2012-04-15 16:28:30 +03:00
InputFile * ifile = input_files [ i ] ;
2011-08-29 10:16:42 +03:00
if ( ifile - > rate_emu )
for ( j = 0 ; j < ifile - > nb_streams ; j + + )
2012-04-15 16:28:30 +03:00
input_streams [ j + ifile - > ist_index ] - > start = av_gettime ( ) ;
2011-08-29 10:16:42 +03:00
}
2011-07-27 21:56:59 +03:00
/* output stream init */
2011-11-20 14:32:27 +03:00
for ( i = 0 ; i < nb_output_files ; i + + ) {
2012-04-15 16:28:30 +03:00
oc = output_files [ i ] - > ctx ;
2011-11-20 14:23:55 +03:00
if ( ! oc - > nb_streams & & ! ( oc - > oformat - > flags & AVFMT_NOSTREAMS ) ) {
av_dump_format ( oc , i , oc - > filename , 1 ) ;
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_ERROR , " Output file #%d does not contain any stream \n " , i ) ;
2011-08-20 00:45:02 +03:00
return AVERROR ( EINVAL ) ;
2011-07-27 21:56:59 +03:00
}
}
2012-04-01 16:08:33 +03:00
/* init complex filtergraphs */
for ( i = 0 ; i < nb_filtergraphs ; i + + )
if ( ( ret = avfilter_graph_config ( filtergraphs [ i ] - > graph , NULL ) ) < 0 )
return ret ;
2011-07-27 21:56:59 +03:00
/* for each output stream, we compute the right encoding parameters */
2011-08-17 11:21:37 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
oc = output_files [ ost - > file_index ] - > ctx ;
2012-04-01 16:08:33 +03:00
ist = get_input_stream ( ost ) ;
2011-07-27 21:56:59 +03:00
2011-10-30 09:10:08 +03:00
if ( ost - > attachment_filename )
continue ;
2011-11-20 14:32:27 +03:00
codec = ost - > st - > codec ;
2011-07-27 21:56:59 +03:00
2012-04-01 16:08:33 +03:00
if ( ist ) {
icodec = ist - > st - > codec ;
ost - > st - > disposition = ist - > st - > disposition ;
codec - > bits_per_raw_sample = icodec - > bits_per_raw_sample ;
codec - > chroma_sample_location = icodec - > chroma_sample_location ;
}
2011-07-27 21:56:59 +03:00
2011-10-23 12:22:33 +03:00
if ( ost - > stream_copy ) {
2012-04-21 08:03:06 +03:00
uint64_t extra_size ;
av_assert0 ( ist & & ! ost - > filter ) ;
extra_size = ( uint64_t ) icodec - > extradata_size + FF_INPUT_BUFFER_PADDING_SIZE ;
2011-07-27 21:56:59 +03:00
2011-08-19 21:26:21 +03:00
if ( extra_size > INT_MAX ) {
2011-08-20 00:45:02 +03:00
return AVERROR ( EINVAL ) ;
2011-08-19 21:26:21 +03:00
}
2011-07-27 21:56:59 +03:00
/* if stream_copy is selected, no need to decode or encode */
2011-11-20 14:32:27 +03:00
codec - > codec_id = icodec - > codec_id ;
2011-07-27 21:56:59 +03:00
codec - > codec_type = icodec - > codec_type ;
2011-11-20 14:32:27 +03:00
if ( ! codec - > codec_tag ) {
if ( ! oc - > oformat - > codec_tag | |
av_codec_get_id ( oc - > oformat - > codec_tag , icodec - > codec_tag ) = = codec - > codec_id | |
av_codec_get_tag ( oc - > oformat - > codec_tag , icodec - > codec_id ) < = 0 )
2011-07-27 21:56:59 +03:00
codec - > codec_tag = icodec - > codec_tag ;
}
2011-11-20 14:32:27 +03:00
codec - > bit_rate = icodec - > bit_rate ;
2011-07-27 21:56:59 +03:00
codec - > rc_max_rate = icodec - > rc_max_rate ;
codec - > rc_buffer_size = icodec - > rc_buffer_size ;
2011-12-14 05:49:06 +03:00
codec - > field_order = icodec - > field_order ;
2011-11-20 14:32:27 +03:00
codec - > extradata = av_mallocz ( extra_size ) ;
2011-08-19 21:26:21 +03:00
if ( ! codec - > extradata ) {
2011-08-20 00:45:02 +03:00
return AVERROR ( ENOMEM ) ;
2011-08-19 21:26:21 +03:00
}
2011-07-27 21:56:59 +03:00
memcpy ( codec - > extradata , icodec - > extradata , icodec - > extradata_size ) ;
2011-11-20 14:32:27 +03:00
codec - > extradata_size = icodec - > extradata_size ;
2011-11-29 13:10:31 +03:00
if ( ! copy_tb ) {
2011-11-20 14:32:27 +03:00
codec - > time_base = icodec - > time_base ;
2011-07-27 21:56:59 +03:00
codec - > time_base . num * = icodec - > ticks_per_frame ;
av_reduce ( & codec - > time_base . num , & codec - > time_base . den ,
codec - > time_base . num , codec - > time_base . den , INT_MAX ) ;
2011-11-20 14:32:27 +03:00
} else
2011-07-27 21:56:59 +03:00
codec - > time_base = ist - > st - > time_base ;
2011-11-20 14:32:27 +03:00
2011-12-29 23:28:29 +03:00
switch ( codec - > codec_type ) {
2011-07-27 21:56:59 +03:00
case AVMEDIA_TYPE_AUDIO :
2011-12-29 23:28:29 +03:00
if ( audio_volume ! = 256 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " -acodec copy and -vol are incompatible (frames are not decoded) \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2011-11-20 14:32:27 +03:00
codec - > channel_layout = icodec - > channel_layout ;
codec - > sample_rate = icodec - > sample_rate ;
codec - > channels = icodec - > channels ;
codec - > frame_size = icodec - > frame_size ;
2011-07-27 21:56:59 +03:00
codec - > audio_service_type = icodec - > audio_service_type ;
2011-11-20 14:32:27 +03:00
codec - > block_align = icodec - > block_align ;
2011-07-27 21:56:59 +03:00
break ;
case AVMEDIA_TYPE_VIDEO :
2011-11-20 14:32:27 +03:00
codec - > pix_fmt = icodec - > pix_fmt ;
codec - > width = icodec - > width ;
codec - > height = icodec - > height ;
codec - > has_b_frames = icodec - > has_b_frames ;
2011-07-27 21:56:59 +03:00
if ( ! codec - > sample_aspect_ratio . num ) {
2011-11-20 14:32:27 +03:00
codec - > sample_aspect_ratio =
2011-07-27 21:56:59 +03:00
ost - > st - > sample_aspect_ratio =
ist - > st - > sample_aspect_ratio . num ? ist - > st - > sample_aspect_ratio :
ist - > st - > codec - > sample_aspect_ratio . num ?
ist - > st - > codec - > sample_aspect_ratio : ( AVRational ) { 0 , 1 } ;
}
break ;
case AVMEDIA_TYPE_SUBTITLE :
2011-11-20 14:32:27 +03:00
codec - > width = icodec - > width ;
2011-07-27 21:56:59 +03:00
codec - > height = icodec - > height ;
break ;
case AVMEDIA_TYPE_DATA :
2011-09-27 11:37:22 +03:00
case AVMEDIA_TYPE_ATTACHMENT :
2011-07-27 21:56:59 +03:00
break ;
default :
abort ( ) ;
}
} else {
2012-04-18 07:53:11 +03:00
if ( ! ost - > enc ) {
/* should only happen when a default codec is not present. */
snprintf ( error , sizeof ( error ) , " Automatic encoder selection "
" failed for output stream #%d:%d. Default encoder for "
" format %s is probably disabled. Please choose an "
" encoder manually. \n " , ost - > file_index , ost - > index ,
oc - > oformat - > name ) ;
ret = AVERROR ( EINVAL ) ;
goto dump_format ;
}
2011-11-20 14:32:27 +03:00
2012-04-01 16:08:33 +03:00
if ( ist )
ist - > decoding_needed = 1 ;
2011-09-27 09:16:26 +03:00
ost - > encoding_needed = 1 ;
2011-11-20 14:32:27 +03:00
2011-12-29 23:28:29 +03:00
switch ( codec - > codec_type ) {
2011-07-27 21:56:59 +03:00
case AVMEDIA_TYPE_AUDIO :
2011-11-20 14:32:27 +03:00
ost - > fifo = av_fifo_alloc ( 1024 ) ;
2011-08-19 21:26:21 +03:00
if ( ! ost - > fifo ) {
2011-08-20 00:45:02 +03:00
return AVERROR ( ENOMEM ) ;
2011-08-19 21:26:21 +03:00
}
2011-11-20 14:32:27 +03:00
2011-10-23 01:29:14 +03:00
if ( ! codec - > sample_rate )
2011-07-27 21:56:59 +03:00
codec - > sample_rate = icodec - > sample_rate ;
choose_sample_rate ( ost - > st , ost - > enc ) ;
2011-12-29 23:28:29 +03:00
codec - > time_base = ( AVRational ) { 1 , codec - > sample_rate } ;
2011-11-20 14:32:27 +03:00
2011-07-27 21:56:59 +03:00
if ( codec - > sample_fmt = = AV_SAMPLE_FMT_NONE )
codec - > sample_fmt = icodec - > sample_fmt ;
choose_sample_fmt ( ost - > st , ost - > enc ) ;
2011-11-20 14:32:27 +03:00
2011-07-27 21:56:59 +03:00
if ( ! codec - > channels )
codec - > channels = icodec - > channels ;
2012-04-05 21:06:28 +03:00
if ( ! codec - > channel_layout )
codec - > channel_layout = icodec - > channel_layout ;
2011-07-27 21:56:59 +03:00
if ( av_get_channel_layout_nb_channels ( codec - > channel_layout ) ! = codec - > channels )
codec - > channel_layout = 0 ;
2011-11-20 14:32:27 +03:00
icodec - > request_channels = codec - > channels ;
2011-07-27 21:56:59 +03:00
ost - > resample_sample_fmt = icodec - > sample_fmt ;
ost - > resample_sample_rate = icodec - > sample_rate ;
ost - > resample_channels = icodec - > channels ;
2012-04-05 21:06:28 +03:00
ost - > resample_channel_layout = icodec - > channel_layout ;
2011-07-27 21:56:59 +03:00
break ;
case AVMEDIA_TYPE_VIDEO :
2012-04-01 16:08:33 +03:00
if ( ! ost - > filter ) {
FilterGraph * fg ;
fg = init_simple_filtergraph ( ist , ost ) ;
if ( configure_video_filters ( fg ) ) {
av_log ( NULL , AV_LOG_FATAL , " Error opening filters! \n " ) ;
exit ( 1 ) ;
}
}
2012-02-07 13:03:33 +03:00
/*
* We want CFR output if and only if one of those is true :
* 1 ) user specified output framerate with - r
* 2 ) user specified - vsync cfr
* 3 ) output format is CFR and the user didn ' t force vsync to
* something else than CFR
*
* in such a case , set ost - > frame_rate
*/
2012-04-01 16:08:33 +03:00
if ( ! ost - > frame_rate . num & & ist & &
2012-02-07 13:03:33 +03:00
( video_sync_method = = VSYNC_CFR | |
( video_sync_method = = VSYNC_AUTO & &
! ( oc - > oformat - > flags & ( AVFMT_NOTIMESTAMPS | AVFMT_VARIABLE_FPS ) ) ) ) ) {
ost - > frame_rate = ist - > st - > r_frame_rate . num ? ist - > st - > r_frame_rate : ( AVRational ) { 25 , 1 } ;
if ( ost - > enc & & ost - > enc - > supported_framerates & & ! ost - > force_fps ) {
int idx = av_find_nearest_q_idx ( ost - > frame_rate , ost - > enc - > supported_framerates ) ;
ost - > frame_rate = ost - > enc - > supported_framerates [ idx ] ;
}
2011-07-27 21:56:59 +03:00
}
2012-02-07 13:03:33 +03:00
if ( ost - > frame_rate . num ) {
codec - > time_base = ( AVRational ) { ost - > frame_rate . den , ost - > frame_rate . num } ;
video_sync_method = VSYNC_CFR ;
2012-04-01 16:08:33 +03:00
} else if ( ist )
2012-02-07 13:03:33 +03:00
codec - > time_base = ist - > st - > time_base ;
2012-04-01 16:08:33 +03:00
else
codec - > time_base = ost - > filter - > filter - > inputs [ 0 ] - > time_base ;
2011-07-27 21:56:59 +03:00
2012-04-01 16:08:33 +03:00
codec - > width = ost - > filter - > filter - > inputs [ 0 ] - > w ;
codec - > height = ost - > filter - > filter - > inputs [ 0 ] - > h ;
codec - > sample_aspect_ratio = ost - > st - > sample_aspect_ratio =
ost - > frame_aspect_ratio ? // overridden by the -aspect cli option
av_d2q ( ost - > frame_aspect_ratio * codec - > height / codec - > width , 255 ) :
ost - > filter - > filter - > inputs [ 0 ] - > sample_aspect_ratio ;
codec - > pix_fmt = ost - > filter - > filter - > inputs [ 0 ] - > format ;
2012-03-23 17:46:30 +03:00
2012-03-23 22:59:09 +03:00
if ( codec - > width ! = icodec - > width | |
codec - > height ! = icodec - > height | |
codec - > pix_fmt ! = icodec - > pix_fmt ) {
2012-03-23 17:46:30 +03:00
codec - > bits_per_raw_sample = 0 ;
}
2011-07-27 21:56:59 +03:00
break ;
case AVMEDIA_TYPE_SUBTITLE :
2012-02-05 16:28:19 +03:00
codec - > time_base = ( AVRational ) { 1 , 1000 } ;
2011-07-27 21:56:59 +03:00
break ;
default :
abort ( ) ;
break ;
}
/* two pass mode */
2011-09-27 09:21:48 +03:00
if ( ( codec - > flags & ( CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2 ) ) ) {
2011-07-27 21:56:59 +03:00
char logfilename [ 1024 ] ;
FILE * f ;
snprintf ( logfilename , sizeof ( logfilename ) , " %s-%d.log " ,
pass_logfilename_prefix ? pass_logfilename_prefix : DEFAULT_PASS_LOGFILENAME_PREFIX ,
i ) ;
2012-03-12 19:42:57 +03:00
if ( ! strcmp ( ost - > enc - > name , " libx264 " ) ) {
av_dict_set ( & ost - > opts , " stats " , logfilename , AV_DICT_DONT_OVERWRITE ) ;
} else {
2012-03-12 19:43:48 +03:00
if ( codec - > flags & CODEC_FLAG_PASS1 ) {
f = fopen ( logfilename , " wb " ) ;
if ( ! f ) {
av_log ( NULL , AV_LOG_FATAL , " Cannot write log file '%s' for pass-1 encoding: %s \n " ,
logfilename , strerror ( errno ) ) ;
exit_program ( 1 ) ;
}
ost - > logfile = f ;
} else {
char * logbuffer ;
size_t logbuffer_size ;
if ( cmdutils_read_file ( logfilename , & logbuffer , & logbuffer_size ) < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Error reading log file '%s' for pass-2 encoding \n " ,
logfilename ) ;
exit_program ( 1 ) ;
}
codec - > stats_in = logbuffer ;
2011-07-27 21:56:59 +03:00
}
2012-03-12 19:42:57 +03:00
}
2011-07-27 21:56:59 +03:00
}
}
}
/* open each encoder */
2011-08-17 11:21:37 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
2011-07-27 21:56:59 +03:00
if ( ost - > encoding_needed ) {
2011-11-20 14:32:27 +03:00
AVCodec * codec = ost - > enc ;
2012-04-01 16:08:33 +03:00
AVCodecContext * dec = NULL ;
if ( ( ist = get_input_stream ( ost ) ) )
dec = ist - > st - > codec ;
if ( dec & & dec - > subtitle_header ) {
2011-07-27 21:56:59 +03:00
ost - > st - > codec - > subtitle_header = av_malloc ( dec - > subtitle_header_size ) ;
if ( ! ost - > st - > codec - > subtitle_header ) {
ret = AVERROR ( ENOMEM ) ;
goto dump_format ;
}
memcpy ( ost - > st - > codec - > subtitle_header , dec - > subtitle_header , dec - > subtitle_header_size ) ;
ost - > st - > codec - > subtitle_header_size = dec - > subtitle_header_size ;
}
2012-01-18 12:53:41 +03:00
if ( ! av_dict_get ( ost - > opts , " threads " , NULL , 0 ) )
av_dict_set ( & ost - > opts , " threads " , " auto " , 0 ) ;
2011-07-27 21:56:59 +03:00
if ( avcodec_open2 ( ost - > st - > codec , codec , & ost - > opts ) < 0 ) {
2011-11-06 15:00:13 +03:00
snprintf ( error , sizeof ( error ) , " Error while opening encoder for output stream #%d:%d - maybe incorrect parameters such as bit_rate, rate, width or height " ,
2011-07-27 21:56:59 +03:00
ost - > file_index , ost - > index ) ;
ret = AVERROR ( EINVAL ) ;
goto dump_format ;
}
assert_codec_experimental ( ost - > st - > codec , 1 ) ;
assert_avoptions ( ost - > opts ) ;
if ( ost - > st - > codec - > bit_rate & & ost - > st - > codec - > bit_rate < 1000 )
av_log ( NULL , AV_LOG_WARNING , " The bitrate parameter is set too low. "
" It takes bits/s as argument, not kbits/s \n " ) ;
extra_size + = ost - > st - > codec - > extradata_size ;
2011-09-02 18:06:30 +03:00
if ( ost - > st - > codec - > me_threshold )
2012-04-15 16:28:30 +03:00
input_streams [ ost - > source_index ] - > st - > codec - > debug | = FF_DEBUG_MV ;
2011-07-27 21:56:59 +03:00
}
}
2011-07-06 09:49:07 +03:00
/* init input streams */
for ( i = 0 ; i < nb_input_streams ; i + + )
2012-04-15 16:03:07 +03:00
if ( ( ret = init_input_stream ( i , error , sizeof ( error ) ) ) < 0 )
2011-07-06 09:49:07 +03:00
goto dump_format ;
2011-07-27 21:56:59 +03:00
2011-08-31 23:24:06 +03:00
/* discard unused programs */
for ( i = 0 ; i < nb_input_files ; i + + ) {
2012-04-15 16:28:30 +03:00
InputFile * ifile = input_files [ i ] ;
2011-08-31 23:24:06 +03:00
for ( j = 0 ; j < ifile - > ctx - > nb_programs ; j + + ) {
AVProgram * p = ifile - > ctx - > programs [ j ] ;
int discard = AVDISCARD_ALL ;
for ( k = 0 ; k < p - > nb_stream_indexes ; k + + )
2012-04-15 16:28:30 +03:00
if ( ! input_streams [ ifile - > ist_index + p - > stream_index [ k ] ] - > discard ) {
2011-08-31 23:24:06 +03:00
discard = AVDISCARD_DEFAULT ;
break ;
}
p - > discard = discard ;
}
}
2011-07-27 21:56:59 +03:00
/* open files and write file headers */
2011-08-17 10:56:08 +03:00
for ( i = 0 ; i < nb_output_files ; i + + ) {
2012-04-15 16:28:30 +03:00
oc = output_files [ i ] - > ctx ;
2011-11-20 14:23:55 +03:00
oc - > interrupt_callback = int_cb ;
2012-04-15 16:28:30 +03:00
if ( avformat_write_header ( oc , & output_files [ i ] - > opts ) < 0 ) {
2011-07-27 21:56:59 +03:00
snprintf ( error , sizeof ( error ) , " Could not write header for output file #%d (incorrect codec parameters ?) " , i ) ;
ret = AVERROR ( EINVAL ) ;
goto dump_format ;
}
2012-04-15 16:28:30 +03:00
assert_avoptions ( output_files [ i ] - > opts ) ;
2011-11-20 14:23:55 +03:00
if ( strcmp ( oc - > oformat - > name , " rtp " ) ) {
2011-07-27 21:56:59 +03:00
want_sdp = 0 ;
}
}
dump_format :
/* dump the file output parameters - cannot be done before in case
of stream copy */
2011-11-20 14:32:27 +03:00
for ( i = 0 ; i < nb_output_files ; i + + ) {
2012-04-15 16:28:30 +03:00
av_dump_format ( output_files [ i ] - > ctx , i , output_files [ i ] - > ctx - > filename , 1 ) ;
2011-07-27 21:56:59 +03:00
}
/* dump the stream mapping */
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_INFO , " Stream mapping: \n " ) ;
2012-04-01 16:08:33 +03:00
for ( i = 0 ; i < nb_input_streams ; i + + ) {
ist = input_streams [ i ] ;
for ( j = 0 ; j < ist - > nb_filters ; j + + ) {
AVFilterLink * link = ist - > filters [ j ] - > filter - > outputs [ 0 ] ;
if ( ist - > filters [ j ] - > graph - > graph_desc ) {
av_log ( NULL , AV_LOG_INFO , " Stream #%d:%d (%s) -> %s " ,
ist - > file_index , ist - > st - > index , ist - > dec ? ist - > dec - > name : " ? " ,
link - > dst - > filter - > name ) ;
if ( link - > dst - > input_count > 1 )
av_log ( NULL , AV_LOG_INFO , " :%s " , link - > dstpad - > name ) ;
if ( nb_filtergraphs > 1 )
av_log ( NULL , AV_LOG_INFO , " (graph %d) " , ist - > filters [ j ] - > graph - > index ) ;
av_log ( NULL , AV_LOG_INFO , " \n " ) ;
}
}
}
2011-09-13 08:31:21 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
2011-10-30 09:10:08 +03:00
if ( ost - > attachment_filename ) {
/* an attached file */
av_log ( NULL , AV_LOG_INFO , " File %s -> Stream #%d:%d \n " ,
ost - > attachment_filename , ost - > file_index , ost - > index ) ;
continue ;
}
2012-04-01 16:08:33 +03:00
if ( ost - > filter & & ost - > filter - > graph - > graph_desc ) {
/* output from a complex graph */
AVFilterLink * link = ost - > filter - > filter - > inputs [ 0 ] ;
av_log ( NULL , AV_LOG_INFO , " %s " , link - > src - > filter - > name ) ;
if ( link - > src - > output_count > 1 )
av_log ( NULL , AV_LOG_INFO , " :%s " , link - > srcpad - > name ) ;
if ( nb_filtergraphs > 1 )
av_log ( NULL , AV_LOG_INFO , " (graph %d) " , ost - > filter - > graph - > index ) ;
av_log ( NULL , AV_LOG_INFO , " -> Stream #%d:%d (%s) \n " , ost - > file_index ,
ost - > index , ost - > enc ? ost - > enc - > name : " ? " ) ;
continue ;
}
2011-11-06 15:00:13 +03:00
av_log ( NULL , AV_LOG_INFO , " Stream #%d:%d -> #%d:%d " ,
2012-04-15 16:28:30 +03:00
input_streams [ ost - > source_index ] - > file_index ,
input_streams [ ost - > source_index ] - > st - > index ,
2011-09-13 08:31:21 +03:00
ost - > file_index ,
ost - > index ) ;
2012-04-15 16:28:30 +03:00
if ( ost - > sync_ist ! = input_streams [ ost - > source_index ] )
2011-11-06 15:00:13 +03:00
av_log ( NULL , AV_LOG_INFO , " [sync #%d:%d] " ,
2011-09-13 08:31:21 +03:00
ost - > sync_ist - > file_index ,
ost - > sync_ist - > st - > index ) ;
2011-10-23 12:22:33 +03:00
if ( ost - > stream_copy )
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_INFO , " (copy) " ) ;
else
2012-04-15 16:28:30 +03:00
av_log ( NULL , AV_LOG_INFO , " (%s -> %s) " , input_streams [ ost - > source_index ] - > dec ?
input_streams [ ost - > source_index ] - > dec - > name : " ? " ,
2011-09-13 08:31:21 +03:00
ost - > enc ? ost - > enc - > name : " ? " ) ;
av_log ( NULL , AV_LOG_INFO , " \n " ) ;
2011-07-27 21:56:59 +03:00
}
if ( ret ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_ERROR , " %s \n " , error ) ;
2011-08-20 00:45:02 +03:00
return ret ;
2011-07-27 21:56:59 +03:00
}
if ( want_sdp ) {
2012-04-15 16:03:07 +03:00
print_sdp ( ) ;
2011-07-27 21:56:59 +03:00
}
2011-08-20 00:45:02 +03:00
return 0 ;
}
/*
* The following code is the main loop of the file converter
*/
2012-04-15 16:03:07 +03:00
static int transcode ( void )
2011-08-20 00:45:02 +03:00
{
int ret , i ;
AVFormatContext * is , * os ;
OutputStream * ost ;
InputStream * ist ;
uint8_t * no_packet ;
2011-12-29 23:28:29 +03:00
int no_packet_count = 0 ;
2011-08-20 00:45:02 +03:00
int64_t timer_start ;
if ( ! ( no_packet = av_mallocz ( nb_input_files ) ) )
exit_program ( 1 ) ;
2012-04-15 16:03:07 +03:00
ret = transcode_init ( ) ;
2011-08-20 00:45:02 +03:00
if ( ret < 0 )
goto fail ;
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_INFO , " Press ctrl-c to stop encoding \n " ) ;
2011-07-27 21:56:59 +03:00
term_init ( ) ;
timer_start = av_gettime ( ) ;
2011-12-29 23:28:29 +03:00
for ( ; received_sigterm = = 0 ; ) {
2012-04-01 16:08:33 +03:00
int file_index , ist_index , past_recording_time = 1 ;
2011-07-27 21:56:59 +03:00
AVPacket pkt ;
2011-08-17 00:41:16 +03:00
int64_t ipts_min ;
2011-07-27 21:56:59 +03:00
2011-08-17 00:41:16 +03:00
ipts_min = INT64_MAX ;
2011-07-27 21:56:59 +03:00
2012-04-01 16:08:33 +03:00
/* check if there's any stream where output is still needed */
2011-08-17 11:21:37 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2011-08-19 18:52:44 +03:00
OutputFile * of ;
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
2012-04-01 16:08:33 +03:00
of = output_files [ ost - > file_index ] ;
os = output_files [ ost - > file_index ] - > ctx ;
if ( ost - > is_past_recording_time | |
2011-08-19 18:52:44 +03:00
( os - > pb & & avio_tell ( os - > pb ) > = of - > limit_filesize ) )
2011-07-27 21:56:59 +03:00
continue ;
2012-04-01 16:08:33 +03:00
if ( ost - > frame_number > ost - > max_frames ) {
int j ;
for ( j = 0 ; j < of - > ctx - > nb_streams ; j + + )
output_streams [ of - > ost_index + j ] - > is_past_recording_time = 1 ;
continue ;
}
past_recording_time = 0 ;
}
if ( past_recording_time )
break ;
/* select the stream that we must read now by looking at the
smallest output pts */
file_index = - 1 ;
for ( i = 0 ; i < nb_input_streams ; i + + ) {
int64_t ipts ;
ist = input_streams [ i ] ;
2012-02-05 23:12:43 +03:00
ipts = ist - > last_dts ;
2012-04-01 16:08:33 +03:00
if ( ist - > discard | | no_packet [ ist - > file_index ] )
continue ;
2012-04-15 16:28:30 +03:00
if ( ! input_files [ ist - > file_index ] - > eof_reached ) {
2011-12-29 23:28:29 +03:00
if ( ipts < ipts_min ) {
2011-07-27 21:56:59 +03:00
ipts_min = ipts ;
2012-04-01 16:08:33 +03:00
file_index = ist - > file_index ;
2011-07-27 21:56:59 +03:00
}
}
}
/* if none, if is finished */
if ( file_index < 0 ) {
2011-12-29 23:28:29 +03:00
if ( no_packet_count ) {
no_packet_count = 0 ;
2011-08-17 11:27:17 +03:00
memset ( no_packet , 0 , nb_input_files ) ;
2011-07-27 21:56:59 +03:00
usleep ( 10000 ) ;
continue ;
}
break ;
}
/* read a frame from it and output it in the fifo */
2012-04-15 16:28:30 +03:00
is = input_files [ file_index ] - > ctx ;
2011-12-29 23:28:29 +03:00
ret = av_read_frame ( is , & pkt ) ;
if ( ret = = AVERROR ( EAGAIN ) ) {
no_packet [ file_index ] = 1 ;
2011-07-27 21:56:59 +03:00
no_packet_count + + ;
continue ;
}
if ( ret < 0 ) {
2012-04-15 16:28:30 +03:00
input_files [ file_index ] - > eof_reached = 1 ;
2012-04-02 21:13:29 +03:00
for ( i = 0 ; i < input_files [ file_index ] - > nb_streams ; i + + ) {
ist = input_streams [ input_files [ file_index ] - > ist_index + i ] ;
if ( ist - > decoding_needed )
output_packet ( ist , NULL ) ;
}
2011-07-27 21:56:59 +03:00
if ( opt_shortest )
break ;
else
continue ;
}
2011-12-29 23:28:29 +03:00
no_packet_count = 0 ;
2011-08-17 11:27:17 +03:00
memset ( no_packet , 0 , nb_input_files ) ;
2011-07-27 21:56:59 +03:00
if ( do_pkt_dump ) {
av_pkt_dump_log2 ( NULL , AV_LOG_DEBUG , & pkt , do_hex_dump ,
is - > streams [ pkt . stream_index ] ) ;
}
/* the following test is needed in case new streams appear
dynamically in stream : we ignore them */
2012-04-15 16:28:30 +03:00
if ( pkt . stream_index > = input_files [ file_index ] - > nb_streams )
2011-07-27 21:56:59 +03:00
goto discard_packet ;
2012-04-15 16:28:30 +03:00
ist_index = input_files [ file_index ] - > ist_index + pkt . stream_index ;
ist = input_streams [ ist_index ] ;
2011-07-27 21:56:59 +03:00
if ( ist - > discard )
goto discard_packet ;
if ( pkt . dts ! = AV_NOPTS_VALUE )
2012-04-15 16:28:30 +03:00
pkt . dts + = av_rescale_q ( input_files [ ist - > file_index ] - > ts_offset , AV_TIME_BASE_Q , ist - > st - > time_base ) ;
2011-07-27 21:56:59 +03:00
if ( pkt . pts ! = AV_NOPTS_VALUE )
2012-04-15 16:28:30 +03:00
pkt . pts + = av_rescale_q ( input_files [ ist - > file_index ] - > ts_offset , AV_TIME_BASE_Q , ist - > st - > time_base ) ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
if ( pkt . pts ! = AV_NOPTS_VALUE )
2011-08-29 10:16:42 +03:00
pkt . pts * = ist - > ts_scale ;
2011-12-29 23:28:29 +03:00
if ( pkt . dts ! = AV_NOPTS_VALUE )
2011-08-29 10:16:42 +03:00
pkt . dts * = ist - > ts_scale ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
//fprintf(stderr, "next:%"PRId64" dts:%"PRId64" off:%"PRId64" %d\n",
2012-02-03 17:32:51 +03:00
// ist->next_dts,
2011-12-29 23:28:29 +03:00
// pkt.dts, input_files[ist->file_index].ts_offset,
// ist->st->codec->codec_type);
2012-02-03 17:32:51 +03:00
if ( pkt . dts ! = AV_NOPTS_VALUE & & ist - > next_dts ! = AV_NOPTS_VALUE
2011-07-27 21:56:59 +03:00
& & ( is - > iformat - > flags & AVFMT_TS_DISCONT ) ) {
2011-12-29 23:28:29 +03:00
int64_t pkt_dts = av_rescale_q ( pkt . dts , ist - > st - > time_base , AV_TIME_BASE_Q ) ;
2012-02-03 17:32:51 +03:00
int64_t delta = pkt_dts - ist - > next_dts ;
2012-02-05 23:12:43 +03:00
if ( ( FFABS ( delta ) > 1LL * dts_delta_threshold * AV_TIME_BASE | | pkt_dts + 1 < ist - > last_dts ) & & ! copy_ts ) {
2012-04-15 16:28:30 +03:00
input_files [ ist - > file_index ] - > ts_offset - = delta ;
2011-12-29 23:28:29 +03:00
av_log ( NULL , AV_LOG_DEBUG ,
" timestamp discontinuity % " PRId64 " , new offset= % " PRId64 " \n " ,
2012-04-15 16:28:30 +03:00
delta , input_files [ ist - > file_index ] - > ts_offset ) ;
2011-07-27 21:56:59 +03:00
pkt . dts - = av_rescale_q ( delta , AV_TIME_BASE_Q , ist - > st - > time_base ) ;
2011-12-29 23:28:29 +03:00
if ( pkt . pts ! = AV_NOPTS_VALUE )
2011-07-27 21:56:59 +03:00
pkt . pts - = av_rescale_q ( delta , AV_TIME_BASE_Q , ist - > st - > time_base ) ;
}
}
2011-12-29 23:28:29 +03:00
// fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->st->index, pkt.size);
2012-03-29 09:51:17 +03:00
if ( output_packet ( ist , & pkt ) < 0 | | poll_filters ( ) < 0 ) {
2011-11-06 15:00:13 +03:00
av_log ( NULL , AV_LOG_ERROR , " Error while decoding stream #%d:%d \n " ,
2011-09-13 08:31:21 +03:00
ist - > file_index , ist - > st - > index ) ;
2011-07-27 21:56:59 +03:00
if ( exit_on_error )
exit_program ( 1 ) ;
av_free_packet ( & pkt ) ;
2011-08-23 23:47:31 +03:00
continue ;
2011-07-27 21:56:59 +03:00
}
discard_packet :
av_free_packet ( & pkt ) ;
/* dump report by using the output first video and audio streams */
2012-04-15 16:03:07 +03:00
print_report ( 0 , timer_start ) ;
2011-07-27 21:56:59 +03:00
}
/* at the end of stream, we must flush the decoder buffers */
for ( i = 0 ; i < nb_input_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ist = input_streams [ i ] ;
2012-04-02 21:13:29 +03:00
if ( ! input_files [ ist - > file_index ] - > eof_reached & & ist - > decoding_needed ) {
2012-04-15 16:03:07 +03:00
output_packet ( ist , NULL ) ;
2011-07-27 21:56:59 +03:00
}
}
2012-03-29 09:51:17 +03:00
poll_filters ( ) ;
2012-04-15 16:03:07 +03:00
flush_encoders ( ) ;
2011-07-27 21:56:59 +03:00
term_exit ( ) ;
/* write the trailer if needed and close file */
2011-12-29 23:28:29 +03:00
for ( i = 0 ; i < nb_output_files ; i + + ) {
2012-04-15 16:28:30 +03:00
os = output_files [ i ] - > ctx ;
2011-07-27 21:56:59 +03:00
av_write_trailer ( os ) ;
}
/* dump report by using the first video and audio streams */
2012-04-15 16:03:07 +03:00
print_report ( 1 , timer_start ) ;
2011-07-27 21:56:59 +03:00
/* close each encoder */
2011-08-17 11:21:37 +03:00
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
2011-07-27 21:56:59 +03:00
if ( ost - > encoding_needed ) {
av_freep ( & ost - > st - > codec - > stats_in ) ;
avcodec_close ( ost - > st - > codec ) ;
}
}
/* close each decoder */
for ( i = 0 ; i < nb_input_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ist = input_streams [ i ] ;
2011-07-27 21:56:59 +03:00
if ( ist - > decoding_needed ) {
avcodec_close ( ist - > st - > codec ) ;
}
}
/* finished ! */
ret = 0 ;
fail :
2011-08-17 11:27:17 +03:00
av_freep ( & no_packet ) ;
2011-07-27 21:56:59 +03:00
2011-08-17 11:21:37 +03:00
if ( output_streams ) {
for ( i = 0 ; i < nb_output_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ost = output_streams [ i ] ;
2011-07-27 21:56:59 +03:00
if ( ost ) {
2011-10-23 12:22:33 +03:00
if ( ost - > stream_copy )
2011-07-27 21:56:59 +03:00
av_freep ( & ost - > st - > codec - > extradata ) ;
if ( ost - > logfile ) {
fclose ( ost - > logfile ) ;
ost - > logfile = NULL ;
}
av_fifo_free ( ost - > fifo ) ; /* works even if fifo is not
initialized but set to zero */
av_freep ( & ost - > st - > codec - > subtitle_header ) ;
av_free ( ost - > forced_kf_pts ) ;
2012-04-05 21:06:28 +03:00
if ( ost - > avr )
avresample_free ( & ost - > avr ) ;
2011-07-27 21:56:59 +03:00
av_dict_free ( & ost - > opts ) ;
}
}
}
return ret ;
}
2011-09-01 09:20:50 +03:00
static double parse_frame_aspect_ratio ( const char * arg )
2011-07-27 21:56:59 +03:00
{
int x = 0 , y = 0 ;
double ar = 0 ;
const char * p ;
char * end ;
p = strchr ( arg , ' : ' ) ;
if ( p ) {
x = strtol ( arg , & end , 10 ) ;
if ( end = = p )
2011-12-29 23:28:29 +03:00
y = strtol ( end + 1 , & end , 10 ) ;
2011-07-27 21:56:59 +03:00
if ( x > 0 & & y > 0 )
ar = ( double ) x / ( double ) y ;
} else
ar = strtod ( arg , NULL ) ;
if ( ! ar ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Incorrect aspect ratio specification. \n " ) ;
2011-09-01 09:20:50 +03:00
exit_program ( 1 ) ;
2011-07-27 21:56:59 +03:00
}
2011-09-01 09:20:50 +03:00
return ar ;
2011-07-27 21:56:59 +03:00
}
2011-08-29 10:16:42 +03:00
static int opt_audio_codec ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
2011-08-29 10:16:42 +03:00
return parse_option ( o , " codec:a " , arg , options ) ;
2011-07-27 21:56:59 +03:00
}
2011-08-29 10:16:42 +03:00
static int opt_video_codec ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
2011-08-29 10:16:42 +03:00
return parse_option ( o , " codec:v " , arg , options ) ;
2011-07-27 21:56:59 +03:00
}
2011-08-29 10:16:42 +03:00
static int opt_subtitle_codec ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
2011-08-29 10:16:42 +03:00
return parse_option ( o , " codec:s " , arg , options ) ;
2011-07-27 21:56:59 +03:00
}
2011-08-29 10:16:42 +03:00
static int opt_data_codec ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
2011-08-29 10:16:42 +03:00
return parse_option ( o , " codec:d " , arg , options ) ;
2011-07-27 21:56:59 +03:00
}
2011-08-28 18:21:56 +03:00
static int opt_map ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
2011-07-29 16:21:51 +03:00
StreamMap * m = NULL ;
int i , negative = 0 , file_idx ;
int sync_file_idx = - 1 , sync_stream_idx ;
char * p , * sync ;
char * map ;
2011-07-27 21:56:59 +03:00
2011-07-29 16:21:51 +03:00
if ( * arg = = ' - ' ) {
negative = 1 ;
arg + + ;
}
map = av_strdup ( arg ) ;
2011-07-27 21:56:59 +03:00
2011-07-29 16:21:51 +03:00
/* parse sync stream first, just pick first matching stream */
if ( sync = strchr ( map , ' , ' ) ) {
* sync = 0 ;
sync_file_idx = strtol ( sync + 1 , & sync , 0 ) ;
if ( sync_file_idx > = nb_input_files | | sync_file_idx < 0 ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid sync file index: %d. \n " , sync_file_idx ) ;
2011-07-29 16:21:51 +03:00
exit_program ( 1 ) ;
}
if ( * sync )
sync + + ;
2012-04-15 16:28:30 +03:00
for ( i = 0 ; i < input_files [ sync_file_idx ] - > nb_streams ; i + + )
if ( check_stream_specifier ( input_files [ sync_file_idx ] - > ctx ,
input_files [ sync_file_idx ] - > ctx - > streams [ i ] , sync ) = = 1 ) {
2011-07-29 16:21:51 +03:00
sync_stream_idx = i ;
break ;
}
2012-04-15 16:28:30 +03:00
if ( i = = input_files [ sync_file_idx ] - > nb_streams ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Sync stream specification in map %s does not "
2011-07-29 16:21:51 +03:00
" match any streams. \n " , arg ) ;
exit_program ( 1 ) ;
}
}
2011-07-27 21:56:59 +03:00
2011-07-29 16:21:51 +03:00
2012-04-01 16:08:33 +03:00
if ( map [ 0 ] = = ' [ ' ) {
/* this mapping refers to lavfi output */
const char * c = map + 1 ;
o - > stream_maps = grow_array ( o - > stream_maps , sizeof ( * o - > stream_maps ) ,
& o - > nb_stream_maps , o - > nb_stream_maps + 1 ) ;
m = & o - > stream_maps [ o - > nb_stream_maps - 1 ] ;
m - > linklabel = av_get_token ( & c , " ] " ) ;
if ( ! m - > linklabel ) {
av_log ( NULL , AV_LOG_ERROR , " Invalid output link label: %s. \n " , map ) ;
exit_program ( 1 ) ;
}
} else {
2012-03-30 23:07:56 +03:00
file_idx = strtol ( map , & p , 0 ) ;
if ( file_idx > = nb_input_files | | file_idx < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Invalid input file index: %d. \n " , file_idx ) ;
exit_program ( 1 ) ;
2011-07-29 16:21:51 +03:00
}
2012-03-30 23:07:56 +03:00
if ( negative )
/* disable some already defined maps */
for ( i = 0 ; i < o - > nb_stream_maps ; i + + ) {
m = & o - > stream_maps [ i ] ;
if ( file_idx = = m - > file_index & &
check_stream_specifier ( input_files [ m - > file_index ] - > ctx ,
input_files [ m - > file_index ] - > ctx - > streams [ m - > stream_index ] ,
* p = = ' : ' ? p + 1 : p ) > 0 )
m - > disabled = 1 ;
}
else
for ( i = 0 ; i < input_files [ file_idx ] - > nb_streams ; i + + ) {
if ( check_stream_specifier ( input_files [ file_idx ] - > ctx , input_files [ file_idx ] - > ctx - > streams [ i ] ,
* p = = ' : ' ? p + 1 : p ) < = 0 )
continue ;
o - > stream_maps = grow_array ( o - > stream_maps , sizeof ( * o - > stream_maps ) ,
& o - > nb_stream_maps , o - > nb_stream_maps + 1 ) ;
m = & o - > stream_maps [ o - > nb_stream_maps - 1 ] ;
m - > file_index = file_idx ;
m - > stream_index = i ;
if ( sync_file_idx > = 0 ) {
m - > sync_file_index = sync_file_idx ;
m - > sync_stream_index = sync_stream_idx ;
} else {
m - > sync_file_index = file_idx ;
m - > sync_stream_index = i ;
}
2011-07-29 16:21:51 +03:00
}
2012-04-01 16:08:33 +03:00
}
2011-07-29 16:21:51 +03:00
if ( ! m ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Stream map '%s' matches no streams. \n " , arg ) ;
2011-07-29 16:21:51 +03:00
exit_program ( 1 ) ;
}
av_freep ( & map ) ;
2011-07-27 21:56:59 +03:00
return 0 ;
}
2011-10-30 09:10:08 +03:00
static int opt_attach ( OptionsContext * o , const char * opt , const char * arg )
{
o - > attachments = grow_array ( o - > attachments , sizeof ( * o - > attachments ) ,
& o - > nb_attachments , o - > nb_attachments + 1 ) ;
o - > attachments [ o - > nb_attachments - 1 ] = arg ;
return 0 ;
}
2011-12-13 13:23:06 +03:00
/**
* Parse a metadata specifier in arg .
* @ param type metadata type is written here - - g ( lobal ) / s ( tream ) / c ( hapter ) / p ( rogram )
* @ param index for type c / p , chapter / program index is written here
* @ param stream_spec for type s , the stream specifier is written here
*/
static void parse_meta_type ( char * arg , char * type , int * index , const char * * stream_spec )
2011-07-27 21:56:59 +03:00
{
2011-08-29 10:16:42 +03:00
if ( * arg ) {
2011-09-05 12:39:09 +03:00
* type = * arg ;
2011-07-27 21:56:59 +03:00
switch ( * arg ) {
case ' g ' :
break ;
case ' s ' :
2011-12-13 13:23:06 +03:00
if ( * ( + + arg ) & & * arg ! = ' : ' ) {
av_log ( NULL , AV_LOG_FATAL , " Invalid metadata specifier %s. \n " , arg ) ;
exit_program ( 1 ) ;
}
* stream_spec = * arg = = ' : ' ? arg + 1 : " " ;
break ;
2011-07-27 21:56:59 +03:00
case ' c ' :
case ' p ' :
2011-07-29 14:07:27 +03:00
if ( * ( + + arg ) = = ' : ' )
* index = strtol ( + + arg , NULL , 0 ) ;
2011-07-27 21:56:59 +03:00
break ;
default :
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid metadata type %c. \n " , * arg ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
} else
* type = ' g ' ;
}
2011-12-13 13:23:06 +03:00
static int copy_metadata ( char * outspec , char * inspec , AVFormatContext * oc , AVFormatContext * ic , OptionsContext * o )
2011-07-27 21:56:59 +03:00
{
2011-12-13 13:23:06 +03:00
AVDictionary * * meta_in = NULL ;
AVDictionary * * meta_out ;
int i , ret = 0 ;
char type_in , type_out ;
const char * istream_spec = NULL , * ostream_spec = NULL ;
int idx_in = 0 , idx_out = 0 ;
2011-07-27 21:56:59 +03:00
2011-12-13 13:23:06 +03:00
parse_meta_type ( inspec , & type_in , & idx_in , & istream_spec ) ;
parse_meta_type ( outspec , & type_out , & idx_out , & ostream_spec ) ;
2011-07-27 21:56:59 +03:00
2011-12-13 13:23:06 +03:00
if ( type_in = = ' g ' | | type_out = = ' g ' )
2011-08-29 10:16:42 +03:00
o - > metadata_global_manual = 1 ;
2011-12-13 13:23:06 +03:00
if ( type_in = = ' s ' | | type_out = = ' s ' )
2011-08-29 10:16:42 +03:00
o - > metadata_streams_manual = 1 ;
2011-12-13 13:23:06 +03:00
if ( type_in = = ' c ' | | type_out = = ' c ' )
2011-08-29 10:16:42 +03:00
o - > metadata_chapters_manual = 1 ;
2011-07-27 21:56:59 +03:00
2011-12-13 13:23:06 +03:00
# define METADATA_CHECK_INDEX(index, nb_elems, desc)\
if ( ( index ) < 0 | | ( index ) > = ( nb_elems ) ) { \
av_log ( NULL , AV_LOG_FATAL , " Invalid %s index %d while processing metadata maps. \n " , \
( desc ) , ( index ) ) ; \
exit_program ( 1 ) ; \
}
# define SET_DICT(type, meta, context, index)\
switch ( type ) { \
case ' g ' : \
meta = & context - > metadata ; \
break ; \
case ' c ' : \
METADATA_CHECK_INDEX ( index , context - > nb_chapters , " chapter " ) \
meta = & context - > chapters [ index ] - > metadata ; \
break ; \
case ' p ' : \
METADATA_CHECK_INDEX ( index , context - > nb_programs , " program " ) \
meta = & context - > programs [ index ] - > metadata ; \
break ; \
} \
SET_DICT ( type_in , meta_in , ic , idx_in ) ;
SET_DICT ( type_out , meta_out , oc , idx_out ) ;
/* for input streams choose first matching stream */
if ( type_in = = ' s ' ) {
for ( i = 0 ; i < ic - > nb_streams ; i + + ) {
if ( ( ret = check_stream_specifier ( ic , ic - > streams [ i ] , istream_spec ) ) > 0 ) {
meta_in = & ic - > streams [ i ] - > metadata ;
break ;
} else if ( ret < 0 )
exit_program ( 1 ) ;
}
if ( ! meta_in ) {
av_log ( NULL , AV_LOG_FATAL , " Stream specifier %s does not match any streams. \n " , istream_spec ) ;
exit_program ( 1 ) ;
}
}
if ( type_out = = ' s ' ) {
for ( i = 0 ; i < oc - > nb_streams ; i + + ) {
if ( ( ret = check_stream_specifier ( oc , oc - > streams [ i ] , ostream_spec ) ) > 0 ) {
meta_out = & oc - > streams [ i ] - > metadata ;
av_dict_copy ( meta_out , * meta_in , AV_DICT_DONT_OVERWRITE ) ;
} else if ( ret < 0 )
exit_program ( 1 ) ;
}
} else
av_dict_copy ( meta_out , * meta_in , AV_DICT_DONT_OVERWRITE ) ;
2011-07-27 21:56:59 +03:00
return 0 ;
}
2011-10-23 12:10:27 +03:00
static AVCodec * find_codec_or_die ( const char * name , enum AVMediaType type , int encoder )
2011-08-13 09:08:13 +03:00
{
const char * codec_string = encoder ? " encoder " : " decoder " ;
AVCodec * codec ;
codec = encoder ?
avcodec_find_encoder_by_name ( name ) :
avcodec_find_decoder_by_name ( name ) ;
2011-12-29 23:28:29 +03:00
if ( ! codec ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Unknown %s '%s' \n " , codec_string , name ) ;
2011-08-13 09:08:13 +03:00
exit_program ( 1 ) ;
}
2011-12-29 23:28:29 +03:00
if ( codec - > type ! = type ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid %s type '%s' \n " , codec_string , name ) ;
2011-08-13 09:08:13 +03:00
exit_program ( 1 ) ;
}
2011-10-23 12:10:27 +03:00
return codec ;
2011-08-13 09:08:13 +03:00
}
2011-10-23 12:10:27 +03:00
static AVCodec * choose_decoder ( OptionsContext * o , AVFormatContext * s , AVStream * st )
2011-08-13 09:08:13 +03:00
{
char * codec_name = NULL ;
2011-08-29 10:16:42 +03:00
MATCH_PER_STREAM_OPT ( codec_names , str , codec_name , s , st ) ;
2011-10-23 12:10:27 +03:00
if ( codec_name ) {
AVCodec * codec = find_codec_or_die ( codec_name , st - > codec - > codec_type , 0 ) ;
st - > codec - > codec_id = codec - > id ;
return codec ;
} else
return avcodec_find_decoder ( st - > codec - > codec_id ) ;
2011-08-13 09:08:13 +03:00
}
2011-07-27 21:56:59 +03:00
2011-07-05 20:38:50 +03:00
/**
* Add all the streams from the given input file to the global
* list of input streams .
*/
2011-08-29 10:16:42 +03:00
static void add_input_streams ( OptionsContext * o , AVFormatContext * ic )
2011-07-05 20:38:50 +03:00
{
2011-12-24 17:08:07 +03:00
int i ;
2011-07-05 20:38:50 +03:00
for ( i = 0 ; i < ic - > nb_streams ; i + + ) {
AVStream * st = ic - > streams [ i ] ;
AVCodecContext * dec = st - > codec ;
2012-04-15 16:28:30 +03:00
InputStream * ist = av_mallocz ( sizeof ( * ist ) ) ;
if ( ! ist )
exit_program ( 1 ) ;
2011-07-05 20:38:50 +03:00
input_streams = grow_array ( input_streams , sizeof ( * input_streams ) , & nb_input_streams , nb_input_streams + 1 ) ;
2012-04-15 16:28:30 +03:00
input_streams [ nb_input_streams - 1 ] = ist ;
2011-07-05 20:38:50 +03:00
ist - > st = st ;
ist - > file_index = nb_input_files ;
ist - > discard = 1 ;
2012-02-24 16:14:24 +03:00
st - > discard = AVDISCARD_ALL ;
2011-07-05 20:38:50 +03:00
ist - > opts = filter_codec_opts ( codec_opts , ist - > st - > codec - > codec_id , ic , st ) ;
2011-11-06 12:32:44 +03:00
ist - > ts_scale = 1.0 ;
MATCH_PER_STREAM_OPT ( ts_scale , dbl , ist - > ts_scale , ic , st ) ;
2011-07-05 20:38:50 +03:00
2011-10-23 12:10:27 +03:00
ist - > dec = choose_decoder ( o , ic , st ) ;
2011-07-05 20:38:50 +03:00
switch ( dec - > codec_type ) {
case AVMEDIA_TYPE_VIDEO :
2012-03-29 08:40:26 +03:00
ist - > resample_height = dec - > height ;
ist - > resample_width = dec - > width ;
ist - > resample_pix_fmt = dec - > pix_fmt ;
2011-07-05 20:38:50 +03:00
break ;
2012-02-24 16:14:24 +03:00
case AVMEDIA_TYPE_AUDIO :
2011-07-05 20:38:50 +03:00
case AVMEDIA_TYPE_DATA :
case AVMEDIA_TYPE_SUBTITLE :
case AVMEDIA_TYPE_ATTACHMENT :
case AVMEDIA_TYPE_UNKNOWN :
break ;
default :
abort ( ) ;
}
}
}
2011-11-02 23:50:54 +03:00
static void assert_file_overwrite ( const char * filename )
{
if ( ! file_overwrite & &
( strchr ( filename , ' : ' ) = = NULL | | filename [ 1 ] = = ' : ' | |
av_strstart ( filename , " file: " , NULL ) ) ) {
if ( avio_check ( filename , 0 ) = = 0 ) {
if ( ! using_stdin ) {
fprintf ( stderr , " File '%s' already exists. Overwrite ? [y/N] " , filename ) ;
fflush ( stderr ) ;
if ( ! read_yesno ( ) ) {
fprintf ( stderr , " Not overwriting - exiting \n " ) ;
exit_program ( 1 ) ;
}
}
else {
fprintf ( stderr , " File '%s' already exists. Exiting. \n " , filename ) ;
exit_program ( 1 ) ;
}
}
}
}
static void dump_attachment ( AVStream * st , const char * filename )
{
int ret ;
AVIOContext * out = NULL ;
AVDictionaryEntry * e ;
if ( ! st - > codec - > extradata_size ) {
av_log ( NULL , AV_LOG_WARNING , " No extradata to dump in stream #%d:%d. \n " ,
nb_input_files - 1 , st - > index ) ;
return ;
}
if ( ! * filename & & ( e = av_dict_get ( st - > metadata , " filename " , NULL , 0 ) ) )
filename = e - > value ;
if ( ! * filename ) {
av_log ( NULL , AV_LOG_FATAL , " No filename specified and no 'filename' tag "
" in stream #%d:%d. \n " , nb_input_files - 1 , st - > index ) ;
exit_program ( 1 ) ;
}
assert_file_overwrite ( filename ) ;
2011-11-07 00:22:04 +03:00
if ( ( ret = avio_open2 ( & out , filename , AVIO_FLAG_WRITE , & int_cb , NULL ) ) < 0 ) {
2011-11-02 23:50:54 +03:00
av_log ( NULL , AV_LOG_FATAL , " Could not open file %s for writing. \n " ,
filename ) ;
exit_program ( 1 ) ;
}
avio_write ( out , st - > codec - > extradata , st - > codec - > extradata_size ) ;
avio_flush ( out ) ;
avio_close ( out ) ;
}
2011-08-28 18:21:56 +03:00
static int opt_input_file ( OptionsContext * o , const char * opt , const char * filename )
2011-07-27 21:56:59 +03:00
{
AVFormatContext * ic ;
AVInputFormat * file_iformat = NULL ;
2011-07-05 20:38:50 +03:00
int err , i , ret ;
2011-07-27 21:56:59 +03:00
int64_t timestamp ;
uint8_t buf [ 128 ] ;
AVDictionary * * opts ;
int orig_nb_streams ; // number of streams before avformat_find_stream_info
2011-08-28 19:20:17 +03:00
if ( o - > format ) {
if ( ! ( file_iformat = av_find_input_format ( o - > format ) ) ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Unknown input format: '%s' \n " , o - > format ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
}
if ( ! strcmp ( filename , " - " ) )
filename = " pipe: " ;
using_stdin | = ! strncmp ( filename , " pipe: " , 5 ) | |
! strcmp ( filename , " /dev/stdin " ) ;
/* get default parameters from command line */
ic = avformat_alloc_context ( ) ;
if ( ! ic ) {
print_error ( filename , AVERROR ( ENOMEM ) ) ;
exit_program ( 1 ) ;
}
2011-09-01 09:20:50 +03:00
if ( o - > nb_audio_sample_rate ) {
snprintf ( buf , sizeof ( buf ) , " %d " , o - > audio_sample_rate [ o - > nb_audio_sample_rate - 1 ] . u . i ) ;
2011-07-27 21:56:59 +03:00
av_dict_set ( & format_opts , " sample_rate " , buf , 0 ) ;
}
2011-09-01 09:20:50 +03:00
if ( o - > nb_audio_channels ) {
2012-04-16 19:51:24 +03:00
/* because we set audio_channels based on both the "ac" and
* " channel_layout " options , we need to check that the specified
* demuxer actually has the " channels " option before setting it */
if ( file_iformat & & file_iformat - > priv_class & &
av_opt_find ( & file_iformat - > priv_class , " channels " , NULL , 0 ,
AV_OPT_SEARCH_FAKE_OBJ ) ) {
snprintf ( buf , sizeof ( buf ) , " %d " ,
o - > audio_channels [ o - > nb_audio_channels - 1 ] . u . i ) ;
av_dict_set ( & format_opts , " channels " , buf , 0 ) ;
}
2011-07-27 21:56:59 +03:00
}
2011-09-01 09:20:50 +03:00
if ( o - > nb_frame_rates ) {
av_dict_set ( & format_opts , " framerate " , o - > frame_rates [ o - > nb_frame_rates - 1 ] . u . str , 0 ) ;
2011-07-27 21:56:59 +03:00
}
2011-09-01 09:20:50 +03:00
if ( o - > nb_frame_sizes ) {
av_dict_set ( & format_opts , " video_size " , o - > frame_sizes [ o - > nb_frame_sizes - 1 ] . u . str , 0 ) ;
2011-07-27 21:56:59 +03:00
}
2011-09-01 09:20:50 +03:00
if ( o - > nb_frame_pix_fmts )
av_dict_set ( & format_opts , " pixel_format " , o - > frame_pix_fmts [ o - > nb_frame_pix_fmts - 1 ] . u . str , 0 ) ;
2011-07-27 21:56:59 +03:00
ic - > flags | = AVFMT_FLAG_NONBLOCK ;
2011-11-07 00:22:04 +03:00
ic - > interrupt_callback = int_cb ;
2011-07-27 21:56:59 +03:00
/* open the input file with generic libav function */
err = avformat_open_input ( & ic , filename , file_iformat , & format_opts ) ;
if ( err < 0 ) {
print_error ( filename , err ) ;
exit_program ( 1 ) ;
}
assert_avoptions ( format_opts ) ;
2011-08-08 15:29:37 +03:00
/* apply forced codec ids */
for ( i = 0 ; i < ic - > nb_streams ; i + + )
2011-10-23 12:10:27 +03:00
choose_decoder ( o , ic , ic - > streams [ i ] ) ;
2011-08-08 15:29:37 +03:00
2011-07-27 21:56:59 +03:00
/* Set AVCodecContext options for avformat_find_stream_info */
opts = setup_find_stream_info_opts ( ic , codec_opts ) ;
orig_nb_streams = ic - > nb_streams ;
/* If not enough info to get the stream parameters, we decode the
first frames to get it . ( used in mpeg case for example ) */
ret = avformat_find_stream_info ( ic , opts ) ;
2011-09-13 08:31:21 +03:00
if ( ret < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " %s: could not find codec parameters \n " , filename ) ;
2011-12-11 12:38:28 +03:00
avformat_close_input ( & ic ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2011-08-28 19:20:17 +03:00
timestamp = o - > start_time ;
2011-07-27 21:56:59 +03:00
/* add the stream start time */
if ( ic - > start_time ! = AV_NOPTS_VALUE )
timestamp + = ic - > start_time ;
/* if seeking requested, we execute it */
2011-08-28 19:20:17 +03:00
if ( o - > start_time ! = 0 ) {
2011-07-27 21:56:59 +03:00
ret = av_seek_frame ( ic , - 1 , timestamp , AVSEEK_FLAG_BACKWARD ) ;
if ( ret < 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_WARNING , " %s: could not seek to position %0.3f \n " ,
filename , ( double ) timestamp / AV_TIME_BASE ) ;
2011-07-27 21:56:59 +03:00
}
}
/* update the current parameters so that they match the one of the input stream */
2011-08-29 10:16:42 +03:00
add_input_streams ( o , ic ) ;
2011-07-27 21:56:59 +03:00
/* dump the file content */
2011-09-13 08:31:21 +03:00
av_dump_format ( ic , nb_input_files , filename , 0 ) ;
2011-07-27 21:56:59 +03:00
input_files = grow_array ( input_files , sizeof ( * input_files ) , & nb_input_files , nb_input_files + 1 ) ;
2012-04-15 16:28:30 +03:00
if ( ! ( input_files [ nb_input_files - 1 ] = av_mallocz ( sizeof ( * input_files [ 0 ] ) ) ) )
exit_program ( 1 ) ;
input_files [ nb_input_files - 1 ] - > ctx = ic ;
input_files [ nb_input_files - 1 ] - > ist_index = nb_input_streams - ic - > nb_streams ;
input_files [ nb_input_files - 1 ] - > ts_offset = o - > input_ts_offset - ( copy_ts ? 0 : timestamp ) ;
input_files [ nb_input_files - 1 ] - > nb_streams = ic - > nb_streams ;
input_files [ nb_input_files - 1 ] - > rate_emu = o - > rate_emu ;
2011-07-27 21:56:59 +03:00
2011-11-02 23:50:54 +03:00
for ( i = 0 ; i < o - > nb_dump_attachment ; i + + ) {
int j ;
for ( j = 0 ; j < ic - > nb_streams ; j + + ) {
AVStream * st = ic - > streams [ j ] ;
if ( check_stream_specifier ( ic , st , o - > dump_attachment [ i ] . specifier ) = = 1 )
dump_attachment ( st , o - > dump_attachment [ i ] . u . str ) ;
}
}
2011-07-27 21:56:59 +03:00
for ( i = 0 ; i < orig_nb_streams ; i + + )
av_dict_free ( & opts [ i ] ) ;
av_freep ( & opts ) ;
2011-08-28 18:21:56 +03:00
reset_options ( o ) ;
2011-07-27 21:56:59 +03:00
return 0 ;
}
2011-08-13 08:55:04 +03:00
static void parse_forced_key_frames ( char * kf , OutputStream * ost ,
AVCodecContext * avctx )
{
char * p ;
int n = 1 , i ;
int64_t t ;
for ( p = kf ; * p ; p + + )
if ( * p = = ' , ' )
n + + ;
ost - > forced_kf_count = n ;
2011-12-29 23:28:29 +03:00
ost - > forced_kf_pts = av_malloc ( sizeof ( * ost - > forced_kf_pts ) * n ) ;
2011-08-13 08:55:04 +03:00
if ( ! ost - > forced_kf_pts ) {
av_log ( NULL , AV_LOG_FATAL , " Could not allocate forced key frames array. \n " ) ;
exit_program ( 1 ) ;
}
for ( i = 0 ; i < n ; i + + ) {
p = i ? strchr ( p , ' , ' ) + 1 : kf ;
t = parse_time_or_die ( " force_key_frames " , p , 1 ) ;
ost - > forced_kf_pts [ i ] = av_rescale_q ( t , AV_TIME_BASE_Q , avctx - > time_base ) ;
}
}
2011-10-10 13:52:13 +03:00
static uint8_t * get_line ( AVIOContext * s )
{
AVIOContext * line ;
uint8_t * buf ;
char c ;
if ( avio_open_dyn_buf ( & line ) < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Could not alloc buffer for reading preset. \n " ) ;
exit_program ( 1 ) ;
}
while ( ( c = avio_r8 ( s ) ) & & c ! = ' \n ' )
avio_w8 ( line , c ) ;
avio_w8 ( line , 0 ) ;
avio_close_dyn_buf ( line , & buf ) ;
return buf ;
}
static int get_preset_file_2 ( const char * preset_name , const char * codec_name , AVIOContext * * s )
{
int i , ret = 1 ;
char filename [ 1000 ] ;
const char * base [ 3 ] = { getenv ( " AVCONV_DATADIR " ) ,
getenv ( " HOME " ) ,
AVCONV_DATADIR ,
} ;
for ( i = 0 ; i < FF_ARRAY_ELEMS ( base ) & & ret ; i + + ) {
if ( ! base [ i ] )
continue ;
if ( codec_name ) {
snprintf ( filename , sizeof ( filename ) , " %s%s/%s-%s.avpreset " , base [ i ] ,
i ! = 1 ? " " : " /.avconv " , codec_name , preset_name ) ;
2011-11-07 00:22:04 +03:00
ret = avio_open2 ( s , filename , AVIO_FLAG_READ , & int_cb , NULL ) ;
2011-10-10 13:52:13 +03:00
}
if ( ret ) {
snprintf ( filename , sizeof ( filename ) , " %s%s/%s.avpreset " , base [ i ] ,
i ! = 1 ? " " : " /.avconv " , preset_name ) ;
2011-11-07 00:22:04 +03:00
ret = avio_open2 ( s , filename , AVIO_FLAG_READ , & int_cb , NULL ) ;
2011-10-10 13:52:13 +03:00
}
}
return ret ;
}
2011-10-23 12:10:27 +03:00
static void choose_encoder ( OptionsContext * o , AVFormatContext * s , OutputStream * ost )
{
char * codec_name = NULL ;
MATCH_PER_STREAM_OPT ( codec_names , str , codec_name , s , ost - > st ) ;
if ( ! codec_name ) {
ost - > st - > codec - > codec_id = av_guess_codec ( s - > oformat , NULL , s - > filename ,
NULL , ost - > st - > codec - > codec_type ) ;
ost - > enc = avcodec_find_encoder ( ost - > st - > codec - > codec_id ) ;
} else if ( ! strcmp ( codec_name , " copy " ) )
2011-10-23 12:22:33 +03:00
ost - > stream_copy = 1 ;
2011-10-23 12:10:27 +03:00
else {
ost - > enc = find_codec_or_die ( codec_name , ost - > st - > codec - > codec_type , 1 ) ;
ost - > st - > codec - > codec_id = ost - > enc - > id ;
}
}
2011-08-29 10:16:42 +03:00
static OutputStream * new_output_stream ( OptionsContext * o , AVFormatContext * oc , enum AVMediaType type )
2011-08-13 09:08:13 +03:00
{
OutputStream * ost ;
2011-06-18 12:43:24 +03:00
AVStream * st = avformat_new_stream ( oc , NULL ) ;
2011-10-10 13:52:13 +03:00
int idx = oc - > nb_streams - 1 , ret = 0 ;
2011-08-31 09:51:15 +03:00
char * bsf = NULL , * next , * codec_tag = NULL ;
2011-08-31 09:45:25 +03:00
AVBitStreamFilterContext * bsfc , * bsfc_prev = NULL ;
2011-09-01 11:04:35 +03:00
double qscale = - 1 ;
2011-10-10 13:52:13 +03:00
char * buf = NULL , * arg = NULL , * preset = NULL ;
AVIOContext * s = NULL ;
2011-08-13 09:08:13 +03:00
if ( ! st ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Could not alloc stream. \n " ) ;
2011-08-13 09:08:13 +03:00
exit_program ( 1 ) ;
}
2011-06-18 12:43:24 +03:00
if ( oc - > nb_streams - 1 < o - > nb_streamid_map )
st - > id = o - > streamid_map [ oc - > nb_streams - 1 ] ;
2011-08-17 11:21:37 +03:00
output_streams = grow_array ( output_streams , sizeof ( * output_streams ) , & nb_output_streams ,
nb_output_streams + 1 ) ;
2012-04-15 16:28:30 +03:00
if ( ! ( ost = av_mallocz ( sizeof ( * ost ) ) ) )
exit_program ( 1 ) ;
output_streams [ nb_output_streams - 1 ] = ost ;
2011-08-13 09:23:44 +03:00
ost - > file_index = nb_output_files ;
2011-12-29 23:28:29 +03:00
ost - > index = idx ;
ost - > st = st ;
2011-08-13 09:08:13 +03:00
st - > codec - > codec_type = type ;
2011-10-23 12:10:27 +03:00
choose_encoder ( o , oc , ost ) ;
2011-08-13 09:08:13 +03:00
if ( ost - > enc ) {
ost - > opts = filter_codec_opts ( codec_opts , ost - > enc - > id , oc , st ) ;
}
avcodec_get_context_defaults3 ( st - > codec , ost - > enc ) ;
st - > codec - > codec_type = type ; // XXX hack, avcodec_get_context_defaults2() sets type to unknown for stream copy
2011-10-10 13:52:13 +03:00
MATCH_PER_STREAM_OPT ( presets , str , preset , oc , st ) ;
if ( preset & & ( ! ( ret = get_preset_file_2 ( preset , ost - > enc - > name , & s ) ) ) ) {
do {
buf = get_line ( s ) ;
if ( ! buf [ 0 ] | | buf [ 0 ] = = ' # ' ) {
av_free ( buf ) ;
continue ;
}
if ( ! ( arg = strchr ( buf , ' = ' ) ) ) {
av_log ( NULL , AV_LOG_FATAL , " Invalid line found in the preset file. \n " ) ;
exit_program ( 1 ) ;
}
* arg + + = 0 ;
av_dict_set ( & ost - > opts , buf , arg , AV_DICT_DONT_OVERWRITE ) ;
av_free ( buf ) ;
} while ( ! s - > eof_reached ) ;
avio_close ( s ) ;
}
if ( ret ) {
av_log ( NULL , AV_LOG_FATAL ,
" Preset %s specified for stream %d:%d, but could not be opened. \n " ,
preset , ost - > file_index , ost - > index ) ;
exit_program ( 1 ) ;
}
2011-11-06 12:32:44 +03:00
ost - > max_frames = INT64_MAX ;
MATCH_PER_STREAM_OPT ( max_frames , i64 , ost - > max_frames , oc , st ) ;
2011-08-29 10:16:42 +03:00
2011-08-31 09:45:25 +03:00
MATCH_PER_STREAM_OPT ( bitstream_filters , str , bsf , oc , st ) ;
while ( bsf ) {
if ( next = strchr ( bsf , ' , ' ) )
* next + + = 0 ;
if ( ! ( bsfc = av_bitstream_filter_init ( bsf ) ) ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Unknown bitstream filter %s \n " , bsf ) ;
2011-08-31 09:45:25 +03:00
exit_program ( 1 ) ;
}
if ( bsfc_prev )
bsfc_prev - > next = bsfc ;
else
ost - > bitstream_filters = bsfc ;
bsfc_prev = bsfc ;
bsf = next ;
}
2011-08-31 09:51:15 +03:00
MATCH_PER_STREAM_OPT ( codec_tags , str , codec_tag , oc , st ) ;
if ( codec_tag ) {
uint32_t tag = strtol ( codec_tag , & next , 0 ) ;
if ( * next )
tag = AV_RL32 ( codec_tag ) ;
st - > codec - > codec_tag = tag ;
}
2011-09-01 11:04:35 +03:00
MATCH_PER_STREAM_OPT ( qscale , dbl , qscale , oc , st ) ;
if ( qscale > = 0 | | same_quant ) {
st - > codec - > flags | = CODEC_FLAG_QSCALE ;
st - > codec - > global_quality = FF_QP2LAMBDA * qscale ;
}
2011-09-27 08:42:29 +03:00
if ( oc - > oformat - > flags & AVFMT_GLOBALHEADER )
st - > codec - > flags | = CODEC_FLAG_GLOBAL_HEADER ;
2011-10-03 22:10:06 +03:00
av_opt_get_int ( sws_opts , " sws_flags " , 0 , & ost - > sws_flags ) ;
2012-03-23 17:46:30 +03:00
ost - > pix_fmts [ 0 ] = ost - > pix_fmts [ 1 ] = PIX_FMT_NONE ;
2011-08-13 09:08:13 +03:00
return ost ;
}
2011-09-02 18:57:56 +03:00
static void parse_matrix_coeffs ( uint16_t * dest , const char * str )
{
int i ;
const char * p = str ;
2011-12-29 23:28:29 +03:00
for ( i = 0 ; ; i + + ) {
2011-09-02 18:57:56 +03:00
dest [ i ] = atoi ( p ) ;
2011-12-29 23:28:29 +03:00
if ( i = = 63 )
2011-09-02 18:57:56 +03:00
break ;
p = strchr ( p , ' , ' ) ;
2011-12-29 23:28:29 +03:00
if ( ! p ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Syntax error in matrix \" %s \" at coeff %d \n " , str , i ) ;
2011-09-02 18:57:56 +03:00
exit_program ( 1 ) ;
}
p + + ;
}
}
2011-08-29 10:16:42 +03:00
static OutputStream * new_video_stream ( OptionsContext * o , AVFormatContext * oc )
2011-07-27 21:56:59 +03:00
{
AVStream * st ;
OutputStream * ost ;
AVCodecContext * video_enc ;
2011-08-29 10:16:42 +03:00
ost = new_output_stream ( o , oc , AVMEDIA_TYPE_VIDEO ) ;
2011-07-27 21:56:59 +03:00
st = ost - > st ;
video_enc = st - > codec ;
2011-10-23 12:22:33 +03:00
if ( ! ost - > stream_copy ) {
2011-09-02 17:06:18 +03:00
const char * p = NULL ;
2011-09-01 09:20:50 +03:00
char * forced_key_frames = NULL , * frame_rate = NULL , * frame_size = NULL ;
2011-09-01 09:20:50 +03:00
char * frame_aspect_ratio = NULL , * frame_pix_fmt = NULL ;
2011-09-02 22:11:00 +03:00
char * intra_matrix = NULL , * inter_matrix = NULL , * filters = NULL ;
2011-11-06 12:32:44 +03:00
int i ;
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( frame_rates , str , frame_rate , oc , st ) ;
if ( frame_rate & & av_parse_video_rate ( & ost - > frame_rate , frame_rate ) < 0 ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid framerate value: %s \n " , frame_rate ) ;
2011-09-01 09:20:50 +03:00
exit_program ( 1 ) ;
}
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( frame_sizes , str , frame_size , oc , st ) ;
if ( frame_size & & av_parse_video_size ( & video_enc - > width , & video_enc - > height , frame_size ) < 0 ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid frame size: %s. \n " , frame_size ) ;
2011-09-01 09:20:50 +03:00
exit_program ( 1 ) ;
}
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( frame_aspect_ratios , str , frame_aspect_ratio , oc , st ) ;
if ( frame_aspect_ratio )
ost - > frame_aspect_ratio = parse_frame_aspect_ratio ( frame_aspect_ratio ) ;
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( frame_pix_fmts , str , frame_pix_fmt , oc , st ) ;
if ( frame_pix_fmt & & ( video_enc - > pix_fmt = av_get_pix_fmt ( frame_pix_fmt ) ) = = PIX_FMT_NONE ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Unknown pixel format requested: %s. \n " , frame_pix_fmt ) ;
2011-09-01 09:20:50 +03:00
exit_program ( 1 ) ;
}
2011-07-27 21:56:59 +03:00
st - > sample_aspect_ratio = video_enc - > sample_aspect_ratio ;
2011-09-02 18:57:56 +03:00
MATCH_PER_STREAM_OPT ( intra_matrices , str , intra_matrix , oc , st ) ;
if ( intra_matrix ) {
if ( ! ( video_enc - > intra_matrix = av_mallocz ( sizeof ( * video_enc - > intra_matrix ) * 64 ) ) ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Could not allocate memory for intra matrix. \n " ) ;
2011-09-02 18:57:56 +03:00
exit_program ( 1 ) ;
}
parse_matrix_coeffs ( video_enc - > intra_matrix , intra_matrix ) ;
}
MATCH_PER_STREAM_OPT ( inter_matrices , str , inter_matrix , oc , st ) ;
if ( inter_matrix ) {
if ( ! ( video_enc - > inter_matrix = av_mallocz ( sizeof ( * video_enc - > inter_matrix ) * 64 ) ) ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Could not allocate memory for inter matrix. \n " ) ;
2011-09-02 18:57:56 +03:00
exit_program ( 1 ) ;
}
parse_matrix_coeffs ( video_enc - > inter_matrix , inter_matrix ) ;
}
2011-07-27 21:56:59 +03:00
2011-09-02 17:06:18 +03:00
MATCH_PER_STREAM_OPT ( rc_overrides , str , p , oc , st ) ;
2011-12-29 23:28:29 +03:00
for ( i = 0 ; p ; i + + ) {
2011-07-27 21:56:59 +03:00
int start , end , q ;
2011-12-29 23:28:29 +03:00
int e = sscanf ( p , " %d,%d,%d " , & start , & end , & q ) ;
if ( e ! = 3 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " error parsing rc_override \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2011-12-29 23:28:29 +03:00
video_enc - > rc_override =
2011-07-27 21:56:59 +03:00
av_realloc ( video_enc - > rc_override ,
2011-12-29 23:28:29 +03:00
sizeof ( RcOverride ) * ( i + 1 ) ) ;
video_enc - > rc_override [ i ] . start_frame = start ;
video_enc - > rc_override [ i ] . end_frame = end ;
if ( q > 0 ) {
video_enc - > rc_override [ i ] . qscale = q ;
video_enc - > rc_override [ i ] . quality_factor = 1.0 ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
else {
video_enc - > rc_override [ i ] . qscale = 0 ;
video_enc - > rc_override [ i ] . quality_factor = - q / 100.0 ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
p = strchr ( p , ' / ' ) ;
if ( p ) p + + ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
video_enc - > rc_override_count = i ;
2011-07-27 21:56:59 +03:00
if ( ! video_enc - > rc_initial_buffer_occupancy )
2011-12-29 23:28:29 +03:00
video_enc - > rc_initial_buffer_occupancy = video_enc - > rc_buffer_size * 3 / 4 ;
video_enc - > intra_dc_precision = intra_dc_precision - 8 ;
2011-07-27 21:56:59 +03:00
/* two pass mode */
if ( do_pass ) {
if ( do_pass = = 1 ) {
video_enc - > flags | = CODEC_FLAG_PASS1 ;
} else {
video_enc - > flags | = CODEC_FLAG_PASS2 ;
}
}
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( forced_key_frames , str , forced_key_frames , oc , st ) ;
2011-07-27 21:56:59 +03:00
if ( forced_key_frames )
parse_forced_key_frames ( forced_key_frames , ost , video_enc ) ;
2011-09-01 09:20:50 +03:00
2011-11-06 12:32:44 +03:00
MATCH_PER_STREAM_OPT ( force_fps , i , ost - > force_fps , oc , st ) ;
2011-09-02 19:03:31 +03:00
2011-11-06 12:32:44 +03:00
ost - > top_field_first = - 1 ;
MATCH_PER_STREAM_OPT ( top_field_first , i , ost - > top_field_first , oc , st ) ;
2011-09-02 22:11:00 +03:00
MATCH_PER_STREAM_OPT ( filters , str , filters , oc , st ) ;
if ( filters )
ost - > avfilter = av_strdup ( filters ) ;
2012-01-10 11:13:32 +03:00
} else {
MATCH_PER_STREAM_OPT ( copy_initial_nonkeyframes , i , ost - > copy_initial_nonkeyframes , oc , st ) ;
2011-07-27 21:56:59 +03:00
}
2011-07-29 09:36:13 +03:00
return ost ;
2011-07-27 21:56:59 +03:00
}
2011-08-29 10:16:42 +03:00
static OutputStream * new_audio_stream ( OptionsContext * o , AVFormatContext * oc )
2011-07-27 21:56:59 +03:00
{
AVStream * st ;
OutputStream * ost ;
AVCodecContext * audio_enc ;
2011-08-29 10:16:42 +03:00
ost = new_output_stream ( o , oc , AVMEDIA_TYPE_AUDIO ) ;
2011-07-27 21:56:59 +03:00
st = ost - > st ;
audio_enc = st - > codec ;
audio_enc - > codec_type = AVMEDIA_TYPE_AUDIO ;
2011-10-23 12:22:33 +03:00
if ( ! ost - > stream_copy ) {
2011-08-31 09:51:15 +03:00
char * sample_fmt = NULL ;
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( audio_channels , i , audio_enc - > channels , oc , st ) ;
2011-08-31 09:51:15 +03:00
MATCH_PER_STREAM_OPT ( sample_fmts , str , sample_fmt , oc , st ) ;
if ( sample_fmt & &
( audio_enc - > sample_fmt = av_get_sample_fmt ( sample_fmt ) ) = = AV_SAMPLE_FMT_NONE ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid sample format '%s' \n " , sample_fmt ) ;
2011-08-31 09:51:15 +03:00
exit_program ( 1 ) ;
}
2011-09-01 09:20:50 +03:00
MATCH_PER_STREAM_OPT ( audio_sample_rate , i , audio_enc - > sample_rate , oc , st ) ;
2011-07-27 21:56:59 +03:00
}
2011-07-29 09:36:13 +03:00
return ost ;
2011-07-27 21:56:59 +03:00
}
2011-08-29 10:16:42 +03:00
static OutputStream * new_data_stream ( OptionsContext * o , AVFormatContext * oc )
2011-07-27 21:56:59 +03:00
{
OutputStream * ost ;
2011-08-29 10:16:42 +03:00
ost = new_output_stream ( o , oc , AVMEDIA_TYPE_DATA ) ;
2011-10-23 12:22:33 +03:00
if ( ! ost - > stream_copy ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Data stream encoding not supported yet (only streamcopy) \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2011-07-29 09:36:13 +03:00
return ost ;
2011-07-27 21:56:59 +03:00
}
2011-09-27 11:37:22 +03:00
static OutputStream * new_attachment_stream ( OptionsContext * o , AVFormatContext * oc )
{
OutputStream * ost = new_output_stream ( o , oc , AVMEDIA_TYPE_ATTACHMENT ) ;
2011-10-23 12:22:33 +03:00
ost - > stream_copy = 1 ;
2011-09-27 11:37:22 +03:00
return ost ;
}
2011-08-29 10:16:42 +03:00
static OutputStream * new_subtitle_stream ( OptionsContext * o , AVFormatContext * oc )
2011-07-27 21:56:59 +03:00
{
AVStream * st ;
OutputStream * ost ;
AVCodecContext * subtitle_enc ;
2011-08-29 10:16:42 +03:00
ost = new_output_stream ( o , oc , AVMEDIA_TYPE_SUBTITLE ) ;
2011-07-27 21:56:59 +03:00
st = ost - > st ;
subtitle_enc = st - > codec ;
subtitle_enc - > codec_type = AVMEDIA_TYPE_SUBTITLE ;
2011-07-29 09:36:13 +03:00
return ost ;
2011-07-27 21:56:59 +03:00
}
/* arg format is "output-stream-index:streamid-value". */
2011-09-02 23:07:30 +03:00
static int opt_streamid ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
int idx ;
char * p ;
char idx_str [ 16 ] ;
av_strlcpy ( idx_str , arg , sizeof ( idx_str ) ) ;
p = strchr ( idx_str , ' : ' ) ;
if ( ! p ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL ,
" Invalid value '%s' for option '%s', required syntax is 'index:value' \n " ,
arg , opt ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
* p + + = ' \0 ' ;
idx = parse_number_or_die ( opt , idx_str , OPT_INT , 0 , INT_MAX ) ;
2011-09-02 23:07:30 +03:00
o - > streamid_map = grow_array ( o - > streamid_map , sizeof ( * o - > streamid_map ) , & o - > nb_streamid_map , idx + 1 ) ;
o - > streamid_map [ idx ] = parse_number_or_die ( opt , p , OPT_INT , 0 , INT_MAX ) ;
2011-07-27 21:56:59 +03:00
return 0 ;
}
2011-08-29 10:16:42 +03:00
static int copy_chapters ( InputFile * ifile , OutputFile * ofile , int copy_metadata )
2011-08-13 08:56:38 +03:00
{
2011-08-28 19:20:17 +03:00
AVFormatContext * is = ifile - > ctx ;
AVFormatContext * os = ofile - > ctx ;
2011-08-13 08:56:38 +03:00
int i ;
for ( i = 0 ; i < is - > nb_chapters ; i + + ) {
AVChapter * in_ch = is - > chapters [ i ] , * out_ch ;
2011-08-28 19:20:17 +03:00
int64_t ts_off = av_rescale_q ( ofile - > start_time - ifile - > ts_offset ,
2011-12-29 23:28:29 +03:00
AV_TIME_BASE_Q , in_ch - > time_base ) ;
2011-08-28 19:20:17 +03:00
int64_t rt = ( ofile - > recording_time = = INT64_MAX ) ? INT64_MAX :
av_rescale_q ( ofile - > recording_time , AV_TIME_BASE_Q , in_ch - > time_base ) ;
2011-08-13 08:56:38 +03:00
if ( in_ch - > end < ts_off )
continue ;
if ( rt ! = INT64_MAX & & in_ch - > start > rt + ts_off )
break ;
out_ch = av_mallocz ( sizeof ( AVChapter ) ) ;
if ( ! out_ch )
return AVERROR ( ENOMEM ) ;
out_ch - > id = in_ch - > id ;
out_ch - > time_base = in_ch - > time_base ;
out_ch - > start = FFMAX ( 0 , in_ch - > start - ts_off ) ;
out_ch - > end = FFMIN ( rt , in_ch - > end - ts_off ) ;
2011-08-29 10:16:42 +03:00
if ( copy_metadata )
2011-08-13 08:56:38 +03:00
av_dict_copy ( & out_ch - > metadata , in_ch - > metadata , 0 ) ;
os - > nb_chapters + + ;
2011-12-29 23:28:29 +03:00
os - > chapters = av_realloc ( os - > chapters , sizeof ( AVChapter ) * os - > nb_chapters ) ;
2011-08-13 08:56:38 +03:00
if ( ! os - > chapters )
return AVERROR ( ENOMEM ) ;
os - > chapters [ os - > nb_chapters - 1 ] = out_ch ;
}
return 0 ;
}
2012-04-01 16:08:33 +03:00
static void init_output_filter ( OutputFilter * ofilter , OptionsContext * o ,
AVFormatContext * oc )
{
OutputStream * ost ;
if ( ofilter - > out_tmp - > filter_ctx - > output_pads [ ofilter - > out_tmp - > pad_idx ] . type ! = AVMEDIA_TYPE_VIDEO ) {
av_log ( NULL , AV_LOG_FATAL , " Only video filters are supported currently. \n " ) ;
exit_program ( 1 ) ;
}
ost = new_video_stream ( o , oc ) ;
ost - > source_index = - 1 ;
ost - > filter = ofilter ;
ofilter - > ost = ost ;
2012-04-21 08:03:06 +03:00
if ( ost - > stream_copy ) {
av_log ( NULL , AV_LOG_ERROR , " Streamcopy requested for output stream %d:%d, "
" which is fed from a complex filtergraph. Filtering and streamcopy "
" cannot be used together. \n " , ost - > file_index , ost - > index ) ;
exit_program ( 1 ) ;
}
2012-04-01 16:08:33 +03:00
if ( configure_output_filter ( ofilter - > graph , ofilter , ofilter - > out_tmp ) < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Error configuring filter. \n " ) ;
exit_program ( 1 ) ;
}
avfilter_inout_free ( & ofilter - > out_tmp ) ;
}
2011-08-28 15:43:54 +03:00
static void opt_output_file ( void * optctx , const char * filename )
2011-07-27 21:56:59 +03:00
{
2011-08-28 18:21:56 +03:00
OptionsContext * o = optctx ;
2011-07-27 21:56:59 +03:00
AVFormatContext * oc ;
2012-04-01 16:08:33 +03:00
int i , j , err ;
2011-07-27 21:56:59 +03:00
AVOutputFormat * file_oformat ;
2011-07-29 09:36:13 +03:00
OutputStream * ost ;
InputStream * ist ;
2011-07-27 21:56:59 +03:00
2012-04-01 16:08:33 +03:00
if ( configure_complex_filters ( ) < 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Error configuring filters. \n " ) ;
exit_program ( 1 ) ;
}
2011-07-27 21:56:59 +03:00
if ( ! strcmp ( filename , " - " ) )
filename = " pipe: " ;
oc = avformat_alloc_context ( ) ;
if ( ! oc ) {
print_error ( filename , AVERROR ( ENOMEM ) ) ;
exit_program ( 1 ) ;
}
2011-08-28 19:20:17 +03:00
if ( o - > format ) {
file_oformat = av_guess_format ( o - > format , NULL , NULL ) ;
2011-07-27 21:56:59 +03:00
if ( ! file_oformat ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Requested output format '%s' is not a suitable output format \n " , o - > format ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
} else {
file_oformat = av_guess_format ( NULL , filename , NULL ) ;
if ( ! file_oformat ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Unable to find a suitable output format for '%s' \n " ,
filename ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
}
oc - > oformat = file_oformat ;
2011-11-07 00:22:04 +03:00
oc - > interrupt_callback = int_cb ;
2011-07-27 21:56:59 +03:00
av_strlcpy ( oc - > filename , filename , sizeof ( oc - > filename ) ) ;
2012-04-01 16:08:33 +03:00
/* create streams for all unlabeled output pads */
for ( i = 0 ; i < nb_filtergraphs ; i + + ) {
FilterGraph * fg = filtergraphs [ i ] ;
for ( j = 0 ; j < fg - > nb_outputs ; j + + ) {
OutputFilter * ofilter = fg - > outputs [ j ] ;
if ( ! ofilter - > out_tmp | | ofilter - > out_tmp - > name )
continue ;
switch ( ofilter - > out_tmp - > filter_ctx - > output_pads [ ofilter - > out_tmp - > pad_idx ] . type ) {
case AVMEDIA_TYPE_VIDEO : o - > video_disable = 1 ; break ;
case AVMEDIA_TYPE_AUDIO : o - > audio_disable = 1 ; break ;
case AVMEDIA_TYPE_SUBTITLE : o - > subtitle_disable = 1 ; break ;
}
init_output_filter ( ofilter , o , oc ) ;
}
}
2011-12-10 19:33:52 +03:00
if ( ! o - > nb_stream_maps ) {
2011-07-29 09:36:13 +03:00
/* pick the "best" stream of each type */
# define NEW_STREAM(type, index)\
if ( index > = 0 ) { \
2011-08-29 10:16:42 +03:00
ost = new_ # # type # # _stream ( o , oc ) ; \
2011-07-29 09:36:13 +03:00
ost - > source_index = index ; \
2012-04-15 16:28:30 +03:00
ost - > sync_ist = input_streams [ index ] ; \
input_streams [ index ] - > discard = 0 ; \
input_streams [ index ] - > st - > discard = AVDISCARD_NONE ; \
2011-07-29 09:36:13 +03:00
}
/* video: highest resolution */
2011-08-31 09:51:15 +03:00
if ( ! o - > video_disable & & oc - > oformat - > video_codec ! = CODEC_ID_NONE ) {
2011-07-29 09:36:13 +03:00
int area = 0 , idx = - 1 ;
for ( i = 0 ; i < nb_input_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ist = input_streams [ i ] ;
2011-07-29 09:36:13 +03:00
if ( ist - > st - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO & &
ist - > st - > codec - > width * ist - > st - > codec - > height > area ) {
area = ist - > st - > codec - > width * ist - > st - > codec - > height ;
idx = i ;
}
}
NEW_STREAM ( video , idx ) ;
}
/* audio: most channels */
2011-08-31 09:51:15 +03:00
if ( ! o - > audio_disable & & oc - > oformat - > audio_codec ! = CODEC_ID_NONE ) {
2011-07-29 09:36:13 +03:00
int channels = 0 , idx = - 1 ;
for ( i = 0 ; i < nb_input_streams ; i + + ) {
2012-04-15 16:28:30 +03:00
ist = input_streams [ i ] ;
2011-07-29 09:36:13 +03:00
if ( ist - > st - > codec - > codec_type = = AVMEDIA_TYPE_AUDIO & &
ist - > st - > codec - > channels > channels ) {
channels = ist - > st - > codec - > channels ;
idx = i ;
}
}
NEW_STREAM ( audio , idx ) ;
}
/* subtitles: pick first */
2011-08-31 09:51:15 +03:00
if ( ! o - > subtitle_disable & & oc - > oformat - > subtitle_codec ! = CODEC_ID_NONE ) {
2011-07-29 09:36:13 +03:00
for ( i = 0 ; i < nb_input_streams ; i + + )
2012-04-15 16:28:30 +03:00
if ( input_streams [ i ] - > st - > codec - > codec_type = = AVMEDIA_TYPE_SUBTITLE ) {
2011-07-29 09:36:13 +03:00
NEW_STREAM ( subtitle , i ) ;
break ;
}
}
/* do something with data? */
2011-07-27 21:56:59 +03:00
} else {
2011-08-28 18:21:56 +03:00
for ( i = 0 ; i < o - > nb_stream_maps ; i + + ) {
StreamMap * map = & o - > stream_maps [ i ] ;
2011-07-29 09:36:13 +03:00
2011-07-29 16:21:51 +03:00
if ( map - > disabled )
continue ;
2011-07-29 09:36:13 +03:00
2012-04-01 16:08:33 +03:00
if ( map - > linklabel ) {
FilterGraph * fg ;
OutputFilter * ofilter = NULL ;
int j , k ;
for ( j = 0 ; j < nb_filtergraphs ; j + + ) {
fg = filtergraphs [ j ] ;
for ( k = 0 ; k < fg - > nb_outputs ; k + + ) {
AVFilterInOut * out = fg - > outputs [ k ] - > out_tmp ;
if ( out & & ! strcmp ( out - > name , map - > linklabel ) ) {
ofilter = fg - > outputs [ k ] ;
goto loop_end ;
}
}
}
loop_end :
if ( ! ofilter ) {
av_log ( NULL , AV_LOG_FATAL , " Output with label '%s' does not exist "
" in any defined filter graph. \n " , map - > linklabel ) ;
exit_program ( 1 ) ;
}
init_output_filter ( ofilter , o , oc ) ;
} else {
2012-03-30 23:07:56 +03:00
ist = input_streams [ input_files [ map - > file_index ] - > ist_index + map - > stream_index ] ;
switch ( ist - > st - > codec - > codec_type ) {
case AVMEDIA_TYPE_VIDEO : ost = new_video_stream ( o , oc ) ; break ;
case AVMEDIA_TYPE_AUDIO : ost = new_audio_stream ( o , oc ) ; break ;
case AVMEDIA_TYPE_SUBTITLE : ost = new_subtitle_stream ( o , oc ) ; break ;
case AVMEDIA_TYPE_DATA : ost = new_data_stream ( o , oc ) ; break ;
case AVMEDIA_TYPE_ATTACHMENT : ost = new_attachment_stream ( o , oc ) ; break ;
default :
av_log ( NULL , AV_LOG_FATAL , " Cannot map stream #%d:%d - unsupported type. \n " ,
map - > file_index , map - > stream_index ) ;
exit_program ( 1 ) ;
}
2011-07-29 09:36:13 +03:00
2012-03-30 23:07:56 +03:00
ost - > source_index = input_files [ map - > file_index ] - > ist_index + map - > stream_index ;
ost - > sync_ist = input_streams [ input_files [ map - > sync_file_index ] - > ist_index +
map - > sync_stream_index ] ;
ist - > discard = 0 ;
ist - > st - > discard = AVDISCARD_NONE ;
2012-04-01 16:08:33 +03:00
}
2011-07-29 09:36:13 +03:00
}
2011-07-27 21:56:59 +03:00
}
2011-10-30 09:10:08 +03:00
/* handle attached files */
for ( i = 0 ; i < o - > nb_attachments ; i + + ) {
AVIOContext * pb ;
uint8_t * attachment ;
const char * p ;
int64_t len ;
2011-11-07 00:22:04 +03:00
if ( ( err = avio_open2 ( & pb , o - > attachments [ i ] , AVIO_FLAG_READ , & int_cb , NULL ) ) < 0 ) {
2011-10-30 09:10:08 +03:00
av_log ( NULL , AV_LOG_FATAL , " Could not open attachment file %s. \n " ,
o - > attachments [ i ] ) ;
exit_program ( 1 ) ;
}
if ( ( len = avio_size ( pb ) ) < = 0 ) {
av_log ( NULL , AV_LOG_FATAL , " Could not get size of the attachment %s. \n " ,
o - > attachments [ i ] ) ;
exit_program ( 1 ) ;
}
if ( ! ( attachment = av_malloc ( len ) ) ) {
av_log ( NULL , AV_LOG_FATAL , " Attachment %s too large to fit into memory. \n " ,
o - > attachments [ i ] ) ;
exit_program ( 1 ) ;
}
avio_read ( pb , attachment , len ) ;
ost = new_attachment_stream ( o , oc ) ;
ost - > stream_copy = 0 ;
ost - > source_index = - 1 ;
ost - > attachment_filename = o - > attachments [ i ] ;
ost - > st - > codec - > extradata = attachment ;
ost - > st - > codec - > extradata_size = len ;
p = strrchr ( o - > attachments [ i ] , ' / ' ) ;
av_dict_set ( & ost - > st - > metadata , " filename " , ( p & & * p ) ? p + 1 : o - > attachments [ i ] , AV_DICT_DONT_OVERWRITE ) ;
avio_close ( pb ) ;
}
2011-08-17 10:56:08 +03:00
output_files = grow_array ( output_files , sizeof ( * output_files ) , & nb_output_files , nb_output_files + 1 ) ;
2012-04-15 16:28:30 +03:00
if ( ! ( output_files [ nb_output_files - 1 ] = av_mallocz ( sizeof ( * output_files [ 0 ] ) ) ) )
exit_program ( 1 ) ;
output_files [ nb_output_files - 1 ] - > ctx = oc ;
output_files [ nb_output_files - 1 ] - > ost_index = nb_output_streams - oc - > nb_streams ;
output_files [ nb_output_files - 1 ] - > recording_time = o - > recording_time ;
2012-02-05 17:49:14 +03:00
if ( o - > recording_time ! = INT64_MAX )
oc - > duration = o - > recording_time ;
2012-04-15 16:28:30 +03:00
output_files [ nb_output_files - 1 ] - > start_time = o - > start_time ;
output_files [ nb_output_files - 1 ] - > limit_filesize = o - > limit_filesize ;
av_dict_copy ( & output_files [ nb_output_files - 1 ] - > opts , format_opts , 0 ) ;
2011-07-27 21:56:59 +03:00
/* check filename in case of an image number is expected */
if ( oc - > oformat - > flags & AVFMT_NEEDNUMBER ) {
if ( ! av_filename_number_test ( oc - > filename ) ) {
print_error ( oc - > filename , AVERROR ( EINVAL ) ) ;
exit_program ( 1 ) ;
}
}
if ( ! ( oc - > oformat - > flags & AVFMT_NOFILE ) ) {
2011-10-05 15:12:42 +03:00
/* test if it already exists to avoid losing precious files */
2011-11-02 23:50:54 +03:00
assert_file_overwrite ( filename ) ;
2011-07-27 21:56:59 +03:00
/* open the file */
2011-11-07 00:22:04 +03:00
if ( ( err = avio_open2 ( & oc - > pb , filename , AVIO_FLAG_WRITE ,
& oc - > interrupt_callback ,
2012-04-15 16:28:30 +03:00
& output_files [ nb_output_files - 1 ] - > opts ) ) < 0 ) {
2011-07-27 21:56:59 +03:00
print_error ( filename , err ) ;
exit_program ( 1 ) ;
}
}
2011-10-09 16:04:38 +03:00
if ( o - > mux_preload ) {
uint8_t buf [ 64 ] ;
snprintf ( buf , sizeof ( buf ) , " %d " , ( int ) ( o - > mux_preload * AV_TIME_BASE ) ) ;
2012-04-15 16:28:30 +03:00
av_dict_set ( & output_files [ nb_output_files - 1 ] - > opts , " preload " , buf , 0 ) ;
2011-10-09 16:04:38 +03:00
}
2011-08-31 09:51:15 +03:00
oc - > max_delay = ( int ) ( o - > mux_max_delay * AV_TIME_BASE ) ;
2011-07-27 21:56:59 +03:00
oc - > flags | = AVFMT_FLAG_NONBLOCK ;
2011-12-13 13:23:06 +03:00
/* copy metadata */
for ( i = 0 ; i < o - > nb_metadata_map ; i + + ) {
char * p ;
int in_file_index = strtol ( o - > metadata_map [ i ] . u . str , & p , 0 ) ;
if ( in_file_index < 0 )
continue ;
if ( in_file_index > = nb_input_files ) {
av_log ( NULL , AV_LOG_FATAL , " Invalid input file index %d while processing metadata maps \n " , in_file_index ) ;
exit_program ( 1 ) ;
}
2012-04-15 16:28:30 +03:00
copy_metadata ( o - > metadata_map [ i ] . specifier , * p ? p + 1 : p , oc , input_files [ in_file_index ] - > ctx , o ) ;
2011-12-13 13:23:06 +03:00
}
2011-07-27 22:39:12 +03:00
/* copy chapters */
2011-08-29 10:16:42 +03:00
if ( o - > chapters_input_file > = nb_input_files ) {
if ( o - > chapters_input_file = = INT_MAX ) {
2011-07-27 22:39:12 +03:00
/* copy chapters from the first input file that has them*/
2011-08-29 10:16:42 +03:00
o - > chapters_input_file = - 1 ;
2011-07-27 22:39:12 +03:00
for ( i = 0 ; i < nb_input_files ; i + + )
2012-04-15 16:28:30 +03:00
if ( input_files [ i ] - > ctx - > nb_chapters ) {
2011-08-29 10:16:42 +03:00
o - > chapters_input_file = i ;
2011-07-27 22:39:12 +03:00
break ;
}
} else {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " Invalid input file index %d in chapter mapping. \n " ,
2011-08-29 10:16:42 +03:00
o - > chapters_input_file ) ;
2011-07-27 22:39:12 +03:00
exit_program ( 1 ) ;
}
}
2011-08-29 10:16:42 +03:00
if ( o - > chapters_input_file > = 0 )
2012-04-15 16:28:30 +03:00
copy_chapters ( input_files [ o - > chapters_input_file ] , output_files [ nb_output_files - 1 ] ,
2011-09-05 15:11:13 +03:00
! o - > metadata_chapters_manual ) ;
2011-07-27 22:39:12 +03:00
2011-07-29 14:07:27 +03:00
/* copy global metadata by default */
2011-08-29 10:16:42 +03:00
if ( ! o - > metadata_global_manual & & nb_input_files )
2012-04-15 16:28:30 +03:00
av_dict_copy ( & oc - > metadata , input_files [ 0 ] - > ctx - > metadata ,
2011-07-29 14:07:27 +03:00
AV_DICT_DONT_OVERWRITE ) ;
2011-08-29 10:16:42 +03:00
if ( ! o - > metadata_streams_manual )
2012-04-15 16:28:30 +03:00
for ( i = output_files [ nb_output_files - 1 ] - > ost_index ; i < nb_output_streams ; i + + ) {
2011-10-30 09:10:08 +03:00
InputStream * ist ;
2012-04-15 16:28:30 +03:00
if ( output_streams [ i ] - > source_index < 0 ) /* this is true e.g. for attached files */
2011-10-30 09:10:08 +03:00
continue ;
2012-04-15 16:28:30 +03:00
ist = input_streams [ output_streams [ i ] - > source_index ] ;
av_dict_copy ( & output_streams [ i ] - > st - > metadata , ist - > st - > metadata , AV_DICT_DONT_OVERWRITE ) ;
2011-07-29 14:07:27 +03:00
}
2011-08-29 10:16:42 +03:00
/* process manually set metadata */
for ( i = 0 ; i < o - > nb_metadata ; i + + ) {
AVDictionary * * m ;
char type , * val ;
2011-12-13 13:23:06 +03:00
const char * stream_spec ;
int index = 0 , j , ret ;
2011-08-29 10:16:42 +03:00
val = strchr ( o - > metadata [ i ] . u . str , ' = ' ) ;
if ( ! val ) {
2011-09-12 22:51:02 +03:00
av_log ( NULL , AV_LOG_FATAL , " No '=' character in metadata string %s. \n " ,
2011-08-29 10:16:42 +03:00
o - > metadata [ i ] . u . str ) ;
exit_program ( 1 ) ;
}
* val + + = 0 ;
2011-12-13 13:23:06 +03:00
parse_meta_type ( o - > metadata [ i ] . specifier , & type , & index , & stream_spec ) ;
if ( type = = ' s ' ) {
for ( j = 0 ; j < oc - > nb_streams ; j + + ) {
if ( ( ret = check_stream_specifier ( oc , oc - > streams [ j ] , stream_spec ) ) > 0 ) {
av_dict_set ( & oc - > streams [ j ] - > metadata , o - > metadata [ i ] . u . str , * val ? val : NULL , 0 ) ;
} else if ( ret < 0 )
exit_program ( 1 ) ;
2011-08-29 10:16:42 +03:00
}
2011-12-13 13:23:06 +03:00
printf ( " ret %d, stream_spec %s \n " , ret , stream_spec ) ;
}
else {
switch ( type ) {
case ' g ' :
m = & oc - > metadata ;
break ;
case ' c ' :
if ( index < 0 | | index > = oc - > nb_chapters ) {
av_log ( NULL , AV_LOG_FATAL , " Invalid chapter index %d in metadata specifier. \n " , index ) ;
exit_program ( 1 ) ;
}
m = & oc - > chapters [ index ] - > metadata ;
break ;
default :
av_log ( NULL , AV_LOG_FATAL , " Invalid metadata specifier %s. \n " , o - > metadata [ i ] . specifier ) ;
2011-08-29 10:16:42 +03:00
exit_program ( 1 ) ;
}
2011-12-13 13:23:06 +03:00
av_dict_set ( m , o - > metadata [ i ] . u . str , * val ? val : NULL , 0 ) ;
2011-08-29 10:16:42 +03:00
}
}
2011-08-28 18:21:56 +03:00
reset_options ( o ) ;
2011-07-27 21:56:59 +03:00
}
/* same option as mencoder */
static int opt_pass ( const char * opt , const char * arg )
{
do_pass = parse_number_or_die ( opt , arg , OPT_INT , 1 , 2 ) ;
return 0 ;
}
static int64_t getutime ( void )
{
# if HAVE_GETRUSAGE
struct rusage rusage ;
getrusage ( RUSAGE_SELF , & rusage ) ;
return ( rusage . ru_utime . tv_sec * 1000000LL ) + rusage . ru_utime . tv_usec ;
# elif HAVE_GETPROCESSTIMES
HANDLE proc ;
FILETIME c , e , k , u ;
proc = GetCurrentProcess ( ) ;
GetProcessTimes ( proc , & c , & e , & k , & u ) ;
return ( ( int64_t ) u . dwHighDateTime < < 32 | u . dwLowDateTime ) / 10 ;
# else
return av_gettime ( ) ;
# endif
}
static int64_t getmaxrss ( void )
{
# if HAVE_GETRUSAGE && HAVE_STRUCT_RUSAGE_RU_MAXRSS
struct rusage rusage ;
getrusage ( RUSAGE_SELF , & rusage ) ;
return ( int64_t ) rusage . ru_maxrss * 1024 ;
# elif HAVE_GETPROCESSMEMORYINFO
HANDLE proc ;
PROCESS_MEMORY_COUNTERS memcounters ;
proc = GetCurrentProcess ( ) ;
memcounters . cb = sizeof ( memcounters ) ;
GetProcessMemoryInfo ( proc , & memcounters , sizeof ( memcounters ) ) ;
return memcounters . PeakPagefileUsage ;
# else
return 0 ;
# endif
}
2011-09-01 11:04:35 +03:00
static int opt_audio_qscale ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " q:a " , arg , options ) ;
}
2011-07-27 21:56:59 +03:00
static void show_usage ( void )
{
printf ( " Hyper fast Audio and Video encoder \n " ) ;
printf ( " usage: %s [options] [[infile options] -i infile]... {[outfile options] outfile}... \n " , program_name ) ;
printf ( " \n " ) ;
}
static void show_help ( void )
{
2011-10-04 15:50:00 +03:00
int flags = AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_ENCODING_PARAM ;
2011-07-27 21:56:59 +03:00
av_log_set_callback ( log_callback_help ) ;
show_usage ( ) ;
show_help_options ( options , " Main options: \n " ,
OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB , 0 ) ;
show_help_options ( options , " \n Advanced options: \n " ,
OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB ,
OPT_EXPERT ) ;
show_help_options ( options , " \n Video options: \n " ,
OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB ,
OPT_VIDEO ) ;
show_help_options ( options , " \n Advanced Video options: \n " ,
OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB ,
OPT_VIDEO | OPT_EXPERT ) ;
show_help_options ( options , " \n Audio options: \n " ,
OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB ,
OPT_AUDIO ) ;
show_help_options ( options , " \n Advanced Audio options: \n " ,
OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB ,
OPT_AUDIO | OPT_EXPERT ) ;
show_help_options ( options , " \n Subtitle options: \n " ,
OPT_SUBTITLE | OPT_GRAB ,
OPT_SUBTITLE ) ;
show_help_options ( options , " \n Audio/Video grab options: \n " ,
OPT_GRAB ,
OPT_GRAB ) ;
printf ( " \n " ) ;
2011-10-04 15:50:00 +03:00
show_help_children ( avcodec_get_class ( ) , flags ) ;
show_help_children ( avformat_get_class ( ) , flags ) ;
show_help_children ( sws_get_class ( ) , flags ) ;
2011-07-27 21:56:59 +03:00
}
2011-08-28 19:20:17 +03:00
static int opt_target ( OptionsContext * o , const char * opt , const char * arg )
2011-07-27 21:56:59 +03:00
{
enum { PAL , NTSC , FILM , UNKNOWN } norm = UNKNOWN ;
2011-12-29 23:28:29 +03:00
static const char * const frame_rates [ ] = { " 25 " , " 30000/1001 " , " 24000/1001 " } ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
if ( ! strncmp ( arg , " pal- " , 4 ) ) {
2011-07-27 21:56:59 +03:00
norm = PAL ;
arg + = 4 ;
2011-12-29 23:28:29 +03:00
} else if ( ! strncmp ( arg , " ntsc- " , 5 ) ) {
2011-07-27 21:56:59 +03:00
norm = NTSC ;
arg + = 5 ;
2011-12-29 23:28:29 +03:00
} else if ( ! strncmp ( arg , " film- " , 5 ) ) {
2011-07-27 21:56:59 +03:00
norm = FILM ;
arg + = 5 ;
} else {
2011-09-01 09:20:50 +03:00
/* Try to determine PAL/NTSC by peeking in the input files */
2011-12-29 23:28:29 +03:00
if ( nb_input_files ) {
2011-09-01 09:20:50 +03:00
int i , j , fr ;
for ( j = 0 ; j < nb_input_files ; j + + ) {
2012-04-15 16:28:30 +03:00
for ( i = 0 ; i < input_files [ j ] - > nb_streams ; i + + ) {
AVCodecContext * c = input_files [ j ] - > ctx - > streams [ i ] - > codec ;
2011-12-29 23:28:29 +03:00
if ( c - > codec_type ! = AVMEDIA_TYPE_VIDEO )
2011-09-01 09:20:50 +03:00
continue ;
fr = c - > time_base . den * 1000 / c - > time_base . num ;
2011-12-29 23:28:29 +03:00
if ( fr = = 25000 ) {
2011-09-01 09:20:50 +03:00
norm = PAL ;
2011-07-27 21:56:59 +03:00
break ;
2011-12-29 23:28:29 +03:00
} else if ( ( fr = = 29970 ) | | ( fr = = 23976 ) ) {
2011-09-01 09:20:50 +03:00
norm = NTSC ;
break ;
}
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
if ( norm ! = UNKNOWN )
2011-09-01 09:20:50 +03:00
break ;
2011-07-27 21:56:59 +03:00
}
}
2011-09-13 08:31:21 +03:00
if ( norm ! = UNKNOWN )
av_log ( NULL , AV_LOG_INFO , " Assuming %s for target. \n " , norm = = PAL ? " PAL " : " NTSC " ) ;
2011-07-27 21:56:59 +03:00
}
2011-12-29 23:28:29 +03:00
if ( norm = = UNKNOWN ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " Could not determine norm (PAL/NTSC/NTSC-Film) for target. \n " ) ;
av_log ( NULL , AV_LOG_FATAL , " Please prefix target with \" pal- \" , \" ntsc- \" or \" film- \" , \n " ) ;
av_log ( NULL , AV_LOG_FATAL , " or set a framerate with \" -r xxx \" . \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
2011-12-29 23:28:29 +03:00
if ( ! strcmp ( arg , " vcd " ) ) {
2011-08-29 10:16:42 +03:00
opt_video_codec ( o , " c:v " , " mpeg1video " ) ;
opt_audio_codec ( o , " c:a " , " mp2 " ) ;
2011-08-28 19:20:17 +03:00
parse_option ( o , " f " , " vcd " , options ) ;
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
parse_option ( o , " s " , norm = = PAL ? " 352x288 " : " 352x240 " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " r " , frame_rates [ norm ] , options ) ;
2011-07-27 21:56:59 +03:00
opt_default ( " g " , norm = = PAL ? " 15 " : " 18 " ) ;
opt_default ( " b " , " 1150000 " ) ;
opt_default ( " maxrate " , " 1150000 " ) ;
opt_default ( " minrate " , " 1150000 " ) ;
opt_default ( " bufsize " , " 327680 " ) ; // 40*1024*8;
2011-08-09 14:56:36 +03:00
opt_default ( " b:a " , " 224000 " ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " ar " , " 44100 " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " ac " , " 2 " , options ) ;
2011-07-27 21:56:59 +03:00
opt_default ( " packetsize " , " 2324 " ) ;
opt_default ( " muxrate " , " 1411200 " ) ; // 2352 * 75 * 8;
/* We have to offset the PTS, so that it is consistent with the SCR.
SCR starts at 36000 , but the first two packs contain only padding
and the first pack from the other stream , respectively , may also have
been written before .
So the real data starts at SCR 36000 + 3 * 1200. */
2011-12-29 23:28:29 +03:00
o - > mux_preload = ( 36000 + 3 * 1200 ) / 90000.0 ; // 0.44
} else if ( ! strcmp ( arg , " svcd " ) ) {
2011-07-27 21:56:59 +03:00
2011-08-29 10:16:42 +03:00
opt_video_codec ( o , " c:v " , " mpeg2video " ) ;
opt_audio_codec ( o , " c:a " , " mp2 " ) ;
2011-08-28 19:20:17 +03:00
parse_option ( o , " f " , " svcd " , options ) ;
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
parse_option ( o , " s " , norm = = PAL ? " 480x576 " : " 480x480 " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " r " , frame_rates [ norm ] , options ) ;
2011-07-27 21:56:59 +03:00
opt_default ( " g " , norm = = PAL ? " 15 " : " 18 " ) ;
opt_default ( " b " , " 2040000 " ) ;
opt_default ( " maxrate " , " 2516000 " ) ;
2011-12-29 23:28:29 +03:00
opt_default ( " minrate " , " 0 " ) ; // 1145000;
opt_default ( " bufsize " , " 1835008 " ) ; // 224*1024*8;
2011-07-27 21:56:59 +03:00
opt_default ( " flags " , " +scan_offset " ) ;
2011-08-09 14:56:36 +03:00
opt_default ( " b:a " , " 224000 " ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " ar " , " 44100 " , options ) ;
2011-07-27 21:56:59 +03:00
opt_default ( " packetsize " , " 2324 " ) ;
2011-12-29 23:28:29 +03:00
} else if ( ! strcmp ( arg , " dvd " ) ) {
2011-07-27 21:56:59 +03:00
2011-08-29 10:16:42 +03:00
opt_video_codec ( o , " c:v " , " mpeg2video " ) ;
opt_audio_codec ( o , " c:a " , " ac3 " ) ;
2011-08-28 19:20:17 +03:00
parse_option ( o , " f " , " dvd " , options ) ;
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
parse_option ( o , " s " , norm = = PAL ? " 720x576 " : " 720x480 " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " r " , frame_rates [ norm ] , options ) ;
2011-07-27 21:56:59 +03:00
opt_default ( " g " , norm = = PAL ? " 15 " : " 18 " ) ;
opt_default ( " b " , " 6000000 " ) ;
opt_default ( " maxrate " , " 9000000 " ) ;
2011-12-29 23:28:29 +03:00
opt_default ( " minrate " , " 0 " ) ; // 1500000;
opt_default ( " bufsize " , " 1835008 " ) ; // 224*1024*8;
2011-07-27 21:56:59 +03:00
opt_default ( " packetsize " , " 2048 " ) ; // from www.mpucoder.com: DVD sectors contain 2048 bytes of data, this is also the size of one pack.
opt_default ( " muxrate " , " 10080000 " ) ; // from mplex project: data_rate = 1260000. mux_rate = data_rate * 8
2011-08-09 14:56:36 +03:00
opt_default ( " b:a " , " 448000 " ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " ar " , " 48000 " , options ) ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
} else if ( ! strncmp ( arg , " dv " , 2 ) ) {
2011-07-27 21:56:59 +03:00
2011-08-28 19:20:17 +03:00
parse_option ( o , " f " , " dv " , options ) ;
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
parse_option ( o , " s " , norm = = PAL ? " 720x576 " : " 720x480 " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " pix_fmt " , ! strncmp ( arg , " dv50 " , 4 ) ? " yuv422p " :
norm = = PAL ? " yuv420p " : " yuv411p " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " r " , frame_rates [ norm ] , options ) ;
2011-07-27 21:56:59 +03:00
2011-09-01 09:20:50 +03:00
parse_option ( o , " ar " , " 48000 " , options ) ;
2011-09-01 09:20:50 +03:00
parse_option ( o , " ac " , " 2 " , options ) ;
2011-07-27 21:56:59 +03:00
} else {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_ERROR , " Unknown target: %s \n " , arg ) ;
2011-07-27 21:56:59 +03:00
return AVERROR ( EINVAL ) ;
}
return 0 ;
}
static int opt_vstats_file ( const char * opt , const char * arg )
{
av_free ( vstats_filename ) ;
2011-12-29 23:28:29 +03:00
vstats_filename = av_strdup ( arg ) ;
2011-07-27 21:56:59 +03:00
return 0 ;
}
static int opt_vstats ( const char * opt , const char * arg )
{
char filename [ 40 ] ;
time_t today2 = time ( NULL ) ;
struct tm * today = localtime ( & today2 ) ;
snprintf ( filename , sizeof ( filename ) , " vstats_%02d%02d%02d.log " , today - > tm_hour , today - > tm_min ,
today - > tm_sec ) ;
return opt_vstats_file ( opt , filename ) ;
}
2011-08-29 10:16:42 +03:00
static int opt_video_frames ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " frames:v " , arg , options ) ;
}
static int opt_audio_frames ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " frames:a " , arg , options ) ;
}
static int opt_data_frames ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " frames:d " , arg , options ) ;
}
2011-08-31 09:51:15 +03:00
static int opt_video_tag ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " tag:v " , arg , options ) ;
}
static int opt_audio_tag ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " tag:a " , arg , options ) ;
}
static int opt_subtitle_tag ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " tag:s " , arg , options ) ;
}
2011-09-02 22:11:00 +03:00
static int opt_video_filters ( OptionsContext * o , const char * opt , const char * arg )
{
return parse_option ( o , " filter:v " , arg , options ) ;
}
2011-11-26 13:01:33 +03:00
static int opt_vsync ( const char * opt , const char * arg )
{
if ( ! av_strcasecmp ( arg , " cfr " ) ) video_sync_method = VSYNC_CFR ;
else if ( ! av_strcasecmp ( arg , " vfr " ) ) video_sync_method = VSYNC_VFR ;
else if ( ! av_strcasecmp ( arg , " passthrough " ) ) video_sync_method = VSYNC_PASSTHROUGH ;
if ( video_sync_method = = VSYNC_AUTO )
video_sync_method = parse_number_or_die ( " vsync " , arg , OPT_INT , VSYNC_AUTO , VSYNC_VFR ) ;
return 0 ;
}
2012-01-30 18:05:28 +03:00
static int opt_deinterlace ( const char * opt , const char * arg )
{
av_log ( NULL , AV_LOG_WARNING , " -%s is deprecated, use -filter:v yadif instead \n " , opt ) ;
do_deinterlace = 1 ;
return 0 ;
}
2012-03-04 18:46:45 +03:00
static int opt_cpuflags ( const char * opt , const char * arg )
{
2012-04-25 20:06:51 +03:00
int flags = av_parse_cpu_flags ( arg ) ;
if ( flags < 0 )
return flags ;
2012-03-04 18:46:45 +03:00
av_set_cpu_flags_mask ( flags ) ;
return 0 ;
}
static void parse_cpuflags ( int argc , char * * argv , const OptionDef * options )
{
int idx = locate_option ( argc , argv , options , " cpuflags " ) ;
if ( idx & & argv [ idx + 1 ] )
opt_cpuflags ( " cpuflags " , argv [ idx + 1 ] ) ;
}
2012-04-06 00:47:20 +03:00
static int opt_channel_layout ( OptionsContext * o , const char * opt , const char * arg )
{
char layout_str [ 32 ] ;
char * stream_str ;
char * ac_str ;
int ret , channels , ac_str_size ;
uint64_t layout ;
layout = av_get_channel_layout ( arg ) ;
if ( ! layout ) {
av_log ( NULL , AV_LOG_ERROR , " Unknown channel layout: %s \n " , arg ) ;
return AVERROR ( EINVAL ) ;
}
snprintf ( layout_str , sizeof ( layout_str ) , " % " PRIu64 , layout ) ;
ret = opt_default ( opt , layout_str ) ;
if ( ret < 0 )
return ret ;
/* set 'ac' option based on channel layout */
channels = av_get_channel_layout_nb_channels ( layout ) ;
snprintf ( layout_str , sizeof ( layout_str ) , " %d " , channels ) ;
stream_str = strchr ( opt , ' : ' ) ;
ac_str_size = 3 + ( stream_str ? strlen ( stream_str ) : 0 ) ;
ac_str = av_mallocz ( ac_str_size ) ;
if ( ! ac_str )
return AVERROR ( ENOMEM ) ;
av_strlcpy ( ac_str , " ac " , 3 ) ;
if ( stream_str )
av_strlcat ( ac_str , stream_str , ac_str_size ) ;
ret = parse_option ( o , ac_str , layout_str , options ) ;
av_free ( ac_str ) ;
return ret ;
}
2012-04-01 16:08:33 +03:00
static int opt_filter_complex ( const char * opt , const char * arg )
{
filtergraphs = grow_array ( filtergraphs , sizeof ( * filtergraphs ) ,
& nb_filtergraphs , nb_filtergraphs + 1 ) ;
if ( ! ( filtergraphs [ nb_filtergraphs - 1 ] = av_mallocz ( sizeof ( * filtergraphs [ 0 ] ) ) ) )
return AVERROR ( ENOMEM ) ;
filtergraphs [ nb_filtergraphs - 1 ] - > index = nb_filtergraphs - 1 ;
filtergraphs [ nb_filtergraphs - 1 ] - > graph_desc = arg ;
return 0 ;
}
2011-08-28 19:20:17 +03:00
# define OFFSET(x) offsetof(OptionsContext, x)
2011-07-27 21:56:59 +03:00
static const OptionDef options [ ] = {
/* main options */
# include "cmdutils_common_opts.h"
2011-08-28 19:20:17 +03:00
{ " f " , HAS_ARG | OPT_STRING | OPT_OFFSET , { . off = OFFSET ( format ) } , " force format " , " fmt " } ,
2011-08-28 18:21:56 +03:00
{ " i " , HAS_ARG | OPT_FUNC2 , { ( void * ) opt_input_file } , " input file name " , " filename " } ,
2011-07-27 21:56:59 +03:00
{ " y " , OPT_BOOL , { ( void * ) & file_overwrite } , " overwrite output files " } ,
2011-08-29 10:16:42 +03:00
{ " c " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( codec_names ) } , " codec name " , " codec " } ,
{ " codec " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( codec_names ) } , " codec name " , " codec " } ,
2011-10-10 13:52:13 +03:00
{ " pre " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( presets ) } , " preset name " , " preset " } ,
2011-12-12 14:45:32 +03:00
{ " map " , HAS_ARG | OPT_EXPERT | OPT_FUNC2 , { ( void * ) opt_map } , " set input stream mapping " , " [-]input_file_id[:stream_specifier][,sync_file_id[:stream_specifier]] " } ,
2011-12-13 13:23:06 +03:00
{ " map_metadata " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( metadata_map ) } , " set metadata information of outfile from infile " ,
2011-07-27 21:56:59 +03:00
" outfile[,metadata]:infile[,metadata] " } ,
2011-08-29 10:16:42 +03:00
{ " map_chapters " , OPT_INT | HAS_ARG | OPT_EXPERT | OPT_OFFSET , { . off = OFFSET ( chapters_input_file ) } , " set chapters mapping " , " input_file_index " } ,
2011-08-28 19:20:17 +03:00
{ " t " , HAS_ARG | OPT_TIME | OPT_OFFSET , { . off = OFFSET ( recording_time ) } , " record or transcode \" duration \" seconds of audio/video " , " duration " } ,
2011-08-28 19:20:17 +03:00
{ " fs " , HAS_ARG | OPT_INT64 | OPT_OFFSET , { . off = OFFSET ( limit_filesize ) } , " set the limit file size in bytes " , " limit_size " } , //
2011-08-28 19:20:17 +03:00
{ " ss " , HAS_ARG | OPT_TIME | OPT_OFFSET , { . off = OFFSET ( start_time ) } , " set the start time offset " , " time_off " } ,
{ " itsoffset " , HAS_ARG | OPT_TIME | OPT_OFFSET , { . off = OFFSET ( input_ts_offset ) } , " set the input ts offset " , " time_off " } ,
2011-08-29 10:16:42 +03:00
{ " itsscale " , HAS_ARG | OPT_DOUBLE | OPT_SPEC , { . off = OFFSET ( ts_scale ) } , " set the input ts scale " , " scale " } ,
2011-08-29 10:16:42 +03:00
{ " metadata " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( metadata ) } , " add metadata " , " string=string " } ,
2011-08-29 10:16:42 +03:00
{ " dframes " , HAS_ARG | OPT_FUNC2 , { ( void * ) opt_data_frames } , " set the number of data frames to record " , " number " } ,
2011-07-27 21:56:59 +03:00
{ " benchmark " , OPT_BOOL | OPT_EXPERT , { ( void * ) & do_benchmark } ,
" add timings for benchmarking " } ,
{ " timelimit " , HAS_ARG , { ( void * ) opt_timelimit } , " set max runtime in seconds " , " limit " } ,
{ " dump " , OPT_BOOL | OPT_EXPERT , { ( void * ) & do_pkt_dump } ,
" dump each input packet " } ,
{ " hex " , OPT_BOOL | OPT_EXPERT , { ( void * ) & do_hex_dump } ,
" when dumping packets, also dump the payload " } ,
2011-08-29 10:16:42 +03:00
{ " re " , OPT_BOOL | OPT_EXPERT | OPT_OFFSET , { . off = OFFSET ( rate_emu ) } , " read input at native frame rate " , " " } ,
2011-08-28 19:20:17 +03:00
{ " target " , HAS_ARG | OPT_FUNC2 , { ( void * ) opt_target } , " specify target file type ( \" vcd \" , \" svcd \" , \" dvd \" , \" dv \" , \" dv50 \" , \" pal-vcd \" , \" ntsc-svcd \" , ...) " , " type " } ,
2011-11-26 13:01:33 +03:00
{ " vsync " , HAS_ARG | OPT_EXPERT , { ( void * ) opt_vsync } , " video sync method " , " " } ,
2011-07-27 21:56:59 +03:00
{ " async " , HAS_ARG | OPT_INT | OPT_EXPERT , { ( void * ) & audio_sync_method } , " audio sync method " , " " } ,
{ " adrift_threshold " , HAS_ARG | OPT_FLOAT | OPT_EXPERT , { ( void * ) & audio_drift_threshold } , " audio drift threshold " , " threshold " } ,
{ " copyts " , OPT_BOOL | OPT_EXPERT , { ( void * ) & copy_ts } , " copy timestamps " } ,
{ " copytb " , OPT_BOOL | OPT_EXPERT , { ( void * ) & copy_tb } , " copy input stream time base when stream copying " } ,
{ " shortest " , OPT_BOOL | OPT_EXPERT , { ( void * ) & opt_shortest } , " finish encoding within shortest input " } , //
{ " dts_delta_threshold " , HAS_ARG | OPT_FLOAT | OPT_EXPERT , { ( void * ) & dts_delta_threshold } , " timestamp discontinuity delta threshold " , " threshold " } ,
{ " xerror " , OPT_BOOL , { ( void * ) & exit_on_error } , " exit on error " , " error " } ,
2011-11-06 12:26:56 +03:00
{ " copyinkf " , OPT_BOOL | OPT_EXPERT | OPT_SPEC , { . off = OFFSET ( copy_initial_nonkeyframes ) } , " copy initial non-keyframes " } ,
2011-08-29 10:16:42 +03:00
{ " frames " , OPT_INT64 | HAS_ARG | OPT_SPEC , { . off = OFFSET ( max_frames ) } , " set the number of frames to record " , " number " } ,
2011-08-31 09:51:15 +03:00
{ " tag " , OPT_STRING | HAS_ARG | OPT_SPEC , { . off = OFFSET ( codec_tags ) } , " force codec tag/fourcc " , " fourcc/tag " } ,
2011-09-01 11:04:35 +03:00
{ " q " , HAS_ARG | OPT_EXPERT | OPT_DOUBLE | OPT_SPEC , { . off = OFFSET ( qscale ) } , " use fixed quality scale (VBR) " , " q " } ,
{ " qscale " , HAS_ARG | OPT_EXPERT | OPT_DOUBLE | OPT_SPEC , { . off = OFFSET ( qscale ) } , " use fixed quality scale (VBR) " , " q " } ,
2011-09-02 22:11:00 +03:00
{ " filter " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( filters ) } , " set stream filterchain " , " filter_list " } ,
2012-04-01 16:08:33 +03:00
{ " filter_complex " , HAS_ARG | OPT_EXPERT , { ( void * ) opt_filter_complex } , " create a complex filtergraph " , " graph_description " } ,
2011-10-09 16:57:30 +03:00
{ " stats " , OPT_BOOL , { & print_stats } , " print progress report during encoding " , } ,
2011-10-30 09:10:08 +03:00
{ " attach " , HAS_ARG | OPT_FUNC2 , { ( void * ) opt_attach } , " add an attachment to the output file " , " filename " } ,
2011-11-02 23:50:54 +03:00
{ " dump_attachment " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( dump_attachment ) } , " extract an attachment into a file " , " filename " } ,
2012-03-04 18:46:45 +03:00
{ " cpuflags " , HAS_ARG | OPT_EXPERT , { ( void * ) opt_cpuflags } , " set CPU flags mask " , " mask " } ,
2011-07-27 21:56:59 +03:00
/* video options */
2011-08-29 10:16:42 +03:00
{ " vframes " , HAS_ARG | OPT_VIDEO | OPT_FUNC2 , { ( void * ) opt_video_frames } , " set the number of video frames to record " , " number " } ,
2011-09-01 09:20:50 +03:00
{ " r " , HAS_ARG | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( frame_rates ) } , " set frame rate (Hz value, fraction or abbreviation) " , " rate " } ,
2011-09-01 09:20:50 +03:00
{ " s " , HAS_ARG | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( frame_sizes ) } , " set frame size (WxH or abbreviation) " , " size " } ,
2011-09-01 09:20:50 +03:00
{ " aspect " , HAS_ARG | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( frame_aspect_ratios ) } , " set aspect ratio (4:3, 16:9 or 1.3333, 1.7777) " , " aspect " } ,
2011-09-01 09:20:50 +03:00
{ " pix_fmt " , HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( frame_pix_fmts ) } , " set pixel format " , " format " } ,
2011-08-31 09:51:15 +03:00
{ " vn " , OPT_BOOL | OPT_VIDEO | OPT_OFFSET , { . off = OFFSET ( video_disable ) } , " disable video " } ,
2011-07-27 21:56:59 +03:00
{ " vdt " , OPT_INT | HAS_ARG | OPT_EXPERT | OPT_VIDEO , { ( void * ) & video_discard } , " discard threshold " , " n " } ,
2011-09-02 17:06:18 +03:00
{ " rc_override " , HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( rc_overrides ) } , " rate control override for specific intervals " , " override " } ,
2011-08-29 10:16:42 +03:00
{ " vcodec " , HAS_ARG | OPT_VIDEO | OPT_FUNC2 , { ( void * ) opt_video_codec } , " force video codec ('copy' to copy stream) " , " codec " } ,
2011-08-09 12:32:16 +03:00
{ " same_quant " , OPT_BOOL | OPT_VIDEO , { ( void * ) & same_quant } ,
2011-07-27 21:56:59 +03:00
" use same quantizer as source (implies VBR) " } ,
{ " pass " , HAS_ARG | OPT_VIDEO , { ( void * ) opt_pass } , " select the pass number (1 or 2) " , " n " } ,
{ " passlogfile " , HAS_ARG | OPT_STRING | OPT_VIDEO , { ( void * ) & pass_logfilename_prefix } , " select two pass log file name prefix " , " prefix " } ,
2012-01-30 18:05:28 +03:00
{ " deinterlace " , OPT_EXPERT | OPT_VIDEO , { ( void * ) opt_deinterlace } ,
" this option is deprecated, use the yadif filter instead " } ,
2011-07-27 21:56:59 +03:00
{ " vstats " , OPT_EXPERT | OPT_VIDEO , { ( void * ) & opt_vstats } , " dump video coding statistics to file " } ,
{ " vstats_file " , HAS_ARG | OPT_EXPERT | OPT_VIDEO , { ( void * ) opt_vstats_file } , " dump video coding statistics to file " , " file " } ,
2011-09-02 22:11:00 +03:00
{ " vf " , HAS_ARG | OPT_VIDEO | OPT_FUNC2 , { ( void * ) opt_video_filters } , " video filters " , " filter list " } ,
2011-09-02 18:57:56 +03:00
{ " intra_matrix " , HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( intra_matrices ) } , " specify intra matrix coeffs " , " matrix " } ,
{ " inter_matrix " , HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC , { . off = OFFSET ( inter_matrices ) } , " specify inter matrix coeffs " , " matrix " } ,
2011-09-02 19:03:31 +03:00
{ " top " , HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_INT | OPT_SPEC , { . off = OFFSET ( top_field_first ) } , " top=1/bottom=0/auto=-1 field first " , " " } ,
2011-07-27 21:56:59 +03:00
{ " dc " , OPT_INT | HAS_ARG | OPT_EXPERT | OPT_VIDEO , { ( void * ) & intra_dc_precision } , " intra_dc_precision " , " precision " } ,
2011-08-31 09:51:15 +03:00
{ " vtag " , HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_FUNC2 , { ( void * ) opt_video_tag } , " force video tag/fourcc " , " fourcc/tag " } ,
2011-07-27 21:56:59 +03:00
{ " qphist " , OPT_BOOL | OPT_EXPERT | OPT_VIDEO , { ( void * ) & qp_hist } , " show QP histogram " } ,
2011-09-01 09:20:50 +03:00
{ " force_fps " , OPT_BOOL | OPT_EXPERT | OPT_VIDEO | OPT_SPEC , { . off = OFFSET ( force_fps ) } , " force the selected framerate, disable the best supported framerate selection " } ,
2011-09-02 23:07:30 +03:00
{ " streamid " , HAS_ARG | OPT_EXPERT | OPT_FUNC2 , { ( void * ) opt_streamid } , " set the value of an outfile streamid " , " streamIndex:value " } ,
2011-09-01 09:20:50 +03:00
{ " force_key_frames " , OPT_STRING | HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_SPEC , { . off = OFFSET ( forced_key_frames ) } , " force key frames at specified timestamps " , " timestamps " } ,
2011-07-27 21:56:59 +03:00
/* audio options */
2011-08-29 10:16:42 +03:00
{ " aframes " , HAS_ARG | OPT_AUDIO | OPT_FUNC2 , { ( void * ) opt_audio_frames } , " set the number of audio frames to record " , " number " } ,
2011-09-01 11:04:35 +03:00
{ " aq " , HAS_ARG | OPT_AUDIO | OPT_FUNC2 , { ( void * ) opt_audio_qscale } , " set audio quality (codec-specific) " , " quality " , } ,
2011-09-01 09:20:50 +03:00
{ " ar " , HAS_ARG | OPT_AUDIO | OPT_INT | OPT_SPEC , { . off = OFFSET ( audio_sample_rate ) } , " set audio sampling rate (in Hz) " , " rate " } ,
2011-09-01 09:20:50 +03:00
{ " ac " , HAS_ARG | OPT_AUDIO | OPT_INT | OPT_SPEC , { . off = OFFSET ( audio_channels ) } , " set number of audio channels " , " channels " } ,
2011-08-31 09:51:15 +03:00
{ " an " , OPT_BOOL | OPT_AUDIO | OPT_OFFSET , { . off = OFFSET ( audio_disable ) } , " disable audio " } ,
2011-08-29 10:16:42 +03:00
{ " acodec " , HAS_ARG | OPT_AUDIO | OPT_FUNC2 , { ( void * ) opt_audio_codec } , " force audio codec ('copy' to copy stream) " , " codec " } ,
2011-08-31 09:51:15 +03:00
{ " atag " , HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_FUNC2 , { ( void * ) opt_audio_tag } , " force audio tag/fourcc " , " fourcc/tag " } ,
2011-07-27 21:56:59 +03:00
{ " vol " , OPT_INT | HAS_ARG | OPT_AUDIO , { ( void * ) & audio_volume } , " change audio volume (256=normal) " , " volume " } , //
2011-08-31 09:51:15 +03:00
{ " sample_fmt " , HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_SPEC | OPT_STRING , { . off = OFFSET ( sample_fmts ) } , " set sample format " , " format " } ,
2012-04-06 00:47:20 +03:00
{ " channel_layout " , HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_FUNC2 , { ( void * ) opt_channel_layout } , " set channel layout " , " layout " } ,
2011-07-27 21:56:59 +03:00
/* subtitle options */
2011-08-31 09:51:15 +03:00
{ " sn " , OPT_BOOL | OPT_SUBTITLE | OPT_OFFSET , { . off = OFFSET ( subtitle_disable ) } , " disable subtitle " } ,
2011-08-29 10:16:42 +03:00
{ " scodec " , HAS_ARG | OPT_SUBTITLE | OPT_FUNC2 , { ( void * ) opt_subtitle_codec } , " force subtitle codec ('copy' to copy stream) " , " codec " } ,
2011-08-31 09:51:15 +03:00
{ " stag " , HAS_ARG | OPT_EXPERT | OPT_SUBTITLE | OPT_FUNC2 , { ( void * ) opt_subtitle_tag } , " force subtitle tag/fourcc " , " fourcc/tag " } ,
2011-07-27 21:56:59 +03:00
/* grab options */
{ " isync " , OPT_BOOL | OPT_EXPERT | OPT_GRAB , { ( void * ) & input_sync } , " sync read on input " , " " } ,
/* muxer options */
2011-08-31 09:51:15 +03:00
{ " muxdelay " , OPT_FLOAT | HAS_ARG | OPT_EXPERT | OPT_OFFSET , { . off = OFFSET ( mux_max_delay ) } , " set the maximum demux-decode delay " , " seconds " } ,
{ " muxpreload " , OPT_FLOAT | HAS_ARG | OPT_EXPERT | OPT_OFFSET , { . off = OFFSET ( mux_preload ) } , " set the initial demux-decode delay " , " seconds " } ,
2011-07-27 21:56:59 +03:00
2011-08-31 09:45:25 +03:00
{ " bsf " , HAS_ARG | OPT_STRING | OPT_SPEC , { . off = OFFSET ( bitstream_filters ) } , " A comma-separated list of bitstream filters " , " bitstream_filters " } ,
2011-07-27 21:56:59 +03:00
/* data codec support */
2011-08-29 10:16:42 +03:00
{ " dcodec " , HAS_ARG | OPT_DATA | OPT_FUNC2 , { ( void * ) opt_data_codec } , " force data codec ('copy' to copy stream) " , " codec " } ,
2011-07-27 21:56:59 +03:00
{ " default " , HAS_ARG | OPT_AUDIO | OPT_VIDEO | OPT_EXPERT , { ( void * ) opt_default } , " generic catch all option " , " " } ,
{ NULL , } ,
} ;
int main ( int argc , char * * argv )
{
2011-08-28 18:21:56 +03:00
OptionsContext o = { 0 } ;
2011-07-27 21:56:59 +03:00
int64_t ti ;
2011-08-28 18:21:56 +03:00
reset_options ( & o ) ;
2011-07-27 21:56:59 +03:00
av_log_set_flags ( AV_LOG_SKIP_REPEATED ) ;
2011-09-26 09:15:37 +03:00
parse_loglevel ( argc , argv , options ) ;
2011-07-27 21:56:59 +03:00
avcodec_register_all ( ) ;
# if CONFIG_AVDEVICE
avdevice_register_all ( ) ;
# endif
avfilter_register_all ( ) ;
av_register_all ( ) ;
2011-11-06 03:47:48 +03:00
avformat_network_init ( ) ;
2011-07-27 21:56:59 +03:00
show_banner ( ) ;
2012-03-04 18:46:45 +03:00
parse_cpuflags ( argc , argv , options ) ;
2011-07-27 21:56:59 +03:00
/* parse options */
2011-08-28 18:21:56 +03:00
parse_options ( & o , argc , argv , options , opt_output_file ) ;
2011-07-27 21:56:59 +03:00
2011-12-29 23:28:29 +03:00
if ( nb_output_files < = 0 & & nb_input_files = = 0 ) {
2011-07-27 21:56:59 +03:00
show_usage ( ) ;
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_WARNING , " Use -h to get full help or, even better, run 'man %s' \n " , program_name ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
/* file converter / grab */
if ( nb_output_files < = 0 ) {
fprintf ( stderr , " At least one output file must be specified \n " ) ;
exit_program ( 1 ) ;
}
if ( nb_input_files = = 0 ) {
2011-09-13 08:31:21 +03:00
av_log ( NULL , AV_LOG_FATAL , " At least one input file must be specified \n " ) ;
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
}
ti = getutime ( ) ;
2012-04-15 16:03:07 +03:00
if ( transcode ( ) < 0 )
2011-07-27 21:56:59 +03:00
exit_program ( 1 ) ;
ti = getutime ( ) - ti ;
if ( do_benchmark ) {
int maxrss = getmaxrss ( ) / 1024 ;
printf ( " bench: utime=%0.3fs maxrss=%ikB \n " , ti / 1000000.0 , maxrss ) ;
}
2011-08-29 08:03:24 +03:00
exit_program ( 0 ) ;
return 0 ;
2011-07-27 21:56:59 +03:00
}