vf_scale_npp.c File Reference

#include <nppi.h>
#include <stdio.h>
#include <string.h>
#include "libavutil/avstring.h"
#include "libavutil/common.h"
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda_internal.h"
#include "libavutil/cuda_check.h"
#include "libavutil/internal.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "formats.h"
#include "internal.h"
#include "scale_eval.h"
#include "video.h"

Go to the source code of this file.

Data Structures
struct	NPPScaleStageContext
struct	NPPScaleContext

Macros
#define	CHECK_CU(x)   FF_CUDA_CHECK_DL(ctx, device_hwctx->internal->cuda_dl, x)
#define	OFFSET(x)   offsetof(NPPScaleContext, x)
#define	FLAGS   (AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM)

Enumerations
enum	ScaleStage { STAGE_DEINTERLEAVE, STAGE_RESIZE, STAGE_INTERLEAVE, STAGE_NB }

Functions
static int	nppscale_init (AVFilterContext *ctx)
static void	nppscale_uninit (AVFilterContext *ctx)
static int	nppscale_query_formats (AVFilterContext *ctx)
static int	init_stage (NPPScaleStageContext *stage, AVBufferRef *device_ctx)
static int	format_is_supported (enum AVPixelFormat fmt)
static enum AVPixelFormat	get_deinterleaved_format (enum AVPixelFormat fmt)
static int	init_processing_chain (AVFilterContext *ctx, int in_width, int in_height, int out_width, int out_height)
static int	nppscale_config_props (AVFilterLink *outlink)
static int	nppscale_deinterleave (AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
static int	nppscale_resize (AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
static int	nppscale_interleave (AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
static int	nppscale_scale (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
static int	nppscale_filter_frame (AVFilterLink *link, AVFrame *in)

Variables
static enum AVPixelFormat	supported_formats []
static enum AVPixelFormat	deinterleaved_formats [][2]
static int(*const	nppscale_process [])(AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)
static const AVOption	options []
static const AVClass	nppscale_class
static const AVFilterPad	nppscale_inputs []
static const AVFilterPad	nppscale_outputs []
AVFilter	ff_vf_scale_npp

Detailed Description

scale video filter

Definition in file vf_scale_npp.c.

Macro Definition Documentation

CHECK_CU

#define CHECK_CU

(

x

)

FF_CUDA_CHECK_DL(ctx, device_hwctx->internal->cuda_dl, x)

Definition at line 43 of file vf_scale_npp.c.

OFFSET

#define OFFSET

(

x

)

offsetof(NPPScaleContext, x)

Definition at line 548 of file vf_scale_npp.c.

FLAGS

#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 549 of file vf_scale_npp.c.

Enumeration Type Documentation

ScaleStage

enum ScaleStage

Enumerator
STAGE_DEINTERLEAVE
STAGE_RESIZE
STAGE_INTERLEAVE
STAGE_NB

Definition at line 55 of file vf_scale_npp.c.

Function Documentation

nppscale_init()

static int nppscale_init ( AVFilterContext *  ctx )

static

Definition at line 107 of file vf_scale_npp.c.

nppscale_uninit()

static void nppscale_uninit ( AVFilterContext *  ctx )

static

Definition at line 134 of file vf_scale_npp.c.

nppscale_query_formats()

static int nppscale_query_formats ( AVFilterContext *  ctx )

static

Definition at line 146 of file vf_scale_npp.c.

init_stage()

static int init_stage ( NPPScaleStageContext *  stage, AVBufferRef *  device_ctx  )

static

Definition at line 156 of file vf_scale_npp.c.

Referenced by init_processing_chain().

format_is_supported()

static int format_is_supported ( enum AVPixelFormat  fmt )

static

Definition at line 208 of file vf_scale_npp.c.

Referenced by init_processing_chain().

get_deinterleaved_format()

static enum AVPixelFormat get_deinterleaved_format ( enum AVPixelFormat  fmt )

static

Definition at line 218 of file vf_scale_npp.c.

Referenced by init_processing_chain().

init_processing_chain()

static int init_processing_chain ( AVFilterContext *  ctx, int  in_width, int  in_height, int  out_width, int  out_height  )

static

Definition at line 232 of file vf_scale_npp.c.

Referenced by nppscale_config_props().

nppscale_config_props()

static int nppscale_config_props ( AVFilterLink *  outlink )

static

Definition at line 339 of file vf_scale_npp.c.

nppscale_deinterleave()

static int nppscale_deinterleave ( AVFilterContext *  ctx, NPPScaleStageContext *  stage, AVFrame *  out, AVFrame *  in  )

static

Definition at line 383 of file vf_scale_npp.c.

nppscale_resize()

static int nppscale_resize ( AVFilterContext *  ctx, NPPScaleStageContext *  stage, AVFrame *  out, AVFrame *  in  )

static

Definition at line 407 of file vf_scale_npp.c.

nppscale_interleave()

static int nppscale_interleave ( AVFilterContext *  ctx, NPPScaleStageContext *  stage, AVFrame *  out, AVFrame *  in  )

static

Definition at line 435 of file vf_scale_npp.c.

nppscale_scale()

static int nppscale_scale ( AVFilterContext *  ctx, AVFrame *  out, AVFrame *  in  )

static

Definition at line 467 of file vf_scale_npp.c.

Referenced by nppscale_filter_frame().

nppscale_filter_frame()

static int nppscale_filter_frame ( AVFilterLink *  link, AVFrame *  in  )

static

Definition at line 504 of file vf_scale_npp.c.

Variable Documentation

supported_formats

enum AVPixelFormat supported_formats[]

static

Initial value:

= {
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_NV12,
    AV_PIX_FMT_YUV444P,
}

Definition at line 45 of file vf_scale_npp.c.

Referenced by format_is_supported().

deinterleaved_formats

enum AVPixelFormat deinterleaved_formats[][2]

static

Initial value:

= {
    { AV_PIX_FMT_NV12, AV_PIX_FMT_YUV420P },
}

Definition at line 51 of file vf_scale_npp.c.

Referenced by get_deinterleaved_format().

nppscale_process

int(*const nppscale_process[])(AVFilterContext *ctx, NPPScaleStageContext *stage, AVFrame *out, AVFrame *in)

static

Initial value:

= {
    [STAGE_DEINTERLEAVE] = nppscale_deinterleave,
    [STAGE_RESIZE]       = nppscale_resize,
    [STAGE_INTERLEAVE]   = nppscale_interleave,
}

Definition at line 460 of file vf_scale_npp.c.

Referenced by nppscale_scale().

options

const AVOption options[]

static

Initial value:

= {
    { "w",      "Output video width",  OFFSET(w_expr),     AV_OPT_TYPE_STRING, { .str = "iw"   }, .flags = FLAGS },
    { "h",      "Output video height", OFFSET(h_expr),     AV_OPT_TYPE_STRING, { .str = "ih"   }, .flags = FLAGS },
    { "format", "Output pixel format", OFFSET(format_str), AV_OPT_TYPE_STRING, { .str = "same" }, .flags = FLAGS },
 
    { "interp_algo", "Interpolation algorithm used for resizing", OFFSET(interp_algo), AV_OPT_TYPE_INT, { .i64 = NPPI_INTER_CUBIC }, 0, INT_MAX, FLAGS, "interp_algo" },
        { "nn",                 "nearest neighbour",                 0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_NN                 }, 0, 0, FLAGS, "interp_algo" },
        { "linear",             "linear",                            0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_LINEAR             }, 0, 0, FLAGS, "interp_algo" },
        { "cubic",              "cubic",                             0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_CUBIC              }, 0, 0, FLAGS, "interp_algo" },
        { "cubic2p_bspline",    "2-parameter cubic (B=1, C=0)",      0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_CUBIC2P_BSPLINE    }, 0, 0, FLAGS, "interp_algo" },
        { "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" },
        { "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" },
        { "super",              "supersampling",                     0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_SUPER              }, 0, 0, FLAGS, "interp_algo" },
        { "lanczos",            "Lanczos",                           0, AV_OPT_TYPE_CONST, { .i64 = NPPI_INTER_LANCZOS            }, 0, 0, FLAGS, "interp_algo" },
    { "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" },
    { "disable",  NULL, 0, AV_OPT_TYPE_CONST, {.i64 = 0 }, 0, 0, FLAGS, "force_oar" },
    { "decrease", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = 1 }, 0, 0, FLAGS, "force_oar" },
    { "increase", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = 2 }, 0, 0, FLAGS, "force_oar" },
    { "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 },
    { NULL },
}

