removed clang-format and clang_complete files from the repo
[dosdemo] / libs / imago / src / filetga.c
1 /*
2 libimago - a multi-format image file input/output library.
3 Copyright (C) 2010-2015 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 /* -- Targa (tga) module -- */
20
21 #include <string.h>
22 #include <stdlib.h>
23 #include "inttypes.h"
24 #include "imago2.h"
25 #include "ftmodule.h"
26
27
28 #if  defined(__i386__) || defined(__ia64__) || defined(WIN32) || \
29     (defined(__alpha__) || defined(__alpha)) || \
30      defined(__arm__) || \
31     (defined(__mips__) && defined(__MIPSEL__)) || \
32      defined(__SYMBIAN32__) || \
33      defined(__x86_64__) || \
34      defined(__LITTLE_ENDIAN__)
35 /* little endian */
36 #define read_int16_le(f)        read_int16(f)
37 #else
38 /* big endian */
39 #define read_int16_le(f)        read_int16_inv(f)
40 #endif  /* endian check */
41
42
43 enum {
44         IMG_NONE,
45         IMG_CMAP,
46         IMG_RGBA,
47         IMG_BW,
48
49         IMG_RLE_CMAP = 9,
50         IMG_RLE_RGBA,
51         IMG_RLE_BW
52 };
53
54 #define IS_RLE(x)       ((x) >= IMG_RLE_CMAP)
55 #define IS_RGBA(x)      ((x) == IMG_RGBA || (x) == IMG_RLE_RGBA)
56
57
58 struct tga_header {
59         uint8_t idlen;                  /* id field length */
60         uint8_t cmap_type;              /* color map type (0:no color map, 1:color map present) */
61         uint8_t img_type;               /* image type:
62                                                          * 0: no image data
63                                                          *      1: uncomp. color-mapped          9: RLE color-mapped
64                                                          *      2: uncomp. true color           10: RLE true color
65                                                          *      3: uncomp. black/white          11: RLE black/white */
66         uint16_t cmap_first;    /* color map first entry index */
67         uint16_t cmap_len;              /* color map length */
68         uint8_t cmap_entry_sz;  /* color map entry size */
69         uint16_t img_x;                 /* X-origin of the image */
70         uint16_t img_y;                 /* Y-origin of the image */
71         uint16_t img_width;             /* image width */
72         uint16_t img_height;    /* image height */
73         uint8_t img_bpp;                /* bits per pixel */
74         uint8_t img_desc;               /* descriptor:
75                                                          * bits 0 - 3: alpha or overlay bits
76                                                          * bits 5 & 4: origin (0 = bottom/left, 1 = top/right)
77                                                          * bits 7 & 6: data interleaving */
78 };
79
80 struct tga_footer {
81         uint32_t ext_off;               /* extension area offset */
82         uint32_t devdir_off;    /* developer directory offset */
83         char sig[18];                           /* signature with . and \0 */
84 };
85
86
87 static int check(struct img_io *io);
88 static int read(struct img_pixmap *img, struct img_io *io);
89 static int write(struct img_pixmap *img, struct img_io *io);
90 static int read_pixel(struct img_io *io, int rdalpha, uint32_t *pix);
91 static int16_t read_int16(struct img_io *io);
92 static int16_t read_int16_inv(struct img_io *io);
93
94 int img_register_tga(void)
95 {
96         static struct ftype_module mod = {".tga", check, read, write};
97         return img_register_module(&mod);
98 }
99
100
101 static int check(struct img_io *io)
102 {
103         struct tga_footer foot;
104         int res = -1;
105         long pos = io->seek(0, SEEK_CUR, io->uptr);
106         io->seek(-18, SEEK_END, io->uptr);
107
108         if(io->read(foot.sig, 17, io->uptr) < 17) {
109                 io->seek(pos, SEEK_SET, io->uptr);
110                 return -1;
111         }
112
113         if(memcmp(foot.sig, "TRUEVISION-XFILE.", 17) == 0) {
114                 res = 0;
115         }
116         io->seek(pos, SEEK_SET, io->uptr);
117         return res;
118 }
119
120 static int iofgetc(struct img_io *io)
121 {
122         char c;
123         return io->read(&c, 1, io->uptr) < 1 ? -1 : c;
124 }
125
126 static int read(struct img_pixmap *img, struct img_io *io)
127 {
128         struct tga_header hdr;
129         int x, y;
130         int i, c;
131         uint32_t ppixel = 0;
132         int rle_mode = 0, rle_pix_left = 0;
133         int rdalpha;
134
135         /* read header */
136         hdr.idlen = iofgetc(io);
137         hdr.cmap_type = iofgetc(io);
138         hdr.img_type = iofgetc(io);
139         hdr.cmap_first = read_int16_le(io);
140         hdr.cmap_len = read_int16_le(io);
141         hdr.cmap_entry_sz = iofgetc(io);
142         hdr.img_x = read_int16_le(io);
143         hdr.img_y = read_int16_le(io);
144         hdr.img_width = read_int16_le(io);
145         hdr.img_height = read_int16_le(io);
146         hdr.img_bpp = iofgetc(io);
147         if((c = iofgetc(io)) == -1) {
148                 return -1;
149         }
150         hdr.img_desc = c;
151
152         if(!IS_RGBA(hdr.img_type)) {
153                 fprintf(stderr, "only true color tga images supported\n");
154                 return -1;
155         }
156
157         io->seek(hdr.idlen, SEEK_CUR, io);      /* skip the image ID */
158
159         /* skip the color map if it exists */
160         if(hdr.cmap_type == 1) {
161                 io->seek(hdr.cmap_len * hdr.cmap_entry_sz / 8, SEEK_CUR, io);
162         }
163
164         x = hdr.img_width;
165         y = hdr.img_height;
166         rdalpha = hdr.img_desc & 0xf;
167
168         /* TODO make this IMG_FMT_RGB24 if there's no alpha channel */
169         if(img_set_pixels(img, x, y, IMG_FMT_RGBA32, 0) == -1) {
170                 return -1;
171         }
172
173         for(i=0; i<y; i++) {
174                 uint32_t *ptr;
175                 int j;
176
177                 ptr = (uint32_t*)img->pixels + ((hdr.img_desc & 0x20) ? i : y - (i + 1)) * x;
178
179                 for(j=0; j<x; j++) {
180                         /* if the image is raw, then just read the next pixel */
181                         if(!IS_RLE(hdr.img_type)) {
182                                 if(read_pixel(io, rdalpha, &ppixel) == -1) {
183                                         return -1;
184                                 }
185                         } else {
186                                 /* otherwise, for RLE... */
187
188                                 /* if we have pixels left in the packet ... */
189                                 if(rle_pix_left) {
190                                         /* if it's a raw packet, read the next pixel, otherwise keep the same */
191                                         if(!rle_mode) {
192                                                 if(read_pixel(io, rdalpha, &ppixel) == -1) {
193                                                         return -1;
194                                                 }
195                                         }
196                                         --rle_pix_left;
197                                 } else {
198                                         /* read RLE packet header */
199                                         unsigned char phdr = iofgetc(io);
200                                         rle_mode = (phdr & 128);                /* last bit shows the mode for this packet (1: rle, 0: raw) */
201                                         rle_pix_left = (phdr & ~128);   /* the rest gives the count of pixels minus one (we also read one here, so no +1) */
202                                         /* and read the first pixel of the packet */
203                                         if(read_pixel(io, rdalpha, &ppixel) == -1) {
204                                                 return -1;
205                                         }
206                                 }
207                         }
208
209                         *ptr++ = ppixel;
210                 }
211         }
212
213         return 0;
214 }
215
216 static int write(struct img_pixmap *img, struct img_io *io)
217 {
218         return -1;      /* TODO */
219 }
220
221 #define PACK_COLOR32(r,g,b,a) \
222         ((((a) & 0xff) << 24) | \
223          (((r) & 0xff) << 0) | \
224          (((g) & 0xff) << 8) | \
225          (((b) & 0xff) << 16))
226
227 static int read_pixel(struct img_io *io, int rdalpha, uint32_t *pix)
228 {
229         int r, g, b, a;
230         b = iofgetc(io);
231         g = iofgetc(io);
232         r = iofgetc(io);
233         a = rdalpha ? iofgetc(io) : 0xff;
234         *pix = PACK_COLOR32(r, g, b, a);
235         return a == -1 || r == -1 ? -1 : 0;
236 }
237
238 static int16_t read_int16(struct img_io *io)
239 {
240         int16_t v;
241         io->read(&v, 2, io);
242         return v;
243 }
244
245 static int16_t read_int16_inv(struct img_io *io)
246 {
247         int16_t v;
248         io->read(&v, 2, io);
249         return ((v >> 8) & 0xff) | (v << 8);
250 }