infcubes crashes
[dosdemo] / libs / imago / src / fileppm.c
1 /*
2 libimago - a multi-format image file input/output library.
3 Copyright (C) 2010 John Tsiombikas <nuclear@member.fsf.org>
4
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU Lesser General Public License as published
7 by the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 This program 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
13 GNU Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public License
16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 */
18
19 /* -- Portable Pixmap (PPM) module -- */
20
21 #include <string.h>
22 #include "imago2.h"
23 #include "ftmodule.h"
24
25 static int check(struct img_io *io);
26 static int read(struct img_pixmap *img, struct img_io *io);
27 static int write(struct img_pixmap *img, struct img_io *io);
28
29 int img_register_ppm(void)
30 {
31         static struct ftype_module mod = {".ppm", check, read, write};
32         return img_register_module(&mod);
33 }
34
35
36 static int check(struct img_io *io)
37 {
38         char id[2];
39         int res = -1;
40         long pos = io->seek(0, SEEK_CUR, io->uptr);
41
42         if(io->read(id, 2, io->uptr) < 2) {
43                 io->seek(pos, SEEK_SET, io->uptr);
44                 return -1;
45         }
46
47         if(id[0] == 'P' && (id[1] == '6' || id[1] == '3')) {
48                 res = 0;
49         }
50         io->seek(pos, SEEK_SET, io->uptr);
51         return res;
52 }
53
54 static int iofgetc(struct img_io *io)
55 {
56         char c;
57         return io->read(&c, 1, io->uptr) < 1 ? -1 : c;
58 }
59
60 static char *iofgets(char *buf, int size, struct img_io *io)
61 {
62         int c;
63         char *ptr = buf;
64
65         while(--size > 0 && (c = iofgetc(io)) != -1) {
66                 *ptr++ = c;
67                 if(c == '\n') break;
68         }
69         *ptr = 0;
70
71         return ptr == buf ? 0 : buf;
72 }
73
74 /* TODO: implement P3 reading */
75 static int read(struct img_pixmap *img, struct img_io *io)
76 {
77         char buf[256];
78         int xsz, ysz, maxval, got_hdrlines = 1;
79
80         if(!iofgets(buf, sizeof buf, io)) {
81                 return -1;
82         }
83         if(!(buf[0] == 'P' && (buf[1] == '6' || buf[1] == '3'))) {
84                 return -1;
85         }
86
87         while(got_hdrlines < 3 && iofgets(buf, sizeof buf, io)) {
88                 if(buf[0] == '#') continue;
89
90                 switch(got_hdrlines) {
91                 case 1:
92                         if(sscanf(buf, "%d %d\n", &xsz, &ysz) < 2) {
93                                 return -1;
94                         }
95                         break;
96
97                 case 2:
98                         if(sscanf(buf, "%d\n", &maxval) < 1) {
99                                 return -1;
100                         }
101                 default:
102                         break;
103                 }
104                 got_hdrlines++;
105         }
106
107         if(xsz < 1 || ysz < 1 || maxval != 255) {
108                 return -1;
109         }
110
111         if(img_set_pixels(img, xsz, ysz, IMG_FMT_RGB24, 0) == -1) {
112                 return -1;
113         }
114
115         if(io->read(img->pixels, xsz * ysz * 3, io->uptr) < (unsigned int)(xsz * ysz * 3)) {
116                 return -1;
117         }
118         return 0;
119 }
120
121 static int write(struct img_pixmap *img, struct img_io *io)
122 {
123         int sz;
124         char buf[256];
125         struct img_pixmap tmpimg;
126
127         img_init(&tmpimg);
128
129         if(img->fmt != IMG_FMT_RGB24) {
130                 if(img_copy(&tmpimg, img) == -1) {
131                         return -1;
132                 }
133                 if(img_convert(&tmpimg, IMG_FMT_RGB24) == -1) {
134                         return -1;
135                 }
136                 img = &tmpimg;
137         }
138
139         sprintf(buf, "P6\n#written by libimago2\n%d %d\n255\n", img->width, img->height);
140         if(io->write(buf, strlen(buf), io->uptr) < strlen(buf)) {
141                 img_destroy(&tmpimg);
142                 return -1;
143         }
144
145         sz = img->width * img->height * 3;
146         if(io->write(img->pixels, sz, io->uptr) < (unsigned int)sz) {
147                 img_destroy(&tmpimg);
148                 return -1;
149         }
150
151         img_destroy(&tmpimg);
152         return 0;
153 }