FFmpeg
vsrc_cellauto.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) Stefano Sabatini 2011
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  * cellular automaton video source, based on Stephen Wolfram "experimentus crucis"
24  */
25 
26 /* #define DEBUG */
27 
28 #include "libavutil/file.h"
29 #include "libavutil/internal.h"
30 #include "libavutil/lfg.h"
31 #include "libavutil/opt.h"
32 #include "libavutil/random_seed.h"
33 #include "libavutil/avstring.h"
34 #include "avfilter.h"
35 #include "internal.h"
36 #include "video.h"
37 
38 typedef struct CellAutoContext {
39  const AVClass *class;
40  int w, h;
41  char *filename;
42  char *rule_str;
43  uint8_t *file_buf;
44  size_t file_bufsize;
45  uint8_t *buf;
47  uint8_t rule;
48  uint64_t pts;
51  int64_t random_seed;
53  int64_t generation; ///< the generation number, starting from 0
55  char *pattern;
57 
58 #define OFFSET(x) offsetof(CellAutoContext, x)
59 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
60 
61 static const AVOption cellauto_options[] = {
62  { "filename", "read initial pattern from file", OFFSET(filename), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, FLAGS },
63  { "f", "read initial pattern from file", OFFSET(filename), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, FLAGS },
64  { "pattern", "set initial pattern", OFFSET(pattern), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, FLAGS },
65  { "p", "set initial pattern", OFFSET(pattern), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, FLAGS },
66  { "rate", "set video rate", OFFSET(frame_rate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, INT_MAX, FLAGS },
67  { "r", "set video rate", OFFSET(frame_rate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, INT_MAX, FLAGS },
68  { "size", "set video size", OFFSET(w), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, FLAGS },
69  { "s", "set video size", OFFSET(w), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, FLAGS },
70  { "rule", "set rule", OFFSET(rule), AV_OPT_TYPE_INT, {.i64 = 110}, 0, 255, FLAGS },
71  { "random_fill_ratio", "set fill ratio for filling initial grid randomly", OFFSET(random_fill_ratio), AV_OPT_TYPE_DOUBLE, {.dbl = 1/M_PHI}, 0, 1, FLAGS },
72  { "ratio", "set fill ratio for filling initial grid randomly", OFFSET(random_fill_ratio), AV_OPT_TYPE_DOUBLE, {.dbl = 1/M_PHI}, 0, 1, FLAGS },
73  { "random_seed", "set the seed for filling the initial grid randomly", OFFSET(random_seed), AV_OPT_TYPE_INT64, {.i64 = -1}, -1, UINT32_MAX, FLAGS },
74  { "seed", "set the seed for filling the initial grid randomly", OFFSET(random_seed), AV_OPT_TYPE_INT64, {.i64 = -1}, -1, UINT32_MAX, FLAGS },
75  { "scroll", "scroll pattern downward", OFFSET(scroll), AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, FLAGS },
76  { "start_full", "start filling the whole video", OFFSET(start_full), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, FLAGS },
77  { "full", "start filling the whole video", OFFSET(start_full), AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, FLAGS },
78  { "stitch", "stitch boundaries", OFFSET(stitch), AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, FLAGS },
79  { NULL }
80 };
81 
82 AVFILTER_DEFINE_CLASS(cellauto);
83 
84 #ifdef DEBUG
85 static void show_cellauto_row(AVFilterContext *ctx)
86 {
87  CellAutoContext *s = ctx->priv;
88  int i;
89  uint8_t *row = s->buf + s->w * s->buf_row_idx;
90  char *line = av_malloc(s->w + 1);
91  if (!line)
92  return;
93 
94  for (i = 0; i < s->w; i++)
95  line[i] = row[i] ? '@' : ' ';
96  line[i] = 0;
97  av_log(ctx, AV_LOG_DEBUG, "generation:%"PRId64" row:%s|\n", s->generation, line);
98  av_free(line);
99 }
100 #endif
101 
103 {
104  CellAutoContext *s = ctx->priv;
105  char *p;
106  int i, w = 0;
107 
108  w = strlen(s->pattern);
109  av_log(ctx, AV_LOG_DEBUG, "w:%d\n", w);
110 
111  if (s->w) {
112  if (w > s->w) {
114  "The specified width is %d which cannot contain the provided string width of %d\n",
115  s->w, w);
116  return AVERROR(EINVAL);
117  }
118  } else {
119  /* width was not specified, set it to width of the provided row */
120  s->w = w;
121  s->h = (double)s->w * M_PHI;
122  }
123 
124  s->buf = av_calloc(s->w, s->h * sizeof(*s->buf));
125  if (!s->buf)
126  return AVERROR(ENOMEM);
127 
128  /* fill buf */
129  p = s->pattern;
130  for (i = (s->w - w)/2;; i++) {
131  av_log(ctx, AV_LOG_DEBUG, "%d %c\n", i, *p == '\n' ? 'N' : *p);
132  if (*p == '\n' || !*p)
133  break;
134  else
135  s->buf[i] = !!av_isgraph(*(p++));
136  }
137 
138  return 0;
139 }
140 
142 {
143  CellAutoContext *s = ctx->priv;
144  int ret;
145 
146  ret = av_file_map(s->filename,
147  &s->file_buf, &s->file_bufsize, 0, ctx);
148  if (ret < 0)
149  return ret;
150 
151  /* create a string based on the read file */
152  s->pattern = av_malloc(s->file_bufsize + 1);
153  if (!s->pattern)
154  return AVERROR(ENOMEM);
155  memcpy(s->pattern, s->file_buf, s->file_bufsize);
156  s->pattern[s->file_bufsize] = 0;
157 
159 }
160 
162 {
163  CellAutoContext *s = ctx->priv;
164  int ret;
165 
166  if (!s->w && !s->filename && !s->pattern)
167  av_opt_set(s, "size", "320x518", 0);
168 
169  if (s->filename && s->pattern) {
170  av_log(ctx, AV_LOG_ERROR, "Only one of the filename or pattern options can be used\n");
171  return AVERROR(EINVAL);
172  }
173 
174  if (s->filename) {
175  if ((ret = init_pattern_from_file(ctx)) < 0)
176  return ret;
177  } else if (s->pattern) {
178  if ((ret = init_pattern_from_string(ctx)) < 0)
179  return ret;
180  } else {
181  /* fill the first row randomly */
182  int i;
183 
184  s->buf = av_calloc(s->w, s->h * sizeof(*s->buf));
185  if (!s->buf)
186  return AVERROR(ENOMEM);
187  if (s->random_seed == -1)
188  s->random_seed = av_get_random_seed();
189 
190  av_lfg_init(&s->lfg, s->random_seed);
191 
192  for (i = 0; i < s->w; i++) {
193  double r = (double)av_lfg_get(&s->lfg) / UINT32_MAX;
194  if (r <= s->random_fill_ratio)
195  s->buf[i] = 1;
196  }
197  }
198 
200  "s:%dx%d r:%d/%d rule:%d stitch:%d scroll:%d full:%d seed:%"PRId64"\n",
201  s->w, s->h, s->frame_rate.num, s->frame_rate.den,
202  s->rule, s->stitch, s->scroll, s->start_full,
203  s->random_seed);
204  return 0;
205 }
206 
208 {
209  CellAutoContext *s = ctx->priv;
210 
211  av_file_unmap(s->file_buf, s->file_bufsize);
212  av_freep(&s->buf);
213  av_freep(&s->pattern);
214 }
215 
216 static int config_props(AVFilterLink *outlink)
217 {
218  CellAutoContext *s = outlink->src->priv;
219 
220  outlink->w = s->w;
221  outlink->h = s->h;
222  outlink->time_base = av_inv_q(s->frame_rate);
223  outlink->frame_rate = s->frame_rate;
224 
225  return 0;
226 }
227 
229 {
230  CellAutoContext *s = ctx->priv;
231  int i, v, pos[3];
232  uint8_t *row, *prev_row = s->buf + s->buf_row_idx * s->w;
233  enum { NW, N, NE };
234 
235  s->buf_prev_row_idx = s->buf_row_idx;
236  s->buf_row_idx = s->buf_row_idx == s->h-1 ? 0 : s->buf_row_idx+1;
237  row = s->buf + s->w * s->buf_row_idx;
238 
239  for (i = 0; i < s->w; i++) {
240  if (s->stitch) {
241  pos[NW] = i-1 < 0 ? s->w-1 : i-1;
242  pos[N] = i;
243  pos[NE] = i+1 == s->w ? 0 : i+1;
244  v = prev_row[pos[NW]]<<2 | prev_row[pos[N]]<<1 | prev_row[pos[NE]];
245  } else {
246  v = 0;
247  v|= i-1 >= 0 ? prev_row[i-1]<<2 : 0;
248  v|= prev_row[i ]<<1 ;
249  v|= i+1 < s->w ? prev_row[i+1] : 0;
250  }
251  row[i] = !!(s->rule & (1<<v));
252  ff_dlog(ctx, "i:%d context:%c%c%c -> cell:%d\n", i,
253  v&4?'@':' ', v&2?'@':' ', v&1?'@':' ', row[i]);
254  }
255 
256  s->generation++;
257 }
258 
259 static void fill_picture(AVFilterContext *ctx, AVFrame *picref)
260 {
261  CellAutoContext *s = ctx->priv;
262  int i, j, k, row_idx = 0;
263  uint8_t *p0 = picref->data[0];
264 
265  if (s->scroll && s->generation >= s->h)
266  /* show on top the oldest row */
267  row_idx = (s->buf_row_idx + 1) % s->h;
268 
269  /* fill the output picture with the whole buffer */
270  for (i = 0; i < s->h; i++) {
271  uint8_t byte = 0;
272  uint8_t *row = s->buf + row_idx*s->w;
273  uint8_t *p = p0;
274  for (k = 0, j = 0; j < s->w; j++) {
275  byte |= row[j]<<(7-k++);
276  if (k==8 || j == s->w-1) {
277  k = 0;
278  *p++ = byte;
279  byte = 0;
280  }
281  }
282  row_idx = (row_idx + 1) % s->h;
283  p0 += picref->linesize[0];
284  }
285 }
286 
287 static int request_frame(AVFilterLink *outlink)
288 {
289  CellAutoContext *s = outlink->src->priv;
290  AVFrame *picref = ff_get_video_buffer(outlink, s->w, s->h);
291  if (!picref)
292  return AVERROR(ENOMEM);
293  picref->sample_aspect_ratio = (AVRational) {1, 1};
294  if (s->generation == 0 && s->start_full) {
295  int i;
296  for (i = 0; i < s->h-1; i++)
297  evolve(outlink->src);
298  }
299  fill_picture(outlink->src, picref);
300  evolve(outlink->src);
301 
302  picref->pts = s->pts++;
303  picref->duration = 1;
304 
305 #ifdef DEBUG
306  show_cellauto_row(outlink->src);
307 #endif
308  return ff_filter_frame(outlink, picref);
309 }
310 
311 static const AVFilterPad cellauto_outputs[] = {
312  {
313  .name = "default",
314  .type = AVMEDIA_TYPE_VIDEO,
315  .request_frame = request_frame,
316  .config_props = config_props,
317  },
318 };
319 
321  .name = "cellauto",
322  .description = NULL_IF_CONFIG_SMALL("Create pattern generated by an elementary cellular automaton."),
323  .priv_size = sizeof(CellAutoContext),
324  .priv_class = &cellauto_class,
325  .init = init,
326  .uninit = uninit,
327  .inputs = NULL,
330 };
ff_get_video_buffer
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:112
CellAutoContext::generation
int64_t generation
the generation number, starting from 0
Definition: vsrc_cellauto.c:53
OFFSET
#define OFFSET(x)
Definition: vsrc_cellauto.c:58
r
const char * r
Definition: vf_curves.c:126
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
M_PHI
#define M_PHI
Definition: mathematics.h:61
av_lfg_init
av_cold void av_lfg_init(AVLFG *c, unsigned int seed)
Definition: lfg.c:32
FLAGS
#define FLAGS
Definition: vsrc_cellauto.c:59
CellAutoContext::frame_rate
AVRational frame_rate
Definition: vsrc_cellauto.c:49
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1018
AVFrame::duration
int64_t duration
Duration of the frame, in the same units as pts.
Definition: frame.h:781
AV_OPT_TYPE_VIDEO_RATE
@ AV_OPT_TYPE_VIDEO_RATE
offset must point to AVRational
Definition: opt.h:248
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:375
AVFrame::pts
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
Definition: frame.h:487
w
uint8_t w
Definition: llviddspenc.c:38
CellAutoContext::rule
uint8_t rule
Definition: vsrc_cellauto.c:47
AVOption
AVOption.
Definition: opt.h:346
AV_LOG_VERBOSE
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:196
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:170
video.h
init_pattern_from_file
static int init_pattern_from_file(AVFilterContext *ctx)
Definition: vsrc_cellauto.c:141
CellAutoContext::random_fill_ratio
double random_fill_ratio
Definition: vsrc_cellauto.c:50
init
static av_cold int init(AVFilterContext *ctx)
Definition: vsrc_cellauto.c:161
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:396
av_malloc
#define av_malloc(s)
Definition: tableprint_vlc.h:30
av_get_random_seed
uint32_t av_get_random_seed(void)
Get a seed to use in conjunction with random functions.
Definition: random_seed.c:167
CellAutoContext
Definition: vsrc_cellauto.c:38
av_file_map
int av_file_map(const char *filename, uint8_t **bufptr, size_t *size, int log_offset, void *log_ctx)
Read the file with name filename, and put its content in a newly allocated buffer or map it with mmap...
Definition: file.c:55
AVFilterContext::priv
void * priv
private data for use by the filter
Definition: avfilter.h:422
av_opt_set
int av_opt_set(void *obj, const char *name, const char *val, int search_flags)
Definition: opt.c:738
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:33
CellAutoContext::pts
uint64_t pts
Definition: vsrc_cellauto.c:48
CellAutoContext::random_seed
int64_t random_seed
Definition: vsrc_cellauto.c:51
config_props
static int config_props(AVFilterLink *outlink)
Definition: vsrc_cellauto.c:216
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
CellAutoContext::lfg
AVLFG lfg
Definition: vsrc_cellauto.c:54
av_cold
#define av_cold
Definition: attributes.h:90
request_frame
static int request_frame(AVFilterLink *outlink)
Definition: vsrc_cellauto.c:287
s
#define s(width, name)
Definition: cbs_vp9.c:198
av_lfg_get
static unsigned int av_lfg_get(AVLFG *c)
Get the next random unsigned 32-bit number using an ALFG.
Definition: lfg.h:53
AV_OPT_TYPE_DOUBLE
@ AV_OPT_TYPE_DOUBLE
Definition: opt.h:237
lfg.h
AV_OPT_TYPE_INT64
@ AV_OPT_TYPE_INT64
Definition: opt.h:236
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:201
ctx
AVFormatContext * ctx
Definition: movenc.c:48
CellAutoContext::file_bufsize
size_t file_bufsize
Definition: vsrc_cellauto.c:44
av_file_unmap
void av_file_unmap(uint8_t *bufptr, size_t size)
Unmap or free the buffer bufptr created by av_file_map().
Definition: file.c:146
CellAutoContext::pattern
char * pattern
Definition: vsrc_cellauto.c:55
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
NULL
#define NULL
Definition: coverity.c:32
AVRational
Rational number (pair of numerator and denominator).
Definition: rational.h:58
AV_PIX_FMT_MONOBLACK
@ AV_PIX_FMT_MONOBLACK
Y , 1bpp, 0 is black, 1 is white, in each byte pixels are ordered from the msb to the lsb.
Definition: pixfmt.h:83
AV_OPT_TYPE_IMAGE_SIZE
@ AV_OPT_TYPE_IMAGE_SIZE
offset must point to two consecutive integers
Definition: opt.h:245
evolve
static void evolve(AVFilterContext *ctx)
Definition: vsrc_cellauto.c:228
double
double
Definition: af_crystalizer.c:131
CellAutoContext::scroll
int scroll
Definition: vsrc_cellauto.c:52
inputs
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several inputs
Definition: filter_design.txt:243
cellauto_outputs
static const AVFilterPad cellauto_outputs[]
Definition: vsrc_cellauto.c:311
CellAutoContext::rule_str
char * rule_str
Definition: vsrc_cellauto.c:42
ff_dlog
#define ff_dlog(a,...)
Definition: tableprint_vlc.h:28
AVLFG
Context structure for the Lagged Fibonacci PRNG.
Definition: lfg.h:33
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(cellauto)
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:106
scroll
static void scroll(AVFilterContext *ctx, AVFrame *in, AVFrame *out)
Definition: vf_scroll.c:103
byte
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_WB24 unsigned int_TMPL AV_WB16 unsigned int_TMPL byte
Definition: bytestream.h:99
for
for(k=2;k<=8;++k)
Definition: h264pred_template.c:425
CellAutoContext::file_buf
uint8_t * file_buf
Definition: vsrc_cellauto.c:43
av_isgraph
static av_const int av_isgraph(int c)
Locale-independent conversion of ASCII isgraph.
Definition: avstring.h:210
fill_picture
static void fill_picture(AVFilterContext *ctx, AVFrame *picref)
Definition: vsrc_cellauto.c:259
line
Definition: graph2dot.c:48
N
#define N
Definition: af_mcompand.c:53
ff_vsrc_cellauto
const AVFilter ff_vsrc_cellauto
Definition: vsrc_cellauto.c:320
internal.h
FILTER_SINGLE_PIXFMT
#define FILTER_SINGLE_PIXFMT(pix_fmt_)
Definition: internal.h:172
CellAutoContext::buf_prev_row_idx
int buf_prev_row_idx
Definition: vsrc_cellauto.c:46
CellAutoContext::stitch
int stitch
Definition: vsrc_cellauto.c:52
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:255
internal.h
av_inv_q
static av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
Definition: rational.h:159
CellAutoContext::h
int h
Definition: vsrc_cellauto.c:40
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:39
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:262
AVFilter
Filter definition.
Definition: avfilter.h:166
ret
ret
Definition: filter_design.txt:187
CellAutoContext::w
int w
Definition: vsrc_cellauto.c:40
pos
unsigned int pos
Definition: spdifenc.c:413
AVFrame::sample_aspect_ratio
AVRational sample_aspect_ratio
Sample aspect ratio for the video frame, 0/1 if unknown/unspecified.
Definition: frame.h:482
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vsrc_cellauto.c:207
random_seed.h
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:235
avfilter.h
init_pattern_from_string
static int init_pattern_from_string(AVFilterContext *ctx)
Definition: vsrc_cellauto.c:102
file.h
AVFilterContext
An instance of a filter.
Definition: avfilter.h:407
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
CellAutoContext::buf_row_idx
int buf_row_idx
Definition: vsrc_cellauto.c:46
av_free
#define av_free(p)
Definition: tableprint_vlc.h:33
CellAutoContext::start_full
int start_full
Definition: vsrc_cellauto.c:52
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Definition: opt.h:251
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:183
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:34
CellAutoContext::filename
char * filename
Definition: vsrc_cellauto.c:41
AVFrame::linesize
int linesize[AV_NUM_DATA_POINTERS]
For video, a positive or negative value, which is typically indicating the size in bytes of each pict...
Definition: frame.h:420
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
cellauto_options
static const AVOption cellauto_options[]
Definition: vsrc_cellauto.c:61
CellAutoContext::buf
uint8_t * buf
Definition: vsrc_cellauto.c:45
avstring.h
AV_OPT_TYPE_STRING
@ AV_OPT_TYPE_STRING
Definition: opt.h:239