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
130
|
/*
* 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
"pos", // frame position
"r", // frame rate
"t", // timestamp expressed in seconds
NULL
};
enum var_name {
VAR_N,
VAR_POS,
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 /* AVFILTER_EQ_H */
|
