1 /*
2 * Original MPlayer filters by Richard Felker, Hampa Hug, Daniel Moreno,
3 * and Michael Niedermeyer.
4 *
5 * Copyright (c) 2014 James Darnley <james.darnley@gmail.com>
6 * Copyright (c) 2015 Arwa Arif <arwaarif1994@gmail.com>
7 *
8 * This file is part of FFmpeg.
9 *
10 * FFmpeg is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * FFmpeg is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License along
21 * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
22 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
23 */
24
25 #ifndef AVFILTER_EQ_H
26 #define AVFILTER_EQ_H
27
28 #include "avfilter.h"
29 #include "libavutil/eval.h"
30
31 static const char *const var_names[] = {
32 "n", // frame count
33 "pos", // frame position
34 "r", // frame rate
35 "t", // timestamp expressed in seconds
36 NULL
37 };
38
39 enum var_name {
40 VAR_N,
41 VAR_POS,
42 VAR_R,
43 VAR_T,
44 VAR_NB
45 };
46
47 typedef struct EQParameters {
48 void (*adjust)(struct EQParameters *eq, uint8_t *dst, int dst_stride,
49 const uint8_t *src, int src_stride, int w, int h);
50
51 uint8_t lut[256];
52
53 double brightness, contrast, gamma, gamma_weight;
54 int lut_clean;
55
56 } EQParameters;
57
58 typedef struct EQContext {
59 const AVClass *class;
60
61 EQParameters param[3];
62
63 char *contrast_expr;
64 AVExpr *contrast_pexpr;
65 double contrast;
66
67 char *brightness_expr;
68 AVExpr *brightness_pexpr;
69 double brightness;
70
71 char *saturation_expr;
72 AVExpr *saturation_pexpr;
73 double saturation;
74
75 char *gamma_expr;
76 AVExpr *gamma_pexpr;
77 double gamma;
78
79 char *gamma_weight_expr;
80 AVExpr *gamma_weight_pexpr;
81 double gamma_weight;
82
83 char *gamma_r_expr;
84 AVExpr *gamma_r_pexpr;
85 double gamma_r;
86
87 char *gamma_g_expr;
88 AVExpr *gamma_g_pexpr;
89 double gamma_g;
90
91 char *gamma_b_expr;
92 AVExpr *gamma_b_pexpr;
93 double gamma_b;
94
95 double var_values[VAR_NB];
96
97 void (*process)(struct EQParameters *par, uint8_t *dst, int dst_stride,
98 const uint8_t *src, int src_stride, int w, int h);
99
100 enum EvalMode { EVAL_MODE_INIT, EVAL_MODE_FRAME, EVAL_MODE_NB } eval_mode;
101 } EQContext;
102
process_c(EQParameters * param,uint8_t * dst,int dst_stride,const uint8_t * src,int src_stride,int w,int h)103 static void process_c(EQParameters *param, uint8_t *dst, int dst_stride,
104 const uint8_t *src, int src_stride, int w, int h)
105 {
106 int contrast = (int) (param->contrast * 256 * 16);
107 int brightness = ((int) (100.0 * param->brightness + 100.0) * 511) / 200 - 128 - contrast / 32;
108
109 for (int y = 0; y < h; y++) {
110 for (int x = 0; x < w; x++) {
111 int pel = ((src[y * src_stride + x] * contrast) >> 12) + brightness;
112
113 if (pel & ~255)
114 pel = (-pel) >> 31;
115
116 dst[y * dst_stride + x] = pel;
117 }
118 }
119 }
120
121 void ff_eq_init_x86(EQContext *eq);
122
ff_eq_init(EQContext * eq)123 static av_unused void ff_eq_init(EQContext *eq)
124 {
125 eq->process = process_c;
126 #if ARCH_X86
127 ff_eq_init_x86(eq);
128 #endif
129 }
130
131 #endif /* AVFILTER_EQ_H */
132