Statistics
| Branch: | Revision:

ffmpeg / libavformat / img2.c @ 470bce2b

History | View | Annotate | Download (11.2 KB)

1 03cfe134 Michael Niedermayer
/*
2
 * Image format
3
 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
4
 * Copyright (c) 2004 Michael Niedermayer
5
 *
6 b78e7197 Diego Biurrun
 * This file is part of FFmpeg.
7
 *
8
 * FFmpeg is free software; you can redistribute it and/or
9 03cfe134 Michael Niedermayer
 * modify it under the terms of the GNU Lesser General Public
10
 * License as published by the Free Software Foundation; either
11 b78e7197 Diego Biurrun
 * version 2.1 of the License, or (at your option) any later version.
12 03cfe134 Michael Niedermayer
 *
13 b78e7197 Diego Biurrun
 * FFmpeg is distributed in the hope that it will be useful,
14 03cfe134 Michael Niedermayer
 * 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 b78e7197 Diego Biurrun
 * License along with FFmpeg; if not, write to the Free Software
20 5509bffa Diego Biurrun
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 03cfe134 Michael Niedermayer
 */
22 245976da Diego Biurrun
23
#include "libavutil/avstring.h"
24 03cfe134 Michael Niedermayer
#include "avformat.h"
25 3375a6a5 Diego Pettenò
#include <strings.h>
26 03cfe134 Michael Niedermayer
27
typedef struct {
28
    int img_first;
29
    int img_last;
30
    int img_number;
31
    int img_count;
32
    int is_pipe;
33
    char path[1024];
34
} VideoData;
35
36
typedef struct {
37
    enum CodecID id;
38
    const char *str;
39
} IdStrMap;
40
41
static const IdStrMap img_tags[] = {
42
    { CODEC_ID_MJPEG     , "jpeg"},
43
    { CODEC_ID_MJPEG     , "jpg"},
44
    { CODEC_ID_LJPEG     , "ljpg"},
45 4eff7cf4 Michael Niedermayer
    { CODEC_ID_PNG       , "png"},
46 042e0add Loren Merritt
    { CODEC_ID_PNG       , "mng"},
47 5b6d5596 Michael Niedermayer
    { CODEC_ID_PPM       , "ppm"},
48 485cfbe8 Carl Eugen Hoyos
    { CODEC_ID_PPM       , "pnm"},
49 5b6d5596 Michael Niedermayer
    { CODEC_ID_PGM       , "pgm"},
50
    { CODEC_ID_PGMYUV    , "pgmyuv"},
51
    { CODEC_ID_PBM       , "pbm"},
52
    { CODEC_ID_PAM       , "pam"},
53 03cfe134 Michael Niedermayer
    { CODEC_ID_MPEG1VIDEO, "mpg1-img"},
54
    { CODEC_ID_MPEG2VIDEO, "mpg2-img"},
55
    { CODEC_ID_MPEG4     , "mpg4-img"},
56
    { CODEC_ID_FFV1      , "ffv1-img"},
57 55cf1959 Michael Niedermayer
    { CODEC_ID_RAWVIDEO  , "y"},
58 9fa62f2a Måns Rullgård
    { CODEC_ID_BMP       , "bmp"},
59 6f9e492a Baptiste Coudurier
    { CODEC_ID_GIF       , "gif"},
60 79f7c32f Kostya Shishkov
    { CODEC_ID_TARGA     , "tga"},
61
    { CODEC_ID_TIFF      , "tiff"},
62 dd0f776c Quoc Cuong Pham
    { CODEC_ID_TIFF      , "tif"},
63 2d99eed1 Xiaohui Sun
    { CODEC_ID_SGI       , "sgi"},
64 8d2fb333 Ivo van Poorten
    { CODEC_ID_PTX       , "ptx"},
65 b4abe1d1 Ivo van Poorten
    { CODEC_ID_PCX       , "pcx"},
66 d43df959 Ivo van Poorten
    { CODEC_ID_SUNRAST   , "sun"},
67
    { CODEC_ID_SUNRAST   , "ras"},
68
    { CODEC_ID_SUNRAST   , "rs"},
69
    { CODEC_ID_SUNRAST   , "im1"},
70
    { CODEC_ID_SUNRAST   , "im8"},
71
    { CODEC_ID_SUNRAST   , "im24"},
72
    { CODEC_ID_SUNRAST   , "sunras"},
73 8b2386dc Benoit Fouet
    { CODEC_ID_JPEG2000  , "jp2"},
74 d9133126 Carl Eugen Hoyos
    { CODEC_ID_NONE      , NULL}
75 03cfe134 Michael Niedermayer
};
76
77 aecf157e Reimar Döffinger
static const int sizes[][2] = {
78 55cf1959 Michael Niedermayer
    { 640, 480 },
79
    { 720, 480 },
80
    { 720, 576 },
81
    { 352, 288 },
82
    { 352, 240 },
83
    { 160, 128 },
84
    { 512, 384 },
85
    { 640, 352 },
86
    { 640, 240 },
87
};
88
89
static int infer_size(int *width_ptr, int *height_ptr, int size)
90
{
91
    int i;
92
93 37d3e066 Aurelien Jacobs
    for(i=0;i<FF_ARRAY_ELEMS(sizes);i++) {
94 55cf1959 Michael Niedermayer
        if ((sizes[i][0] * sizes[i][1]) == size) {
95
            *width_ptr = sizes[i][0];
96
            *height_ptr = sizes[i][1];
97
            return 0;
98
        }
99
    }
100
    return -1;
101
}
102 03cfe134 Michael Niedermayer
static enum CodecID av_str2id(const IdStrMap *tags, const char *str)
103
{
104 3ed02129 Michael Niedermayer
    str= strrchr(str, '.');
105
    if(!str) return CODEC_ID_NONE;
106
    str++;
107 03cfe134 Michael Niedermayer
108
    while (tags->id) {
109 3375a6a5 Diego Pettenò
        if (!strcasecmp(str, tags->str))
110
            return tags->id;
111 03cfe134 Michael Niedermayer
112
        tags++;
113
    }
114
    return CODEC_ID_NONE;
115
}
116
117
/* return -1 if no image found */
118 115329f1 Diego Biurrun
static int find_image_range(int *pfirst_index, int *plast_index,
119 03cfe134 Michael Niedermayer
                            const char *path)
