123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796 |
- /*
- * Muxer/output file setup.
- *
- * This file is part of FFmpeg.
- *
- * FFmpeg 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.
- *
- * FFmpeg 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 FFmpeg; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- */
- #include <string.h>
- #include "cmdutils.h"
- #include "ffmpeg.h"
- #include "ffmpeg_mux.h"
- #include "fopen_utf8.h"
- #include "libavformat/avformat.h"
- #include "libavformat/avio.h"
- #include "libavcodec/avcodec.h"
- #include "libavfilter/avfilter.h"
- #include "libavutil/avassert.h"
- #include "libavutil/avstring.h"
- #include "libavutil/avutil.h"
- #include "libavutil/bprint.h"
- #include "libavutil/dict.h"
- #include "libavutil/display.h"
- #include "libavutil/getenv_utf8.h"
- #include "libavutil/intreadwrite.h"
- #include "libavutil/log.h"
- #include "libavutil/mem.h"
- #include "libavutil/opt.h"
- #include "libavutil/parseutils.h"
- #include "libavutil/pixdesc.h"
- #define DEFAULT_PASS_LOGFILENAME_PREFIX "ffmpeg2pass"
- static const char *const opt_name_apad[] = {"apad", NULL};
- static const char *const opt_name_autoscale[] = {"autoscale", NULL};
- static const char *const opt_name_bits_per_raw_sample[] = {"bits_per_raw_sample", NULL};
- static const char *const opt_name_bitstream_filters[] = {"bsf", "absf", "vbsf", NULL};
- static const char *const opt_name_copy_initial_nonkeyframes[] = {"copyinkf", NULL};
- static const char *const opt_name_copy_prior_start[] = {"copypriorss", NULL};
- static const char *const opt_name_disposition[] = {"disposition", NULL};
- static const char *const opt_name_enc_time_bases[] = {"enc_time_base", NULL};
- static const char *const opt_name_enc_stats_pre[] = {"enc_stats_pre", NULL};
- static const char *const opt_name_enc_stats_post[] = {"enc_stats_post", NULL};
- static const char *const opt_name_mux_stats[] = {"mux_stats", NULL};
- static const char *const opt_name_enc_stats_pre_fmt[] = {"enc_stats_pre_fmt", NULL};
- static const char *const opt_name_enc_stats_post_fmt[] = {"enc_stats_post_fmt", NULL};
- static const char *const opt_name_mux_stats_fmt[] = {"mux_stats_fmt", NULL};
- static const char *const opt_name_filters[] = {"filter", "af", "vf", NULL};
- static const char *const opt_name_filter_scripts[] = {"filter_script", NULL};
- static const char *const opt_name_fix_sub_duration_heartbeat[] = {"fix_sub_duration_heartbeat", NULL};
- static const char *const opt_name_fps_mode[] = {"fps_mode", NULL};
- static const char *const opt_name_force_fps[] = {"force_fps", NULL};
- static const char *const opt_name_forced_key_frames[] = {"forced_key_frames", NULL};
- static const char *const opt_name_frame_aspect_ratios[] = {"aspect", NULL};
- static const char *const opt_name_intra_matrices[] = {"intra_matrix", NULL};
- static const char *const opt_name_inter_matrices[] = {"inter_matrix", NULL};
- static const char *const opt_name_chroma_intra_matrices[] = {"chroma_intra_matrix", NULL};
- static const char *const opt_name_max_frame_rates[] = {"fpsmax", NULL};
- static const char *const opt_name_max_frames[] = {"frames", "aframes", "vframes", "dframes", NULL};
- static const char *const opt_name_max_muxing_queue_size[] = {"max_muxing_queue_size", NULL};
- static const char *const opt_name_muxing_queue_data_threshold[] = {"muxing_queue_data_threshold", NULL};
- static const char *const opt_name_pass[] = {"pass", NULL};
- static const char *const opt_name_passlogfiles[] = {"passlogfile", NULL};
- static const char *const opt_name_presets[] = {"pre", "apre", "vpre", "spre", NULL};
- static const char *const opt_name_qscale[] = {"q", "qscale", NULL};
- static const char *const opt_name_rc_overrides[] = {"rc_override", NULL};
- static const char *const opt_name_time_bases[] = {"time_base", NULL};
- static const char *const opt_name_audio_channels[] = {"ac", NULL};
- static const char *const opt_name_audio_ch_layouts[] = {"channel_layout", "ch_layout", NULL};
- static const char *const opt_name_audio_sample_rate[] = {"ar", NULL};
- static const char *const opt_name_frame_sizes[] = {"s", NULL};
- static const char *const opt_name_frame_pix_fmts[] = {"pix_fmt", NULL};
- static const char *const opt_name_sample_fmts[] = {"sample_fmt", NULL};
- static int check_opt_bitexact(void *ctx, const AVDictionary *opts,
- const char *opt_name, int flag)
- {
- const AVDictionaryEntry *e = av_dict_get(opts, opt_name, NULL, 0);
- if (e) {
- const AVOption *o = av_opt_find(ctx, opt_name, NULL, 0, 0);
- int val = 0;
- if (!o)
- return 0;
- av_opt_eval_flags(ctx, o, e->value, &val);
- return !!(val & flag);
- }
- return 0;
- }
- static int choose_encoder(const OptionsContext *o, AVFormatContext *s,
- OutputStream *ost, const AVCodec **enc)
- {
- enum AVMediaType type = ost->type;
- char *codec_name = NULL;
- *enc = NULL;
- MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, ost->st);
- if (type != AVMEDIA_TYPE_VIDEO &&
- type != AVMEDIA_TYPE_AUDIO &&
- type != AVMEDIA_TYPE_SUBTITLE) {
- if (codec_name && strcmp(codec_name, "copy")) {
- const char *type_str = av_get_media_type_string(type);
- av_log(ost, AV_LOG_FATAL,
- "Encoder '%s' specified, but only '-codec copy' supported "
- "for %s streams\n", codec_name, type_str);
- return AVERROR(ENOSYS);
- }
- return 0;
- }
- if (!codec_name) {
- ost->par_in->codec_id = av_guess_codec(s->oformat, NULL, s->url, NULL, ost->type);
- *enc = avcodec_find_encoder(ost->par_in->codec_id);
- if (!*enc) {
- av_log(ost, AV_LOG_FATAL, "Automatic encoder selection failed "
- "Default encoder for format %s (codec %s) is "
- "probably disabled. Please choose an encoder manually.\n",
- s->oformat->name, avcodec_get_name(ost->par_in->codec_id));
- return AVERROR_ENCODER_NOT_FOUND;
- }
- } else if (strcmp(codec_name, "copy")) {
- int ret = find_codec(ost, codec_name, ost->type, 1, enc);
- if (ret < 0)
- return ret;
- ost->par_in->codec_id = (*enc)->id;
- }
- return 0;
- }
- static char *get_line(AVIOContext *s, AVBPrint *bprint)
- {
- char c;
- while ((c = avio_r8(s)) && c != '\n')
- av_bprint_chars(bprint, c, 1);
- if (!av_bprint_is_complete(bprint))
- return NULL;
- return bprint->str;
- }
- static int get_preset_file_2(const char *preset_name, const char *codec_name, AVIOContext **s)
- {
- int i, ret = -1;
- char filename[1000];
- char *env_avconv_datadir = getenv_utf8("AVCONV_DATADIR");
- char *env_home = getenv_utf8("HOME");
- const char *base[3] = { env_avconv_datadir,
- env_home,
- AVCONV_DATADIR,
- };
- for (i = 0; i < FF_ARRAY_ELEMS(base) && ret < 0; 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);
- ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
- }
- if (ret < 0) {
- snprintf(filename, sizeof(filename), "%s%s/%s.avpreset", base[i],
- i != 1 ? "" : "/.avconv", preset_name);
- ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
- }
- }
- freeenv_utf8(env_home);
- freeenv_utf8(env_avconv_datadir);
- return ret;
- }
- typedef struct EncStatsFile {
- char *path;
- AVIOContext *io;
- } EncStatsFile;
- static EncStatsFile *enc_stats_files;
- static int nb_enc_stats_files;
- static int enc_stats_get_file(AVIOContext **io, const char *path)
- {
- EncStatsFile *esf;
- int ret;
- for (int i = 0; i < nb_enc_stats_files; i++)
- if (!strcmp(path, enc_stats_files[i].path)) {
- *io = enc_stats_files[i].io;
- return 0;
- }
- ret = GROW_ARRAY(enc_stats_files, nb_enc_stats_files);
- if (ret < 0)
- return ret;
- esf = &enc_stats_files[nb_enc_stats_files - 1];
- ret = avio_open2(&esf->io, path, AVIO_FLAG_WRITE, &int_cb, NULL);
- if (ret < 0) {
- av_log(NULL, AV_LOG_ERROR, "Error opening stats file '%s': %s\n",
- path, av_err2str(ret));
- return ret;
- }
- esf->path = av_strdup(path);
- if (!esf->path)
- return AVERROR(ENOMEM);
- *io = esf->io;
- return 0;
- }
- void of_enc_stats_close(void)
- {
- for (int i = 0; i < nb_enc_stats_files; i++) {
- av_freep(&enc_stats_files[i].path);
- avio_closep(&enc_stats_files[i].io);
- }
- av_freep(&enc_stats_files);
- nb_enc_stats_files = 0;
- }
- static int unescape(char **pdst, size_t *dst_len,
- const char **pstr, char delim)
- {
- const char *str = *pstr;
- char *dst;
- size_t len, idx;
- *pdst = NULL;
- len = strlen(str);
- if (!len)
- return 0;
- dst = av_malloc(len + 1);
- if (!dst)
- return AVERROR(ENOMEM);
- for (idx = 0; *str; idx++, str++) {
- if (str[0] == '\\' && str[1])
- str++;
- else if (*str == delim)
- break;
- dst[idx] = *str;
- }
- if (!idx) {
- av_freep(&dst);
- return 0;
- }
- dst[idx] = 0;
- *pdst = dst;
- *dst_len = idx;
- *pstr = str;
- return 0;
- }
- static int enc_stats_init(OutputStream *ost, EncStats *es, int pre,
- const char *path, const char *fmt_spec)
- {
- static const struct {
- enum EncStatsType type;
- const char *str;
- int pre_only:1;
- int post_only:1;
- int need_input_data:1;
- } fmt_specs[] = {
- { ENC_STATS_FILE_IDX, "fidx" },
- { ENC_STATS_STREAM_IDX, "sidx" },
- { ENC_STATS_FRAME_NUM, "n" },
- { ENC_STATS_FRAME_NUM_IN, "ni", 0, 0, 1 },
- { ENC_STATS_TIMEBASE, "tb" },
- { ENC_STATS_TIMEBASE_IN, "tbi", 0, 0, 1 },
- { ENC_STATS_PTS, "pts" },
- { ENC_STATS_PTS_TIME, "t" },
- { ENC_STATS_PTS_IN, "ptsi", 0, 0, 1 },
- { ENC_STATS_PTS_TIME_IN, "ti", 0, 0, 1 },
- { ENC_STATS_DTS, "dts", 0, 1 },
- { ENC_STATS_DTS_TIME, "dt", 0, 1 },
- { ENC_STATS_SAMPLE_NUM, "sn", 1 },
- { ENC_STATS_NB_SAMPLES, "samp", 1 },
- { ENC_STATS_PKT_SIZE, "size", 0, 1 },
- { ENC_STATS_BITRATE, "br", 0, 1 },
- { ENC_STATS_AVG_BITRATE, "abr", 0, 1 },
- };
- const char *next = fmt_spec;
- int ret;
- while (*next) {
- EncStatsComponent *c;
- char *val;
- size_t val_len;
- // get the sequence up until next opening brace
- ret = unescape(&val, &val_len, &next, '{');
- if (ret < 0)
- return ret;
- if (val) {
- ret = GROW_ARRAY(es->components, es->nb_components);
- if (ret < 0) {
- av_freep(&val);
- return ret;
- }
- c = &es->components[es->nb_components - 1];
- c->type = ENC_STATS_LITERAL;
- c->str = val;
- c->str_len = val_len;
- }
- if (!*next)
- break;
- next++;
- // get the part inside braces
- ret = unescape(&val, &val_len, &next, '}');
- if (ret < 0)
- return ret;
- if (!val) {
- av_log(NULL, AV_LOG_ERROR,
- "Empty formatting directive in: %s\n", fmt_spec);
- return AVERROR(EINVAL);
- }
- if (!*next) {
- av_log(NULL, AV_LOG_ERROR,
- "Missing closing brace in: %s\n", fmt_spec);
- ret = AVERROR(EINVAL);
- goto fail;
- }
- next++;
- ret = GROW_ARRAY(es->components, es->nb_components);
- if (ret < 0)
- goto fail;
- c = &es->components[es->nb_components - 1];
- for (size_t i = 0; i < FF_ARRAY_ELEMS(fmt_specs); i++) {
- if (!strcmp(val, fmt_specs[i].str)) {
- if ((pre && fmt_specs[i].post_only) || (!pre && fmt_specs[i].pre_only)) {
- av_log(NULL, AV_LOG_ERROR,
- "Format directive '%s' may only be used %s-encoding\n",
- val, pre ? "post" : "pre");
- ret = AVERROR(EINVAL);
- goto fail;
- }
- c->type = fmt_specs[i].type;
- if (fmt_specs[i].need_input_data && !ost->ist) {
- av_log(ost, AV_LOG_WARNING,
- "Format directive '%s' is unavailable, because "
- "this output stream has no associated input stream\n",
- val);
- }
- break;
- }
- }
- if (!c->type) {
- av_log(NULL, AV_LOG_ERROR, "Invalid format directive: %s\n", val);
- ret = AVERROR(EINVAL);
- goto fail;
- }
- fail:
- av_freep(&val);
- if (ret < 0)
- return ret;
- }
- ret = enc_stats_get_file(&es->io, path);
- if (ret < 0)
- return ret;
- return 0;
- }
- static const char *output_stream_item_name(void *obj)
- {
- const MuxStream *ms = obj;
- return ms->log_name;
- }
- static const AVClass output_stream_class = {
- .class_name = "OutputStream",
- .version = LIBAVUTIL_VERSION_INT,
- .item_name = output_stream_item_name,
- .category = AV_CLASS_CATEGORY_MUXER,
- };
- static MuxStream *mux_stream_alloc(Muxer *mux, enum AVMediaType type)
- {
- const char *type_str = av_get_media_type_string(type);
- MuxStream *ms;
- ms = allocate_array_elem(&mux->of.streams, sizeof(*ms), &mux->of.nb_streams);
- if (!ms)
- return NULL;
- ms->ost.file_index = mux->of.index;
- ms->ost.index = mux->of.nb_streams - 1;
- ms->ost.type = type;
- ms->ost.class = &output_stream_class;
- snprintf(ms->log_name, sizeof(ms->log_name), "%cost#%d:%d",
- type_str ? *type_str : '?', mux->of.index, ms->ost.index);
- return ms;
- }
- static int ost_get_filters(const OptionsContext *o, AVFormatContext *oc,
- OutputStream *ost, char **dst)
- {
- const char *filters = NULL, *filters_script = NULL;
- MATCH_PER_STREAM_OPT(filter_scripts, str, filters_script, oc, ost->st);
- MATCH_PER_STREAM_OPT(filters, str, filters, oc, ost->st);
- if (!ost->enc) {
- if (filters_script || filters) {
- av_log(ost, AV_LOG_ERROR,
- "%s '%s' was specified, but codec copy was selected. "
- "Filtering and streamcopy cannot be used together.\n",
- filters ? "Filtergraph" : "Filtergraph script",
- filters ? filters : filters_script);
- return AVERROR(ENOSYS);
- }
- return 0;
- }
- if (!ost->ist) {
- if (filters_script || filters) {
- av_log(ost, AV_LOG_ERROR,
- "%s '%s' was specified for a stream fed from a complex "
- "filtergraph. Simple and complex filtering cannot be used "
- "together for the same stream.\n",
- filters ? "Filtergraph" : "Filtergraph script",
- filters ? filters : filters_script);
- return AVERROR(EINVAL);
- }
- return 0;
- }
- if (filters_script && filters) {
- av_log(ost, AV_LOG_ERROR, "Both -filter and -filter_script set\n");
- return AVERROR(EINVAL);
- }
- if (filters_script)
- *dst = file_read(filters_script);
- else if (filters)
- *dst = av_strdup(filters);
- else
- *dst = av_strdup(ost->type == AVMEDIA_TYPE_VIDEO ? "null" : "anull");
- return *dst ? 0 : AVERROR(ENOMEM);
- }
- static int parse_matrix_coeffs(void *logctx, uint16_t *dest, const char *str)
- {
- int i;
- const char *p = str;
- for (i = 0;; i++) {
- dest[i] = atoi(p);
- if (i == 63)
- break;
- p = strchr(p, ',');
- if (!p) {
- av_log(logctx, AV_LOG_FATAL,
- "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
- return AVERROR(EINVAL);
- }
- p++;
- }
- return 0;
- }
- static int fmt_in_list(const int *formats, int format)
- {
- for (; *formats != -1; formats++)
- if (*formats == format)
- return 1;
- return 0;
- }
- static enum AVPixelFormat
- choose_pixel_fmt(const AVCodec *codec, enum AVPixelFormat target)
- {
- const enum AVPixelFormat *p = codec->pix_fmts;
- const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(target);
- //FIXME: This should check for AV_PIX_FMT_FLAG_ALPHA after PAL8 pixel format without alpha is implemented
- int has_alpha = desc ? desc->nb_components % 2 == 0 : 0;
- enum AVPixelFormat best= AV_PIX_FMT_NONE;
- for (; *p != AV_PIX_FMT_NONE; p++) {
- best = av_find_best_pix_fmt_of_2(best, *p, target, has_alpha, NULL);
- if (*p == target)
- break;
- }
- if (*p == AV_PIX_FMT_NONE) {
- if (target != AV_PIX_FMT_NONE)
- av_log(NULL, AV_LOG_WARNING,
- "Incompatible pixel format '%s' for codec '%s', auto-selecting format '%s'\n",
- av_get_pix_fmt_name(target),
- codec->name,
- av_get_pix_fmt_name(best));
- return best;
- }
- return target;
- }
- static enum AVPixelFormat pix_fmt_parse(OutputStream *ost, const char *name)
- {
- const enum AVPixelFormat *fmts = ost->enc_ctx->codec->pix_fmts;
- enum AVPixelFormat fmt;
- fmt = av_get_pix_fmt(name);
- if (fmt == AV_PIX_FMT_NONE) {
- av_log(ost, AV_LOG_FATAL, "Unknown pixel format requested: %s.\n", name);
- return AV_PIX_FMT_NONE;
- }
- /* when the user specified-format is an alias for an endianness-specific
- * one (e.g. rgb48 -> rgb48be/le), it gets translated into the native
- * endianness by av_get_pix_fmt();
- * the following code handles the case when the native endianness is not
- * supported by the encoder, but the other one is */
- if (fmts && !fmt_in_list(fmts, fmt)) {
- const char *name_canonical = av_get_pix_fmt_name(fmt);
- int len = strlen(name_canonical);
- if (strcmp(name, name_canonical) &&
- (!strcmp(name_canonical + len - 2, "le") ||
- !strcmp(name_canonical + len - 2, "be"))) {
- char name_other[64];
- enum AVPixelFormat fmt_other;
- snprintf(name_other, sizeof(name_other), "%s%ce",
- name, name_canonical[len - 2] == 'l' ? 'b' : 'l');
- fmt_other = av_get_pix_fmt(name_other);
- if (fmt_other != AV_PIX_FMT_NONE && fmt_in_list(fmts, fmt_other)) {
- av_log(ost, AV_LOG_VERBOSE, "Mapping pixel format %s->%s\n",
- name, name_other);
- fmt = fmt_other;
- }
- }
- }
- if (fmts && !fmt_in_list(fmts, fmt))
- fmt = choose_pixel_fmt(ost->enc_ctx->codec, fmt);
- return fmt;
- }
- static int new_stream_video(Muxer *mux, const OptionsContext *o,
- OutputStream *ost)
- {
- AVFormatContext *oc = mux->fc;
- AVStream *st;
- char *frame_rate = NULL, *max_frame_rate = NULL, *frame_aspect_ratio = NULL;
- int ret = 0;
- st = ost->st;
- MATCH_PER_STREAM_OPT(frame_rates, str, frame_rate, oc, st);
- if (frame_rate && av_parse_video_rate(&ost->frame_rate, frame_rate) < 0) {
- av_log(ost, AV_LOG_FATAL, "Invalid framerate value: %s\n", frame_rate);
- return AVERROR(EINVAL);
- }
- MATCH_PER_STREAM_OPT(max_frame_rates, str, max_frame_rate, oc, st);
- if (max_frame_rate && av_parse_video_rate(&ost->max_frame_rate, max_frame_rate) < 0) {
- av_log(ost, AV_LOG_FATAL, "Invalid maximum framerate value: %s\n", max_frame_rate);
- return AVERROR(EINVAL);
- }
- if (frame_rate && max_frame_rate) {
- av_log(ost, AV_LOG_ERROR, "Only one of -fpsmax and -r can be set for a stream.\n");
- return AVERROR(EINVAL);
- }
- MATCH_PER_STREAM_OPT(frame_aspect_ratios, str, frame_aspect_ratio, oc, st);
- if (frame_aspect_ratio) {
- AVRational q;
- if (av_parse_ratio(&q, frame_aspect_ratio, 255, 0, NULL) < 0 ||
- q.num <= 0 || q.den <= 0) {
- av_log(ost, AV_LOG_FATAL, "Invalid aspect ratio: %s\n", frame_aspect_ratio);
- return AVERROR(EINVAL);
- }
- ost->frame_aspect_ratio = q;
- }
- if (ost->enc_ctx) {
- AVCodecContext *video_enc = ost->enc_ctx;
- const char *p = NULL, *fps_mode = NULL;
- char *frame_size = NULL;
- char *frame_pix_fmt = NULL;
- char *intra_matrix = NULL, *inter_matrix = NULL;
- char *chroma_intra_matrix = NULL;
- int do_pass = 0;
- int i;
- MATCH_PER_STREAM_OPT(frame_sizes, str, frame_size, oc, st);
- if (frame_size) {
- ret = av_parse_video_size(&video_enc->width, &video_enc->height, frame_size);
- if (ret < 0) {
- av_log(ost, AV_LOG_FATAL, "Invalid frame size: %s.\n", frame_size);
- return AVERROR(EINVAL);
- }
- }
- MATCH_PER_STREAM_OPT(frame_pix_fmts, str, frame_pix_fmt, oc, st);
- if (frame_pix_fmt && *frame_pix_fmt == '+') {
- ost->keep_pix_fmt = 1;
- if (!*++frame_pix_fmt)
- frame_pix_fmt = NULL;
- }
- if (frame_pix_fmt) {
- video_enc->pix_fmt = pix_fmt_parse(ost, frame_pix_fmt);
- if (video_enc->pix_fmt == AV_PIX_FMT_NONE)
- return AVERROR(EINVAL);
- }
- 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)))
- return AVERROR(ENOMEM);
- ret = parse_matrix_coeffs(ost, video_enc->intra_matrix, intra_matrix);
- if (ret < 0)
- return ret;
- }
- MATCH_PER_STREAM_OPT(chroma_intra_matrices, str, chroma_intra_matrix, oc, st);
- if (chroma_intra_matrix) {
- uint16_t *p = av_mallocz(sizeof(*video_enc->chroma_intra_matrix) * 64);
- if (!p)
- return AVERROR(ENOMEM);
- video_enc->chroma_intra_matrix = p;
- ret = parse_matrix_coeffs(ost, p, chroma_intra_matrix);
- if (ret < 0)
- return ret;
- }
- 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)))
- return AVERROR(ENOMEM);
- ret = parse_matrix_coeffs(ost, video_enc->inter_matrix, inter_matrix);
- if (ret < 0)
- return ret;
- }
- MATCH_PER_STREAM_OPT(rc_overrides, str, p, oc, st);
- for (i = 0; p; i++) {
- int start, end, q;
- int e = sscanf(p, "%d,%d,%d", &start, &end, &q);
- if (e != 3) {
- av_log(ost, AV_LOG_FATAL, "error parsing rc_override\n");
- return AVERROR(EINVAL);
- }
- video_enc->rc_override =
- av_realloc_array(video_enc->rc_override,
- i + 1, sizeof(RcOverride));
- if (!video_enc->rc_override) {
- av_log(ost, AV_LOG_FATAL, "Could not (re)allocate memory for rc_override.\n");
- return AVERROR(ENOMEM);
- }
- 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;
- }
- else {
- video_enc->rc_override[i].qscale = 0;
- video_enc->rc_override[i].quality_factor = -q/100.0;
- }
- p = strchr(p, '/');
- if (p) p++;
- }
- video_enc->rc_override_count = i;
- #if FFMPEG_OPT_PSNR
- if (do_psnr) {
- av_log(ost, AV_LOG_WARNING, "The -psnr option is deprecated, use -flags +psnr\n");
- video_enc->flags|= AV_CODEC_FLAG_PSNR;
- }
- #endif
- /* two pass mode */
- MATCH_PER_STREAM_OPT(pass, i, do_pass, oc, st);
- if (do_pass) {
- if (do_pass & 1) {
- video_enc->flags |= AV_CODEC_FLAG_PASS1;
- av_dict_set(&ost->encoder_opts, "flags", "+pass1", AV_DICT_APPEND);
- }
- if (do_pass & 2) {
- video_enc->flags |= AV_CODEC_FLAG_PASS2;
- av_dict_set(&ost->encoder_opts, "flags", "+pass2", AV_DICT_APPEND);
- }
- }
- MATCH_PER_STREAM_OPT(passlogfiles, str, ost->logfile_prefix, oc, st);
- if (ost->logfile_prefix &&
- !(ost->logfile_prefix = av_strdup(ost->logfile_prefix)))
- return AVERROR(ENOMEM);
- if (do_pass) {
- int ost_idx = -1;
- char logfilename[1024];
- FILE *f;
- /* compute this stream's global index */
- for (int i = 0; i <= ost->file_index; i++)
- ost_idx += output_files[i]->nb_streams;
- snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
- ost->logfile_prefix ? ost->logfile_prefix :
- DEFAULT_PASS_LOGFILENAME_PREFIX,
- ost_idx);
- if (!strcmp(ost->enc_ctx->codec->name, "libx264")) {
- av_dict_set(&ost->encoder_opts, "stats", logfilename, AV_DICT_DONT_OVERWRITE);
- } else {
- if (video_enc->flags & AV_CODEC_FLAG_PASS2) {
- char *logbuffer = file_read(logfilename);
- if (!logbuffer) {
- av_log(ost, AV_LOG_FATAL, "Error reading log file '%s' for pass-2 encoding\n",
- logfilename);
- return AVERROR(EIO);
- }
- video_enc->stats_in = logbuffer;
- }
- if (video_enc->flags & AV_CODEC_FLAG_PASS1) {
- f = fopen_utf8(logfilename, "wb");
- if (!f) {
- av_log(ost, AV_LOG_FATAL,
- "Cannot write log file '%s' for pass-1 encoding: %s\n",
- logfilename, strerror(errno));
- return AVERROR(errno);
- }
- ost->logfile = f;
- }
- }
- }
- MATCH_PER_STREAM_OPT(force_fps, i, ost->force_fps, oc, st);
- #if FFMPEG_OPT_TOP
- ost->top_field_first = -1;
- MATCH_PER_STREAM_OPT(top_field_first, i, ost->top_field_first, oc, st);
- if (ost->top_field_first >= 0)
- av_log(ost, AV_LOG_WARNING, "-top is deprecated, use the setfield filter instead\n");
- #endif
- ost->vsync_method = video_sync_method;
- MATCH_PER_STREAM_OPT(fps_mode, str, fps_mode, oc, st);
- if (fps_mode) {
- ret = parse_and_set_vsync(fps_mode, &ost->vsync_method, ost->file_index, ost->index, 0);
- if (ret < 0)
- return ret;
- }
- if ((ost->frame_rate.num || ost->max_frame_rate.num) &&
- !(ost->vsync_method == VSYNC_AUTO ||
- ost->vsync_method == VSYNC_CFR || ost->vsync_method == VSYNC_VSCFR)) {
- av_log(ost, AV_LOG_FATAL, "One of -r/-fpsmax was specified "
- "together a non-CFR -vsync/-fps_mode. This is contradictory.\n");
- return AVERROR(EINVAL);
- }
- if (ost->vsync_method == VSYNC_AUTO) {
- if (ost->frame_rate.num || ost->max_frame_rate.num) {
- ost->vsync_method = VSYNC_CFR;
- } else if (!strcmp(oc->oformat->name, "avi")) {
- ost->vsync_method = VSYNC_VFR;
- } else {
- ost->vsync_method = (oc->oformat->flags & AVFMT_VARIABLE_FPS) ?
- ((oc->oformat->flags & AVFMT_NOTIMESTAMPS) ?
- VSYNC_PASSTHROUGH : VSYNC_VFR) :
- VSYNC_CFR;
- }
- if (ost->ist && ost->vsync_method == VSYNC_CFR) {
- const InputFile *ifile = input_files[ost->ist->file_index];
- if (ifile->nb_streams == 1 && ifile->input_ts_offset == 0)
- ost->vsync_method = VSYNC_VSCFR;
- }
- if (ost->vsync_method == VSYNC_CFR && copy_ts) {
- ost->vsync_method = VSYNC_VSCFR;
- }
- }
- ost->is_cfr = (ost->vsync_method == VSYNC_CFR || ost->vsync_method == VSYNC_VSCFR);
- }
- return 0;
- }
- static int new_stream_audio(Muxer *mux, const OptionsContext *o,
- OutputStream *ost)
- {
- AVFormatContext *oc = mux->fc;
- AVStream *st;
- int ret = 0;
- st = ost->st;
- if (ost->enc_ctx) {
- AVCodecContext *audio_enc = ost->enc_ctx;
- int channels = 0;
- char *layout = NULL;
- char *sample_fmt = NULL;
- MATCH_PER_STREAM_OPT(audio_channels, i, channels, oc, st);
- if (channels) {
- audio_enc->ch_layout.order = AV_CHANNEL_ORDER_UNSPEC;
- audio_enc->ch_layout.nb_channels = channels;
- }
- MATCH_PER_STREAM_OPT(audio_ch_layouts, str, layout, oc, st);
- if (layout) {
- if (av_channel_layout_from_string(&audio_enc->ch_layout, layout) < 0) {
- #if FF_API_OLD_CHANNEL_LAYOUT
- uint64_t mask;
- AV_NOWARN_DEPRECATED({
- mask = av_get_channel_layout(layout);
- })
- if (!mask) {
- #endif
- av_log(ost, AV_LOG_FATAL, "Unknown channel layout: %s\n", layout);
- return AVERROR(EINVAL);
- #if FF_API_OLD_CHANNEL_LAYOUT
- }
- av_log(ost, AV_LOG_WARNING, "Channel layout '%s' uses a deprecated syntax.\n",
- layout);
- av_channel_layout_from_mask(&audio_enc->ch_layout, mask);
- #endif
- }
- }
- 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) {
- av_log(ost, AV_LOG_FATAL, "Invalid sample format '%s'\n", sample_fmt);
- return AVERROR(EINVAL);
- }
- MATCH_PER_STREAM_OPT(audio_sample_rate, i, audio_enc->sample_rate, oc, st);
- MATCH_PER_STREAM_OPT(apad, str, ost->apad, oc, st);
- ost->apad = av_strdup(ost->apad);
- #if FFMPEG_OPT_MAP_CHANNEL
- /* check for channel mapping for this audio stream */
- for (int n = 0; n < o->nb_audio_channel_maps; n++) {
- AudioChannelMap *map = &o->audio_channel_maps[n];
- if ((map->ofile_idx == -1 || ost->file_index == map->ofile_idx) &&
- (map->ostream_idx == -1 || ost->st->index == map->ostream_idx)) {
- InputStream *ist;
- if (map->channel_idx == -1) {
- ist = NULL;
- } else if (!ost->ist) {
- av_log(ost, AV_LOG_FATAL, "Cannot determine input stream for channel mapping %d.%d\n",
- ost->file_index, ost->st->index);
- continue;
- } else {
- ist = ost->ist;
- }
- if (!ist || (ist->file_index == map->file_idx && ist->index == map->stream_idx)) {
- ret = av_reallocp_array(&ost->audio_channels_map,
- ost->audio_channels_mapped + 1,
- sizeof(*ost->audio_channels_map));
- if (ret < 0)
- return ret;
- ost->audio_channels_map[ost->audio_channels_mapped++] = map->channel_idx;
- }
- }
- }
- #endif
- }
- return 0;
- }
- static int new_stream_attachment(Muxer *mux, const OptionsContext *o,
- OutputStream *ost)
- {
- ost->finished = 1;
- return 0;
- }
- static int new_stream_subtitle(Muxer *mux, const OptionsContext *o,
- OutputStream *ost)
- {
- AVStream *st;
- st = ost->st;
- if (ost->enc_ctx) {
- AVCodecContext *subtitle_enc = ost->enc_ctx;
- AVCodecDescriptor const *input_descriptor =
- avcodec_descriptor_get(ost->ist->par->codec_id);
- AVCodecDescriptor const *output_descriptor =
- avcodec_descriptor_get(subtitle_enc->codec_id);
- int input_props = 0, output_props = 0;
- char *frame_size = NULL;
- MATCH_PER_STREAM_OPT(frame_sizes, str, frame_size, mux->fc, st);
- if (frame_size) {
- int ret = av_parse_video_size(&subtitle_enc->width, &subtitle_enc->height, frame_size);
- if (ret < 0) {
- av_log(ost, AV_LOG_FATAL, "Invalid frame size: %s.\n", frame_size);
- return ret;
- }
- }
- if (input_descriptor)
- input_props = input_descriptor->props & (AV_CODEC_PROP_TEXT_SUB | AV_CODEC_PROP_BITMAP_SUB);
- if (output_descriptor)
- output_props = output_descriptor->props & (AV_CODEC_PROP_TEXT_SUB | AV_CODEC_PROP_BITMAP_SUB);
- if (input_props && output_props && input_props != output_props) {
- av_log(ost, AV_LOG_ERROR,
- "Subtitle encoding currently only possible from text to text "
- "or bitmap to bitmap\n");
- return AVERROR(EINVAL);
- }
- }
- return 0;
- }
- static int streamcopy_init(const Muxer *mux, OutputStream *ost)
- {
- MuxStream *ms = ms_from_ost(ost);
- const InputStream *ist = ost->ist;
- const InputFile *ifile = input_files[ist->file_index];
- AVCodecParameters *par = ost->par_in;
- uint32_t codec_tag = par->codec_tag;
- AVCodecContext *codec_ctx = NULL;
- AVDictionary *codec_opts = NULL;
- AVRational fr = ost->frame_rate;
- int ret = 0;
- codec_ctx = avcodec_alloc_context3(NULL);
- if (!codec_ctx)
- return AVERROR(ENOMEM);
- ret = avcodec_parameters_to_context(codec_ctx, ist->par);
- if (ret >= 0)
- ret = av_opt_set_dict(codec_ctx, &ost->encoder_opts);
- if (ret < 0) {
- av_log(ost, AV_LOG_FATAL,
- "Error setting up codec context options.\n");
- goto fail;
- }
- ret = avcodec_parameters_from_context(par, codec_ctx);
- if (ret < 0) {
- av_log(ost, AV_LOG_FATAL,
- "Error getting reference codec parameters.\n");
- goto fail;
- }
- if (!codec_tag) {
- const struct AVCodecTag * const *ct = mux->fc->oformat->codec_tag;
- unsigned int codec_tag_tmp;
- if (!ct || av_codec_get_id (ct, par->codec_tag) == par->codec_id ||
- !av_codec_get_tag2(ct, par->codec_id, &codec_tag_tmp))
- codec_tag = par->codec_tag;
- }
- par->codec_tag = codec_tag;
- if (!fr.num)
- fr = ist->framerate;
- if (fr.num)
- ost->st->avg_frame_rate = fr;
- else
- ost->st->avg_frame_rate = ist->st->avg_frame_rate;
- ret = avformat_transfer_internal_stream_timing_info(mux->fc->oformat,
- ost->st, ist->st, copy_tb);
- if (ret < 0)
- goto fail;
- // copy timebase while removing common factors
- if (ost->st->time_base.num <= 0 || ost->st->time_base.den <= 0) {
- if (fr.num)
- ost->st->time_base = av_inv_q(fr);
- else
- ost->st->time_base = av_add_q(av_stream_get_codec_timebase(ost->st), (AVRational){0, 1});
- }
- if (!ms->copy_prior_start) {
- ms->ts_copy_start = (mux->of.start_time == AV_NOPTS_VALUE) ?
- 0 : mux->of.start_time;
- if (copy_ts && ifile->start_time != AV_NOPTS_VALUE) {
- ms->ts_copy_start = FFMAX(ms->ts_copy_start,
- ifile->start_time + ifile->ts_offset);
- }
- }
- for (int i = 0; i < ist->st->codecpar->nb_coded_side_data; i++) {
- const AVPacketSideData *sd_src = &ist->st->codecpar->coded_side_data[i];
- AVPacketSideData *sd_dst;
- sd_dst = av_packet_side_data_new(&ost->st->codecpar->coded_side_data,
- &ost->st->codecpar->nb_coded_side_data,
- sd_src->type, sd_src->size, 0);
- if (!sd_dst) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- memcpy(sd_dst->data, sd_src->data, sd_src->size);
- }
- #if FFMPEG_ROTATION_METADATA
- if (ost->rotate_overridden) {
- AVPacketSideData *sd = av_packet_side_data_new(&ost->st->codecpar->coded_side_data,
- &ost->st->codecpar->nb_coded_side_data,
- AV_PKT_DATA_DISPLAYMATRIX,
- sizeof(int32_t) * 9, 0);
- if (sd)
- av_display_rotation_set((int32_t *)sd->data, -ost->rotate_override_value);
- }
- #endif
- switch (par->codec_type) {
- case AVMEDIA_TYPE_AUDIO:
- if ((par->block_align == 1 || par->block_align == 1152 || par->block_align == 576) &&
- par->codec_id == AV_CODEC_ID_MP3)
- par->block_align = 0;
- if (par->codec_id == AV_CODEC_ID_AC3)
- par->block_align = 0;
- break;
- case AVMEDIA_TYPE_VIDEO: {
- AVRational sar;
- if (ost->frame_aspect_ratio.num) { // overridden by the -aspect cli option
- sar =
- av_mul_q(ost->frame_aspect_ratio,
- (AVRational){ par->height, par->width });
- av_log(ost, AV_LOG_WARNING, "Overriding aspect ratio "
- "with stream copy may produce invalid files\n");
- }
- else if (ist->st->sample_aspect_ratio.num)
- sar = ist->st->sample_aspect_ratio;
- else
- sar = par->sample_aspect_ratio;
- ost->st->sample_aspect_ratio = par->sample_aspect_ratio = sar;
- ost->st->avg_frame_rate = ist->st->avg_frame_rate;
- ost->st->r_frame_rate = ist->st->r_frame_rate;
- break;
- }
- }
- fail:
- avcodec_free_context(&codec_ctx);
- av_dict_free(&codec_opts);
- return ret;
- }
- static int ost_add(Muxer *mux, const OptionsContext *o, enum AVMediaType type,
- InputStream *ist, OutputFilter *ofilter,
- OutputStream **post)
- {
- AVFormatContext *oc = mux->fc;
- MuxStream *ms;
- OutputStream *ost;
- const AVCodec *enc;
- AVStream *st;
- int ret = 0;
- const char *bsfs = NULL, *time_base = NULL;
- char *filters = NULL, *next, *codec_tag = NULL;
- double qscale = -1;
- int i;
- st = avformat_new_stream(oc, NULL);
- if (!st)
- return AVERROR(ENOMEM);
- ms = mux_stream_alloc(mux, type);
- if (!ms)
- return AVERROR(ENOMEM);
- ost = &ms->ost;
- if (o->streamid) {
- AVDictionaryEntry *e;
- char idx[16], *p;
- snprintf(idx, sizeof(idx), "%d", ost->index);
- e = av_dict_get(o->streamid, idx, NULL, 0);
- if (e) {
- st->id = strtol(e->value, &p, 0);
- if (!e->value[0] || *p) {
- av_log(ost, AV_LOG_FATAL, "Invalid stream id: %s\n", e->value);
- return AVERROR(EINVAL);
- }
- }
- }
- ost->par_in = avcodec_parameters_alloc();
- if (!ost->par_in)
- return AVERROR(ENOMEM);
- ms->muxing_queue = av_fifo_alloc2(8, sizeof(AVPacket*), 0);
- if (!ms->muxing_queue)
- return AVERROR(ENOMEM);
- ms->last_mux_dts = AV_NOPTS_VALUE;
- ost->st = st;
- ost->ist = ist;
- ost->kf.ref_pts = AV_NOPTS_VALUE;
- ost->par_in->codec_type = type;
- st->codecpar->codec_type = type;
- ret = choose_encoder(o, oc, ost, &enc);
- if (ret < 0) {
- av_log(ost, AV_LOG_FATAL, "Error selecting an encoder\n");
- return ret;
- }
- if (enc) {
- ost->enc_ctx = avcodec_alloc_context3(enc);
- if (!ost->enc_ctx)
- return AVERROR(ENOMEM);
- ret = enc_alloc(&ost->enc, enc);
- if (ret < 0)
- return ret;
- av_strlcat(ms->log_name, "/", sizeof(ms->log_name));
- av_strlcat(ms->log_name, enc->name, sizeof(ms->log_name));
- } else {
- if (ofilter) {
- av_log(ost, AV_LOG_ERROR,
- "Streamcopy requested for output stream fed "
- "from a complex filtergraph. Filtering and streamcopy "
- "cannot be used together.\n");
- return AVERROR(EINVAL);
- }
- av_strlcat(ms->log_name, "/copy", sizeof(ms->log_name));
- }
- av_log(ost, AV_LOG_VERBOSE, "Created %s stream from ",
- av_get_media_type_string(type));
- if (ist)
- av_log(ost, AV_LOG_VERBOSE, "input stream %d:%d",
- ist->file_index, ist->index);
- else if (ofilter)
- av_log(ost, AV_LOG_VERBOSE, "complex filtergraph %d:[%s]\n",
- ofilter->graph->index, ofilter->name);
- else if (type == AVMEDIA_TYPE_ATTACHMENT)
- av_log(ost, AV_LOG_VERBOSE, "attached file");
- else av_assert0(0);
- av_log(ost, AV_LOG_VERBOSE, "\n");
- ms->pkt = av_packet_alloc();
- if (!ms->pkt)
- return AVERROR(ENOMEM);
- if (ost->enc_ctx) {
- AVCodecContext *enc = ost->enc_ctx;
- AVIOContext *s = NULL;
- char *buf = NULL, *arg = NULL, *preset = NULL;
- const char *enc_stats_pre = NULL, *enc_stats_post = NULL, *mux_stats = NULL;
- const char *enc_time_base = NULL;
- ret = filter_codec_opts(o->g->codec_opts, enc->codec_id,
- oc, st, enc->codec, &ost->encoder_opts);
- if (ret < 0)
- return ret;
- MATCH_PER_STREAM_OPT(presets, str, preset, oc, st);
- ost->autoscale = 1;
- MATCH_PER_STREAM_OPT(autoscale, i, ost->autoscale, oc, st);
- if (preset && (!(ret = get_preset_file_2(preset, enc->codec->name, &s)))) {
- AVBPrint bprint;
- av_bprint_init(&bprint, 0, AV_BPRINT_SIZE_UNLIMITED);
- do {
- av_bprint_clear(&bprint);
- buf = get_line(s, &bprint);
- if (!buf) {
- ret = AVERROR(ENOMEM);
- break;
- }
- if (!buf[0] || buf[0] == '#')
- continue;
- if (!(arg = strchr(buf, '='))) {
- av_log(ost, AV_LOG_FATAL, "Invalid line found in the preset file.\n");
- ret = AVERROR(EINVAL);
- break;
- }
- *arg++ = 0;
- av_dict_set(&ost->encoder_opts, buf, arg, AV_DICT_DONT_OVERWRITE);
- } while (!s->eof_reached);
- av_bprint_finalize(&bprint, NULL);
- avio_closep(&s);
- }
- if (ret) {
- av_log(ost, AV_LOG_FATAL,
- "Preset %s specified, but could not be opened.\n", preset);
- return ret;
- }
- MATCH_PER_STREAM_OPT(enc_stats_pre, str, enc_stats_pre, oc, st);
- if (enc_stats_pre &&
- (type == AVMEDIA_TYPE_VIDEO || type == AVMEDIA_TYPE_AUDIO)) {
- const char *format = "{fidx} {sidx} {n} {t}";
- MATCH_PER_STREAM_OPT(enc_stats_pre_fmt, str, format, oc, st);
- ret = enc_stats_init(ost, &ost->enc_stats_pre, 1, enc_stats_pre, format);
- if (ret < 0)
- return ret;
- }
- MATCH_PER_STREAM_OPT(enc_stats_post, str, enc_stats_post, oc, st);
- if (enc_stats_post &&
- (type == AVMEDIA_TYPE_VIDEO || type == AVMEDIA_TYPE_AUDIO)) {
- const char *format = "{fidx} {sidx} {n} {t}";
- MATCH_PER_STREAM_OPT(enc_stats_post_fmt, str, format, oc, st);
- ret = enc_stats_init(ost, &ost->enc_stats_post, 0, enc_stats_post, format);
- if (ret < 0)
- return ret;
- }
- MATCH_PER_STREAM_OPT(mux_stats, str, mux_stats, oc, st);
- if (mux_stats &&
- (type == AVMEDIA_TYPE_VIDEO || type == AVMEDIA_TYPE_AUDIO)) {
- const char *format = "{fidx} {sidx} {n} {t}";
- MATCH_PER_STREAM_OPT(mux_stats_fmt, str, format, oc, st);
- ret = enc_stats_init(ost, &ms->stats, 0, mux_stats, format);
- if (ret < 0)
- return ret;
- }
- MATCH_PER_STREAM_OPT(enc_time_bases, str, enc_time_base, oc, st);
- if (enc_time_base) {
- AVRational q;
- if (!strcmp(enc_time_base, "demux")) {
- q = (AVRational){ ENC_TIME_BASE_DEMUX, 0 };
- } else if (!strcmp(enc_time_base, "filter")) {
- q = (AVRational){ ENC_TIME_BASE_FILTER, 0 };
- } else {
- ret = av_parse_ratio(&q, enc_time_base, INT_MAX, 0, NULL);
- if (ret < 0 || q.den <= 0
- #if !FFMPEG_OPT_ENC_TIME_BASE_NUM
- || q.num < 0
- #endif
- ) {
- av_log(ost, AV_LOG_FATAL, "Invalid time base: %s\n", enc_time_base);
- return ret < 0 ? ret : AVERROR(EINVAL);
- }
- #if FFMPEG_OPT_ENC_TIME_BASE_NUM
- if (q.num < 0)
- av_log(ost, AV_LOG_WARNING, "-enc_time_base -1 is deprecated,"
- " use -enc_timebase demux\n");
- #endif
- }
- ost->enc_timebase = q;
- }
- } else {
- ret = filter_codec_opts(o->g->codec_opts, AV_CODEC_ID_NONE, oc, st,
- NULL, &ost->encoder_opts);
- if (ret < 0)
- return ret;
- }
- if (o->bitexact) {
- ost->bitexact = 1;
- } else if (ost->enc_ctx) {
- ost->bitexact = check_opt_bitexact(ost->enc_ctx, ost->encoder_opts, "flags",
- AV_CODEC_FLAG_BITEXACT);
- }
- MATCH_PER_STREAM_OPT(time_bases, str, time_base, oc, st);
- if (time_base) {
- AVRational q;
- if (av_parse_ratio(&q, time_base, INT_MAX, 0, NULL) < 0 ||
- q.num <= 0 || q.den <= 0) {
- av_log(ost, AV_LOG_FATAL, "Invalid time base: %s\n", time_base);
- return AVERROR(EINVAL);
- }
- st->time_base = q;
- }
- ms->max_frames = INT64_MAX;
- MATCH_PER_STREAM_OPT(max_frames, i64, ms->max_frames, oc, st);
- for (i = 0; i<o->nb_max_frames; i++) {
- char *p = o->max_frames[i].specifier;
- if (!*p && type != AVMEDIA_TYPE_VIDEO) {
- av_log(ost, AV_LOG_WARNING, "Applying unspecific -frames to non video streams, maybe you meant -vframes ?\n");
- break;
- }
- }
- ms->copy_prior_start = -1;
- MATCH_PER_STREAM_OPT(copy_prior_start, i, ms->copy_prior_start, oc ,st);
- MATCH_PER_STREAM_OPT(bitstream_filters, str, bsfs, oc, st);
- if (bsfs && *bsfs) {
- ret = av_bsf_list_parse_str(bsfs, &ms->bsf_ctx);
- if (ret < 0) {
- av_log(ost, AV_LOG_ERROR, "Error parsing bitstream filter sequence '%s': %s\n", bsfs, av_err2str(ret));
- return ret;
- }
- }
- MATCH_PER_STREAM_OPT(codec_tags, str, codec_tag, oc, st);
- if (codec_tag) {
- uint32_t tag = strtol(codec_tag, &next, 0);
- if (*next) {
- uint8_t buf[4] = { 0 };
- memcpy(buf, codec_tag, FFMIN(sizeof(buf), strlen(codec_tag)));
- tag = AV_RL32(buf);
- }
- ost->st->codecpar->codec_tag = tag;
- ost->par_in->codec_tag = tag;
- if (ost->enc_ctx)
- ost->enc_ctx->codec_tag = tag;
- }
- MATCH_PER_STREAM_OPT(qscale, dbl, qscale, oc, st);
- if (ost->enc_ctx && qscale >= 0) {
- ost->enc_ctx->flags |= AV_CODEC_FLAG_QSCALE;
- ost->enc_ctx->global_quality = FF_QP2LAMBDA * qscale;
- }
- ms->max_muxing_queue_size = 128;
- MATCH_PER_STREAM_OPT(max_muxing_queue_size, i, ms->max_muxing_queue_size, oc, st);
- ms->muxing_queue_data_threshold = 50*1024*1024;
- MATCH_PER_STREAM_OPT(muxing_queue_data_threshold, i, ms->muxing_queue_data_threshold, oc, st);
- MATCH_PER_STREAM_OPT(bits_per_raw_sample, i, ost->bits_per_raw_sample,
- oc, st);
- MATCH_PER_STREAM_OPT(fix_sub_duration_heartbeat, i, ost->fix_sub_duration_heartbeat,
- oc, st);
- if (oc->oformat->flags & AVFMT_GLOBALHEADER && ost->enc_ctx)
- ost->enc_ctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
- av_dict_copy(&ost->sws_dict, o->g->sws_dict, 0);
- av_dict_copy(&ost->swr_opts, o->g->swr_opts, 0);
- if (ost->enc_ctx && av_get_exact_bits_per_sample(ost->enc_ctx->codec_id) == 24)
- av_dict_set(&ost->swr_opts, "output_sample_bits", "24", 0);
- ost->last_mux_dts = AV_NOPTS_VALUE;
- MATCH_PER_STREAM_OPT(copy_initial_nonkeyframes, i,
- ms->copy_initial_nonkeyframes, oc, st);
- switch (type) {
- case AVMEDIA_TYPE_VIDEO: ret = new_stream_video (mux, o, ost); break;
- case AVMEDIA_TYPE_AUDIO: ret = new_stream_audio (mux, o, ost); break;
- case AVMEDIA_TYPE_SUBTITLE: ret = new_stream_subtitle (mux, o, ost); break;
- case AVMEDIA_TYPE_ATTACHMENT: ret = new_stream_attachment(mux, o, ost); break;
- }
- if (ret < 0)
- return ret;
- if (type == AVMEDIA_TYPE_VIDEO || type == AVMEDIA_TYPE_AUDIO) {
- ret = ost_get_filters(o, oc, ost, &filters);
- if (ret < 0)
- return ret;
- }
- if (ost->enc &&
- (type == AVMEDIA_TYPE_VIDEO || type == AVMEDIA_TYPE_AUDIO)) {
- if (ofilter) {
- ost->filter = ofilter;
- ret = ofilter_bind_ost(ofilter, ost);
- if (ret < 0)
- return ret;
- } else {
- ret = init_simple_filtergraph(ost->ist, ost, filters);
- if (ret < 0) {
- av_log(ost, AV_LOG_ERROR,
- "Error initializing a simple filtergraph\n");
- return ret;
- }
- }
- } else if (ost->ist) {
- ret = ist_output_add(ost->ist, ost);
- if (ret < 0) {
- av_log(ost, AV_LOG_ERROR,
- "Error binding an input stream\n");
- return ret;
- }
- }
- if (ost->ist && !ost->enc) {
- ret = streamcopy_init(mux, ost);
- if (ret < 0)
- return ret;
- }
- // copy estimated duration as a hint to the muxer
- if (ost->ist && ost->ist->st->duration > 0) {
- ms->stream_duration = ist->st->duration;
- ms->stream_duration_tb = ist->st->time_base;
- }
- if (post)
- *post = ost;
- return 0;
- }
- static int map_auto_video(Muxer *mux, const OptionsContext *o)
- {
- AVFormatContext *oc = mux->fc;
- InputStream *best_ist = NULL;
- int best_score = 0;
- int qcr;
- /* video: highest resolution */
- if (av_guess_codec(oc->oformat, NULL, oc->url, NULL, AVMEDIA_TYPE_VIDEO) == AV_CODEC_ID_NONE)
- return 0;
- qcr = avformat_query_codec(oc->oformat, oc->oformat->video_codec, 0);
- for (int j = 0; j < nb_input_files; j++) {
- InputFile *ifile = input_files[j];
- InputStream *file_best_ist = NULL;
- int file_best_score = 0;
- for (int i = 0; i < ifile->nb_streams; i++) {
- InputStream *ist = ifile->streams[i];
- int score;
- if (ist->user_set_discard == AVDISCARD_ALL ||
- ist->st->codecpar->codec_type != AVMEDIA_TYPE_VIDEO)
- continue;
- score = ist->st->codecpar->width * ist->st->codecpar->height
- + 100000000 * !!(ist->st->event_flags & AVSTREAM_EVENT_FLAG_NEW_PACKETS)
- + 5000000*!!(ist->st->disposition & AV_DISPOSITION_DEFAULT);
- if((qcr!=MKTAG('A', 'P', 'I', 'C')) && (ist->st->disposition & AV_DISPOSITION_ATTACHED_PIC))
- score = 1;
- if (score > file_best_score) {
- if((qcr==MKTAG('A', 'P', 'I', 'C')) && !(ist->st->disposition & AV_DISPOSITION_ATTACHED_PIC))
- continue;
- file_best_score = score;
- file_best_ist = ist;
- }
- }
- if (file_best_ist) {
- if((qcr == MKTAG('A', 'P', 'I', 'C')) ||
- !(file_best_ist->st->disposition & AV_DISPOSITION_ATTACHED_PIC))
- file_best_score -= 5000000*!!(file_best_ist->st->disposition & AV_DISPOSITION_DEFAULT);
- if (file_best_score > best_score) {
- best_score = file_best_score;
- best_ist = file_best_ist;
- }
- }
- }
- if (best_ist)
- return ost_add(mux, o, AVMEDIA_TYPE_VIDEO, best_ist, NULL, NULL);
- return 0;
- }
- static int map_auto_audio(Muxer *mux, const OptionsContext *o)
- {
- AVFormatContext *oc = mux->fc;
- InputStream *best_ist = NULL;
- int best_score = 0;
- /* audio: most channels */
- if (av_guess_codec(oc->oformat, NULL, oc->url, NULL, AVMEDIA_TYPE_AUDIO) == AV_CODEC_ID_NONE)
- return 0;
- for (int j = 0; j < nb_input_files; j++) {
- InputFile *ifile = input_files[j];
- InputStream *file_best_ist = NULL;
- int file_best_score = 0;
- for (int i = 0; i < ifile->nb_streams; i++) {
- InputStream *ist = ifile->streams[i];
- int score;
- if (ist->user_set_discard == AVDISCARD_ALL ||
- ist->st->codecpar->codec_type != AVMEDIA_TYPE_AUDIO)
- continue;
- score = ist->st->codecpar->ch_layout.nb_channels
- + 100000000 * !!(ist->st->event_flags & AVSTREAM_EVENT_FLAG_NEW_PACKETS)
- + 5000000*!!(ist->st->disposition & AV_DISPOSITION_DEFAULT);
- if (score > file_best_score) {
- file_best_score = score;
- file_best_ist = ist;
- }
- }
- if (file_best_ist) {
- file_best_score -= 5000000*!!(file_best_ist->st->disposition & AV_DISPOSITION_DEFAULT);
- if (file_best_score > best_score) {
- best_score = file_best_score;
- best_ist = file_best_ist;
- }
- }
- }
- if (best_ist)
- return ost_add(mux, o, AVMEDIA_TYPE_AUDIO, best_ist, NULL, NULL);
- return 0;
- }
- static int map_auto_subtitle(Muxer *mux, const OptionsContext *o)
- {
- AVFormatContext *oc = mux->fc;
- char *subtitle_codec_name = NULL;
- /* subtitles: pick first */
- MATCH_PER_TYPE_OPT(codec_names, str, subtitle_codec_name, oc, "s");
- if (!avcodec_find_encoder(oc->oformat->subtitle_codec) && !subtitle_codec_name)
- return 0;
- for (InputStream *ist = ist_iter(NULL); ist; ist = ist_iter(ist))
- if (ist->st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) {
- AVCodecDescriptor const *input_descriptor =
- avcodec_descriptor_get(ist->st->codecpar->codec_id);
- AVCodecDescriptor const *output_descriptor = NULL;
- AVCodec const *output_codec =
- avcodec_find_encoder(oc->oformat->subtitle_codec);
- int input_props = 0, output_props = 0;
- if (ist->user_set_discard == AVDISCARD_ALL)
- continue;
- if (output_codec)
- output_descriptor = avcodec_descriptor_get(output_codec->id);
- if (input_descriptor)
- input_props = input_descriptor->props & (AV_CODEC_PROP_TEXT_SUB | AV_CODEC_PROP_BITMAP_SUB);
- if (output_descriptor)
- output_props = output_descriptor->props & (AV_CODEC_PROP_TEXT_SUB | AV_CODEC_PROP_BITMAP_SUB);
- if (subtitle_codec_name ||
- input_props & output_props ||
- // Map dvb teletext which has neither property to any output subtitle encoder
- input_descriptor && output_descriptor &&
- (!input_descriptor->props ||
- !output_descriptor->props)) {
- return ost_add(mux, o, AVMEDIA_TYPE_SUBTITLE, ist, NULL, NULL);
- }
- }
- return 0;
- }
- static int map_auto_data(Muxer *mux, const OptionsContext *o)
- {
- AVFormatContext *oc = mux->fc;
- /* Data only if codec id match */
- enum AVCodecID codec_id = av_guess_codec(oc->oformat, NULL, oc->url, NULL, AVMEDIA_TYPE_DATA);
- if (codec_id == AV_CODEC_ID_NONE)
- return 0;
- for (InputStream *ist = ist_iter(NULL); ist; ist = ist_iter(ist)) {
- if (ist->user_set_discard == AVDISCARD_ALL)
- continue;
- if (ist->st->codecpar->codec_type == AVMEDIA_TYPE_DATA &&
- ist->st->codecpar->codec_id == codec_id) {
- int ret = ost_add(mux, o, AVMEDIA_TYPE_DATA, ist, NULL, NULL);
- if (ret < 0)
- return ret;
- }
- }
- return 0;
- }
- static int map_manual(Muxer *mux, const OptionsContext *o, const StreamMap *map)
- {
- InputStream *ist;
- int ret;
- if (map->disabled)
- return 0;
- 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++) {
- const char *linklabel = fg->outputs[k]->linklabel;
- if (linklabel && !strcmp(linklabel, map->linklabel)) {
- ofilter = fg->outputs[k];
- goto loop_end;
- }
- }
- }
- loop_end:
- if (!ofilter) {
- av_log(mux, AV_LOG_FATAL, "Output with label '%s' does not exist "
- "in any defined filter graph, or was already used elsewhere.\n", map->linklabel);
- return AVERROR(EINVAL);
- }
- av_log(mux, AV_LOG_VERBOSE, "Creating output stream from an explicitly "
- "mapped complex filtergraph %d, output [%s]\n", fg->index, map->linklabel);
- ret = ost_add(mux, o, ofilter->type, NULL, ofilter, NULL);
- if (ret < 0)
- return ret;
- } else {
- ist = input_files[map->file_index]->streams[map->stream_index];
- if (ist->user_set_discard == AVDISCARD_ALL) {
- av_log(mux, AV_LOG_FATAL, "Stream #%d:%d is disabled and cannot be mapped.\n",
- map->file_index, map->stream_index);
- return AVERROR(EINVAL);
- }
- if(o->subtitle_disable && ist->st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE)
- return 0;
- if(o-> audio_disable && ist->st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO)
- return 0;
- if(o-> video_disable && ist->st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
- return 0;
- if(o-> data_disable && ist->st->codecpar->codec_type == AVMEDIA_TYPE_DATA)
- return 0;
- if (ist->st->codecpar->codec_type == AVMEDIA_TYPE_UNKNOWN &&
- !copy_unknown_streams) {
- av_log(mux, ignore_unknown_streams ? AV_LOG_WARNING : AV_LOG_FATAL,
- "Cannot map stream #%d:%d - unsupported type.\n",
- map->file_index, map->stream_index);
- if (!ignore_unknown_streams) {
- av_log(mux, AV_LOG_FATAL,
- "If you want unsupported types ignored instead "
- "of failing, please use the -ignore_unknown option\n"
- "If you want them copied, please use -copy_unknown\n");
- return AVERROR(EINVAL);
- }
- return 0;
- }
- ret = ost_add(mux, o, ist->st->codecpar->codec_type, ist, NULL, NULL);
- if (ret < 0)
- return ret;
- }
- return 0;
- }
- static int of_add_attachments(Muxer *mux, const OptionsContext *o)
- {
- OutputStream *ost;
- int err;
- for (int i = 0; i < o->nb_attachments; i++) {
- AVIOContext *pb;
- uint8_t *attachment;
- const char *p;
- int64_t len;
- if ((err = avio_open2(&pb, o->attachments[i], AVIO_FLAG_READ, &int_cb, NULL)) < 0) {
- av_log(mux, AV_LOG_FATAL, "Could not open attachment file %s.\n",
- o->attachments[i]);
- return err;
- }
- if ((len = avio_size(pb)) <= 0) {
- av_log(mux, AV_LOG_FATAL, "Could not get size of the attachment %s.\n",
- o->attachments[i]);
- err = len ? len : AVERROR_INVALIDDATA;
- goto read_fail;
- }
- if (len > INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
- av_log(mux, AV_LOG_FATAL, "Attachment %s too large.\n",
- o->attachments[i]);
- err = AVERROR(ERANGE);
- goto read_fail;
- }
- attachment = av_malloc(len + AV_INPUT_BUFFER_PADDING_SIZE);
- if (!attachment) {
- err = AVERROR(ENOMEM);
- goto read_fail;
- }
- err = avio_read(pb, attachment, len);
- if (err < 0)
- av_log(mux, AV_LOG_FATAL, "Error reading attachment file %s: %s\n",
- o->attachments[i], av_err2str(err));
- else if (err != len) {
- av_log(mux, AV_LOG_FATAL, "Could not read all %"PRId64" bytes for "
- "attachment file %s\n", len, o->attachments[i]);
- err = AVERROR(EIO);
- }
- read_fail:
- avio_closep(&pb);
- if (err < 0)
- return err;
- memset(attachment + len, 0, AV_INPUT_BUFFER_PADDING_SIZE);
- av_log(mux, AV_LOG_VERBOSE, "Creating attachment stream from file %s\n",
- o->attachments[i]);
- err = ost_add(mux, o, AVMEDIA_TYPE_ATTACHMENT, NULL, NULL, &ost);
- if (err < 0) {
- av_freep(&attachment);
- return err;
- }
- ost->attachment_filename = o->attachments[i];
- ost->par_in->extradata = attachment;
- ost->par_in->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);
- }
- return 0;
- }
- static int create_streams(Muxer *mux, const OptionsContext *o)
- {
- static int (* const map_func[])(Muxer *mux, const OptionsContext *o) = {
- [AVMEDIA_TYPE_VIDEO] = map_auto_video,
- [AVMEDIA_TYPE_AUDIO] = map_auto_audio,
- [AVMEDIA_TYPE_SUBTITLE] = map_auto_subtitle,
- [AVMEDIA_TYPE_DATA] = map_auto_data,
- };
- AVFormatContext *oc = mux->fc;
- int auto_disable =
- o->video_disable * (1 << AVMEDIA_TYPE_VIDEO) |
- o->audio_disable * (1 << AVMEDIA_TYPE_AUDIO) |
- o->subtitle_disable * (1 << AVMEDIA_TYPE_SUBTITLE) |
- o->data_disable * (1 << AVMEDIA_TYPE_DATA);
- int ret;
- /* create streams for all unlabeled output pads */
- for (int i = 0; i < nb_filtergraphs; i++) {
- FilterGraph *fg = filtergraphs[i];
- for (int j = 0; j < fg->nb_outputs; j++) {
- OutputFilter *ofilter = fg->outputs[j];
- if (ofilter->linklabel || ofilter->ost)
- continue;
- auto_disable |= 1 << ofilter->type;
- av_log(mux, AV_LOG_VERBOSE, "Creating output stream from unlabeled "
- "output of complex filtergraph %d.", fg->index);
- if (!o->nb_stream_maps)
- av_log(mux, AV_LOG_VERBOSE, " This overrides automatic %s mapping.",
- av_get_media_type_string(ofilter->type));
- av_log(mux, AV_LOG_VERBOSE, "\n");
- ret = ost_add(mux, o, ofilter->type, NULL, ofilter, NULL);
- if (ret < 0)
- return ret;
- }
- }
- if (!o->nb_stream_maps) {
- av_log(mux, AV_LOG_VERBOSE, "No explicit maps, mapping streams automatically...\n");
- /* pick the "best" stream of each type */
- for (int i = 0; i < FF_ARRAY_ELEMS(map_func); i++) {
- if (!map_func[i] || auto_disable & (1 << i))
- continue;
- ret = map_func[i](mux, o);
- if (ret < 0)
- return ret;
- }
- } else {
- av_log(mux, AV_LOG_VERBOSE, "Adding streams from explicit maps...\n");
- for (int i = 0; i < o->nb_stream_maps; i++) {
- ret = map_manual(mux, o, &o->stream_maps[i]);
- if (ret < 0)
- return ret;
- }
- }
- ret = of_add_attachments(mux, o);
- if (ret < 0)
- return ret;
- if (!oc->nb_streams && !(oc->oformat->flags & AVFMT_NOSTREAMS)) {
- av_dump_format(oc, nb_output_files - 1, oc->url, 1);
- av_log(mux, AV_LOG_ERROR, "Output file does not contain any stream\n");
- return AVERROR(EINVAL);
- }
- return 0;
- }
- static int setup_sync_queues(Muxer *mux, AVFormatContext *oc, int64_t buf_size_us)
- {
- OutputFile *of = &mux->of;
- int nb_av_enc = 0, nb_audio_fs = 0, nb_interleaved = 0;
- int limit_frames = 0, limit_frames_av_enc = 0;
- #define IS_AV_ENC(ost, type) \
- (ost->enc_ctx && (type == AVMEDIA_TYPE_VIDEO || type == AVMEDIA_TYPE_AUDIO))
- #define IS_INTERLEAVED(type) (type != AVMEDIA_TYPE_ATTACHMENT)
- for (int i = 0; i < oc->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- MuxStream *ms = ms_from_ost(ost);
- enum AVMediaType type = ost->type;
- ost->sq_idx_encode = -1;
- ost->sq_idx_mux = -1;
- nb_interleaved += IS_INTERLEAVED(type);
- nb_av_enc += IS_AV_ENC(ost, type);
- nb_audio_fs += (ost->enc_ctx && type == AVMEDIA_TYPE_AUDIO &&
- !(ost->enc_ctx->codec->capabilities & AV_CODEC_CAP_VARIABLE_FRAME_SIZE));
- limit_frames |= ms->max_frames < INT64_MAX;
- limit_frames_av_enc |= (ms->max_frames < INT64_MAX) && IS_AV_ENC(ost, type);
- }
- if (!((nb_interleaved > 1 && of->shortest) ||
- (nb_interleaved > 0 && limit_frames) ||
- nb_audio_fs))
- return 0;
- /* we use a sync queue before encoding when:
- * - 'shortest' is in effect and we have two or more encoded audio/video
- * streams
- * - at least one encoded audio/video stream is frame-limited, since
- * that has similar semantics to 'shortest'
- * - at least one audio encoder requires constant frame sizes
- */
- if ((of->shortest && nb_av_enc > 1) || limit_frames_av_enc || nb_audio_fs) {
- of->sq_encode = sq_alloc(SYNC_QUEUE_FRAMES, buf_size_us, mux);
- if (!of->sq_encode)
- return AVERROR(ENOMEM);
- for (int i = 0; i < oc->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- MuxStream *ms = ms_from_ost(ost);
- enum AVMediaType type = ost->type;
- if (!IS_AV_ENC(ost, type))
- continue;
- ost->sq_idx_encode = sq_add_stream(of->sq_encode,
- of->shortest || ms->max_frames < INT64_MAX);
- if (ost->sq_idx_encode < 0)
- return ost->sq_idx_encode;
- if (ms->max_frames != INT64_MAX)
- sq_limit_frames(of->sq_encode, ost->sq_idx_encode, ms->max_frames);
- }
- }
- /* if there are any additional interleaved streams, then ALL the streams
- * are also synchronized before sending them to the muxer */
- if (nb_interleaved > nb_av_enc) {
- mux->sq_mux = sq_alloc(SYNC_QUEUE_PACKETS, buf_size_us, mux);
- if (!mux->sq_mux)
- return AVERROR(ENOMEM);
- mux->sq_pkt = av_packet_alloc();
- if (!mux->sq_pkt)
- return AVERROR(ENOMEM);
- for (int i = 0; i < oc->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- MuxStream *ms = ms_from_ost(ost);
- enum AVMediaType type = ost->type;
- if (!IS_INTERLEAVED(type))
- continue;
- ost->sq_idx_mux = sq_add_stream(mux->sq_mux,
- of->shortest || ms->max_frames < INT64_MAX);
- if (ost->sq_idx_mux < 0)
- return ost->sq_idx_mux;
- if (ms->max_frames != INT64_MAX)
- sq_limit_frames(mux->sq_mux, ost->sq_idx_mux, ms->max_frames);
- }
- }
- #undef IS_AV_ENC
- #undef IS_INTERLEAVED
- return 0;
- }
- static int of_add_programs(Muxer *mux, const OptionsContext *o)
- {
- AVFormatContext *oc = mux->fc;
- /* process manually set programs */
- for (int i = 0; i < o->nb_program; i++) {
- AVDictionary *dict = NULL;
- const AVDictionaryEntry *e;
- AVProgram *program;
- int ret, progid = i + 1;
- ret = av_dict_parse_string(&dict, o->program[i].u.str, "=", ":",
- AV_DICT_MULTIKEY);
- if (ret < 0) {
- av_log(mux, AV_LOG_ERROR, "Error parsing program specification %s\n",
- o->program[i].u.str);
- return ret;
- }
- e = av_dict_get(dict, "program_num", NULL, 0);
- if (e) {
- progid = strtol(e->value, NULL, 0);
- av_dict_set(&dict, e->key, NULL, 0);
- }
- program = av_new_program(oc, progid);
- if (!program) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- e = av_dict_get(dict, "title", NULL, 0);
- if (e) {
- av_dict_set(&program->metadata, e->key, e->value, 0);
- av_dict_set(&dict, e->key, NULL, 0);
- }
- e = NULL;
- while (e = av_dict_get(dict, "st", e, 0)) {
- int st_num = strtol(e->value, NULL, 0);
- av_program_add_stream_index(oc, progid, st_num);
- }
- // make sure that nothing but "st" entries are left in the dict
- e = NULL;
- while (e = av_dict_iterate(dict, e)) {
- if (!strcmp(e->key, "st"))
- continue;
- av_log(mux, AV_LOG_FATAL, "Unknown program key %s.\n", e->key);
- ret = AVERROR(EINVAL);
- goto fail;
- }
- fail:
- av_dict_free(&dict);
- if (ret < 0)
- return ret;
- }
- return 0;
- }
- /**
- * Parse a metadata specifier passed as 'arg' parameter.
- * @param arg metadata string to parse
- * @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 int parse_meta_type(void *logctx, const char *arg,
- char *type, int *index, const char **stream_spec)
- {
- if (*arg) {
- *type = *arg;
- switch (*arg) {
- case 'g':
- break;
- case 's':
- if (*(++arg) && *arg != ':') {
- av_log(logctx, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", arg);
- return AVERROR(EINVAL);
- }
- *stream_spec = *arg == ':' ? arg + 1 : "";
- break;
- case 'c':
- case 'p':
- if (*(++arg) == ':')
- *index = strtol(++arg, NULL, 0);
- break;
- default:
- av_log(logctx, AV_LOG_FATAL, "Invalid metadata type %c.\n", *arg);
- return AVERROR(EINVAL);
- }
- } else
- *type = 'g';
- return 0;
- }
- static int of_add_metadata(OutputFile *of, AVFormatContext *oc,
- const OptionsContext *o)
- {
- for (int i = 0; i < o->nb_metadata; i++) {
- AVDictionary **m;
- char type, *val;
- const char *stream_spec;
- int index = 0, ret = 0;
- val = strchr(o->metadata[i].u.str, '=');
- if (!val) {
- av_log(of, AV_LOG_FATAL, "No '=' character in metadata string %s.\n",
- o->metadata[i].u.str);
- return AVERROR(EINVAL);
- }
- *val++ = 0;
- ret = parse_meta_type(of, o->metadata[i].specifier, &type, &index, &stream_spec);
- if (ret < 0)
- return ret;
- if (type == 's') {
- for (int j = 0; j < oc->nb_streams; j++) {
- OutputStream *ost = of->streams[j];
- if ((ret = check_stream_specifier(oc, oc->streams[j], stream_spec)) > 0) {
- #if FFMPEG_ROTATION_METADATA
- if (!strcmp(o->metadata[i].u.str, "rotate")) {
- char *tail;
- double theta = av_strtod(val, &tail);
- if (!*tail) {
- ost->rotate_overridden = 1;
- ost->rotate_override_value = theta;
- }
- av_log(ost, AV_LOG_WARNING,
- "Conversion of a 'rotate' metadata key to a "
- "proper display matrix rotation is deprecated. "
- "See -display_rotation for setting rotation "
- "instead.");
- } else {
- #endif
- av_dict_set(&oc->streams[j]->metadata, o->metadata[i].u.str, *val ? val : NULL, 0);
- #if FFMPEG_ROTATION_METADATA
- }
- #endif
- } else if (ret < 0)
- return ret;
- }
- } else {
- switch (type) {
- case 'g':
- m = &oc->metadata;
- break;
- case 'c':
- if (index < 0 || index >= oc->nb_chapters) {
- av_log(of, AV_LOG_FATAL, "Invalid chapter index %d in metadata specifier.\n", index);
- return AVERROR(EINVAL);
- }
- m = &oc->chapters[index]->metadata;
- break;
- case 'p':
- if (index < 0 || index >= oc->nb_programs) {
- av_log(of, AV_LOG_FATAL, "Invalid program index %d in metadata specifier.\n", index);
- return AVERROR(EINVAL);
- }
- m = &oc->programs[index]->metadata;
- break;
- default:
- av_log(of, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", o->metadata[i].specifier);
- return AVERROR(EINVAL);
- }
- av_dict_set(m, o->metadata[i].u.str, *val ? val : NULL, 0);
- }
- }
- return 0;
- }
- static int copy_chapters(InputFile *ifile, OutputFile *ofile, AVFormatContext *os,
- int copy_metadata)
- {
- AVFormatContext *is = ifile->ctx;
- AVChapter **tmp;
- int i;
- tmp = av_realloc_f(os->chapters, is->nb_chapters + os->nb_chapters, sizeof(*os->chapters));
- if (!tmp)
- return AVERROR(ENOMEM);
- os->chapters = tmp;
- for (i = 0; i < is->nb_chapters; i++) {
- AVChapter *in_ch = is->chapters[i], *out_ch;
- int64_t start_time = (ofile->start_time == AV_NOPTS_VALUE) ? 0 : ofile->start_time;
- int64_t ts_off = av_rescale_q(start_time - ifile->ts_offset,
- AV_TIME_BASE_Q, in_ch->time_base);
- int64_t rt = (ofile->recording_time == INT64_MAX) ? INT64_MAX :
- av_rescale_q(ofile->recording_time, AV_TIME_BASE_Q, in_ch->time_base);
- 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);
- if (copy_metadata)
- av_dict_copy(&out_ch->metadata, in_ch->metadata, 0);
- os->chapters[os->nb_chapters++] = out_ch;
- }
- return 0;
- }
- static int copy_metadata(Muxer *mux, AVFormatContext *ic,
- const char *outspec, const char *inspec,
- int *metadata_global_manual, int *metadata_streams_manual,
- int *metadata_chapters_manual)
- {
- AVFormatContext *oc = mux->fc;
- AVDictionary **meta_in = NULL;
- AVDictionary **meta_out = NULL;
- int i, ret = 0;
- char type_in, type_out;
- const char *istream_spec = NULL, *ostream_spec = NULL;
- int idx_in = 0, idx_out = 0;
- ret = parse_meta_type(mux, inspec, &type_in, &idx_in, &istream_spec);
- if (ret >= 0)
- ret = parse_meta_type(mux, outspec, &type_out, &idx_out, &ostream_spec);
- if (ret < 0)
- return ret;
- if (type_in == 'g' || type_out == 'g' || (!*outspec && !ic))
- *metadata_global_manual = 1;
- if (type_in == 's' || type_out == 's' || (!*outspec && !ic))
- *metadata_streams_manual = 1;
- if (type_in == 'c' || type_out == 'c' || (!*outspec && !ic))
- *metadata_chapters_manual = 1;
- /* ic is NULL when just disabling automatic mappings */
- if (!ic)
- return 0;
- #define METADATA_CHECK_INDEX(index, nb_elems, desc)\
- if ((index) < 0 || (index) >= (nb_elems)) {\
- av_log(mux, AV_LOG_FATAL, "Invalid %s index %d while processing metadata maps.\n",\
- (desc), (index));\
- return AVERROR(EINVAL);\
- }
- #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;\
- case 's':\
- break; /* handled separately below */ \
- default: av_assert0(0);\
- }\
- 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)
- return ret;
- }
- if (!meta_in) {
- av_log(mux, AV_LOG_FATAL, "Stream specifier %s does not match any streams.\n", istream_spec);
- return AVERROR(EINVAL);
- }
- }
- 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)
- return ret;
- }
- } else
- av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
- return 0;
- }
- static int copy_meta(Muxer *mux, const OptionsContext *o)
- {
- OutputFile *of = &mux->of;
- AVFormatContext *oc = mux->fc;
- int chapters_input_file = o->chapters_input_file;
- int metadata_global_manual = 0;
- int metadata_streams_manual = 0;
- int metadata_chapters_manual = 0;
- int ret;
- /* copy metadata */
- for (int 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 >= nb_input_files) {
- av_log(mux, AV_LOG_FATAL, "Invalid input file index %d while "
- "processing metadata maps\n", in_file_index);
- return AVERROR(EINVAL);
- }
- ret = copy_metadata(mux,
- in_file_index >= 0 ? input_files[in_file_index]->ctx : NULL,
- o->metadata_map[i].specifier, *p ? p + 1 : p,
- &metadata_global_manual, &metadata_streams_manual,
- &metadata_chapters_manual);
- if (ret < 0)
- return ret;
- }
- /* copy chapters */
- if (chapters_input_file >= nb_input_files) {
- if (chapters_input_file == INT_MAX) {
- /* copy chapters from the first input file that has them*/
- chapters_input_file = -1;
- for (int i = 0; i < nb_input_files; i++)
- if (input_files[i]->ctx->nb_chapters) {
- chapters_input_file = i;
- break;
- }
- } else {
- av_log(mux, AV_LOG_FATAL, "Invalid input file index %d in chapter mapping.\n",
- chapters_input_file);
- return AVERROR(EINVAL);
- }
- }
- if (chapters_input_file >= 0)
- copy_chapters(input_files[chapters_input_file], of, oc,
- !metadata_chapters_manual);
- /* copy global metadata by default */
- if (!metadata_global_manual && nb_input_files){
- av_dict_copy(&oc->metadata, input_files[0]->ctx->metadata,
- AV_DICT_DONT_OVERWRITE);
- if (of->recording_time != INT64_MAX)
- av_dict_set(&oc->metadata, "duration", NULL, 0);
- av_dict_set(&oc->metadata, "creation_time", NULL, 0);
- av_dict_set(&oc->metadata, "company_name", NULL, 0);
- av_dict_set(&oc->metadata, "product_name", NULL, 0);
- av_dict_set(&oc->metadata, "product_version", NULL, 0);
- }
- if (!metadata_streams_manual)
- for (int i = 0; i < of->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- if (!ost->ist) /* this is true e.g. for attached files */
- continue;
- av_dict_copy(&ost->st->metadata, ost->ist->st->metadata, AV_DICT_DONT_OVERWRITE);
- if (ost->enc_ctx) {
- av_dict_set(&ost->st->metadata, "encoder", NULL, 0);
- }
- }
- return 0;
- }
- static int set_dispositions(Muxer *mux, const OptionsContext *o)
- {
- OutputFile *of = &mux->of;
- AVFormatContext *ctx = mux->fc;
- // indexed by type+1, because AVMEDIA_TYPE_UNKNOWN=-1
- int nb_streams[AVMEDIA_TYPE_NB + 1] = { 0 };
- int have_default[AVMEDIA_TYPE_NB + 1] = { 0 };
- int have_manual = 0;
- int ret = 0;
- const char **dispositions;
- dispositions = av_calloc(ctx->nb_streams, sizeof(*dispositions));
- if (!dispositions)
- return AVERROR(ENOMEM);
- // first, copy the input dispositions
- for (int i = 0; i < ctx->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- nb_streams[ost->type + 1]++;
- MATCH_PER_STREAM_OPT(disposition, str, dispositions[i], ctx, ost->st);
- have_manual |= !!dispositions[i];
- if (ost->ist) {
- ost->st->disposition = ost->ist->st->disposition;
- if (ost->st->disposition & AV_DISPOSITION_DEFAULT)
- have_default[ost->type + 1] = 1;
- }
- }
- if (have_manual) {
- // process manually set dispositions - they override the above copy
- for (int i = 0; i < ctx->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- const char *disp = dispositions[i];
- if (!disp)
- continue;
- ret = av_opt_set(ost->st, "disposition", disp, 0);
- if (ret < 0)
- goto finish;
- }
- } else {
- // For each media type with more than one stream, find a suitable stream to
- // mark as default, unless one is already marked default.
- // "Suitable" means the first of that type, skipping attached pictures.
- for (int i = 0; i < ctx->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- enum AVMediaType type = ost->type;
- if (nb_streams[type + 1] < 2 || have_default[type + 1] ||
- ost->st->disposition & AV_DISPOSITION_ATTACHED_PIC)
- continue;
- ost->st->disposition |= AV_DISPOSITION_DEFAULT;
- have_default[type + 1] = 1;
- }
- }
- finish:
- av_freep(&dispositions);
- return ret;
- }
- const char *const forced_keyframes_const_names[] = {
- "n",
- "n_forced",
- "prev_forced_n",
- "prev_forced_t",
- "t",
- NULL
- };
- static int compare_int64(const void *a, const void *b)
- {
- return FFDIFFSIGN(*(const int64_t *)a, *(const int64_t *)b);
- }
- static int parse_forced_key_frames(void *log, KeyframeForceCtx *kf,
- const Muxer *mux, const char *spec)
- {
- const char *p;
- int n = 1, i, ret, size, index = 0;
- int64_t t, *pts;
- for (p = spec; *p; p++)
- if (*p == ',')
- n++;
- size = n;
- pts = av_malloc_array(size, sizeof(*pts));
- if (!pts)
- return AVERROR(ENOMEM);
- p = spec;
- for (i = 0; i < n; i++) {
- char *next = strchr(p, ',');
- if (next)
- *next++ = 0;
- if (strstr(p, "chapters") == p) {
- AVChapter * const *ch = mux->fc->chapters;
- unsigned int nb_ch = mux->fc->nb_chapters;
- int j;
- if (nb_ch > INT_MAX - size ||
- !(pts = av_realloc_f(pts, size += nb_ch - 1,
- sizeof(*pts))))
- return AVERROR(ENOMEM);
- if (p[8]) {
- ret = av_parse_time(&t, p + 8, 1);
- if (ret < 0) {
- av_log(log, AV_LOG_ERROR,
- "Invalid chapter time offset: %s\n", p + 8);
- goto fail;
- }
- } else
- t = 0;
- for (j = 0; j < nb_ch; j++) {
- const AVChapter *c = ch[j];
- av_assert1(index < size);
- pts[index++] = av_rescale_q(c->start, c->time_base,
- AV_TIME_BASE_Q) + t;
- }
- } else {
- av_assert1(index < size);
- ret = av_parse_time(&t, p, 1);
- if (ret < 0) {
- av_log(log, AV_LOG_ERROR, "Invalid keyframe time: %s\n", p);
- goto fail;
- }
- pts[index++] = t;
- }
- p = next;
- }
- av_assert0(index == size);
- qsort(pts, size, sizeof(*pts), compare_int64);
- kf->nb_pts = size;
- kf->pts = pts;
- return 0;
- fail:
- av_freep(&pts);
- return ret;
- }
- static int process_forced_keyframes(Muxer *mux, const OptionsContext *o)
- {
- for (int i = 0; i < mux->of.nb_streams; i++) {
- OutputStream *ost = mux->of.streams[i];
- const char *forced_keyframes = NULL;
- MATCH_PER_STREAM_OPT(forced_key_frames, str, forced_keyframes, mux->fc, ost->st);
- if (!(ost->type == AVMEDIA_TYPE_VIDEO &&
- ost->enc_ctx && forced_keyframes))
- continue;
- if (!strncmp(forced_keyframes, "expr:", 5)) {
- int ret = av_expr_parse(&ost->kf.pexpr, forced_keyframes + 5,
- forced_keyframes_const_names, NULL, NULL, NULL, NULL, 0, NULL);
- if (ret < 0) {
- av_log(ost, AV_LOG_ERROR,
- "Invalid force_key_frames expression '%s'\n", forced_keyframes + 5);
- return ret;
- }
- ost->kf.expr_const_values[FKF_N] = 0;
- ost->kf.expr_const_values[FKF_N_FORCED] = 0;
- ost->kf.expr_const_values[FKF_PREV_FORCED_N] = NAN;
- ost->kf.expr_const_values[FKF_PREV_FORCED_T] = NAN;
- // Don't parse the 'forced_keyframes' in case of 'keep-source-keyframes',
- // parse it only for static kf timings
- } else if (!strcmp(forced_keyframes, "source")) {
- ost->kf.type = KF_FORCE_SOURCE;
- #if FFMPEG_OPT_FORCE_KF_SOURCE_NO_DROP
- } else if (!strcmp(forced_keyframes, "source_no_drop")) {
- av_log(ost, AV_LOG_WARNING, "The 'source_no_drop' value for "
- "-force_key_frames is deprecated, use just 'source'\n");
- ost->kf.type = KF_FORCE_SOURCE;
- #endif
- } else {
- int ret = parse_forced_key_frames(ost, &ost->kf, mux, forced_keyframes);
- if (ret < 0)
- return ret;
- }
- }
- return 0;
- }
- static int validate_enc_avopt(Muxer *mux, const AVDictionary *codec_avopt)
- {
- const AVClass *class = avcodec_get_class();
- const AVClass *fclass = avformat_get_class();
- const OutputFile *of = &mux->of;
- AVDictionary *unused_opts;
- const AVDictionaryEntry *e;
- unused_opts = strip_specifiers(codec_avopt);
- for (int i = 0; i < of->nb_streams; i++) {
- e = NULL;
- while ((e = av_dict_iterate(of->streams[i]->encoder_opts, e)))
- av_dict_set(&unused_opts, e->key, NULL, 0);
- }
- e = NULL;
- while ((e = av_dict_iterate(unused_opts, e))) {
- const AVOption *option = av_opt_find(&class, e->key, NULL, 0,
- AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ);
- const AVOption *foption = av_opt_find(&fclass, e->key, NULL, 0,
- AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ);
- if (!option || foption)
- continue;
- if (!(option->flags & AV_OPT_FLAG_ENCODING_PARAM)) {
- av_log(mux, AV_LOG_ERROR, "Codec AVOption %s (%s) is not an "
- "encoding option.\n", e->key, option->help ? option->help : "");
- return AVERROR(EINVAL);
- }
- // gop_timecode is injected by generic code but not always used
- if (!strcmp(e->key, "gop_timecode"))
- continue;
- av_log(mux, AV_LOG_WARNING, "Codec AVOption %s (%s) has not been used "
- "for any stream. The most likely reason is either wrong type "
- "(e.g. a video option with no video streams) or that it is a "
- "private option of some encoder which was not actually used for "
- "any stream.\n", e->key, option->help ? option->help : "");
- }
- av_dict_free(&unused_opts);
- return 0;
- }
- static int init_output_stream_nofilter(OutputStream *ost)
- {
- int ret = 0;
- if (ost->enc_ctx) {
- ret = enc_open(ost, NULL);
- if (ret < 0)
- return ret;
- } else {
- ret = of_stream_init(output_files[ost->file_index], ost);
- if (ret < 0)
- return ret;
- }
- return ret;
- }
- static const char *output_file_item_name(void *obj)
- {
- const Muxer *mux = obj;
- return mux->log_name;
- }
- static const AVClass output_file_class = {
- .class_name = "OutputFile",
- .version = LIBAVUTIL_VERSION_INT,
- .item_name = output_file_item_name,
- .category = AV_CLASS_CATEGORY_MUXER,
- };
- static Muxer *mux_alloc(void)
- {
- Muxer *mux = allocate_array_elem(&output_files, sizeof(*mux), &nb_output_files);
- if (!mux)
- return NULL;
- mux->of.class = &output_file_class;
- mux->of.index = nb_output_files - 1;
- snprintf(mux->log_name, sizeof(mux->log_name), "out#%d", mux->of.index);
- return mux;
- }
- int of_open(const OptionsContext *o, const char *filename)
- {
- Muxer *mux;
- AVFormatContext *oc;
- int err;
- OutputFile *of;
- int64_t recording_time = o->recording_time;
- int64_t stop_time = o->stop_time;
- mux = mux_alloc();
- if (!mux)
- return AVERROR(ENOMEM);
- of = &mux->of;
- if (stop_time != INT64_MAX && recording_time != INT64_MAX) {
- stop_time = INT64_MAX;
- av_log(mux, AV_LOG_WARNING, "-t and -to cannot be used together; using -t.\n");
- }
- if (stop_time != INT64_MAX && recording_time == INT64_MAX) {
- int64_t start_time = o->start_time == AV_NOPTS_VALUE ? 0 : o->start_time;
- if (stop_time <= start_time) {
- av_log(mux, AV_LOG_ERROR, "-to value smaller than -ss; aborting.\n");
- return AVERROR(EINVAL);
- } else {
- recording_time = stop_time - start_time;
- }
- }
- of->recording_time = recording_time;
- of->start_time = o->start_time;
- of->shortest = o->shortest;
- mux->thread_queue_size = o->thread_queue_size > 0 ? o->thread_queue_size : 8;
- mux->limit_filesize = o->limit_filesize;
- av_dict_copy(&mux->opts, o->g->format_opts, 0);
- if (!strcmp(filename, "-"))
- filename = "pipe:";
- err = avformat_alloc_output_context2(&oc, NULL, o->format, filename);
- if (!oc) {
- av_log(mux, AV_LOG_FATAL, "Error initializing the muxer for %s: %s\n",
- filename, av_err2str(err));
- return err;
- }
- mux->fc = oc;
- av_strlcat(mux->log_name, "/", sizeof(mux->log_name));
- av_strlcat(mux->log_name, oc->oformat->name, sizeof(mux->log_name));
- if (strcmp(oc->oformat->name, "rtp"))
- want_sdp = 0;
- of->format = oc->oformat;
- if (recording_time != INT64_MAX)
- oc->duration = recording_time;
- oc->interrupt_callback = int_cb;
- if (o->bitexact) {
- oc->flags |= AVFMT_FLAG_BITEXACT;
- of->bitexact = 1;
- } else {
- of->bitexact = check_opt_bitexact(oc, mux->opts, "fflags",
- AVFMT_FLAG_BITEXACT);
- }
- /* create all output streams for this file */
- err = create_streams(mux, o);
- if (err < 0)
- return err;
- /* check if all codec options have been used */
- err = validate_enc_avopt(mux, o->g->codec_opts);
- if (err < 0)
- return err;
- /* check filename in case of an image number is expected */
- if (oc->oformat->flags & AVFMT_NEEDNUMBER && !av_filename_number_test(oc->url)) {
- av_log(mux, AV_LOG_FATAL,
- "Output filename '%s' does not contain a numeric pattern like "
- "'%%d', which is required by output format '%s'.\n",
- oc->url, oc->oformat->name);
- return AVERROR(EINVAL);
- }
- if (!(oc->oformat->flags & AVFMT_NOFILE)) {
- /* test if it already exists to avoid losing precious files */
- err = assert_file_overwrite(filename);
- if (err < 0)
- return err;
- /* open the file */
- if ((err = avio_open2(&oc->pb, filename, AVIO_FLAG_WRITE,
- &oc->interrupt_callback,
- &mux->opts)) < 0) {
- av_log(mux, AV_LOG_FATAL, "Error opening output %s: %s\n",
- filename, av_err2str(err));
- return err;
- }
- } else if (strcmp(oc->oformat->name, "image2")==0 && !av_filename_number_test(filename)) {
- err = assert_file_overwrite(filename);
- if (err < 0)
- return err;
- }
- if (o->mux_preload) {
- av_dict_set_int(&mux->opts, "preload", o->mux_preload*AV_TIME_BASE, 0);
- }
- oc->max_delay = (int)(o->mux_max_delay * AV_TIME_BASE);
- /* copy metadata and chapters from input files */
- err = copy_meta(mux, o);
- if (err < 0)
- return err;
- err = of_add_programs(mux, o);
- if (err < 0)
- return err;
- err = of_add_metadata(of, oc, o);
- if (err < 0)
- return err;
- err = set_dispositions(mux, o);
- if (err < 0) {
- av_log(mux, AV_LOG_FATAL, "Error setting output stream dispositions\n");
- return err;
- }
- // parse forced keyframe specifications;
- // must be done after chapters are created
- err = process_forced_keyframes(mux, o);
- if (err < 0) {
- av_log(mux, AV_LOG_FATAL, "Error processing forced keyframes\n");
- return err;
- }
- err = setup_sync_queues(mux, oc, o->shortest_buf_duration * AV_TIME_BASE);
- if (err < 0) {
- av_log(mux, AV_LOG_FATAL, "Error setting up output sync queues\n");
- return err;
- }
- of->url = filename;
- /* initialize stream copy and subtitle/data streams.
- * Encoded AVFrame based streams will get initialized when the first AVFrame
- * is received in do_video_out
- */
- for (int i = 0; i < of->nb_streams; i++) {
- OutputStream *ost = of->streams[i];
- if (ost->filter)
- continue;
- err = init_output_stream_nofilter(ost);
- if (err < 0)
- return err;
- }
- /* write the header for files with no streams */
- if (of->format->flags & AVFMT_NOSTREAMS && oc->nb_streams == 0) {
- int ret = mux_check_init(mux);
- if (ret < 0)
- return ret;
- }
- return 0;
- }
|