FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
img2dec.c
Go to the documentation of this file.
1 /*
2  * Image format
3  * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
4  * Copyright (c) 2004 Michael Niedermayer
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #define _BSD_SOURCE
24 #include <sys/stat.h>
25 #include "libavutil/avstring.h"
26 #include "libavutil/log.h"
27 #include "libavutil/opt.h"
28 #include "libavutil/pixdesc.h"
29 #include "libavutil/parseutils.h"
30 #include "libavutil/intreadwrite.h"
31 #include "avformat.h"
32 #include "avio_internal.h"
33 #include "internal.h"
34 #include "img2.h"
35 
36 #if HAVE_GLOB
37 /* Locally define as 0 (bitwise-OR no-op) any missing glob options that
38  are non-posix glibc/bsd extensions. */
39 #ifndef GLOB_NOMAGIC
40 #define GLOB_NOMAGIC 0
41 #endif
42 #ifndef GLOB_BRACE
43 #define GLOB_BRACE 0
44 #endif
45 
46 #endif /* HAVE_GLOB */
47 
48 static const int sizes[][2] = {
49  { 640, 480 },
50  { 720, 480 },
51  { 720, 576 },
52  { 352, 288 },
53  { 352, 240 },
54  { 160, 128 },
55  { 512, 384 },
56  { 640, 352 },
57  { 640, 240 },
58 };
59 
60 static int infer_size(int *width_ptr, int *height_ptr, int size)
61 {
62  int i;
63 
64  for (i = 0; i < FF_ARRAY_ELEMS(sizes); i++) {
65  if ((sizes[i][0] * sizes[i][1]) == size) {
66  *width_ptr = sizes[i][0];
67  *height_ptr = sizes[i][1];
68  return 0;
69  }
70  }
71 
72  return -1;
73 }
74 
75 static int is_glob(const char *path)
76 {
77 #if HAVE_GLOB
78  size_t span = 0;
79  const char *p = path;
80 
81  while (p = strchr(p, '%')) {
82  if (*(++p) == '%') {
83  ++p;
84  continue;
85  }
86  if (span = strspn(p, "*?[]{}"))
87  break;
88  }
89  /* Did we hit a glob char or get to the end? */
90  return span != 0;
91 #else
92  return 0;
93 #endif
94 }
95 
96 /**
97  * Get index range of image files matched by path.
98  *
99  * @param pfirst_index pointer to index updated with the first number in the range
100  * @param plast_index pointer to index updated with the last number in the range
101  * @param path path which has to be matched by the image files in the range
102  * @param start_index minimum accepted value for the first index in the range
103  * @return -1 if no image file could be found
104  */
105 static int find_image_range(int *pfirst_index, int *plast_index,
106  const char *path, int start_index, int start_index_range)
107 {
108  char buf[1024];
109  int range, last_index, range1, first_index;
110 
111  /* find the first image */
112  for (first_index = start_index; first_index < start_index + start_index_range; first_index++) {
113  if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0) {
114  *pfirst_index =
115  *plast_index = 1;
116  if (avio_check(buf, AVIO_FLAG_READ) > 0)
117  return 0;
118  return -1;
119  }
120  if (avio_check(buf, AVIO_FLAG_READ) > 0)
121  break;
122  }
123  if (first_index == start_index + start_index_range)
124  goto fail;
125 
126  /* find the last image */
127  last_index = first_index;
128  for (;;) {
129  range = 0;
130  for (;;) {
131  if (!range)
132  range1 = 1;
133  else
134  range1 = 2 * range;
135  if (av_get_frame_filename(buf, sizeof(buf), path,
136  last_index + range1) < 0)
137  goto fail;
138  if (avio_check(buf, AVIO_FLAG_READ) <= 0)
139  break;
140  range = range1;
141  /* just in case... */
142  if (range >= (1 << 30))
143  goto fail;
144  }
145  /* we are sure than image last_index + range exists */
146  if (!range)
147  break;
148  last_index += range;
149  }
150  *pfirst_index = first_index;
151  *plast_index = last_index;
152  return 0;
153 
154 fail:
155  return -1;
156 }
157 
159 {
160  if (p->filename && ff_guess_image2_codec(p->filename)) {
162  return AVPROBE_SCORE_MAX;
163  else if (is_glob(p->filename))
164  return AVPROBE_SCORE_MAX;
165  else if (p->filename[strcspn(p->filename, "*?{")]) // probably PT_GLOB
166  return AVPROBE_SCORE_EXTENSION + 2; // score chosen to be a tad above the image pipes
167  else if (p->buf_size == 0)
168  return 0;
169  else if (av_match_ext(p->filename, "raw") || av_match_ext(p->filename, "gif"))
170  return 5;
171  else
173  }
174  return 0;
175 }
176 
178 {
179  VideoDemuxData *s = s1->priv_data;
180  int first_index = 1, last_index = 1;
181  AVStream *st;
183 
185 
186  st = avformat_new_stream(s1, NULL);
187  if (!st) {
188  return AVERROR(ENOMEM);
189  }
190 
191  if (s->pixel_format &&
192  (pix_fmt = av_get_pix_fmt(s->pixel_format)) == AV_PIX_FMT_NONE) {
193  av_log(s1, AV_LOG_ERROR, "No such pixel format: %s.\n",
194  s->pixel_format);
195  return AVERROR(EINVAL);
196  }
197 
198  av_strlcpy(s->path, s1->filename, sizeof(s->path));
199  s->img_number = 0;
200  s->img_count = 0;
201 
202  /* find format */
203  if (s1->iformat->flags & AVFMT_NOFILE)
204  s->is_pipe = 0;
205  else {
206  s->is_pipe = 1;
208  }
209 
210  if (s->ts_from_file == 2) {
211 #if !HAVE_STRUCT_STAT_ST_MTIM_TV_NSEC
212  av_log(s1, AV_LOG_ERROR, "POSIX.1-2008 not supported, nanosecond file timestamps unavailable\n");
213  return AVERROR(ENOSYS);
214 #endif
215  avpriv_set_pts_info(st, 64, 1, 1000000000);
216  } else if (s->ts_from_file)
217  avpriv_set_pts_info(st, 64, 1, 1);
218  else
220 
221  if (s->width && s->height) {
222  st->codec->width = s->width;
223  st->codec->height = s->height;
224  }
225 
226  if (!s->is_pipe) {
227  if (s->pattern_type == PT_GLOB_SEQUENCE) {
228  s->use_glob = is_glob(s->path);
229  if (s->use_glob) {
230 #if HAVE_GLOB
231  char *p = s->path, *q, *dup;
232  int gerr;
233 #endif
234 
235  av_log(s1, AV_LOG_WARNING, "Pattern type 'glob_sequence' is deprecated: "
236  "use pattern_type 'glob' instead\n");
237 #if HAVE_GLOB
238  dup = q = av_strdup(p);
239  while (*q) {
240  /* Do we have room for the next char and a \ insertion? */
241  if ((p - s->path) >= (sizeof(s->path) - 2))
242  break;
243  if (*q == '%' && strspn(q + 1, "%*?[]{}"))
244  ++q;
245  else if (strspn(q, "\\*?[]{}"))
246  *p++ = '\\';
247  *p++ = *q++;
248  }
249  *p = 0;
250  av_free(dup);
251 
252  gerr = glob(s->path, GLOB_NOCHECK|GLOB_BRACE|GLOB_NOMAGIC, NULL, &s->globstate);
253  if (gerr != 0) {
254  return AVERROR(ENOENT);
255  }
256  first_index = 0;
257  last_index = s->globstate.gl_pathc - 1;
258 #endif
259  }
260  }
261  if ((s->pattern_type == PT_GLOB_SEQUENCE && !s->use_glob) || s->pattern_type == PT_SEQUENCE) {
262  if (find_image_range(&first_index, &last_index, s->path,
263  s->start_number, s->start_number_range) < 0) {
264  av_log(s1, AV_LOG_ERROR,
265  "Could find no file with path '%s' and index in the range %d-%d\n",
266  s->path, s->start_number, s->start_number + s->start_number_range - 1);
267  return AVERROR(ENOENT);
268  }
269  } else if (s->pattern_type == PT_GLOB) {
270 #if HAVE_GLOB
271  int gerr;
272  gerr = glob(s->path, GLOB_NOCHECK|GLOB_BRACE|GLOB_NOMAGIC, NULL, &s->globstate);
273  if (gerr != 0) {
274  return AVERROR(ENOENT);
275  }
276  first_index = 0;
277  last_index = s->globstate.gl_pathc - 1;
278  s->use_glob = 1;
279 #else
280  av_log(s1, AV_LOG_ERROR,
281  "Pattern type 'glob' was selected but globbing "
282  "is not supported by this libavformat build\n");
283  return AVERROR(ENOSYS);
284 #endif
285  } else if (s->pattern_type != PT_GLOB_SEQUENCE && s->pattern_type != PT_NONE) {
286  av_log(s1, AV_LOG_ERROR,
287  "Unknown value '%d' for pattern_type option\n", s->pattern_type);
288  return AVERROR(EINVAL);
289  }
290  s->img_first = first_index;
291  s->img_last = last_index;
292  s->img_number = first_index;
293  /* compute duration */
294  if (!s->ts_from_file) {
295  st->start_time = 0;
296  st->duration = last_index - first_index + 1;
297  }
298  }
299 
300  if (s1->video_codec_id) {
302  st->codec->codec_id = s1->video_codec_id;
303  } else if (s1->audio_codec_id) {
305  st->codec->codec_id = s1->audio_codec_id;
306  } else if (s1->iformat->raw_codec_id) {
308  st->codec->codec_id = s1->iformat->raw_codec_id;
309  } else {
310  const char *str = strrchr(s->path, '.');
311  s->split_planes = str && !av_strcasecmp(str + 1, "y");
313  if (s1->pb) {
314  int probe_buffer_size = 2048;
315  uint8_t *probe_buffer = av_realloc(NULL, probe_buffer_size + AVPROBE_PADDING_SIZE);
317  AVProbeData pd = { 0 };
318 
319  if (!probe_buffer)
320  return AVERROR(ENOMEM);
321 
322  probe_buffer_size = avio_read(s1->pb, probe_buffer, probe_buffer_size);
323  if (probe_buffer_size < 0) {
324  av_free(probe_buffer);
325  return probe_buffer_size;
326  }
327  memset(probe_buffer + probe_buffer_size, 0, AVPROBE_PADDING_SIZE);
328 
329  pd.buf = probe_buffer;
330  pd.buf_size = probe_buffer_size;
331  pd.filename = s1->filename;
332 
333  while ((fmt = av_iformat_next(fmt))) {
334  if (fmt->read_header != ff_img_read_header ||
335  !fmt->read_probe ||
336  (fmt->flags & AVFMT_NOFILE) ||
337  !fmt->raw_codec_id)
338  continue;
339  if (fmt->read_probe(&pd) > 0) {
340  st->codec->codec_id = fmt->raw_codec_id;
341  break;
342  }
343  }
344  if (s1->flags & AVFMT_FLAG_CUSTOM_IO) {
345  avio_seek(s1->pb, 0, SEEK_SET);
346  } else
347  ffio_rewind_with_probe_data(s1->pb, &probe_buffer, probe_buffer_size);
348  }
349  if (st->codec->codec_id == AV_CODEC_ID_NONE)
351  if (st->codec->codec_id == AV_CODEC_ID_LJPEG)
353  if (st->codec->codec_id == AV_CODEC_ID_ALIAS_PIX) // we cannot distingiush this from BRENDER_PIX
355  }
356  if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
357  pix_fmt != AV_PIX_FMT_NONE)
358  st->codec->pix_fmt = pix_fmt;
359 
360  return 0;
361 }
362 
364 {
365  VideoDemuxData *s = s1->priv_data;
366  char filename_bytes[1024];
367  char *filename = filename_bytes;
368  int i, res;
369  int size[3] = { 0 }, ret[3] = { 0 };
370  AVIOContext *f[3] = { NULL };
371  AVCodecContext *codec = s1->streams[0]->codec;
372 
373  if (!s->is_pipe) {
374  /* loop over input */
375  if (s->loop && s->img_number > s->img_last) {
376  s->img_number = s->img_first;
377  }
378  if (s->img_number > s->img_last)
379  return AVERROR_EOF;
380  if (s->pattern_type == PT_NONE) {
381  av_strlcpy(filename_bytes, s->path, sizeof(filename_bytes));
382  } else if (s->use_glob) {
383 #if HAVE_GLOB
384  filename = s->globstate.gl_pathv[s->img_number];
385 #endif
386  } else {
387  if (av_get_frame_filename(filename_bytes, sizeof(filename_bytes),
388  s->path,
389  s->img_number) < 0 && s->img_number > 1)
390  return AVERROR(EIO);
391  }
392  for (i = 0; i < 3; i++) {
393  if (avio_open2(&f[i], filename, AVIO_FLAG_READ,
394  &s1->interrupt_callback, NULL) < 0) {
395  if (i >= 1)
396  break;
397  av_log(s1, AV_LOG_ERROR, "Could not open file : %s\n",
398  filename);
399  return AVERROR(EIO);
400  }
401  size[i] = avio_size(f[i]);
402 
403  if (!s->split_planes)
404  break;
405  filename[strlen(filename) - 1] = 'U' + i;
406  }
407 
408  if (codec->codec_id == AV_CODEC_ID_NONE) {
409  AVProbeData pd = { 0 };
410  AVInputFormat *ifmt;
412  int ret;
413  int score = 0;
414 
415  ret = avio_read(f[0], header, PROBE_BUF_MIN);
416  if (ret < 0)
417  return ret;
418  memset(header + ret, 0, sizeof(header) - ret);
419  avio_skip(f[0], -ret);
420  pd.buf = header;
421  pd.buf_size = ret;
422  pd.filename = filename;
423 
424  ifmt = av_probe_input_format3(&pd, 1, &score);
425  if (ifmt && ifmt->read_packet == ff_img_read_packet && ifmt->raw_codec_id)
426  codec->codec_id = ifmt->raw_codec_id;
427  }
428 
429  if (codec->codec_id == AV_CODEC_ID_RAWVIDEO && !codec->width)
430  infer_size(&codec->width, &codec->height, size[0]);
431  } else {
432  f[0] = s1->pb;
433  if (avio_feof(f[0]) && s->loop && s->is_pipe)
434  avio_seek(f[0], 0, SEEK_SET);
435  if (avio_feof(f[0]))
436  return AVERROR_EOF;
437  if (s->frame_size > 0) {
438  size[0] = s->frame_size;
439  } else if (!s1->streams[0]->parser) {
440  size[0] = avio_size(s1->pb);
441  } else {
442  size[0] = 4096;
443  }
444  }
445 
446  res = av_new_packet(pkt, size[0] + size[1] + size[2]);
447  if (res < 0)
448  return res;
449  pkt->stream_index = 0;
450  pkt->flags |= AV_PKT_FLAG_KEY;
451  if (s->ts_from_file) {
452  struct stat img_stat;
453  if (stat(filename, &img_stat))
454  return AVERROR(EIO);
455  pkt->pts = (int64_t)img_stat.st_mtime;
456 #if HAVE_STRUCT_STAT_ST_MTIM_TV_NSEC
457  if (s->ts_from_file == 2)
458  pkt->pts = 1000000000*pkt->pts + img_stat.st_mtim.tv_nsec;
459 #endif
460  av_add_index_entry(s1->streams[0], s->img_number, pkt->pts, 0, 0, AVINDEX_KEYFRAME);
461  } else if (!s->is_pipe) {
462  pkt->pts = s->pts;
463  }
464 
465  if (s->is_pipe)
466  pkt->pos = avio_tell(f[0]);
467 
468  pkt->size = 0;
469  for (i = 0; i < 3; i++) {
470  if (f[i]) {
471  ret[i] = avio_read(f[i], pkt->data + pkt->size, size[i]);
472  if (s->loop && s->is_pipe && ret[i] == AVERROR_EOF) {
473  if (avio_seek(f[i], 0, SEEK_SET) >= 0) {
474  pkt->pos = 0;
475  ret[i] = avio_read(f[i], pkt->data + pkt->size, size[i]);
476  }
477  }
478  if (!s->is_pipe)
479  avio_closep(&f[i]);
480  if (ret[i] > 0)
481  pkt->size += ret[i];
482  }
483  }
484 
485  if (ret[0] <= 0 || ret[1] < 0 || ret[2] < 0) {
486  av_free_packet(pkt);
487  if (ret[0] < 0) {
488  return ret[0];
489  } else if (ret[1] < 0) {
490  return ret[1];
491  } else if (ret[2] < 0)
492  return ret[2];
493  return AVERROR_EOF;
494  } else {
495  s->img_count++;
496  s->img_number++;
497  s->pts++;
498  return 0;
499  }
500 }
501 
502 static int img_read_close(struct AVFormatContext* s1)
503 {
504 #if HAVE_GLOB
505  VideoDemuxData *s = s1->priv_data;
506  if (s->use_glob) {
507  globfree(&s->globstate);
508  }
509 #endif
510  return 0;
511 }
512 
513 static int img_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
514 {
516  AVStream *st = s->streams[0];
517 
518  if (s1->ts_from_file) {
519  int index = av_index_search_timestamp(st, timestamp, flags);
520  if(index < 0)
521  return -1;
522  s1->img_number = st->index_entries[index].pos;
523  return 0;
524  }
525 
526  if (timestamp < 0 || !s1->loop && timestamp > s1->img_last - s1->img_first)
527  return -1;
528  s1->img_number = timestamp%(s1->img_last - s1->img_first + 1) + s1->img_first;
529  s1->pts = timestamp;
530  return 0;
531 }
532 
533 #define OFFSET(x) offsetof(VideoDemuxData, x)
534 #define DEC AV_OPT_FLAG_DECODING_PARAM
536  { "framerate", "set the video framerate", OFFSET(framerate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, 0, DEC },
537  { "loop", "force loop over input file sequence", OFFSET(loop), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, 1, DEC },
538 
539  { "pattern_type", "set pattern type", OFFSET(pattern_type), AV_OPT_TYPE_INT, {.i64=PT_GLOB_SEQUENCE}, 0, INT_MAX, DEC, "pattern_type"},
540  { "glob_sequence","select glob/sequence pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB_SEQUENCE}, INT_MIN, INT_MAX, DEC, "pattern_type" },
541  { "glob", "select glob pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB }, INT_MIN, INT_MAX, DEC, "pattern_type" },
542  { "sequence", "select sequence pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_SEQUENCE }, INT_MIN, INT_MAX, DEC, "pattern_type" },
543  { "none", "disable pattern matching", 0, AV_OPT_TYPE_CONST, {.i64=PT_NONE }, INT_MIN, INT_MAX, DEC, "pattern_type" },
544 
545  { "pixel_format", "set video pixel format", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC },
546  { "start_number", "set first number in the sequence", OFFSET(start_number), AV_OPT_TYPE_INT, {.i64 = 0 }, INT_MIN, INT_MAX, DEC },
547  { "start_number_range", "set range for looking at the first sequence number", OFFSET(start_number_range), AV_OPT_TYPE_INT, {.i64 = 5}, 1, INT_MAX, DEC },
548  { "video_size", "set video size", OFFSET(width), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, DEC },
549  { "frame_size", "force frame size in bytes", OFFSET(frame_size), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, INT_MAX, DEC },
550  { "ts_from_file", "set frame timestamp from file's one", OFFSET(ts_from_file), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, 2, DEC, "ts_type" },
551  { "none", "none", 0, AV_OPT_TYPE_CONST, {.i64 = 0 }, 0, 2, DEC, "ts_type" },
552  { "sec", "second precision", 0, AV_OPT_TYPE_CONST, {.i64 = 1 }, 0, 2, DEC, "ts_type" },
553  { "ns", "nano second precision", 0, AV_OPT_TYPE_CONST, {.i64 = 2 }, 0, 2, DEC, "ts_type" },
554  { NULL },
555 };
556 
557 #if CONFIG_IMAGE2_DEMUXER
558 static const AVClass img2_class = {
559  .class_name = "image2 demuxer",
560  .item_name = av_default_item_name,
561  .option = ff_img_options,
562  .version = LIBAVUTIL_VERSION_INT,
563 };
564 AVInputFormat ff_image2_demuxer = {
565  .name = "image2",
566  .long_name = NULL_IF_CONFIG_SMALL("image2 sequence"),
567  .priv_data_size = sizeof(VideoDemuxData),
573  .flags = AVFMT_NOFILE,
574  .priv_class = &img2_class,
575 };
576 #endif
577 #if CONFIG_IMAGE2PIPE_DEMUXER
578 static const AVClass img2pipe_class = {
579  .class_name = "image2pipe demuxer",
580  .item_name = av_default_item_name,
581  .option = ff_img_options,
582  .version = LIBAVUTIL_VERSION_INT,
583 };
584 AVInputFormat ff_image2pipe_demuxer = {
585  .name = "image2pipe",
586  .long_name = NULL_IF_CONFIG_SMALL("piped image2 sequence"),
587  .priv_data_size = sizeof(VideoDemuxData),
590  .priv_class = &img2pipe_class,
591 };
592 #endif
593 
594 static int bmp_probe(AVProbeData *p)
595 {
596  const uint8_t *b = p->buf;
597  int ihsize;
598 
599  if (AV_RB16(b) != 0x424d)
600  return 0;
601 
602  ihsize = AV_RL32(b+14);
603  if (ihsize < 12 || ihsize > 255)
604  return 0;
605 
606  if (!AV_RN32(b + 6)) {
607  return AVPROBE_SCORE_EXTENSION + 1;
608  }
609  return AVPROBE_SCORE_EXTENSION / 4;
610 }
611 
612 static int dds_probe(AVProbeData *p)
613 {
614  const uint8_t *b = p->buf;
615 
616  if ( AV_RB64(b) == 0x444453207c000000
617  && AV_RL32(b + 8)
618  && AV_RL32(b + 12))
619  return AVPROBE_SCORE_MAX - 1;
620  return 0;
621 }
622 
623 static int dpx_probe(AVProbeData *p)
624 {
625  const uint8_t *b = p->buf;
626  int w, h;
627  int is_big = (AV_RN32(b) == AV_RN32("SDPX"));
628 
629  if (p->buf_size < 0x304+8)
630  return 0;
631  w = is_big ? AV_RB32(p->buf + 0x304) : AV_RL32(p->buf + 0x304);
632  h = is_big ? AV_RB32(p->buf + 0x308) : AV_RL32(p->buf + 0x308);
633  if (w <= 0 || h <= 0)
634  return 0;
635 
636  if (is_big || AV_RN32(b) == AV_RN32("XPDS"))
637  return AVPROBE_SCORE_EXTENSION + 1;
638  return 0;
639 }
640 
641 static int exr_probe(AVProbeData *p)
642 {
643  const uint8_t *b = p->buf;
644 
645  if (AV_RL32(b) == 20000630)
646  return AVPROBE_SCORE_EXTENSION + 1;
647  return 0;
648 }
649 
650 static int j2k_probe(AVProbeData *p)
651 {
652  const uint8_t *b = p->buf;
653 
654  if (AV_RB64(b) == 0x0000000c6a502020 ||
655  AV_RB32(b) == 0xff4fff51)
656  return AVPROBE_SCORE_EXTENSION + 1;
657  return 0;
658 }
659 
660 static int jpeg_probe(AVProbeData *p)
661 {
662  const uint8_t *b = p->buf;
663  int i, state = 0xD8, exif_size = 0;
664 
665  if (AV_RB16(b) != 0xFFD8 ||
666  AV_RB32(b) == 0xFFD8FFF7)
667  return 0;
668 
669  b += 2;
670  if (AV_RB16(b) == 0xFFE1 && AV_RB32(b + 4) == AV_RB32("Exif")) {
671  exif_size = AV_RB16(b + 2) + 2;
672  b += exif_size;
673  }
674  for (i = 0; i + exif_size < p->buf_size - 2; i++) {
675  int c;
676  if (b[i] != 0xFF)
677  continue;
678  c = b[i + 1];
679  switch (c) {
680  case 0xD8:
681  return 0;
682  case 0xC0:
683  case 0xC1:
684  case 0xC2:
685  case 0xC3:
686  case 0xC5:
687  case 0xC6:
688  case 0xC7:
689  if (state != 0xD8)
690  return 0;
691  state = 0xC0;
692  break;
693  case 0xDA:
694  if (state != 0xC0)
695  return 0;
696  state = 0xDA;
697  break;
698  case 0xD9:
699  if (state != 0xDA)
700  return 0;
701  state = 0xD9;
702  break;
703  default:
704  if ( (c >= 0x02 && c <= 0xBF)
705  || c == 0xC8)
706  return 0;
707  }
708  }
709 
710  if (state == 0xD9)
711  return AVPROBE_SCORE_EXTENSION + 1;
712  return AVPROBE_SCORE_EXTENSION / 8;
713 }
714 
715 static int jpegls_probe(AVProbeData *p)
716 {
717  const uint8_t *b = p->buf;
718 
719  if (AV_RB32(b) == 0xffd8fff7)
720  return AVPROBE_SCORE_EXTENSION + 1;
721  return 0;
722 }
723 
724 static int qdraw_probe(AVProbeData *p)
725 {
726  const uint8_t *b = p->buf;
727 
728  if ( p->buf_size >= 528
729  && (AV_RB64(b + 520) & 0xFFFFFFFFFFFF) == 0x001102ff0c00
730  && AV_RB16(b + 520)
731  && AV_RB16(b + 518))
732  return AVPROBE_SCORE_MAX * 3 / 4;
733  if ( (AV_RB64(b + 8) & 0xFFFFFFFFFFFF) == 0x001102ff0c00
734  && AV_RB16(b + 8)
735  && AV_RB16(b + 6))
736  return AVPROBE_SCORE_EXTENSION / 4;
737  return 0;
738 }
739 
740 static int pictor_probe(AVProbeData *p)
741 {
742  const uint8_t *b = p->buf;
743 
744  if (AV_RL16(b) == 0x1234)
745  return AVPROBE_SCORE_EXTENSION / 4;
746  return 0;
747 }
748 
749 static int png_probe(AVProbeData *p)
750 {
751  const uint8_t *b = p->buf;
752 
753  if (AV_RB64(b) == 0x89504e470d0a1a0a)
754  return AVPROBE_SCORE_MAX - 1;
755  return 0;
756 }
757 
758 static int sgi_probe(AVProbeData *p)
759 {
760  const uint8_t *b = p->buf;
761 
762  if (AV_RB16(b) == 474 &&
763  (b[2] & ~1) == 0 &&
764  (b[3] & ~3) == 0 && b[3] &&
765  (AV_RB16(b + 4) & ~7) == 0 && AV_RB16(b + 4))
766  return AVPROBE_SCORE_EXTENSION + 1;
767  return 0;
768 }
769 
771 {
772  const uint8_t *b = p->buf;
773 
774  if (AV_RB32(b) == 0x59a66a95)
775  return AVPROBE_SCORE_EXTENSION + 1;
776  return 0;
777 }
778 
779 static int tiff_probe(AVProbeData *p)
780 {
781  const uint8_t *b = p->buf;
782 
783  if (AV_RB32(b) == 0x49492a00 ||
784  AV_RB32(b) == 0x4D4D002a)
785  return AVPROBE_SCORE_EXTENSION + 1;
786  return 0;
787 }
788 
789 static int webp_probe(AVProbeData *p)
790 {
791  const uint8_t *b = p->buf;
792 
793  if (AV_RB32(b) == 0x52494646 &&
794  AV_RB32(b + 8) == 0x57454250)
795  return AVPROBE_SCORE_MAX - 1;
796  return 0;
797 }
798 
799 #define IMAGEAUTO_DEMUXER(imgname, codecid)\
800 static const AVClass imgname ## _class = {\
801  .class_name = AV_STRINGIFY(imgname) " demuxer",\
802  .item_name = av_default_item_name,\
803  .option = ff_img_options,\
804  .version = LIBAVUTIL_VERSION_INT,\
805 };\
806 AVInputFormat ff_image_ ## imgname ## _pipe_demuxer = {\
807  .name = AV_STRINGIFY(imgname) "_pipe",\
808  .long_name = NULL_IF_CONFIG_SMALL("piped " AV_STRINGIFY(imgname) " sequence"),\
809  .priv_data_size = sizeof(VideoDemuxData),\
810  .read_probe = imgname ## _probe,\
811  .read_header = ff_img_read_header,\
812  .read_packet = ff_img_read_packet,\
813  .priv_class = & imgname ## _class,\
814  .flags = AVFMT_GENERIC_INDEX, \
815  .raw_codec_id = codecid,\
816 };
817 
int img_number
Definition: img2.h:44
#define NULL
Definition: coverity.c:32
const char * s
Definition: avisynth_c.h:631
Bytestream IO Context.
Definition: avio.h:111
static enum AVPixelFormat pix_fmt
int height
Set by a private option.
Definition: img2.h:51
static int img_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Definition: img2dec.c:513
int64_t avio_size(AVIOContext *s)
Get the filesize.
Definition: aviobuf.c:282
void av_free_packet(AVPacket *pkt)
Free a packet.
Definition: avpacket.c:280
static int bmp_probe(AVProbeData *p)
Definition: img2dec.c:594
AVIOInterruptCB interrupt_callback
Custom interrupt callbacks for the I/O layer.
Definition: avformat.h:1523
AVOption.
Definition: opt.h:255
int av_add_index_entry(AVStream *st, int64_t pos, int64_t timestamp, int size, int distance, int flags)
Add an index entry into a sorted list.
Definition: utils.c:1745
#define OFFSET(x)
Definition: img2dec.c:533
static int qdraw_probe(AVProbeData *p)
Definition: img2dec.c:724
const char * fmt
Definition: avisynth_c.h:632
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:182
#define LIBAVUTIL_VERSION_INT
Definition: version.h:62
const char * filename
Definition: avformat.h:449
int64_t pos
byte position in stream, -1 if unknown
Definition: avcodec.h:1448
int ffio_rewind_with_probe_data(AVIOContext *s, unsigned char **buf, int buf_size)
Rewind the AVIOContext using the specified buffer containing the first buf_size bytes of the file...
Definition: aviobuf.c:869
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:4083
int64_t pos
Definition: avformat.h:784
char path[1024]
Definition: img2.h:49
static int j2k_probe(AVProbeData *p)
Definition: img2dec.c:650
static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, int flags)
Definition: libcdio.c:153
static int img_read_close(struct AVFormatContext *s1)
Definition: img2dec.c:502
int num
numerator
Definition: rational.h:44
int size
Definition: avcodec.h:1424
const char * b
Definition: vf_curves.c:109
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:204
#define AVIO_FLAG_READ
read-only
Definition: avio.h:485
#define AVPROBE_PADDING_SIZE
extra allocated bytes at the end of the probe buffer
Definition: avformat.h:462
AVIndexEntry * index_entries
Only used if the format does not support seeking natively.
Definition: avformat.h:1045
int start_number_range
Definition: img2.h:60
Definition: img2.h:37
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1722
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:277
AVInputFormat * av_probe_input_format3(AVProbeData *pd, int is_opened, int *score_ret)
Guess the file format.
Definition: format.c:169
static AVPacket pkt
int ctx_flags
Flags signalling stream properties.
Definition: avformat.h:1322
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_RB16
Definition: bytestream.h:87
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:87
static int png_probe(AVProbeData *p)
Definition: img2dec.c:749
int img_count
Definition: img2.h:46
char * pixel_format
Set by a private option.
Definition: img2.h:50
#define IMAGEAUTO_DEMUXER(imgname, codecid)
Definition: img2dec.c:799
Format I/O context.
Definition: avformat.h:1273
int avio_check(const char *url, int flags)
Return AVIO_FLAG_* access flags corresponding to the access permissions of the resource in url...
Definition: avio.c:425
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:72
uint8_t
static int dds_probe(AVProbeData *p)
Definition: img2dec.c:612
#define AVFMTCTX_NOHEADER
signal that no header is present (streams are added dynamically)
Definition: avformat.h:1232
AVOptions.
int pattern_type
PatternType.
Definition: img2.h:54
int flags
Can use flags: AVFMT_NOFILE, AVFMT_NEEDNUMBER, AVFMT_SHOW_IDS, AVFMT_GENERIC_INDEX, AVFMT_TS_DISCONT, AVFMT_NOBINSEARCH, AVFMT_NOGENSEARCH, AVFMT_NO_BYTE_SEEK, AVFMT_SEEK_TO_PTS.
Definition: avformat.h:642
static int pictor_probe(AVProbeData *p)
Definition: img2dec.c:740
enum AVStreamParseType need_parsing
Definition: avformat.h:1034
int ff_img_read_header(AVFormatContext *s1)
Definition: img2dec.c:177
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:3749
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_RB32
Definition: bytestream.h:87
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1341
int flags
Flags modifying the (de)muxer behaviour.
Definition: avformat.h:1384
uint8_t * data
Definition: avcodec.h:1423
int(* read_header)(struct AVFormatContext *)
Read the format header and initialize the AVFormatContext structure.
Definition: avformat.h:693
static int img_read_probe(AVProbeData *p)
Definition: img2dec.c:158
#define AVERROR_EOF
End of file.
Definition: error.h:55
static int webp_probe(AVProbeData *p)
Definition: img2dec.c:789
static av_cold int read_close(AVFormatContext *ctx)
Definition: libcdio.c:145
int av_match_ext(const char *filename, const char *extensions)
Return a positive value if the given filename has one of the given extensions, 0 otherwise.
Definition: format.c:81
const AVOption ff_img_options[]
Definition: img2dec.c:535
ptrdiff_t size
Definition: opengl_enc.c:101
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:390
static const uint8_t header[24]
Definition: sdr2.c:67
enum AVCodecID video_codec_id
Forced video codec_id.
Definition: avformat.h:1431
#define av_log(a,...)
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:538
static int exr_probe(AVProbeData *p)
Definition: img2dec.c:641
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: avcodec.h:1469
static int is_glob(const char *path)
Definition: img2dec.c:75
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:83
#define AVINDEX_KEYFRAME
Definition: avformat.h:791
int64_t pts
Definition: img2.h:45
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int av_index_search_timestamp(AVStream *st, int64_t timestamp, int flags)
Get the index for a specific timestamp.
Definition: utils.c:1844
int(* read_probe)(AVProbeData *)
Tell if a given file has a chance of being parsed as this format.
Definition: avformat.h:686
av_default_item_name
static const int sizes[][2]
Definition: img2dec.c:48
int loop
Definition: img2.h:53
#define AVERROR(e)
Definition: error.h:43
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:175
int ts_from_file
Definition: img2.h:62
size_t av_strlcpy(char *dst, const char *src, size_t size)
Copy the string src to dst, but no more than size - 1 bytes, and null-terminate dst.
Definition: avstring.c:83
#define fail()
Definition: checkasm.h:57
int flags
A combination of AV_PKT_FLAG values.
Definition: avcodec.h:1429
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:861
int buf_size
Size of buf except extra allocated bytes.
Definition: avformat.h:451
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:450
char filename[1024]
input or output filename
Definition: avformat.h:1349
static int jpeg_probe(AVProbeData *p)
Definition: img2dec.c:660
enum AVCodecID audio_codec_id
Forced audio codec_id.
Definition: avformat.h:1437
int av_strcasecmp(const char *a, const char *b)
Locale-independent case-insensitive compare.
Definition: avstring.c:213
static int read_probe(AVProbeData *pd)
Definition: jvdec.c:55
int is_pipe
Definition: img2.h:47
int width
picture width / height.
Definition: avcodec.h:1681
int width
Definition: img2.h:51
static struct @197 state
Definition: img2.h:35
static int sgi_probe(AVProbeData *p)
Definition: img2dec.c:758
#define AVFMT_FLAG_CUSTOM_IO
The caller has supplied a custom AVIOContext, don't avio_close() it.
Definition: avformat.h:1392
int start_number
Definition: img2.h:59
#define FF_ARRAY_ELEMS(a)
int av_get_frame_filename(char *buf, int buf_size, const char *path, int number)
Return in 'buf' the path with 'd' replaced by a number.
Definition: utils.c:3908
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:623
#define PROBE_BUF_MIN
size of probe buffer, for guessing file type from file contents
Definition: internal.h:31
Stream structure.
Definition: avformat.h:842
#define DEC
Definition: img2dec.c:534
int frame_size
Definition: img2.h:61
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
int frame_size
Definition: mxfenc.c:1805
enum AVMediaType codec_type
Definition: avcodec.h:1510
enum AVCodecID codec_id
Definition: avcodec.h:1519
char * av_strdup(const char *s)
Duplicate the string s.
Definition: mem.c:267
AVIOContext * pb
I/O context.
Definition: avformat.h:1315
static int loop
Definition: ffplay.c:334
int split_planes
use independent file for each Y, U, V plane
Definition: img2.h:48
main external API structure.
Definition: avcodec.h:1502
int av_filename_number_test(const char *filename)
Check whether filename actually is a numbered sequence generator.
Definition: utils.c:261
int(* read_packet)(struct AVFormatContext *, AVPacket *pkt)
Read one packet and put it in 'pkt'.
Definition: avformat.h:704
void * buf
Definition: avisynth_c.h:553
Describe the class of an AVClass context structure.
Definition: log.h:67
int index
Definition: gxfenc.c:89
static int infer_size(int *width_ptr, int *height_ptr, int size)
Definition: img2dec.c:60
offset must point to AVRational
Definition: opt.h:235
int avio_open2(AVIOContext **s, const char *url, int flags, const AVIOInterruptCB *int_cb, AVDictionary **options)
Create and initialize a AVIOContext for accessing the resource indicated by url.
Definition: aviobuf.c:918
#define s1
Definition: regdef.h:38
#define AVPROBE_SCORE_EXTENSION
score for file extension
Definition: avformat.h:458
offset must point to two consecutive integers
Definition: opt.h:232
This structure contains the data a format has to probe a file.
Definition: avformat.h:448
#define AV_RN32(p)
Definition: intreadwrite.h:364
misc parsing utilities
int img_last
Definition: img2.h:43
static int sunrast_probe(AVProbeData *p)
Definition: img2dec.c:770
int raw_codec_id
Raw demuxers store their codec ID here.
Definition: avformat.h:674
static int flags
Definition: cpu.c:47
int64_t duration
Decoding: duration of the stream, in stream time base.
Definition: avformat.h:901
static int dpx_probe(AVProbeData *p)
Definition: img2dec.c:623
#define AVPROBE_SCORE_MAX
maximum score
Definition: avformat.h:460
static int tiff_probe(AVProbeData *p)
Definition: img2dec.c:779
full parsing and repack
Definition: avformat.h:774
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_RB64
Definition: bytestream.h:87
Main libavformat public API header.
if(ret< 0)
Definition: vf_mcdeint.c:280
#define AVFMT_NOFILE
Demuxer will use avio_open, no opened file should be provided by the caller.
Definition: avformat.h:465
void * av_realloc(void *ptr, size_t size)
Allocate or reallocate a block of memory.
Definition: mem.c:145
static int find_image_range(int *pfirst_index, int *plast_index, const char *path, int start_index, int start_index_range)
Get index range of image files matched by path.
Definition: img2dec.c:105
int64_t start_time
Decoding: pts of the first frame of the stream in presentation order, in stream time base...
Definition: avformat.h:894
static double c[64]
int den
denominator
Definition: rational.h:45
struct AVInputFormat * iformat
The input container format.
Definition: avformat.h:1285
enum AVCodecID ff_guess_image2_codec(const char *filename)
Definition: img2.c:104
#define av_free(p)
AVRational framerate
Set by a private option.
Definition: img2.h:52
struct AVCodecParserContext * parser
Definition: avformat.h:1035
void * priv_data
Format private data.
Definition: avformat.h:1301
static int jpegls_probe(AVProbeData *p)
Definition: img2dec.c:715
int use_glob
Definition: img2.h:55
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:628
int avio_feof(AVIOContext *s)
feof() equivalent for AVIOContext.
Definition: aviobuf.c:301
enum AVPixelFormat av_get_pix_fmt(const char *name)
Return the pixel format corresponding to name.
Definition: pixdesc.c:2062
int stream_index
Definition: avcodec.h:1425
int img_first
Definition: img2.h:42
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:87
AVPixelFormat
Pixel format.
Definition: pixfmt.h:61
This structure stores compressed data.
Definition: avcodec.h:1400
int avio_closep(AVIOContext **s)
Close the resource accessed by the AVIOContext *s, free it and set the pointer pointing to it to NULL...
Definition: aviobuf.c:959
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1416
int ff_img_read_packet(AVFormatContext *s1, AVPacket *pkt)
Definition: img2dec.c:363
AVInputFormat * av_iformat_next(const AVInputFormat *f)
If f is NULL, returns the first registered input format, if f is non-NULL, returns the next registere...
Definition: format.c:45
static int width