120
{
121
    char buf[1024];
122
    int range, last_index, range1, first_index;
123
124
    /* find the first image */
125
    for(first_index = 0; first_index < 5; first_index++) {
126 5c07cf53 Michel Bardiaux
        if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0){
127 115329f1 Diego Biurrun
            *pfirst_index =
128 61fb3183 Michael Niedermayer
            *plast_index = 1;
129
            return 0;
130
        }
131 03cfe134 Michael Niedermayer
        if (url_exist(buf))
132
            break;
133
    }
134
    if (first_index == 5)
135
        goto fail;
136 115329f1 Diego Biurrun
137 03cfe134 Michael Niedermayer
    /* find the last image */
138
    last_index = first_index;
139
    for(;;) {
140
        range = 0;
141
        for(;;) {
142
            if (!range)
143
                range1 = 1;
144
            else
145
                range1 = 2 * range;
146 5c07cf53 Michel Bardiaux
            if (av_get_frame_filename(buf, sizeof(buf), path,
147
                                      last_index + range1) < 0)
148 03cfe134 Michael Niedermayer
                goto fail;
149
            if (!url_exist(buf))
150
                break;
151
            range = range1;
152
            /* just in case... */
153
            if (range >= (1 << 30))
154
                goto fail;
155
        }
156
        /* we are sure than image last_index + range exists */
157
        if (!range)
158
            break;
159
        last_index += range;
160
    }
161
    *pfirst_index = first_index;
162
    *plast_index = last_index;
163
    return 0;
164
 fail:
165
    return -1;
