1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351 |
- /*
- * Copyright (c) 2000-2003 Fabrice Bellard
- *
- * 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
- */
- /**
- * @file
- * multimedia converter based on the FFmpeg libraries
- */
- #include "config.h"
- #include <errno.h>
- #include <limits.h>
- #include <stdatomic.h>
- #include <stdint.h>
- #include <stdlib.h>
- #include <string.h>
- #include <time.h>
- #if HAVE_IO_H
- #include <io.h>
- #endif
- #if HAVE_UNISTD_H
- #include <unistd.h>
- #endif
- #if HAVE_SYS_RESOURCE_H
- #include <sys/time.h>
- #include <sys/types.h>
- #include <sys/resource.h>
- #elif HAVE_GETPROCESSTIMES
- #include <windows.h>
- #endif
- #if HAVE_GETPROCESSMEMORYINFO
- #include <windows.h>
- #include <psapi.h>
- #endif
- #if HAVE_SETCONSOLECTRLHANDLER
- #include <windows.h>
- #endif
- #if HAVE_SYS_SELECT_H
- #include <sys/select.h>
- #endif
- #if HAVE_TERMIOS_H
- #include <fcntl.h>
- #include <sys/ioctl.h>
- #include <sys/time.h>
- #include <termios.h>
- #elif HAVE_KBHIT
- #include <conio.h>
- #endif
- #include "libavutil/avassert.h"
- #include "libavutil/avstring.h"
- #include "libavutil/bprint.h"
- #include "libavutil/channel_layout.h"
- #include "libavutil/dict.h"
- #include "libavutil/display.h"
- #include "libavutil/fifo.h"
- #include "libavutil/hwcontext.h"
- #include "libavutil/imgutils.h"
- #include "libavutil/intreadwrite.h"
- #include "libavutil/libm.h"
- #include "libavutil/mathematics.h"
- #include "libavutil/opt.h"
- #include "libavutil/parseutils.h"
- #include "libavutil/pixdesc.h"
- #include "libavutil/samplefmt.h"
- #include "libavutil/thread.h"
- #include "libavutil/threadmessage.h"
- #include "libavutil/time.h"
- #include "libavutil/timestamp.h"
- #include "libavcodec/version.h"
- #include "libavformat/avformat.h"
- #include "libavdevice/avdevice.h"
- #include "libswresample/swresample.h"
- #include "cmdutils.h"
- #include "ffmpeg.h"
- #include "sync_queue.h"
- const char program_name[] = "ffmpeg";
- const int program_birth_year = 2000;
- FILE *vstats_file;
- typedef struct BenchmarkTimeStamps {
- int64_t real_usec;
- int64_t user_usec;
- int64_t sys_usec;
- } BenchmarkTimeStamps;
- static BenchmarkTimeStamps get_benchmark_time_stamps(void);
- static int64_t getmaxrss(void);
- unsigned nb_output_dumped = 0;
- static BenchmarkTimeStamps current_time;
- AVIOContext *progress_avio = NULL;
- InputFile **input_files = NULL;
- int nb_input_files = 0;
- OutputFile **output_files = NULL;
- int nb_output_files = 0;
- FilterGraph **filtergraphs;
- int nb_filtergraphs;
- #if HAVE_TERMIOS_H
- /* init terminal so that we can grab keys */
- static struct termios oldtty;
- static int restore_tty;
- #endif
- /* sub2video hack:
- Convert subtitles to video with alpha to insert them in filter graphs.
- This is a temporary solution until libavfilter gets real subtitles support.
- */
- static void sub2video_heartbeat(InputFile *infile, int64_t pts, AVRational tb)
- {
- /* When a frame is read from a file, examine all sub2video streams in
- the same file and send the sub2video frame again. Otherwise, decoded
- video frames could be accumulating in the filter graph while a filter
- (possibly overlay) is desperately waiting for a subtitle frame. */
- for (int i = 0; i < infile->nb_streams; i++) {
- InputStream *ist = infile->streams[i];
- if (ist->dec_ctx->codec_type != AVMEDIA_TYPE_SUBTITLE)
- continue;
- for (int j = 0; j < ist->nb_filters; j++)
- ifilter_sub2video_heartbeat(ist->filters[j], pts, tb);
- }
- }
- /* end of sub2video hack */
- static void term_exit_sigsafe(void)
- {
- #if HAVE_TERMIOS_H
- if(restore_tty)
- tcsetattr (0, TCSANOW, &oldtty);
- #endif
- }
- void term_exit(void)
- {
- av_log(NULL, AV_LOG_QUIET, "%s", "");
- term_exit_sigsafe();
- }
- static volatile int received_sigterm = 0;
- static volatile int received_nb_signals = 0;
- static atomic_int transcode_init_done = ATOMIC_VAR_INIT(0);
- static volatile int ffmpeg_exited = 0;
- static int64_t copy_ts_first_pts = AV_NOPTS_VALUE;
- static void
- sigterm_handler(int sig)
- {
- int ret;
- received_sigterm = sig;
- received_nb_signals++;
- term_exit_sigsafe();
- if(received_nb_signals > 3) {
- ret = write(2/*STDERR_FILENO*/, "Received > 3 system signals, hard exiting\n",
- strlen("Received > 3 system signals, hard exiting\n"));
- if (ret < 0) { /* Do nothing */ };
- exit(123);
- }
- }
- #if HAVE_SETCONSOLECTRLHANDLER
- static BOOL WINAPI CtrlHandler(DWORD fdwCtrlType)
- {
- av_log(NULL, AV_LOG_DEBUG, "\nReceived windows signal %ld\n", fdwCtrlType);
- switch (fdwCtrlType)
- {
- case CTRL_C_EVENT:
- case CTRL_BREAK_EVENT:
- sigterm_handler(SIGINT);
- return TRUE;
- case CTRL_CLOSE_EVENT:
- case CTRL_LOGOFF_EVENT:
- case CTRL_SHUTDOWN_EVENT:
- sigterm_handler(SIGTERM);
- /* Basically, with these 3 events, when we return from this method the
- process is hard terminated, so stall as long as we need to
- to try and let the main thread(s) clean up and gracefully terminate
- (we have at most 5 seconds, but should be done far before that). */
- while (!ffmpeg_exited) {
- Sleep(0);
- }
- return TRUE;
- default:
- av_log(NULL, AV_LOG_ERROR, "Received unknown windows signal %ld\n", fdwCtrlType);
- return FALSE;
- }
- }
- #endif
- #ifdef __linux__
- #define SIGNAL(sig, func) \
- do { \
- action.sa_handler = func; \
- sigaction(sig, &action, NULL); \
- } while (0)
- #else
- #define SIGNAL(sig, func) \
- signal(sig, func)
- #endif
- void term_init(void)
- {
- #if defined __linux__
- struct sigaction action = {0};
- action.sa_handler = sigterm_handler;
- /* block other interrupts while processing this one */
- sigfillset(&action.sa_mask);
- /* restart interruptible functions (i.e. don't fail with EINTR) */
- action.sa_flags = SA_RESTART;
- #endif
- #if HAVE_TERMIOS_H
- if (stdin_interaction) {
- struct termios tty;
- if (tcgetattr (0, &tty) == 0) {
- oldtty = tty;
- restore_tty = 1;
- tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
- |INLCR|IGNCR|ICRNL|IXON);
- tty.c_oflag |= OPOST;
- tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
- tty.c_cflag &= ~(CSIZE|PARENB);
- tty.c_cflag |= CS8;
- tty.c_cc[VMIN] = 1;
- tty.c_cc[VTIME] = 0;
- tcsetattr (0, TCSANOW, &tty);
- }
- SIGNAL(SIGQUIT, sigterm_handler); /* Quit (POSIX). */
- }
- #endif
- SIGNAL(SIGINT , sigterm_handler); /* Interrupt (ANSI). */
- SIGNAL(SIGTERM, sigterm_handler); /* Termination (ANSI). */
- #ifdef SIGXCPU
- SIGNAL(SIGXCPU, sigterm_handler);
- #endif
- #ifdef SIGPIPE
- signal(SIGPIPE, SIG_IGN); /* Broken pipe (POSIX). */
- #endif
- #if HAVE_SETCONSOLECTRLHANDLER
- SetConsoleCtrlHandler((PHANDLER_ROUTINE) CtrlHandler, TRUE);
- #endif
- }
- /* read a key without blocking */
- static int read_key(void)
- {
- unsigned char ch;
- #if HAVE_TERMIOS_H
- int n = 1;
- struct timeval tv;
- fd_set rfds;
- FD_ZERO(&rfds);
- FD_SET(0, &rfds);
- tv.tv_sec = 0;
- tv.tv_usec = 0;
- n = select(1, &rfds, NULL, NULL, &tv);
- if (n > 0) {
- n = read(0, &ch, 1);
- if (n == 1)
- return ch;
- return n;
- }
- #elif HAVE_KBHIT
- # if HAVE_PEEKNAMEDPIPE && HAVE_GETSTDHANDLE
- static int is_pipe;
- static HANDLE input_handle;
- DWORD dw, nchars;
- if(!input_handle){
- input_handle = GetStdHandle(STD_INPUT_HANDLE);
- is_pipe = !GetConsoleMode(input_handle, &dw);
- }
- if (is_pipe) {
- /* When running under a GUI, you will end here. */
- if (!PeekNamedPipe(input_handle, NULL, 0, NULL, &nchars, NULL)) {
- // input pipe may have been closed by the program that ran ffmpeg
- return -1;
- }
- //Read it
- if(nchars != 0) {
- read(0, &ch, 1);
- return ch;
- }else{
- return -1;
- }
- }
- # endif
- if(kbhit())
- return(getch());
- #endif
- return -1;
- }
- static int decode_interrupt_cb(void *ctx)
- {
- return received_nb_signals > atomic_load(&transcode_init_done);
- }
- const AVIOInterruptCB int_cb = { decode_interrupt_cb, NULL };
- static void ffmpeg_cleanup(int ret)
- {
- int i;
- if (do_benchmark) {
- int maxrss = getmaxrss() / 1024;
- av_log(NULL, AV_LOG_INFO, "bench: maxrss=%ikB\n", maxrss);
- }
- for (i = 0; i < nb_filtergraphs; i++)
- fg_free(&filtergraphs[i]);
- av_freep(&filtergraphs);
- for (i = 0; i < nb_output_files; i++)
- of_free(&output_files[i]);
- for (i = 0; i < nb_input_files; i++)
- ifile_close(&input_files[i]);
- if (vstats_file) {
- if (fclose(vstats_file))
- av_log(NULL, AV_LOG_ERROR,
- "Error closing vstats file, loss of information possible: %s\n",
- av_err2str(AVERROR(errno)));
- }
- av_freep(&vstats_filename);
- of_enc_stats_close();
- hw_device_free_all();
- av_freep(&filter_nbthreads);
- av_freep(&input_files);
- av_freep(&output_files);
- uninit_opts();
- avformat_network_deinit();
- if (received_sigterm) {
- av_log(NULL, AV_LOG_INFO, "Exiting normally, received signal %d.\n",
- (int) received_sigterm);
- } else if (ret && atomic_load(&transcode_init_done)) {
- av_log(NULL, AV_LOG_INFO, "Conversion failed!\n");
- }
- term_exit();
- ffmpeg_exited = 1;
- }
- OutputStream *ost_iter(OutputStream *prev)
- {
- int of_idx = prev ? prev->file_index : 0;
- int ost_idx = prev ? prev->index + 1 : 0;
- for (; of_idx < nb_output_files; of_idx++) {
- OutputFile *of = output_files[of_idx];
- if (ost_idx < of->nb_streams)
- return of->streams[ost_idx];
- ost_idx = 0;
- }
- return NULL;
- }
- InputStream *ist_iter(InputStream *prev)
- {
- int if_idx = prev ? prev->file_index : 0;
- int ist_idx = prev ? prev->index + 1 : 0;
- for (; if_idx < nb_input_files; if_idx++) {
- InputFile *f = input_files[if_idx];
- if (ist_idx < f->nb_streams)
- return f->streams[ist_idx];
- ist_idx = 0;
- }
- return NULL;
- }
- FrameData *frame_data(AVFrame *frame)
- {
- if (!frame->opaque_ref) {
- FrameData *fd;
- frame->opaque_ref = av_buffer_allocz(sizeof(*fd));
- if (!frame->opaque_ref)
- return NULL;
- fd = (FrameData*)frame->opaque_ref->data;
- fd->dec.frame_num = UINT64_MAX;
- fd->dec.pts = AV_NOPTS_VALUE;
- }
- return (FrameData*)frame->opaque_ref->data;
- }
- void remove_avoptions(AVDictionary **a, AVDictionary *b)
- {
- const AVDictionaryEntry *t = NULL;
- while ((t = av_dict_iterate(b, t))) {
- av_dict_set(a, t->key, NULL, AV_DICT_MATCH_CASE);
- }
- }
- int check_avoptions(AVDictionary *m)
- {
- const AVDictionaryEntry *t;
- if ((t = av_dict_get(m, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
- av_log(NULL, AV_LOG_FATAL, "Option %s not found.\n", t->key);
- return AVERROR_OPTION_NOT_FOUND;
- }
- return 0;
- }
- void update_benchmark(const char *fmt, ...)
- {
- if (do_benchmark_all) {
- BenchmarkTimeStamps t = get_benchmark_time_stamps();
- va_list va;
- char buf[1024];
- if (fmt) {
- va_start(va, fmt);
- vsnprintf(buf, sizeof(buf), fmt, va);
- va_end(va);
- av_log(NULL, AV_LOG_INFO,
- "bench: %8" PRIu64 " user %8" PRIu64 " sys %8" PRIu64 " real %s \n",
- t.user_usec - current_time.user_usec,
- t.sys_usec - current_time.sys_usec,
- t.real_usec - current_time.real_usec, buf);
- }
- current_time = t;
- }
- }
- void close_output_stream(OutputStream *ost)
- {
- OutputFile *of = output_files[ost->file_index];
- ost->finished |= ENCODER_FINISHED;
- if (ost->sq_idx_encode >= 0)
- sq_send(of->sq_encode, ost->sq_idx_encode, SQFRAME(NULL));
- }
- static void print_report(int is_last_report, int64_t timer_start, int64_t cur_time)
- {
- AVBPrint buf, buf_script;
- int64_t total_size = of_filesize(output_files[0]);
- int vid;
- double bitrate;
- double speed;
- int64_t pts = AV_NOPTS_VALUE;
- static int64_t last_time = -1;
- static int first_report = 1;
- uint64_t nb_frames_dup = 0, nb_frames_drop = 0;
- int mins, secs, us;
- int64_t hours;
- const char *hours_sign;
- int ret;
- float t;
- if (!print_stats && !is_last_report && !progress_avio)
- return;
- if (!is_last_report) {
- if (last_time == -1) {
- last_time = cur_time;
- }
- if (((cur_time - last_time) < stats_period && !first_report) ||
- (first_report && nb_output_dumped < nb_output_files))
- return;
- last_time = cur_time;
- }
- t = (cur_time-timer_start) / 1000000.0;
- vid = 0;
- av_bprint_init(&buf, 0, AV_BPRINT_SIZE_AUTOMATIC);
- av_bprint_init(&buf_script, 0, AV_BPRINT_SIZE_AUTOMATIC);
- for (OutputStream *ost = ost_iter(NULL); ost; ost = ost_iter(ost)) {
- const float q = ost->enc ? ost->quality / (float) FF_QP2LAMBDA : -1;
- if (vid && ost->type == AVMEDIA_TYPE_VIDEO) {
- av_bprintf(&buf, "q=%2.1f ", q);
- av_bprintf(&buf_script, "stream_%d_%d_q=%.1f\n",
- ost->file_index, ost->index, q);
- }
- if (!vid && ost->type == AVMEDIA_TYPE_VIDEO && ost->filter) {
- float fps;
- uint64_t frame_number = atomic_load(&ost->packets_written);
- fps = t > 1 ? frame_number / t : 0;
- av_bprintf(&buf, "frame=%5"PRId64" fps=%3.*f q=%3.1f ",
- frame_number, fps < 9.95, fps, q);
- av_bprintf(&buf_script, "frame=%"PRId64"\n", frame_number);
- av_bprintf(&buf_script, "fps=%.2f\n", fps);
- av_bprintf(&buf_script, "stream_%d_%d_q=%.1f\n",
- ost->file_index, ost->index, q);
- if (is_last_report)
- av_bprintf(&buf, "L");
- nb_frames_dup = ost->filter->nb_frames_dup;
- nb_frames_drop = ost->filter->nb_frames_drop;
- vid = 1;
- }
- /* compute min output value */
- if (ost->last_mux_dts != AV_NOPTS_VALUE) {
- if (pts == AV_NOPTS_VALUE || ost->last_mux_dts > pts)
- pts = ost->last_mux_dts;
- if (copy_ts) {
- if (copy_ts_first_pts == AV_NOPTS_VALUE && pts > 1)
- copy_ts_first_pts = pts;
- if (copy_ts_first_pts != AV_NOPTS_VALUE)
- pts -= copy_ts_first_pts;
- }
- }
- }
- us = FFABS64U(pts) % AV_TIME_BASE;
- secs = FFABS64U(pts) / AV_TIME_BASE % 60;
- mins = FFABS64U(pts) / AV_TIME_BASE / 60 % 60;
- hours = FFABS64U(pts) / AV_TIME_BASE / 3600;
- hours_sign = (pts < 0) ? "-" : "";
- bitrate = pts != AV_NOPTS_VALUE && pts && total_size >= 0 ? total_size * 8 / (pts / 1000.0) : -1;
- speed = pts != AV_NOPTS_VALUE && t != 0.0 ? (double)pts / AV_TIME_BASE / t : -1;
- if (total_size < 0) av_bprintf(&buf, "size=N/A time=");
- else av_bprintf(&buf, "size=%8.0fkB time=", total_size / 1024.0);
- if (pts == AV_NOPTS_VALUE) {
- av_bprintf(&buf, "N/A ");
- } else {
- av_bprintf(&buf, "%s%02"PRId64":%02d:%02d.%02d ",
- hours_sign, hours, mins, secs, (100 * us) / AV_TIME_BASE);
- }
- if (bitrate < 0) {
- av_bprintf(&buf, "bitrate=N/A");
- av_bprintf(&buf_script, "bitrate=N/A\n");
- }else{
- av_bprintf(&buf, "bitrate=%6.1fkbits/s", bitrate);
- av_bprintf(&buf_script, "bitrate=%6.1fkbits/s\n", bitrate);
- }
- if (total_size < 0) av_bprintf(&buf_script, "total_size=N/A\n");
- else av_bprintf(&buf_script, "total_size=%"PRId64"\n", total_size);
- if (pts == AV_NOPTS_VALUE) {
- av_bprintf(&buf_script, "out_time_us=N/A\n");
- av_bprintf(&buf_script, "out_time_ms=N/A\n");
- av_bprintf(&buf_script, "out_time=N/A\n");
- } else {
- av_bprintf(&buf_script, "out_time_us=%"PRId64"\n", pts);
- av_bprintf(&buf_script, "out_time_ms=%"PRId64"\n", pts);
- av_bprintf(&buf_script, "out_time=%s%02"PRId64":%02d:%02d.%06d\n",
- hours_sign, hours, mins, secs, us);
- }
- if (nb_frames_dup || nb_frames_drop)
- av_bprintf(&buf, " dup=%"PRId64" drop=%"PRId64, nb_frames_dup, nb_frames_drop);
- av_bprintf(&buf_script, "dup_frames=%"PRId64"\n", nb_frames_dup);
- av_bprintf(&buf_script, "drop_frames=%"PRId64"\n", nb_frames_drop);
- if (speed < 0) {
- av_bprintf(&buf, " speed=N/A");
- av_bprintf(&buf_script, "speed=N/A\n");
- } else {
- av_bprintf(&buf, " speed=%4.3gx", speed);
- av_bprintf(&buf_script, "speed=%4.3gx\n", speed);
- }
- if (print_stats || is_last_report) {
- const char end = is_last_report ? '\n' : '\r';
- if (print_stats==1 && AV_LOG_INFO > av_log_get_level()) {
- fprintf(stderr, "%s %c", buf.str, end);
- } else
- av_log(NULL, AV_LOG_INFO, "%s %c", buf.str, end);
- fflush(stderr);
- }
- av_bprint_finalize(&buf, NULL);
- if (progress_avio) {
- av_bprintf(&buf_script, "progress=%s\n",
- is_last_report ? "end" : "continue");
- avio_write(progress_avio, buf_script.str,
- FFMIN(buf_script.len, buf_script.size - 1));
- avio_flush(progress_avio);
- av_bprint_finalize(&buf_script, NULL);
- if (is_last_report) {
- if ((ret = avio_closep(&progress_avio)) < 0)
- av_log(NULL, AV_LOG_ERROR,
- "Error closing progress log, loss of information possible: %s\n", av_err2str(ret));
- }
- }
- first_report = 0;
- }
- int copy_av_subtitle(AVSubtitle *dst, const AVSubtitle *src)
- {
- int ret = AVERROR_BUG;
- AVSubtitle tmp = {
- .format = src->format,
- .start_display_time = src->start_display_time,
- .end_display_time = src->end_display_time,
- .num_rects = 0,
- .rects = NULL,
- .pts = src->pts
- };
- if (!src->num_rects)
- goto success;
- if (!(tmp.rects = av_calloc(src->num_rects, sizeof(*tmp.rects))))
- return AVERROR(ENOMEM);
- for (int i = 0; i < src->num_rects; i++) {
- AVSubtitleRect *src_rect = src->rects[i];
- AVSubtitleRect *dst_rect;
- if (!(dst_rect = tmp.rects[i] = av_mallocz(sizeof(*tmp.rects[0])))) {
- ret = AVERROR(ENOMEM);
- goto cleanup;
- }
- tmp.num_rects++;
- dst_rect->type = src_rect->type;
- dst_rect->flags = src_rect->flags;
- dst_rect->x = src_rect->x;
- dst_rect->y = src_rect->y;
- dst_rect->w = src_rect->w;
- dst_rect->h = src_rect->h;
- dst_rect->nb_colors = src_rect->nb_colors;
- if (src_rect->text)
- if (!(dst_rect->text = av_strdup(src_rect->text))) {
- ret = AVERROR(ENOMEM);
- goto cleanup;
- }
- if (src_rect->ass)
- if (!(dst_rect->ass = av_strdup(src_rect->ass))) {
- ret = AVERROR(ENOMEM);
- goto cleanup;
- }
- for (int j = 0; j < 4; j++) {
- // SUBTITLE_BITMAP images are special in the sense that they
- // are like PAL8 images. first pointer to data, second to
- // palette. This makes the size calculation match this.
- size_t buf_size = src_rect->type == SUBTITLE_BITMAP && j == 1 ?
- AVPALETTE_SIZE :
- src_rect->h * src_rect->linesize[j];
- if (!src_rect->data[j])
- continue;
- if (!(dst_rect->data[j] = av_memdup(src_rect->data[j], buf_size))) {
- ret = AVERROR(ENOMEM);
- goto cleanup;
- }
- dst_rect->linesize[j] = src_rect->linesize[j];
- }
- }
- success:
- *dst = tmp;
- return 0;
- cleanup:
- avsubtitle_free(&tmp);
- return ret;
- }
- static void subtitle_free(void *opaque, uint8_t *data)
- {
- AVSubtitle *sub = (AVSubtitle*)data;
- avsubtitle_free(sub);
- av_free(sub);
- }
- int subtitle_wrap_frame(AVFrame *frame, AVSubtitle *subtitle, int copy)
- {
- AVBufferRef *buf;
- AVSubtitle *sub;
- int ret;
- if (copy) {
- sub = av_mallocz(sizeof(*sub));
- ret = sub ? copy_av_subtitle(sub, subtitle) : AVERROR(ENOMEM);
- if (ret < 0) {
- av_freep(&sub);
- return ret;
- }
- } else {
- sub = av_memdup(subtitle, sizeof(*subtitle));
- if (!sub)
- return AVERROR(ENOMEM);
- memset(subtitle, 0, sizeof(*subtitle));
- }
- buf = av_buffer_create((uint8_t*)sub, sizeof(*sub),
- subtitle_free, NULL, 0);
- if (!buf) {
- avsubtitle_free(sub);
- av_freep(&sub);
- return AVERROR(ENOMEM);
- }
- frame->buf[0] = buf;
- return 0;
- }
- int trigger_fix_sub_duration_heartbeat(OutputStream *ost, const AVPacket *pkt)
- {
- OutputFile *of = output_files[ost->file_index];
- int64_t signal_pts = av_rescale_q(pkt->pts, pkt->time_base,
- AV_TIME_BASE_Q);
- if (!ost->fix_sub_duration_heartbeat || !(pkt->flags & AV_PKT_FLAG_KEY))
- // we are only interested in heartbeats on streams configured, and
- // only on random access points.
- return 0;
- for (int i = 0; i < of->nb_streams; i++) {
- OutputStream *iter_ost = of->streams[i];
- InputStream *ist = iter_ost->ist;
- int ret = AVERROR_BUG;
- if (iter_ost == ost || !ist || !ist->decoding_needed ||
- ist->dec_ctx->codec_type != AVMEDIA_TYPE_SUBTITLE)
- // We wish to skip the stream that causes the heartbeat,
- // output streams without an input stream, streams not decoded
- // (as fix_sub_duration is only done for decoded subtitles) as
- // well as non-subtitle streams.
- continue;
- if ((ret = fix_sub_duration_heartbeat(ist, signal_pts)) < 0)
- return ret;
- }
- return 0;
- }
- /* pkt = NULL means EOF (needed to flush decoder buffers) */
- static int process_input_packet(InputStream *ist, const AVPacket *pkt, int no_eof)
- {
- InputFile *f = input_files[ist->file_index];
- int64_t dts_est = AV_NOPTS_VALUE;
- int ret = 0;
- int eof_reached = 0;
- if (ist->decoding_needed) {
- ret = dec_packet(ist, pkt, no_eof);
- if (ret < 0 && ret != AVERROR_EOF)
- return ret;
- }
- if (ret == AVERROR_EOF || (!pkt && !ist->decoding_needed))
- eof_reached = 1;
- if (pkt && pkt->opaque_ref) {
- DemuxPktData *pd = (DemuxPktData*)pkt->opaque_ref->data;
- dts_est = pd->dts_est;
- }
- if (f->recording_time != INT64_MAX) {
- int64_t start_time = 0;
- if (copy_ts) {
- start_time += f->start_time != AV_NOPTS_VALUE ? f->start_time : 0;
- start_time += start_at_zero ? 0 : f->start_time_effective;
- }
- if (dts_est >= f->recording_time + start_time)
- pkt = NULL;
- }
- for (int oidx = 0; oidx < ist->nb_outputs; oidx++) {
- OutputStream *ost = ist->outputs[oidx];
- if (ost->enc || (!pkt && no_eof))
- continue;
- ret = of_streamcopy(ost, pkt, dts_est);
- if (ret < 0)
- return ret;
- }
- return !eof_reached;
- }
- static void print_stream_maps(void)
- {
- av_log(NULL, AV_LOG_INFO, "Stream mapping:\n");
- for (InputStream *ist = ist_iter(NULL); ist; ist = ist_iter(ist)) {
- for (int j = 0; j < ist->nb_filters; j++) {
- if (!filtergraph_is_simple(ist->filters[j]->graph)) {
- av_log(NULL, AV_LOG_INFO, " Stream #%d:%d (%s) -> %s",
- ist->file_index, ist->index, ist->dec ? ist->dec->name : "?",
- ist->filters[j]->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");
- }
- }
- }
- for (OutputStream *ost = ost_iter(NULL); ost; ost = ost_iter(ost)) {
- 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;
- }
- if (ost->filter && !filtergraph_is_simple(ost->filter->graph)) {
- /* output from a complex graph */
- av_log(NULL, AV_LOG_INFO, " %s", ost->filter->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_ctx->codec->name);
- continue;
- }
- av_log(NULL, AV_LOG_INFO, " Stream #%d:%d -> #%d:%d",
- ost->ist->file_index,
- ost->ist->index,
- ost->file_index,
- ost->index);
- if (ost->enc_ctx) {
- const AVCodec *in_codec = ost->ist->dec;
- const AVCodec *out_codec = ost->enc_ctx->codec;
- const char *decoder_name = "?";
- const char *in_codec_name = "?";
- const char *encoder_name = "?";
- const char *out_codec_name = "?";
- const AVCodecDescriptor *desc;
- if (in_codec) {
- decoder_name = in_codec->name;
- desc = avcodec_descriptor_get(in_codec->id);
- if (desc)
- in_codec_name = desc->name;
- if (!strcmp(decoder_name, in_codec_name))
- decoder_name = "native";
- }
- if (out_codec) {
- encoder_name = out_codec->name;
- desc = avcodec_descriptor_get(out_codec->id);
- if (desc)
- out_codec_name = desc->name;
- if (!strcmp(encoder_name, out_codec_name))
- encoder_name = "native";
- }
- av_log(NULL, AV_LOG_INFO, " (%s (%s) -> %s (%s))",
- in_codec_name, decoder_name,
- out_codec_name, encoder_name);
- } else
- av_log(NULL, AV_LOG_INFO, " (copy)");
- av_log(NULL, AV_LOG_INFO, "\n");
- }
- }
- /**
- * Select the output stream to process.
- *
- * @retval 0 an output stream was selected
- * @retval AVERROR(EAGAIN) need to wait until more input is available
- * @retval AVERROR_EOF no more streams need output
- */
- static int choose_output(OutputStream **post)
- {
- int64_t opts_min = INT64_MAX;
- OutputStream *ost_min = NULL;
- for (OutputStream *ost = ost_iter(NULL); ost; ost = ost_iter(ost)) {
- int64_t opts;
- if (ost->filter && ost->filter->last_pts != AV_NOPTS_VALUE) {
- opts = ost->filter->last_pts;
- } else {
- opts = ost->last_mux_dts == AV_NOPTS_VALUE ?
- INT64_MIN : ost->last_mux_dts;
- }
- if (!ost->initialized && !ost->finished) {
- ost_min = ost;
- break;
- }
- if (!ost->finished && opts < opts_min) {
- opts_min = opts;
- ost_min = ost;
- }
- }
- if (!ost_min)
- return AVERROR_EOF;
- *post = ost_min;
- return ost_min->unavailable ? AVERROR(EAGAIN) : 0;
- }
- static void set_tty_echo(int on)
- {
- #if HAVE_TERMIOS_H
- struct termios tty;
- if (tcgetattr(0, &tty) == 0) {
- if (on) tty.c_lflag |= ECHO;
- else tty.c_lflag &= ~ECHO;
- tcsetattr(0, TCSANOW, &tty);
- }
- #endif
- }
- static int check_keyboard_interaction(int64_t cur_time)
- {
- int i, key;
- static int64_t last_time;
- if (received_nb_signals)
- return AVERROR_EXIT;
- /* read_key() returns 0 on EOF */
- if (cur_time - last_time >= 100000) {
- key = read_key();
- last_time = cur_time;
- }else
- key = -1;
- if (key == 'q') {
- av_log(NULL, AV_LOG_INFO, "\n\n[q] command received. Exiting.\n\n");
- return AVERROR_EXIT;
- }
- if (key == '+') av_log_set_level(av_log_get_level()+10);
- if (key == '-') av_log_set_level(av_log_get_level()-10);
- if (key == 'c' || key == 'C'){
- char buf[4096], target[64], command[256], arg[256] = {0};
- double time;
- int k, n = 0;
- fprintf(stderr, "\nEnter command: <target>|all <time>|-1 <command>[ <argument>]\n");
- i = 0;
- set_tty_echo(1);
- while ((k = read_key()) != '\n' && k != '\r' && i < sizeof(buf)-1)
- if (k > 0)
- buf[i++] = k;
- buf[i] = 0;
- set_tty_echo(0);
- fprintf(stderr, "\n");
- if (k > 0 &&
- (n = sscanf(buf, "%63[^ ] %lf %255[^ ] %255[^\n]", target, &time, command, arg)) >= 3) {
- av_log(NULL, AV_LOG_DEBUG, "Processing command target:%s time:%f command:%s arg:%s",
- target, time, command, arg);
- for (i = 0; i < nb_filtergraphs; i++)
- fg_send_command(filtergraphs[i], time, target, command, arg,
- key == 'C');
- } else {
- av_log(NULL, AV_LOG_ERROR,
- "Parse error, at least 3 arguments were expected, "
- "only %d given in string '%s'\n", n, buf);
- }
- }
- if (key == '?'){
- fprintf(stderr, "key function\n"
- "? show this help\n"
- "+ increase verbosity\n"
- "- decrease verbosity\n"
- "c Send command to first matching filter supporting it\n"
- "C Send/Queue command to all matching filters\n"
- "h dump packets/hex press to cycle through the 3 states\n"
- "q quit\n"
- "s Show QP histogram\n"
- );
- }
- return 0;
- }
- static void reset_eagain(void)
- {
- int i;
- for (i = 0; i < nb_input_files; i++)
- input_files[i]->eagain = 0;
- for (OutputStream *ost = ost_iter(NULL); ost; ost = ost_iter(ost))
- ost->unavailable = 0;
- }
- static void decode_flush(InputFile *ifile)
- {
- for (int i = 0; i < ifile->nb_streams; i++) {
- InputStream *ist = ifile->streams[i];
- if (ist->discard || !ist->decoding_needed)
- continue;
- dec_packet(ist, NULL, 1);
- }
- }
- /*
- * Return
- * - 0 -- one packet was read and processed
- * - AVERROR(EAGAIN) -- no packets were available for selected file,
- * this function should be called again
- * - AVERROR_EOF -- this function should not be called again
- */
- static int process_input(int file_index)
- {
- InputFile *ifile = input_files[file_index];
- InputStream *ist;
- AVPacket *pkt;
- int ret, i;
- ret = ifile_get_packet(ifile, &pkt);
- if (ret == AVERROR(EAGAIN)) {
- ifile->eagain = 1;
- return ret;
- }
- if (ret == 1) {
- /* the input file is looped: flush the decoders */
- decode_flush(ifile);
- return AVERROR(EAGAIN);
- }
- if (ret < 0) {
- if (ret != AVERROR_EOF) {
- av_log(ifile, AV_LOG_ERROR,
- "Error retrieving a packet from demuxer: %s\n", av_err2str(ret));
- if (exit_on_error)
- return ret;
- }
- for (i = 0; i < ifile->nb_streams; i++) {
- ist = ifile->streams[i];
- if (!ist->discard) {
- ret = process_input_packet(ist, NULL, 0);
- if (ret>0)
- return 0;
- else if (ret < 0)
- return ret;
- }
- /* mark all outputs that don't go through lavfi as finished */
- for (int oidx = 0; oidx < ist->nb_outputs; oidx++) {
- OutputStream *ost = ist->outputs[oidx];
- OutputFile *of = output_files[ost->file_index];
- ret = of_output_packet(of, ost, NULL);
- if (ret < 0)
- return ret;
- }
- }
- ifile->eof_reached = 1;
- return AVERROR(EAGAIN);
- }
- reset_eagain();
- ist = ifile->streams[pkt->stream_index];
- sub2video_heartbeat(ifile, pkt->pts, pkt->time_base);
- ret = process_input_packet(ist, pkt, 0);
- av_packet_free(&pkt);
- return ret < 0 ? ret : 0;
- }
- /**
- * Run a single step of transcoding.
- *
- * @return 0 for success, <0 for error
- */
- static int transcode_step(OutputStream *ost)
- {
- InputStream *ist = NULL;
- int ret;
- if (ost->filter) {
- if ((ret = fg_transcode_step(ost->filter->graph, &ist)) < 0)
- return ret;
- if (!ist)
- return 0;
- } else {
- ist = ost->ist;
- av_assert0(ist);
- }
- ret = process_input(ist->file_index);
- if (ret == AVERROR(EAGAIN)) {
- if (input_files[ist->file_index]->eagain)
- ost->unavailable = 1;
- return 0;
- }
- if (ret < 0)
- return ret == AVERROR_EOF ? 0 : ret;
- // process_input() above might have caused output to become available
- // in multiple filtergraphs, so we process all of them
- for (int i = 0; i < nb_filtergraphs; i++) {
- ret = reap_filters(filtergraphs[i], 0);
- if (ret < 0)
- return ret;
- }
- return 0;
- }
- /*
- * The following code is the main loop of the file converter
- */
- static int transcode(int *err_rate_exceeded)
- {
- int ret = 0, i;
- InputStream *ist;
- int64_t timer_start;
- print_stream_maps();
- *err_rate_exceeded = 0;
- atomic_store(&transcode_init_done, 1);
- if (stdin_interaction) {
- av_log(NULL, AV_LOG_INFO, "Press [q] to stop, [?] for help\n");
- }
- timer_start = av_gettime_relative();
- while (!received_sigterm) {
- OutputStream *ost;
- int64_t cur_time= av_gettime_relative();
- /* if 'q' pressed, exits */
- if (stdin_interaction)
- if (check_keyboard_interaction(cur_time) < 0)
- break;
- ret = choose_output(&ost);
- if (ret == AVERROR(EAGAIN)) {
- reset_eagain();
- av_usleep(10000);
- ret = 0;
- continue;
- } else if (ret < 0) {
- av_log(NULL, AV_LOG_VERBOSE, "No more output streams to write to, finishing.\n");
- ret = 0;
- break;
- }
- ret = transcode_step(ost);
- if (ret < 0 && ret != AVERROR_EOF) {
- av_log(NULL, AV_LOG_ERROR, "Error while filtering: %s\n", av_err2str(ret));
- break;
- }
- /* dump report by using the output first video and audio streams */
- print_report(0, timer_start, cur_time);
- }
- /* at the end of stream, we must flush the decoder buffers */
- for (ist = ist_iter(NULL); ist; ist = ist_iter(ist)) {
- float err_rate;
- if (!input_files[ist->file_index]->eof_reached) {
- int err = process_input_packet(ist, NULL, 0);
- ret = err_merge(ret, err);
- }
- err_rate = (ist->frames_decoded || ist->decode_errors) ?
- ist->decode_errors / (ist->frames_decoded + ist->decode_errors) : 0.f;
- if (err_rate > max_error_rate) {
- av_log(ist, AV_LOG_FATAL, "Decode error rate %g exceeds maximum %g\n",
- err_rate, max_error_rate);
- *err_rate_exceeded = 1;
- } else if (err_rate)
- av_log(ist, AV_LOG_VERBOSE, "Decode error rate %g\n", err_rate);
- }
- ret = err_merge(ret, enc_flush());
- term_exit();
- /* write the trailer if needed */
- for (i = 0; i < nb_output_files; i++) {
- int err = of_write_trailer(output_files[i]);
- ret = err_merge(ret, err);
- }
- /* dump report by using the first video and audio streams */
- print_report(1, timer_start, av_gettime_relative());
- return ret;
- }
- static BenchmarkTimeStamps get_benchmark_time_stamps(void)
- {
- BenchmarkTimeStamps time_stamps = { av_gettime_relative() };
- #if HAVE_GETRUSAGE
- struct rusage rusage;
- getrusage(RUSAGE_SELF, &rusage);
- time_stamps.user_usec =
- (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
- time_stamps.sys_usec =
- (rusage.ru_stime.tv_sec * 1000000LL) + rusage.ru_stime.tv_usec;
- #elif HAVE_GETPROCESSTIMES
- HANDLE proc;
- FILETIME c, e, k, u;
- proc = GetCurrentProcess();
- GetProcessTimes(proc, &c, &e, &k, &u);
- time_stamps.user_usec =
- ((int64_t)u.dwHighDateTime << 32 | u.dwLowDateTime) / 10;
- time_stamps.sys_usec =
- ((int64_t)k.dwHighDateTime << 32 | k.dwLowDateTime) / 10;
- #else
- time_stamps.user_usec = time_stamps.sys_usec = 0;
- #endif
- return time_stamps;
- }
- 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
- }
- int main(int argc, char **argv)
- {
- int ret, err_rate_exceeded;
- BenchmarkTimeStamps ti;
- init_dynload();
- setvbuf(stderr,NULL,_IONBF,0); /* win32 runtime needs this */
- av_log_set_flags(AV_LOG_SKIP_REPEATED);
- parse_loglevel(argc, argv, options);
- #if CONFIG_AVDEVICE
- avdevice_register_all();
- #endif
- avformat_network_init();
- show_banner(argc, argv, options);
- /* parse options and open all input/output files */
- ret = ffmpeg_parse_options(argc, argv);
- if (ret < 0)
- goto finish;
- if (nb_output_files <= 0 && nb_input_files == 0) {
- show_usage();
- av_log(NULL, AV_LOG_WARNING, "Use -h to get full help or, even better, run 'man %s'\n", program_name);
- ret = 1;
- goto finish;
- }
- if (nb_output_files <= 0) {
- av_log(NULL, AV_LOG_FATAL, "At least one output file must be specified\n");
- ret = 1;
- goto finish;
- }
- current_time = ti = get_benchmark_time_stamps();
- ret = transcode(&err_rate_exceeded);
- if (ret >= 0 && do_benchmark) {
- int64_t utime, stime, rtime;
- current_time = get_benchmark_time_stamps();
- utime = current_time.user_usec - ti.user_usec;
- stime = current_time.sys_usec - ti.sys_usec;
- rtime = current_time.real_usec - ti.real_usec;
- av_log(NULL, AV_LOG_INFO,
- "bench: utime=%0.3fs stime=%0.3fs rtime=%0.3fs\n",
- utime / 1000000.0, stime / 1000000.0, rtime / 1000000.0);
- }
- ret = received_nb_signals ? 255 :
- err_rate_exceeded ? 69 : ret;
- finish:
- if (ret == AVERROR_EXIT)
- ret = 0;
- ffmpeg_cleanup(ret);
- return ret;
- }
|