FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
vf_stack.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2015 Paul B. Mahol
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 #include "libavutil/avstring.h"
22 #include "libavutil/imgutils.h"
23 #include "libavutil/opt.h"
24 #include "libavutil/pixdesc.h"
25 
26 #include "avfilter.h"
27 #include "formats.h"
28 #include "internal.h"
29 #include "framesync.h"
30 #include "video.h"
31 
32 typedef struct StackContext {
33  const AVClass *class;
35  int nb_inputs;
36  int shortest;
38  int nb_planes;
39 
42 } StackContext;
43 
45 {
47  int fmt, ret;
48 
49  for (fmt = 0; av_pix_fmt_desc_get(fmt); fmt++) {
51  if (!(desc->flags & AV_PIX_FMT_FLAG_PAL ||
54  (ret = ff_add_format(&pix_fmts, fmt)) < 0)
55  return ret;
56  }
57 
58  return ff_set_common_formats(ctx, pix_fmts);
59 }
60 
62 {
63  StackContext *s = ctx->priv;
64  int i, ret;
65 
66  if (!strcmp(ctx->filter->name, "vstack"))
67  s->is_vertical = 1;
68 
69  s->frames = av_calloc(s->nb_inputs, sizeof(*s->frames));
70  if (!s->frames)
71  return AVERROR(ENOMEM);
72 
73  for (i = 0; i < s->nb_inputs; i++) {
74  AVFilterPad pad = { 0 };
75 
77  pad.name = av_asprintf("input%d", i);
78  if (!pad.name)
79  return AVERROR(ENOMEM);
80 
81  if ((ret = ff_insert_inpad(ctx, i, &pad)) < 0) {
82  av_freep(&pad.name);
83  return ret;
84  }
85  }
86 
87  return 0;
88 }
89 
90 static int process_frame(FFFrameSync *fs)
91 {
92  AVFilterContext *ctx = fs->parent;
93  AVFilterLink *outlink = ctx->outputs[0];
94  StackContext *s = fs->opaque;
95  AVFrame **in = s->frames;
96  AVFrame *out;
97  int i, p, ret, offset[4] = { 0 };
98 
99  for (i = 0; i < s->nb_inputs; i++) {
100  if ((ret = ff_framesync_get_frame(&s->fs, i, &in[i], 0)) < 0)
101  return ret;
102  }
103 
104  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
105  if (!out)
106  return AVERROR(ENOMEM);
107  out->pts = av_rescale_q(s->fs.pts, s->fs.time_base, outlink->time_base);
108 
109  for (i = 0; i < s->nb_inputs; i++) {
110  AVFilterLink *inlink = ctx->inputs[i];
111  int linesize[4];
112  int height[4];
113 
114  if ((ret = av_image_fill_linesizes(linesize, inlink->format, inlink->w)) < 0) {
115  av_frame_free(&out);
116  return ret;
117  }
118 
119  height[1] = height[2] = AV_CEIL_RSHIFT(inlink->h, s->desc->log2_chroma_h);
120  height[0] = height[3] = inlink->h;
121 
122  for (p = 0; p < s->nb_planes; p++) {
123  if (s->is_vertical) {
124  av_image_copy_plane(out->data[p] + offset[p] * out->linesize[p],
125  out->linesize[p],
126  in[i]->data[p],
127  in[i]->linesize[p],
128  linesize[p], height[p]);
129  offset[p] += height[p];
130  } else {
131  av_image_copy_plane(out->data[p] + offset[p],
132  out->linesize[p],
133  in[i]->data[p],
134  in[i]->linesize[p],
135  linesize[p], height[p]);
136  offset[p] += linesize[p];
137  }
138  }
139  }
140 
141  return ff_filter_frame(outlink, out);
142 }
143 
144 static int config_output(AVFilterLink *outlink)
145 {
146  AVFilterContext *ctx = outlink->src;
147  StackContext *s = ctx->priv;
148  AVRational time_base = ctx->inputs[0]->time_base;
149  AVRational frame_rate = ctx->inputs[0]->frame_rate;
150  int height = ctx->inputs[0]->h;
151  int width = ctx->inputs[0]->w;
152  FFFrameSyncIn *in;
153  int i, ret;
154 
155  if (s->is_vertical) {
156  for (i = 1; i < s->nb_inputs; i++) {
157  if (ctx->inputs[i]->w != width) {
158  av_log(ctx, AV_LOG_ERROR, "Input %d width %d does not match input %d width %d.\n", i, ctx->inputs[i]->w, 0, width);
159  return AVERROR(EINVAL);
160  }
161  height += ctx->inputs[i]->h;
162  }
163  } else {
164  for (i = 1; i < s->nb_inputs; i++) {
165  if (ctx->inputs[i]->h != height) {
166  av_log(ctx, AV_LOG_ERROR, "Input %d height %d does not match input %d height %d.\n", i, ctx->inputs[i]->h, 0, height);
167  return AVERROR(EINVAL);
168  }
169  width += ctx->inputs[i]->w;
170  }
171  }
172 
173  s->desc = av_pix_fmt_desc_get(outlink->format);
174  if (!s->desc)
175  return AVERROR_BUG;
177 
178  outlink->w = width;
179  outlink->h = height;
180  outlink->time_base = time_base;
181  outlink->frame_rate = frame_rate;
182 
183  if ((ret = ff_framesync_init(&s->fs, ctx, s->nb_inputs)) < 0)
184  return ret;
185 
186  in = s->fs.in;
187  s->fs.opaque = s;
189 
190  for (i = 0; i < s->nb_inputs; i++) {
191  AVFilterLink *inlink = ctx->inputs[i];
192 
193  in[i].time_base = inlink->time_base;
194  in[i].sync = 1;
195  in[i].before = EXT_STOP;
196  in[i].after = s->shortest ? EXT_STOP : EXT_INFINITY;
197  }
198 
199  return ff_framesync_configure(&s->fs);
200 }
201 
203 {
204  StackContext *s = ctx->priv;
205  int i;
206 
207  ff_framesync_uninit(&s->fs);
208  av_freep(&s->frames);
209 
210  for (i = 0; i < ctx->nb_inputs; i++)
211  av_freep(&ctx->input_pads[i].name);
212 }
213 
215 {
216  StackContext *s = ctx->priv;
217  return ff_framesync_activate(&s->fs);
218 }
219 
220 #define OFFSET(x) offsetof(StackContext, x)
221 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
222 static const AVOption stack_options[] = {
223  { "inputs", "set number of inputs", OFFSET(nb_inputs), AV_OPT_TYPE_INT, {.i64=2}, 2, INT_MAX, .flags = FLAGS },
224  { "shortest", "force termination when the shortest input terminates", OFFSET(shortest), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, .flags = FLAGS },
225  { NULL },
226 };
227 
228 static const AVFilterPad outputs[] = {
229  {
230  .name = "default",
231  .type = AVMEDIA_TYPE_VIDEO,
232  .config_props = config_output,
233  },
234  { NULL }
235 };
236 
237 #if CONFIG_HSTACK_FILTER
238 
239 #define hstack_options stack_options
240 AVFILTER_DEFINE_CLASS(hstack);
241 
242 AVFilter ff_vf_hstack = {
243  .name = "hstack",
244  .description = NULL_IF_CONFIG_SMALL("Stack video inputs horizontally."),
245  .priv_size = sizeof(StackContext),
246  .priv_class = &hstack_class,
248  .outputs = outputs,
249  .init = init,
250  .uninit = uninit,
251  .activate = activate,
253 };
254 
255 #endif /* CONFIG_HSTACK_FILTER */
256 
257 #if CONFIG_VSTACK_FILTER
258 
259 #define vstack_options stack_options
260 AVFILTER_DEFINE_CLASS(vstack);
261 
262 AVFilter ff_vf_vstack = {
263  .name = "vstack",
264  .description = NULL_IF_CONFIG_SMALL("Stack video inputs vertically."),
265  .priv_size = sizeof(StackContext),
266  .priv_class = &vstack_class,
268  .outputs = outputs,
269  .init = init,
270  .uninit = uninit,
271  .activate = activate,
273 };
274 
275 #endif /* CONFIG_VSTACK_FILTER */
#define AV_PIX_FMT_FLAG_PAL
Pixel format has a palette in data[1], values are indexes in this palette.
Definition: pixdesc.h:132
#define NULL
Definition: coverity.c:32
#define FLAGS
Definition: vf_stack.c:221
const char * s
Definition: avisynth_c.h:768
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2419
This structure describes decoded (raw) audio or video data.
Definition: frame.h:201
AVOption.
Definition: opt.h:246
const char * fmt
Definition: avisynth_c.h:769
int shortest
Definition: vf_stack.c:36
misc image utilities
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2459
Main libavfilter public API header.
const char * desc
Definition: nvenc.c:60
int(* on_event)(struct FFFrameSync *fs)
Callback called when a frame event is ready.
Definition: framesync.h:172
#define AVFILTER_FLAG_DYNAMIC_INPUTS
The number of the filter inputs is not determined just by AVFilter.inputs.
Definition: avfilter.h:105
enum AVMediaType type
AVFilterPad type.
Definition: internal.h:65
int ff_framesync_configure(FFFrameSync *fs)
Configure a frame sync structure.
Definition: framesync.c:117
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:92
int64_t pts
Timestamp of the current event.
Definition: framesync.h:167
enum FFFrameSyncExtMode before
Extrapolation mode for timestamps before the first frame.
Definition: framesync.h:86
void * av_calloc(size_t nmemb, size_t size)
Non-inlined equivalent of av_mallocz_array().
Definition: mem.c:230
const char * name
Pad name.
Definition: internal.h:60
AVFilterContext * parent
Parent filter context.
Definition: framesync.h:152
AVFilterLink ** inputs
array of pointers to input links
Definition: avfilter.h:346
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1151
#define av_cold
Definition: attributes.h:82
static int query_formats(AVFilterContext *ctx)
Definition: vf_stack.c:44
AVOptions.
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
Definition: frame.h:294
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_stack.c:202
FFFrameSyncIn * in
Pointer to array of inputs.
Definition: framesync.h:203
#define height
static int flags
Definition: log.c:57
enum FFFrameSyncExtMode after
Extrapolation mode for timestamps after the last frame.
Definition: framesync.h:91
Input stream structure.
Definition: framesync.h:81
#define av_log(a,...)
A filter pad used for either input or output.
Definition: internal.h:54
int64_t av_rescale_q(int64_t a, AVRational bq, AVRational cq)
Rescale a 64-bit integer by 2 rational numbers.
Definition: mathematics.c:142
AVFilterPad * input_pads
array of input pads
Definition: avfilter.h:345
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int ff_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
A helper for query_formats() which sets all links to the same list of formats.
Definition: formats.c:568
static int config_output(AVFilterLink *outlink)
Definition: vf_stack.c:144
uint8_t log2_chroma_h
Amount to shift the luma height right to find the chroma height.
Definition: pixdesc.h:101
void ff_framesync_uninit(FFFrameSync *fs)
Free all memory currently allocated.
Definition: framesync.c:293
static const AVOption stack_options[]
Definition: vf_stack.c:222
Frame sync structure.
Definition: framesync.h:146
#define AVERROR(e)
Definition: error.h:43
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:163
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:179
void * priv
private data for use by the filter
Definition: avfilter.h:353
uint16_t width
Definition: gdv.c:47
#define AV_PIX_FMT_FLAG_HWACCEL
Pixel format is an HW accelerated format.
Definition: pixdesc.h:140
AVRational time_base
Time base for the incoming frames.
Definition: framesync.h:96
int ff_add_format(AVFilterFormats **avff, int64_t fmt)
Add fmt to the list of media formats contained in *avff.
Definition: formats.c:337
int ff_framesync_activate(FFFrameSync *fs)
Examine the frames in the filter's input and try to produce output.
Definition: framesync.c:344
static const uint8_t offset[127][2]
Definition: vf_spp.c:92
char * av_asprintf(const char *fmt,...)
Definition: avstring.c:113
uint64_t flags
Combination of AV_PIX_FMT_FLAG_...
Definition: pixdesc.h:106
unsigned nb_inputs
number of input pads
Definition: avfilter.h:347
AVFrame ** frames
Definition: vf_stack.c:40
const AVPixFmtDescriptor * desc
Definition: vf_stack.c:34
#define OFFSET(x)
Definition: vf_stack.c:220
AVFormatContext * ctx
Definition: movenc.c:48
AVRational time_base
Time base for the output events.
Definition: framesync.h:162
void * opaque
Opaque pointer, not used by the API.
Definition: framesync.h:177
int nb_planes
Definition: vf_stack.c:38
FFFrameSync fs
Definition: vf_stack.c:41
Extend the frame to infinity.
Definition: framesync.h:75
static av_cold int init(AVFilterContext *ctx)
Definition: vf_stack.c:61
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:232
int ff_framesync_init(FFFrameSync *fs, AVFilterContext *parent, unsigned nb_in)
Initialize a frame sync structure.
Definition: framesync.c:77
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
int nb_inputs
Definition: vf_stack.c:35
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_log(ac->avr, AV_LOG_TRACE,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> in
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
Definition: error.h:50
unsigned sync
Synchronization level: frames on input at the highest sync level will generate output frame events...
Definition: framesync.h:139
Describe the class of an AVClass context structure.
Definition: log.h:67
Filter definition.
Definition: avfilter.h:144
int av_image_fill_linesizes(int linesizes[4], enum AVPixelFormat pix_fmt, int width)
Fill plane linesizes for an image with pixel format pix_fmt and width width.
Definition: imgutils.c:89
Rational number (pair of numerator and denominator).
Definition: rational.h:58
const char * name
Filter name.
Definition: avfilter.h:148
#define AV_PIX_FMT_FLAG_BITSTREAM
All values of a component are bit-wise packed end to end.
Definition: pixdesc.h:136
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:350
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:266
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:215
static int process_frame(FFFrameSync *fs)
Definition: vf_stack.c:90
static const AVFilterPad outputs[]
Definition: vf_stack.c:228
Completely stop all streams with this one.
Definition: framesync.h:65
#define AVFILTER_DEFINE_CLASS(fname)
Definition: internal.h:334
int is_vertical
Definition: vf_stack.c:37
A list of supported formats for one end of a filter link.
Definition: formats.h:64
An instance of a filter.
Definition: avfilter.h:338
FILE * out
Definition: movenc.c:54
#define av_freep(p)
static int activate(AVFilterContext *ctx)
Definition: vf_stack.c:214
void av_image_copy_plane(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize, int bytewidth, int height)
Copy image plane from src to dst.
Definition: imgutils.c:337
internal API functions
int ff_framesync_get_frame(FFFrameSync *fs, unsigned in, AVFrame **rframe, unsigned get)
Get the current frame in an input.
Definition: framesync.c:256
const AVFilter * filter
the AVFilter of which this is an instance
Definition: avfilter.h:341
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58
static int ff_insert_inpad(AVFilterContext *f, unsigned index, AVFilterPad *p)
Insert a new input pad for the filter.
Definition: internal.h:277