Browse code

avfilter/vf_boxblur: Support 10bit planar formats

Michael Niedermayer authored on 2014/12/21 22:59:05
Showing 1 changed files
... ...
@@ -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