Line | Branch | Exec | Source |
---|---|---|---|
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 | #if FF_API_FRAME_PKT | ||
34 | "pos", // frame position | ||
35 | #endif | ||
36 | "r", // frame rate | ||
37 | "t", // timestamp expressed in seconds | ||
38 | NULL | ||
39 | }; | ||
40 | |||
41 | enum var_name { | ||
42 | VAR_N, | ||
43 | #if FF_API_FRAME_PKT | ||
44 | VAR_POS, | ||
45 | #endif | ||
46 | VAR_R, | ||
47 | VAR_T, | ||
48 | VAR_NB | ||
49 | }; | ||
50 | |||
51 | typedef struct EQParameters { | ||
52 | void (*adjust)(struct EQParameters *eq, uint8_t *dst, int dst_stride, | ||
53 | const uint8_t *src, int src_stride, int w, int h); | ||
54 | |||
55 | uint8_t lut[256]; | ||
56 | |||
57 | double brightness, contrast, gamma, gamma_weight; | ||
58 | int lut_clean; | ||
59 | |||
60 | } EQParameters; | ||
61 | |||
62 | typedef struct EQContext { | ||
63 | const AVClass *class; | ||
64 | |||
65 | EQParameters param[3]; | ||
66 | |||
67 | char *contrast_expr; | ||
68 | AVExpr *contrast_pexpr; | ||
69 | double contrast; | ||
70 | |||
71 | char *brightness_expr; | ||
72 | AVExpr *brightness_pexpr; | ||
73 | double brightness; | ||
74 | |||
75 | char *saturation_expr; | ||
76 | AVExpr *saturation_pexpr; | ||
77 | double saturation; | ||
78 | |||
79 | char *gamma_expr; | ||
80 | AVExpr *gamma_pexpr; | ||
81 | double gamma; | ||
82 | |||
83 | char *gamma_weight_expr; | ||
84 | AVExpr *gamma_weight_pexpr; | ||
85 | double gamma_weight; | ||
86 | |||
87 | char *gamma_r_expr; | ||
88 | AVExpr *gamma_r_pexpr; | ||
89 | double gamma_r; | ||
90 | |||
91 | char *gamma_g_expr; | ||
92 | AVExpr *gamma_g_pexpr; | ||
93 | double gamma_g; | ||
94 | |||
95 | char *gamma_b_expr; | ||
96 | AVExpr *gamma_b_pexpr; | ||
97 | double gamma_b; | ||
98 | |||
99 | double var_values[VAR_NB]; | ||
100 | |||
101 | void (*process)(struct EQParameters *par, uint8_t *dst, int dst_stride, | ||
102 | const uint8_t *src, int src_stride, int w, int h); | ||
103 | |||
104 | enum EvalMode { EVAL_MODE_INIT, EVAL_MODE_FRAME, EVAL_MODE_NB } eval_mode; | ||
105 | } EQContext; | ||
106 | |||
107 | 3 | static void process_c(EQParameters *param, uint8_t *dst, int dst_stride, | |
108 | const uint8_t *src, int src_stride, int w, int h) | ||
109 | { | ||
110 | 3 | int contrast = (int) (param->contrast * 256 * 16); | |
111 | 3 | int brightness = ((int) (100.0 * param->brightness + 100.0) * 511) / 200 - 128 - contrast / 32; | |
112 | |||
113 |
2/2✓ Branch 0 taken 768 times.
✓ Branch 1 taken 3 times.
|
771 | for (int y = 0; y < h; y++) { |
114 |
2/2✓ Branch 0 taken 196608 times.
✓ Branch 1 taken 768 times.
|
197376 | for (int x = 0; x < w; x++) { |
115 | 196608 | int pel = ((src[y * src_stride + x] * contrast) >> 12) + brightness; | |
116 | |||
117 |
2/2✓ Branch 0 taken 108887 times.
✓ Branch 1 taken 87721 times.
|
196608 | if (pel & ~255) |
118 | 108887 | pel = (-pel) >> 31; | |
119 | |||
120 | 196608 | dst[y * dst_stride + x] = pel; | |
121 | } | ||
122 | } | ||
123 | 3 | } | |
124 | |||
125 | void ff_eq_init_x86(EQContext *eq); | ||
126 | |||
127 | 13 | static av_unused void ff_eq_init(EQContext *eq) | |
128 | { | ||
129 | 13 | eq->process = process_c; | |
130 | #if ARCH_X86 | ||
131 | 13 | ff_eq_init_x86(eq); | |
132 | #endif | ||
133 | 13 | } | |
134 | |||
135 | #endif /* AVFILTER_EQ_H */ | ||
136 |