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