FFmpeg
vf_scale_npp.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 /**
20  * @file
21  * scale video filter
22  */
23 
24 #include <nppi.h>
25 #include <stdio.h>
26 #include <string.h>
27 
28 #include "libavutil/avstring.h"
29 #include "libavutil/common.h"
30 #include "libavutil/hwcontext.h"
32 #include "libavutil/cuda_check.h"
33 #include "libavutil/internal.h"
34 #include "libavutil/opt.h"
35 #include "libavutil/pixdesc.h"
36 
37 #include "avfilter.h"
38 #include "formats.h"
39 #include "internal.h"
40 #include "scale_eval.h"
41 #include "video.h"
42 
43 #define CHECK_CU(x) FF_CUDA_CHECK_DL(ctx, device_hwctx->internal->cuda_dl, x)
44 
45 static const enum AVPixelFormat supported_formats[] = {
49 };
50 
51 static const enum AVPixelFormat deinterleaved_formats[][2] = {
53 };
54 
55 enum ScaleStage {
60 };
61 
62 typedef struct NPPScaleStageContext {
66 
67  struct {
68  int width;
69  int height;
70  } planes_in[3], planes_out[3];
71 
75 
76 typedef struct NPPScaleContext {
77  const AVClass *class;
78 
82 
84 
85  /**
86  * New dimensions. Special values are:
87  * 0 = original width/height
88  * -1 = keep original aspect
89  */
90  int w, h;
91 
92  /**
93  * Output sw format. AV_PIX_FMT_NONE for no conversion.
94  */
96 
97  char *w_expr; ///< width expression string
98  char *h_expr; ///< height expression string
99  char *format_str;
100 
103 
106 
108 {
109  NPPScaleContext *s = ctx->priv;
110  int i;
111 
112  if (!strcmp(s->format_str, "same")) {
113  s->format = AV_PIX_FMT_NONE;
114  } else {
115  s->format = av_get_pix_fmt(s->format_str);
116  if (s->format == AV_PIX_FMT_NONE) {
117  av_log(ctx, AV_LOG_ERROR, "Unrecognized pixel format: %s\n", s->format_str);
118  return AVERROR(EINVAL);
119  }
120  }
121 
122  for (i = 0; i < FF_ARRAY_ELEMS(s->stages); i++) {
123  s->stages[i].frame = av_frame_alloc();
124  if (!s->stages[i].frame)
125  return AVERROR(ENOMEM);
126  }
127  s->tmp_frame = av_frame_alloc();
128  if (!s->tmp_frame)
129  return AVERROR(ENOMEM);
130 
131  return 0;
132 }
133 
135 {
136  NPPScaleContext *s = ctx->priv;
137  int i;
138 
139  for (i = 0; i < FF_ARRAY_ELEMS(s->stages); i++) {
140  av_frame_free(&s->stages[i].frame);
141  av_buffer_unref(&s->stages[i].frames_ctx);
142  }
143  av_frame_free(&s->tmp_frame);
144 }
145 
147 {
148  static const enum AVPixelFormat pixel_formats[] = {
150  };
151  AVFilterFormats *pix_fmts = ff_make_format_list(pixel_formats);
152 
154 }
155 
156 static int init_stage(NPPScaleStageContext *stage, AVBufferRef *device_ctx)
157 {
158  AVBufferRef *out_ref = NULL;
159  AVHWFramesContext *out_ctx;
160  int in_sw, in_sh, out_sw, out_sh;
161  int ret, i;
162 
163  av_pix_fmt_get_chroma_sub_sample(stage->in_fmt, &in_sw, &in_sh);
164  av_pix_fmt_get_chroma_sub_sample(stage->out_fmt, &out_sw, &out_sh);
165  if (!stage->planes_out[0].width) {
166  stage->planes_out[0].width = stage->planes_in[0].width;
167  stage->planes_out[0].height = stage->planes_in[0].height;
168  }
169 
170  for (i = 1; i < FF_ARRAY_ELEMS(stage->planes_in); i++) {
171  stage->planes_in[i].width = stage->planes_in[0].width >> in_sw;
172  stage->planes_in[i].height = stage->planes_in[0].height >> in_sh;
173  stage->planes_out[i].width = stage->planes_out[0].width >> out_sw;
174  stage->planes_out[i].height = stage->planes_out[0].height >> out_sh;
175  }
176 
177  out_ref = av_hwframe_ctx_alloc(device_ctx);
178  if (!out_ref)
179  return AVERROR(ENOMEM);
180  out_ctx = (AVHWFramesContext*)out_ref->data;
181 
182  out_ctx->format = AV_PIX_FMT_CUDA;
183  out_ctx->sw_format = stage->out_fmt;
184  out_ctx->width = FFALIGN(stage->planes_out[0].width, 32);
185  out_ctx->height = FFALIGN(stage->planes_out[0].height, 32);
186 
187  ret = av_hwframe_ctx_init(out_ref);
188  if (ret < 0)
189  goto fail;
190 
191  av_frame_unref(stage->frame);
192  ret = av_hwframe_get_buffer(out_ref, stage->frame, 0);
193  if (ret < 0)
194  goto fail;
195 
196  stage->frame->width = stage->planes_out[0].width;
197  stage->frame->height = stage->planes_out[0].height;
198 
199  av_buffer_unref(&stage->frames_ctx);
200  stage->frames_ctx = out_ref;
201 
202  return 0;
203 fail:
204  av_buffer_unref(&out_ref);
205  return ret;
206 }
207 
208 static int format_is_supported(enum AVPixelFormat fmt)
209 {
210  int i;
211 
212  for (i = 0; i < FF_ARRAY_ELEMS(supported_formats); i++)
213  if (supported_formats[i] == fmt)
214  return 1;
215  return 0;
216 }
217 
219 {
221  int i, planes;
222 
224  if (planes == desc->nb_components)
225  return fmt;
226  for (i = 0; i < FF_ARRAY_ELEMS(deinterleaved_formats); i++)
227  if (deinterleaved_formats[i][0] == fmt)
228  return deinterleaved_formats[i][1];
229  return AV_PIX_FMT_NONE;
230 }
231 
232 static int init_processing_chain(AVFilterContext *ctx, int in_width, int in_height,
233  int out_width, int out_height)
234 {
235  NPPScaleContext *s = ctx->priv;
236 
237  AVHWFramesContext *in_frames_ctx;
238 
239  enum AVPixelFormat in_format;
240  enum AVPixelFormat out_format;
241  enum AVPixelFormat in_deinterleaved_format;
242  enum AVPixelFormat out_deinterleaved_format;
243 
244  int i, ret, last_stage = -1;
245 
246  /* check that we have a hw context */
247  if (!ctx->inputs[0]->hw_frames_ctx) {
248  av_log(ctx, AV_LOG_ERROR, "No hw context provided on input\n");
249  return AVERROR(EINVAL);
250  }
251  in_frames_ctx = (AVHWFramesContext*)ctx->inputs[0]->hw_frames_ctx->data;
252  in_format = in_frames_ctx->sw_format;
253  out_format = (s->format == AV_PIX_FMT_NONE) ? in_format : s->format;
254 
255  if (!format_is_supported(in_format)) {
256  av_log(ctx, AV_LOG_ERROR, "Unsupported input format: %s\n",
257  av_get_pix_fmt_name(in_format));
258  return AVERROR(ENOSYS);
259  }
260  if (!format_is_supported(out_format)) {
261  av_log(ctx, AV_LOG_ERROR, "Unsupported output format: %s\n",
262  av_get_pix_fmt_name(out_format));
263  return AVERROR(ENOSYS);
264  }
265 
266  in_deinterleaved_format = get_deinterleaved_format(in_format);
267  out_deinterleaved_format = get_deinterleaved_format(out_format);
268  if (in_deinterleaved_format == AV_PIX_FMT_NONE ||
269  out_deinterleaved_format == AV_PIX_FMT_NONE)
270  return AVERROR_BUG;
271 
272  /* figure out which stages need to be done */
273  if (in_width != out_width || in_height != out_height ||
274  in_deinterleaved_format != out_deinterleaved_format) {
275  s->stages[STAGE_RESIZE].stage_needed = 1;
276 
277  if (s->interp_algo == NPPI_INTER_SUPER &&
278  (out_width > in_width && out_height > in_height)) {
279  s->interp_algo = NPPI_INTER_LANCZOS;
280  av_log(ctx, AV_LOG_WARNING, "super-sampling not supported for output dimensions, using lanczos instead.\n");
281  }
282  if (s->interp_algo == NPPI_INTER_SUPER &&
283  !(out_width < in_width && out_height < in_height)) {
284  s->interp_algo = NPPI_INTER_CUBIC;
285  av_log(ctx, AV_LOG_WARNING, "super-sampling not supported for output dimensions, using cubic instead.\n");
286  }
287  }
288 
289  if (!s->stages[STAGE_RESIZE].stage_needed && in_format == out_format)
290  s->passthrough = 1;
291 
292  if (!s->passthrough) {
293  if (in_format != in_deinterleaved_format)
294  s->stages[STAGE_DEINTERLEAVE].stage_needed = 1;
295  if (out_format != out_deinterleaved_format)
296  s->stages[STAGE_INTERLEAVE].stage_needed = 1;
297  }
298 
299  s->stages[STAGE_DEINTERLEAVE].in_fmt = in_format;
300  s->stages[STAGE_DEINTERLEAVE].out_fmt = in_deinterleaved_format;
301  s->stages[STAGE_DEINTERLEAVE].planes_in[0].width = in_width;
302  s->stages[STAGE_DEINTERLEAVE].planes_in[0].height = in_height;
303 
304  s->stages[STAGE_RESIZE].in_fmt = in_deinterleaved_format;
305  s->stages[STAGE_RESIZE].out_fmt = out_deinterleaved_format;
306  s->stages[STAGE_RESIZE].planes_in[0].width = in_width;
307  s->stages[STAGE_RESIZE].planes_in[0].height = in_height;
308  s->stages[STAGE_RESIZE].planes_out[0].width = out_width;
309  s->stages[STAGE_RESIZE].planes_out[0].height = out_height;
310 
311  s->stages[STAGE_INTERLEAVE].in_fmt = out_deinterleaved_format;
312  s->stages[STAGE_INTERLEAVE].out_fmt = out_format;
313  s->stages[STAGE_INTERLEAVE].planes_in[0].width = out_width;
314  s->stages[STAGE_INTERLEAVE].planes_in[0].height = out_height;
315 
316  /* init the hardware contexts */
317  for (i = 0; i < FF_ARRAY_ELEMS(s->stages); i++) {
318  if (!s->stages[i].stage_needed)
319  continue;
320 
321  ret = init_stage(&s->stages[i], in_frames_ctx->device_ref);
322  if (ret < 0)
323  return ret;
324 
325  last_stage = i;
326  }
327 
328  if (last_stage >= 0)
329  ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(s->stages[last_stage].frames_ctx);
330  else
331  ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(ctx->inputs[0]->hw_frames_ctx);
332 
333  if (!ctx->outputs[0]->hw_frames_ctx)
334  return AVERROR(ENOMEM);
335 
336  return 0;
337 }
338 
340 {
341  AVFilterContext *ctx = outlink->src;
342  AVFilterLink *inlink = outlink->src->inputs[0];
343  NPPScaleContext *s = ctx->priv;
344  int w, h;
345  int ret;
346 
348  s->w_expr, s->h_expr,
349  inlink, outlink,
350  &w, &h)) < 0)
351  goto fail;
352 
354  s->force_original_aspect_ratio, s->force_divisible_by);
355 
356  if (((int64_t)h * inlink->w) > INT_MAX ||
357  ((int64_t)w * inlink->h) > INT_MAX)
358  av_log(ctx, AV_LOG_ERROR, "Rescaled value for width or height is too big.\n");
359 
360  outlink->w = w;
361  outlink->h = h;
362 
364  if (ret < 0)
365  return ret;
366 
367  av_log(ctx, AV_LOG_VERBOSE, "w:%d h:%d -> w:%d h:%d\n",
368  inlink->w, inlink->h, outlink->w, outlink->h);
369 
370  if (inlink->sample_aspect_ratio.num)
371  outlink->sample_aspect_ratio = av_mul_q((AVRational){outlink->h*inlink->w,
372  outlink->w*inlink->h},
373  inlink->sample_aspect_ratio);
374  else
375  outlink->sample_aspect_ratio = inlink->sample_aspect_ratio;
376 
377  return 0;
378 
379 fail:
380  return ret;
381 }
382 
384  AVFrame *out, AVFrame *in)
385 {
386  AVHWFramesContext *in_frames_ctx = (AVHWFramesContext*)in->hw_frames_ctx->data;
387  NppStatus err;
388 
389  switch (in_frames_ctx->sw_format) {
390  case AV_PIX_FMT_NV12:
391  err = nppiYCbCr420_8u_P2P3R(in->data[0], in->linesize[0],
392  in->data[1], in->linesize[1],
393  out->data, out->linesize,
394  (NppiSize){ in->width, in->height });
395  break;
396  default:
397  return AVERROR_BUG;
398  }
399  if (err != NPP_SUCCESS) {
400  av_log(ctx, AV_LOG_ERROR, "NPP deinterleave error: %d\n", err);
401  return AVERROR_UNKNOWN;
402  }
403 
404  return 0;
405 }
406 
408  AVFrame *out, AVFrame *in)
409 {
410  NPPScaleContext *s = ctx->priv;
411  NppStatus err;
412  int i;
413 
414  for (i = 0; i < FF_ARRAY_ELEMS(stage->planes_in) && i < FF_ARRAY_ELEMS(in->data) && in->data[i]; i++) {
415  int iw = stage->planes_in[i].width;
416  int ih = stage->planes_in[i].height;
417  int ow = stage->planes_out[i].width;
418  int oh = stage->planes_out[i].height;
419 
420  err = nppiResizeSqrPixel_8u_C1R(in->data[i], (NppiSize){ iw, ih },
421  in->linesize[i], (NppiRect){ 0, 0, iw, ih },
422  out->data[i], out->linesize[i],
423  (NppiRect){ 0, 0, ow, oh },
424  (double)ow / iw, (double)oh / ih,
425  0.0, 0.0, s->interp_algo);
426  if (err != NPP_SUCCESS) {
427  av_log(ctx, AV_LOG_ERROR, "NPP resize error: %d\n", err);
428  return AVERROR_UNKNOWN;
429  }
430  }
431 
432  return 0;
433 }
434 
436  AVFrame *out, AVFrame *in)
437 {
438  AVHWFramesContext *out_frames_ctx = (AVHWFramesContext*)out->hw_frames_ctx->data;
439  NppStatus err;
440 
441  switch (out_frames_ctx->sw_format) {
442  case AV_PIX_FMT_NV12:
443  err = nppiYCbCr420_8u_P3P2R((const uint8_t**)in->data,
444  in->linesize,
445  out->data[0], out->linesize[0],
446  out->data[1], out->linesize[1],
447  (NppiSize){ in->width, in->height });
448  break;
449  default:
450  return AVERROR_BUG;
451  }
452  if (err != NPP_SUCCESS) {
453  av_log(ctx, AV_LOG_ERROR, "NPP deinterleave error: %d\n", err);
454  return AVERROR_UNKNOWN;
455  }
456 
457  return 0;
458 }
459 
461  AVFrame *out, AVFrame *in) = {
465 };
466 
468 {
469  NPPScaleContext *s = ctx->priv;
470  AVFrame *src = in;
471  int i, ret, last_stage = -1;
472 
473  for (i = 0; i < FF_ARRAY_ELEMS(s->stages); i++) {
474  if (!s->stages[i].stage_needed)
475  continue;
476 
477  ret = nppscale_process[i](ctx, &s->stages[i], s->stages[i].frame, src);
478  if (ret < 0)
479  return ret;
480 
481  src = s->stages[i].frame;
482  last_stage = i;
483  }
484  if (last_stage < 0)
485  return AVERROR_BUG;
486 
487  ret = av_hwframe_get_buffer(src->hw_frames_ctx, s->tmp_frame, 0);
488  if (ret < 0)
489  return ret;
490 
491  s->tmp_frame->width = src->width;
492  s->tmp_frame->height = src->height;
493 
495  av_frame_move_ref(src, s->tmp_frame);
496 
498  if (ret < 0)
499  return ret;
500 
501  return 0;
502 }
503 
505 {
506  AVFilterContext *ctx = link->dst;
507  NPPScaleContext *s = ctx->priv;
508  AVFilterLink *outlink = ctx->outputs[0];
509  AVHWFramesContext *frames_ctx = (AVHWFramesContext*)outlink->hw_frames_ctx->data;
510  AVCUDADeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
511 
512  AVFrame *out = NULL;
513  CUcontext dummy;
514  int ret = 0;
515 
516  if (s->passthrough)
517  return ff_filter_frame(outlink, in);
518 
519  out = av_frame_alloc();
520  if (!out) {
521  ret = AVERROR(ENOMEM);
522  goto fail;
523  }
524 
525  ret = CHECK_CU(device_hwctx->internal->cuda_dl->cuCtxPushCurrent(device_hwctx->cuda_ctx));
526  if (ret < 0)
527  goto fail;
528 
529  ret = nppscale_scale(ctx, out, in);
530 
531  CHECK_CU(device_hwctx->internal->cuda_dl->cuCtxPopCurrent(&dummy));
532  if (ret < 0)
533  goto fail;
534 
535  av_reduce(&out->sample_aspect_ratio.num, &out->sample_aspect_ratio.den,
536  (int64_t)in->sample_aspect_ratio.num * outlink->h * link->w,
537  (int64_t)in->sample_aspect_ratio.den * outlink->w * link->h,
538  INT_MAX);
539 
540  av_frame_free(&in);
541  return ff_filter_frame(outlink, out);
542 fail:
543  av_frame_free(&in);
544  av_frame_free(&out);
545  return ret;
546 }
547 
548 #define OFFSET(x) offsetof(NPPScaleContext, x)
549 #define FLAGS (AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM)
550 static const AVOption options[] = {
551  { "w", "Output video width", OFFSET(w_expr), AV_OPT_TYPE_STRING, { .str = "iw" }, .flags = FLAGS },
552  { "h", "Output video height", OFFSET(h_expr), AV_OPT_TYPE_STRING, { .str = "ih" }, .flags = FLAGS },
553  { "format", "Output pixel format", OFFSET(format_str), AV_OPT_TYPE_STRING, { .str = "same" }, .flags = FLAGS },
554 
555  { "interp_algo", "Interpolation algorithm used for resizing", OFFSET(interp_algo), AV_OPT_TYPE_INT, { .i64 = NPPI_INTER_CUBIC }, 0, INT_MAX, FLAGS, "interp_algo" },
556  { "nn", "nearest neighbour", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_NN }, 0, 0, FLAGS, "interp_algo" },
557  { "linear", "linear", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_LINEAR }, 0, 0, FLAGS, "interp_algo" },
558  { "cubic", "cubic", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_CUBIC }, 0, 0, FLAGS, "interp_algo" },
559  { "cubic2p_bspline", "2-parameter cubic (B=1, C=0)", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_CUBIC2P_BSPLINE }, 0, 0, FLAGS, "interp_algo" },
560  { "cubic2p_catmullrom", "2-parameter cubic (B=0, C=1/2)", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_CUBIC2P_CATMULLROM }, 0, 0, FLAGS, "interp_algo" },
561  { "cubic2p_b05c03", "2-parameter cubic (B=1/2, C=3/10)", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_CUBIC2P_B05C03 }, 0, 0, FLAGS, "interp_algo" },
562  { "super", "supersampling", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_SUPER }, 0, 0, FLAGS, "interp_algo" },
563  { "lanczos", "Lanczos", 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_LANCZOS }, 0, 0, FLAGS, "interp_algo" },
564  { "force_original_aspect_ratio", "decrease or increase w/h if necessary to keep the original AR", OFFSET(force_original_aspect_ratio), AV_OPT_TYPE_INT, { .i64 = 0}, 0, 2, FLAGS, "force_oar" },
565  { "disable", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = 0 }, 0, 0, FLAGS, "force_oar" },
566  { "decrease", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = 1 }, 0, 0, FLAGS, "force_oar" },
567  { "increase", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = 2 }, 0, 0, FLAGS, "force_oar" },
568  { "force_divisible_by", "enforce that the output resolution is divisible by a defined integer when force_original_aspect_ratio is used", OFFSET(force_divisible_by), AV_OPT_TYPE_INT, { .i64 = 1}, 1, 256, FLAGS },
569  { NULL },
570 };
571 
572 static const AVClass nppscale_class = {
573  .class_name = "nppscale",
574  .item_name = av_default_item_name,
575  .option = options,
576  .version = LIBAVUTIL_VERSION_INT,
577 };
578 
579 static const AVFilterPad nppscale_inputs[] = {
580  {
581  .name = "default",
582  .type = AVMEDIA_TYPE_VIDEO,
583  .filter_frame = nppscale_filter_frame,
584  },
585  { NULL }
586 };
587 
588 static const AVFilterPad nppscale_outputs[] = {
589  {
590  .name = "default",
591  .type = AVMEDIA_TYPE_VIDEO,
592  .config_props = nppscale_config_props,
593  },
594  { NULL }
595 };
596 
598  .name = "scale_npp",
599  .description = NULL_IF_CONFIG_SMALL("NVIDIA Performance Primitives video "
600  "scaling and format conversion"),
601 
602  .init = nppscale_init,
603  .uninit = nppscale_uninit,
604  .query_formats = nppscale_query_formats,
605 
606  .priv_size = sizeof(NPPScaleContext),
607  .priv_class = &nppscale_class,
608 
611 
612  .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
613 };
format_is_supported
static int format_is_supported(enum AVPixelFormat fmt)
Definition: vf_scale_npp.c:208
AVHWDeviceContext::hwctx
void * hwctx
The format-specific data, allocated and freed by libavutil along with this context.
Definition: hwcontext.h:92
NPPScaleContext::passthrough
int passthrough
Definition: vf_scale_npp.c:81
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:200
AV_PIX_FMT_CUDA
@ AV_PIX_FMT_CUDA
HW acceleration through CUDA.
Definition: pixfmt.h:235
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
nppscale_inputs
static const AVFilterPad nppscale_inputs[]
Definition: vf_scale_npp.c:579
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
ff_make_format_list
AVFilterFormats * ff_make_format_list(const int *fmts)
Create a list of supported formats.
Definition: formats.c:286
hwcontext_cuda_internal.h
out
FILE * out
Definition: movenc.c:54
FF_FILTER_FLAG_HWFRAME_AWARE
#define FF_FILTER_FLAG_HWFRAME_AWARE
The filter is aware of hardware frames, and any hardware frame context should not be automatically pr...
Definition: internal.h:339
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1096
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2573
AVBufferRef::data
uint8_t * data
The data buffer.
Definition: buffer.h:92
CHECK_CU
#define CHECK_CU(x)
Definition: vf_scale_npp.c:43
AVHWFramesContext::format
enum AVPixelFormat format
The pixel format identifying the underlying HW surface type.
Definition: hwcontext.h:209
inlink
The exact code depends on how similar the blocks are and how related they are to the and needs to apply these operations to the correct inlink or outlink if there are several Macros are available to factor that when no extra processing is inlink
Definition: filter_design.txt:212
ScaleStage
ScaleStage
Definition: vf_scale_npp.c:55
nppscale_class
static const AVClass nppscale_class
Definition: vf_scale_npp.c:572
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:203
av_hwframe_ctx_init
int av_hwframe_ctx_init(AVBufferRef *ref)
Finalize the context before use.
Definition: hwcontext.c:333
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:318
pixdesc.h
AVFrame::width
int width
Definition: frame.h:376
w
uint8_t w
Definition: llviddspenc.c:39
av_hwframe_ctx_alloc
AVBufferRef * av_hwframe_ctx_alloc(AVBufferRef *device_ref_in)
Allocate an AVHWFramesContext tied to a given device context.
Definition: hwcontext.c:247
NPPScaleStageContext::stage_needed
int stage_needed
Definition: vf_scale_npp.c:63
AVOption
AVOption.
Definition: opt.h:248
supported_formats
static enum AVPixelFormat supported_formats[]
Definition: vf_scale_npp.c:45
AV_LOG_VERBOSE
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:210
ff_scale_eval_dimensions
int ff_scale_eval_dimensions(void *log_ctx, const char *w_expr, const char *h_expr, AVFilterLink *inlink, AVFilterLink *outlink, int *ret_w, int *ret_h)
Parse and evaluate string expressions for width and height.
Definition: scale_eval.c:57
STAGE_NB
@ STAGE_NB
Definition: vf_scale_npp.c:59
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:149
AVERROR_UNKNOWN
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:71
AVHWFramesContext::width
int width
The allocated dimensions of the frames in this pool.
Definition: hwcontext.h:229
video.h
NPPScaleStageContext::planes_in
struct NPPScaleStageContext::@234 planes_in[3]
NPPScaleStageContext::frame
AVFrame * frame
Definition: vf_scale_npp.c:73
AVFilterFormats
A list of supported formats for one end of a filter link.
Definition: formats.h:65
formats.h
av_pix_fmt_count_planes
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2613
fail
#define fail()
Definition: checkasm.h:133
get_deinterleaved_format
static enum AVPixelFormat get_deinterleaved_format(enum AVPixelFormat fmt)
Definition: vf_scale_npp.c:218
STAGE_RESIZE
@ STAGE_RESIZE
Definition: vf_scale_npp.c:57
NPPScaleContext::tmp_frame
AVFrame * tmp_frame
Definition: vf_scale_npp.c:80
av_pix_fmt_get_chroma_sub_sample
int av_pix_fmt_get_chroma_sub_sample(enum AVPixelFormat pix_fmt, int *h_shift, int *v_shift)
Utility function to access log2_chroma_w log2_chroma_h from the pixel format AVPixFmtDescriptor.
Definition: pixdesc.c:2601
av_reduce
int av_reduce(int *dst_num, int *dst_den, int64_t num, int64_t den, int64_t max)
Reduce a fraction.
Definition: rational.c:35
NPPScaleContext::shift_width
int shift_width
Definition: vf_scale_npp.c:83
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:54
av_frame_alloc
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:190
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
ff_set_common_formats
int ff_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
A helper for query_formats() which sets all links to the same list of formats.
Definition: formats.c:587
AVHWFramesContext::height
int height
Definition: hwcontext.h:229
s
#define s(width, name)
Definition: cbs_vp9.c:257
nppscale_uninit
static void nppscale_uninit(AVFilterContext *ctx)
Definition: vf_scale_npp.c:134
NPPScaleContext::stages
NPPScaleStageContext stages[STAGE_NB]
Definition: vf_scale_npp.c:79
nppscale_query_formats
static int nppscale_query_formats(AVFilterContext *ctx)
Definition: vf_scale_npp.c:146
outputs
static const AVFilterPad outputs[]
Definition: af_acontrast.c:203
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:309
ctx
AVFormatContext * ctx
Definition: movenc.c:48
NPPScaleContext::force_original_aspect_ratio
int force_original_aspect_ratio
Definition: vf_scale_npp.c:101
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
link
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a link
Definition: filter_design.txt:23
NPPScaleContext
Definition: vf_scale_npp.c:76
if
if(ret)
Definition: filter_design.txt:179
FLAGS
#define FLAGS
Definition: vf_scale_npp.c:549
init_stage
static int init_stage(NPPScaleStageContext *stage, AVBufferRef *device_ctx)
Definition: vf_scale_npp.c:156
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:67
nppscale_interleave
static int nppscale_interleave(AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
Definition: vf_scale_npp.c:435
NULL
#define NULL
Definition: coverity.c:32
AVHWFramesContext::sw_format
enum AVPixelFormat sw_format
The pixel format identifying the actual data layout of the hardware frames.
Definition: hwcontext.h:222
av_frame_copy_props
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:658
av_buffer_unref
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it.
Definition: buffer.c:125
AVRational
Rational number (pair of numerator and denominator).
Definition: rational.h:58
AVHWFramesContext::device_ref
AVBufferRef * device_ref
A reference to the parent AVHWDeviceContext.
Definition: hwcontext.h:141
AVFilterContext::inputs
AVFilterLink ** inputs
array of pointers to input links
Definition: avfilter.h:349
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:235
src
#define src
Definition: vp8dsp.c:255
NPPScaleContext::h_expr
char * h_expr
height expression string
Definition: vf_scale_npp.c:98
inputs
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several inputs
Definition: filter_design.txt:243
nppscale_init
static int nppscale_init(AVFilterContext *ctx)
Definition: vf_scale_npp.c:107
NPPScaleContext::shift_height
int shift_height
Definition: vf_scale_npp.c:83
NPPScaleStageContext::height
int height
Definition: vf_scale_npp.c:69
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
NPPScaleStageContext::out_fmt
enum AVPixelFormat out_fmt
Definition: vf_scale_npp.c:65
OFFSET
#define OFFSET(x)
Definition: vf_scale_npp.c:548
NPPScaleStageContext::frames_ctx
AVBufferRef * frames_ctx
Definition: vf_scale_npp.c:72
ff_vf_scale_npp
AVFilter ff_vf_scale_npp
Definition: vf_scale_npp.c:597
nppscale_filter_frame
static int nppscale_filter_frame(AVFilterLink *link, AVFrame *in)
Definition: vf_scale_npp.c:504
scale_eval.h
nppscale_config_props
static int nppscale_config_props(AVFilterLink *outlink)
Definition: vf_scale_npp.c:339
init_processing_chain
static int init_processing_chain(AVFilterContext *ctx, int in_width, int in_height, int out_width, int out_height)
Definition: vf_scale_npp.c:232
nppscale_resize
static int nppscale_resize(AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
Definition: vf_scale_npp.c:407
internal.h
in
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
Definition: audio_convert.c:326
i
int i
Definition: input.c:407
NPPScaleContext::h
int h
Definition: vf_scale_npp.c:90
internal.h
common.h
av_frame_move_ref
void av_frame_move_ref(AVFrame *dst, AVFrame *src)
Move everything contained in src to dst and reset src.
Definition: frame.c:582
uint8_t
uint8_t
Definition: audio_convert.c:194
av_frame_unref
void av_frame_unref(AVFrame *frame)
Unreference all the buffers referenced by frame and reset the frame fields.
Definition: frame.c:553
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:60
NPPScaleContext::w
int w
New dimensions.
Definition: vf_scale_npp.c:90
NPPScaleStageContext::planes_out
struct NPPScaleStageContext::@234 planes_out[3]
AVFilter
Filter definition.
Definition: avfilter.h:145
AVHWFramesContext
This struct describes a set or pool of "hardware" frames (i.e.
Definition: hwcontext.h:124
AVCUDADeviceContext
This struct is allocated as AVHWDeviceContext.hwctx.
Definition: hwcontext_cuda.h:42
ret
ret
Definition: filter_design.txt:187
NPPScaleContext::format
enum AVPixelFormat format
Output sw format.
Definition: vf_scale_npp.c:95
AV_PIX_FMT_NV12
@ AV_PIX_FMT_NV12
planar YUV 4:2:0, 12bpp, 1 plane for Y and 1 plane for the UV components, which are interleaved (firs...
Definition: pixfmt.h:89
AVClass::class_name
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:72
AVHWFramesContext::device_ctx
AVHWDeviceContext * device_ctx
The parent AVHWDeviceContext.
Definition: hwcontext.h:149
cuda_check.h
NPPScaleContext::interp_algo
int interp_algo
Definition: vf_scale_npp.c:104
ff_scale_adjust_dimensions
int ff_scale_adjust_dimensions(AVFilterLink *inlink, int *ret_w, int *ret_h, int force_original_aspect_ratio, int force_divisible_by)
Transform evaluated width and height obtained from ff_scale_eval_dimensions into actual target width ...
Definition: scale_eval.c:113
NPPScaleContext::w_expr
char * w_expr
width expression string
Definition: vf_scale_npp.c:97
av_get_pix_fmt
enum AVPixelFormat av_get_pix_fmt(const char *name)
Return the pixel format corresponding to name.
Definition: pixdesc.c:2501
NPPScaleStageContext::in_fmt
enum AVPixelFormat in_fmt
Definition: vf_scale_npp.c:64
AVFrame::height
int height
Definition: frame.h:376
NPPScaleContext::format_str
char * format_str
Definition: vf_scale_npp.c:99
STAGE_INTERLEAVE
@ STAGE_INTERLEAVE
Definition: vf_scale_npp.c:58
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
dummy
int dummy
Definition: motion.c:64
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:225
avfilter.h
nppscale_process
static int(*const nppscale_process[])(AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
Definition: vf_scale_npp.c:460
nppscale_outputs
static const AVFilterPad nppscale_outputs[]
Definition: vf_scale_npp.c:588
NPPScaleContext::force_divisible_by
int force_divisible_by
Definition: vf_scale_npp.c:102
av_mul_q
AVRational av_mul_q(AVRational b, AVRational c)
Multiply two rationals.
Definition: rational.c:80
av_buffer_ref
AVBufferRef * av_buffer_ref(AVBufferRef *buf)
Create a new reference to an AVBuffer.
Definition: buffer.c:93
NPPScaleStageContext::width
int width
Definition: vf_scale_npp.c:68
AV_PIX_FMT_YUV444P
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
NPPScaleStageContext
Definition: vf_scale_npp.c:62
AVFilterContext
An instance of a filter.
Definition: avfilter.h:341
STAGE_DEINTERLEAVE
@ STAGE_DEINTERLEAVE
Definition: vf_scale_npp.c:56
planes
static const struct @322 planes[]
desc
const char * desc
Definition: libsvtav1.c:79
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AVBufferRef
A reference to a data buffer.
Definition: buffer.h:84
options
static const AVOption options[]
Definition: vf_scale_npp.c:550
nppscale_deinterleave
static int nppscale_deinterleave(AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
Definition: vf_scale_npp.c:383
AVPixFmtDescriptor
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
FFALIGN
#define FFALIGN(x, a)
Definition: macros.h:48
deinterleaved_formats
static enum AVPixelFormat deinterleaved_formats[][2]
Definition: vf_scale_npp.c:51
hwcontext.h
AVERROR_BUG
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
Definition: error.h:50
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
h
h
Definition: vp9dsp_template.c:2038
avstring.h
AV_OPT_TYPE_STRING
@ AV_OPT_TYPE_STRING
Definition: opt.h:229
int
int
Definition: ffmpeg_filter.c:170
av_hwframe_get_buffer
int av_hwframe_get_buffer(AVBufferRef *hwframe_ref, AVFrame *frame, int flags)
Allocate a new frame attached to the given AVHWFramesContext.
Definition: hwcontext.c:502
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Definition: opt.h:234
av_get_pix_fmt_name
const char * av_get_pix_fmt_name(enum AVPixelFormat pix_fmt)
Return the short name for a pixel format, NULL in case pix_fmt is unknown.
Definition: pixdesc.c:2489
nppscale_scale
static int nppscale_scale(AVFilterContext *ctx, AVFrame *out, AVFrame *in)
Definition: vf_scale_npp.c:467