Statistics
| Branch: | Revision:

ffmpeg / libavformat / img2.c @ 4647c0ec

History | View | Annotate | Download (10.6 KB)

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
#include "avformat.h"
23

    
24
typedef struct {
25
    int img_first;
26
    int img_last;
27
    int img_number;
28
    int img_count;
29
    int is_pipe;
30
    char path[1024];
31
} VideoData;
32

    
33
typedef struct {
34
    enum CodecID id;
35
    const char *str;
36
} IdStrMap;
37

    
38
static const IdStrMap img_tags[] = {
39
    { CODEC_ID_MJPEG     , "jpeg"},
40
    { CODEC_ID_MJPEG     , "jpg"},
41
    { CODEC_ID_LJPEG     , "ljpg"},
42
    { CODEC_ID_PNG       , "png"},
43
    { CODEC_ID_PPM       , "ppm"},
44
    { CODEC_ID_PGM       , "pgm"},
45
    { CODEC_ID_PGMYUV    , "pgmyuv"},
46
    { CODEC_ID_PBM       , "pbm"},
47
    { CODEC_ID_PAM       , "pam"},
48
    { CODEC_ID_MPEG1VIDEO, "mpg1-img"},
49
    { CODEC_ID_MPEG2VIDEO, "mpg2-img"},
50
    { CODEC_ID_MPEG4     , "mpg4-img"},
51
    { CODEC_ID_FFV1      , "ffv1-img"},
52
    { CODEC_ID_RAWVIDEO  , "y"},
53
    { CODEC_ID_BMP       , "bmp"},
54
    { CODEC_ID_GIF       , "gif"},
55
    { CODEC_ID_TARGA     , "tga"},
56
    { CODEC_ID_TIFF      , "tiff"},
57
    {0, NULL}
58
};
59

    
60
static int sizes[][2] = {
61
    { 640, 480 },
62
    { 720, 480 },
63
    { 720, 576 },
64
    { 352, 288 },
65
    { 352, 240 },
66
    { 160, 128 },
67
    { 512, 384 },
68
    { 640, 352 },
69
    { 640, 240 },
70
};
71

    
72
static int infer_size(int *width_ptr, int *height_ptr, int size)
73
{
74
    int i;
75

    
76
    for(i=0;i<sizeof(sizes)/sizeof(sizes[0]);i++) {
77
        if ((sizes[i][0] * sizes[i][1]) == size) {
78
            *width_ptr = sizes[i][0];
79
            *height_ptr = sizes[i][1];
80
            return 0;
81
        }
82
    }
83
    return -1;
84
}
85
static enum CodecID av_str2id(const IdStrMap *tags, const char *str)
86
{
87
    str= strrchr(str, '.');
88
    if(!str) return CODEC_ID_NONE;
89
    str++;
90

    
91
    while (tags->id) {
92
        int i;
93
        for(i=0; toupper(tags->str[i]) == toupper(str[i]); i++){
94
            if(tags->str[i]==0 && str[i]==0)
95
                return tags->id;
96
        }
97

    
98
        tags++;
99
    }
100
    return CODEC_ID_NONE;
101
}
102

    
103
/* return -1 if no image found */
104
static int find_image_range(int *pfirst_index, int *plast_index,
105
                            const char *path)
106
{
107
    char buf[1024];
108
    int range, last_index, range1, first_index;
109

    
110
    /* find the first image */
111
    for(first_index = 0; first_index < 5; first_index++) {
112
        if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0){
113
            *pfirst_index =
114
            *plast_index = 1;
115
            return 0;
116
        }
117
        if (url_exist(buf))
118
            break;
119
    }
120
    if (first_index == 5)
121
        goto fail;
122

    
123
    /* find the last image */
124
    last_index = first_index;
125
    for(;;) {
126
        range = 0;
127
        for(;;) {
128
            if (!range)
129
                range1 = 1;
130
            else
131
                range1 = 2 * range;
132
            if (av_get_frame_filename(buf, sizeof(buf), path,
133
                                      last_index + range1) < 0)
134
                goto fail;
135
            if (!url_exist(buf))
136
                break;
137
            range = range1;
138
            /* just in case... */
139
            if (range >= (1 << 30))
140
                goto fail;
141
        }
142
        /* we are sure than image last_index + range exists */
143
        if (!range)
144
            break;
145
        last_index += range;
146
    }
147
    *pfirst_index = first_index;
148
    *plast_index = last_index;
149
    return 0;
150
 fail:
151
    return -1;
