Mercurial > libavformat.hg

comparison pnm.c @ 20:3d52de18ecc3 libavformat

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added still image support
author bellard
date Sat, 11 Jan 2003 05:02:14 +0000
parents
children a58a8a53eb46
comparison
equal deleted inserted replaced
19:81e87c8de3dc 20:3d52de18ecc3
1 /*
2 * PNM image format
3 * Copyright (c) 2002, 2003 Fabrice Bellard.
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 */
19 #include "avformat.h"
20
21 static inline int pnm_space(int c)
22 {
23 return (c == ' ' || c == '\n' || c == '\r' || c == '\t');
24 }
25
26 static void pnm_get(ByteIOContext *f, char *str, int buf_size)
27 {
28 char *s;
29 int c;
30
31 /* skip spaces and comments */
32 for(;;) {
33 c = url_fgetc(f);
34 if (c == '#') {
35 do {
36 c = url_fgetc(f);
37 } while (c != '\n' && c != URL_EOF);
38 } else if (!pnm_space(c)) {
39 break;
40 }
41 }
42
43 s = str;
44 while (c != URL_EOF && !pnm_space(c)) {
45 if ((s - str) < buf_size - 1)
46 *s++ = c;
47 c = url_fgetc(f);
48 }
49 *s = '\0';
50 }
51
52 static int pnm_read1(ByteIOContext *f,
53 int (*alloc_cb)(void *opaque, AVImageInfo *info), void *opaque,
54 int allow_yuv)
55 {
56 int i, n, linesize, h;
57 char buf1[32];
58 unsigned char *ptr;
59 AVImageInfo info1, *info = &info1;
60 int ret;
61
62 pnm_get(f, buf1, sizeof(buf1));
63 if (!strcmp(buf1, "P4")) {
64 info->pix_fmt = PIX_FMT_MONOWHITE;
65 } else if (!strcmp(buf1, "P5")) {
66 if (allow_yuv)
67 info->pix_fmt = PIX_FMT_YUV420P;
68 else
69 info->pix_fmt = PIX_FMT_GRAY8;
70 } else if (!strcmp(buf1, "P6")) {
71 info->pix_fmt = PIX_FMT_RGB24;
72 } else {
73 return AVERROR_INVALIDDATA;
74 }
75 pnm_get(f, buf1, sizeof(buf1));
76 info->width = atoi(buf1);
77 if (info->width <= 0)
78 return AVERROR_INVALIDDATA;
79 pnm_get(f, buf1, sizeof(buf1));
80 info->height = atoi(buf1);
81 if (info->height <= 0)
82 return AVERROR_INVALIDDATA;
83 if (info->pix_fmt != PIX_FMT_MONOWHITE) {
84 pnm_get(f, buf1, sizeof(buf1));
85 }
86
87 /* more check if YUV420 */
88 if (info->pix_fmt == PIX_FMT_YUV420P) {
89 if ((info->width & 1) != 0)
90 return AVERROR_INVALIDDATA;
91 h = (info->height * 2);
92 if ((h % 3) != 0)
93 return AVERROR_INVALIDDATA;
94 h /= 3;
95 info->height = h;
96 }
97
98 ret = alloc_cb(opaque, info);
99 if (ret)
100 return ret;
101
102 switch(info->pix_fmt) {
103 default:
104 return AVERROR_INVALIDDATA;
105 case PIX_FMT_RGB24:
106 n = info->width * 3;
107 goto do_read;
108 case PIX_FMT_GRAY8:
109 n = info->width;
110 goto do_read;
111 case PIX_FMT_MONOWHITE:
112 n = (info->width + 7) >> 3;
113 do_read:
114 ptr = info->pict.data[0];
115 linesize = info->pict.linesize[0];
116 for(i = 0; i < info->height; i++) {
117 get_buffer(f, ptr, n);
118 ptr += linesize;
119 }
120 break;
121 case PIX_FMT_YUV420P:
122 {
123 unsigned char *ptr1, *ptr2;
124
125 n = info->width;
126 ptr = info->pict.data[0];
127 linesize = info->pict.linesize[0];
128 for(i = 0; i < info->height; i++) {
129 get_buffer(f, ptr, n);
130 ptr += linesize;
131 }
132 ptr1 = info->pict.data[1];
133 ptr2 = info->pict.data[2];
134 n >>= 1;
135 h = info->height >> 1;
136 for(i = 0; i < h; i++) {
137 get_buffer(f, ptr1, n);
138 get_buffer(f, ptr2, n);
139 ptr1 += info->pict.linesize[1];
140 ptr2 += info->pict.linesize[2];
141 }
142 }
143 break;
144 }
145 return 0;
146 }
147
148 static int pnm_read(ByteIOContext *f,
149 int (*alloc_cb)(void *opaque, AVImageInfo *info), void *opaque)
150 {
151 return pnm_read1(f, alloc_cb, opaque, 0);
152 }
153
154 static int pgmyuv_read(ByteIOContext *f,
155 int (*alloc_cb)(void *opaque, AVImageInfo *info), void *opaque)
156 {
157 return pnm_read1(f, alloc_cb, opaque, 1);
158 }
159
160 static int pnm_write(ByteIOContext *pb, AVImageInfo *info)
161 {
162 int i, h, h1, c, n, linesize;
163 char buf[100];
164 UINT8 *ptr, *ptr1, *ptr2;
165
166 h = info->height;
167 h1 = h;
168 switch(info->pix_fmt) {
169 case PIX_FMT_MONOWHITE:
170 c = '4';
171 n = (info->width + 7) >> 3;
172 break;
173 case PIX_FMT_GRAY8:
174 c = '5';
175 n = info->width;
176 break;
177 case PIX_FMT_RGB24:
178 c = '6';
179 n = info->width * 3;
180 break;
181 case PIX_FMT_YUV420P:
182 c = '5';
183 n = info->width;
184 h1 = (h * 3) / 2;
185 break;
186 default:
187 return AVERROR_INVALIDDATA;
188 }
189 snprintf(buf, sizeof(buf),
190 "P%c\n%d %d\n",
191 c, info->width, h1);
192 put_buffer(pb, buf, strlen(buf));
193 if (info->pix_fmt != PIX_FMT_MONOWHITE) {
194 snprintf(buf, sizeof(buf),
195 "%d\n", 255);
196 put_buffer(pb, buf, strlen(buf));
197 }
198
199 ptr = info->pict.data[0];
200 linesize = info->pict.linesize[0];
201 for(i=0;i<h;i++) {
202 put_buffer(pb, ptr, n);
203 ptr += linesize;
204 }
205
206 if (info->pix_fmt == PIX_FMT_YUV420P) {
207 h >>= 1;
208 n >>= 1;
209 ptr1 = info->pict.data[1];
210 ptr2 = info->pict.data[2];
211 for(i=0;i<h;i++) {
212 put_buffer(pb, ptr1, n);
213 put_buffer(pb, ptr2, n);
214 ptr1 += info->pict.linesize[1];
215 ptr2 += info->pict.linesize[2];
216 }
217 }
218 put_flush_packet(pb);
219 return 0;
220 }
221
222 static int pnm_probe(AVProbeData *pd)
223 {
224 const char *p = pd->buf;
225 if (pd->buf_size >= 8 &&
226 p[0] == 'P' &&
227 p[1] >= '4' && p[1] <= '6' &&
228 p[2] == '\n')
229 return AVPROBE_SCORE_MAX;
230 else
231 return 0;
232 }
233
234 static int pgmyuv_probe(AVProbeData *pd)
235 {
236 if (match_ext(pd->filename, "pgmyuv"))
237 return AVPROBE_SCORE_MAX;
238 else
239 return 0;
240 }
241
242 AVImageFormat pnm_image_format = {
243 "pnm",
244 NULL,
245 pnm_probe,
246 pnm_read,
247 0,
248 NULL,
249 };
250
251 AVImageFormat pbm_image_format = {
252 "pbm",
253 "pbm",
254 NULL,
255 NULL,
256 (1 << PIX_FMT_MONOWHITE),
257 pnm_write,
258 };
259
260 AVImageFormat pgm_image_format = {
261 "pgm",
262 "pgm",
263 NULL,
264 NULL,
265 (1 << PIX_FMT_GRAY8),
266 pnm_write,
267 };
268
269 AVImageFormat ppm_image_format = {
270 "ppm",
271 "ppm",
272 NULL,
273 NULL,
274 (1 << PIX_FMT_RGB24),
275 pnm_write,
276 };
277
278 AVImageFormat pgmyuv_image_format = {
279 "pgmyuv",
280 NULL,
281 pgmyuv_probe,
282 pgmyuv_read,
283 (1 << PIX_FMT_YUV420P),
284 pnm_write,
285 };