33#define FF_BUFQUEUE_SIZE 129
40#define SIZE FF_BUFQUEUE_SIZE
68#define OFFSET(x) offsetof(ATADenoiseContext, x)
69#define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
70#define VF AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
126 if (!(
s->size & 1)) {
130 s->radius =
s->size / 2;
140#define WFILTER_ROW(type, name) \
141static void fweight_row##name(const uint8_t *ssrc, uint8_t *ddst, \
142 const uint8_t *ssrcf[SIZE], \
143 int w, int mid, int size, \
144 int thra, int thrb, const float *weights) \
146 const type *src = (const type *)ssrc; \
147 const type **srcf = (const type **)ssrcf; \
148 type *dst = (type *)ddst; \
150 for (int x = 0; x < w; x++) { \
151 const int srcx = src[x]; \
152 unsigned lsumdiff = 0, rsumdiff = 0; \
153 unsigned ldiff, rdiff; \
158 for (int j = mid - 1, i = mid + 1; j >= 0 && i < size; j--, i++) { \
159 srcjx = srcf[j][x]; \
161 ldiff = FFABS(srcx - srcjx); \
163 if (ldiff > thra || \
166 sum += srcjx * weights[j]; \
167 wsum += weights[j]; \
169 srcix = srcf[i][x]; \
171 rdiff = FFABS(srcx - srcix); \
173 if (rdiff > thra || \
176 sum += srcix * weights[i]; \
177 wsum += weights[i]; \
180 dst[x] = lrintf(sum / wsum); \
187#define WFILTER_ROW_SERIAL(type, name) \
188static void fweight_row##name##_serial(const uint8_t *ssrc, uint8_t *ddst, \
189 const uint8_t *ssrcf[SIZE], \
190 int w, int mid, int size, \
191 int thra, int thrb, \
192 const float *weights) \
194 const type *src = (const type *)ssrc; \
195 const type **srcf = (const type **)ssrcf; \
196 type *dst = (type *)ddst; \
198 for (int x = 0; x < w; x++) { \
199 const int srcx = src[x]; \
200 unsigned lsumdiff = 0, rsumdiff = 0; \
201 unsigned ldiff, rdiff; \
206 for (int j = mid - 1; j >= 0; j--) { \
207 srcjx = srcf[j][x]; \
209 ldiff = FFABS(srcx - srcjx); \
211 if (ldiff > thra || \
214 sum += srcjx * weights[j]; \
215 wsum += weights[j]; \
218 for (int i = mid + 1; i < size; i++) { \
219 srcix = srcf[i][x]; \
221 rdiff = FFABS(srcx - srcix); \
223 if (rdiff > thra || \
226 sum += srcix * weights[i]; \
227 wsum += weights[i]; \
230 dst[x] = lrintf(sum / wsum); \
237#define FILTER_ROW(type, name) \
238static void filter_row##name(const uint8_t *ssrc, uint8_t *ddst, \
239 const uint8_t *ssrcf[SIZE], \
240 int w, int mid, int size, \
241 int thra, int thrb, const float *weights) \
243 const type *src = (const type *)ssrc; \
244 const type **srcf = (const type **)ssrcf; \
245 type *dst = (type *)ddst; \
247 for (int x = 0; x < w; x++) { \
248 const int srcx = src[x]; \
249 unsigned lsumdiff = 0, rsumdiff = 0; \
250 unsigned ldiff, rdiff; \
251 unsigned sum = srcx; \
255 for (int j = mid - 1, i = mid + 1; j >= 0 && i < size; j--, i++) { \
256 srcjx = srcf[j][x]; \
258 ldiff = FFABS(srcx - srcjx); \
260 if (ldiff > thra || \
266 srcix = srcf[i][x]; \
268 rdiff = FFABS(srcx - srcix); \
270 if (rdiff > thra || \
277 dst[x] = (sum + ((r + l + 1) >> 1)) / (r + l + 1); \
284#define FILTER_ROW_SERIAL(type, name) \
285static void filter_row##name##_serial(const uint8_t *ssrc, uint8_t *ddst, \
286 const uint8_t *ssrcf[SIZE], \
287 int w, int mid, int size, \
288 int thra, int thrb, \
289 const float *weights) \
291 const type *src = (const type *)ssrc; \
292 const type **srcf = (const type **)ssrcf; \
293 type *dst = (type *)ddst; \
295 for (int x = 0; x < w; x++) { \
296 const int srcx = src[x]; \
297 unsigned lsumdiff = 0, rsumdiff = 0; \
298 unsigned ldiff, rdiff; \
299 unsigned sum = srcx; \
303 for (int j = mid - 1; j >= 0; j--) { \
304 srcjx = srcf[j][x]; \
306 ldiff = FFABS(srcx - srcjx); \
308 if (ldiff > thra || \
315 for (int i = mid + 1; i < size; i++) { \
316 srcix = srcf[i][x]; \
318 rdiff = FFABS(srcx - srcix); \
320 if (rdiff > thra || \
327 dst[x] = (sum + ((r + l + 1) >> 1)) / (r + l + 1); \
340 const int size =
s->size;
341 const int mid =
s->mid;
344 for (p = 0; p <
s->nb_planes; p++) {
345 const float *
weights =
s->weights[p];
346 const int h =
s->planeheight[p];
347 const int w =
s->planewidth[p];
352 const int thra =
s->thra[p];
353 const int thrb =
s->thrb[p];
354 const uint8_t **
data = (
const uint8_t **)
s->data[p];
355 const int *linesize = (
const int *)
s->linesize[p];
356 const uint8_t *srcf[
SIZE];
358 if (!((1 << p) &
s->planes)) {
374 srcf[
i] += linesize[
i];
388 s->nb_planes =
desc->nb_components;
391 s->planeheight[0] =
s->planeheight[3] = inlink->
h;
393 s->planewidth[0] =
s->planewidth[3] = inlink->
w;
395 depth =
desc->comp[0].depth;
401 for (
int p = 0; p <
s->nb_planes; p++) {
402 if (depth == 8 &&
s->sigma[p] == INT16_MAX)
403 s->dsp.filter_row[p] =
s->algorithm ==
PARALLEL ? filter_row8 : filter_row8_serial;
404 else if (
s->sigma[p] == INT16_MAX)
405 s->dsp.filter_row[p] =
s->algorithm ==
PARALLEL ? filter_row16 : filter_row16_serial;
406 else if (depth == 8 &&
s->sigma[p] < INT16_MAX)
407 s->dsp.filter_row[p] =
s->algorithm ==
PARALLEL ? fweight_row8 : fweight_row8_serial;
408 else if (
s->sigma[p] < INT16_MAX)
409 s->dsp.filter_row[p] =
s->algorithm ==
PARALLEL ? fweight_row16 : fweight_row16_serial;
412 s->thra[0] =
s->fthra[0] * (1 << depth) - 1;
413 s->thra[1] =
s->fthra[1] * (1 << depth) - 1;
414 s->thra[2] =
s->fthra[2] * (1 << depth) - 1;
415 s->thrb[0] =
s->fthrb[0] * (1 << depth) - 1;
416 s->thrb[1] =
s->fthrb[1] * (1 << depth) - 1;
417 s->thrb[2] =
s->fthrb[2] * (1 << depth) - 1;
419 for (
int p = 0; p <
s->nb_planes; p++) {
420 float sigma =
s->radius *
s->sigma[p];
422 s->weights[p][
s->radius] = 1.f;
423 for (
int n = 1; n <=
s->radius; n++) {
424 s->weights[p][
s->radius + n] =
425 s->weights[p][
s->radius - n] =
expf(-0.5 * (n + 1) * (n + 1) / (sigma * sigma));
429#if ARCH_X86 && HAVE_X86ASM
444 if (
s->q.available !=
s->size) {
445 if (
s->q.available <
s->mid) {
446 for (
i = 0;
i <
s->mid;
i++) {
455 if (
s->q.available <
s->size) {
464 if (!
ctx->is_disabled) {
473 for (
i = 0;
i <
s->size;
i++) {
476 s->data[0][
i] =
frame->data[0];
477 s->data[1][
i] =
frame->data[1];
478 s->data[2][
i] =
frame->data[2];
479 s->linesize[0][
i] =
frame->linesize[0];
480 s->linesize[1][
i] =
frame->linesize[1];
481 s->linesize[2][
i] =
frame->linesize[2];
565 .p.name =
"atadenoise",
567 .p.priv_class = &atadenoise_class,
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
static const AVFilterPad inputs[]
static const AVFilterPad outputs[]
static int config_input(AVFilterLink *inlink)
static int process_command(AVFilterContext *ctx, const char *cmd, const char *args, char *res, int res_len, int flags)
static int request_frame(AVFilterLink *outlink)
const FFFilter ff_vf_atadenoise
static AVFormatContext * ctx
void ff_atadenoise_init_x86(ATADenoiseDSPContext *dsp, int depth, int algorithm, const float *sigma)
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
int ff_filter_process_command(AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)
Generic processing of user supplied commands that are set in the same way as the filter options.
int ff_request_frame(AVFilterLink *link)
Request an input frame from the filter at the other end of the link.
int ff_filter_execute(AVFilterContext *ctx, avfilter_action_func *func, void *arg, int *ret, int nb_jobs)
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Main libavfilter public API header.
static void ff_bufqueue_add(void *log, struct FFBufQueue *queue, AVFrame *buf)
Add a buffer to the queue.
static void ff_bufqueue_discard_all(struct FFBufQueue *queue)
Unref and remove all buffers from the queue.
static AVFrame * ff_bufqueue_peek(struct FFBufQueue *queue, unsigned index)
Get a buffer from the queue without altering it.
static AVFrame * ff_bufqueue_get(struct FFBufQueue *queue)
Get the first buffer from the queue and remove it.
#define flags(name, subs,...)
#define i(width, name, range_min, range_max)
#define AV_CEIL_RSHIFT(a, b)
static int filter_frame(DBEDecodeContext *s, AVFrame *frame)
int(* init)(AVBSFContext *ctx)
@ AV_OPT_TYPE_CONST
Special option type for declaring named constants.
@ AV_OPT_TYPE_FLAGS
Underlying C type is unsigned int.
@ AV_OPT_TYPE_INT
Underlying C type is int.
@ AV_OPT_TYPE_FLOAT
Underlying C type is float.
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
#define AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
Same as AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, except that the filter will have its filter_frame() c...
#define AVERROR_EOF
End of file.
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
AVFrame * av_frame_clone(const AVFrame *src)
Create a new frame that references the same data as src.
#define AV_LOG_WARNING
Something somehow does not look correct.
void av_image_copy_plane(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize, int bytewidth, int height)
Copy image plane from src to dst.
int av_image_fill_linesizes(int linesizes[4], enum AVPixelFormat pix_fmt, int width)
Fill plane linesizes for an image with pixel format pix_fmt and width width.
static const int weights[]
static av_cold void uninit(AVBitStreamFilterContext *ctx)
#define FILTER_INPUTS(array)
#define FILTER_OUTPUTS(array)
static int ff_slice_pos(int total, int jobnr, int nb_jobs)
Compute the boundary index for a slice when work of size total is split into nb_jobs slices.
#define FILTER_PIXFMTS_ARRAY(array)
#define AVFILTER_DEFINE_CLASS(fname)
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
static const struct @257111027162314367033347246032313251342043035002 planes[]
static int slice_end(AVCodecContext *avctx, AVFrame *pict, int *got_output)
Handle slice ends.
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
#define AV_PIX_FMT_GBRAP12
#define AV_PIX_FMT_YUV420P16
#define AV_PIX_FMT_YUV444P12
#define AV_PIX_FMT_YUV444P9
#define AV_PIX_FMT_YUV420P10
#define AV_PIX_FMT_YUV440P12
#define AV_PIX_FMT_GBRAP16
#define AV_PIX_FMT_YUV422P9
#define AV_PIX_FMT_YUVA444P10
#define AV_PIX_FMT_YUVA420P16
#define AV_PIX_FMT_YUV420P12
#define AV_PIX_FMT_YUVA420P10
#define AV_PIX_FMT_YUVA422P9
#define AV_PIX_FMT_YUV422P12
#define AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_GRAY12
#define AV_PIX_FMT_GBRP12
#define AV_PIX_FMT_YUV420P9
#define AV_PIX_FMT_YUVA420P9
#define AV_PIX_FMT_YUVA422P10
#define AV_PIX_FMT_YUV420P14
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
@ AV_PIX_FMT_YUV422P
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
@ AV_PIX_FMT_YUVA420P
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
@ AV_PIX_FMT_YUVA444P
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
@ AV_PIX_FMT_YUVA422P
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
#define AV_PIX_FMT_YUVA422P12
#define AV_PIX_FMT_YUV422P14
#define AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GRAY14
#define AV_PIX_FMT_YUV422P16
#define AV_PIX_FMT_YUV440P10
#define AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GBRAP10
#define AV_PIX_FMT_YUVA444P16
#define AV_PIX_FMT_YUVA422P16
#define AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_YUV444P14
#define AV_PIX_FMT_YUVA444P9
#define AV_PIX_FMT_GBRP14
#define AV_PIX_FMT_YUVA444P12
#define AV_PIX_FMT_YUV444P16
#define AV_PIX_FMT_YUV444P10
int(* filter_slice)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Describe the class of an AVClass context structure.
A link between two filters.
int w
agreed upon image width
int h
agreed upon image height
AVFilterContext * src
source filter
AVFilterContext * dst
dest filter
int format
agreed upon media format
A filter pad used for either input or output.
This structure describes decoded (raw) audio or video data.
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
int linesize[AV_NUM_DATA_POINTERS]
For video, a positive or negative value, which is typically indicating the size in bytes of each pict...
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Structure holding the queue.
Used for passing data between threads.
static enum AVPixelFormat pixel_fmts[]
static int process_command(AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)
static int filter_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
#define FILTER_ROW(type, name)
#define FILTER_ROW_SERIAL(type, name)
#define WFILTER_ROW(type, name)
static int config_input(AVFilterLink *inlink)
static int request_frame(AVFilterLink *outlink)
static int filter_frame(AVFilterLink *inlink, AVFrame *buf)
static const AVOption atadenoise_options[]
static av_cold void uninit(AVFilterContext *ctx)
#define WFILTER_ROW_SERIAL(type, name)
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
static int slice_start(SliceContext *sc, VVCContext *s, VVCFrameContext *fc, const CodedBitstreamUnit *unit, const int is_first_slice)