152
}
153

    
154

    
155
static int image_probe(AVProbeData *p)
156
{
157
    if (av_filename_number_test(p->filename) && av_str2id(img_tags, p->filename))
158
        return AVPROBE_SCORE_MAX;
159
    else
160
        return 0;
161
}
162

    
163
enum CodecID av_guess_image2_codec(const char *filename){
164
    return av_str2id(img_tags, filename);
165
}
166

    
167
static int img_read_header(AVFormatContext *s1, AVFormatParameters *ap)
168
{
169
    VideoData *s = s1->priv_data;
170
    int first_index, last_index;
171
    AVStream *st;
172

    
173
    s1->ctx_flags |= AVFMTCTX_NOHEADER;
174

    
175
    st = av_new_stream(s1, 0);
176
    if (!st) {
177
        return -ENOMEM;
178
    }
179

    
180
    pstrcpy(s->path, sizeof(s->path), s1->filename);
181
    s->img_number = 0;
182
    s->img_count = 0;
183

    
184
    /* find format */
185
    if (s1->iformat->flags & AVFMT_NOFILE)
186
        s->is_pipe = 0;
187
    else{
188
        s->is_pipe = 1;
189
        st->need_parsing= 1;
190
    }
191

    
192
    if (!ap->time_base.num) {
193
        av_set_pts_info(st, 60, 1, 25);
194
    } else {
195
        av_set_pts_info(st, 60, ap->time_base.num, ap->time_base.den);
196
    }
197

    
198
    if(ap->width && ap->height){
199
        st->codec->width = ap->width;
200
        st->codec->height= ap->height;
201
    }
202

    
203
    if (!s->is_pipe) {
204
        if (find_image_range(&first_index, &last_index, s->path) < 0)
205
            return AVERROR_IO;
206
        s->img_first = first_index;
207
        s->img_last = last_index;
208
        s->img_number = first_index;
209
        /* compute duration */
210
        st->start_time = 0;
211
        st->duration = last_index - first_index + 1;
212
    }
213

    
214
    if(ap->video_codec_id){
215
        st->codec->codec_type = CODEC_TYPE_VIDEO;
216
        st->codec->codec_id = ap->video_codec_id;
217
    }else if(ap->audio_codec_id){
218
        st->codec->codec_type = CODEC_TYPE_AUDIO;
219
        st->codec->codec_id = ap->audio_codec_id;
220
    }else{
221
        st->codec->codec_type = CODEC_TYPE_VIDEO;
222
        st->codec->codec_id = av_str2id(img_tags, s->path);
223
    }
224
    if(st->codec->codec_type == CODEC_TYPE_VIDEO && ap->pix_fmt != PIX_FMT_NONE)
225
        st->codec->pix_fmt = ap->pix_fmt;
226

    
227
    return 0;
228
}
229

    
230
static int img_read_packet(AVFormatContext *s1, AVPacket *pkt)
231
{
232
    VideoData *s = s1->priv_data;
233
    char filename[1024];
234
    int i;
235
    int size[3]={0}, ret[3]={0};
236
    ByteIOContext f1[3], *f[3]= {&f1[0], &f1[1], &f1[2]};
237
    AVCodecContext *codec= s1->streams[0]->codec;
238

    
239
    if (!s->is_pipe) {
240
        /* loop over input */
241
        if (s1->loop_input && s->img_number > s->img_last) {
242
            s->img_number = s->img_first;
243
        }
244
        if (av_get_frame_filename(filename, sizeof(filename),
245
                                  s->path, s->img_number)<0 && s->img_number > 1)
246
            return AVERROR_IO;
247
        for(i=0; i<3; i++){
248
            if (url_fopen(f[i], filename, URL_RDONLY) < 0)
249
                return AVERROR_IO;
250
            size[i]= url_fsize(f[i]);
251

    
252
            if(codec->codec_id != CODEC_ID_RAWVIDEO)
253
                break;
254
            filename[ strlen(filename) - 1 ]= 'U' + i;
255
        }
256

    
257
        if(codec->codec_id == CODEC_ID_RAWVIDEO && !codec->width)
258
            infer_size(&codec->width, &codec->height, size[0]);
259
    } else {
260
        f[0] = &s1->pb;
261
        if (url_feof(f[0]))
262
            return AVERROR_IO;
263
        size[0]= 4096;
264
    }
265

    
266
    av_new_packet(pkt, size[0] + size[1] + size[2]);
267
    pkt->stream_index = 0;
268
    pkt->flags |= PKT_FLAG_KEY;
269

    
270
    pkt->size= 0;
271
    for(i=0; i<3; i++){
272
        if(size[i]){
273
            ret[i]= get_buffer(f[i], pkt->data + pkt->size, size[i]);
274
            if (!s->is_pipe)
275
                url_fclose(f[i]);
276
            if(ret[i]>0)
277
                pkt->size += ret[i];
278
        }
279
    }
280

    
281
    if (ret[0] <= 0 || ret[1]<0 || ret[2]<0) {
282
        av_free_packet(pkt);
283
        return AVERROR_IO; /* signal EOF */
284
    } else {
285
        s->img_count++;
286
        s->img_number++;
287
        return 0;
288
    }
289
}
290

    
291
static int img_read_close(AVFormatContext *s1)
292
{
293
    return 0;
294
}
295

    
296
#ifdef CONFIG_MUXERS
297
/******************************************************/
298
/* image output */
299

    
300
static int img_write_header(AVFormatContext *s)
301
{
302
    VideoData *img = s->priv_data;
303

    
304
    img->img_number = 1;
305
    pstrcpy(img->path, sizeof(img->path), s->filename);
306

    
307
    /* find format */
308
    if (s->oformat->flags & AVFMT_NOFILE)
309
        img->is_pipe = 0;
310
    else
311
        img->is_pipe = 1;
312

    
313
    return 0;
314
}
315

    
316
static int img_write_packet(AVFormatContext *s, AVPacket *pkt)
317
{
318
    VideoData *img = s->priv_data;
319
    ByteIOContext pb1[3], *pb[3]= {&pb1[0], &pb1[1], &pb1[2]};
320
    char filename[1024];
321
    AVCodecContext *codec= s->streams[ pkt->stream_index ]->codec;
322
    int i;
323

    
324
    if (!img->is_pipe) {
325
        if (av_get_frame_filename(filename, sizeof(filename),
326
                                  img->path, img->img_number) < 0 && img->img_number>1)
327
            return AVERROR_IO;
328
        for(i=0; i<3; i++){
329
            if (url_fopen(pb[i], filename, URL_WRONLY) < 0)
330
                return AVERROR_IO;
331

    
332
            if(codec->codec_id != CODEC_ID_RAWVIDEO)
333
                break;
334
            filename[ strlen(filename) - 1 ]= 'U' + i;
335
        }
336
    } else {
337
        pb[0] = &s->pb;
338
    }
339

    
340
    if(codec->codec_id == CODEC_ID_RAWVIDEO){
341
        int ysize = codec->width * codec->height;
342
        put_buffer(pb[0], pkt->data        , ysize);
343
        put_buffer(pb[1], pkt->data + ysize, (pkt->size - ysize)/2);
344
        put_buffer(pb[2], pkt->data + ysize +(pkt->size - ysize)/2, (pkt->size - ysize)/2);
345
        put_flush_packet(pb[1]);
346
        put_flush_packet(pb[2]);
347
        url_fclose(pb[1]);
348
        url_fclose(pb[2]);
349
    }else{
350
        put_buffer(pb[0], pkt->data, pkt->size);
351
    }
352
    put_flush_packet(pb[0]);
353
    if (!img->is_pipe) {
354
        url_fclose(pb[0]);
355
    }
356

    
357
    img->img_number++;
358
    return 0;
359
}
360

    
361
static int img_write_trailer(AVFormatContext *s)
362
{
363
    return 0;
364
}
365

    
366
#endif /* CONFIG_MUXERS */
367

    
368
/* input */
369
#ifdef CONFIG_IMAGE2_DEMUXER
370
AVInputFormat image2_demuxer = {
371
    "image2",
372
    "image2 sequence",
373
    sizeof(VideoData),
374
    image_probe,
375
    img_read_header,
376
    img_read_packet,
377
    img_read_close,
378
    NULL,
379
    NULL,
380
    AVFMT_NOFILE,
381
};
382
#endif
383
#ifdef CONFIG_IMAGE2PIPE_DEMUXER
384
AVInputFormat image2pipe_demuxer = {
385
    "image2pipe",
386
    "piped image2 sequence",
387
    sizeof(VideoData),
388
    NULL, /* no probe */
389
    img_read_header,
390
    img_read_packet,
391
    img_read_close,
392
    NULL,
393
};
394
#endif
395

    
396
/* output */
397
#ifdef CONFIG_IMAGE2_MUXER
398
AVOutputFormat image2_muxer = {
399
    "image2",
400
    "image2 sequence",
401
    "",
402
    "",
403
    sizeof(VideoData),
404
    CODEC_ID_NONE,
405
    CODEC_ID_MJPEG,
406
    img_write_header,
407
    img_write_packet,
408
    img_write_trailer,
409
    AVFMT_NOFILE,
410
};
411
#endif
412
#ifdef CONFIG_IMAGE2PIPE_MUXER
413
AVOutputFormat image2pipe_muxer = {
414
    "image2pipe",
415
    "piped image2 sequence",
416
    "",
417
    "",
418
    sizeof(VideoData),
419
    CODEC_ID_NONE,
420
    CODEC_ID_MJPEG,
421
    img_write_header,
422
    img_write_packet,
423
    img_write_trailer,
424
};
425
#endif