Definition at line 550 of file vf_scale_npp.c.

nppscale_class

const AVClass nppscale_class

static

Initial value:

= {
    .class_name = "nppscale",
    .item_name  = av_default_item_name,
    .option     = options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 572 of file vf_scale_npp.c.

nppscale_inputs

const AVFilterPad nppscale_inputs[]

static

Initial value:

= {
    {
        .name        = "default",
        .type        = AVMEDIA_TYPE_VIDEO,
        .filter_frame = nppscale_filter_frame,
    },
    { NULL }
}

Definition at line 579 of file vf_scale_npp.c.

nppscale_outputs

const AVFilterPad nppscale_outputs[]

static

Initial value:

= {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
        .config_props = nppscale_config_props,
    },
    { NULL }
}

Definition at line 588 of file vf_scale_npp.c.

ff_vf_scale_npp

AVFilter ff_vf_scale_npp

Initial value:

= {
    .name      = "scale_npp",
    .description = NULL_IF_CONFIG_SMALL("NVIDIA Performance Primitives video "
                                        "scaling and format conversion"),
 
    .init          = nppscale_init,
    .uninit        = nppscale_uninit,
    .query_formats = nppscale_query_formats,
 
    .priv_size = sizeof(NPPScaleContext),
    .priv_class = &nppscale_class,
 
    .inputs    = nppscale_inputs,
    .outputs   = nppscale_outputs,
 
    .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}

Definition at line 597 of file vf_scale_npp.c.