166
}
167
168
169
static int image_probe(AVProbeData *p)
170
{
171 e825b500 Joakim Plate
    if (p->filename && av_str2id(img_tags, p->filename)) {
172 d1e8675c Baptiste Coudurier
        if (av_filename_number_test(p->filename))
173
            return AVPROBE_SCORE_MAX;
174
        else
175
            return AVPROBE_SCORE_MAX/2;
176
    }
177
    return 0;
178 03cfe134 Michael Niedermayer
}
179
180 5b6d5596 Michael Niedermayer
enum CodecID av_guess_image2_codec(const char *filename){
181
    return av_str2id(img_tags, filename);
182
}
183
184 03cfe134 Michael Niedermayer
static int img_read_header(AVFormatContext *s1, AVFormatParameters *ap)
185
{
186
    VideoData *s = s1->priv_data;
187
    int first_index, last_index;
188
    AVStream *st;
189
190
    s1->ctx_flags |= AVFMTCTX_NOHEADER;
191
192
    st = av_new_stream(s1, 0);
193
    if (!st) {
194 8fa36ae0 François Revol
        return AVERROR(ENOMEM);
195 03cfe134 Michael Niedermayer
    }
196
197 75e61b0e Måns Rullgård
    av_strlcpy(s->path, s1->filename, sizeof(s->path));
198 03cfe134 Michael Niedermayer
    s->img_number = 0;
199
    s->img_count = 0;
200 115329f1 Diego Biurrun
201 03cfe134 Michael Niedermayer
    /* find format */
202
    if (s1->iformat->flags & AVFMT_NOFILE)
203
        s->is_pipe = 0;
204 146210ca Michael Niedermayer
    else{
205 03cfe134 Michael Niedermayer
        s->is_pipe = 1;
206 57004ff1 Aurelien Jacobs
        st->need_parsing = AVSTREAM_PARSE_FULL;
207 146210ca Michael Niedermayer
    }
208 115329f1 Diego Biurrun
209 c04c3282 Michael Niedermayer
    if (!ap->time_base.num) {
210 c0df9d75 Michael Niedermayer
        av_set_pts_info(st, 60, 1, 25);
211 03cfe134 Michael Niedermayer
    } else {
212 c0df9d75 Michael Niedermayer
        av_set_pts_info(st, 60, ap->time_base.num, ap->time_base.den);
213 03cfe134 Michael Niedermayer
    }
214 115329f1 Diego Biurrun
215 c04c3282 Michael Niedermayer
    if(ap->width && ap->height){
216 01f4895c Michael Niedermayer
        st->codec->width = ap->width;
217
        st->codec->height= ap->height;
218 55cf1959 Michael Niedermayer
    }
219 115329f1 Diego Biurrun
220 03cfe134 Michael Niedermayer
    if (!s->is_pipe) {
221
        if (find_image_range(&first_index, &last_index, s->path) < 0)
222 6f3e0b21 Panagiotis Issaris
            return AVERROR(EIO);
223 03cfe134 Michael Niedermayer
        s->img_first = first_index;
224
        s->img_last = last_index;
225
        s->img_number = first_index;
226
        /* compute duration */
227
        st->start_time = 0;
228 c0df9d75 Michael Niedermayer
        st->duration = last_index - first_index + 1;
229 03cfe134 Michael Niedermayer
    }
230 115329f1 Diego Biurrun
231 5b6d5596 Michael Niedermayer
    if(ap->video_codec_id){
232 01f4895c Michael Niedermayer
        st->codec->codec_type = CODEC_TYPE_VIDEO;
233
        st->codec->codec_id = ap->video_codec_id;
234 5b6d5596 Michael Niedermayer
    }else if(ap->audio_codec_id){
235 01f4895c Michael Niedermayer
        st->codec->codec_type = CODEC_TYPE_AUDIO;
236
        st->codec->codec_id = ap->audio_codec_id;
237 5b6d5596 Michael Niedermayer
    }else{
238 01f4895c Michael Niedermayer
        st->codec->codec_type = CODEC_TYPE_VIDEO;
239
        st->codec->codec_id = av_str2id(img_tags, s->path);
240 5b6d5596 Michael Niedermayer
    }
241 01f4895c Michael Niedermayer
    if(st->codec->codec_type == CODEC_TYPE_VIDEO && ap->pix_fmt != PIX_FMT_NONE)
242
        st->codec->pix_fmt = ap->pix_fmt;
243 03cfe134 Michael Niedermayer
244
    return 0;
245
}
246
247
static int img_read_packet(AVFormatContext *s1, AVPacket *pkt)
248
{
249
    VideoData *s = s1->priv_data;
250
    char filename[1024];
251 55cf1959 Michael Niedermayer
    int i;
252
    int size[3]={0}, ret[3]={0};
253 899681cd Björn Axelsson
    ByteIOContext *f[3];
254 01f4895c Michael Niedermayer
    AVCodecContext *codec= s1->streams[0]->codec;
255 03cfe134 Michael Niedermayer
256
    if (!s->is_pipe) {
257
        /* loop over input */
258 5894e1bb Víctor Paesa
        if (s1->loop_input && s->img_number > s->img_last) {
259 03cfe134 Michael Niedermayer
            s->img_number = s->img_first;
260 b339fb00 Michael Niedermayer
        }
261 5c07cf53 Michel Bardiaux
        if (av_get_frame_filename(filename, sizeof(filename),
262
                                  s->path, s->img_number)<0 && s->img_number > 1)
263 6f3e0b21 Panagiotis Issaris
            return AVERROR(EIO);
264 55cf1959 Michael Niedermayer
        for(i=0; i<3; i++){
265 899681cd Björn Axelsson
            if (url_fopen(&f[i], filename, URL_RDONLY) < 0)
266 6f3e0b21 Panagiotis Issaris
                return AVERROR(EIO);
267 a965c478 Aurelien Jacobs
            size[i]= url_fsize(f[i]);
268 115329f1 Diego Biurrun
269 55cf1959 Michael Niedermayer
            if(codec->codec_id != CODEC_ID_RAWVIDEO)
270
                break;
271
            filename[ strlen(filename) - 1 ]= 'U' + i;
272
        }
273 115329f1 Diego Biurrun
274 55cf1959 Michael Niedermayer
        if(codec->codec_id == CODEC_ID_RAWVIDEO && !codec->width)
275
            infer_size(&codec->width, &codec->height, size[0]);
276 03cfe134 Michael Niedermayer
    } else {
277 899681cd Björn Axelsson
        f[0] = s1->pb;
278 55cf1959 Michael Niedermayer
        if (url_feof(f[0]))
279 6f3e0b21 Panagiotis Issaris
            return AVERROR(EIO);
280 55cf1959 Michael Niedermayer
        size[0]= 4096;
281 03cfe134 Michael Niedermayer
    }
282
283 55cf1959 Michael Niedermayer
    av_new_packet(pkt, size[0] + size[1] + size[2]);
284 03cfe134 Michael Niedermayer
    pkt->stream_index = 0;
285
    pkt->flags |= PKT_FLAG_KEY;
286
287 55cf1959 Michael Niedermayer
    pkt->size= 0;
288
    for(i=0; i<3; i++){
289
        if(size[i]){
290
            ret[i]= get_buffer(f[i], pkt->data + pkt->size, size[i]);
291
            if (!s->is_pipe)
292
                url_fclose(f[i]);
293
            if(ret[i]>0)
294
                pkt->size += ret[i];
295
        }
296 03cfe134 Michael Niedermayer
    }
297
298 55cf1959 Michael Niedermayer
    if (ret[0] <= 0 || ret[1]<0 || ret[2]<0) {
299 03cfe134 Michael Niedermayer
        av_free_packet(pkt);
300 6f3e0b21 Panagiotis Issaris
        return AVERROR(EIO); /* signal EOF */
301 03cfe134 Michael Niedermayer
    } else {
302
        s->img_count++;
303
        s->img_number++;
304
        return 0;
305
    }
306
}
307
308 8212568a Diego Biurrun
#if defined(CONFIG_IMAGE2_MUXER) || defined(CONFIG_IMAGE2PIPE_MUXER)
309 03cfe134 Michael Niedermayer
/******************************************************/
310
/* image output */
311
312
static int img_write_header(AVFormatContext *s)
313
{
314
    VideoData *img = s->priv_data;
315
316
    img->img_number = 1;
317 75e61b0e Måns Rullgård
    av_strlcpy(img->path, s->filename, sizeof(img->path));
318 03cfe134 Michael Niedermayer
319
    /* find format */
320
    if (s->oformat->flags & AVFMT_NOFILE)
321
        img->is_pipe = 0;
322
    else
323
        img->is_pipe = 1;
324 115329f1 Diego Biurrun
325 03cfe134 Michael Niedermayer
    return 0;
326
}
327
328
static int img_write_packet(AVFormatContext *s, AVPacket *pkt)
329
{
330
    VideoData *img = s->priv_data;
331 899681cd Björn Axelsson
    ByteIOContext *pb[3];
332 03cfe134 Michael Niedermayer
    char filename[1024];
333 01f4895c Michael Niedermayer
    AVCodecContext *codec= s->streams[ pkt->stream_index ]->codec;
334 55cf1959 Michael Niedermayer
    int i;
335 03cfe134 Michael Niedermayer
336
    if (!img->is_pipe) {
337 5c07cf53 Michel Bardiaux
        if (av_get_frame_filename(filename, sizeof(filename),
338
                                  img->path, img->img_number) < 0 && img->img_number>1)
339 6f3e0b21 Panagiotis Issaris
            return AVERROR(EIO);
340 55cf1959 Michael Niedermayer
        for(i=0; i<3; i++){
341 899681cd Björn Axelsson
            if (url_fopen(&pb[i], filename, URL_WRONLY) < 0)
342 6f3e0b21 Panagiotis Issaris
                return AVERROR(EIO);
343 115329f1 Diego Biurrun
344 55cf1959 Michael Niedermayer
            if(codec->codec_id != CODEC_ID_RAWVIDEO)
345
                break;
346
            filename[ strlen(filename) - 1 ]= 'U' + i;
347
        }
348 03cfe134 Michael Niedermayer
    } else {
349 899681cd Björn Axelsson
        pb[0] = s->pb;
350 03cfe134 Michael Niedermayer
    }
351 115329f1 Diego Biurrun
352 55cf1959 Michael Niedermayer
    if(codec->codec_id == CODEC_ID_RAWVIDEO){
353 2c34596f Michael Niedermayer
        int ysize = codec->width * codec->height;
354
        put_buffer(pb[0], pkt->data        , ysize);
355
        put_buffer(pb[1], pkt->data + ysize, (pkt->size - ysize)/2);
356
        put_buffer(pb[2], pkt->data + ysize +(pkt->size - ysize)/2, (pkt->size - ysize)/2);
357 55cf1959 Michael Niedermayer
        put_flush_packet(pb[1]);
358
        put_flush_packet(pb[2]);
359
        url_fclose(pb[1]);
360
        url_fclose(pb[2]);
361
    }else{
362
        put_buffer(pb[0], pkt->data, pkt->size);
363
    }
364
    put_flush_packet(pb[0]);
365 03cfe134 Michael Niedermayer
    if (!img->is_pipe) {
366 55cf1959 Michael Niedermayer
        url_fclose(pb[0]);
367 03cfe134 Michael Niedermayer
    }
368
369
    img->img_number++;
370
    return 0;
371
}
372
373 8212568a Diego Biurrun
#endif /* defined(CONFIG_IMAGE2_MUXER) || defined(CONFIG_IMAGE2PIPE_MUXER) */
374 8228bff5 Diego Biurrun
375 03cfe134 Michael Niedermayer
/* input */
376 ff70e601 Måns Rullgård
#ifdef CONFIG_IMAGE2_DEMUXER
377
AVInputFormat image2_demuxer = {
378 03cfe134 Michael Niedermayer
    "image2",
379 bde15e74 Stefano Sabatini
    NULL_IF_CONFIG_SMALL("image2 sequence"),
380 03cfe134 Michael Niedermayer
    sizeof(VideoData),
381
    image_probe,
382
    img_read_header,
383
    img_read_packet,
384 bea91b8c Ramiro Polla
    NULL,
385 03cfe134 Michael Niedermayer
    NULL,
386
    NULL,
387 61fb3183 Michael Niedermayer
    AVFMT_NOFILE,
388 03cfe134 Michael Niedermayer
};
389 ff70e601 Måns Rullgård
#endif
390
#ifdef CONFIG_IMAGE2PIPE_DEMUXER
391
AVInputFormat image2pipe_demuxer = {
392 03cfe134 Michael Niedermayer
    "image2pipe",
393 bde15e74 Stefano Sabatini
    NULL_IF_CONFIG_SMALL("piped image2 sequence"),
394 03cfe134 Michael Niedermayer
    sizeof(VideoData),
395
    NULL, /* no probe */
396
    img_read_header,
397
    img_read_packet,
398
};
399 ff70e601 Måns Rullgård
#endif
400 03cfe134 Michael Niedermayer
401
/* output */
402 ff70e601 Måns Rullgård
#ifdef CONFIG_IMAGE2_MUXER
403
AVOutputFormat image2_muxer = {
404 03cfe134 Michael Niedermayer
    "image2",
405 bde15e74 Stefano Sabatini
    NULL_IF_CONFIG_SMALL("image2 sequence"),
406 03cfe134 Michael Niedermayer
    "",
407 a8e41613 Carl Eugen Hoyos
    "bmp,jpeg,jpg,ljpg,pam,pbm,pgm,pgmyuv,png,ppm,sgi,tif,tiff",
408 03cfe134 Michael Niedermayer
    sizeof(VideoData),
409
    CODEC_ID_NONE,
410
    CODEC_ID_MJPEG,
411
    img_write_header,
412
    img_write_packet,
413 82213f68 Baptiste Coudurier
    NULL,
414 644b0c4c Michael Niedermayer
    .flags= AVFMT_NOTIMESTAMPS | AVFMT_NOFILE
415 03cfe134 Michael Niedermayer
};
416 ff70e601 Måns Rullgård
#endif
417
#ifdef CONFIG_IMAGE2PIPE_MUXER
418
AVOutputFormat image2pipe_muxer = {
419 03cfe134 Michael Niedermayer
    "image2pipe",
420 bde15e74 Stefano Sabatini
    NULL_IF_CONFIG_SMALL("piped image2 sequence"),
421 03cfe134 Michael Niedermayer
    "",
422
    "",
423
    sizeof(VideoData),
424
    CODEC_ID_NONE,
425
    CODEC_ID_MJPEG,
426
    img_write_header,
427
    img_write_packet,
428 644b0c4c Michael Niedermayer
    .flags= AVFMT_NOTIMESTAMPS
429 03cfe134 Michael Niedermayer
};
430 8228bff5 Diego Biurrun
#endif