mirror of
https://github.com/mpv-player/mpv
synced 2024-12-25 00:02:13 +00:00
e6f543ebec
Example see edl-mpv.rst. What is this useful for? No clue...
295 lines
9.9 KiB
C
295 lines
9.9 KiB
C
/*
|
|
* This file is part of MPlayer.
|
|
*
|
|
* MPlayer 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.
|
|
*
|
|
* MPlayer 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 MPlayer; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include <inttypes.h>
|
|
#include <ctype.h>
|
|
#include <math.h>
|
|
|
|
#include "talloc.h"
|
|
|
|
#include "player/core.h"
|
|
#include "common/msg.h"
|
|
#include "demux/demux.h"
|
|
#include "options/path.h"
|
|
#include "bstr/bstr.h"
|
|
#include "common/common.h"
|
|
#include "stream/stream.h"
|
|
|
|
struct tl_part {
|
|
char *filename; // what is stream_open()ed
|
|
double offset; // offset into the source file
|
|
double length; // length of the part (-1 if rest of the file)
|
|
bool chapter_ts;
|
|
};
|
|
|
|
struct tl_parts {
|
|
struct tl_part *parts;
|
|
int num_parts;
|
|
};
|
|
|
|
// Parse a time (absolute file time or duration). Currently equivalent to a
|
|
// number. Return false on failure.
|
|
static bool parse_time(bstr str, double *out_time)
|
|
{
|
|
bstr rest;
|
|
double time = bstrtod(str, &rest);
|
|
if (!str.len || rest.len || !isfinite(time))
|
|
return false;
|
|
*out_time = time;
|
|
return true;
|
|
}
|
|
|
|
/* Returns a list of parts, or NULL on parse error.
|
|
* Syntax (without file header or URI prefix):
|
|
* url ::= <entry> ( (';' | '\n') <entry> )*
|
|
* entry ::= <param> ( <param> ',' )*
|
|
* param ::= [<string> '='] (<string> | '%' <number> '%' <bytes>)
|
|
*/
|
|
static struct tl_parts *parse_edl(bstr str)
|
|
{
|
|
struct tl_parts *tl = talloc_zero(NULL, struct tl_parts);
|
|
while (str.len) {
|
|
if (bstr_eatstart0(&str, "#"))
|
|
bstr_split_tok(str, "\n", &(bstr){0}, &str);
|
|
if (bstr_eatstart0(&str, "\n") || bstr_eatstart0(&str, ";"))
|
|
continue;
|
|
struct tl_part p = { .length = -1 };
|
|
int nparam = 0;
|
|
while (1) {
|
|
bstr name, val;
|
|
// Check if it's of the form "name=..."
|
|
int next = bstrcspn(str, "=%,;\n");
|
|
if (next > 0 && next < str.len && str.start[next] == '=') {
|
|
name = bstr_splice(str, 0, next);
|
|
str = bstr_cut(str, next + 1);
|
|
} else {
|
|
const char *names[] = {"file", "start", "length"}; // implied name
|
|
name = bstr0(nparam < 3 ? names[nparam] : "-");
|
|
}
|
|
if (bstr_eatstart0(&str, "%")) {
|
|
int len = bstrtoll(str, &str, 0);
|
|
if (!bstr_startswith0(str, "%") || (len > str.len - 1))
|
|
goto error;
|
|
val = bstr_splice(str, 1, len + 1);
|
|
str = bstr_cut(str, len + 1);
|
|
} else {
|
|
next = bstrcspn(str, ",;\n");
|
|
val = bstr_splice(str, 0, next);
|
|
str = bstr_cut(str, next);
|
|
}
|
|
// Interpret parameters. Explicitly ignore unknown ones.
|
|
if (bstr_equals0(name, "file")) {
|
|
p.filename = bstrto0(tl, val);
|
|
} else if (bstr_equals0(name, "start")) {
|
|
if (!parse_time(val, &p.offset))
|
|
goto error;
|
|
} else if (bstr_equals0(name, "length")) {
|
|
if (!parse_time(val, &p.length))
|
|
goto error;
|
|
} else if (bstr_equals0(name, "timestamps")) {
|
|
if (bstr_equals0(val, "chapters"))
|
|
p.chapter_ts = true;
|
|
}
|
|
nparam++;
|
|
if (!bstr_eatstart0(&str, ","))
|
|
break;
|
|
}
|
|
if (!p.filename)
|
|
goto error;
|
|
MP_TARRAY_APPEND(tl, tl->parts, tl->num_parts, p);
|
|
}
|
|
if (!tl->num_parts)
|
|
goto error;
|
|
return tl;
|
|
error:
|
|
talloc_free(tl);
|
|
return NULL;
|
|
}
|
|
|
|
static struct demuxer *open_file(char *filename, struct MPContext *mpctx)
|
|
{
|
|
struct MPOpts *opts = mpctx->opts;
|
|
struct demuxer *d = NULL;
|
|
struct stream *s = stream_open(filename, mpctx->global);
|
|
if (s) {
|
|
stream_enable_cache_percent(&s,
|
|
opts->stream_cache_size,
|
|
opts->stream_cache_def_size,
|
|
opts->stream_cache_min_percent,
|
|
opts->stream_cache_seek_min_percent);
|
|
d = demux_open(s, NULL, NULL, mpctx->global);
|
|
}
|
|
if (!d) {
|
|
MP_ERR(mpctx, "EDL: Could not open source file '%s'.\n",
|
|
filename);
|
|
free_stream(s);
|
|
}
|
|
return d;
|
|
}
|
|
|
|
static struct demuxer *open_source(struct MPContext *mpctx, char *filename)
|
|
{
|
|
for (int n = 0; n < mpctx->num_sources; n++) {
|
|
struct demuxer *d = mpctx->sources[n];
|
|
if (strcmp(d->stream->url, filename) == 0)
|
|
return d;
|
|
}
|
|
struct demuxer *d = open_file(filename, mpctx);
|
|
if (d)
|
|
MP_TARRAY_APPEND(NULL, mpctx->sources, mpctx->num_sources, d);
|
|
return d;
|
|
}
|
|
|
|
// Append all chapters from src to the chapters array.
|
|
// Ignore chapters outside of the given time range.
|
|
static void copy_chapters(struct chapter **chapters, int *num_chapters,
|
|
struct demuxer *src, double start, double len,
|
|
double dest_offset)
|
|
{
|
|
int count = demuxer_chapter_count(src);
|
|
for (int n = 0; n < count; n++) {
|
|
double time = demuxer_chapter_time(src, n);
|
|
if (time >= start && time <= start + len) {
|
|
struct chapter ch = {
|
|
.start = dest_offset + time - start,
|
|
.name = talloc_steal(*chapters, demuxer_chapter_name(src, n)),
|
|
};
|
|
MP_TARRAY_APPEND(NULL, *chapters, *num_chapters, ch);
|
|
}
|
|
}
|
|
}
|
|
|
|
// return length of the source in seconds, or -1 if unknown
|
|
static double source_get_length(struct demuxer *demuxer)
|
|
{
|
|
double time;
|
|
// <= 0 means DEMUXER_CTRL_NOTIMPL or DEMUXER_CTRL_DONTKNOW
|
|
if (demux_control(demuxer, DEMUXER_CTRL_GET_TIME_LENGTH, &time) <= 0)
|
|
time = -1;
|
|
return time;
|
|
}
|
|
|
|
static void resolve_timestamps(struct tl_part *part, struct demuxer *demuxer)
|
|
{
|
|
if (part->chapter_ts) {
|
|
double start = demuxer_chapter_time(demuxer, part->offset);
|
|
double length = part->length;
|
|
double end = length;
|
|
if (end >= 0)
|
|
end = demuxer_chapter_time(demuxer, part->offset + part->length);
|
|
if (end >= 0 && start >= 0)
|
|
length = end - start;
|
|
part->offset = start;
|
|
part->length = length;
|
|
}
|
|
}
|
|
|
|
static void build_timeline(struct MPContext *mpctx, struct tl_parts *parts)
|
|
{
|
|
struct chapter *chapters = talloc_new(NULL);
|
|
int num_chapters = 0;
|
|
struct timeline_part *timeline = talloc_array_ptrtype(NULL, timeline,
|
|
parts->num_parts + 1);
|
|
double starttime = 0;
|
|
for (int n = 0; n < parts->num_parts; n++) {
|
|
struct tl_part *part = &parts->parts[n];
|
|
struct demuxer *source = open_source(mpctx, part->filename);
|
|
if (!source)
|
|
goto error;
|
|
|
|
resolve_timestamps(part, source);
|
|
|
|
double len = source_get_length(source);
|
|
if (len <= 0) {
|
|
MP_WARN(mpctx, "EDL: source file '%s' has unknown duration.\n",
|
|
part->filename);
|
|
}
|
|
|
|
// Unkown length => use rest of the file. If duration is unknown, make
|
|
// something up.
|
|
if (part->length < 0)
|
|
part->length = (len < 0 ? 1 : len) - part->offset;
|
|
|
|
if (len > 0) {
|
|
double partlen = part->offset + part->length;
|
|
if (partlen > len) {
|
|
MP_WARN(mpctx, "EDL: entry %d uses %f "
|
|
"seconds, but file has only %f seconds.\n",
|
|
n, partlen, len);
|
|
}
|
|
}
|
|
|
|
// Add a chapter between each file.
|
|
struct chapter ch = {
|
|
.start = starttime,
|
|
.name = talloc_strdup(chapters, part->filename),
|
|
};
|
|
MP_TARRAY_APPEND(NULL, chapters, num_chapters, ch);
|
|
|
|
// Also copy the source file's chapters for the relevant parts
|
|
copy_chapters(&chapters, &num_chapters, source, part->offset,
|
|
part->length, starttime);
|
|
|
|
timeline[n] = (struct timeline_part) {
|
|
.start = starttime,
|
|
.source_start = part->offset,
|
|
.source = source,
|
|
};
|
|
|
|
starttime += part->length;
|
|
}
|
|
timeline[parts->num_parts] = (struct timeline_part) {.start = starttime};
|
|
mpctx->timeline = timeline;
|
|
mpctx->num_timeline_parts = parts->num_parts;
|
|
mpctx->chapters = chapters;
|
|
mpctx->num_chapters = num_chapters;
|
|
return;
|
|
|
|
error:
|
|
talloc_free(timeline);
|
|
talloc_free(chapters);
|
|
}
|
|
|
|
// For security, don't allow relative or absolute paths, only plain filenames.
|
|
// Also, make these filenames relative to the edl source file.
|
|
static void fix_filenames(struct tl_parts *parts, char *source_path)
|
|
{
|
|
struct bstr dirname = mp_dirname(source_path);
|
|
for (int n = 0; n < parts->num_parts; n++) {
|
|
struct tl_part *part = &parts->parts[n];
|
|
char *filename = mp_basename(part->filename); // plain filename only
|
|
part->filename = mp_path_join(parts, dirname, bstr0(filename));
|
|
}
|
|
}
|
|
|
|
void build_mpv_edl_timeline(struct MPContext *mpctx)
|
|
{
|
|
struct tl_parts *parts = parse_edl(mpctx->demuxer->file_contents);
|
|
if (!parts) {
|
|
MP_ERR(mpctx, "Error in EDL.\n");
|
|
return;
|
|
}
|
|
// Source is .edl and not edl:// => don't allow arbitrary paths
|
|
if (mpctx->demuxer->stream->uncached_type != STREAMTYPE_EDL)
|
|
fix_filenames(parts, mpctx->demuxer->filename);
|
|
build_timeline(mpctx, parts);
|
|
talloc_free(parts);
|
|
}
|