47 #define ZIMG_ALIGNMENT 64
48 #define MIN_TILESIZE 64
49 #define MAX_THREADS 64
158 zimg_image_format_default(&
s->src_format, ZIMG_API_VERSION);
159 zimg_image_format_default(&
s->dst_format, ZIMG_API_VERSION);
160 zimg_image_format_default(&
s->src_format_tmp, ZIMG_API_VERSION);
161 zimg_image_format_default(&
s->dst_format_tmp, ZIMG_API_VERSION);
163 zimg_image_format_default(&
s->alpha_src_format, ZIMG_API_VERSION);
164 zimg_image_format_default(&
s->alpha_dst_format, ZIMG_API_VERSION);
165 zimg_image_format_default(&
s->alpha_src_format_tmp, ZIMG_API_VERSION);
166 zimg_image_format_default(&
s->alpha_dst_format_tmp, ZIMG_API_VERSION);
168 zimg_graph_builder_params_default(&
s->params, ZIMG_API_VERSION);
169 zimg_graph_builder_params_default(&
s->params_tmp, ZIMG_API_VERSION);
170 zimg_graph_builder_params_default(&
s->alpha_params, ZIMG_API_VERSION);
171 zimg_graph_builder_params_default(&
s->alpha_params_tmp, ZIMG_API_VERSION);
173 if (
s->size_str && (
s->w_expr ||
s->h_expr)) {
175 "Size and width/height expressions cannot be set at the same time.\n");
179 if (
s->w_expr && !
s->h_expr)
180 FFSWAP(
char *,
s->w_expr,
s->size_str);
186 "Invalid size '%s'\n",
s->size_str);
189 snprintf(buf,
sizeof(buf)-1,
"%d",
s->w);
191 snprintf(buf,
sizeof(buf)-1,
"%d",
s->h);
238 slice_size = (out_h + (
s->nb_threads / 2)) /
s->nb_threads;
241 s->out_slice_start[0] = 0;
242 s->out_slice_end[0] =
FFMIN(out_h, slice_size);
243 for (
int i = 1;
i <
s->nb_threads - 1;
i++) {
244 s->out_slice_start[
i] =
s->out_slice_end[
i-1];
245 s->out_slice_end[
i] =
s->out_slice_start[
i] + slice_size;
248 if (
s->nb_threads > 1) {
249 s->out_slice_start[
s->nb_threads - 1] =
s->out_slice_end[
s->nb_threads - 2];
250 s->out_slice_end[
s->nb_threads - 1] = out_h;
253 for (
int i = 0;
i <
s->nb_threads;
i++) {
254 s->in_slice_start[
i] =
s->out_slice_start[
i] * in_h / (
double)out_h;
255 s->in_slice_end[
i] =
s->out_slice_end[
i] * in_h / (
double)out_h;
267 double var_values[
VARS_NB], res;
270 int factor_w, factor_h;
278 (
double)
inlink->sample_aspect_ratio.num /
inlink->sample_aspect_ratio.den : 1;
334 if (
s->force_original_aspect_ratio) {
338 if (
s->force_original_aspect_ratio == 1) {
347 if (
w > INT_MAX ||
h > INT_MAX ||
357 if (
inlink->sample_aspect_ratio.num){
364 inlink->sample_aspect_ratio.num,
inlink->sample_aspect_ratio.den,
371 "Error when evaluating the expression '%s'.\n"
372 "Maybe the expression for out_w:'%s' or for out_h:'%s' is self-referencing.\n",
373 expr,
s->w_expr,
s->h_expr);
380 int err_code = zimg_get_last_error(err_msg,
sizeof(err_msg));
389 switch (chroma_location) {
392 return ZIMG_CHROMA_LEFT;
394 return ZIMG_CHROMA_CENTER;
396 return ZIMG_CHROMA_TOP_LEFT;
398 return ZIMG_CHROMA_TOP;
400 return ZIMG_CHROMA_BOTTOM_LEFT;
402 return ZIMG_CHROMA_BOTTOM;
404 return ZIMG_CHROMA_LEFT;
409 switch (colorspace) {
411 return ZIMG_MATRIX_RGB;
413 return ZIMG_MATRIX_709;
415 return ZIMG_MATRIX_UNSPECIFIED;
417 return ZIMG_MATRIX_FCC;
419 return ZIMG_MATRIX_470BG;
421 return ZIMG_MATRIX_170M;
423 return ZIMG_MATRIX_240M;
425 return ZIMG_MATRIX_YCGCO;
427 return ZIMG_MATRIX_2020_NCL;
429 return ZIMG_MATRIX_2020_CL;
431 return ZIMG_MATRIX_CHROMATICITY_DERIVED_NCL;
433 return ZIMG_MATRIX_CHROMATICITY_DERIVED_CL;
435 return ZIMG_MATRIX_ICTCP;
437 return ZIMG_MATRIX_UNSPECIFIED;
444 return ZIMG_TRANSFER_UNSPECIFIED;
446 return ZIMG_TRANSFER_709;
448 return ZIMG_TRANSFER_470_M;
450 return ZIMG_TRANSFER_470_BG;
452 return ZIMG_TRANSFER_601;
454 return ZIMG_TRANSFER_240M;
456 return ZIMG_TRANSFER_LINEAR;
458 return ZIMG_TRANSFER_LOG_100;
460 return ZIMG_TRANSFER_LOG_316;
462 return ZIMG_TRANSFER_IEC_61966_2_4;
464 return ZIMG_TRANSFER_2020_10;
466 return ZIMG_TRANSFER_2020_12;
468 return ZIMG_TRANSFER_ST2084;
470 return ZIMG_TRANSFER_ARIB_B67;
472 return ZIMG_TRANSFER_IEC_61966_2_1;
474 return ZIMG_TRANSFER_UNSPECIFIED;
481 return ZIMG_PRIMARIES_UNSPECIFIED;
483 return ZIMG_PRIMARIES_709;
485 return ZIMG_PRIMARIES_470_M;
487 return ZIMG_PRIMARIES_470_BG;
489 return ZIMG_PRIMARIES_170M;
491 return ZIMG_PRIMARIES_240M;
493 return ZIMG_PRIMARIES_FILM;
495 return ZIMG_PRIMARIES_2020;
497 return ZIMG_PRIMARIES_ST428;
499 return ZIMG_PRIMARIES_ST431_2;
501 return ZIMG_PRIMARIES_ST432_1;
503 return ZIMG_PRIMARIES_EBU3213_E;
505 return ZIMG_PRIMARIES_UNSPECIFIED;
513 return ZIMG_RANGE_LIMITED;
515 return ZIMG_RANGE_FULL;
517 return ZIMG_RANGE_LIMITED;
523 case ZIMG_RANGE_LIMITED:
525 case ZIMG_RANGE_FULL:
534 return ((img_fmt0->chroma_location != img_fmt1->chroma_location) ||
535 #
if ZIMG_API_VERSION >= 0x204
536 (img_fmt0->alpha != img_fmt1->alpha) ||
538 (img_fmt0->color_family != img_fmt1->color_family) ||
539 (img_fmt0->color_primaries != img_fmt1->color_primaries) ||
540 (img_fmt0->depth != img_fmt1->depth) ||
541 (img_fmt0->field_parity != img_fmt1->field_parity) ||
542 (img_fmt0->height != img_fmt1->height) ||
543 (img_fmt0->matrix_coefficients != img_fmt1->matrix_coefficients) ||
544 (img_fmt0->pixel_range != img_fmt1->pixel_range) ||
545 (img_fmt0->pixel_type != img_fmt1->pixel_type) ||
546 (img_fmt0->subsample_h != img_fmt1->subsample_h) ||
547 (img_fmt0->subsample_w != img_fmt1->subsample_w) ||
548 (img_fmt0->transfer_characteristics != img_fmt1->transfer_characteristics) ||
549 (img_fmt0->width != img_fmt1->width));
557 int ret = (parm0->allow_approximate_gamma != parm1->allow_approximate_gamma) ||
558 (parm0->dither_type != parm1->dither_type) ||
559 (parm0->resample_filter != parm1->resample_filter) ||
560 (parm0->resample_filter_uv != parm1->resample_filter_uv);
562 if ((
isnan(parm0->nominal_peak_luminance) == 0) || (
isnan(parm1->nominal_peak_luminance) == 0))
563 ret =
ret || (parm0->nominal_peak_luminance != parm1->nominal_peak_luminance);
564 if ((
isnan(parm0->filter_param_a) == 0) || (
isnan(parm1->filter_param_a) == 0))
565 ret =
ret || (parm0->filter_param_a != parm1->filter_param_a);
566 if ((
isnan(parm0->filter_param_a_uv) == 0) || (
isnan(parm1->filter_param_a_uv) == 0))
567 ret =
ret || (parm0->filter_param_a_uv != parm1->filter_param_a_uv);
568 if ((
isnan(parm0->filter_param_b) == 0) || (
isnan(parm1->filter_param_b) == 0))
569 ret =
ret || (parm0->filter_param_b != parm1->filter_param_b);
570 if ((
isnan(parm0->filter_param_b_uv) == 0) || (
isnan(parm1->filter_param_b_uv) == 0))
571 ret =
ret || (parm0->filter_param_b_uv != parm1->filter_param_b_uv);
577 int colorspace,
int primaries,
int transfer,
int range,
int location)
599 zimg_image_format src_format;
600 zimg_image_format dst_format;
601 zimg_image_format alpha_src_format;
602 zimg_image_format alpha_dst_format;
603 const double in_slice_start =
s->in_slice_start[job_nr];
604 const double in_slice_end =
s->in_slice_end[job_nr];
605 const int out_slice_start =
s->out_slice_start[job_nr];
606 const int out_slice_end =
s->out_slice_end[job_nr];
608 src_format =
s->src_format;
609 dst_format =
s->dst_format;
613 src_format.active_region.width = in->
width;
614 src_format.active_region.height = in_slice_end - in_slice_start;
615 src_format.active_region.left = 0;
616 src_format.active_region.top = in_slice_start;
618 dst_format.width =
out->width;
619 dst_format.height = out_slice_end - out_slice_start;
621 if (
s->graph[job_nr]) {
622 zimg_filter_graph_free(
s->graph[job_nr]);
624 s->graph[job_nr] = zimg_filter_graph_build(&src_format, &dst_format, &
s->params);
625 if (!
s->graph[job_nr])
628 ret = zimg_filter_graph_get_tmp_size(
s->graph[job_nr], &
size);
639 alpha_src_format =
s->alpha_src_format;
640 alpha_dst_format =
s->alpha_dst_format;
643 alpha_src_format.active_region.width = in->
width;
644 alpha_src_format.active_region.height = in_slice_end - in_slice_start;
645 alpha_src_format.active_region.left = 0;
646 alpha_src_format.active_region.top = in_slice_start;
648 alpha_dst_format.width =
out->width;
649 alpha_dst_format.height = out_slice_end - out_slice_start;
651 if (
s->alpha_graph[job_nr]) {
652 zimg_filter_graph_free(
s->alpha_graph[job_nr]);
654 s->alpha_graph[job_nr] = zimg_filter_graph_build(&alpha_src_format, &alpha_dst_format, &
s->alpha_params);
655 if (!
s->alpha_graph[job_nr])
668 for (plane = 0; plane <
planes; plane++) {
669 int p =
desc->comp[plane].plane;
676 aligned->format = (*frame)->format;
677 aligned->width = (*frame)->width;
678 aligned->height = (*frame)->height;
702 if (
s->colorspace != -1)
703 frame->colorspace = (
int)
s->dst_format.matrix_coefficients;
705 if (
s->primaries != -1)
706 frame->color_primaries = (
int)
s->dst_format.color_primaries;
712 frame->color_trc = (
int)
s->dst_format.transfer_characteristics;
714 if (
s->chromal != -1)
715 frame->chroma_location = (
int)
s->dst_format.chroma_location + 1;
725 zimg_image_buffer_const src_buf = { ZIMG_API_VERSION };
726 zimg_image_buffer dst_buf = { ZIMG_API_VERSION };
727 const int out_slice_start =
s->out_slice_start[job_nr];
744 for (
int i = 0;
i < 3;
i++) {
745 const int vsamp =
i >= 1 ?
td->odesc->log2_chroma_h : 0;
747 p =
td->desc->comp[
i].plane;
749 src_buf.plane[
i].data =
td->in->data[p];
750 src_buf.plane[
i].stride =
td->in->linesize[p];
751 src_buf.plane[
i].mask = -1;
753 p =
td->odesc->comp[
i].plane;
754 dst_buf.plane[
i].data =
td->out->data[p] +
td->out->linesize[p] * (out_slice_start >> vsamp);
755 dst_buf.plane[
i].stride =
td->out->linesize[p];
756 dst_buf.plane[
i].mask = -1;
758 if (!
s->graph[job_nr])
760 ret = zimg_filter_graph_process(
s->graph[job_nr], &src_buf, &dst_buf,
s->tmp[job_nr], 0, 0, 0, 0);
765 src_buf.plane[0].data =
td->in->data[3];
766 src_buf.plane[0].stride =
td->in->linesize[3];
767 src_buf.plane[0].mask = -1;
769 dst_buf.plane[0].data =
td->out->data[3] +
td->out->linesize[3] * out_slice_start;
770 dst_buf.plane[0].stride =
td->out->linesize[3];
771 dst_buf.plane[0].mask = -1;
773 if (!
s->alpha_graph[job_nr])
775 ret = zimg_filter_graph_process(
s->alpha_graph[job_nr], &src_buf, &dst_buf,
s->tmp[job_nr], 0, 0, 0, 0);
797 (
link->w != outlink->
w) ||
798 (
link->h != outlink->
h) ||
800 (
s->src_format.chroma_location !=
s->dst_format.chroma_location) ||
801 (
s->src_format.color_family !=
s->dst_format.color_family) ||
802 (
s->src_format.color_primaries !=
s->dst_format.color_primaries) ||
803 (
s->src_format.depth !=
s->dst_format.depth) ||
804 (
s->src_format.matrix_coefficients !=
s->dst_format.matrix_coefficients) ||
805 (
s->src_format.field_parity !=
s->dst_format.field_parity) ||
806 (
s->src_format.pixel_range !=
s->dst_format.pixel_range) ||
807 (
s->src_format.pixel_type !=
s->dst_format.pixel_type) ||
808 (
s->src_format.transfer_characteristics !=
s->dst_format.transfer_characteristics)
824 snprintf(buf,
sizeof(buf)-1,
"%d", outlink->
w);
826 snprintf(buf,
sizeof(buf)-1,
"%d", outlink->
h);
838 s->out_colorspace =
out->colorspace;
839 s->out_trc =
out->color_trc;
840 s->out_primaries =
out->color_primaries;
841 s->out_range =
out->color_range;
845 zimg_image_format_default(&
s->src_format, ZIMG_API_VERSION);
846 zimg_image_format_default(&
s->dst_format, ZIMG_API_VERSION);
847 zimg_graph_builder_params_default(&
s->params, ZIMG_API_VERSION);
850 s->primaries_in,
s->trc_in,
s->range_in,
s->chromal_in);
852 s->primaries,
s->trc,
s->range,
s->chromal);
855 s->params.dither_type =
s->dither;
856 s->params.cpu_type = ZIMG_CPU_AUTO;
857 s->params.resample_filter =
s->filter;
858 s->params.resample_filter_uv =
s->filter;
859 s->params.nominal_peak_luminance =
s->nominal_peak_luminance;
860 s->params.allow_approximate_gamma =
s->approximate_gamma;
861 s->params.filter_param_a =
s->params.filter_param_a_uv =
s->param_a;
862 s->params.filter_param_b =
s->params.filter_param_b_uv =
s->param_b;
865 zimg_image_format_default(&
s->alpha_src_format, ZIMG_API_VERSION);
866 zimg_image_format_default(&
s->alpha_dst_format, ZIMG_API_VERSION);
867 zimg_graph_builder_params_default(&
s->alpha_params, ZIMG_API_VERSION);
869 s->alpha_params.dither_type =
s->dither;
870 s->alpha_params.cpu_type = ZIMG_CPU_AUTO;
871 s->alpha_params.resample_filter =
s->filter;
873 s->alpha_src_format.width = in->
width;
874 s->alpha_src_format.height = in->
height;
875 s->alpha_src_format.depth =
desc->comp[0].depth;
876 s->alpha_src_format.pixel_type = (
desc->flags &
AV_PIX_FMT_FLAG_FLOAT) ? ZIMG_PIXEL_FLOAT :
desc->comp[0].depth > 8 ? ZIMG_PIXEL_WORD : ZIMG_PIXEL_BYTE;
877 s->alpha_src_format.color_family = ZIMG_COLOR_GREY;
879 s->alpha_dst_format.depth = odesc->
comp[0].
depth;
881 s->alpha_dst_format.color_family = ZIMG_COLOR_GREY;
885 av_reduce(&
out->sample_aspect_ratio.num, &
out->sample_aspect_ratio.den,
895 memset(
s->jobs_ret, 0,
s->nb_threads *
sizeof(*
s->jobs_ret));
897 for (
int i = 0;
ret >= 0 &&
i <
s->nb_threads;
i++)
898 if (
s->jobs_ret[
i] < 0)
899 ret =
s->jobs_ret[
i];
906 s->src_format_tmp =
s->src_format;
907 s->dst_format_tmp =
s->dst_format;
908 s->params_tmp =
s->params;
910 s->alpha_src_format_tmp =
s->alpha_src_format;
911 s->alpha_dst_format_tmp =
s->alpha_dst_format;
912 s->alpha_params_tmp =
s->alpha_params;
918 for (y = 0; y <
out->height; y++) {
919 for (x = 0; x <
out->width; x++) {
925 for (y = 0; y < outlink->
h; y++)
926 memset(
out->data[3] + y *
out->linesize[3], 0xff, outlink->
w);
947 for (
int i = 0;
i <
s->nb_threads;
i++) {
950 zimg_filter_graph_free(
s->graph[
i]);
953 if (
s->alpha_graph[
i]) {
954 zimg_filter_graph_free(
s->alpha_graph[
i]);
961 char *res,
int res_len,
int flags)
966 if ( !strcmp(cmd,
"width") || !strcmp(cmd,
"w")
967 || !strcmp(cmd,
"height") || !strcmp(cmd,
"h")) {
984 #define OFFSET(x) offsetof(ZScaleContext, x)
985 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
986 #define TFLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
1000 {
"error_diffusion", 0, 0,
AV_OPT_TYPE_CONST, {.i64 = ZIMG_DITHER_ERROR_DIFFUSION}, 0, 0,
FLAGS,
"dither" },
1022 {
"unspecified", 0, 0,
AV_OPT_TYPE_CONST, {.i64 = ZIMG_PRIMARIES_UNSPECIFIED}, 0, 0,
FLAGS,
"primaries" },
1059 {
"iec61966-2-4", 0, 0,
AV_OPT_TYPE_CONST, {.i64 = ZIMG_TRANSFER_IEC_61966_2_4},0, 0,
FLAGS,
"transfer" },
1060 {
"iec61966-2-1", 0, 0,
AV_OPT_TYPE_CONST, {.i64 = ZIMG_TRANSFER_IEC_61966_2_1},0, 0,
FLAGS,
"transfer" },
1081 {
"chroma-derived-nc",0, 0,
AV_OPT_TYPE_CONST, {.i64 = ZIMG_MATRIX_CHROMATICITY_DERIVED_NCL}, 0, 0,
FLAGS,
"matrix" },
1082 {
"chroma-derived-c", 0, 0,
AV_OPT_TYPE_CONST, {.i64 = ZIMG_MATRIX_CHROMATICITY_DERIVED_CL}, 0, 0,
FLAGS,
"matrix" },
1084 {
"in_range",
"set input color range",
OFFSET(range_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, ZIMG_RANGE_FULL,
FLAGS,
"range" },
1087 {
"primariesin",
"set input color primaries",
OFFSET(primaries_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
FLAGS,
"primaries" },
1089 {
"transferin",
"set input transfer characteristic",
OFFSET(trc_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
FLAGS,
"transfer" },
1090 {
"tin",
"set input transfer characteristic",
OFFSET(trc_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
FLAGS,
"transfer" },
1091 {
"matrixin",
"set input colorspace matrix",
OFFSET(colorspace_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
FLAGS,
"matrix" },
1093 {
"chromal",
"set output chroma location",
OFFSET(chromal),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, ZIMG_CHROMA_BOTTOM,
FLAGS,
"chroma" },
1094 {
"c",
"set output chroma location",
OFFSET(chromal),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, ZIMG_CHROMA_BOTTOM,
FLAGS,
"chroma" },
1102 {
"chromalin",
"set input chroma location",
OFFSET(chromal_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, ZIMG_CHROMA_BOTTOM,
FLAGS,
"chroma" },
1103 {
"cin",
"set input chroma location",
OFFSET(chromal_in),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, ZIMG_CHROMA_BOTTOM,
FLAGS,
"chroma" },
1106 {
"param_a",
"parameter A, which is parameter \"b\" for bicubic, "
1135 .priv_class = &zscale_class,