d448fbe118002ec7c83fa61722b8f736286f06b2
[openjpeg.git] / jp3d / codec / volume_to_jp3d.c
1 /*\r
2  * Copyright (c) 2001-2003, David Janssens\r
3  * Copyright (c) 2002-2003, Yannick Verschueren\r
4  * Copyright (c) 2003-2005, Francois Devaux and Antonin Descampe\r
5  * Copyright (c) 2005, Herv� Drolon, FreeImage Team\r
6  * Copyright (c) 2002-2005, Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium\r
7  * Copyright (c) 2006, M�nica D�ez Garc�a, Image Processing Laboratory, University of Valladolid, Spain\r
8  * All rights reserved.\r
9  *\r
10  * Redistribution and use in source and binary forms, with or without\r
11  * modification, are permitted provided that the following conditions\r
12  * are met:\r
13  * 1. Redistributions of source code must retain the above copyright\r
14  *    notice, this list of conditions and the following disclaimer.\r
15  * 2. Redistributions in binary form must reproduce the above copyright\r
16  *    notice, this list of conditions and the following disclaimer in the\r
17  *    documentation and/or other materials provided with the distribution.\r
18  *\r
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'\r
20  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE\r
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE\r
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE\r
23  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR\r
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF\r
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS\r
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN\r
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)\r
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE\r
29  * POSSIBILITY OF SUCH DAMAGE.\r
30  */\r
31 #include <stdio.h>\r
32 #include <string.h>\r
33 #include <stdlib.h>\r
34 \r
35 #include "../libjp3dvm/openjpeg.h"\r
36 #include "getopt.h"\r
37 #include "convert.h"\r
38 \r
39 #ifdef _WIN32\r
40 #include <windows.h>\r
41 #else\r
42 #define stricmp strcasecmp\r
43 #define strnicmp strncasecmp\r
44 #endif /* _WIN32 */\r
45 \r
46 /* ----------------------------------------------------------------------- */\r
47 \r
48 void encode_help_display() {\r
49         fprintf(stdout,"List of parameters for the JPEG2000 Part 10 encoder:\n");\r
50         fprintf(stdout,"------------\n");\r
51         fprintf(stdout,"\n");\r
52         fprintf(stdout,"Required Parameters (except with -h):\n");\r
53         fprintf(stdout,"\n");\r
54         fprintf(stdout,"-i           : source file  (-i source.bin or source*.pgx) \n");\r
55         fprintf(stdout,"\n");\r
56         fprintf(stdout,"-m           : source characteristics file (-m imgfile.img) \n");\r
57         fprintf(stdout,"\n");\r
58         fprintf(stdout,"-o           : destination file (-o dest.jp3d) \n");\r
59         fprintf(stdout,"\n");\r
60         fprintf(stdout,"Optional Parameters:\n");\r
61         fprintf(stdout,"\n");\r
62         fprintf(stdout,"-h           : display the help information \n ");\r
63         fprintf(stdout,"\n");\r
64         fprintf(stdout,"-n           : number of resolutions (-n 3,3,3) \n");\r
65         fprintf(stdout,"\n");\r
66         fprintf(stdout,"-I           : use the irreversible transforms: ICT + DWT 9-7 (-I) \n");\r
67         fprintf(stdout,"\n");\r
68         fprintf(stdout,"-C           : coding algorithm (-C 2EB) [2EB, 3EB] \n");\r
69         fprintf(stdout,"\n");\r
70         fprintf(stdout,"-r           : different compression ratios for successive layers (-r 20,10,5)\n ");\r
71         fprintf(stdout,"                 - The rate specified for each quality level is the desired compression factor.\n");\r
72         fprintf(stdout,"                 - Rate 1 means lossless compression\n");\r
73         fprintf(stdout,"               (options -r and -q cannot be used together)\n ");\r
74         fprintf(stdout,"\n");\r
75         fprintf(stdout,"-q           : different psnr for successive layers (-q 30,40,50) \n ");\r
76         fprintf(stdout,"               (options -r and -q cannot be used together)\n ");\r
77         fprintf(stdout,"\n");\r
78         fprintf(stdout,"-b           : size of code block (-b 32,32,32) \n");\r
79         fprintf(stdout,"\n");\r
80         fprintf(stdout,"-c           : size of precinct (-c 128,128,128) \n");\r
81         fprintf(stdout,"\n");\r
82         fprintf(stdout,"-t           : size of tile (-t 512,512,512) \n");\r
83         fprintf(stdout,"\n");\r
84         fprintf(stdout,"-p           : progression order (-p LRCP) [LRCP, RLCP, RPCL, PCRL, CPRL] \n");\r
85         fprintf(stdout,"\n");\r
86         fprintf(stdout,"-s           : subsampling factor (-s 2,2,2) [-s X,Y,Z] \n");\r
87         fprintf(stdout,"                          - Remark: subsampling bigger than 2 can produce error\n");\r
88         fprintf(stdout,"\n");\r
89         fprintf(stdout,"-SOP         : write SOP marker before each packet \n");\r
90         fprintf(stdout,"\n");\r
91         fprintf(stdout,"-EPH         : write EPH marker after each header packet \n");\r
92         fprintf(stdout,"\n");\r
93         fprintf(stdout,"-M           : code-block style (-M 0) [1=BYPASS(LAZY) 2=RESET 4=RESTART(TERMALL)\n");\r
94         fprintf(stdout,"                 8=VSC 16=PTERM 32=SEGSYM 64=3DCTXT] \n");\r
95         fprintf(stdout,"                 Indicate multiple modes by adding their values. \n");\r
96         fprintf(stdout,"                 ex: RESTART(4) + RESET(2) + SEGMARK(32) = -M 38\n");\r
97         fprintf(stdout,"\n");\r
98         fprintf(stdout,"-D           : define DC offset (-D 12) \n");\r
99         fprintf(stdout,"\n");\r
100         fprintf(stdout,"-x           : create an index file *.Idx (-x index_name.Idx) \n");\r
101         fprintf(stdout,"\n");\r
102         fprintf(stdout,"-ROI         : c=%%d,U=%%d : quantization indices upshifted \n");\r
103         fprintf(stdout,"               for component c=%%d [%%d = 0,1,2]\n");\r
104         fprintf(stdout,"               with a value of U=%%d [0 <= %%d <= 37] (i.e. -ROI:c=0,U=25) \n");\r
105         fprintf(stdout,"\n");\r
106         fprintf(stdout,"-d           : offset of the origin of the volume (-d 150,300,100) \n");\r
107         fprintf(stdout,"\n");\r
108         fprintf(stdout,"-l           : offset of the origin of the tiles (-l 100,75,25) \n");\r
109         fprintf(stdout,"\n");\r
110         fprintf(stdout,"\n");\r
111         fprintf(stdout,"DEFAULT CODING:\n");\r
112         fprintf(stdout,"------------\n");\r
113         fprintf(stdout,"\n");\r
114         fprintf(stdout," * Lossless\n");\r
115         fprintf(stdout," * 1 tile\n");\r
116         fprintf(stdout," * Size of precinct : 2^15 x 2^15 x 2^15 (means 1 precinct)\n");\r
117         fprintf(stdout," * Size of code-block : 64 x 64 x 64\n");\r
118         fprintf(stdout," * Number of resolutions in x, y and z axis: 3\n");\r
119         fprintf(stdout," * No SOP marker in the codestream\n");\r
120         fprintf(stdout," * No EPH marker in the codestream\n");\r
121         fprintf(stdout," * No sub-sampling in x, y or z direction\n");\r
122         fprintf(stdout," * No mode switch activated\n");\r
123         fprintf(stdout," * Progression order: LRCP\n");\r
124         fprintf(stdout," * No index file\n");\r
125         fprintf(stdout," * No ROI upshifted\n");\r
126         fprintf(stdout," * No offset of the origin of the volume\n");\r
127         fprintf(stdout," * No offset of the origin of the tiles\n");\r
128         fprintf(stdout," * Reversible DWT 5-3 on each 2D slice\n");\r
129         fprintf(stdout," * Coding algorithm: 2D-EBCOT \n");\r
130         fprintf(stdout,"\n");\r
131         fprintf(stdout,"REMARKS:\n");\r
132         fprintf(stdout,"---------\n");\r
133         fprintf(stdout,"\n");\r
134         fprintf(stdout,"- The markers written to the main_header are : SOC SIZ COD QCD COM.\n");\r
135         fprintf(stdout,"- COD and QCD markers will never appear in the tile_header.\n");\r
136         fprintf(stdout,"\n");\r
137         fprintf(stdout,"- You need enough disk space memory (twice the original) to encode \n");\r
138         fprintf(stdout,"the volume,i.e. for a 1.5 GB volume you need a minimum of 3GB of disk memory)\n");\r
139         fprintf(stdout,"\n");\r
140         fprintf(stdout,"- When loading *.pgx files, a relative path to directory is needed for input argument \n");\r
141         fprintf(stdout," followed by the common prefix of the slices and a '*' character representing sequential numeration.\n");\r
142         fprintf(stdout,"( -i relativepath/slices*.pgx )\n");\r
143         fprintf(stdout,"\n");\r
144         fprintf(stdout," - The index file has the structure below:\n");\r
145         fprintf(stdout,"\n");\r
146         fprintf(stdout,"\t      Image_height Image_width Image_depth\n");\r
147         fprintf(stdout,"\t      Progression order: 0 (LRCP)\n");\r
148         fprintf(stdout,"\t      Tiles_size_X Tiles_size_Y Tiles_size_Z\n");\r
149         fprintf(stdout,"\t      Components_nb\n");\r
150         fprintf(stdout,"\t      Layers_nb\n");\r
151         fprintf(stdout,"\t      Decomposition_levels\n");\r
152         fprintf(stdout,"\t      [Precincts_size_X_res_Nr Precincts_size_Y_res_Nr Precincts_size_Z_res_Nr]\n\t  ...\n");\r
153         fprintf(stdout,"\t      [Precincts_size_X_res_0 Precincts_size_Y_res_0 Precincts_size_Z_res_0]\n");\r
154         fprintf(stdout,"\t      Main_header_end_position\n");\r
155         fprintf(stdout,"\t      Codestream_size\n");\r
156         fprintf(stdout,"\t      Tile_0 [start_pos end_header end_pos TotalDisto NumPix MaxMSE]\n");\r
157         fprintf(stdout,"\t      ...\n");\r
158         fprintf(stdout,"\t      Tile_Nt [  ''         ''        ''        ''       ''    ''  ]\n");\r
159         fprintf(stdout,"\t  Tpacket_0 [Tile layer res. comp. prec. start_pos end_pos disto]\n");\r
160         fprintf(stdout,"\t  ...\n");\r
161         fprintf(stdout,"\t  Tpacket_Np [''   ''    ''   ''    ''       ''       ''     '' ]\n");\r
162         fprintf(stdout,"\t  MaxDisto\n");\r
163         fprintf(stdout,"\t  TotalDisto\n\n");\r
164         fprintf(stdout,"\n");\r
165 \r
166 }\r
167 \r
168 OPJ_PROG_ORDER give_progression(char progression[4]) {\r
169         if(strncmp(progression, "LRCP", 4) == 0) {\r
170                 return LRCP;\r
171         }\r
172         if(strncmp(progression, "RLCP", 4) == 0) {\r
173                 return RLCP;\r
174         }\r
175         if(strncmp(progression, "RPCL", 4) == 0) {\r
176                 return RPCL;\r
177         }\r
178         if(strncmp(progression, "PCRL", 4) == 0) {\r
179                 return PCRL;\r
180         }\r
181         if(strncmp(progression, "CPRL", 4) == 0) {\r
182                 return CPRL;\r
183         }\r
184 \r
185         return PROG_UNKNOWN;\r
186 }\r
187 \r
188 OPJ_TRANSFORM give_transform(char transform[4]) {\r
189         if(strncmp(transform, "2DWT", 4) == 0) {\r
190                 return TRF_2D_DWT;\r
191         }\r
192         if(strncmp(transform, "3DWT", 4) == 0) {\r
193                 return TRF_3D_DWT;\r
194         }\r
195         return TRF_UNKNOWN;\r
196 }\r
197 \r
198 OPJ_ENTROPY_CODING give_coding(char coding[3]) {\r
199 \r
200         if(strncmp(coding, "2EB", 3) == 0) {\r
201                 return ENCOD_2EB;\r
202         }\r
203         if(strncmp(coding, "3EB", 3) == 0) {\r
204                 return ENCOD_3EB;\r
205         }\r
206         /*if(strncmp(coding, "2GR", 3) == 0) {\r
207                 return ENCOD_2GR;\r
208         }\r
209         if(strncmp(coding, "3GR", 3) == 0) {\r
210                 return ENCOD_3GR;\r
211         }*/\r
212 \r
213         return ENCOD_UNKNOWN;\r
214 }\r
215 \r
216 int get_file_format(char *filename) {\r
217         int i;\r
218         static const char *extension[] = {"pgx", "bin", "img", "j3d", "jp3d", "j2k"};\r
219         static const int format[] = { PGX_DFMT, BIN_DFMT, IMG_DFMT, J3D_CFMT, J3D_CFMT, J2K_CFMT};\r
220         char * ext = strrchr(filename, '.');\r
221         if (ext) {\r
222                 ext++;\r
223         for(i = 0; i < sizeof(format)/sizeof(*format); i++) {\r
224                         if(strnicmp(ext, extension[i], 3) == 0) {\r
225                 return format[i];\r
226                         }\r
227                 }\r
228         }\r
229 \r
230         return -1;\r
231 }\r
232 \r
233 /* ------------------------------------------------------------------------------------ */\r
234 \r
235 int parse_cmdline_encoder(int argc, char **argv, opj_cparameters_t *parameters) {\r
236         int i, value;\r
237 \r
238         /* parse the command line */\r
239 \r
240         while (1) {\r
241                 int c = getopt(argc, argv, "i:m:o:r:q:f:t:n:c:b:x:p:s:d:hP:S:E:M:D:R:l:T:C:A:I");\r
242                 if (c == -1)\r
243                         break;\r
244                 switch (c) {\r
245                         case 'i':                       /* input file */\r
246                         {\r
247                                 char *infile = optarg;\r
248                                 parameters->decod_format = get_file_format(infile);\r
249                                 switch(parameters->decod_format) {\r
250                                         case PGX_DFMT:\r
251                                         case BIN_DFMT:\r
252                                         case IMG_DFMT:\r
253                                                 break;\r
254                                         default:\r
255                                                 fprintf(stdout, "[ERROR] Unrecognized format for infile : %s [accept only *.pgx or *.bin] !!\n\n", infile);\r
256                                                 return 1;\r
257                                                 break;\r
258                                 }\r
259                                 strncpy(parameters->infile, infile, MAX_PATH);\r
260                                 fprintf(stdout, "[INFO] Infile: %s \n", parameters->infile);\r
261 \r
262                         }\r
263                         break;\r
264                                 \r
265                                 /* ----------------------------------------------------- */\r
266                         case 'm':                       /* input IMG file */\r
267                         {\r
268                                 char *imgfile = optarg;\r
269                                 int imgformat = get_file_format(imgfile);\r
270                                 switch(imgformat) {\r
271                                         case IMG_DFMT:\r
272                                                 break;\r
273                                         default:\r
274                                                 fprintf(stdout, "[ERROR] Unrecognized format for imgfile : %s [accept only *.img] !!\n\n", imgfile);\r
275                                                 return 1;\r
276                                                 break;\r
277                                 }\r
278                                 strncpy(parameters->imgfile, imgfile, MAX_PATH);\r
279                                 fprintf(stdout, "[INFO] Imgfile: %s Format: %d\n", parameters->imgfile, imgformat);\r
280                         }\r
281                         break;\r
282                                 \r
283                                 /* ----------------------------------------------------- */\r
284                         case 'o':                       /* output file */\r
285                         {\r
286                                 char *outfile = optarg;\r
287                                 parameters->cod_format = get_file_format(outfile);\r
288                                 switch(parameters->cod_format) {\r
289                                         case J3D_CFMT:\r
290                                         case J2K_CFMT:\r
291                                         case LSE_CFMT:\r
292                                                 break;\r
293                                         default:\r
294                                                 fprintf(stdout, "[ERROR] Unknown output format volume %s [only *.j2k, *.lse3d or *.jp3d]!! \n", outfile);\r
295                                                 return 1;\r
296                                                 break;\r
297                                 }\r
298                                 strncpy(parameters->outfile, outfile, MAX_PATH);\r
299                                 fprintf(stdout, "[INFO] Outfile: %s \n", parameters->outfile);\r
300                         }\r
301                         break;\r
302 \r
303                                 /* ----------------------------------------------------- */\r
304                         \r
305                         case 'r':                       /* define compression rates for each layer */\r
306                         {\r
307                                 char *s = optarg;\r
308                                 while (sscanf(s, "%f", &parameters->tcp_rates[parameters->tcp_numlayers]) == 1) {\r
309                                         parameters->tcp_numlayers++;\r
310                                         while (*s && *s != ',') {\r
311                                                 s++;\r
312                                         }\r
313                                         if (!*s)\r
314                                                 break;\r
315                                         s++;\r
316                                 }\r
317                                 parameters->cp_disto_alloc = 1;\r
318                         }\r
319                         break;\r
320                                 \r
321                                 /* ----------------------------------------------------- */\r
322                         \r
323                         case 'q':                       /* define distorsion (PSNR) for each layer */\r
324                         {\r
325                                 char *s = optarg;\r
326                                 while (sscanf(s, "%f", &parameters->tcp_distoratio[parameters->tcp_numlayers]) == 1) {\r
327                                         parameters->tcp_numlayers++;\r
328                                         while (*s && *s != ',') {\r
329                                                 s++;\r
330                                         }\r
331                                         if (!*s)\r
332                                                 break;\r
333                                         s++;\r
334                                 }\r
335                                 parameters->cp_fixed_quality = 1;\r
336                         }\r
337                         break;\r
338                                 \r
339                                 /* ----------------------------------------------------- */\r
340 \r
341                         case 'f':       \r
342                         {\r
343                                 fprintf(stdout, "/---------------------------------------------------\\\n");\r
344                                 fprintf(stdout, "|  Fixed layer allocation option not implemented !!  |\n");\r
345                                 fprintf(stdout, "\\---------------------------------------------------/\n");\r
346                                 /*int *row = NULL, *col = NULL;\r
347                                 int numlayers = 0, matrix_width = 0;\r
348 \r
349                                 char *s = optarg;\r
350                                 sscanf(s, "%d", &numlayers);\r
351                                 s++;\r
352                                 if (numlayers > 9)\r
353                                         s++;\r
354 \r
355                                 parameters->tcp_numlayers = numlayers;\r
356                                 matrix_width = parameters->numresolution[0] + parameters->numresolution[1] + parameters->numresolution[2];\r
357                                 parameters->cp_matrice = (int *) malloc(numlayers * matrix_width * sizeof(int));\r
358                                 s = s + 2;\r
359 \r
360                                 for (i = 0; i < numlayers; i++) {\r
361                                         row = &parameters->cp_matrice[i * matrix_width];\r
362                                         col = row;\r
363                                         parameters->tcp_rates[i] = 1;\r
364                                         sscanf(s, "%d,", &col[0]);\r
365                                         s += 2;\r
366                                         if (col[0] > 9)\r
367                                                 s++;\r
368                                         col[1] = 0;\r
369                                         col[2] = 0;\r
370                                         for (j = 1; j < matrix_width; j++) {\r
371                                                 col += 3; j+=2;\r
372                                                 sscanf(s, "%d,%d,%d", &col[0], &col[1], &col[2]);\r
373                                                 s += 6;\r
374                                                 if (col[0] > 9)\r
375                                                         s++;\r
376                                                 if (col[1] > 9)\r
377                                                         s++;\r
378                                                 if (col[2] > 9)\r
379                                                         s++;\r
380                                         }\r
381                                         if (i < numlayers - 1)\r
382                                                 s++;\r
383                                 }\r
384                                 parameters->cp_fixed_alloc = 1; */\r
385                         }\r
386                         break;\r
387                                 \r
388                                 /* ----------------------------------------------------- */\r
389 \r
390                         case 't':                       /* tiles */\r
391                         {\r
392                                 if (sscanf(optarg, "%d,%d,%d", &parameters->cp_tdx, &parameters->cp_tdy, &parameters->cp_tdz) !=3) {\r
393                                         fprintf(stdout, "[ERROR] '-t' 'dimensions of tiles' argument error !  [-t tdx,tdy,tdz]\n");\r
394                                         return 1;\r
395                                 }\r
396                                 parameters->tile_size_on = true;\r
397                         }\r
398                         break;\r
399                                 \r
400                                 /* ----------------------------------------------------- */\r
401                         \r
402                         case 'n':                       /* resolution */\r
403                         {\r
404                                 int aux;\r
405                                 aux = sscanf(optarg, "%d,%d,%d", &parameters->numresolution[0], &parameters->numresolution[1], &parameters->numresolution[2]);\r
406                                 if (aux == 2) \r
407                                         parameters->numresolution[2] = 1;\r
408                                 else if (aux == 1) {\r
409                                         parameters->numresolution[1] = parameters->numresolution[0];\r
410                                         parameters->numresolution[2] = 1;\r
411                                 }else if (aux == 0){\r
412                                         parameters->numresolution[0] = 1;\r
413                                         parameters->numresolution[1] = 1;\r
414                                         parameters->numresolution[2] = 1;\r
415                                 }\r
416                         }\r
417                         break;\r
418                                 \r
419                                 /* ----------------------------------------------------- */\r
420                         case 'c':                       /* precinct dimension */\r
421                         {\r
422                                 char sep;\r
423                                 int res_spec = 0;\r
424                                 int aux;\r
425                                 char *s = optarg;\r
426                                 do {\r
427                                         sep = 0;\r
428                                         aux = sscanf(s, "[%d,%d,%d]%c", &parameters->prct_init[0][res_spec], &parameters->prct_init[1][res_spec], &parameters->prct_init[2][res_spec], &sep);\r
429                                         if (sep == ',' && aux != 4) {\r
430                                                 fprintf(stdout, "[ERROR] '-c' 'dimensions of precincts' argument error !  [-c [prcx_res0,prcy_res0,prcz_res0],...,[prcx_resN,prcy_resN,prcz_resN]]\n");\r
431                                                 return 1;\r
432                                         }\r
433                                         parameters->csty |= 0x01;\r
434                                         res_spec++;\r
435                                         s = strpbrk(s, "]") + 2;\r
436                                 }\r
437                                 while (sep == ',');\r
438                                 parameters->res_spec = res_spec; /* number of precinct size specifications */\r
439                         }\r
440                         break;\r
441                                 \r
442                                 /* ----------------------------------------------------- */\r
443                         \r
444                         case 'b':                       /* code-block dimension */\r
445                         {\r
446                                 int cblockw_init = 0, cblockh_init = 0, cblockl_init = 0;\r
447                                 if (sscanf(optarg, "%d,%d,%d", &cblockw_init, &cblockh_init, &cblockl_init) != 3) {\r
448                                         fprintf(stdout, "[ERROR] '-b' 'dimensions of codeblocks' argument error !  [-b cblkx,cblky,cblkz]\n");\r
449                                         return 1;\r
450                                 }\r
451                                 if (cblockw_init * cblockh_init * cblockl_init > (1<<18) || cblockw_init > 1024 || cblockw_init < 4 || cblockh_init > 1024 || cblockh_init < 4 || cblockl_init > 1024 || cblockl_init < 4) {\r
452                                         fprintf(stdout,"[ERROR] Size of code_block error (option -b) !!\n\nRestriction :\n * width*height*length<=4096\n * 4<=width,height,length<= 1024\n\n");\r
453                                         return 1;\r
454                                 }\r
455                                 parameters->cblock_init[0] = cblockw_init;\r
456                                 parameters->cblock_init[1] = cblockh_init;\r
457                                 parameters->cblock_init[2] = cblockl_init;\r
458                         }\r
459                         break;\r
460                                 \r
461                                 /* ----------------------------------------------------- */\r
462                         \r
463                         case 'x':                       /* creation of index file */\r
464                         {\r
465                                 char *index = optarg;\r
466                                 strncpy(parameters->index, index, MAX_PATH);\r
467                                 parameters->index_on = 1;\r
468                         }\r
469                         break;\r
470                                 \r
471                                 /* ----------------------------------------------------- */\r
472                         \r
473                         case 'p':                       /* progression order */\r
474                         {\r
475                                 char progression[4];\r
476 \r
477                                 strncpy(progression, optarg, 4);\r
478                                 parameters->prog_order = give_progression(progression);\r
479                                 if (parameters->prog_order == -1) {\r
480                                         fprintf(stdout, "[ERROR] Unrecognized progression order [LRCP, RLCP, RPCL, PCRL, CPRL] !!\n");\r
481                                         return 1;\r
482                                 }\r
483                         }\r
484                         break;\r
485                                 \r
486                                 /* ----------------------------------------------------- */\r
487                         \r
488                         case 's':                       /* subsampling factor */\r
489                         {\r
490                                 if (sscanf(optarg, "%d,%d,%d", &parameters->subsampling_dx, &parameters->subsampling_dy, &parameters->subsampling_dz) != 2) {\r
491                                         fprintf(stdout, "[ERROR] '-s' sub-sampling argument error !  [-s dx,dy,dz]\n");\r
492                                         return 1;\r
493                                 }\r
494                         }\r
495                         break;\r
496                                 \r
497                                 /* ----------------------------------------------------- */\r
498                         \r
499                         case 'd':                       /* coordonnate of the reference grid */\r
500                         {\r
501                                 if (sscanf(optarg, "%d,%d,%d", &parameters->volume_offset_x0, &parameters->volume_offset_y0, &parameters->volume_offset_z0) != 3) {\r
502                                         fprintf(stdout, "[ERROR] -d 'coordonnate of the reference grid' argument error !! [-d x0,y0,z0]\n");\r
503                                         return 1;\r
504                                 }\r
505                         }\r
506                         break;\r
507                                 \r
508                                 /* ----------------------------------------------------- */\r
509                         \r
510                         case 'h':                       /* display an help description */\r
511                         {\r
512                                 encode_help_display();\r
513                                 return 1;\r
514                         }\r
515                         break;\r
516                                 \r
517                                 /* ----------------------------------------------------- */\r
518 \r
519                         case 'P':                       /* POC */\r
520                         {\r
521                                 int numpocs = 0;                /* number of progression order change (POC) default 0 */\r
522                                 opj_poc_t *POC = NULL;  /* POC : used in case of Progression order change */\r
523 \r
524                                 char *s = optarg;\r
525                                 POC = parameters->POC;\r
526 \r
527                                 fprintf(stdout, "/----------------------------------\\\n");\r
528                                 fprintf(stdout, "|  POC option not fully tested !!  |\n");\r
529                                 fprintf(stdout, "\\----------------------------------/\n");\r
530                                 \r
531                                 while (sscanf(s, "T%d=%d,%d,%d,%d,%d,%s", &POC[numpocs].tile,\r
532                                                                                                                 &POC[numpocs].resno0, &POC[numpocs].compno0,\r
533                                                                                                                 &POC[numpocs].layno1, &POC[numpocs].resno1,\r
534                                                                                                                 &POC[numpocs].compno1, POC[numpocs].progorder) == 7) {\r
535                                         POC[numpocs].prg = give_progression(POC[numpocs].progorder);\r
536                                         /* POC[numpocs].tile; */\r
537                                         numpocs++;\r
538                                         while (*s && *s != '/') {\r
539                                                 s++;\r
540                                         }\r
541                                         if (!*s) {\r
542                                                 break;\r
543                                         }\r
544                                         s++;\r
545                                 }\r
546                                 parameters->numpocs = numpocs;\r
547                         }\r
548                         break;\r
549                                 \r
550                                 /* ------------------------------------------------------ */\r
551                                 \r
552                         case 'S':                       /* SOP marker */\r
553                         {\r
554                                 parameters->csty |= 0x02;\r
555                         }\r
556                         break;\r
557                                 \r
558                                 /* ------------------------------------------------------ */\r
559                         \r
560                         case 'E':                       /* EPH marker */\r
561                         {\r
562                                 parameters->csty |= 0x04;\r
563                         }\r
564                         break;\r
565                                 \r
566                                 /* ------------------------------------------------------ */\r
567                         \r
568                         case 'M':                       /* Codification mode switch */\r
569                         {\r
570                                 fprintf(stdout, "[INFO] Mode switch option not fully tested !!\n");\r
571                                 value = 0;\r
572                                 if (sscanf(optarg, "%d", &value) == 1) {\r
573                                         for (i = 0; i <= 6; i++) {\r
574                                                 int cache = value & (1 << i);\r
575                                                 if (cache)\r
576                                                         parameters->mode |= (1 << i);\r
577                                         }\r
578                                 }\r
579                         }\r
580                         break;\r
581                                 \r
582                                 /* ------------------------------------------------------ */\r
583                         \r
584                         case 'D':                       /* DCO */\r
585                         {\r
586                                 if (sscanf(optarg, "%d", &parameters->dcoffset) != 1) {\r
587                                         fprintf(stdout, "[ERROR] DC offset error !! [-D %d]\n",parameters->dcoffset);\r
588                                         return 1;\r
589                                 }\r
590                         }\r
591                         break;\r
592                                 \r
593                                 /* ------------------------------------------------------ */\r
594                         \r
595                         case 'R':                       /* ROI */\r
596                         {\r
597                                 if (sscanf(optarg, "OI:c=%d,U=%d", &parameters->roi_compno, &parameters->roi_shift) != 2) {\r
598                                         fprintf(stdout, "[ERROR] ROI error !! [-ROI:c='compno',U='shift']\n");\r
599                                         return 1;\r
600                                 }\r
601                         }\r
602                         break;\r
603                                 \r
604                                 /* ------------------------------------------------------ */\r
605                         \r
606                         case 'l':                       /* Tile offset */\r
607                         {\r
608                                 if (sscanf(optarg, "%d,%d,%d", &parameters->cp_tx0, &parameters->cp_ty0, &parameters->cp_tz0) != 3) {\r
609                                         fprintf(stdout, "[ERROR] -l 'tile offset' argument error !! [-l X0,Y0,Z0]");\r
610                                         return 1;\r
611                                 }\r
612                         }\r
613                         break;\r
614                                 \r
615                                 /* ------------------------------------------------------ \r
616                                 \r
617                         case 'T':                       // Tranformation of original data (2D-DWT/3D-DWT/3D-RLS/2D-DWT+1D-RLS) \r
618                         {\r
619                                 char transform[4];\r
620 \r
621                                 strncpy(transform, optarg, 4);\r
622                                 parameters->transform_format = give_transform(transform);\r
623                                 if (parameters->transform_format == -1) {\r
624                                         fprintf(stdout, "[ERROR] -T 'Transform domain' argument error !! [-T 2DWT, 3DWT, 3RLS or 3LSE only]");\r
625                     return 1;\r
626                                 }\r
627                         }\r
628                         break;\r
629                                 \r
630                                  ------------------------------------------------------ */\r
631                         \r
632                         case 'C':                       /* Coding of transformed data */\r
633                         {\r
634                                 char coding[3];\r
635 \r
636                                 strncpy(coding, optarg, 3);\r
637                                 parameters->encoding_format = give_coding(coding);\r
638                                 if (parameters->encoding_format == -1) {\r
639                                         fprintf(stdout, "[ERROR] -C 'Coding algorithm' argument error !! [-C 2EB, 3EB, 2GR, 3GR or GRI only]");\r
640                     return 1;\r
641                                 }\r
642                         }\r
643                         break;\r
644                         \r
645                         /* ------------------------------------------------------ */\r
646                         \r
647                         case 'I':                       /* reversible or not */\r
648                         {\r
649                                 parameters->irreversible = 1;\r
650                         }\r
651                         break;\r
652                                 \r
653                         default:\r
654                                 fprintf(stdout, "[ERROR] This option is not valid \"-%c %s\"\n", c, optarg);\r
655                                 return 1;\r
656                 }\r
657         }\r
658 \r
659         /* check for possible errors */\r
660 \r
661         if((parameters->infile[0] == 0) || (parameters->outfile[0] == 0)) {\r
662                 fprintf(stdout, "usage: jp3d_vm_enc -i volume-file -o jp3d-file (+ options)\n");\r
663                 return 1;\r
664         }\r
665 \r
666         if((parameters->decod_format == BIN_DFMT) && (parameters->imgfile[0] == 0)) {\r
667                 fprintf(stdout, "usage: jp3d_vm_enc -i bin-volume-file -m img-file -o jp3d-file (+ options)\n");\r
668                 return 1;\r
669         }\r
670 \r
671         if((parameters->decod_format != BIN_DFMT) && (parameters->decod_format != PGX_DFMT) && (parameters->decod_format != IMG_DFMT)) {\r
672                 fprintf(stdout, "usage: jp3d_vm_enc -i input-volume-file [*.bin,*.pgx,*.img] -o jp3d-file [*.jp3d,*.j2k] (+ options)\n");\r
673                 return 1;\r
674         }\r
675         if((parameters->cod_format != J3D_CFMT) && (parameters->cod_format != J2K_CFMT)) {\r
676                 fprintf(stdout, "usage: jp3d_vm_enc -i input-volume-file [*.bin,*.pgx,*.img] -o jp3d-file [*.jp3d,*.j2k] (+ options)\n");\r
677                 return 1;\r
678         }\r
679 \r
680         if((parameters->encoding_format == ENCOD_2GR || parameters->encoding_format == ENCOD_3GR) && parameters->transform_format != TRF_3D_LSE && parameters->transform_format != TRF_3D_RLS) {\r
681                 fprintf(stdout, "[ERROR] Entropy coding options -C [2GR,3GR] are only compatible with predictive-based transform algorithms: -T [3RLS,3LSE].\n");\r
682                 return 1;\r
683         }\r
684         if (parameters->encoding_format == ENCOD_3EB)\r
685                 parameters->mode |= (1 << 6);\r
686 \r
687         if ((parameters->mode >> 6) & 1) {\r
688                 parameters->encoding_format = ENCOD_3EB;\r
689         }\r
690 \r
691         if((parameters->numresolution[2] == 0 || (parameters->numresolution[1] == 0) || (parameters->numresolution[0] == 0))) {\r
692                 fprintf(stdout, "[ERROR] -n 'resolution levels' argument error ! Resolutions must be greater than 1 in order to perform DWT.\n");\r
693                 return 1;\r
694         }\r
695         if (parameters->numresolution[1] != parameters->numresolution[0]) {\r
696                 fprintf(stdout, "[ERROR] -n 'resolution levels' argument error ! Resolutions in X and Y axis must be the same in this implementation.\n");\r
697                 return 1;\r
698         }\r
699         \r
700         if (parameters->numresolution[2] > parameters->numresolution[0]) {\r
701                 fprintf(stdout, "[ERROR] -n 'resolution levels' argument error ! Resolutions in Z axis must be lower than in X-Y axis.\n");\r
702                 return 1;\r
703         }\r
704         \r
705         if (parameters->dcoffset >= 128 && parameters->dcoffset <= -128) {\r
706                 fprintf(stdout, "[ERROR] -D 'DC offset' argument error ! Value must be -128<=DCO<=128.\n");\r
707                 return 1;\r
708         }\r
709 \r
710         if(parameters->numresolution[2] != 1) {\r
711                 parameters->transform_format = TRF_3D_DWT;\r
712                 //fprintf(stdout, "[Warning] Resolution level in axial dim > 1 : 3D-DWT will be performed... \n");\r
713         } else if (parameters->numresolution[2] == 1) {\r
714                 parameters->transform_format = TRF_2D_DWT;\r
715                 //fprintf(stdout, "[Warning] Resolution level in axial dim == 1 : 2D-DWT will be performed... \n");\r
716         }\r
717         \r
718         if ((parameters->cod_format == J2K_CFMT) && (parameters->transform_format != TRF_2D_DWT || parameters->encoding_format != ENCOD_2EB)) {\r
719                 fprintf(stdout, "[WARNING] Incompatible options -o *.j2k and defined transform or encoding algorithm. Latter will be ignored\n");\r
720                 parameters->transform_format = TRF_2D_DWT;\r
721                 parameters->encoding_format = ENCOD_2EB;\r
722         }\r
723 \r
724         if ((parameters->cp_disto_alloc || parameters->cp_fixed_alloc || parameters->cp_fixed_quality) && (!(parameters->cp_disto_alloc ^ parameters->cp_fixed_quality))) {\r
725                 fprintf(stdout, "[ERROR] Options -r and -q cannot be used together !!\n");\r
726                 return 1;\r
727         }                               /* mod fixed_quality */\r
728 \r
729         /* if no rate entered, lossless by default */\r
730         if (parameters->tcp_numlayers == 0) {\r
731                 parameters->tcp_rates[0] = 0.0; /* MOD antonin : losslessbug */\r
732                 parameters->tcp_numlayers++;\r
733                 parameters->cp_disto_alloc = 1;\r
734         }\r
735 \r
736         if((parameters->cp_tx0 > parameters->volume_offset_x0) || (parameters->cp_ty0 > parameters->volume_offset_y0) || (parameters->cp_tz0 > parameters->volume_offset_z0)) {\r
737                 fprintf(stdout, "[ERROR] Tile offset dimension is unnappropriate --> TX0(%d)<=IMG_X0(%d) TYO(%d)<=IMG_Y0(%d) TZO(%d)<=IMG_Z0(%d)\n",\r
738                         parameters->cp_tx0, parameters->volume_offset_x0, parameters->cp_ty0, parameters->volume_offset_y0, \r
739                         parameters->cp_tz0, parameters->volume_offset_z0);\r
740                 return 1;\r
741         }\r
742 \r
743         for (i = 0; i < parameters->numpocs; i++) {\r
744                 if (parameters->POC[i].prg == -1) {\r
745                         fprintf(stdout,"[ERROR] Unrecognized progression order in option -P (POC n %d) [LRCP, RLCP, RPCL, PCRL, CPRL] !!\n",i + 1);\r
746                 }\r
747         }\r
748         return 0;\r
749 }\r
750 \r
751 /* -------------------------------------------------------------------------- */\r
752 \r
753 /**\r
754 sample error callback expecting a FILE* client object\r
755 */\r
756 void error_callback(const char *msg, void *client_data) {\r
757         FILE *stream = (FILE*)client_data;\r
758         fprintf(stream, "[ERROR] %s", msg);\r
759 }\r
760 /**\r
761 sample warning callback expecting a FILE* client object\r
762 */\r
763 void warning_callback(const char *msg, void *client_data) {\r
764         FILE *stream = (FILE*)client_data;\r
765         fprintf(stream, "[WARNING] %s", msg);\r
766 }\r
767 /**\r
768 sample debug callback expecting a FILE* client object\r
769 */\r
770 void info_callback(const char *msg, void *client_data) {\r
771         FILE *stream = (FILE*)client_data;\r
772         fprintf(stream, "[INFO] %s", msg);\r
773 }\r
774 \r
775 /* -------------------------------------------------------------------------- */\r
776 \r
777 int main(int argc, char **argv) {\r
778         bool bSuccess;\r
779         bool delete_comment = true;\r
780         opj_cparameters_t parameters;   /* compression parameters */\r
781         opj_event_mgr_t event_mgr;              /* event manager */\r
782         opj_volume_t *volume = NULL;\r
783 \r
784         /* \r
785         configure the event callbacks (not required)\r
786         setting of each callback is optionnal \r
787         */\r
788         memset(&event_mgr, 0, sizeof(opj_event_mgr_t));\r
789         event_mgr.error_handler = error_callback;\r
790         event_mgr.warning_handler = warning_callback;\r
791         event_mgr.info_handler = info_callback;\r
792 \r
793         /* set encoding parameters to default values */\r
794         opj_set_default_encoder_parameters(&parameters);\r
795 \r
796         /* parse input and get user encoding parameters */\r
797         if(parse_cmdline_encoder(argc, argv, &parameters) == 1) {\r
798                 return 0;\r
799         }\r
800 \r
801         if(parameters.cp_comment == NULL) {\r
802                 parameters.cp_comment = "Created by OpenJPEG version JP3D";\r
803                 /* no need to delete parameters.cp_comment on exit */\r
804                 delete_comment = false;\r
805         }\r
806         \r
807         /* encode the destination volume */\r
808         /* ---------------------------- */\r
809         if (parameters.cod_format == J3D_CFMT || parameters.cod_format == J2K_CFMT)     {\r
810                 int codestream_length, pixels, bitsin;\r
811                 opj_cio_t *cio = NULL;\r
812                 FILE *f = NULL;\r
813                 opj_cinfo_t* cinfo = NULL;\r
814                 \r
815                 /* decode the source volume */\r
816                 /* ----------------------- */\r
817                 switch (parameters.decod_format) {\r
818                         case PGX_DFMT: \r
819                                 fprintf(stdout, "[INFO] Loading pgx file(s)\n");\r
820                                 volume = pgxtovolume(parameters.infile, &parameters);\r
821                                 if (!volume) {\r
822                                         fprintf(stdout, "[ERROR] Unable to load pgx files\n");\r
823                                         return 1;\r
824                                 }\r
825                                 break;\r
826                         \r
827                         case BIN_DFMT:\r
828                                 fprintf(stdout, "[INFO] Loading bin file\n");\r
829                                 volume = bintovolume(parameters.infile, parameters.imgfile, &parameters);\r
830                                 if (!volume) {\r
831                                         fprintf(stdout, "[ERROR] Unable to load bin file\n");\r
832                                         return 1;\r
833                                 }\r
834                                 break;\r
835 \r
836                         case IMG_DFMT:\r
837                                 fprintf(stdout, "[INFO] Loading img file\n");\r
838                                 volume = imgtovolume(parameters.infile, &parameters);\r
839                                 if (!volume) {\r
840                                         fprintf(stderr, "[ERROR] Unable to load img file\n");\r
841                                         return 1;\r
842                                 }\r
843                                 break;\r
844                 }\r
845                 \r
846                 /* get a JP3D or J2K compressor handle */\r
847                 if (parameters.cod_format == J3D_CFMT) \r
848             cinfo = opj_create_compress(CODEC_J3D);\r
849                 else if (parameters.cod_format == J2K_CFMT) \r
850             cinfo = opj_create_compress(CODEC_J2K);\r
851 \r
852                 /* catch events using our callbacks and give a local context */\r
853                 opj_set_event_mgr((opj_common_ptr)cinfo, &event_mgr, stdout);                   \r
854 \r
855                 /* setup the encoder parameters using the current volume and using user parameters */\r
856                 opj_setup_encoder(cinfo, &parameters, volume);\r
857                 \r
858                 /* open a byte stream for writing */\r
859                 /* allocate memory for all tiles */\r
860                 cio = opj_cio_open((opj_common_ptr)cinfo, NULL, 0);\r
861 \r
862                 /* encode the volume */\r
863                 //fprintf(stdout, "[INFO] Encode the volume\n");\r
864                 bSuccess = opj_encode(cinfo, cio, volume, parameters.index);\r
865                 if (!bSuccess) {\r
866                         opj_cio_close(cio);\r
867                         fprintf(stdout, "[ERROR] Failed to encode volume\n");\r
868                         return 1;\r
869                 }\r
870                 codestream_length = cio_tell(cio);\r
871                 pixels =(volume->x1 - volume->x0) * (volume->y1 - volume->y0) * (volume->z1 - volume->z0);\r
872                 bitsin = pixels * volume->comps[0].prec;\r
873                 fprintf(stdout, "[RESULT] Volume: %d x %d x %d (x %d bpv)\n Codestream: %d B,  Ratio: %5.3f bpv,  (%5.3f : 1) \n", \r
874                         (volume->x1 - volume->x0),(volume->y1 - volume->y0),(volume->z1 - volume->z0),volume->comps[0].prec,\r
875                         codestream_length, ((double)codestream_length * 8.0/(double)pixels), ((double)bitsin/(8.0*(double)codestream_length)));\r
876 \r
877                 /* write the buffer to disk */\r
878                 f = fopen(parameters.outfile, "wb");\r
879                 if (!f) {\r
880                         fprintf(stdout, "[ERROR] Failed to open %s for writing\n", parameters.outfile);\r
881                         return 1;\r
882                 }\r
883                 fwrite(cio->buffer, 1, codestream_length, f);\r
884                 fclose(f);\r
885 \r
886                 /* close and free the byte stream */\r
887                 opj_cio_close(cio);\r
888 \r
889                 /* free remaining compression structures */\r
890                 opj_destroy_compress(cinfo);\r
891         } else {\r
892                 fprintf(stdout, "[ERROR] Cod_format != JP3d !!! \n");\r
893                 return 1;\r
894         }\r
895 \r
896         /* free user parameters structure */\r
897         if(delete_comment) {\r
898                 if(parameters.cp_comment) free(parameters.cp_comment);\r
899         }\r
900         if(parameters.cp_matrice) free(parameters.cp_matrice);\r
901 \r
902         /* free volume data */\r
903         opj_volume_destroy(volume);\r
904         \r
905         return 0;\r
906 }\r