FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavfilter/vf_transpose.c
Date: 2025-10-24 10:15:11
Exec Total Coverage
Lines: 159 175 90.9%
Functions: 17 17 100.0%
Branches: 91 105 86.7%
Line Branch Exec Source
1 /*
2 * Copyright (c) 2010 Stefano Sabatini
3 * Copyright (c) 2008 Vitor Sessak
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file
24 * transposition filter
25 * Based on MPlayer libmpcodecs/vf_rotate.c.
26 */
27
28 #include <stdio.h>
29
30 #include "libavutil/avassert.h"
31 #include "libavutil/imgutils.h"
32 #include "libavutil/internal.h"
33 #include "libavutil/intreadwrite.h"
34 #include "libavutil/opt.h"
35 #include "libavutil/pixdesc.h"
36
37 #include "avfilter.h"
38 #include "filters.h"
39 #include "formats.h"
40 #include "video.h"
41 #include "transpose.h"
42
43 typedef struct TransContext {
44 const AVClass *class;
45 int hsub, vsub;
46 int planes;
47 int pixsteps[4];
48
49 int passthrough; ///< PassthroughType, landscape passthrough mode enabled
50 int dir; ///< TransposeDir
51
52 TransVtable vtables[4];
53 } TransContext;
54
55 163 static int query_formats(const AVFilterContext *ctx,
56 AVFilterFormatsConfig **cfg_in,
57 AVFilterFormatsConfig **cfg_out)
58 {
59 163 AVFilterFormats *pix_fmts = NULL;
60 const AVPixFmtDescriptor *desc;
61 int fmt, ret;
62
63
2/2
✓ Branch 1 taken 43521 times.
✓ Branch 2 taken 163 times.
 43684 for (fmt = 0; desc = av_pix_fmt_desc_get(fmt); fmt++) {
64
2/2
✓ Branch 0 taken 43358 times.
✓ Branch 1 taken 163 times.
 43521 if (!(desc->flags & AV_PIX_FMT_FLAG_PAL ||
65
2/2
✓ Branch 0 taken 40750 times.
✓ Branch 1 taken 2608 times.
 43358 desc->flags & AV_PIX_FMT_FLAG_HWACCEL ||
66
2/2
✓ Branch 0 taken 39772 times.
✓ Branch 1 taken 978 times.
 40750 desc->flags & AV_PIX_FMT_FLAG_BITSTREAM ||
67
3/4
✓ Branch 0 taken 31948 times.
✓ Branch 1 taken 7824 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 31948 times.
 71720 desc->log2_chroma_w != desc->log2_chroma_h) &&
68 31948 (ret = ff_add_format(&pix_fmts, fmt)) < 0)
69 ✗ return ret;
70 }
71
72
73 163 return ff_set_common_formats2(ctx, cfg_in, cfg_out, pix_fmts);
74 }
75
76 558844 static inline void transpose_block_8_c(uint8_t *src, ptrdiff_t src_linesize,
77 uint8_t *dst, ptrdiff_t dst_linesize,
78 int w, int h)
79 {
80 int x, y;
81
2/2
✓ Branch 0 taken 4494856 times.
✓ Branch 1 taken 558844 times.
 5053700 for (y = 0; y < h; y++, dst += dst_linesize, src++)
82
2/2
✓ Branch 0 taken 37061136 times.
✓ Branch 1 taken 4494856 times.
 41555992 for (x = 0; x < w; x++)
83 37061136 dst[x] = src[x*src_linesize];
84 558844 }
85
86 557333 static void transpose_8x8_8_c(uint8_t *src, ptrdiff_t src_linesize,
87 uint8_t *dst, ptrdiff_t dst_linesize)
88 {
89 557333 transpose_block_8_c(src, src_linesize, dst, dst_linesize, 8, 8);
90 557333 }
91
92 682704 static inline void transpose_block_16_c(uint8_t *src, ptrdiff_t src_linesize,
93 uint8_t *dst, ptrdiff_t dst_linesize,
94 int w, int h)
95 {
96 int x, y;
97
2/2
✓ Branch 0 taken 5461632 times.
✓ Branch 1 taken 682704 times.
 6144336 for (y = 0; y < h; y++, dst += dst_linesize, src += 2)
98
2/2
✓ Branch 0 taken 43693056 times.
✓ Branch 1 taken 5461632 times.
 49154688 for (x = 0; x < w; x++)
99 43693056 *((uint16_t *)(dst + 2*x)) = *((uint16_t *)(src + x*src_linesize));
100 682704 }
101
102 682704 static void transpose_8x8_16_c(uint8_t *src, ptrdiff_t src_linesize,
103 uint8_t *dst, ptrdiff_t dst_linesize)
104 {
105 682704 transpose_block_16_c(src, src_linesize, dst, dst_linesize, 8, 8);
106 682704 }
107
108 9504 static inline void transpose_block_24_c(uint8_t *src, ptrdiff_t src_linesize,
109 uint8_t *dst, ptrdiff_t dst_linesize,
110 int w, int h)
111 {
112 int x, y;
113
2/2
✓ Branch 0 taken 76032 times.
✓ Branch 1 taken 9504 times.
 85536 for (y = 0; y < h; y++, dst += dst_linesize) {
114
2/2
✓ Branch 0 taken 608256 times.
✓ Branch 1 taken 76032 times.
 684288 for (x = 0; x < w; x++) {
115 608256 int32_t v = AV_RB24(src + x*src_linesize + y*3);
116 608256 AV_WB24(dst + 3*x, v);
117 }
118 }
119 9504 }
120
121 9504 static void transpose_8x8_24_c(uint8_t *src, ptrdiff_t src_linesize,
122 uint8_t *dst, ptrdiff_t dst_linesize)
123 {
124 9504 transpose_block_24_c(src, src_linesize, dst, dst_linesize, 8, 8);
125 9504 }
126
127 131472 static inline void transpose_block_32_c(uint8_t *src, ptrdiff_t src_linesize,
128 uint8_t *dst, ptrdiff_t dst_linesize,
129 int w, int h)
130 {
131 int x, y;
132
2/2
✓ Branch 0 taken 1051776 times.
✓ Branch 1 taken 131472 times.
 1183248 for (y = 0; y < h; y++, dst += dst_linesize, src += 4) {
133
2/2
✓ Branch 0 taken 8414208 times.
✓ Branch 1 taken 1051776 times.
 9465984 for (x = 0; x < w; x++)
134 8414208 *((uint32_t *)(dst + 4*x)) = *((uint32_t *)(src + x*src_linesize));
135 }
136 131472 }
137
138 131472 static void transpose_8x8_32_c(uint8_t *src, ptrdiff_t src_linesize,
139 uint8_t *dst, ptrdiff_t dst_linesize)
140 {
141 131472 transpose_block_32_c(src, src_linesize, dst, dst_linesize, 8, 8);
142 131472 }
143
144 19008 static inline void transpose_block_48_c(uint8_t *src, ptrdiff_t src_linesize,
145 uint8_t *dst, ptrdiff_t dst_linesize,
146 int w, int h)
147 {
148 int x, y;
149
2/2
✓ Branch 0 taken 152064 times.
✓ Branch 1 taken 19008 times.
 171072 for (y = 0; y < h; y++, dst += dst_linesize, src += 6) {
150
2/2
✓ Branch 0 taken 1216512 times.
✓ Branch 1 taken 152064 times.
 1368576 for (x = 0; x < w; x++) {
151 1216512 int64_t v = AV_RB48(src + x*src_linesize);
152 1216512 AV_WB48(dst + 6*x, v);
153 }
154 }
155 19008 }
156
157 19008 static void transpose_8x8_48_c(uint8_t *src, ptrdiff_t src_linesize,
158 uint8_t *dst, ptrdiff_t dst_linesize)
159 {
160 19008 transpose_block_48_c(src, src_linesize, dst, dst_linesize, 8, 8);
161 19008 }
162
163 31680 static inline void transpose_block_64_c(uint8_t *src, ptrdiff_t src_linesize,
164 uint8_t *dst, ptrdiff_t dst_linesize,
165 int w, int h)
166 {
167 int x, y;
168
2/2
✓ Branch 0 taken 253440 times.
✓ Branch 1 taken 31680 times.
 285120 for (y = 0; y < h; y++, dst += dst_linesize, src += 8)
169
2/2
✓ Branch 0 taken 2027520 times.
✓ Branch 1 taken 253440 times.
 2280960 for (x = 0; x < w; x++)
170 2027520 *((uint64_t *)(dst + 8*x)) = *((uint64_t *)(src + x*src_linesize));
171 31680 }
172
173 31680 static void transpose_8x8_64_c(uint8_t *src, ptrdiff_t src_linesize,
174 uint8_t *dst, ptrdiff_t dst_linesize)
175 {
176 31680 transpose_block_64_c(src, src_linesize, dst, dst_linesize, 8, 8);
177 31680 }
178
179 162 static int config_props_output(AVFilterLink *outlink)
180 {
181 162 AVFilterContext *ctx = outlink->src;
182 162 TransContext *s = ctx->priv;
183 162 AVFilterLink *inlink = ctx->inputs[0];
184 162 const AVPixFmtDescriptor *desc_out = av_pix_fmt_desc_get(outlink->format);
185 162 const AVPixFmtDescriptor *desc_in = av_pix_fmt_desc_get(inlink->format);
186
187
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 162 times.
 162 if (s->dir&4) {
188 ✗ av_log(ctx, AV_LOG_WARNING,
189 "dir values greater than 3 are deprecated, use the passthrough option instead\n");
190 ✗ s->dir &= 3;
191 ✗ s->passthrough = TRANSPOSE_PT_TYPE_LANDSCAPE;
192 }
193
194
3/4
✓ Branch 0 taken 161 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 161 times.
✗ Branch 3 not taken.
 162 if ((inlink->w >= inlink->h && s->passthrough == TRANSPOSE_PT_TYPE_LANDSCAPE) ||
195
3/4
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 161 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
 162 (inlink->w <= inlink->h && s->passthrough == TRANSPOSE_PT_TYPE_PORTRAIT)) {
196 ✗ av_log(ctx, AV_LOG_VERBOSE,
197 "w:%d h:%d -> w:%d h:%d (passthrough mode)\n",
198 inlink->w, inlink->h, inlink->w, inlink->h);
199 ✗ return 0;
200 } else {
201 162 s->passthrough = TRANSPOSE_PT_TYPE_NONE;
202 }
203
204 162 s->hsub = desc_in->log2_chroma_w;
205 162 s->vsub = desc_in->log2_chroma_h;
206 162 s->planes = av_pix_fmt_count_planes(outlink->format);
207
208
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 162 times.
 162 av_assert0(desc_in->nb_components == desc_out->nb_components);
209
210
211 162 av_image_fill_max_pixsteps(s->pixsteps, NULL, desc_out);
212
213 162 outlink->w = inlink->h;
214 162 outlink->h = inlink->w;
215
216
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 161 times.
 162 if (inlink->sample_aspect_ratio.num)
217 1 outlink->sample_aspect_ratio = av_div_q((AVRational) { 1, 1 },
218 inlink->sample_aspect_ratio);
219 else
220 161 outlink->sample_aspect_ratio = inlink->sample_aspect_ratio;
221
222
2/2
✓ Branch 0 taken 648 times.
✓ Branch 1 taken 162 times.
 810 for (int i = 0; i < 4; i++) {
223 648 TransVtable *v = &s->vtables[i];
224
7/7
✓ Branch 0 taken 57 times.
✓ Branch 1 taken 241 times.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 46 times.
✓ Branch 4 taken 6 times.
✓ Branch 5 taken 10 times.
✓ Branch 6 taken 285 times.
 648 switch (s->pixsteps[i]) {
225 57 case 1: v->transpose_block = transpose_block_8_c;
226 57 v->transpose_8x8 = transpose_8x8_8_c; break;
227 241 case 2: v->transpose_block = transpose_block_16_c;
228 241 v->transpose_8x8 = transpose_8x8_16_c; break;
229 3 case 3: v->transpose_block = transpose_block_24_c;
230 3 v->transpose_8x8 = transpose_8x8_24_c; break;
231 46 case 4: v->transpose_block = transpose_block_32_c;
232 46 v->transpose_8x8 = transpose_8x8_32_c; break;
233 6 case 6: v->transpose_block = transpose_block_48_c;
234 6 v->transpose_8x8 = transpose_8x8_48_c; break;
235 10 case 8: v->transpose_block = transpose_block_64_c;
236 10 v->transpose_8x8 = transpose_8x8_64_c; break;
237 }
238 }
239
240 #if ARCH_X86
241
2/2
✓ Branch 0 taken 648 times.
✓ Branch 1 taken 162 times.
 810 for (int i = 0; i < 4; i++) {
242 648 TransVtable *v = &s->vtables[i];
243
244 648 ff_transpose_init_x86(v, s->pixsteps[i]);
245 }
246 #endif
247
248 324 av_log(ctx, AV_LOG_VERBOSE,
249 "w:%d h:%d dir:%d -> w:%d h:%d rotation:%s vflip:%d\n",
250 inlink->w, inlink->h, s->dir, outlink->w, outlink->h,
251
3/4
✓ Branch 0 taken 161 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 161 times.
 162 s->dir == 1 || s->dir == 3 ? "clockwise" : "counterclockwise",
252
3/4
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 161 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
 162 s->dir == 0 || s->dir == 3);
253 162 return 0;
254 }
255
256 312 static AVFrame *get_video_buffer(AVFilterLink *inlink, int w, int h)
257 {
258 312 TransContext *s = inlink->dst->priv;
259
260 312 return s->passthrough ?
261
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 312 times.
 624 ff_null_get_video_buffer (inlink, w, h) :
262 312 ff_default_get_video_buffer(inlink, w, h);
263 }
264
265 typedef struct ThreadData {
266 AVFrame *in, *out;
267 } ThreadData;
268
269 871 static int filter_slice(AVFilterContext *ctx, void *arg, int jobnr,
270 int nb_jobs)
271 {
272 871 TransContext *s = ctx->priv;
273 871 ThreadData *td = arg;
274 871 AVFrame *out = td->out;
275 871 AVFrame *in = td->in;
276 int plane;
277
278
2/2
✓ Branch 0 taken 2367 times.
✓ Branch 1 taken 871 times.
 3238 for (plane = 0; plane < s->planes; plane++) {
279
4/4
✓ Branch 0 taken 1630 times.
✓ Branch 1 taken 737 times.
✓ Branch 2 taken 705 times.
✓ Branch 3 taken 925 times.
 2367 int hsub = plane == 1 || plane == 2 ? s->hsub : 0;
280
4/4
✓ Branch 0 taken 1630 times.
✓ Branch 1 taken 737 times.
✓ Branch 2 taken 705 times.
✓ Branch 3 taken 925 times.
 2367 int vsub = plane == 1 || plane == 2 ? s->vsub : 0;
281 2367 int pixstep = s->pixsteps[plane];
282 2367 int inh = AV_CEIL_RSHIFT(in->height, vsub);
283 2367 int outw = AV_CEIL_RSHIFT(out->width, hsub);
284 2367 int outh = AV_CEIL_RSHIFT(out->height, vsub);
285 2367 int start = (outh * jobnr ) / nb_jobs;
286 2367 int end = (outh * (jobnr+1)) / nb_jobs;
287 uint8_t *dst, *src;
288 int dstlinesize, srclinesize;
289 int x, y;
290 2367 TransVtable *v = &s->vtables[plane];
291
292 2367 dstlinesize = out->linesize[plane];
293 2367 dst = out->data[plane] + start * dstlinesize;
294 2367 src = in->data[plane];
295 2367 srclinesize = in->linesize[plane];
296
297
2/2
✓ Branch 0 taken 315 times.
✓ Branch 1 taken 2052 times.
 2367 if (s->dir & 1) {
298 315 src += in->linesize[plane] * (inh - 1);
299 315 srclinesize *= -1;
300 }
301
302
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2367 times.
 2367 if (s->dir & 2) {
303 ✗ dst = out->data[plane] + dstlinesize * (outh - start - 1);
304 ✗ dstlinesize *= -1;
305 }
306
307
2/2
✓ Branch 0 taken 45143 times.
✓ Branch 1 taken 2367 times.
 47510 for (y = start; y < end - 7; y += 8) {
308
2/2
✓ Branch 0 taken 1431701 times.
✓ Branch 1 taken 45143 times.
 1476844 for (x = 0; x < outw - 7; x += 8) {
309 1431701 v->transpose_8x8(src + x * srclinesize + y * pixstep,
310 srclinesize,
311 1431701 dst + (y - start) * dstlinesize + x * pixstep,
312 dstlinesize);
313 }
314
3/4
✓ Branch 0 taken 209 times.
✓ Branch 1 taken 44934 times.
✓ Branch 2 taken 209 times.
✗ Branch 3 not taken.
 45143 if (outw - x > 0 && end - y > 0)
315 209 v->transpose_block(src + x * srclinesize + y * pixstep,
316 srclinesize,
317 209 dst + (y - start) * dstlinesize + x * pixstep,
318 dstlinesize, outw - x, end - y);
319 }
320
321
2/2
✓ Branch 0 taken 1302 times.
✓ Branch 1 taken 1065 times.
 2367 if (end - y > 0)
322 1302 v->transpose_block(src + 0 * srclinesize + y * pixstep,
323 srclinesize,
324 1302 dst + (y - start) * dstlinesize + 0 * pixstep,
325 dstlinesize, outw, end - y);
326 }
327
328 871 return 0;
329 }
330
331 471 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
332 {
333 471 int err = 0;
334 471 AVFilterContext *ctx = inlink->dst;
335 471 TransContext *s = ctx->priv;
336 471 AVFilterLink *outlink = ctx->outputs[0];
337 ThreadData td;
338 AVFrame *out;
339
340
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
 471 if (s->passthrough)
341 ✗ return ff_filter_frame(outlink, in);
342
343 471 out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
344
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
 471 if (!out) {
345 ✗ err = AVERROR(ENOMEM);
346 ✗ goto fail;
347 }
348
349 471 err = av_frame_copy_props(out, in);
350
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
 471 if (err < 0)
351 ✗ goto fail;
352
353
2/2
✓ Branch 0 taken 470 times.
✓ Branch 1 taken 1 times.
 471 if (in->sample_aspect_ratio.num == 0) {
354 470 out->sample_aspect_ratio = in->sample_aspect_ratio;
355 } else {
356 1 out->sample_aspect_ratio.num = in->sample_aspect_ratio.den;
357 1 out->sample_aspect_ratio.den = in->sample_aspect_ratio.num;
358 }
359
360 471 td.in = in, td.out = out;
361 471 ff_filter_execute(ctx, filter_slice, &td, NULL,
362
1/2
✓ Branch 0 taken 471 times.
✗ Branch 1 not taken.
 471 FFMIN(outlink->h, ff_filter_get_nb_threads(ctx)));
363 471 av_frame_free(&in);
364 471 return ff_filter_frame(outlink, out);
365
366 ✗ fail:
367 ✗ av_frame_free(&in);
368 ✗ av_frame_free(&out);
369 ✗ return err;
370 }
371
372 #define OFFSET(x) offsetof(TransContext, x)
373 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
374
375 static const AVOption transpose_options[] = {
376 { "dir", "set transpose direction", OFFSET(dir), AV_OPT_TYPE_INT, { .i64 = TRANSPOSE_CCLOCK_FLIP }, 0, 7, FLAGS, .unit = "dir" },
377 { "cclock_flip", "rotate counter-clockwise with vertical flip", 0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CCLOCK_FLIP }, .flags=FLAGS, .unit = "dir" },
378 { "clock", "rotate clockwise", 0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CLOCK }, .flags=FLAGS, .unit = "dir" },
379 { "cclock", "rotate counter-clockwise", 0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CCLOCK }, .flags=FLAGS, .unit = "dir" },
380 { "clock_flip", "rotate clockwise with vertical flip", 0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CLOCK_FLIP }, .flags=FLAGS, .unit = "dir" },
381
382 { "passthrough", "do not apply transposition if the input matches the specified geometry",
383 OFFSET(passthrough), AV_OPT_TYPE_INT, {.i64=TRANSPOSE_PT_TYPE_NONE}, 0, INT_MAX, FLAGS, .unit = "passthrough" },
384 { "none", "always apply transposition", 0, AV_OPT_TYPE_CONST, {.i64=TRANSPOSE_PT_TYPE_NONE}, INT_MIN, INT_MAX, FLAGS, .unit = "passthrough" },
385 { "portrait", "preserve portrait geometry", 0, AV_OPT_TYPE_CONST, {.i64=TRANSPOSE_PT_TYPE_PORTRAIT}, INT_MIN, INT_MAX, FLAGS, .unit = "passthrough" },
386 { "landscape", "preserve landscape geometry", 0, AV_OPT_TYPE_CONST, {.i64=TRANSPOSE_PT_TYPE_LANDSCAPE}, INT_MIN, INT_MAX, FLAGS, .unit = "passthrough" },
387
388 { NULL }
389 };
390
391 AVFILTER_DEFINE_CLASS(transpose);
392
393 static const AVFilterPad avfilter_vf_transpose_inputs[] = {
394 {
395 .name = "default",
396 .type = AVMEDIA_TYPE_VIDEO,
397 .get_buffer.video = get_video_buffer,
398 .filter_frame = filter_frame,
399 },
400 };
401
402 static const AVFilterPad avfilter_vf_transpose_outputs[] = {
403 {
404 .name = "default",
405 .config_props = config_props_output,
406 .type = AVMEDIA_TYPE_VIDEO,
407 },
408 };
409
410 const FFFilter ff_vf_transpose = {
411 .p.name = "transpose",
412 .p.description = NULL_IF_CONFIG_SMALL("Transpose input video."),
413 .p.priv_class = &transpose_class,
414 .p.flags = AVFILTER_FLAG_SLICE_THREADS,
415 .priv_size = sizeof(TransContext),
416 FILTER_INPUTS(avfilter_vf_transpose_inputs),
417 FILTER_OUTPUTS(avfilter_vf_transpose_outputs),
418 FILTER_QUERY_FUNC2(query_formats),
419 };
420