3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
4 * Copyright (c) 2004 Michael Niedermayer
6 * This library 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 of the License, or (at your option) any later version.
11 * This library 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.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
36 static const IdStrMap img_tags[] = {
37 { CODEC_ID_MJPEG , "jpeg"},
38 { CODEC_ID_MJPEG , "jpg"},
39 { CODEC_ID_LJPEG , "ljpg"},
40 // { CODEC_ID_PNG , "png"},
41 { CODEC_ID_MPEG1VIDEO, "mpg1-img"},
42 { CODEC_ID_MPEG2VIDEO, "mpg2-img"},
43 { CODEC_ID_MPEG4 , "mpg4-img"},
44 { CODEC_ID_FFV1 , "ffv1-img"},
48 static enum CodecID av_str2id(const IdStrMap *tags, const char *str)
50 str= strrchr(str, '.');
51 if(!str) return CODEC_ID_NONE;
56 for(i=0; toupper(tags->str[i]) == toupper(str[i]); i++){
57 if(tags->str[i]==0 && str[i]==0)
66 static const char *av_id2str(const IdStrMap *tags, enum CodecID id)
76 /* return -1 if no image found */
77 static int find_image_range(int *pfirst_index, int *plast_index,
81 int range, last_index, range1, first_index;
83 /* find the first image */
84 for(first_index = 0; first_index < 5; first_index++) {
85 if (get_frame_filename(buf, sizeof(buf), path, first_index) < 0){
96 /* find the last image */
97 last_index = first_index;
105 if (get_frame_filename(buf, sizeof(buf), path,
106 last_index + range1) < 0)
111 /* just in case... */
112 if (range >= (1 << 30))
115 /* we are sure than image last_index + range exists */
120 *pfirst_index = first_index;
121 *plast_index = last_index;
128 static int image_probe(AVProbeData *p)
130 if (filename_number_test(p->filename) >= 0 && av_str2id(img_tags, p->filename))
131 return AVPROBE_SCORE_MAX;
136 static int img_read_header(AVFormatContext *s1, AVFormatParameters *ap)
138 VideoData *s = s1->priv_data;
139 int first_index, last_index;
142 s1->ctx_flags |= AVFMTCTX_NOHEADER;
144 st = av_new_stream(s1, 0);
150 strcpy(s->path, s1->filename);
155 if (s1->iformat->flags & AVFMT_NOFILE)
162 if (!ap || !ap->frame_rate) {
163 st->codec.frame_rate = 25;
164 st->codec.frame_rate_base = 1;
166 st->codec.frame_rate = ap->frame_rate;
167 st->codec.frame_rate_base = ap->frame_rate_base;
171 if (find_image_range(&first_index, &last_index, s->path) < 0)
173 s->img_first = first_index;
174 s->img_last = last_index;
175 s->img_number = first_index;
176 /* compute duration */
178 st->duration = ((int64_t)AV_TIME_BASE *
179 (last_index - first_index + 1) *
180 st->codec.frame_rate_base) / st->codec.frame_rate;
183 st->codec.codec_type = CODEC_TYPE_VIDEO;
184 st->codec.codec_id = av_str2id(img_tags, s->path);
193 static int img_read_packet(AVFormatContext *s1, AVPacket *pkt)
195 VideoData *s = s1->priv_data;
198 ByteIOContext f1, *f;
201 /* loop over input */
202 /* if (loop_input && s->img_number > s->img_last) {
203 s->img_number = s->img_first;
205 if (get_frame_filename(filename, sizeof(filename),
206 s->path, s->img_number)<0 && s->img_number > 1)
209 if (url_fopen(f, filename, URL_RDONLY) < 0)
218 av_new_packet(pkt, 4096);
220 av_new_packet(pkt, url_filesize(url_fileno(f)));
222 pkt->stream_index = 0;
223 pkt->flags |= PKT_FLAG_KEY;
225 ret = get_buffer(f, pkt->data, pkt->size);
232 return AVERROR_IO; /* signal EOF */
240 static int img_read_close(AVFormatContext *s1)
245 /******************************************************/
248 static int img_write_header(AVFormatContext *s)
250 VideoData *img = s->priv_data;
253 strcpy(img->path, s->filename);
256 if (s->oformat->flags & AVFMT_NOFILE)
264 static int img_write_packet(AVFormatContext *s, AVPacket *pkt)
266 VideoData *img = s->priv_data;
267 ByteIOContext pb1, *pb;
271 if (get_frame_filename(filename, sizeof(filename),
272 img->path, img->img_number) < 0 && img->img_number>1)
275 if (url_fopen(pb, filename, URL_WRONLY) < 0)
281 put_buffer(pb, pkt->data, pkt->size);
282 put_flush_packet(pb);
291 static int img_write_trailer(AVFormatContext *s)
298 static AVInputFormat image2_iformat = {
311 static AVInputFormat image2pipe_iformat = {
313 "piped image2 sequence",
325 static AVOutputFormat image2_oformat = {
339 static AVOutputFormat image2pipe_oformat = {
341 "piped image2 sequence",
354 av_register_input_format(&image2_iformat);
355 av_register_output_format(&image2_oformat);
357 av_register_input_format(&image2pipe_iformat);
358 av_register_output_format(&image2pipe_oformat);