Reformatage : indent -kr -i2 -ci2 $(find . -name '*.c') $(find . -name '*.h')
[openjpeg.git] / codec / j2k_to_image.c
1 /* Copyright (c) 2001 David Janssens
2  * Copyright (c) 2002-2003 Yannick Verschueren
3  * Copyright (c) 2002-2003 Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium
4  * 
5  * All rights reserved. 
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
16  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
19  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25  * POSSIBILITY OF SUCH DAMAGE.
26  */
27
28
29
30 #include <openjpeg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 //#include <unistd.h>
35
36 int ceildiv(int a, int b)
37 {
38   return (a + b - 1) / b;
39 }
40
41 int main(int argc, char **argv)
42 {
43   FILE *f;
44   char *src, *src_name;
45   char *dest, S1, S2, S3;
46   int len;
47   j2k_image_t *img;
48   j2k_cp_t *cp;
49   j2k_option_t option;
50   int w, wr, wrr, h, hr, hrr, max;
51   int i, image_type = -1, compno, pad;
52   int adjust;
53
54   if (argc < 3) {
55     fprintf(stderr,
56             "usage: %s j2k-file image-file -reduce n (<- optional)\n",
57             argv[0]);
58     return 1;
59   }
60
61   f = fopen(argv[1], "rb");
62   if (!f) {
63     fprintf(stderr, "failed to open %s for reading\n", argv[1]);
64     return 1;
65   }
66
67   dest = argv[2];
68
69   option.reduce_on = 0;
70   option.reduce_value = 0;
71
72   /* OPTION REDUCE IS ACTIVE */
73   if (argc == 5) {
74     if (strcmp(argv[3], "-reduce")) {
75       fprintf(stderr,
76               "usage: options " "-reduce n"
77               " where n is the factor of reduction [%s]\n", argv[3]);
78       return 1;
79     }
80     option.reduce_on = 1;
81     sscanf(argv[4], "%d", &option.reduce_value);
82   }
83
84   while (*dest) {
85     dest++;
86   }
87   dest--;
88   S3 = *dest;
89   dest--;
90   S2 = *dest;
91   dest--;
92   S1 = *dest;
93
94   if ((S1 == 'p' && S2 == 'g' && S3 == 'x')
95       || (S1 == 'P' && S2 == 'G' && S3 == 'X')) {
96     image_type = 0;
97   }
98
99   if ((S1 == 'p' && S2 == 'n' && S3 == 'm')
100       || (S1 == 'P' && S2 == 'N' && S3 == 'M') || (S1 == 'p' && S2 == 'g'
101                                                    && S3 == 'm')
102       || (S1 == 'P' && S2 == 'G' && S3 == 'M') || (S1 == 'P' && S2 == 'P'
103                                                    && S3 == 'M')
104       || (S1 == 'p' && S2 == 'p' && S3 == 'm')) {
105     image_type = 1;
106   }
107
108   if ((S1 == 'b' && S2 == 'm' && S3 == 'p')
109       || (S1 == 'B' && S2 == 'M' && S3 == 'P')) {
110     image_type = 2;
111   }
112
113   if (image_type == -1) {
114     fprintf(stderr,
115             "!! Unrecognized format for infile : %c%c%c [accept only *.pnm, *.pgm, *.ppm, *.pgx or *.bmp] !!\n\n",
116             S1, S2, S3);
117     return 1;
118   }
119
120   fseek(f, 0, SEEK_END);
121   len = ftell(f);
122   fseek(f, 0, SEEK_SET);
123   src = (char *) malloc(len);
124   fread(src, 1, len, f);
125   fclose(f);
126
127   src_name = argv[1];
128   while (*src_name) {
129     src_name++;
130   }
131   src_name--;
132   S3 = *src_name;
133   src_name--;
134   S2 = *src_name;
135   src_name--;
136   S1 = *src_name;
137
138   if (S1 == 'j' && S2 == '2' && S3 == 'k') {
139     if (!j2k_decode(src, len, &img, &cp, option)) {
140       fprintf(stderr, "j2k_to_image: failed to decode image!\n");
141       return 1;
142     }
143   } else {
144     if (S1 == 'j' && S2 == 'p' && S3 == 't') {
145       if (!j2k_decode_jpt_stream(src, len, &img, &cp)) {
146         fprintf(stderr, "j2k_to_image: failed to decode image!\n");
147         return 1;
148       }
149     } else {
150       fprintf(stderr,
151               "j2k_to_image : Unknown format image *.%c%c%c [only *.j2k or *.jpt]!! \n",
152               S1, S2, S3);
153       return 1;
154     }
155   }
156
157   free(src);
158   /* ------------------  CREATE OUT IMAGE WITH THE RIGHT FORMAT ----------------------- */
159
160   /* ---------------------------- / */
161   /* /                            / */
162   /* /  FORMAT : PNM, PGM or PPM  / */
163   /* /                            / */
164   /* ---------------------------- / */
165
166   switch (image_type) {
167   case 1:                       /* PNM PGM PPM */
168     if (img->numcomps == 3 && img->comps[0].dx == img->comps[1].dx
169         && img->comps[1].dx == img->comps[2].dx
170         && img->comps[0].dy == img->comps[1].dy
171         && img->comps[1].dy == img->comps[2].dy
172         && img->comps[0].prec == img->comps[1].prec
173         && img->comps[1].prec == img->comps[2].prec) {
174       f = fopen(argv[2], "wb");
175       w = ceildiv(img->x1 - img->x0, img->comps[0].dx);
176       // wr = ceildiv(int_ceildivpow2(img->x1 - img->x0,img->factor),img->comps[0].dx);
177       wr = img->comps[0].w;
178       wrr = int_ceildivpow2(img->comps[0].w, img->comps[0].factor);
179
180       h = ceildiv(img->y1 - img->y0, img->comps[0].dy);
181       // hr = ceildiv(int_ceildivpow2(img->y1 - img->y0,img->factor), img->comps[0].dy);
182       hr = img->comps[0].h;
183       hrr = int_ceildivpow2(img->comps[0].h, img->comps[0].factor);
184
185       max = img->comps[0].prec > 8 ? 255 : (1 << img->comps[0].prec) - 1;
186
187       img->comps[0].x0 =
188         int_ceildivpow2(img->comps[0].x0 -
189                         int_ceildiv(img->x0, img->comps[0].dx),
190                         img->comps[0].factor);
191       img->comps[0].y0 =
192         int_ceildivpow2(img->comps[0].y0 -
193                         int_ceildiv(img->y0, img->comps[0].dy),
194                         img->comps[0].factor);
195
196
197       fprintf(f, "P6\n# %d %d %d %d %d\n%d %d\n%d\n",
198               cp->tcps[cp->tileno[0]].tccps[0].numresolutions, w, h,
199               img->comps[0].x0, img->comps[0].y0, wrr, hrr, max);
200       adjust = img->comps[0].prec > 8 ? img->comps[0].prec - 8 : 0;
201       for (i = 0; i < wrr * hrr; i++) {
202         char r, g, b;
203         r = img->comps[0].data[i / wrr * wr + i % wrr];
204         r += (img->comps[0].sgnd ? 1 << (img->comps[0].prec - 1) : 0);
205         r = r >> adjust;
206
207         g = img->comps[1].data[i / wrr * wr + i % wrr];
208         g += (img->comps[1].sgnd ? 1 << (img->comps[1].prec - 1) : 0);
209         g = g >> adjust;
210
211         b = img->comps[2].data[i / wrr * wr + i % wrr];
212         b += (img->comps[2].sgnd ? 1 << (img->comps[2].prec - 1) : 0);
213         b = b >> adjust;
214
215         fprintf(f, "%c%c%c", r, g, b);
216       }
217       fclose(f);
218     } else {
219       for (compno = 0; compno < img->numcomps; compno++) {
220         char name[256];
221         if (img->numcomps > 1) {
222           sprintf(name, "%d.%s", compno, argv[2]);
223         } else {
224           sprintf(name, "%s", argv[2]);
225         }
226         f = fopen(name, "wb");
227         w = ceildiv(img->x1 - img->x0, img->comps[compno].dx);
228         // wr = ceildiv(int_ceildivpow2(img->x1 - img->x0,img->factor),img->comps[compno].dx);
229         wr = img->comps[compno].w;
230         wrr =
231           int_ceildivpow2(img->comps[compno].w, img->comps[compno].factor);
232
233         h = ceildiv(img->y1 - img->y0, img->comps[compno].dy);
234         // hr = ceildiv(int_ceildivpow2(img->y1 - img->y0,img->factor), img->comps[compno].dy);
235         hr = img->comps[compno].h;
236         hrr =
237           int_ceildivpow2(img->comps[compno].h, img->comps[compno].factor);
238
239         max =
240           img->comps[compno].prec >
241           8 ? 255 : (1 << img->comps[compno].prec) - 1;
242
243         img->comps[compno].x0 =
244           int_ceildivpow2(img->comps[compno].x0 -
245                           int_ceildiv(img->x0,
246                                       img->comps[compno].dx),
247                           img->comps[compno].factor);
248         img->comps[compno].y0 =
249           int_ceildivpow2(img->comps[compno].y0 -
250                           int_ceildiv(img->y0,
251                                       img->comps[compno].dy),
252                           img->comps[compno].factor);
253
254         fprintf(f, "P5\n# %d %d %d %d %d\n%d %d\n%d\n",
255                 cp->tcps[cp->tileno[0]].tccps[compno].
256                 numresolutions, w, h, img->comps[compno].x0,
257                 img->comps[compno].y0, wrr, hrr, max);
258         adjust =
259           img->comps[compno].prec > 8 ? img->comps[compno].prec - 8 : 0;
260         for (i = 0; i < wrr * hrr; i++) {
261           char l;
262           l = img->comps[compno].data[i / wrr * wr + i % wrr];
263           l += (img->comps[compno].
264                 sgnd ? 1 << (img->comps[compno].prec - 1) : 0);
265           l = l >> adjust;
266           fprintf(f, "%c", l);
267         }
268         fclose(f);
269       }
270     }
271     break;
272
273     /* ------------------------ / */
274     /* /                        / */
275     /* /     FORMAT : PGX       / */
276     /* /                        / */
277     /* /----------------------- / */
278   case 0:                       /* PGX */
279     for (compno = 0; compno < img->numcomps; compno++) {
280       j2k_comp_t *comp = &img->comps[compno];
281       char name[256];
282       if (img->numcomps > 1)
283         sprintf(name, "%d_%s", compno, argv[2]);
284       else
285         sprintf(name, "%s", argv[2]);
286
287       f = fopen(name, "wb");
288       // w = ceildiv(img->x1 - img->x0, comp->dx);
289       // wr = ceildiv(int_ceildivpow2(img->x1 - img->x0,img->factor), comp->dx);
290       w = img->comps[compno].w;
291       wr = int_ceildivpow2(img->comps[compno].w,
292                            img->comps[compno].factor);
293
294       // h = ceildiv(img->y1 - img->y0, comp->dy);
295       // hr = ceildiv(int_ceildivpow2(img->y1 - img->y0,img->factor), comp->dy);
296       h = img->comps[compno].h;
297       hr = int_ceildivpow2(img->comps[compno].h,
298                            img->comps[compno].factor);
299
300       fprintf(f, "PG LM %c %d %d %d\n", comp->sgnd ? '-' : '+',
301               comp->prec, wr, hr);
302       for (i = 0; i < wr * hr; i++) {
303         int v = img->comps[compno].data[i / wr * w + i % wr];
304         if (comp->prec <= 8) {
305           char c = (char) v;
306           fwrite(&c, 1, 1, f);
307         } else if (comp->prec <= 16) {
308           short s = (short) v;
309           fwrite(&s, 2, 1, f);
310         } else {
311           fwrite(&v, 4, 1, f);
312         }
313       }
314       fclose(f);
315     }
316     break;
317
318     /* ------------------------ / */
319     /* /                        / */
320     /* /     FORMAT : BMP       / */
321     /* /                        / */
322     /* /----------------------- / */
323
324   case 2:                       /* BMP */
325     if (img->numcomps == 3 && img->comps[0].dx == img->comps[1].dx
326         && img->comps[1].dx == img->comps[2].dx
327         && img->comps[0].dy == img->comps[1].dy
328         && img->comps[1].dy == img->comps[2].dy
329         && img->comps[0].prec == img->comps[1].prec
330         && img->comps[1].prec == img->comps[2].prec) {
331       /* -->> -->> -->> -->>
332
333          24 bits color
334
335          <<-- <<-- <<-- <<-- */
336
337       f = fopen(argv[2], "wb");
338       // w = ceildiv(img->x1 - img->x0, img->comps[0].dx);
339       // wr = ceildiv(int_ceildivpow2(img->x1 - img->x0,img->factor), img->comps[0].dx);
340       w = img->comps[0].w;
341       wr = int_ceildivpow2(img->comps[0].w, img->comps[0].factor);
342
343       // h = ceildiv(img->y1 - img->y0, img->comps[0].dy);
344       // hr = ceildiv(int_ceildivpow2(img->y1 - img->y0,img->factor), img->comps[0].dy);
345       h = img->comps[0].h;
346       hr = int_ceildivpow2(img->comps[0].h, img->comps[0].factor);
347
348       fprintf(f, "BM");
349
350       /* FILE HEADER */
351       /* ------------- */
352       fprintf(f, "%c%c%c%c",
353               (unsigned char) (hr * wr * 3 + 3 * hr * (wr % 2) +
354                                54) & 0xff,
355               (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2) + 54)
356                                >> 8) & 0xff,
357               (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2) + 54)
358                                >> 16) & 0xff,
359               (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2) + 54)
360                                >> 24) & 0xff);
361       fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
362               ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
363       fprintf(f, "%c%c%c%c", (54) & 0xff, ((54) >> 8) & 0xff,
364               ((54) >> 16) & 0xff, ((54) >> 24) & 0xff);
365
366       /* INFO HEADER   */
367       /* ------------- */
368       fprintf(f, "%c%c%c%c", (40) & 0xff, ((40) >> 8) & 0xff,
369               ((40) >> 16) & 0xff, ((40) >> 24) & 0xff);
370       fprintf(f, "%c%c%c%c", (unsigned char) ((wr) & 0xff),
371               (unsigned char) ((wr) >> 8) & 0xff,
372               (unsigned char) ((wr) >> 16) & 0xff,
373               (unsigned char) ((wr) >> 24) & 0xff);
374       fprintf(f, "%c%c%c%c", (unsigned char) ((hr) & 0xff),
375               (unsigned char) ((hr) >> 8) & 0xff,
376               (unsigned char) ((hr) >> 16) & 0xff,
377               (unsigned char) ((hr) >> 24) & 0xff);
378       fprintf(f, "%c%c", (1) & 0xff, ((1) >> 8) & 0xff);
379       fprintf(f, "%c%c", (24) & 0xff, ((24) >> 8) & 0xff);
380       fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
381               ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
382       fprintf(f, "%c%c%c%c",
383               (unsigned char) (3 * hr * wr +
384                                3 * hr * (wr % 2)) & 0xff,
385               (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2)) >>
386                                8) & 0xff,
387               (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2)) >>
388                                16) & 0xff,
389               (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2)) >>
390                                24) & 0xff);
391       fprintf(f, "%c%c%c%c", (7834) & 0xff, ((7834) >> 8) & 0xff,
392               ((7834) >> 16) & 0xff, ((7834) >> 24) & 0xff);
393       fprintf(f, "%c%c%c%c", (7834) & 0xff, ((7834) >> 8) & 0xff,
394               ((7834) >> 16) & 0xff, ((7834) >> 24) & 0xff);
395       fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
396               ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
397       fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
398               ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
399
400       for (i = 0; i < wr * hr; i++) {
401         unsigned char R, G, B;
402         /* a modifier */
403         // R = img->comps[0].data[w * h - ((i) / (w) + 1) * w + (i) % (w)];
404         R = img->comps[0].data[w * hr - ((i) / (wr) + 1) * w + (i) % (wr)];
405         // G = img->comps[1].data[w * h - ((i) / (w) + 1) * w + (i) % (w)];
406         G = img->comps[1].data[w * hr - ((i) / (wr) + 1) * w + (i) % (wr)];
407         // B = img->comps[2].data[w * h - ((i) / (w) + 1) * w + (i) % (w)];
408         B = img->comps[2].data[w * hr - ((i) / (wr) + 1) * w + (i) % (wr)];
409         fprintf(f, "%c%c%c", B, G, R);
410
411         if ((i + 1) % wr == 0) {
412           for (pad = (3 * wr) % 4 ? 4 - (3 * wr) % 4 : 0; pad > 0; pad--)       /* ADD */
413             fprintf(f, "%c", 0);
414         }
415       }
416       fclose(f);
417     } else {                    /* Gray-scale */
418
419       /* -->> -->> -->> -->>
420
421          8 bits non code (Gray scale)
422
423          <<-- <<-- <<-- <<-- */
424       f = fopen(argv[2], "wb");
425       // w = ceildiv(img->x1 - img->x0, img->comps[0].dx);
426       // wr = ceildiv(int_ceildivpow2(img->x1 - img->x0,img->factor), img->comps[0].dx);
427       w = img->comps[0].w;
428       wr = int_ceildivpow2(img->comps[0].w, img->comps[0].factor);
429
430       // h = ceildiv(img->y1 - img->y0, img->comps[0].dy);
431       // hr = ceildiv(int_ceildivpow2(img->y1 - img->y0,img->factor), img->comps[0].dy);
432       h = img->comps[0].h;
433       hr = int_ceildivpow2(img->comps[0].h, img->comps[0].factor);
434
435       fprintf(f, "BM");
436
437       /* FILE HEADER */
438       /* ------------- */
439       fprintf(f, "%c%c%c%c",
440               (unsigned char) (hr * wr + 54 + 1024 +
441                                hr * (wr % 2)) & 0xff,
442               (unsigned char) ((hr * wr + 54 + 1024 + hr * (wr % 2))
443                                >> 8) & 0xff,
444               (unsigned char) ((hr * wr + 54 + 1024 + hr * (wr % 2))
445                                >> 16) & 0xff,
446               (unsigned char) ((hr * wr + 54 + 1024 + wr * (wr % 2))
447                                >> 24) & 0xff);
448       fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
449               ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
450       fprintf(f, "%c%c%c%c", (54 + 1024) & 0xff,
451               ((54 + 1024) >> 8) & 0xff, ((54 + 1024) >> 16) & 0xff,
452               ((54 + 1024) >> 24) & 0xff);
453
454       /* INFO HEADER */
455       /* ------------- */
456       fprintf(f, "%c%c%c%c", (40) & 0xff, ((40) >> 8) & 0xff,
457               ((40) >> 16) & 0xff, ((40) >> 24) & 0xff);
458       fprintf(f, "%c%c%c%c", (unsigned char) ((wr) & 0xff),
459               (unsigned char) ((wr) >> 8) & 0xff,
460               (unsigned char) ((wr) >> 16) & 0xff,
461               (unsigned char) ((wr) >> 24) & 0xff);
462       fprintf(f, "%c%c%c%c", (unsigned char) ((hr) & 0xff),
463               (unsigned char) ((hr) >> 8) & 0xff,
464               (unsigned char) ((hr) >> 16) & 0xff,
465               (unsigned char) ((hr) >> 24) & 0xff);
466       fprintf(f, "%c%c", (1) & 0xff, ((1) >> 8) & 0xff);
467       fprintf(f, "%c%c", (8) & 0xff, ((8) >> 8) & 0xff);
468       fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
469               ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
470       fprintf(f, "%c%c%c%c",
471               (unsigned char) (hr * wr + hr * (wr % 2)) & 0xff,
472               (unsigned char) ((hr * wr + hr * (wr % 2)) >> 8) &
473               0xff,
474               (unsigned char) ((hr * wr + hr * (wr % 2)) >> 16) &
475               0xff,
476               (unsigned char) ((hr * wr + hr * (wr % 2)) >> 24) & 0xff);
477       fprintf(f, "%c%c%c%c", (7834) & 0xff, ((7834) >> 8) & 0xff,
478               ((7834) >> 16) & 0xff, ((7834) >> 24) & 0xff);
479       fprintf(f, "%c%c%c%c", (7834) & 0xff, ((7834) >> 8) & 0xff,
480               ((7834) >> 16) & 0xff, ((7834) >> 24) & 0xff);
481       fprintf(f, "%c%c%c%c", (256) & 0xff, ((256) >> 8) & 0xff,
482               ((256) >> 16) & 0xff, ((256) >> 24) & 0xff);
483       fprintf(f, "%c%c%c%c", (256) & 0xff, ((256) >> 8) & 0xff,
484               ((256) >> 16) & 0xff, ((256) >> 24) & 0xff);
485     }
486
487     for (i = 0; i < 256; i++) {
488       fprintf(f, "%c%c%c%c", i, i, i, 0);
489     }
490
491     for (i = 0; i < wr * hr; i++) {
492       /* a modifier !! */
493       // fprintf(f, "%c", img->comps[0].data[w * h - ((i) / (w) + 1) * w + (i) % (w)]);
494       fprintf(f, "%c",
495               img->comps[0].data[w * hr - ((i) / (wr) + 1) * w +
496                                  (i) % (wr)]);
497       /*if (((i + 1) % w == 0 && w % 2))
498          fprintf(f, "%c", 0); */
499       if ((i + 1) % wr == 0) {
500         for (pad = wr % 4 ? 4 - wr % 4 : 0; pad > 0; pad--)     /* ADD */
501           fprintf(f, "%c", 0);
502       }
503     }
504     break;
505   default:
506     break;
507   }
508
509   return 0;
510 }