f53df7da64
Code submissions have continually suffered from formatting inconsistencies that constantly have to be addressed. Using clang-format simplifies this by making code formatting more consistent, and allows automation of the code formatting so that maintainers can focus more on the code itself instead of code formatting.
276 lines
6.7 KiB
C
276 lines
6.7 KiB
C
/******************************************************************************
|
|
Copyright (C) 2014 by Ruwen Hahn <palana@stunned.de>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
******************************************************************************/
|
|
|
|
#include "media-remux.h"
|
|
|
|
#include "../util/base.h"
|
|
#include "../util/bmem.h"
|
|
#include "../util/platform.h"
|
|
|
|
#include <libavformat/avformat.h>
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
|
|
#if LIBAVCODEC_VERSION_MAJOR >= 58
|
|
#define CODEC_FLAG_GLOBAL_H AV_CODEC_FLAG_GLOBAL_HEADER
|
|
#else
|
|
#define CODEC_FLAG_GLOBAL_H CODEC_FLAG_GLOBAL_HEADER
|
|
#endif
|
|
|
|
struct media_remux_job {
|
|
int64_t in_size;
|
|
AVFormatContext *ifmt_ctx, *ofmt_ctx;
|
|
};
|
|
|
|
static inline void init_size(media_remux_job_t job, const char *in_filename)
|
|
{
|
|
#ifdef _MSC_VER
|
|
struct _stat64 st = {0};
|
|
_stat64(in_filename, &st);
|
|
#else
|
|
struct stat st = {0};
|
|
stat(in_filename, &st);
|
|
#endif
|
|
job->in_size = st.st_size;
|
|
}
|
|
|
|
static inline bool init_input(media_remux_job_t job, const char *in_filename)
|
|
{
|
|
int ret = avformat_open_input(&job->ifmt_ctx, in_filename, NULL, NULL);
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR, "media_remux: Could not open input file '%s'",
|
|
in_filename);
|
|
return false;
|
|
}
|
|
|
|
ret = avformat_find_stream_info(job->ifmt_ctx, NULL);
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR, "media_remux: Failed to retrieve input stream"
|
|
" information");
|
|
return false;
|
|
}
|
|
|
|
#ifndef _NDEBUG
|
|
av_dump_format(job->ifmt_ctx, 0, in_filename, false);
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
static inline bool init_output(media_remux_job_t job, const char *out_filename)
|
|
{
|
|
int ret;
|
|
|
|
avformat_alloc_output_context2(&job->ofmt_ctx, NULL, NULL,
|
|
out_filename);
|
|
if (!job->ofmt_ctx) {
|
|
blog(LOG_ERROR, "media_remux: Could not create output context");
|
|
return false;
|
|
}
|
|
|
|
for (unsigned i = 0; i < job->ifmt_ctx->nb_streams; i++) {
|
|
AVStream *in_stream = job->ifmt_ctx->streams[i];
|
|
AVStream *out_stream = avformat_new_stream(
|
|
job->ofmt_ctx, in_stream->codec->codec);
|
|
if (!out_stream) {
|
|
blog(LOG_ERROR, "media_remux: Failed to allocate output"
|
|
" stream");
|
|
return false;
|
|
}
|
|
|
|
#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 48, 101)
|
|
AVCodecParameters *par = avcodec_parameters_alloc();
|
|
ret = avcodec_parameters_from_context(par, in_stream->codec);
|
|
if (ret == 0)
|
|
ret = avcodec_parameters_to_context(out_stream->codec,
|
|
par);
|
|
avcodec_parameters_free(&par);
|
|
#else
|
|
ret = avcodec_copy_context(out_stream->codec, in_stream->codec);
|
|
#endif
|
|
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR, "media_remux: Failed to copy context");
|
|
return false;
|
|
}
|
|
out_stream->time_base = out_stream->codec->time_base;
|
|
|
|
av_dict_copy(&out_stream->metadata, in_stream->metadata, 0);
|
|
|
|
out_stream->codec->codec_tag = 0;
|
|
if (job->ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
|
|
out_stream->codec->flags |= CODEC_FLAG_GLOBAL_H;
|
|
}
|
|
|
|
#ifndef _NDEBUG
|
|
av_dump_format(job->ofmt_ctx, 0, out_filename, true);
|
|
#endif
|
|
|
|
if (!(job->ofmt_ctx->oformat->flags & AVFMT_NOFILE)) {
|
|
ret = avio_open(&job->ofmt_ctx->pb, out_filename,
|
|
AVIO_FLAG_WRITE);
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR,
|
|
"media_remux: Failed to open output"
|
|
" file '%s'",
|
|
out_filename);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool media_remux_job_create(media_remux_job_t *job, const char *in_filename,
|
|
const char *out_filename)
|
|
{
|
|
if (!job)
|
|
return false;
|
|
|
|
*job = NULL;
|
|
if (!os_file_exists(in_filename))
|
|
return false;
|
|
|
|
if (strcmp(in_filename, out_filename) == 0)
|
|
return false;
|
|
|
|
*job = (media_remux_job_t)bzalloc(sizeof(struct media_remux_job));
|
|
if (!*job)
|
|
return false;
|
|
|
|
init_size(*job, in_filename);
|
|
|
|
av_register_all();
|
|
|
|
if (!init_input(*job, in_filename))
|
|
goto fail;
|
|
|
|
if (!init_output(*job, out_filename))
|
|
goto fail;
|
|
|
|
return true;
|
|
|
|
fail:
|
|
media_remux_job_destroy(*job);
|
|
return false;
|
|
}
|
|
|
|
static inline void process_packet(AVPacket *pkt, AVStream *in_stream,
|
|
AVStream *out_stream)
|
|
{
|
|
pkt->pts = av_rescale_q_rnd(pkt->pts, in_stream->time_base,
|
|
out_stream->time_base,
|
|
AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX);
|
|
pkt->dts = av_rescale_q_rnd(pkt->dts, in_stream->time_base,
|
|
out_stream->time_base,
|
|
AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX);
|
|
pkt->duration = (int)av_rescale_q(pkt->duration, in_stream->time_base,
|
|
out_stream->time_base);
|
|
pkt->pos = -1;
|
|
}
|
|
|
|
static inline int process_packets(media_remux_job_t job,
|
|
media_remux_progress_callback callback,
|
|
void *data)
|
|
{
|
|
AVPacket pkt;
|
|
|
|
int ret, throttle = 0;
|
|
for (;;) {
|
|
ret = av_read_frame(job->ifmt_ctx, &pkt);
|
|
if (ret < 0) {
|
|
if (ret != AVERROR_EOF)
|
|
blog(LOG_ERROR,
|
|
"media_remux: Error reading"
|
|
" packet: %s",
|
|
av_err2str(ret));
|
|
break;
|
|
}
|
|
|
|
if (callback != NULL && throttle++ > 10) {
|
|
float progress = pkt.pos / (float)job->in_size * 100.f;
|
|
if (!callback(data, progress))
|
|
break;
|
|
throttle = 0;
|
|
}
|
|
|
|
process_packet(&pkt, job->ifmt_ctx->streams[pkt.stream_index],
|
|
job->ofmt_ctx->streams[pkt.stream_index]);
|
|
|
|
ret = av_interleaved_write_frame(job->ofmt_ctx, &pkt);
|
|
av_packet_unref(&pkt);
|
|
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR, "media_remux: Error muxing packet: %s",
|
|
av_err2str(ret));
|
|
break;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool media_remux_job_process(media_remux_job_t job,
|
|
media_remux_progress_callback callback, void *data)
|
|
{
|
|
int ret;
|
|
bool success = false;
|
|
|
|
if (!job)
|
|
return success;
|
|
|
|
ret = avformat_write_header(job->ofmt_ctx, NULL);
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR, "media_remux: Error opening output file: %s",
|
|
av_err2str(ret));
|
|
return success;
|
|
}
|
|
|
|
if (callback != NULL)
|
|
callback(data, 0.f);
|
|
|
|
ret = process_packets(job, callback, data);
|
|
success = ret >= 0 || ret == AVERROR_EOF;
|
|
|
|
ret = av_write_trailer(job->ofmt_ctx);
|
|
if (ret < 0) {
|
|
blog(LOG_ERROR, "media_remux: av_write_trailer: %s",
|
|
av_err2str(ret));
|
|
success = false;
|
|
}
|
|
|
|
if (callback != NULL)
|
|
callback(data, 100.f);
|
|
|
|
return success;
|
|
}
|
|
|
|
void media_remux_job_destroy(media_remux_job_t job)
|
|
{
|
|
if (!job)
|
|
return;
|
|
|
|
avformat_close_input(&job->ifmt_ctx);
|
|
|
|
if (job->ofmt_ctx && !(job->ofmt_ctx->oformat->flags & AVFMT_NOFILE))
|
|
avio_close(job->ofmt_ctx->pb);
|
|
|
|
avformat_free_context(job->ofmt_ctx);
|
|
|
|
bfree(job);
|
|
}
|