Statistics
| Branch: | Revision:

ffmpeg / libavfilter / libmpcodecs / vf_hue.c @ e4852fb3

History | View | Annotate | Download (6.01 KB)

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

    
19
#include <stdio.h>
20
#include <stdlib.h>
21
#include <string.h>
22
#include <inttypes.h>
23
#include <math.h>
24

    
25
#include "config.h"
26
#include "mp_msg.h"
27
#include "cpudetect.h"
28

    
29
#include "img_format.h"
30
#include "mp_image.h"
31
#include "vf.h"
32

    
33
#include "libvo/video_out.h"
34

    
35
#include "m_option.h"
36
#include "m_struct.h"
37

    
38
static struct vf_priv_s {
39
        uint8_t *buf[2];
40
        float hue;
41
        float saturation;
42
} const vf_priv_dflt = {
43
  {NULL, NULL},
44
  0.0,
45
  1.0,
46
};
47

    
48
static void process_C(uint8_t *udst, uint8_t *vdst, uint8_t *usrc, uint8_t *vsrc, int dststride, int srcstride,
49
                    int w, int h, float hue, float sat)
50
{
51
        int i;
52
        const int s= rint(sin(hue) * (1<<16) * sat);
53
        const int c= rint(cos(hue) * (1<<16) * sat);
54

    
55
        while (h--) {
56
                for (i = 0; i<w; i++)
57
                {
58
                        const int u= usrc[i] - 128;
59
                        const int v= vsrc[i] - 128;
60
                        int new_u= (c*u - s*v + (1<<15) + (128<<16))>>16;
61
                        int new_v= (s*u + c*v + (1<<15) + (128<<16))>>16;
62
                        if(new_u & 768) new_u= (-new_u)>>31;
63
                        if(new_v & 768) new_v= (-new_v)>>31;
64
                        udst[i]= new_u;
65
                        vdst[i]= new_v;
66
                }
67
                usrc += srcstride;
68
                vsrc += srcstride;
69
                udst += dststride;
70
                vdst += dststride;
71
        }
72
}
73

    
74
static void (*process)(uint8_t *udst, uint8_t *vdst, uint8_t *usrc, uint8_t *vsrc, int dststride, int srcstride,
75
                    int w, int h, float hue, float sat);
76

    
77
/* FIXME: add packed yuv version of process */
78

    
79
static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts)
80
{
81
        mp_image_t *dmpi;
82

    
83
        dmpi=vf_get_image(vf->next, mpi->imgfmt,
84
                          MP_IMGTYPE_EXPORT, 0,
85
                          mpi->w, mpi->h);
86

    
87
        dmpi->planes[0] = mpi->planes[0];
88
        dmpi->stride[0] = mpi->stride[0];
89
        dmpi->stride[1] = mpi->stride[1];
90
        dmpi->stride[2] = mpi->stride[2];
91

    
92
        if (!vf->priv->buf[0]){
93
                vf->priv->buf[0] = malloc(mpi->stride[1]*mpi->h >> mpi->chroma_y_shift);
94
                vf->priv->buf[1] = malloc(mpi->stride[2]*mpi->h >> mpi->chroma_y_shift);
95
        }
96

    
97
        if (vf->priv->hue == 0 && vf->priv->saturation == 1){
98
                dmpi->planes[1] = mpi->planes[1];
99
                dmpi->planes[2] = mpi->planes[2];
100
        }else {
101
                dmpi->planes[1] = vf->priv->buf[0];
102
                dmpi->planes[2] = vf->priv->buf[1];
103
                process(dmpi->planes[1], dmpi->planes[2],
104
                        mpi->planes[1], mpi->planes[2],
105
                        dmpi->stride[1],mpi->stride[1],
106
                        mpi->w>> mpi->chroma_x_shift, mpi->h>> mpi->chroma_y_shift,
107
                        vf->priv->hue, vf->priv->saturation);
108
        }
109

    
110
        return vf_next_put_image(vf,dmpi, pts);
111
}
112

    
113
static int control(struct vf_instance *vf, int request, void* data)
114
{
115
        vf_equalizer_t *eq;
116

    
117
        switch (request) {
118
        case VFCTRL_SET_EQUALIZER:
119
                eq = data;
120
                if (!strcmp(eq->item,"hue")) {
121
                        vf->priv->hue = eq->value * M_PI / 100;
122
                        return CONTROL_TRUE;
123
                } else if (!strcmp(eq->item,"saturation")) {
124
                        vf->priv->saturation = (eq->value + 100)/100.0;
125
                        return CONTROL_TRUE;
126
                }
127
                break;
128
        case VFCTRL_GET_EQUALIZER:
129
                eq = data;
130
                if (!strcmp(eq->item,"hue")) {
131
                        eq->value = rint(vf->priv->hue *100 / M_PI);
132
                        return CONTROL_TRUE;
133
                }else if (!strcmp(eq->item,"saturation")) {
134
                        eq->value = rint(vf->priv->saturation*100 - 100);
135
                        return CONTROL_TRUE;
136
                }
137
                break;
138
        }
139
        return vf_next_control(vf, request, data);
140
}
141

    
142
static int query_format(struct vf_instance *vf, unsigned int fmt)
143
{
144
        switch (fmt) {
145
        case IMGFMT_YVU9:
146
        case IMGFMT_IF09:
147
        case IMGFMT_YV12:
148
        case IMGFMT_I420:
149
        case IMGFMT_IYUV:
150
        case IMGFMT_CLPL:
151
        case IMGFMT_444P:
152
        case IMGFMT_422P:
153
        case IMGFMT_411P:
154
                return vf_next_query_format(vf, fmt);
155
        }
156
        return 0;
157
}
158

    
159
static void uninit(struct vf_instance *vf)
160
{
161
        free(vf->priv->buf[0]);
162
        free(vf->priv->buf[1]);
163
        free(vf->priv);
164
}
165

    
166
static int vf_open(vf_instance_t *vf, char *args)
167
{
168
        vf->control=control;
169
        vf->query_format=query_format;
170
        vf->put_image=put_image;
171
        vf->uninit=uninit;
172

    
173
        vf->priv->hue *= M_PI / 180.0;
174

    
175
        process = process_C;
176
        return 1;
177
}
178

    
179
#define ST_OFF(f) M_ST_OFF(struct vf_priv_s,f)
180
static const m_option_t vf_opts_fields[] = {
181
  {"hue", ST_OFF(hue), CONF_TYPE_FLOAT, M_OPT_RANGE,-180.0 ,180.0, NULL},
182
  {"saturation", ST_OFF(saturation), CONF_TYPE_FLOAT, M_OPT_RANGE,-10.0 ,10.0, NULL},
183
  { NULL, NULL, 0, 0, 0, 0,  NULL }
184
};
185

    
186
static const m_struct_t vf_opts = {
187
  "hue",
188
  sizeof(struct vf_priv_s),
189
  &vf_priv_dflt,
190
  vf_opts_fields
191
};
192

    
193
const vf_info_t vf_info_hue = {
194
        "hue changer",
195
        "hue",
196
        "Michael Niedermayer",
197
        "",
198
        vf_open,
199
        &vf_opts
200
};