...
|
...
|
@@ -124,6 +124,9 @@ static int query_formats(AVFilterContext *ctx)
|
124
|
124
|
AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ420P,
|
125
|
125
|
AV_PIX_FMT_YUVJ440P,
|
126
|
126
|
AV_PIX_FMT_GBRP,
|
|
127
|
+ AV_PIX_FMT_YUV444P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV420P10,
|
|
128
|
+ AV_PIX_FMT_YUVA420P10,
|
|
129
|
+ AV_PIX_FMT_GBRP10,
|
127
|
130
|
AV_PIX_FMT_NONE
|
128
|
131
|
};
|
129
|
132
|
|
...
|
...
|
@@ -142,8 +145,8 @@ static int config_input(AVFilterLink *inlink)
|
142
|
142
|
char *expr;
|
143
|
143
|
int ret;
|
144
|
144
|
|
145
|
|
- if (!(s->temp[0] = av_malloc(FFMAX(w, h))) ||
|
146
|
|
- !(s->temp[1] = av_malloc(FFMAX(w, h))))
|
|
145
|
+ if (!(s->temp[0] = av_malloc(2*FFMAX(w, h))) ||
|
|
146
|
+ !(s->temp[1] = av_malloc(2*FFMAX(w, h))))
|
147
|
147
|
return AVERROR(ENOMEM);
|
148
|
148
|
|
149
|
149
|
s->hsub = desc->log2_chroma_w;
|
...
|
...
|
@@ -203,7 +206,7 @@ static int config_input(AVFilterLink *inlink)
|
203
|
203
|
return 0;
|
204
|
204
|
}
|
205
|
205
|
|
206
|
|
-static inline void blur(uint8_t *dst, int dst_step, const uint8_t *src, int src_step,
|
|
206
|
+static inline void blur8(uint8_t *dst, int dst_step, const uint8_t *src, int src_step,
|
207
|
207
|
int len, int radius)
|
208
|
208
|
{
|
209
|
209
|
/* Naive boxblur would sum source pixels from x-radius .. x+radius
|
...
|
...
|
@@ -245,34 +248,77 @@ static inline void blur(uint8_t *dst, int dst_step, const uint8_t *src, int src_
|
245
|
245
|
}
|
246
|
246
|
}
|
247
|
247
|
|
|
248
|
+static inline void blur16(uint16_t *dst, int dst_step, const uint16_t *src, int src_step,
|
|
249
|
+ int len, int radius)
|
|
250
|
+{
|
|
251
|
+ const int length = radius*2 + 1;
|
|
252
|
+ const int inv = ((1<<16) + length/2)/length;
|
|
253
|
+ int x, sum = src[radius*src_step];
|
|
254
|
+
|
|
255
|
+ for (x = 0; x < radius; x++)
|
|
256
|
+ sum += src[x*src_step]<<1;
|
|
257
|
+
|
|
258
|
+ sum = sum*inv + (1<<15);
|
|
259
|
+
|
|
260
|
+ for (x = 0; x <= radius; x++) {
|
|
261
|
+ sum += (src[(radius+x)*src_step] - src[(radius-x)*src_step])*inv;
|
|
262
|
+ dst[x*dst_step] = sum>>16;
|
|
263
|
+ }
|
|
264
|
+
|
|
265
|
+ for (; x < len-radius; x++) {
|
|
266
|
+ sum += (src[(radius+x)*src_step] - src[(x-radius-1)*src_step])*inv;
|
|
267
|
+ dst[x*dst_step] = sum >>16;
|
|
268
|
+ }
|
|
269
|
+
|
|
270
|
+ for (; x < len; x++) {
|
|
271
|
+ sum += (src[(2*len-radius-x-1)*src_step] - src[(x-radius-1)*src_step])*inv;
|
|
272
|
+ dst[x*dst_step] = sum>>16;
|
|
273
|
+ }
|
|
274
|
+}
|
|
275
|
+
|
|
276
|
+static inline void blur(uint8_t *dst, int dst_step, const uint8_t *src, int src_step,
|
|
277
|
+ int len, int radius, int pixsize)
|
|
278
|
+{
|
|
279
|
+ if (pixsize == 1) blur8 (dst, dst_step , src, src_step , len, radius);
|
|
280
|
+ else blur16((uint16_t*)dst, dst_step>>1, (const uint16_t*)src, src_step>>1, len, radius);
|
|
281
|
+}
|
|
282
|
+
|
248
|
283
|
static inline void blur_power(uint8_t *dst, int dst_step, const uint8_t *src, int src_step,
|
249
|
|
- int len, int radius, int power, uint8_t *temp[2])
|
|
284
|
+ int len, int radius, int power, uint8_t *temp[2], int pixsize)
|
250
|
285
|
{
|
251
|
286
|
uint8_t *a = temp[0], *b = temp[1];
|
252
|
287
|
|
253
|
288
|
if (radius && power) {
|
254
|
|
- blur(a, 1, src, src_step, len, radius);
|
|
289
|
+ blur(a, pixsize, src, src_step, len, radius, pixsize);
|
255
|
290
|
for (; power > 2; power--) {
|
256
|
291
|
uint8_t *c;
|
257
|
|
- blur(b, 1, a, 1, len, radius);
|
|
292
|
+ blur(b, pixsize, a, pixsize, len, radius, pixsize);
|
258
|
293
|
c = a; a = b; b = c;
|
259
|
294
|
}
|
260
|
295
|
if (power > 1) {
|
261
|
|
- blur(dst, dst_step, a, 1, len, radius);
|
|
296
|
+ blur(dst, dst_step, a, pixsize, len, radius, pixsize);
|
262
|
297
|
} else {
|
263
|
298
|
int i;
|
264
|
|
- for (i = 0; i < len; i++)
|
265
|
|
- dst[i*dst_step] = a[i];
|
|
299
|
+ if (pixsize == 1) {
|
|
300
|
+ for (i = 0; i < len; i++)
|
|
301
|
+ dst[i*dst_step] = a[i];
|
|
302
|
+ } else
|
|
303
|
+ for (i = 0; i < len; i++)
|
|
304
|
+ *(uint16_t*)(dst + i*dst_step) = ((uint16_t*)a)[i];
|
266
|
305
|
}
|
267
|
306
|
} else {
|
268
|
307
|
int i;
|
269
|
|
- for (i = 0; i < len; i++)
|
270
|
|
- dst[i*dst_step] = src[i*src_step];
|
|
308
|
+ if (pixsize == 1) {
|
|
309
|
+ for (i = 0; i < len; i++)
|
|
310
|
+ dst[i*dst_step] = src[i*src_step];
|
|
311
|
+ } else
|
|
312
|
+ for (i = 0; i < len; i++)
|
|
313
|
+ *(uint16_t*)(dst + i*dst_step) = *(uint16_t*)(src + i*src_step);
|
271
|
314
|
}
|
272
|
315
|
}
|
273
|
316
|
|
274
|
317
|
static void hblur(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize,
|
275
|
|
- int w, int h, int radius, int power, uint8_t *temp[2])
|
|
318
|
+ int w, int h, int radius, int power, uint8_t *temp[2], int pixsize)
|
276
|
319
|
{
|
277
|
320
|
int y;
|
278
|
321
|
|
...
|
...
|
@@ -280,12 +326,12 @@ static void hblur(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_li
|
280
|
280
|
return;
|
281
|
281
|
|
282
|
282
|
for (y = 0; y < h; y++)
|
283
|
|
- blur_power(dst + y*dst_linesize, 1, src + y*src_linesize, 1,
|
284
|
|
- w, radius, power, temp);
|
|
283
|
+ blur_power(dst + y*dst_linesize, pixsize, src + y*src_linesize, pixsize,
|
|
284
|
+ w, radius, power, temp, pixsize);
|
285
|
285
|
}
|
286
|
286
|
|
287
|
287
|
static void vblur(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize,
|
288
|
|
- int w, int h, int radius, int power, uint8_t *temp[2])
|
|
288
|
+ int w, int h, int radius, int power, uint8_t *temp[2], int pixsize)
|
289
|
289
|
{
|
290
|
290
|
int x;
|
291
|
291
|
|
...
|
...
|
@@ -293,8 +339,8 @@ static void vblur(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_li
|
293
|
293
|
return;
|
294
|
294
|
|
295
|
295
|
for (x = 0; x < w; x++)
|
296
|
|
- blur_power(dst + x, dst_linesize, src + x, src_linesize,
|
297
|
|
- h, radius, power, temp);
|
|
296
|
+ blur_power(dst + x*pixsize, dst_linesize, src + x*pixsize, src_linesize,
|
|
297
|
+ h, radius, power, temp, pixsize);
|
298
|
298
|
}
|
299
|
299
|
|
300
|
300
|
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
|
...
|
...
|
@@ -307,6 +353,9 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
|
307
|
307
|
int cw = FF_CEIL_RSHIFT(inlink->w, s->hsub), ch = FF_CEIL_RSHIFT(in->height, s->vsub);
|
308
|
308
|
int w[4] = { inlink->w, cw, cw, inlink->w };
|
309
|
309
|
int h[4] = { in->height, ch, ch, in->height };
|
|
310
|
+ const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
|
|
311
|
+ const int depth = desc->comp[0].depth_minus1 + 1;
|
|
312
|
+ const int pixsize = (depth+7)/8;
|
310
|
313
|
|
311
|
314
|
out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
|
312
|
315
|
if (!out) {
|
...
|
...
|
@@ -319,13 +368,13 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
|
319
|
319
|
hblur(out->data[plane], out->linesize[plane],
|
320
|
320
|
in ->data[plane], in ->linesize[plane],
|
321
|
321
|
w[plane], h[plane], s->radius[plane], s->power[plane],
|
322
|
|
- s->temp);
|
|
322
|
+ s->temp, pixsize);
|
323
|
323
|
|
324
|
324
|
for (plane = 0; plane < 4 && in->data[plane] && in->linesize[plane]; plane++)
|
325
|
325
|
vblur(out->data[plane], out->linesize[plane],
|
326
|
326
|
out->data[plane], out->linesize[plane],
|
327
|
327
|
w[plane], h[plane], s->radius[plane], s->power[plane],
|
328
|
|
- s->temp);
|
|
328
|
+ s->temp, pixsize);
|
329
|
329
|
|
330
|
330
|
av_frame_free(&in);
|
331
|
331
|
|