]> git.sesse.net Git - ffmpeg/blobdiff - libavformat/img2dec.c
Merge commit 'cb45553f577f8e0ebfe05d3287e1b6fa5859b967'
[ffmpeg] / libavformat / img2dec.c
index c493f1add7a0b3749417957166d457ba65082cc7..f492bba8299cd2538ae4e98f6f7b3220534334a3 100644 (file)
@@ -61,6 +61,7 @@ typedef struct {
 #endif
     int start_number;
     int start_number_range;
+    int frame_size;
 } VideoDemuxData;
 
 static const int sizes[][2] = {
@@ -79,13 +80,14 @@ static int infer_size(int *width_ptr, int *height_ptr, int size)
 {
     int i;
 
-    for(i=0;i<FF_ARRAY_ELEMS(sizes);i++) {
+    for (i = 0; i < FF_ARRAY_ELEMS(sizes); i++) {
         if ((sizes[i][0] * sizes[i][1]) == size) {
-            *width_ptr = sizes[i][0];
+            *width_ptr  = sizes[i][0];
             *height_ptr = sizes[i][1];
             return 0;
         }
     }
+
     return -1;
 }
 
@@ -127,9 +129,9 @@ static int find_image_range(int *pfirst_index, int *plast_index,
 
     /* find the first image */
     for (first_index = start_index; first_index < start_index + start_index_range; first_index++) {
-        if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0){
+        if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0) {
             *pfirst_index =
-            *plast_index = 1;
+            *plast_index  = 1;
             if (avio_check(buf, AVIO_FLAG_READ) > 0)
                 return 0;
             return -1;
@@ -142,9 +144,9 @@ static int find_image_range(int *pfirst_index, int *plast_index,
 
     /* find the last image */
     last_index = first_index;
-    for(;;) {
+    for (;;) {
         range = 0;
-        for(;;) {
+        for (;;) {
             if (!range)
                 range1 = 1;
             else
@@ -165,29 +167,29 @@ static int find_image_range(int *pfirst_index, int *plast_index,
         last_index += range;
     }
     *pfirst_index = first_index;
-    *plast_index = last_index;
+    *plast_index  = last_index;
     return 0;
- fail:
+
+fail:
     return -1;
 }
 
-
-static int read_probe(AVProbeData *p)
+static int img_read_probe(AVProbeData *p)
 {
     if (p->filename && ff_guess_image2_codec(p->filename)) {
         if (av_filename_number_test(p->filename))
             return AVPROBE_SCORE_MAX;
         else if (is_glob(p->filename))
             return AVPROBE_SCORE_MAX;
-        else if(av_match_ext(p->filename, "raw"))
+        else if (av_match_ext(p->filename, "raw") || av_match_ext(p->filename, "gif"))
             return 5;
         else
-            return AVPROBE_SCORE_MAX/2;
+            return AVPROBE_SCORE_MAX / 2;
     }
     return 0;
 }
 
-static int read_header(AVFormatContext *s1)
+static int img_read_header(AVFormatContext *s1)
 {
     VideoDemuxData *s = s1->priv_data;
     int first_index, last_index, ret = 0;
@@ -203,28 +205,33 @@ static int read_header(AVFormatContext *s1)
         return AVERROR(ENOMEM);
     }
 
-    if (s->pixel_format && (pix_fmt = av_get_pix_fmt(s->pixel_format)) == AV_PIX_FMT_NONE) {
-        av_log(s1, AV_LOG_ERROR, "No such pixel format: %s.\n", s->pixel_format);
+    if (s->pixel_format &&
+        (pix_fmt = av_get_pix_fmt(s->pixel_format)) == AV_PIX_FMT_NONE) {
+        av_log(s1, AV_LOG_ERROR, "No such pixel format: %s.\n",
+               s->pixel_format);
         return AVERROR(EINVAL);
     }
-    if (s->video_size && (ret = av_parse_video_size(&width, &height, s->video_size)) < 0) {
-        av_log(s, AV_LOG_ERROR, "Could not parse video size: %s.\n", s->video_size);
+    if (s->video_size &&
+        (ret = av_parse_video_size(&width, &height, s->video_size)) < 0) {
+        av_log(s, AV_LOG_ERROR,
+               "Could not parse video size: %s.\n", s->video_size);
         return ret;
     }
     if ((ret = av_parse_video_rate(&framerate, s->framerate)) < 0) {
-        av_log(s, AV_LOG_ERROR, "Could not parse framerate: %s.\n", s->framerate);
+        av_log(s, AV_LOG_ERROR,
+               "Could not parse framerate: %s.\n", s->framerate);
         return ret;
     }
 
     av_strlcpy(s->path, s1->filename, sizeof(s->path));
     s->img_number = 0;
-    s->img_count = 0;
+    s->img_count  = 0;
 
     /* find format */
     if (s1->iformat->flags & AVFMT_NOFILE)
         s->is_pipe = 0;
-    else{
-        s->is_pipe = 1;
+    else {
+        s->is_pipe       = 1;
         st->need_parsing = AVSTREAM_PARSE_FULL;
     }
 
@@ -297,43 +304,44 @@ static int read_header(AVFormatContext *s1)
                    "Unknown value '%d' for pattern_type option\n", s->pattern_type);
             return AVERROR(EINVAL);
         }
-        s->img_first = first_index;
-        s->img_last = last_index;
+        s->img_first  = first_index;
+        s->img_last   = last_index;
         s->img_number = first_index;
         /* compute duration */
         st->start_time = 0;
-        st->duration = last_index - first_index + 1;
+        st->duration   = last_index - first_index + 1;
     }
 
-    if(s1->video_codec_id){
+    if (s1->video_codec_id) {
         st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
-        st->codec->codec_id = s1->video_codec_id;
-    }else if(s1->audio_codec_id){
+        st->codec->codec_id   = s1->video_codec_id;
+    } else if (s1->audio_codec_id) {
         st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
-        st->codec->codec_id = s1->audio_codec_id;
-    }else{
-        const char *str= strrchr(s->path, '.');
-        s->split_planes = str && !av_strcasecmp(str + 1, "y");
+        st->codec->codec_id   = s1->audio_codec_id;
+    } else {
+        const char *str = strrchr(s->path, '.');
+        s->split_planes       = str && !av_strcasecmp(str + 1, "y");
         st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
-        st->codec->codec_id = ff_guess_image2_codec(s->path);
+        st->codec->codec_id   = ff_guess_image2_codec(s->path);
         if (st->codec->codec_id == AV_CODEC_ID_LJPEG)
             st->codec->codec_id = AV_CODEC_ID_MJPEG;
     }
-    if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO && pix_fmt != AV_PIX_FMT_NONE)
+    if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
+        pix_fmt != AV_PIX_FMT_NONE)
         st->codec->pix_fmt = pix_fmt;
 
     return 0;
 }
 
-static int read_packet(AVFormatContext *s1, AVPacket *pkt)
+static int img_read_packet(AVFormatContext *s1, AVPacket *pkt)
 {
     VideoDemuxData *s = s1->priv_data;
     char filename_bytes[1024];
     char *filename = filename_bytes;
     int i;
-    int size[3]={0}, ret[3]={0};
-    AVIOContext *f[3] = {NULL};
-    AVCodecContext *codec= s1->streams[0]->codec;
+    int size[3]           = { 0 }, ret[3] = { 0 };
+    AVIOContext *f[3]     = { NULL };
+    AVCodecContext *codec = s1->streams[0]->codec;
 
     if (!s->is_pipe) {
         /* loop over input */
@@ -348,50 +356,56 @@ static int read_packet(AVFormatContext *s1, AVPacket *pkt)
 #endif
         } else {
         if (av_get_frame_filename(filename_bytes, sizeof(filename_bytes),
-                                  s->path, s->img_number)<0 && s->img_number > 1)
+                                  s->path,
+                                  s->img_number) < 0 && s->img_number > 1)
             return AVERROR(EIO);
         }
-        for(i=0; i<3; i++){
+        for (i = 0; i < 3; i++) {
             if (avio_open2(&f[i], filename, AVIO_FLAG_READ,
                            &s1->interrupt_callback, NULL) < 0) {
-                if(i>=1)
+                if (i >= 1)
                     break;
-                av_log(s1, AV_LOG_ERROR, "Could not open file : %s\n",filename);
+                av_log(s1, AV_LOG_ERROR, "Could not open file : %s\n",
+                       filename);
                 return AVERROR(EIO);
             }
-            size[i]= avio_size(f[i]);
+            size[i] = avio_size(f[i]);
 
-            if(!s->split_planes)
+            if (!s->split_planes)
                 break;
-            filename[ strlen(filename) - 1 ]= 'U' + i;
+            filename[strlen(filename) - 1] = 'U' + i;
         }
 
-        if(codec->codec_id == AV_CODEC_ID_RAWVIDEO && !codec->width)
+        if (codec->codec_id == AV_CODEC_ID_RAWVIDEO && !codec->width)
             infer_size(&codec->width, &codec->height, size[0]);
     } else {
         f[0] = s1->pb;
         if (url_feof(f[0]))
             return AVERROR(EIO);
-        size[0]= 4096;
+        if (s->frame_size > 0) {
+            size[0] = s->frame_size;
+        } else {
+            size[0] = 4096;
+        }
     }
 
     if (av_new_packet(pkt, size[0] + size[1] + size[2]) < 0)
         return AVERROR(ENOMEM);
     pkt->stream_index = 0;
-    pkt->flags |= AV_PKT_FLAG_KEY;
+    pkt->flags       |= AV_PKT_FLAG_KEY;
 
-    pkt->size= 0;
-    for(i=0; i<3; i++){
-        if(f[i]){
-            ret[i]= avio_read(f[i], pkt->data + pkt->size, size[i]);
+    pkt->size = 0;
+    for (i = 0; i < 3; i++) {
+        if (f[i]) {
+            ret[i] = avio_read(f[i], pkt->data + pkt->size, size[i]);
             if (!s->is_pipe)
                 avio_close(f[i]);
-            if(ret[i]>0)
+            if (ret[i] > 0)
                 pkt->size += ret[i];
         }
     }
 
-    if (ret[0] <= 0 || ret[1]<0 || ret[2]<0) {
+    if (ret[0] <= 0 || ret[1] < 0 || ret[2] < 0) {
         av_free_packet(pkt);
         return AVERROR(EIO); /* signal EOF */
     } else {
@@ -401,7 +415,7 @@ static int read_packet(AVFormatContext *s1, AVPacket *pkt)
     }
 }
 
-static int read_close(struct AVFormatContext* s1)
+static int img_read_close(struct AVFormatContext* s1)
 {
     VideoDemuxData *s = s1->priv_data;
 #if HAVE_GLOB
@@ -415,18 +429,19 @@ static int read_close(struct AVFormatContext* s1)
 #define OFFSET(x) offsetof(VideoDemuxData, x)
 #define DEC AV_OPT_FLAG_DECODING_PARAM
 static const AVOption options[] = {
-    { "framerate",    "set the video framerate",             OFFSET(framerate),    AV_OPT_TYPE_STRING, {.str = "25"}, 0, 0, DEC },
-    { "loop",         "force loop over input file sequence", OFFSET(loop),         AV_OPT_TYPE_INT,    {.i64 = 0},    0, 1, DEC },
+    { "framerate",    "set the video framerate",             OFFSET(framerate),    AV_OPT_TYPE_STRING, {.str = "25"}, 0, 0,       DEC },
+    { "loop",         "force loop over input file sequence", OFFSET(loop),         AV_OPT_TYPE_INT,    {.i64 = 0   }, 0, 1,       DEC },
 
-    { "pattern_type", "set pattern type",                    OFFSET(pattern_type), AV_OPT_TYPE_INT, {.i64=PT_GLOB_SEQUENCE}, 0, INT_MAX, DEC, "pattern_type"},
-    { "glob_sequence","glob/sequence pattern type",          0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB_SEQUENCE}, INT_MIN, INT_MAX, DEC, "pattern_type" },
-    { "glob",         "glob pattern type",                   0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB},          INT_MIN, INT_MAX, DEC, "pattern_type" },
-    { "sequence",     "glob pattern type",                   0, AV_OPT_TYPE_CONST, {.i64=PT_SEQUENCE},      INT_MIN, INT_MAX, DEC, "pattern_type" },
+    { "pattern_type", "set pattern type",                    OFFSET(pattern_type), AV_OPT_TYPE_INT,    {.i64=PT_GLOB_SEQUENCE}, 0,       INT_MAX, DEC, "pattern_type"},
+    { "glob_sequence","glob/sequence pattern type",          0,                    AV_OPT_TYPE_CONST,  {.i64=PT_GLOB_SEQUENCE}, INT_MIN, INT_MAX, DEC, "pattern_type" },
+    { "glob",         "glob pattern type",                   0,                    AV_OPT_TYPE_CONST,  {.i64=PT_GLOB         }, INT_MIN, INT_MAX, DEC, "pattern_type" },
+    { "sequence",     "glob pattern type",                   0,                    AV_OPT_TYPE_CONST,  {.i64=PT_SEQUENCE     }, INT_MIN, INT_MAX, DEC, "pattern_type" },
 
-    { "pixel_format", "set video pixel format",              OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC },
-    { "start_number", "set first number in the sequence",    OFFSET(start_number), AV_OPT_TYPE_INT,    {.i64 = 0},    0, INT_MAX, DEC },
+    { "pixel_format", "set video pixel format",              OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0,       DEC },
+    { "start_number", "set first number in the sequence",    OFFSET(start_number), AV_OPT_TYPE_INT,    {.i64 = 0   }, 0, INT_MAX, DEC },
     { "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 },
-    { "video_size",   "set video size",                      OFFSET(video_size),   AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC },
+    { "video_size",   "set video size",                      OFFSET(video_size),   AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0,       DEC },
+    { "frame_size",   "force frame size in bytes",           OFFSET(frame_size),   AV_OPT_TYPE_INT,    {.i64 = 0   }, 0, INT_MAX, DEC },
     { NULL },
 };
 
@@ -441,10 +456,10 @@ AVInputFormat ff_image2_demuxer = {
     .name           = "image2",
     .long_name      = NULL_IF_CONFIG_SMALL("image2 sequence"),
     .priv_data_size = sizeof(VideoDemuxData),
-    .read_probe     = read_probe,
-    .read_header    = read_header,
-    .read_packet    = read_packet,
-    .read_close     = read_close,
+    .read_probe     = img_read_probe,
+    .read_header    = img_read_header,
+    .read_packet    = img_read_packet,
+    .read_close     = img_read_close,
     .flags          = AVFMT_NOFILE,
     .priv_class     = &img2_class,
 };
@@ -460,8 +475,8 @@ AVInputFormat ff_image2pipe_demuxer = {
     .name           = "image2pipe",
     .long_name      = NULL_IF_CONFIG_SMALL("piped image2 sequence"),
     .priv_data_size = sizeof(VideoDemuxData),
-    .read_header    = read_header,
-    .read_packet    = read_packet,
+    .read_header    = img_read_header,
+    .read_packet    = img_read_packet,
     .priv_class     = &img2pipe_class,
 };
 #endif