2014-04-12 15:51:19 +00:00
|
|
|
/*
|
|
|
|
* This file is part of mpv.
|
|
|
|
*
|
|
|
|
* mpv 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.
|
|
|
|
*
|
|
|
|
* mpv 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 mpv. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <inttypes.h>
|
|
|
|
#include <pthread.h>
|
|
|
|
#include <assert.h>
|
|
|
|
|
|
|
|
#include <VapourSynth.h>
|
|
|
|
#include <VSScript.h>
|
|
|
|
#include <VSHelper.h>
|
|
|
|
|
|
|
|
#include "common/msg.h"
|
|
|
|
#include "options/m_option.h"
|
|
|
|
|
|
|
|
#include "video/img_format.h"
|
|
|
|
#include "video/mp_image.h"
|
|
|
|
#include "video/sws_utils.h"
|
|
|
|
#include "vf.h"
|
|
|
|
|
|
|
|
struct vf_priv_s {
|
|
|
|
VSCore *vscore;
|
|
|
|
const VSAPI *vsapi;
|
|
|
|
VSScript *se;
|
|
|
|
VSNodeRef *out_node;
|
|
|
|
VSNodeRef *in_node;
|
|
|
|
|
2014-04-14 16:03:13 +00:00
|
|
|
struct mp_image_params fmt_in;
|
2014-04-12 15:51:19 +00:00
|
|
|
|
|
|
|
pthread_mutex_t lock;
|
|
|
|
pthread_cond_t wakeup;
|
|
|
|
|
|
|
|
// --- the following members are all protected by lock
|
2014-04-28 20:22:55 +00:00
|
|
|
struct mp_image *next_image;// used to compute frame duration of oldest image
|
2014-04-12 15:51:19 +00:00
|
|
|
struct mp_image **buffered; // oldest image first
|
|
|
|
int num_buffered;
|
2014-04-28 20:22:55 +00:00
|
|
|
double prev_pts; // pts of last frame returned
|
2014-04-12 15:51:19 +00:00
|
|
|
int in_frameno; // frame number of buffered[0] (the oldest)
|
|
|
|
int out_frameno; // frame number of last requested frame
|
|
|
|
bool getting_frame; // getAsyncFrame is in progress
|
|
|
|
struct mp_image *got_frame; // frame callback result
|
|
|
|
bool failed; // frame callback returned with an error
|
|
|
|
bool shutdown; // ask node to return
|
|
|
|
bool in_node_active; // node might still be called
|
|
|
|
|
|
|
|
// --- options
|
|
|
|
char *cfg_file;
|
|
|
|
int cfg_maxbuffer;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct mpvs_fmt {
|
|
|
|
VSPresetFormat vs;
|
|
|
|
int mp;
|
|
|
|
};
|
|
|
|
|
|
|
|
static const struct mpvs_fmt mpvs_fmt_table[] = {
|
|
|
|
{pfYUV420P8, IMGFMT_420P},
|
|
|
|
{pfYUV422P8, IMGFMT_422P},
|
|
|
|
{pfYUV444P8, IMGFMT_444P},
|
|
|
|
{pfYUV410P8, IMGFMT_410P},
|
|
|
|
{pfYUV411P8, IMGFMT_411P},
|
|
|
|
{pfYUV440P8, IMGFMT_440P},
|
|
|
|
{pfYUV420P9, IMGFMT_420P9},
|
|
|
|
{pfYUV422P9, IMGFMT_422P9},
|
|
|
|
{pfYUV444P9, IMGFMT_444P9},
|
|
|
|
{pfYUV420P10, IMGFMT_420P10},
|
|
|
|
{pfYUV422P10, IMGFMT_422P10},
|
|
|
|
{pfYUV444P10, IMGFMT_444P10},
|
|
|
|
{pfYUV420P16, IMGFMT_420P16},
|
|
|
|
{pfYUV422P16, IMGFMT_422P16},
|
|
|
|
{pfYUV444P16, IMGFMT_444P16},
|
|
|
|
{pfNone}
|
|
|
|
};
|
|
|
|
|
|
|
|
static VSPresetFormat mp_to_vs(int imgfmt)
|
|
|
|
{
|
|
|
|
for (int n = 0; mpvs_fmt_table[n].mp; n++) {
|
|
|
|
if (mpvs_fmt_table[n].mp == imgfmt)
|
|
|
|
return mpvs_fmt_table[n].vs;
|
|
|
|
}
|
|
|
|
return pfNone;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int mp_from_vs(VSPresetFormat vs)
|
|
|
|
{
|
|
|
|
for (int n = 0; mpvs_fmt_table[n].mp; n++) {
|
|
|
|
if (mpvs_fmt_table[n].vs == vs)
|
|
|
|
return mpvs_fmt_table[n].mp;
|
|
|
|
}
|
|
|
|
return pfNone;
|
|
|
|
}
|
|
|
|
|
2014-04-28 21:41:20 +00:00
|
|
|
static void copy_mp_to_vs_frame_props(struct vf_priv_s *p, VSMap *map,
|
|
|
|
struct mp_image *img)
|
|
|
|
{
|
|
|
|
struct mp_image_params *params = &img->params;
|
|
|
|
if (params->d_w > 0 && params->d_h > 0) {
|
|
|
|
p->vsapi->propSetInt(map, "_SARNum", params->d_w, 0);
|
|
|
|
p->vsapi->propSetInt(map, "_SARDen", params->d_h, 0);
|
|
|
|
}
|
|
|
|
if (params->colorlevels) {
|
|
|
|
p->vsapi->propSetInt(map, "_ColorRange",
|
|
|
|
params->colorlevels == MP_CSP_LEVELS_TV, 0);
|
|
|
|
}
|
|
|
|
// The docs explicitly say it uses libavcodec values.
|
|
|
|
p->vsapi->propSetInt(map, "_ColorSpace",
|
|
|
|
mp_csp_to_avcol_spc(params->colorspace), 0);
|
|
|
|
char pict_type = 0;
|
|
|
|
switch (img->pict_type) {
|
|
|
|
case 1: pict_type = 'I'; break;
|
|
|
|
case 2: pict_type = 'P'; break;
|
|
|
|
case 3: pict_type = 'B'; break;
|
|
|
|
}
|
|
|
|
if (pict_type)
|
|
|
|
p->vsapi->propSetData(map, "_PictType", &pict_type, 1, 0);
|
|
|
|
}
|
|
|
|
|
2014-04-12 15:51:19 +00:00
|
|
|
static struct mp_image map_vs_frame(struct vf_priv_s *p, const VSFrameRef *ref,
|
|
|
|
bool w)
|
|
|
|
{
|
|
|
|
const VSFormat *fmt = p->vsapi->getFrameFormat(ref);
|
|
|
|
|
|
|
|
struct mp_image img = {0};
|
|
|
|
mp_image_setfmt(&img, mp_from_vs(fmt->id));
|
|
|
|
mp_image_set_size(&img, p->vsapi->getFrameWidth(ref, 0),
|
|
|
|
p->vsapi->getFrameHeight(ref, 0));
|
|
|
|
|
|
|
|
for (int n = 0; n < img.num_planes; n++) {
|
|
|
|
if (w) {
|
|
|
|
img.planes[n] = p->vsapi->getWritePtr((VSFrameRef *)ref, n);
|
|
|
|
} else {
|
|
|
|
img.planes[n] = (uint8_t *)p->vsapi->getReadPtr(ref, n);
|
|
|
|
}
|
|
|
|
img.stride[n] = p->vsapi->getStride(ref, n);
|
|
|
|
}
|
|
|
|
|
|
|
|
return img;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void drain_oldest_buffered_frame(struct vf_priv_s *p)
|
|
|
|
{
|
|
|
|
if (!p->num_buffered)
|
|
|
|
return;
|
|
|
|
talloc_free(p->buffered[0]);
|
|
|
|
for (int n = 0; n < p->num_buffered - 1; n++)
|
|
|
|
p->buffered[n] = p->buffered[n + 1];
|
|
|
|
p->num_buffered--;
|
|
|
|
p->in_frameno++;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void VS_CC vs_frame_done(void *userData, const VSFrameRef *f, int n,
|
|
|
|
VSNodeRef *node, const char *errorMsg)
|
|
|
|
{
|
|
|
|
struct vf_instance *vf = userData;
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
|
|
|
|
pthread_mutex_lock(&p->lock);
|
|
|
|
assert(p->getting_frame);
|
|
|
|
assert(!p->got_frame);
|
|
|
|
p->getting_frame = false;
|
|
|
|
|
|
|
|
if (f) {
|
|
|
|
struct mp_image img = map_vs_frame(p, f, false);
|
|
|
|
img.pts = MP_NOPTS_VALUE;
|
|
|
|
const VSMap *map = p->vsapi->getFramePropsRO(f);
|
|
|
|
if (map) {
|
2014-04-28 20:22:55 +00:00
|
|
|
int err1, err2;
|
|
|
|
int num = p->vsapi->propGetInt(map, "_DurationNum", 0, &err1);
|
|
|
|
int den = p->vsapi->propGetInt(map, "_DurationDen", 0, &err2);
|
|
|
|
if (!err1 && !err2 && p->prev_pts != MP_NOPTS_VALUE) {
|
|
|
|
img.pts = p->prev_pts;
|
|
|
|
p->prev_pts += num / (double)den;
|
|
|
|
}
|
2014-04-12 15:51:19 +00:00
|
|
|
}
|
|
|
|
if (img.pts == MP_NOPTS_VALUE)
|
|
|
|
MP_ERR(vf, "No PTS after filter!\n");
|
|
|
|
p->got_frame = mp_image_new_copy(&img);
|
|
|
|
p->vsapi->freeFrame(f);
|
|
|
|
} else {
|
|
|
|
p->failed = true;
|
|
|
|
MP_ERR(vf, "Filter error: %s\n", errorMsg);
|
|
|
|
}
|
|
|
|
pthread_cond_broadcast(&p->wakeup);
|
|
|
|
pthread_mutex_unlock(&p->lock);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int filter_ext(struct vf_instance *vf, struct mp_image *mpi)
|
|
|
|
{
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
int ret = 0;
|
|
|
|
|
2014-04-28 19:45:36 +00:00
|
|
|
if (!p->out_node) {
|
|
|
|
talloc_free(mpi);
|
2014-04-12 15:51:19 +00:00
|
|
|
return -1;
|
2014-04-28 19:45:36 +00:00
|
|
|
}
|
2014-04-12 15:51:19 +00:00
|
|
|
|
2014-04-28 20:22:55 +00:00
|
|
|
MPSWAP(struct mp_image *, p->next_image, mpi);
|
|
|
|
|
video: don't drop last frame when deinterlacing with yadif
Or in other words, add support for properly draining remaining frames
from video filters. vf_yadif is buffering at least one frame, and the
buffered frame was not retrieved on EOF.
For most filters, ignore this for now, and just adjust them to the
changed semantics of filter_ext. But for vf_lavfi (used by vf_yadif),
real support is implemented. libavfilter handles this simply by passing
a NULL frame to av_buffersrc_add_frame(), so we just have to make
mp_to_av() handle NULL arguments.
In load_next_vo_frame(), we first try to output a frame buffered in the
VO, then the filter, and then (if EOF is reached and there's still no
new frame) the VO again, with draining enabled. I guess this was
implemented slightly incorrectly before, because the filter chain still
could have had remaining output frames.
2014-04-28 17:44:35 +00:00
|
|
|
if (!mpi)
|
|
|
|
return 0;
|
|
|
|
|
2014-04-28 20:22:55 +00:00
|
|
|
// Turn PTS into frame duration (the pts field is abused for storing it)
|
|
|
|
if (p->prev_pts == MP_NOPTS_VALUE)
|
|
|
|
p->prev_pts = mpi->pts;
|
|
|
|
mpi->pts = p->next_image ? p->next_image->pts - mpi->pts : 0;
|
|
|
|
|
2014-04-12 15:51:19 +00:00
|
|
|
// Try to get new frames until we get rid of the input mpi.
|
|
|
|
pthread_mutex_lock(&p->lock);
|
|
|
|
while (1) {
|
|
|
|
// Not sure what we do on errors, but at least don't deadlock.
|
|
|
|
if (p->failed) {
|
|
|
|
p->failed = false;
|
|
|
|
talloc_free(mpi);
|
|
|
|
ret = -1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (mpi && p->num_buffered < MP_TALLOC_ELEMS(p->buffered)) {
|
|
|
|
p->buffered[p->num_buffered++] = talloc_steal(p->buffered, mpi);
|
|
|
|
mpi = NULL;
|
|
|
|
pthread_cond_broadcast(&p->wakeup);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (p->got_frame) {
|
|
|
|
vf_add_output_frame(vf, p->got_frame);
|
|
|
|
p->got_frame = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!p->getting_frame) {
|
|
|
|
// Note: this assumes getFrameAsync() will never call infiltGetFrame
|
|
|
|
// (if it does, we would deadlock)
|
|
|
|
p->getting_frame = true;
|
|
|
|
p->failed = false;
|
|
|
|
p->vsapi->getFrameAsync(p->out_frameno++, p->out_node,
|
|
|
|
vs_frame_done, vf);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!mpi)
|
|
|
|
break;
|
|
|
|
pthread_cond_wait(&p->wakeup, &p->lock);
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&p->lock);
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void VS_CC infiltInit(VSMap *in, VSMap *out, void **instanceData,
|
|
|
|
VSNode *node, VSCore *core, const VSAPI *vsapi)
|
|
|
|
{
|
|
|
|
struct vf_instance *vf = *instanceData;
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
|
|
|
|
// Note: this is called from createFilter, so no need for locking.
|
|
|
|
|
2014-04-14 16:03:13 +00:00
|
|
|
VSVideoInfo fmt = {
|
|
|
|
.format = p->vsapi->getFormatPreset(mp_to_vs(p->fmt_in.imgfmt), p->vscore),
|
|
|
|
.width = p->fmt_in.w,
|
|
|
|
.height = p->fmt_in.h,
|
|
|
|
};
|
|
|
|
if (!fmt.format) {
|
|
|
|
p->vsapi->setError(out, "Unsupported input format.\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
p->vsapi->setVideoInfo(&fmt, 1, node);
|
|
|
|
p->in_node_active = true;
|
2014-04-12 15:51:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static const VSFrameRef *VS_CC infiltGetFrame(int frameno, int activationReason,
|
|
|
|
void **instanceData, void **frameData,
|
|
|
|
VSFrameContext *frameCtx, VSCore *core,
|
|
|
|
const VSAPI *vsapi)
|
|
|
|
{
|
|
|
|
struct vf_instance *vf = *instanceData;
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
VSFrameRef *ret = NULL;
|
|
|
|
|
|
|
|
pthread_mutex_lock(&p->lock);
|
|
|
|
while (1) {
|
|
|
|
if (p->shutdown)
|
|
|
|
break;
|
|
|
|
if (frameno < p->in_frameno) {
|
|
|
|
p->vsapi->setFilterError("Requesting a frame too far in the past. "
|
|
|
|
"Try increasing the maxbuffer suboption",
|
|
|
|
frameCtx);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (frameno >= p->in_frameno + MP_TALLOC_ELEMS(p->buffered)) {
|
|
|
|
// Too far in the future. Remove frames, so that the main thread can
|
|
|
|
// queue new frames.
|
|
|
|
if (p->num_buffered) {
|
|
|
|
drain_oldest_buffered_frame(p);
|
|
|
|
pthread_cond_broadcast(&p->wakeup);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (frameno < p->in_frameno + p->num_buffered) {
|
|
|
|
struct mp_image *img = p->buffered[frameno - p->in_frameno];
|
|
|
|
const VSFormat *vsfmt =
|
|
|
|
vsapi->getFormatPreset(mp_to_vs(img->imgfmt), core);
|
|
|
|
ret = vsapi->newVideoFrame(vsfmt, img->w, img->h, NULL, core);
|
|
|
|
if (!ret) {
|
|
|
|
p->vsapi->setFilterError("Could not allocate VS frame", frameCtx);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
struct mp_image vsframe = map_vs_frame(p, ret, true);
|
|
|
|
mp_image_copy(&vsframe, img);
|
|
|
|
VSMap *map = p->vsapi->getFramePropsRW(ret);
|
2014-04-28 20:22:55 +00:00
|
|
|
if (map) {
|
|
|
|
int res = 1e6;
|
|
|
|
int dur = img->pts * res + 0.5;
|
|
|
|
p->vsapi->propSetInt(map, "_DurationNum", dur, 0);
|
|
|
|
p->vsapi->propSetInt(map, "_DurationDen", res, 0);
|
2014-04-28 21:41:20 +00:00
|
|
|
copy_mp_to_vs_frame_props(p, map, img);
|
2014-04-28 20:22:55 +00:00
|
|
|
}
|
2014-04-12 15:51:19 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
pthread_cond_wait(&p->wakeup, &p->lock);
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&p->lock);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void VS_CC infiltFree(void *instanceData, VSCore *core, const VSAPI *vsapi)
|
|
|
|
{
|
|
|
|
struct vf_instance *vf = instanceData;
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
|
|
|
|
pthread_mutex_lock(&p->lock);
|
|
|
|
p->in_node_active = false;
|
|
|
|
pthread_cond_broadcast(&p->wakeup);
|
|
|
|
pthread_mutex_unlock(&p->lock);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void destroy_vs(struct vf_instance *vf)
|
|
|
|
{
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
|
2014-04-14 18:49:14 +00:00
|
|
|
// Wait until our frame callback returns.
|
|
|
|
pthread_mutex_lock(&p->lock);
|
|
|
|
p->shutdown = true;
|
|
|
|
pthread_cond_broadcast(&p->wakeup);
|
|
|
|
while (p->getting_frame)
|
|
|
|
pthread_cond_wait(&p->wakeup, &p->lock);
|
|
|
|
pthread_mutex_unlock(&p->lock);
|
|
|
|
|
2014-04-12 15:51:19 +00:00
|
|
|
if (p->in_node)
|
|
|
|
p->vsapi->freeNode(p->in_node);
|
|
|
|
if (p->out_node)
|
|
|
|
p->vsapi->freeNode(p->out_node);
|
|
|
|
p->in_node = p->out_node = NULL;
|
|
|
|
|
2014-04-14 16:03:13 +00:00
|
|
|
if (p->se)
|
|
|
|
vsscript_freeScript(p->se);
|
|
|
|
|
|
|
|
p->se = NULL;
|
|
|
|
p->vsapi = NULL;
|
|
|
|
p->vscore = NULL;
|
|
|
|
|
|
|
|
assert(!p->in_node_active);
|
|
|
|
|
2014-04-12 15:51:19 +00:00
|
|
|
p->shutdown = false;
|
|
|
|
talloc_free(p->got_frame);
|
|
|
|
p->got_frame = NULL;
|
|
|
|
// Kill queued frames too
|
|
|
|
for (int n = 0; n < p->num_buffered; n++)
|
|
|
|
talloc_free(p->buffered[n]);
|
2014-04-14 16:03:13 +00:00
|
|
|
p->num_buffered = 0;
|
2014-04-28 20:22:55 +00:00
|
|
|
talloc_free(p->next_image);
|
|
|
|
p->next_image = NULL;
|
|
|
|
p->prev_pts = MP_NOPTS_VALUE;
|
2014-04-12 15:51:19 +00:00
|
|
|
p->out_frameno = p->in_frameno = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int reinit_vs(struct vf_instance *vf)
|
|
|
|
{
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
VSMap *vars = NULL, *in = NULL, *out = NULL;
|
|
|
|
int res = -1;
|
|
|
|
|
|
|
|
destroy_vs(vf);
|
|
|
|
|
2014-04-14 16:03:13 +00:00
|
|
|
// First load an empty script to get a VSScript, so that we get the vsapi
|
|
|
|
// and vscore.
|
|
|
|
if (vsscript_evaluateScript(&p->se, "", NULL, 0))
|
|
|
|
goto error;
|
|
|
|
p->vsapi = vsscript_getVSApi();
|
|
|
|
p->vscore = vsscript_getCore(p->se);
|
|
|
|
if (!p->vsapi || !p->vscore)
|
|
|
|
goto error;
|
|
|
|
|
2014-04-12 15:51:19 +00:00
|
|
|
in = p->vsapi->createMap();
|
|
|
|
out = p->vsapi->createMap();
|
|
|
|
vars = p->vsapi->createMap();
|
|
|
|
if (!in || !out || !vars)
|
|
|
|
goto error;
|
|
|
|
|
|
|
|
p->vsapi->createFilter(in, out, "Input", infiltInit, infiltGetFrame,
|
|
|
|
infiltFree, fmSerial, 0, vf, p->vscore);
|
|
|
|
int vserr;
|
|
|
|
p->in_node = p->vsapi->propGetNode(out, "clip", 0, &vserr);
|
|
|
|
if (!p->in_node)
|
|
|
|
goto error;
|
|
|
|
|
|
|
|
if (p->vsapi->propSetNode(vars, "video_in", p->in_node, 0))
|
|
|
|
goto error;
|
|
|
|
|
|
|
|
vsscript_setVariable(p->se, vars);
|
|
|
|
|
|
|
|
if (vsscript_evaluateFile(&p->se, p->cfg_file, 0)) {
|
|
|
|
MP_FATAL(vf, "Script evaluation failed:\n%s\n", vsscript_getError(p->se));
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
p->out_node = vsscript_getOutput(p->se, 0);
|
|
|
|
if (!p->out_node)
|
|
|
|
goto error;
|
|
|
|
|
|
|
|
const VSVideoInfo *vi = p->vsapi->getVideoInfo(p->out_node);
|
|
|
|
if (!isConstantFormat(vi)) {
|
|
|
|
MP_FATAL(vf, "Video format is required to be constant.\n");
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
res = 0;
|
|
|
|
error:
|
2014-04-14 16:03:13 +00:00
|
|
|
if (p->vsapi) {
|
|
|
|
p->vsapi->freeMap(in);
|
|
|
|
p->vsapi->freeMap(out);
|
|
|
|
p->vsapi->freeMap(vars);
|
|
|
|
}
|
2014-04-12 15:51:19 +00:00
|
|
|
if (res < 0)
|
|
|
|
destroy_vs(vf);
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int config(struct vf_instance *vf, int width, int height,
|
|
|
|
int d_width, int d_height, unsigned int flags,
|
|
|
|
unsigned int fmt)
|
|
|
|
{
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
|
2014-04-14 16:03:13 +00:00
|
|
|
p->fmt_in = (struct mp_image_params){
|
|
|
|
.imgfmt = fmt,
|
|
|
|
.w = width,
|
|
|
|
.h = height,
|
2014-04-12 15:51:19 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
if (reinit_vs(vf) < 0)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
const VSVideoInfo *vi = p->vsapi->getVideoInfo(p->out_node);
|
|
|
|
fmt = mp_from_vs(vi->format->id);
|
|
|
|
if (!fmt) {
|
2014-04-13 10:30:57 +00:00
|
|
|
MP_FATAL(vf, "Unsupported output format.\n");
|
2014-04-12 15:51:19 +00:00
|
|
|
destroy_vs(vf);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-04-27 15:05:26 +00:00
|
|
|
vf_rescale_dsize(&d_width, &d_height, width, height, vi->width, vi->height);
|
|
|
|
|
|
|
|
return vf_next_config(vf, vi->width, vi->height, d_width, d_height, flags, fmt);
|
2014-04-12 15:51:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static int query_format(struct vf_instance *vf, unsigned int fmt)
|
|
|
|
{
|
|
|
|
return mp_to_vs(fmt) != pfNone ? VFCAP_CSP_SUPPORTED : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int control(vf_instance_t *vf, int request, void *data)
|
|
|
|
{
|
|
|
|
switch (request) {
|
|
|
|
case VFCTRL_SEEK_RESET:
|
|
|
|
if (reinit_vs(vf) < 0)
|
|
|
|
return CONTROL_ERROR;
|
|
|
|
return CONTROL_OK;
|
|
|
|
}
|
|
|
|
return CONTROL_UNKNOWN;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void uninit(struct vf_instance *vf)
|
|
|
|
{
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
|
|
|
|
|
|
|
destroy_vs(vf);
|
2014-04-14 16:03:13 +00:00
|
|
|
vsscript_finalize();
|
2014-04-12 15:51:19 +00:00
|
|
|
|
|
|
|
pthread_cond_destroy(&p->wakeup);
|
|
|
|
pthread_mutex_destroy(&p->lock);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int vf_open(vf_instance_t *vf)
|
|
|
|
{
|
|
|
|
struct vf_priv_s *p = vf->priv;
|
2014-04-14 16:03:13 +00:00
|
|
|
if (!vsscript_init()) {
|
|
|
|
MP_FATAL(vf, "Could not initialize VapourSynth scripting.\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
if (!p->cfg_file || !p->cfg_file[0]) {
|
|
|
|
MP_FATAL(vf, "'file' parameter must be set.\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-04-12 15:51:19 +00:00
|
|
|
pthread_mutex_init(&p->lock, NULL);
|
|
|
|
pthread_cond_init(&p->wakeup, NULL);
|
|
|
|
vf->reconfig = NULL;
|
|
|
|
vf->config = config;
|
|
|
|
vf->filter_ext = filter_ext;
|
|
|
|
vf->filter = NULL;
|
|
|
|
vf->query_format = query_format;
|
|
|
|
vf->control = control;
|
|
|
|
vf->uninit = uninit;
|
|
|
|
p->buffered = talloc_array(vf, struct mp_image *, p->cfg_maxbuffer);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
#define OPT_BASE_STRUCT struct vf_priv_s
|
|
|
|
static const m_option_t vf_opts_fields[] = {
|
|
|
|
OPT_STRING("file", cfg_file, 0),
|
|
|
|
OPT_INTRANGE("maxbuffer", cfg_maxbuffer, 0, 1, 9999, OPTDEF_INT(5)),
|
|
|
|
{0}
|
|
|
|
};
|
|
|
|
|
|
|
|
const vf_info_t vf_info_vapoursynth = {
|
|
|
|
.description = "vapoursynth bridge",
|
|
|
|
.name = "vapoursynth",
|
|
|
|
.open = vf_open,
|
|
|
|
.priv_size = sizeof(struct vf_priv_s),
|
|
|
|
.options = vf_opts_fields,
|
|
|
|
};
|