annotate img2.c @ 497:d95e74ef39e0 libavformat

image2 / image API cleanup phase-1 with this its possible to encode&decode any video codec to individual (1 file per frame) files
author michael
date Thu, 15 Jul 2004 18:32:54 +0000
parents
children 7c8687cb7f27
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
497
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
1 /*
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
2 * Image format
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
4 * Copyright (c) 2004 Michael Niedermayer
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
5 *
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
6 * This library is free software; you can redistribute it and/or
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
7 * modify it under the terms of the GNU Lesser General Public
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
8 * License as published by the Free Software Foundation; either
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
9 * version 2 of the License, or (at your option) any later version.
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
10 *
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
11 * This library is distributed in the hope that it will be useful,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
14 * Lesser General Public License for more details.
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
15 *
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
16 * You should have received a copy of the GNU Lesser General Public
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
17 * License along with this library; if not, write to the Free Software
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
19 */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
20 #include "avformat.h"
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
21
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
22 typedef struct {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
23 int img_first;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
24 int img_last;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
25 int img_number;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
26 int img_count;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
27 int is_pipe;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
28 char path[1024];
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
29 } VideoData;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
30
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
31 typedef struct {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
32 enum CodecID id;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
33 const char *str;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
34 } IdStrMap;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
35
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
36 static const IdStrMap img_tags[] = {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
37 { CODEC_ID_MJPEG , "jpeg"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
38 { CODEC_ID_MJPEG , "jpg"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
39 { CODEC_ID_LJPEG , "ljpg"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
40 { CODEC_ID_MPEG1VIDEO, "mpg1-img"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
41 { CODEC_ID_MPEG2VIDEO, "mpg2-img"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
42 { CODEC_ID_MPEG4 , "mpg4-img"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
43 { CODEC_ID_FFV1 , "ffv1-img"},
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
44 {0, NULL}
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
45 };
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
46
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
47 static enum CodecID av_str2id(const IdStrMap *tags, const char *str)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
48 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
49 while(*str && *str!='.') str++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
50 if(*str) str++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
51
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
52 while (tags->id) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
53 int i;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
54 for(i=0; toupper(tags->str[i]) == toupper(str[i]); i++){
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
55 if(tags->str[i]==0 && str[i]==0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
56 return tags->id;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
57 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
58
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
59 tags++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
60 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
61 return CODEC_ID_NONE;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
62 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
63
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
64 static const char *av_id2str(const IdStrMap *tags, enum CodecID id)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
65 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
66 while (tags->id) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
67 if(tags->id == id)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
68 return tags->str;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
69 tags++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
70 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
71 return NULL;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
72 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
73
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
74 /* return -1 if no image found */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
75 static int find_image_range(int *pfirst_index, int *plast_index,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
76 const char *path)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
77 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
78 char buf[1024];
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
79 int range, last_index, range1, first_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
80
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
81 /* find the first image */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
82 for(first_index = 0; first_index < 5; first_index++) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
83 if (get_frame_filename(buf, sizeof(buf), path, first_index) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
84 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
85 if (url_exist(buf))
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
86 break;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
87 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
88 if (first_index == 5)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
89 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
90
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
91 /* find the last image */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
92 last_index = first_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
93 for(;;) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
94 range = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
95 for(;;) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
96 if (!range)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
97 range1 = 1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
98 else
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
99 range1 = 2 * range;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
100 if (get_frame_filename(buf, sizeof(buf), path,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
101 last_index + range1) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
102 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
103 if (!url_exist(buf))
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
104 break;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
105 range = range1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
106 /* just in case... */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
107 if (range >= (1 << 30))
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
108 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
109 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
110 /* we are sure than image last_index + range exists */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
111 if (!range)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
112 break;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
113 last_index += range;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
114 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
115 *pfirst_index = first_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
116 *plast_index = last_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
117 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
118 fail:
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
119 return -1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
120 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
121
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
122
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
123 static int image_probe(AVProbeData *p)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
124 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
125 if (filename_number_test(p->filename) >= 0 && av_str2id(img_tags, p->filename))
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
126 return AVPROBE_SCORE_MAX;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
127 else
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
128 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
129 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
130
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
131 static int img_read_header(AVFormatContext *s1, AVFormatParameters *ap)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
132 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
133 VideoData *s = s1->priv_data;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
134 int first_index, last_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
135 char buf[1024];
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
136 ByteIOContext pb1, *f = &pb1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
137 AVStream *st;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
138
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
139 s1->ctx_flags |= AVFMTCTX_NOHEADER;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
140
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
141 st = av_new_stream(s1, 0);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
142 if (!st) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
143 av_free(s);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
144 return -ENOMEM;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
145 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
146
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
147 strcpy(s->path, s1->filename);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
148 s->img_number = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
149 s->img_count = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
150
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
151 /* find format */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
152 if (s1->iformat->flags & AVFMT_NOFILE)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
153 s->is_pipe = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
154 else
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
155 s->is_pipe = 1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
156
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
157 if (!ap || !ap->frame_rate) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
158 st->codec.frame_rate = 25;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
159 st->codec.frame_rate_base = 1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
160 } else {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
161 st->codec.frame_rate = ap->frame_rate;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
162 st->codec.frame_rate_base = ap->frame_rate_base;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
163 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
164
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
165 if (!s->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
166 if (find_image_range(&first_index, &last_index, s->path) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
167 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
168 s->img_first = first_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
169 s->img_last = last_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
170 s->img_number = first_index;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
171 /* compute duration */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
172 st->start_time = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
173 st->duration = ((int64_t)AV_TIME_BASE *
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
174 (last_index - first_index + 1) *
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
175 st->codec.frame_rate_base) / st->codec.frame_rate;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
176 if (get_frame_filename(buf, sizeof(buf), s->path, s->img_number) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
177 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
178 if (url_fopen(f, buf, URL_RDONLY) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
179 goto fail;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
180 } else {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
181 f = &s1->pb;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
182 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
183
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
184 if (!s->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
185 url_fclose(f);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
186 } else {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
187 url_fseek(f, 0, SEEK_SET);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
188 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
189
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
190 st->codec.codec_type = CODEC_TYPE_VIDEO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
191 st->codec.codec_id = av_str2id(img_tags, s->path);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
192
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
193 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
194
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
195 if (!s->is_pipe)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
196 url_fclose(f);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
197 fail:
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
198 av_free(s);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
199 return AVERROR_IO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
200 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
201
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
202 static int img_read_packet(AVFormatContext *s1, AVPacket *pkt)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
203 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
204 VideoData *s = s1->priv_data;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
205 char filename[1024];
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
206 int ret;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
207 ByteIOContext f1, *f;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
208
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
209 if (!s->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
210 /* loop over input */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
211 /* if (loop_input && s->img_number > s->img_last) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
212 s->img_number = s->img_first;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
213 }*/
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
214 if (get_frame_filename(filename, sizeof(filename),
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
215 s->path, s->img_number) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
216 return AVERROR_IO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
217 f = &f1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
218 if (url_fopen(f, filename, URL_RDONLY) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
219 return AVERROR_IO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
220 } else {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
221 f = &s1->pb;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
222 if (url_feof(f))
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
223 return AVERROR_IO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
224 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
225
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
226 if (s->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
227 av_new_packet(pkt, 4096);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
228 }else{
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
229 av_new_packet(pkt, url_filesize(url_fileno(f)));
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
230 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
231 pkt->stream_index = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
232 pkt->flags |= PKT_FLAG_KEY;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
233
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
234 ret = get_buffer(f, pkt->data, pkt->size);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
235 if (!s->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
236 url_fclose(f);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
237 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
238
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
239 if (ret <= 0) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
240 av_free_packet(pkt);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
241 return AVERROR_IO; /* signal EOF */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
242 } else {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
243 s->img_count++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
244 s->img_number++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
245 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
246 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
247 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
248
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
249 static int img_read_close(AVFormatContext *s1)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
250 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
251 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
252 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
253
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
254 /******************************************************/
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
255 /* image output */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
256
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
257 static int img_write_header(AVFormatContext *s)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
258 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
259 VideoData *img = s->priv_data;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
260
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
261 img->img_number = 1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
262 strcpy(img->path, s->filename);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
263
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
264 /* find format */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
265 if (s->oformat->flags & AVFMT_NOFILE)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
266 img->is_pipe = 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
267 else
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
268 img->is_pipe = 1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
269
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
270 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
271 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
272
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
273 static int img_write_packet(AVFormatContext *s, AVPacket *pkt)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
274 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
275 VideoData *img = s->priv_data;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
276 ByteIOContext pb1, *pb;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
277 char filename[1024];
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
278
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
279 if (!img->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
280 if (get_frame_filename(filename, sizeof(filename),
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
281 img->path, img->img_number) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
282 return AVERROR_IO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
283 pb = &pb1;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
284 if (url_fopen(pb, filename, URL_WRONLY) < 0)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
285 return AVERROR_IO;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
286 } else {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
287 pb = &s->pb;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
288 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
289
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
290 put_buffer(pb, pkt->data, pkt->size);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
291 put_flush_packet(pb);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
292 if (!img->is_pipe) {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
293 url_fclose(pb);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
294 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
295
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
296 img->img_number++;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
297 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
298 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
299
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
300 static int img_write_trailer(AVFormatContext *s)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
301 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
302 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
303 }
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
304
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
305 /* input */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
306
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
307 static AVInputFormat image2_iformat = {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
308 "image2",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
309 "image2 sequence",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
310 sizeof(VideoData),
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
311 image_probe,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
312 img_read_header,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
313 img_read_packet,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
314 img_read_close,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
315 NULL,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
316 NULL,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
317 AVFMT_NOFILE | AVFMT_NEEDNUMBER,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
318 };
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
319
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
320 static AVInputFormat image2pipe_iformat = {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
321 "image2pipe",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
322 "piped image2 sequence",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
323 sizeof(VideoData),
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
324 NULL, /* no probe */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
325 img_read_header,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
326 img_read_packet,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
327 img_read_close,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
328 NULL,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
329 };
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
330
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
331
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
332 /* output */
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
333
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
334 static AVOutputFormat image2_oformat = {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
335 "image2",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
336 "image2 sequence",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
337 "",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
338 "",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
339 sizeof(VideoData),
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
340 CODEC_ID_NONE,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
341 CODEC_ID_MJPEG,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
342 img_write_header,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
343 img_write_packet,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
344 img_write_trailer,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
345 AVFMT_NOFILE | AVFMT_NEEDNUMBER,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
346 };
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
347
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
348 static AVOutputFormat image2pipe_oformat = {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
349 "image2pipe",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
350 "piped image2 sequence",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
351 "",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
352 "",
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
353 sizeof(VideoData),
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
354 CODEC_ID_NONE,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
355 CODEC_ID_MJPEG,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
356 img_write_header,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
357 img_write_packet,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
358 img_write_trailer,
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
359 };
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
360
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
361 int img2_init(void)
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
362 {
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
363 av_register_input_format(&image2_iformat);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
364 av_register_output_format(&image2_oformat);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
365
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
366 av_register_input_format(&image2pipe_iformat);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
367 av_register_output_format(&image2pipe_oformat);
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
368
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
369 return 0;
d95e74ef39e0 image2 / image API cleanup phase-1
michael
parents:
diff changeset
370 }