static const VSFrameRef *_vszimg_get_frame(struct vszimg_data *data, const VSFrameRef *src_frame, VSCore *core, const VSAPI *vsapi, char *err_msg, size_t err_msg_size)
{
struct vszimg_graph_data *graph_data = NULL;
VSFrameRef *dst_frame = NULL;
VSFrameRef *ret = NULL;
void *tmp = NULL;
struct callback_data unpack_cb_data = { 0 };
struct callback_data pack_cb_data = { 0 };
zimg_image_format src_format;
zimg_image_format dst_format;
const VSFormat *src_vsformat;
const VSFormat *dst_vsformat;
const VSMap *src_props;
VSMap *dst_props;
size_t tmp_size;
zimg_image_format_default(&src_format, ZIMG_API_VERSION);
zimg_image_format_default(&dst_format, ZIMG_API_VERSION);
src_props = vsapi->getFramePropsRO(src_frame);
src_vsformat = vsapi->getFrameFormat(src_frame);
dst_vsformat = data->vi.format ? data->vi.format : src_vsformat;
src_format.width = vsapi->getFrameWidth(src_frame, 0);
src_format.height = vsapi->getFrameHeight(src_frame, 0);
dst_format.width = data->vi.width ? (unsigned)data->vi.width : src_format.width;
dst_format.height = data->vi.height ? (unsigned)data->vi.height : src_format.height;
if (translate_vsformat(src_vsformat, &src_format, err_msg))
goto fail;
if (translate_vsformat(dst_vsformat, &dst_format, err_msg))
goto fail;
_vszimg_set_src_colorspace(data, &src_format);
if (import_frame_props(vsapi, src_props, &src_format, err_msg))
goto fail;
_vszimg_set_dst_colorspace(data, &src_format, &dst_format);
if (!(graph_data = _vszimg_get_graph_data(data, &src_format, &dst_format, err_msg, err_msg_size)))
goto fail;
dst_frame = vsapi->newVideoFrame(dst_vsformat, dst_format.width, dst_format.height, src_frame, core);
dst_props = vsapi->getFramePropsRW(dst_frame);
if (callback_init_unpack(&unpack_cb_data, graph_data->graph, src_frame, &src_format, src_vsformat, core, vsapi, err_msg, err_msg_size))
goto fail;
if (callback_init_pack(&pack_cb_data, graph_data->graph, dst_frame, &dst_format, dst_vsformat, core, vsapi, err_msg, err_msg_size))
goto fail;
if (zimg_filter_graph_get_tmp_size(graph_data->graph, &tmp_size)) {
format_zimg_error(err_msg, err_msg_size);
goto fail;
}
VS_ALIGNED_MALLOC(&tmp, tmp_size, 64);
if (!tmp) {
sprintf(err_msg, "error allocating temporary buffer");
goto fail;
}
if (zimg_filter_graph_process(graph_data->graph,
&unpack_cb_data.line_buf.c,
&pack_cb_data.line_buf.m,
tmp,
unpack_cb_data.cb,
&unpack_cb_data,
pack_cb_data.cb,
&pack_cb_data))
{
format_zimg_error(err_msg, err_msg_size);
goto fail;
}
propagate_sar(vsapi, src_props, dst_props, &src_format, &dst_format);
export_frame_props(vsapi, &dst_format, dst_props);
ret = dst_frame;
dst_frame = NULL;
fail:
_vszimg_release_graph_data(data, graph_data);
vsapi->freeFrame(dst_frame);
VS_ALIGNED_FREE(tmp);
callback_destroy(&unpack_cb_data, vsapi);
callback_destroy(&pack_cb_data, vsapi);
return ret;
}
static void VS_CC vszimg_create(const VSMap *in, VSMap *out, void *userData, VSCore *core, const VSAPI *vsapi)
{
struct vszimg_data *data = NULL;
const VSVideoInfo *node_vi;
const VSFormat *node_fmt;
int format_id;
char err_msg[64];
#define FAIL_BAD_VALUE(name) \
do { \
sprintf(err_msg, "%s: bad value", (name)); \
goto fail; \
} while (0)
#define TRY_GET_ENUM(name, out, flag, table) \
do { \
int enum_tmp; \
if (tryGetEnum(vsapi, in, (name), &enum_tmp, &(flag), (table), ARRAY_SIZE((table)))) \
FAIL_BAD_VALUE(name); \
if ((flag)) \
(out) = enum_tmp; \
} while (0)
#define TRY_GET_ENUM_STR(name, out, table) \
do { \
int enum_tmp; \
vszimg_bool flag; \
if (tryGetEnumStr(vsapi, in, (name), &enum_tmp, &flag, (table), ARRAY_SIZE((table)))) \
FAIL_BAD_VALUE(name); \
if ((flag)) \
(out) = enum_tmp; \
} while (0)
if (!(data = malloc(sizeof(*data)))) {
sprintf(err_msg, "error allocating vszimg_data");
goto fail;
}
_vszimg_default_init(data);
if (vszimg_mutex_init(&data->graph_mutex)) {
sprintf(err_msg, "error initializing mutex");
goto fail;
}
data->graph_mutex_initialized = VSZIMG_TRUE;
data->node = vsapi->propGetNode(in, "clip", 0, NULL);
node_vi = vsapi->getVideoInfo(data->node);
node_fmt = node_vi->format;
data->vi = *node_vi;
zimg_graph_builder_params_default(&data->params, ZIMG_API_VERSION);
if (propGetUintDef(vsapi, in, "width", &data->vi.width, node_vi->width))
FAIL_BAD_VALUE("width");
if (propGetUintDef(vsapi, in, "height", &data->vi.height, node_vi->height))
FAIL_BAD_VALUE("height");
if (propGetSintDef(vsapi, in, "format", &format_id, pfNone))
FAIL_BAD_VALUE("format");
data->vi.format = (format_id == pfNone) ? node_fmt : vsapi->getFormatPreset(format_id, core);
TRY_GET_ENUM("matrix", data->matrix, data->have_matrix, g_matrix_table);
TRY_GET_ENUM("transfer", data->transfer, data->have_transfer, g_transfer_table);
TRY_GET_ENUM("primaries", data->primaries, data->have_primaries, g_primaries_table);
TRY_GET_ENUM("range", data->range, data->have_range, g_range_table);
TRY_GET_ENUM("chromaloc", data->chromaloc, data->have_chromaloc, g_chromaloc_table);
TRY_GET_ENUM("matrix_in", data->matrix_in, data->have_matrix_in, g_matrix_table);
TRY_GET_ENUM("transfer_in", data->transfer_in, data->have_transfer_in, g_transfer_table);
TRY_GET_ENUM("primaries_in", data->primaries_in, data->have_primaries_in, g_primaries_table);
TRY_GET_ENUM("range_in", data->range_in, data->have_range_in, g_range_table);
TRY_GET_ENUM("chromaloc_in", data->chromaloc_in, data->have_chromaloc_in, g_chromaloc_table);
TRY_GET_ENUM_STR("resample_filter", data->params.resample_filter, g_resample_filter_table);
data->params.filter_param_a = propGetFloatDef(vsapi, in, "filter_param_a", data->params.filter_param_a);
data->params.filter_param_b = propGetFloatDef(vsapi, in, "filter_param_b", data->params.filter_param_b);
TRY_GET_ENUM_STR("resample_filter_uv", data->params.resample_filter_uv, g_resample_filter_table);
data->params.filter_param_a_uv = propGetFloatDef(vsapi, in, "filter_param_a_uv", data->params.filter_param_a_uv);
data->params.filter_param_b_uv = propGetFloatDef(vsapi, in, "filter_param_b_uv", data->params.filter_param_b_uv);
TRY_GET_ENUM_STR("dither_type", data->params.dither_type, g_dither_type_table);
TRY_GET_ENUM_STR("cpu_type", data->params.cpu_type, g_cpu_type_table);
#undef FAIL_BAD_VALUE
#undef TRY_GET_ENUM
#undef TRY_GET_ENUM_STR
/* Basic compatibility check. */
if (isConstantFormat(node_vi) && isConstantFormat(&data->vi)) {
zimg_image_format src_format;
zimg_image_format dst_format;
zimg_image_format_default(&src_format, ZIMG_API_VERSION);
zimg_image_format_default(&dst_format, ZIMG_API_VERSION);
src_format.width = node_vi->width;
src_format.height = node_vi->height;
dst_format.width = data->vi.width;
dst_format.height = data->vi.height;
if (translate_vsformat(node_vi->format, &src_format, err_msg))
goto fail;
if (translate_vsformat(data->vi.format, &dst_format, err_msg))
goto fail;
}
vsapi->createFilter(in, out, "format", vszimg_init, vszimg_get_frame, vszimg_free, fmParallel, 0, data, core);
return;
fail:
vsapi->setError(out, err_msg);
_vszimg_destroy(data, vsapi);
free(data);
}
static const VSFrameRef * VS_CC vszimg_get_frame(int n, int activationReason, void **instanceData, void **frameData, VSFrameContext *frameCtx, VSCore *core, const VSAPI *vsapi)
{
struct vszimg_data *data = *instanceData;
struct vszimg_graph_data *graph_data = NULL;
const VSFrameRef *src_frame = NULL;
VSFrameRef *dst_frame = NULL;
VSFrameRef *ret = NULL;
void *tmp = NULL;
char err_msg[1024];
int err_flag = 1;
if (activationReason == arInitial) {
vsapi->requestFrameFilter(n, data->node, frameCtx);
} else if (activationReason == arAllFramesReady) {
zimg_image_buffer_const src_buf = { ZIMG_API_VERSION };
zimg_image_buffer dst_buf = { { ZIMG_API_VERSION } };
zimg_image_format src_format;
zimg_image_format dst_format;
const VSFormat *src_vsformat;
const VSFormat *dst_vsformat;
const VSMap *src_props;
VSMap *dst_props;
size_t tmp_size;
int p;
zimg2_image_format_default(&src_format, ZIMG_API_VERSION);
zimg2_image_format_default(&dst_format, ZIMG_API_VERSION);
src_frame = vsapi->getFrameFilter(n, data->node, frameCtx);
src_props = vsapi->getFramePropsRO(src_frame);
src_vsformat = vsapi->getFrameFormat(src_frame);
dst_vsformat = data->vi.format ? data->vi.format : src_vsformat;
src_format.width = vsapi->getFrameWidth(src_frame, 0);
src_format.height = vsapi->getFrameHeight(src_frame, 0);
dst_format.width = data->vi.width ? (unsigned)data->vi.width : src_format.width;
dst_format.height = data->vi.height ? (unsigned)data->vi.height : src_format.height;
if (translate_vsformat(src_vsformat, &src_format, err_msg))
goto fail;
if (translate_vsformat(dst_vsformat, &dst_format, err_msg))
goto fail;
_vszimg_set_src_colorspace(data, &src_format);
if (import_frame_props(vsapi, src_props, &src_format, err_msg))
goto fail;
_vszimg_set_dst_colorspace(data, &src_format, &dst_format);
if (!(graph_data = _vszimg_get_graph_data(data, &src_format, &dst_format, err_msg, sizeof(err_msg))))
goto fail;
dst_frame = vsapi->newVideoFrame(dst_vsformat, dst_format.width, dst_format.height, src_frame, core);
dst_props = vsapi->getFramePropsRW(dst_frame);
for (p = 0; p < src_vsformat->numPlanes; ++p) {
src_buf.data[p] = vsapi->getReadPtr(src_frame, p);
src_buf.stride[p] = vsapi->getStride(src_frame, p);
src_buf.mask[p] = -1;
}
for (p = 0; p < dst_vsformat->numPlanes; ++p) {
dst_buf.m.data[p] = vsapi->getWritePtr(dst_frame, p);
dst_buf.m.stride[p] = vsapi->getStride(dst_frame, p);
dst_buf.m.mask[p] = -1;
}
if (zimg2_filter_graph_get_tmp_size(graph_data->graph, &tmp_size)) {
format_zimg_error(err_msg, sizeof(err_msg));
goto fail;
}
VS_ALIGNED_MALLOC(&tmp, tmp_size, 64);
if (!tmp) {
sprintf(err_msg, "error allocating temporary buffer");
goto fail;
}
if (zimg2_filter_graph_process(graph_data->graph, &src_buf, &dst_buf, tmp, NULL, NULL, NULL, NULL)) {
format_zimg_error(err_msg, sizeof(err_msg));
goto fail;
}
propagate_sar(vsapi, src_props, dst_props, &src_format, &dst_format);
export_frame_props(vsapi, &dst_format, dst_props);
ret = dst_frame;
dst_frame = NULL;
}
err_flag = 0;
fail:
if (err_flag)
vsapi->setFilterError(err_msg, frameCtx);
_vszimg_release_graph_data(data, graph_data);
vsapi->freeFrame(src_frame);
vsapi->freeFrame(dst_frame);
return ret;
}
