38 #define CHECK_CU(x) FF_CUDA_CHECK_DL(ctx, ctx->hwctx->internal->cuda_dl, x) 39 #define DIV_UP(a, b) ( ((a) + (b) - 1) / (b) ) 87 if (formats[
i] == fmt)
113 uint8_t* main_data,
int main_linesize,
114 int main_width,
int main_height,
115 uint8_t* overlay_data,
int overlay_linesize,
116 int overlay_width,
int overlay_height,
117 uint8_t* alpha_data,
int alpha_linesize,
118 int alpha_adj_x,
int alpha_adj_y) {
122 void* kernel_args[] = {
124 &main_data, &main_linesize,
125 &overlay_data, &overlay_linesize,
126 &overlay_width, &overlay_height,
127 &alpha_data, &alpha_linesize,
128 &alpha_adj_x, &alpha_adj_y,
152 AVFrame *input_main, *input_overlay;
175 ret =
CHECK_CU(cu->cuCtxPushCurrent(cuda_ctx));
189 input_overlay->
data[3], input_overlay->
linesize[3], 1, 1);
210 input_overlay->
width / 2, input_overlay->
height / 2,
211 input_overlay->
data[3], input_overlay->
linesize[3], 2, 2);
218 input_overlay->
width / 2, input_overlay->
height / 2,
219 input_overlay->
data[3], input_overlay->
linesize[3], 2, 2);
224 CHECK_CU(cu->cuCtxPopCurrent(&dummy));
228 CHECK_CU(cu->cuCtxPopCurrent(&dummy));
258 CHECK_CU(cu->cuCtxPopCurrent(&dummy));
295 extern char vf_overlay_cuda_ptx[];
307 CUcontext
dummy, cuda_ctx;
326 if (!frames_ctx_overlay) {
366 err =
CHECK_CU(cu->cuCtxPushCurrent(cuda_ctx));
373 CHECK_CU(cu->cuCtxPopCurrent(&dummy));
379 CHECK_CU(cu->cuCtxPopCurrent(&dummy));
383 CHECK_CU(cu->cuCtxPopCurrent(&dummy));
396 #define OFFSET(x) offsetof(OverlayCUDAContext, x) 397 #define FLAGS (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM) 400 {
"x",
"Overlay x position",
402 {
"y",
"Overlay y position",
404 {
"eof_action",
"Action to take when encountering EOF from secondary input ",
439 .
name =
"overlay_cuda",
442 .priv_class = &overlay_cuda_class,
447 .
inputs = overlay_cuda_inputs,
448 .
outputs = overlay_cuda_outputs,
449 .preinit = overlay_cuda_framesync_preinit,
This struct aggregates all the (hardware/vendor-specific) "high-level" state, i.e.
#define FF_FILTER_FLAG_HWFRAME_AWARE
The filter is aware of hardware frames, and any hardware frame context should not be automatically pr...
FRAMESYNC_DEFINE_CLASS(overlay_cuda, OverlayCUDAContext, fs)
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it...
This structure describes decoded (raw) audio or video data.
Main libavfilter public API header.
Memory handling functions.
static av_cold int init(AVCodecContext *avctx)
AVCUDADeviceContextInternal * internal
static enum AVPixelFormat supported_overlay_formats[]
int ff_framesync_configure(FFFrameSync *fs)
Configure a frame sync structure.
const char * name
Pad name.
AVFilterContext * parent
Parent filter context.
AVFilterLink ** inputs
array of pointers to input links
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
static int overlay_cuda_call_kernel(OverlayCUDAContext *ctx, int x_position, int y_position, uint8_t *main_data, int main_linesize, int main_width, int main_height, uint8_t *overlay_data, int overlay_linesize, int overlay_width, int overlay_height, uint8_t *alpha_data, int alpha_linesize, int alpha_adj_x, int alpha_adj_y)
Call overlay kernell for a plane.
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
static av_cold int uninit(AVCodecContext *avctx)
int ff_framesync_init_dualinput(FFFrameSync *fs, AVFilterContext *parent)
Initialize a frame sync structure for dualinput.
int ff_framesync_dualinput_get(FFFrameSync *fs, AVFrame **f0, AVFrame **f1)
static const AVFilterPad overlay_cuda_inputs[]
AVBufferRef * hw_device_ctx
A filter pad used for either input or output.
static int format_is_supported(const enum AVPixelFormat formats[], enum AVPixelFormat fmt)
Helper to find out if provided format is supported by filter.
A link between two filters.
#define i(width, name, range_min, range_max)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
void ff_framesync_uninit(FFFrameSync *fs)
Free all memory currently allocated.
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
void * priv
private data for use by the filter
planar YUV 4:2:0, 12bpp, 1 plane for Y and 1 plane for the UV components, which are interleaved (firs...
AVRational time_base
Define the time base used by the PTS of the frames/samples which will pass through this link...
int ff_framesync_activate(FFFrameSync *fs)
Examine the frames in the filter's input and try to produce output.
int(* on_event)(struct FFFrameSync *fs)
Callback called when a frame event is ready.
static const AVOption overlay_cuda_options[]
enum AVPixelFormat in_format_main
AVBufferRef * hw_frames_ctx
For hwaccel pixel formats, this should be a reference to the AVHWFramesContext describing the frames...
AVRational time_base
Time base for the output events.
static int activate(AVFilterContext *ctx)
FFmpeg internal API for CUDA.
AVFilterContext * src
source filter
static const AVFilterPad inputs[]
HW acceleration through CUDA.
static av_cold int overlay_cuda_init(AVFilterContext *avctx)
Initialize overlay_cuda.
static const AVFilterPad outputs[]
static int overlay_cuda_activate(AVFilterContext *avctx)
Activate overlay_cuda.
enum AVPixelFormat in_format_overlay
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
uint8_t * data
The data buffer.
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
static av_cold void overlay_cuda_uninit(AVFilterContext *avctx)
Uninitialize overlay_cuda.
This struct is allocated as AVHWDeviceContext.hwctx.
Describe the class of an AVClass context structure.
This struct describes a set or pool of "hardware" frames (i.e.
const char * name
Filter name.
AVFilterLink ** outputs
array of pointers to output links
static enum AVPixelFormat pix_fmts[]
int av_frame_make_writable(AVFrame *frame)
Ensure that the frame data is writable, avoiding data copy if possible.
static const AVFilterPad overlay_cuda_outputs[]
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
AVBufferRef * device_ref
A reference to the parent AVHWDeviceContext.
A reference to a data buffer.
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
static int query_formats(AVFilterContext *ctx)
AVBufferRef * av_buffer_ref(AVBufferRef *buf)
Create a new reference to an AVBuffer.
static int overlay_cuda_blend(FFFrameSync *fs)
Perform blend overlay picture over main picture.
static int formats_match(const enum AVPixelFormat format_main, const enum AVPixelFormat format_overlay)
Helper checks if we can process main and overlay pixel formats.
static int overlay_cuda_config_output(AVFilterLink *outlink)
Configure output.
AVCUDADeviceContext * hwctx
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.
enum AVPixelFormat sw_format
The pixel format identifying the actual data layout of the hardware frames.
AVPixelFormat
Pixel format.
AVFilter ff_vf_overlay_cuda
static int overlay_cuda_query_formats(AVFilterContext *avctx)
Query formats.
static enum AVPixelFormat supported_main_formats[]