Statistics
| Branch: | Revision:

ffmpeg / libavfilter / libmpcodecs / vf_boxblur.c @ e4852fb3

History | View | Annotate | Download (6.75 KB)

1
/*
2
 * Copyright (C) 2002 Michael Niedermayer <michaelni@gmx.at>
3
 *
4
 * This file is part of MPlayer.
5
 *
6
 * MPlayer is free software; you can redistribute it and/or modify
7
 * it under the terms of the GNU General Public License as published by
8
 * the Free Software Foundation; either version 2 of the License, or
9
 * (at your option) any later version.
10
 *
11
 * MPlayer is distributed in the hope that it will be useful,
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 * GNU General Public License for more details.
15
 *
16
 * You should have received a copy of the GNU General Public License along
17
 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
18
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19
 */
20

    
21
#include <stdio.h>
22
#include <stdlib.h>
23
#include <string.h>
24
#include <inttypes.h>
25
#include <assert.h>
26

    
27
#include "mp_msg.h"
28
#include "img_format.h"
29
#include "mp_image.h"
30
#include "vf.h"
31

    
32

    
33
//===========================================================================//
34

    
35
typedef struct FilterParam{
36
        int radius;
37
        int power;
38
}FilterParam;
39

    
40
struct vf_priv_s {
41
        FilterParam lumaParam;
42
        FilterParam chromaParam;
43
};
44

    
45

    
46
/***************************************************************************/
47

    
48

    
49
static int config(struct vf_instance *vf,
50
        int width, int height, int d_width, int d_height,
51
        unsigned int flags, unsigned int outfmt){
52

    
53
        return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
54
}
55

    
56
static inline void blur(uint8_t *dst, uint8_t *src, int w, int radius, int dstStep, int srcStep){
57
        int x;
58
        const int length= radius*2 + 1;
59
        const int inv= ((1<<16) + length/2)/length;
60

    
61
        int sum= 0;
62

    
63
        for(x=0; x<radius; x++){
64
                sum+= src[x*srcStep]<<1;
65
        }
66
        sum+= src[radius*srcStep];
67

    
68
        for(x=0; x<=radius; x++){
69
                sum+= src[(radius+x)*srcStep] - src[(radius-x)*srcStep];
70
                dst[x*dstStep]= (sum*inv + (1<<15))>>16;
71
        }
72

    
73
        for(; x<w-radius; x++){
74
                sum+= src[(radius+x)*srcStep] - src[(x-radius-1)*srcStep];
75
                dst[x*dstStep]= (sum*inv + (1<<15))>>16;
76
        }
77

    
78
        for(; x<w; x++){
79
                sum+= src[(2*w-radius-x-1)*srcStep] - src[(x-radius-1)*srcStep];
80
                dst[x*dstStep]= (sum*inv + (1<<15))>>16;
81
        }
82
}
83

    
84
static inline void blur2(uint8_t *dst, uint8_t *src, int w, int radius, int power, int dstStep, int srcStep){
85
        uint8_t temp[2][4096];
86
        uint8_t *a= temp[0], *b=temp[1];
87

    
88
        if(radius){
89
                blur(a, src, w, radius, 1, srcStep);
90
                for(; power>2; power--){
91
                        uint8_t *c;
92
                        blur(b, a, w, radius, 1, 1);
93
                        c=a; a=b; b=c;
94
                }
95
                if(power>1)
96
                        blur(dst, a, w, radius, dstStep, 1);
97
                else{
98
                        int i;
99
                        for(i=0; i<w; i++)
100
                                dst[i*dstStep]= a[i];
101
                }
102
        }else{
103
                int i;
104
                for(i=0; i<w; i++)
105
                        dst[i*dstStep]= src[i*srcStep];
106
        }
107
}
108

    
109
static void hBlur(uint8_t *dst, uint8_t *src, int w, int h, int dstStride, int srcStride, int radius, int power){
110
        int y;
111

    
112
        if(radius==0 && dst==src) return;
113

    
114
        for(y=0; y<h; y++){
115
                blur2(dst + y*dstStride, src + y*srcStride, w, radius, power, 1, 1);
116
        }
117
}
118

    
119
//FIXME optimize (x before y !!!)
120
static void vBlur(uint8_t *dst, uint8_t *src, int w, int h, int dstStride, int srcStride, int radius, int power){
121
        int x;
122

    
123
        if(radius==0 && dst==src) return;
124

    
125
        for(x=0; x<w; x++){
126
                blur2(dst + x, src + x, h, radius, power, dstStride, srcStride);
127
        }
128
}
129

    
130
static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){
131
        int cw= mpi->w >> mpi->chroma_x_shift;
132
        int ch= mpi->h >> mpi->chroma_y_shift;
133

    
134
        mp_image_t *dmpi=vf_get_image(vf->next,mpi->imgfmt,
135
                MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE | MP_IMGFLAG_READABLE,
136
                mpi->w,mpi->h);
137

    
138
        assert(mpi->flags&MP_IMGFLAG_PLANAR);
139

    
140
        hBlur(dmpi->planes[0], mpi->planes[0], mpi->w,mpi->h,
141
                dmpi->stride[0], mpi->stride[0], vf->priv->lumaParam.radius, vf->priv->lumaParam.power);
142
        hBlur(dmpi->planes[1], mpi->planes[1], cw,ch,
143
                dmpi->stride[1], mpi->stride[1], vf->priv->chromaParam.radius, vf->priv->chromaParam.power);
144
        hBlur(dmpi->planes[2], mpi->planes[2], cw,ch,
145
                dmpi->stride[2], mpi->stride[2], vf->priv->chromaParam.radius, vf->priv->chromaParam.power);
146

    
147
        vBlur(dmpi->planes[0], dmpi->planes[0], mpi->w,mpi->h,
148
                dmpi->stride[0], dmpi->stride[0], vf->priv->lumaParam.radius, vf->priv->lumaParam.power);
149
        vBlur(dmpi->planes[1], dmpi->planes[1], cw,ch,
150
                dmpi->stride[1], dmpi->stride[1], vf->priv->chromaParam.radius, vf->priv->chromaParam.power);
151
        vBlur(dmpi->planes[2], dmpi->planes[2], cw,ch,
152
                dmpi->stride[2], dmpi->stride[2], vf->priv->chromaParam.radius, vf->priv->chromaParam.power);
153

    
154
        return vf_next_put_image(vf,dmpi, pts);
155
}
156

    
157
//===========================================================================//
158

    
159
static int query_format(struct vf_instance *vf, unsigned int fmt){
160
        switch(fmt)
161
        {
162
        case IMGFMT_YV12:
163
        case IMGFMT_I420:
164
        case IMGFMT_IYUV:
165
        case IMGFMT_YVU9:
166
        case IMGFMT_444P:
167
        case IMGFMT_422P:
168
        case IMGFMT_411P:
169
                return vf_next_query_format(vf, fmt);
170
        }
171
        return 0;
172
}
173

    
174
static int vf_open(vf_instance_t *vf, char *args){
175
        int e;
176

    
177
        vf->config=config;
178
        vf->put_image=put_image;
179
//        vf->get_image=get_image;
180
        vf->query_format=query_format;
181
        vf->priv=malloc(sizeof(struct vf_priv_s));
182
        memset(vf->priv, 0, sizeof(struct vf_priv_s));
183

    
184
        if(args==NULL) return 0;
185

    
186
        e=sscanf(args, "%d:%d:%d:%d",
187
                &vf->priv->lumaParam.radius,
188
                &vf->priv->lumaParam.power,
189
                &vf->priv->chromaParam.radius,
190
                &vf->priv->chromaParam.power
191
                );
192

    
193
        if(e==2){
194
                vf->priv->chromaParam.radius= vf->priv->lumaParam.radius;
195
                vf->priv->chromaParam.power = vf->priv->lumaParam.power;
196
        }else if(e!=4)
197
                return 0;
198

    
199
        if(vf->priv->lumaParam.radius < 0) return 0;
200
        if(vf->priv->chromaParam.radius < 0) return 0;
201

    
202
        return 1;
203
}
204

    
205
const vf_info_t vf_info_boxblur = {
206
    "box blur",
207
    "boxblur",
208
    "Michael Niedermayer",
209
    "",
210
    vf_open,
211
    NULL
212
};
213

    
214
//===========================================================================//