1 |
|
|
/* |
2 |
|
|
* Copyright (c) 2010 Bobby Bingham |
3 |
|
|
* |
4 |
|
|
* This file is part of FFmpeg. |
5 |
|
|
* |
6 |
|
|
* FFmpeg is free software; you can redistribute it and/or |
7 |
|
|
* modify it under the terms of the GNU Lesser General Public |
8 |
|
|
* License as published by the Free Software Foundation; either |
9 |
|
|
* version 2.1 of the License, or (at your option) any later version. |
10 |
|
|
* |
11 |
|
|
* FFmpeg is distributed in the hope that it will be useful, |
12 |
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
13 |
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
14 |
|
|
* Lesser General Public License for more details. |
15 |
|
|
* |
16 |
|
|
* You should have received a copy of the GNU Lesser General Public |
17 |
|
|
* License along with FFmpeg; if not, write to the Free Software |
18 |
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
19 |
|
|
*/ |
20 |
|
|
|
21 |
|
|
/** |
22 |
|
|
* @file |
23 |
|
|
* aspect ratio modification video filters |
24 |
|
|
*/ |
25 |
|
|
|
26 |
|
|
#include <float.h> |
27 |
|
|
|
28 |
|
|
#include "libavutil/common.h" |
29 |
|
|
#include "libavutil/eval.h" |
30 |
|
|
#include "libavutil/mathematics.h" |
31 |
|
|
#include "libavutil/opt.h" |
32 |
|
|
#include "libavutil/parseutils.h" |
33 |
|
|
#include "libavutil/pixdesc.h" |
34 |
|
|
|
35 |
|
|
#include "avfilter.h" |
36 |
|
|
#include "internal.h" |
37 |
|
|
#include "video.h" |
38 |
|
|
|
39 |
|
|
static const char *const var_names[] = { |
40 |
|
|
"w", |
41 |
|
|
"h", |
42 |
|
|
"a", "dar", |
43 |
|
|
"sar", |
44 |
|
|
"hsub", |
45 |
|
|
"vsub", |
46 |
|
|
NULL |
47 |
|
|
}; |
48 |
|
|
|
49 |
|
|
enum var_name { |
50 |
|
|
VAR_W, |
51 |
|
|
VAR_H, |
52 |
|
|
VAR_A, VAR_DAR, |
53 |
|
|
VAR_SAR, |
54 |
|
|
VAR_HSUB, |
55 |
|
|
VAR_VSUB, |
56 |
|
|
VARS_NB |
57 |
|
|
}; |
58 |
|
|
|
59 |
|
|
typedef struct AspectContext { |
60 |
|
|
const AVClass *class; |
61 |
|
|
AVRational dar; |
62 |
|
|
AVRational sar; |
63 |
|
|
int max; |
64 |
|
|
char *ratio_expr; |
65 |
|
|
} AspectContext; |
66 |
|
|
|
67 |
|
63 |
static int filter_frame(AVFilterLink *link, AVFrame *frame) |
68 |
|
|
{ |
69 |
|
63 |
AspectContext *s = link->dst->priv; |
70 |
|
|
|
71 |
|
63 |
frame->sample_aspect_ratio = s->sar; |
72 |
|
63 |
return ff_filter_frame(link->dst->outputs[0], frame); |
73 |
|
|
} |
74 |
|
|
|
75 |
|
|
#define OFFSET(x) offsetof(AspectContext, x) |
76 |
|
|
#define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM |
77 |
|
|
|
78 |
|
7 |
static inline void compute_dar(AVRational *dar, AVRational sar, int w, int h) |
79 |
|
|
{ |
80 |
✓✓✓✗
|
7 |
if (sar.num && sar.den) { |
81 |
|
3 |
av_reduce(&dar->num, &dar->den, sar.num * (int64_t)w, sar.den * (int64_t)h, INT_MAX); |
82 |
|
|
} else { |
83 |
|
4 |
av_reduce(&dar->num, &dar->den, w, h, INT_MAX); |
84 |
|
|
} |
85 |
|
7 |
} |
86 |
|
|
|
87 |
|
5 |
static int get_aspect_ratio(AVFilterLink *inlink, AVRational *aspect_ratio) |
88 |
|
|
{ |
89 |
|
5 |
AVFilterContext *ctx = inlink->dst; |
90 |
|
5 |
AspectContext *s = inlink->dst->priv; |
91 |
|
5 |
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); |
92 |
|
|
double var_values[VARS_NB], res; |
93 |
|
|
int ret; |
94 |
|
|
|
95 |
|
5 |
var_values[VAR_W] = inlink->w; |
96 |
|
5 |
var_values[VAR_H] = inlink->h; |
97 |
|
5 |
var_values[VAR_A] = (double) inlink->w / inlink->h; |
98 |
|
10 |
var_values[VAR_SAR] = inlink->sample_aspect_ratio.num ? |
99 |
✓✓ |
5 |
(double) inlink->sample_aspect_ratio.num / inlink->sample_aspect_ratio.den : 1; |
100 |
|
5 |
var_values[VAR_DAR] = var_values[VAR_A] * var_values[VAR_SAR]; |
101 |
|
5 |
var_values[VAR_HSUB] = 1 << desc->log2_chroma_w; |
102 |
|
5 |
var_values[VAR_VSUB] = 1 << desc->log2_chroma_h; |
103 |
|
|
|
104 |
|
|
/* evaluate new aspect ratio*/ |
105 |
|
5 |
ret = av_expr_parse_and_eval(&res, s->ratio_expr, |
106 |
|
|
var_names, var_values, |
107 |
|
|
NULL, NULL, NULL, NULL, NULL, 0, ctx); |
108 |
✗✓ |
5 |
if (ret < 0) { |
109 |
|
|
ret = av_parse_ratio(aspect_ratio, s->ratio_expr, s->max, 0, ctx); |
110 |
|
|
} else |
111 |
|
5 |
*aspect_ratio = av_d2q(res, s->max); |
112 |
|
|
|
113 |
✗✓ |
5 |
if (ret < 0) { |
114 |
|
|
av_log(ctx, AV_LOG_ERROR, |
115 |
|
|
"Error when evaluating the expression '%s'\n", s->ratio_expr); |
116 |
|
|
return ret; |
117 |
|
|
} |
118 |
✓✗✗✓
|
5 |
if (aspect_ratio->num < 0 || aspect_ratio->den <= 0) { |
119 |
|
|
av_log(ctx, AV_LOG_ERROR, |
120 |
|
|
"Invalid string '%s' for aspect ratio\n", s->ratio_expr); |
121 |
|
|
return AVERROR(EINVAL); |
122 |
|
|
} |
123 |
|
5 |
return 0; |
124 |
|
|
} |
125 |
|
|
|
126 |
|
|
#if CONFIG_SETDAR_FILTER |
127 |
|
|
|
128 |
|
3 |
static int setdar_config_props(AVFilterLink *outlink) |
129 |
|
|
{ |
130 |
|
3 |
AVFilterContext *ctx = outlink->src; |
131 |
|
3 |
AVFilterLink *inlink = ctx->inputs[0]; |
132 |
|
3 |
AspectContext *s = ctx->priv; |
133 |
|
|
AVRational dar; |
134 |
|
|
AVRational old_dar; |
135 |
|
3 |
AVRational old_sar = inlink->sample_aspect_ratio; |
136 |
|
|
int ret; |
137 |
|
|
|
138 |
✗✓ |
3 |
if ((ret = get_aspect_ratio(inlink, &s->dar))) |
139 |
|
|
return ret; |
140 |
|
|
|
141 |
✓✗✓✗
|
3 |
if (s->dar.num && s->dar.den) { |
142 |
|
3 |
av_reduce(&s->sar.num, &s->sar.den, |
143 |
|
3 |
s->dar.num * inlink->h, |
144 |
|
3 |
s->dar.den * inlink->w, INT_MAX); |
145 |
|
3 |
outlink->sample_aspect_ratio = s->sar; |
146 |
|
3 |
dar = s->dar; |
147 |
|
|
} else { |
148 |
|
|
outlink->sample_aspect_ratio = (AVRational){ 1, 1 }; |
149 |
|
|
dar = (AVRational){ inlink->w, inlink->h }; |
150 |
|
|
} |
151 |
|
|
|
152 |
|
3 |
compute_dar(&old_dar, old_sar, inlink->w, inlink->h); |
153 |
|
3 |
av_log(ctx, AV_LOG_VERBOSE, "w:%d h:%d dar:%d/%d sar:%d/%d -> dar:%d/%d sar:%d/%d\n", |
154 |
|
|
inlink->w, inlink->h, old_dar.num, old_dar.den, old_sar.num, old_sar.den, |
155 |
|
|
dar.num, dar.den, outlink->sample_aspect_ratio.num, outlink->sample_aspect_ratio.den); |
156 |
|
|
|
157 |
|
3 |
return 0; |
158 |
|
|
} |
159 |
|
|
|
160 |
|
|
static const AVOption setdar_options[] = { |
161 |
|
|
{ "dar", "set display aspect ratio", OFFSET(ratio_expr), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS }, |
162 |
|
|
{ "ratio", "set display aspect ratio", OFFSET(ratio_expr), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS }, |
163 |
|
|
{ "r", "set display aspect ratio", OFFSET(ratio_expr), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS }, |
164 |
|
|
{ "max", "set max value for nominator or denominator in the ratio", OFFSET(max), AV_OPT_TYPE_INT, {.i64=100}, 1, INT_MAX, FLAGS }, |
165 |
|
|
{ NULL } |
166 |
|
|
}; |
167 |
|
|
|
168 |
|
|
AVFILTER_DEFINE_CLASS(setdar); |
169 |
|
|
|
170 |
|
|
static const AVFilterPad avfilter_vf_setdar_inputs[] = { |
171 |
|
|
{ |
172 |
|
|
.name = "default", |
173 |
|
|
.type = AVMEDIA_TYPE_VIDEO, |
174 |
|
|
.filter_frame = filter_frame, |
175 |
|
|
}, |
176 |
|
|
{ NULL } |
177 |
|
|
}; |
178 |
|
|
|
179 |
|
|
static const AVFilterPad avfilter_vf_setdar_outputs[] = { |
180 |
|
|
{ |
181 |
|
|
.name = "default", |
182 |
|
|
.type = AVMEDIA_TYPE_VIDEO, |
183 |
|
|
.config_props = setdar_config_props, |
184 |
|
|
}, |
185 |
|
|
{ NULL } |
186 |
|
|
}; |
187 |
|
|
|
188 |
|
|
AVFilter ff_vf_setdar = { |
189 |
|
|
.name = "setdar", |
190 |
|
|
.description = NULL_IF_CONFIG_SMALL("Set the frame display aspect ratio."), |
191 |
|
|
.priv_size = sizeof(AspectContext), |
192 |
|
|
.priv_class = &setdar_class, |
193 |
|
|
.inputs = avfilter_vf_setdar_inputs, |
194 |
|
|
.outputs = avfilter_vf_setdar_outputs, |
195 |
|
|
}; |
196 |
|
|
|
197 |
|
|
#endif /* CONFIG_SETDAR_FILTER */ |
198 |
|
|
|
199 |
|
|
#if CONFIG_SETSAR_FILTER |
200 |
|
|
|
201 |
|
2 |
static int setsar_config_props(AVFilterLink *outlink) |
202 |
|
|
{ |
203 |
|
2 |
AVFilterContext *ctx = outlink->src; |
204 |
|
2 |
AVFilterLink *inlink = ctx->inputs[0]; |
205 |
|
2 |
AspectContext *s = ctx->priv; |
206 |
|
2 |
AVRational old_sar = inlink->sample_aspect_ratio; |
207 |
|
|
AVRational old_dar, dar; |
208 |
|
|
int ret; |
209 |
|
|
|
210 |
✗✓ |
2 |
if ((ret = get_aspect_ratio(inlink, &s->sar))) |
211 |
|
|
return ret; |
212 |
|
|
|
213 |
|
2 |
outlink->sample_aspect_ratio = s->sar; |
214 |
|
|
|
215 |
|
2 |
compute_dar(&old_dar, old_sar, inlink->w, inlink->h); |
216 |
|
2 |
compute_dar(&dar, s->sar, inlink->w, inlink->h); |
217 |
|
2 |
av_log(ctx, AV_LOG_VERBOSE, "w:%d h:%d sar:%d/%d dar:%d/%d -> sar:%d/%d dar:%d/%d\n", |
218 |
|
|
inlink->w, inlink->h, old_sar.num, old_sar.den, old_dar.num, old_dar.den, |
219 |
|
|
outlink->sample_aspect_ratio.num, outlink->sample_aspect_ratio.den, dar.num, dar.den); |
220 |
|
|
|
221 |
|
2 |
return 0; |
222 |
|
|
} |
223 |
|
|
|
224 |
|
|
static const AVOption setsar_options[] = { |
225 |
|
|
{ "sar", "set sample (pixel) aspect ratio", OFFSET(ratio_expr), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS }, |
226 |
|
|
{ "ratio", "set sample (pixel) aspect ratio", OFFSET(ratio_expr), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS }, |
227 |
|
|
{ "r", "set sample (pixel) aspect ratio", OFFSET(ratio_expr), AV_OPT_TYPE_STRING, { .str = "0" }, .flags = FLAGS }, |
228 |
|
|
{ "max", "set max value for nominator or denominator in the ratio", OFFSET(max), AV_OPT_TYPE_INT, {.i64=100}, 1, INT_MAX, FLAGS }, |
229 |
|
|
{ NULL } |
230 |
|
|
}; |
231 |
|
|
|
232 |
|
|
AVFILTER_DEFINE_CLASS(setsar); |
233 |
|
|
|
234 |
|
|
static const AVFilterPad avfilter_vf_setsar_inputs[] = { |
235 |
|
|
{ |
236 |
|
|
.name = "default", |
237 |
|
|
.type = AVMEDIA_TYPE_VIDEO, |
238 |
|
|
.filter_frame = filter_frame, |
239 |
|
|
}, |
240 |
|
|
{ NULL } |
241 |
|
|
}; |
242 |
|
|
|
243 |
|
|
static const AVFilterPad avfilter_vf_setsar_outputs[] = { |
244 |
|
|
{ |
245 |
|
|
.name = "default", |
246 |
|
|
.type = AVMEDIA_TYPE_VIDEO, |
247 |
|
|
.config_props = setsar_config_props, |
248 |
|
|
}, |
249 |
|
|
{ NULL } |
250 |
|
|
}; |
251 |
|
|
|
252 |
|
|
AVFilter ff_vf_setsar = { |
253 |
|
|
.name = "setsar", |
254 |
|
|
.description = NULL_IF_CONFIG_SMALL("Set the pixel sample aspect ratio."), |
255 |
|
|
.priv_size = sizeof(AspectContext), |
256 |
|
|
.priv_class = &setsar_class, |
257 |
|
|
.inputs = avfilter_vf_setsar_inputs, |
258 |
|
|
.outputs = avfilter_vf_setsar_outputs, |
259 |
|
|
}; |
260 |
|
|
|
261 |
|
|
#endif /* CONFIG_SETSAR_FILTER */ |