[2.1] Create the 2.1 branch
[openjpeg.git] / tests / compare_images.c
1 /*
2  * Copyright (c) 2011-2012, Centre National d'Etudes Spatiales (CNES), France 
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
15  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
18  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
19  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
20  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
21  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
22  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
23  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24  * POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 /*
28  * compare_images.c
29  *
30  *  Created on: 8 juil. 2011
31  *      Author: mickael
32  */
33
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <math.h>
37 #include <string.h>
38 #include <ctype.h>
39 #include <assert.h>
40
41 #include "opj_apps_config.h"
42 #include "opj_getopt.h"
43
44 #include "openjpeg.h"
45 #include "format_defs.h"
46 #include "convert.h"
47
48 #ifdef OPJ_HAVE_LIBTIFF
49 #include <tiffio.h> /* TIFFSetWarningHandler */
50 #endif /* OPJ_HAVE_LIBTIFF */
51
52 /*******************************************************************************
53  * Parse MSE and PEAK input values (
54  * separator = ":"
55  *******************************************************************************/
56 static double* parseToleranceValues( char* inArg, const int nbcomp)
57 {
58   double* outArgs= malloc((size_t)nbcomp * sizeof(double));
59   int it_comp = 0;
60   const char delims[] = ":";
61   char *result = strtok( inArg, delims );
62
63   while( (result != NULL) && (it_comp < nbcomp ))
64     {
65     outArgs[it_comp] = atof(result);
66     result = strtok( NULL, delims );
67     it_comp++;
68     }
69
70   if (it_comp != nbcomp)
71     {
72     free(outArgs);
73     return NULL;
74     }
75   /* else */
76   return outArgs;
77 }
78
79 /*******************************************************************************
80  * Command line help function
81  *******************************************************************************/
82 static void compare_images_help_display(void)
83 {
84   fprintf(stdout,"\nList of parameters for the compare_images function  \n");
85   fprintf(stdout,"\n");
86   fprintf(stdout,"  -b \t REQUIRED \t filename to the reference/baseline PGX/TIF/PNM image \n");
87   fprintf(stdout,"  -t \t REQUIRED \t filename to the test PGX/TIF/PNM image\n");
88   fprintf(stdout,"  -n \t REQUIRED \t number of component of the image (used to generate correct filename)\n");
89   fprintf(stdout,"  -m \t OPTIONAL \t list of MSE tolerances, separated by : (size must correspond to the number of component) of \n");
90   fprintf(stdout,"  -p \t OPTIONAL \t list of PEAK tolerances, separated by : (size must correspond to the number of component) \n");
91   fprintf(stdout,"  -s \t OPTIONAL \t 1 or 2 filename separator to take into account PGX/PNM image with different components, "
92                                       "please indicate b or t before separator to indicate respectively the separator "
93                                       "for ref/base file and for test file.  \n");
94   fprintf(stdout,"  -d \t OPTIONAL \t indicate if you want to run this function as conformance test or as non regression test\n");
95   fprintf(stdout,"\n");
96 }
97
98 static int get_decod_format_from_string(const char *filename)
99 {
100   const int dot = '.';
101   char * ext = strrchr(filename, dot);
102   if( strcmp(ext,".pgx") == 0 ) return PGX_DFMT;
103   if( strcmp(ext,".tif") == 0 ) return TIF_DFMT;
104   if( strcmp(ext,".ppm") == 0 ) return PXM_DFMT;
105   return -1;
106 }
107
108
109 /*******************************************************************************
110  * Create filenames from a filename using separator and nb components
111  * (begin from 0)
112  *******************************************************************************/
113 static char* createMultiComponentsFilename(const char* inFilename, const int indexF, const char* separator)
114 {
115   char s[255];
116   char *outFilename, *ptr;
117   const char token = '.';
118   size_t posToken = 0;
119   int decod_format;
120
121   /*printf("inFilename = %s\n", inFilename);*/
122   if ((ptr = strrchr(inFilename, token)) != NULL)
123     {
124     posToken = strlen(inFilename) - strlen(ptr);
125     /*printf("Position of %c character inside inFilename = %d\n", token, posToken);*/
126     }
127   else
128     {
129     /*printf("Token %c not found\n", token);*/
130     outFilename = (char*)malloc(1);
131     outFilename[0] = '\0';
132     return outFilename;
133     }
134
135   outFilename = (char*)malloc((posToken + 7) * sizeof(char)); /*6*/
136
137   strncpy(outFilename, inFilename, posToken);
138   outFilename[posToken] = '\0';
139   strcat(outFilename, separator);
140   sprintf(s, "%i", indexF);
141   strcat(outFilename, s);
142
143   decod_format = get_decod_format_from_string(inFilename);
144   if( decod_format == PGX_DFMT )
145     {
146     strcat(outFilename, ".pgx");
147     }
148   else if( decod_format == PXM_DFMT )
149     {
150     strcat(outFilename, ".pgm");
151     }
152
153   /*printf("outfilename: %s\n", outFilename);*/
154   return outFilename;
155 }
156
157 /*******************************************************************************
158  *
159  *******************************************************************************/
160 static opj_image_t* readImageFromFilePPM(const char* filename, int nbFilenamePGX, const char *separator)
161 {
162   int it_file;
163   opj_image_t* image_read = NULL;
164   opj_image_t* image = NULL;
165   opj_cparameters_t parameters;
166   opj_image_cmptparm_t* param_image_read;
167   int** data;
168
169   /* If separator is empty => nb file to read is equal to one*/
170   if ( strlen(separator) == 0 )
171     nbFilenamePGX = 1;
172
173   /* set encoding parameters to default values */
174   opj_set_default_encoder_parameters(&parameters);
175   parameters.decod_format = PXM_DFMT;
176   strcpy(parameters.infile, filename);
177
178   /* Allocate memory*/
179   param_image_read = malloc((size_t)nbFilenamePGX * sizeof(opj_image_cmptparm_t));
180   data = malloc((size_t)nbFilenamePGX * sizeof(*data));
181
182   for (it_file = 0; it_file < nbFilenamePGX; it_file++)
183     {
184     /* Create the right filename*/
185     char *filenameComponentPGX;
186     if (strlen(separator) == 0)
187       {
188       filenameComponentPGX = malloc((strlen(filename) + 1) * sizeof(*filenameComponentPGX));
189       strcpy(filenameComponentPGX, filename);
190       }
191     else
192       filenameComponentPGX = createMultiComponentsFilename(filename, it_file, separator);
193
194     /* Read the tif file corresponding to the component */
195     image_read = pnmtoimage(filenameComponentPGX, &parameters);
196     if (!image_read)
197       {
198       int it_free_data;
199       fprintf(stderr, "Unable to load ppm file: %s\n", filenameComponentPGX);
200
201       free(param_image_read);
202
203       for (it_free_data = 0; it_free_data < it_file; it_free_data++) {
204         free(data[it_free_data]);
205       }
206       free(data);
207
208       free(filenameComponentPGX);
209
210       return NULL;
211       }
212
213     /* Set the image_read parameters*/
214     param_image_read[it_file].x0 = 0;
215     param_image_read[it_file].y0 = 0;
216     param_image_read[it_file].dx = 0;
217     param_image_read[it_file].dy = 0;
218     param_image_read[it_file].h = image_read->comps->h;
219     param_image_read[it_file].w = image_read->comps->w;
220     param_image_read[it_file].bpp = image_read->comps->bpp;
221     param_image_read[it_file].prec = image_read->comps->prec;
222     param_image_read[it_file].sgnd = image_read->comps->sgnd;
223
224     /* Copy data*/
225     data[it_file] = malloc(param_image_read[it_file].h * param_image_read[it_file].w * sizeof(int));
226     memcpy(data[it_file], image_read->comps->data, image_read->comps->h * image_read->comps->w * sizeof(int));
227
228     /* Free memory*/
229     opj_image_destroy(image_read);
230     free(filenameComponentPGX);
231     }
232
233   image = opj_image_create((OPJ_UINT32)nbFilenamePGX, param_image_read, OPJ_CLRSPC_UNSPECIFIED);
234   for (it_file = 0; it_file < nbFilenamePGX; it_file++)
235     {
236     /* Copy data into output image and free memory*/
237     memcpy(image->comps[it_file].data, data[it_file], image->comps[it_file].h * image->comps[it_file].w * sizeof(int));
238     free(data[it_file]);
239     }
240
241   /* Free memory*/
242   free(param_image_read);
243   free(data);
244
245   return image;
246 }
247
248 static opj_image_t* readImageFromFileTIF(const char* filename, int nbFilenamePGX, const char *separator)
249 {
250   opj_image_t* image_read = NULL;
251   opj_cparameters_t parameters;
252   (void)nbFilenamePGX;
253   (void)separator;
254
255   /* conformance test suite produce annoying warning/error:
256    * TIFFReadDirectory: Warning, /.../data/baseline/conformance/jp2_1.tif: unknown field with tag 37724 (0x935c) encountered.
257    * TIFFOpen: /.../data/baseline/nonregression/opj_jp2_1.tif: Cannot open.
258    * On Win32 this open a message box by default, so remove it from the test suite:
259    */
260 #ifdef OPJ_HAVE_LIBTIFF
261   TIFFSetWarningHandler(NULL);
262   TIFFSetErrorHandler(NULL);
263 #endif
264
265   if ( strlen(separator) != 0 ) return NULL;
266
267   /* set encoding parameters to default values */
268   opj_set_default_encoder_parameters(&parameters);
269   parameters.decod_format = TIF_DFMT;
270   strcpy(parameters.infile, filename);
271
272   /* Read the tif file corresponding to the component */
273 #ifdef OPJ_HAVE_LIBTIFF
274   image_read = tiftoimage(filename, &parameters);
275 #endif
276   if (!image_read)
277     {
278     fprintf(stderr, "Unable to load TIF file\n");
279     return NULL;
280     }
281
282   /* \postconditions */
283   assert( image_read->numcomps == 1 || image_read->numcomps == 3 );
284   return image_read;
285 }
286
287 static opj_image_t* readImageFromFilePGX(const char* filename, int nbFilenamePGX, const char *separator)
288 {
289   int it_file;
290   opj_image_t* image_read = NULL;
291   opj_image_t* image = NULL;
292   opj_cparameters_t parameters;
293   opj_image_cmptparm_t* param_image_read;
294   int** data;
295
296   /* If separator is empty => nb file to read is equal to one*/
297   if ( strlen(separator) == 0 )
298     nbFilenamePGX = 1;
299
300   /* set encoding parameters to default values */
301   opj_set_default_encoder_parameters(&parameters);
302   parameters.decod_format = PGX_DFMT;
303   strcpy(parameters.infile, filename);
304
305   /* Allocate memory*/
306   param_image_read = malloc((size_t)nbFilenamePGX * sizeof(opj_image_cmptparm_t));
307   data = malloc((size_t)nbFilenamePGX * sizeof(*data));
308
309   for (it_file = 0; it_file < nbFilenamePGX; it_file++)
310     {
311     /* Create the right filename*/
312     char *filenameComponentPGX;
313     if (strlen(separator) == 0)
314       {
315       filenameComponentPGX = malloc((strlen(filename) + 1) * sizeof(*filenameComponentPGX));
316       strcpy(filenameComponentPGX, filename);
317       }
318     else
319       filenameComponentPGX = createMultiComponentsFilename(filename, it_file, separator);
320
321     /* Read the pgx file corresponding to the component */
322     image_read = pgxtoimage(filenameComponentPGX, &parameters);
323     if (!image_read)
324       {
325       int it_free_data;
326       fprintf(stderr, "Unable to load pgx file\n");
327
328       free(param_image_read);
329
330       for (it_free_data = 0; it_free_data < it_file; it_free_data++) {
331         free(data[it_free_data]);
332       }
333       free(data);
334
335       free(filenameComponentPGX);
336
337       return NULL;
338       }
339
340     /* Set the image_read parameters*/
341     param_image_read[it_file].x0 = 0;
342     param_image_read[it_file].y0 = 0;
343     param_image_read[it_file].dx = 0;
344     param_image_read[it_file].dy = 0;
345     param_image_read[it_file].h = image_read->comps->h;
346     param_image_read[it_file].w = image_read->comps->w;
347     param_image_read[it_file].bpp = image_read->comps->bpp;
348     param_image_read[it_file].prec = image_read->comps->prec;
349     param_image_read[it_file].sgnd = image_read->comps->sgnd;
350
351     /* Copy data*/
352     data[it_file] = malloc(param_image_read[it_file].h * param_image_read[it_file].w * sizeof(int));
353     memcpy(data[it_file], image_read->comps->data, image_read->comps->h * image_read->comps->w * sizeof(int));
354
355     /* Free memory*/
356     opj_image_destroy(image_read);
357     free(filenameComponentPGX);
358     }
359
360   image = opj_image_create((OPJ_UINT32)nbFilenamePGX, param_image_read, OPJ_CLRSPC_UNSPECIFIED);
361   for (it_file = 0; it_file < nbFilenamePGX; it_file++)
362     {
363     /* Copy data into output image and free memory*/
364     memcpy(image->comps[it_file].data, data[it_file], image->comps[it_file].h * image->comps[it_file].w * sizeof(int));
365     free(data[it_file]);
366     }
367
368   /* Free memory*/
369   free(param_image_read);
370   free(data);
371
372   return image;
373 }
374
375 #if defined(OPJ_HAVE_LIBPNG) && 0 /* remove for now */
376 /*******************************************************************************
377  *
378  *******************************************************************************/
379 static int imageToPNG(const opj_image_t* image, const char* filename, int num_comp_select)
380 {
381   opj_image_cmptparm_t param_image_write;
382   opj_image_t* image_write = NULL;
383
384   param_image_write.x0 = 0;
385   param_image_write.y0 = 0;
386   param_image_write.dx = 0;
387   param_image_write.dy = 0;
388   param_image_write.h = image->comps[num_comp_select].h;
389   param_image_write.w = image->comps[num_comp_select].w;
390   param_image_write.bpp = image->comps[num_comp_select].bpp;
391   param_image_write.prec = image->comps[num_comp_select].prec;
392   param_image_write.sgnd = image->comps[num_comp_select].sgnd;
393
394   image_write = opj_image_create(1u, &param_image_write, OPJ_CLRSPC_GRAY);
395   memcpy(image_write->comps->data, image->comps[num_comp_select].data, param_image_write.h * param_image_write.w * sizeof(int));
396
397   imagetopng(image_write, filename);
398
399   opj_image_destroy(image_write);
400
401   return EXIT_SUCCESS;
402 }
403 #endif
404
405 typedef struct test_cmp_parameters
406 {
407   /**  */
408   char* base_filename;
409   /**  */
410   char* test_filename;
411   /** Number of components */
412   int nbcomp;
413   /**  */
414   double* tabMSEvalues;
415   /**  */
416   double* tabPEAKvalues;
417   /**  */
418   int nr_flag;
419   /**  */
420   char separator_base[2];
421   /**  */
422   char separator_test[2];
423
424 } test_cmp_parameters;
425
426 /* return decode format PGX / TIF / PPM , return -1 on error */
427 static int get_decod_format(test_cmp_parameters* param)
428 {
429   int base_format = get_decod_format_from_string( param->base_filename );
430   int test_format = get_decod_format_from_string( param->test_filename );
431   if( base_format != test_format ) return -1;
432   /* handle case -1: */
433   return base_format;
434 }
435
436 /*******************************************************************************
437  * Parse command line
438  *******************************************************************************/
439 static int parse_cmdline_cmp(int argc, char **argv, test_cmp_parameters* param)
440 {
441   char *MSElistvalues = NULL;  char *PEAKlistvalues= NULL;
442   char *separatorList = NULL;
443   size_t sizemembasefile, sizememtestfile;
444   int index, flagM=0, flagP=0;
445   const char optlist[] = "b:t:n:m:p:s:d";
446   int c;
447
448   /* Init parameters*/
449   param->base_filename = NULL;
450   param->test_filename = NULL;
451   param->nbcomp = 0;
452   param->tabMSEvalues = NULL;
453   param->tabPEAKvalues = NULL;
454   param->nr_flag = 0;
455   param->separator_base[0] = 0;
456   param->separator_test[0] = 0;
457
458   opj_opterr = 0;
459
460   while ((c = opj_getopt(argc, argv, optlist)) != -1)
461     switch (c)
462       {
463       case 'b':
464         sizemembasefile = strlen(opj_optarg) + 1;
465         param->base_filename = (char*) malloc(sizemembasefile);
466         strcpy(param->base_filename, opj_optarg);
467         /*printf("param->base_filename = %s [%d / %d]\n", param->base_filename, strlen(param->base_filename), sizemembasefile );*/
468         break;
469       case 't':
470         sizememtestfile = strlen(opj_optarg) + 1;
471         param->test_filename = (char*) malloc(sizememtestfile);
472         strcpy(param->test_filename, opj_optarg);
473         /*printf("param->test_filename = %s [%d / %d]\n", param->test_filename, strlen(param->test_filename), sizememtestfile);*/
474        break;
475       case 'n':
476         param->nbcomp = atoi(opj_optarg);
477         break;
478       case 'm':
479         MSElistvalues = opj_optarg;
480         flagM = 1;
481         break;
482       case 'p':
483         PEAKlistvalues = opj_optarg;
484         flagP = 1;
485         break;
486       case 'd':
487         param->nr_flag = 1;
488         break;
489       case 's':
490         separatorList = opj_optarg;
491         break;
492       case '?':
493         if ((opj_optopt == 'b') || (opj_optopt == 't') || (opj_optopt == 'n') || (opj_optopt == 'p') || (opj_optopt == 'm') || (opj_optopt
494             == 's'))
495           fprintf(stderr, "Option -%c requires an argument.\n", opj_optopt);
496         else
497           if (isprint(opj_optopt)) fprintf(stderr, "Unknown option `-%c'.\n", opj_optopt);
498           else fprintf(stderr, "Unknown option character `\\x%x'.\n", opj_optopt);
499         return 1;
500       default:
501         fprintf(stderr, "WARNING -> this option is not valid \"-%c %s\"\n", c, opj_optarg);
502         break;
503       }
504
505   if (opj_optind != argc)
506     {
507     for (index = opj_optind; index < argc; index++)
508       fprintf(stderr,"Non-option argument %s\n", argv[index]);
509     return 1;
510     }
511
512   if (param->nbcomp == 0)
513     {
514     fprintf(stderr,"Need to indicate the number of components !\n");
515     return 1;
516     }
517   /* else */
518   if ( flagM && flagP )
519     {
520     param->tabMSEvalues = parseToleranceValues( MSElistvalues, param->nbcomp);
521     param->tabPEAKvalues = parseToleranceValues( PEAKlistvalues, param->nbcomp);
522     if ( (param->tabMSEvalues == NULL) || (param->tabPEAKvalues == NULL))
523       {
524       fprintf(stderr,"MSE and PEAK values are not correct (respectively need %d values)\n",param->nbcomp);
525       return 1;
526       }
527     }
528
529   /* Get separators after corresponding letter (b or t)*/
530   if (separatorList != NULL)
531     {
532     if( (strlen(separatorList) ==2) || (strlen(separatorList) ==4) )
533       {
534       /* keep original string*/
535       size_t sizeseplist = strlen(separatorList)+1;
536       char* separatorList2 = (char*)malloc( sizeseplist );
537       strcpy(separatorList2, separatorList);
538       /*printf("separatorList2 = %s [%d / %d]\n", separatorList2, strlen(separatorList2), sizeseplist);*/
539
540       if (strlen(separatorList) == 2) /* one separator behind b or t*/
541         {
542         char *resultT = NULL;
543         resultT = strtok(separatorList2, "t");
544         if (strlen(resultT) == strlen(separatorList)) /* didn't find t character, try to find b*/
545           {
546           char *resultB = NULL;
547           resultB = strtok(resultT, "b");
548           if (strlen(resultB) == 1)
549             {
550             param->separator_base[0] = separatorList[1];
551             param->separator_base[1] = 0;
552             param->separator_test[0] = 0;
553             }
554           else /* not found b*/
555             {
556             free(separatorList2);
557             return 1;
558             }
559           }
560         else /* found t*/
561           {
562           param->separator_base[0] = 0;
563           param->separator_test[0] = separatorList[1];
564           param->separator_test[1] = 0;
565           }
566         /*printf("sep b = %s [%d] and sep t = %s [%d]\n",param->separator_base, strlen(param->separator_base), param->separator_test, strlen(param->separator_test) );*/
567         }
568       else /* == 4 characters we must found t and b*/
569         {
570         char *resultT = NULL;
571         resultT = strtok(separatorList2, "t");
572         if (strlen(resultT) == 3) /* found t in first place*/
573           {
574           char *resultB = NULL;
575           resultB = strtok(resultT, "b");
576           if (strlen(resultB) == 1) /* found b after t*/
577             {
578             param->separator_test[0] = separatorList[1];
579             param->separator_test[1] = 0;
580             param->separator_base[0] = separatorList[3];
581             param->separator_base[1] = 0;
582             }
583           else /* didn't find b after t*/
584             {
585             free(separatorList2);
586             return 1;
587             }
588           }
589         else /* == 2, didn't find t in first place*/
590           {
591           char *resultB = NULL;
592           resultB = strtok(resultT, "b");
593           if (strlen(resultB) == 1) /* found b in first place*/
594             {
595             param->separator_base[0] = separatorList[1];
596             param->separator_base[1] = 0;
597             param->separator_test[0] = separatorList[3];
598             param->separator_test[1] = 0;
599             }
600           else /* didn't found b in first place => problem*/
601             {
602             free(separatorList2);
603             return 1;
604             }
605           }
606         }
607       free(separatorList2);
608       }
609     else /* wrong number of argument after -s*/
610       {
611       return 1;
612       }
613     }
614   else
615     {
616     if (param->nbcomp == 1)
617       {
618       assert( param->separator_base[0] == 0 );
619       assert( param->separator_test[0] == 0 );
620       }
621     else
622       {
623       fprintf(stderr,"If number of component is > 1, we need separator\n");
624       return 1;
625       }
626     }
627
628
629   if ( (param->nr_flag) && (flagP || flagM) )
630     {
631     fprintf(stderr,"Wrong input parameters list: it is non-regression test or tolerance comparison\n");
632     return 1;
633     }
634   if ( (!param->nr_flag) && (!flagP || !flagM) )
635     {
636     fprintf(stderr,"Wrong input parameters list: it is non-regression test or tolerance comparison\n");
637     return 1;
638     }
639
640   return 0;
641 }
642
643 /*******************************************************************************
644  * MAIN
645  *******************************************************************************/
646 int main(int argc, char **argv)
647 {
648   test_cmp_parameters inParam;
649   OPJ_UINT32 it_comp, itpxl;
650   int failed = 1;
651   int nbFilenamePGXbase = 0, nbFilenamePGXtest = 0;
652   char *filenamePNGtest= NULL, *filenamePNGbase = NULL, *filenamePNGdiff = NULL;
653   size_t memsizebasefilename, memsizetestfilename;
654   size_t memsizedifffilename;
655   int valueDiff = 0, nbPixelDiff = 0;
656   double sumDiff = 0.0;
657   /* Structures to store image parameters and data*/
658   opj_image_t *imageBase = NULL, *imageTest = NULL, *imageDiff = NULL;
659   opj_image_cmptparm_t* param_image_diff = NULL;
660   int decod_format;
661
662   /* Get parameters from command line*/
663   if( parse_cmdline_cmp(argc, argv, &inParam) )
664     {
665     compare_images_help_display();
666     goto cleanup;
667     }
668
669   /* Display Parameters*/
670   printf("******Parameters********* \n");
671   printf(" base_filename = %s\n"
672          " test_filename = %s\n"
673          " nb of Components = %d\n"
674          " Non regression test = %d\n"
675          " separator Base = %s\n"
676          " separator Test = %s\n",
677          inParam.base_filename, inParam.test_filename, inParam.nbcomp,
678          inParam.nr_flag, inParam.separator_base, inParam.separator_test);
679
680   if ( (inParam.tabMSEvalues != NULL) && (inParam.tabPEAKvalues != NULL))
681     {
682     int it_comp2;
683     printf(" MSE values = [");
684     for (it_comp2 = 0; it_comp2 < inParam.nbcomp; it_comp2++)
685       printf(" %f ", inParam.tabMSEvalues[it_comp2]);
686     printf("]\n");
687     printf(" PEAK values = [");
688     for (it_comp2 = 0; it_comp2 < inParam.nbcomp; it_comp2++)
689       printf(" %f ", inParam.tabPEAKvalues[it_comp2]);
690     printf("]\n");
691     printf(" Non-regression test = %d\n", inParam.nr_flag);
692     }
693
694   if (strlen(inParam.separator_base) != 0)
695     nbFilenamePGXbase = inParam.nbcomp;
696
697   if (strlen(inParam.separator_test) != 0)
698     nbFilenamePGXtest = inParam.nbcomp;
699
700   printf(" NbFilename to generate from base filename = %d\n", nbFilenamePGXbase);
701   printf(" NbFilename to generate from test filename = %d\n", nbFilenamePGXtest);
702   printf("************************* \n");
703
704   /*----------BASELINE IMAGE--------*/
705   memsizebasefilename = strlen(inParam.test_filename) + 1 + 5 + 2 + 4;
706   memsizetestfilename = strlen(inParam.test_filename) + 1 + 5 + 2 + 4;
707
708   decod_format = get_decod_format(&inParam);
709   if( decod_format == -1 )
710     {
711     fprintf( stderr, "Unhandled file format\n" );
712     goto cleanup;
713     }
714   assert( decod_format == PGX_DFMT || decod_format == TIF_DFMT || decod_format == PXM_DFMT );
715
716   if( decod_format == PGX_DFMT )
717     {
718     imageBase = readImageFromFilePGX( inParam.base_filename, nbFilenamePGXbase, inParam.separator_base);
719     if ( imageBase == NULL )
720       goto cleanup;
721     }
722   else if( decod_format == TIF_DFMT )
723     {
724     imageBase = readImageFromFileTIF( inParam.base_filename, nbFilenamePGXbase, "");
725     if ( imageBase == NULL )
726       goto cleanup;
727     }
728   else if( decod_format == PXM_DFMT )
729     {
730     imageBase = readImageFromFilePPM( inParam.base_filename, nbFilenamePGXbase, inParam.separator_base);
731     if ( imageBase == NULL )
732       goto cleanup;
733     }
734
735   filenamePNGbase = (char*) malloc(memsizebasefilename);
736   strcpy(filenamePNGbase, inParam.test_filename);
737   strcat(filenamePNGbase, ".base");
738   /*printf("filenamePNGbase = %s [%d / %d octets]\n",filenamePNGbase, strlen(filenamePNGbase),memsizebasefilename );*/
739
740   /*----------TEST IMAGE--------*/
741
742   if( decod_format == PGX_DFMT )
743     {
744     imageTest = readImageFromFilePGX(inParam.test_filename, nbFilenamePGXtest, inParam.separator_test);
745     if ( imageTest == NULL )
746       goto cleanup;
747     }
748   else if( decod_format == TIF_DFMT )
749     {
750     imageTest = readImageFromFileTIF(inParam.test_filename, nbFilenamePGXtest, "");
751     if ( imageTest == NULL )
752       goto cleanup;
753     }
754   else if( decod_format == PXM_DFMT )
755     {
756     imageTest = readImageFromFilePPM(inParam.test_filename, nbFilenamePGXtest, inParam.separator_test);
757     if ( imageTest == NULL )
758       goto cleanup;
759     }
760
761   filenamePNGtest = (char*) malloc(memsizetestfilename);
762   strcpy(filenamePNGtest, inParam.test_filename);
763   strcat(filenamePNGtest, ".test");
764   /*printf("filenamePNGtest = %s [%d / %d octets]\n",filenamePNGtest, strlen(filenamePNGtest),memsizetestfilename );*/
765
766   /*----------DIFF IMAGE--------*/
767
768   /* Allocate memory*/
769   param_image_diff = malloc( imageBase->numcomps * sizeof(opj_image_cmptparm_t));
770
771   /* Comparison of header parameters*/
772   printf("Step 1 -> Header comparison\n");
773
774   /* check dimensions (issue 286)*/
775   if(imageBase->numcomps != imageTest->numcomps )
776     {
777     printf("ERROR: dim mismatch (%d><%d)\n", imageBase->numcomps, imageTest->numcomps);
778     goto cleanup;
779     }
780
781   for (it_comp = 0; it_comp < imageBase->numcomps; it_comp++)
782     {
783     param_image_diff[it_comp].x0 = 0;
784     param_image_diff[it_comp].y0 = 0;
785     param_image_diff[it_comp].dx = 0;
786     param_image_diff[it_comp].dy = 0;
787     param_image_diff[it_comp].sgnd = 0;
788     param_image_diff[it_comp].prec = 8;
789     param_image_diff[it_comp].bpp = 1;
790     param_image_diff[it_comp].h = imageBase->comps[it_comp].h;
791     param_image_diff[it_comp].w = imageBase->comps[it_comp].w;
792
793     if (imageBase->comps[it_comp].sgnd != imageTest->comps[it_comp].sgnd)
794       {
795       printf("ERROR: sign mismatch [comp %d] (%d><%d)\n", it_comp, ((imageBase->comps)[it_comp]).sgnd, ((imageTest->comps)[it_comp]).sgnd);
796       goto cleanup;
797       }
798
799     if (((imageBase->comps)[it_comp]).prec != ((imageTest->comps)[it_comp]).prec)
800       {
801       printf("ERROR: prec mismatch [comp %d] (%d><%d)\n", it_comp, ((imageBase->comps)[it_comp]).prec, ((imageTest->comps)[it_comp]).prec);
802       goto cleanup;
803       }
804
805     if (((imageBase->comps)[it_comp]).bpp != ((imageTest->comps)[it_comp]).bpp)
806       {
807       printf("ERROR: byte per pixel mismatch [comp %d] (%d><%d)\n", it_comp, ((imageBase->comps)[it_comp]).bpp, ((imageTest->comps)[it_comp]).bpp);
808       goto cleanup;
809       }
810
811     if (((imageBase->comps)[it_comp]).h != ((imageTest->comps)[it_comp]).h)
812       {
813       printf("ERROR: height mismatch [comp %d] (%d><%d)\n", it_comp, ((imageBase->comps)[it_comp]).h, ((imageTest->comps)[it_comp]).h);
814       goto cleanup;
815       }
816
817     if (((imageBase->comps)[it_comp]).w != ((imageTest->comps)[it_comp]).w)
818       {
819       printf("ERROR: width mismatch [comp %d] (%d><%d)\n", it_comp, ((imageBase->comps)[it_comp]).w, ((imageTest->comps)[it_comp]).w);
820       goto cleanup;
821       }
822     }
823
824    imageDiff = opj_image_create(imageBase->numcomps, param_image_diff, OPJ_CLRSPC_UNSPECIFIED);
825    /* Free memory*/
826    free(param_image_diff); param_image_diff = NULL;
827
828    /* Measurement computation*/
829    printf("Step 2 -> measurement comparison\n");
830
831    memsizedifffilename = strlen(inParam.test_filename) + 1 + 5 + 2 + 4;
832    filenamePNGdiff = (char*) malloc(memsizedifffilename);
833    strcpy(filenamePNGdiff, inParam.test_filename);
834    strcat(filenamePNGdiff, ".diff");
835    /*printf("filenamePNGdiff = %s [%d / %d octets]\n",filenamePNGdiff, strlen(filenamePNGdiff),memsizedifffilename );*/
836
837    /* Compute pixel diff*/
838    for (it_comp = 0; it_comp < imageDiff->numcomps; it_comp++)
839      {
840      double SE=0,PEAK=0;
841      double MSE=0;
842      for (itpxl = 0; itpxl < ((imageDiff->comps)[it_comp]).w * ((imageDiff->comps)[it_comp]).h; itpxl++)
843        {
844        if (abs( ((imageBase->comps)[it_comp]).data[itpxl] - ((imageTest->comps)[it_comp]).data[itpxl] ) > 0)
845          {
846          valueDiff = ((imageBase->comps)[it_comp]).data[itpxl] - ((imageTest->comps)[it_comp]).data[itpxl];
847          ((imageDiff->comps)[it_comp]).data[itpxl] = abs(valueDiff);
848          sumDiff += valueDiff;
849          nbPixelDiff++;
850
851          SE += (double)valueDiff * valueDiff;
852          PEAK = (PEAK > abs(valueDiff)) ? PEAK : abs(valueDiff);
853          }
854        else
855          ((imageDiff->comps)[it_comp]).data[itpxl] = 0;
856        }/* h*w loop */
857
858      MSE = SE / ( ((imageDiff->comps)[it_comp]).w * ((imageDiff->comps)[it_comp]).h );
859
860      if (!inParam.nr_flag && (inParam.tabMSEvalues != NULL) && (inParam.tabPEAKvalues != NULL))
861        { /* Conformance test*/
862        printf("<DartMeasurement name=\"PEAK_%d\" type=\"numeric/double\"> %f </DartMeasurement> \n", it_comp, PEAK);
863        printf("<DartMeasurement name=\"MSE_%d\" type=\"numeric/double\"> %f </DartMeasurement> \n", it_comp, MSE);
864
865        if ( (MSE > inParam.tabMSEvalues[it_comp]) || (PEAK > inParam.tabPEAKvalues[it_comp]) )
866          {
867          printf("ERROR: MSE (%f) or PEAK (%f) values produced by the decoded file are greater "
868            "than the allowable error (respectively %f and %f) \n",
869            MSE, PEAK, inParam.tabMSEvalues[it_comp], inParam.tabPEAKvalues[it_comp]);
870          goto cleanup;
871          }
872        }
873      else  /* Non regression-test */
874        {
875        if ( nbPixelDiff > 0)
876          {
877          char it_compc[255];
878          it_compc[0] = 0;
879
880          printf("<DartMeasurement name=\"NumberOfPixelsWithDifferences_%d\" type=\"numeric/int\"> %d </DartMeasurement> \n", it_comp, nbPixelDiff);
881          printf("<DartMeasurement name=\"ComponentError_%d\" type=\"numeric/double\"> %f </DartMeasurement> \n", it_comp, sumDiff);
882
883 #ifdef OPJ_HAVE_LIBPNG
884            {
885            char *filenamePNGbase_it_comp, *filenamePNGtest_it_comp, *filenamePNGdiff_it_comp;
886
887            filenamePNGbase_it_comp = (char*) malloc(memsizebasefilename);
888            strcpy(filenamePNGbase_it_comp,filenamePNGbase);
889
890            filenamePNGtest_it_comp = (char*) malloc(memsizetestfilename);
891            strcpy(filenamePNGtest_it_comp,filenamePNGtest);
892
893            filenamePNGdiff_it_comp = (char*) malloc(memsizedifffilename);
894            strcpy(filenamePNGdiff_it_comp,filenamePNGdiff);
895
896            sprintf(it_compc, "_%i", it_comp);
897            strcat(it_compc,".png");
898            strcat(filenamePNGbase_it_comp, it_compc);
899            /*printf("filenamePNGbase_it = %s [%d / %d octets]\n",filenamePNGbase_it_comp, strlen(filenamePNGbase_it_comp),memsizebasefilename );*/
900            strcat(filenamePNGtest_it_comp, it_compc);
901            /*printf("filenamePNGtest_it = %s [%d / %d octets]\n",filenamePNGtest_it_comp, strlen(filenamePNGtest_it_comp),memsizetestfilename );*/
902            strcat(filenamePNGdiff_it_comp, it_compc);
903            /*printf("filenamePNGdiff_it = %s [%d / %d octets]\n",filenamePNGdiff_it_comp, strlen(filenamePNGdiff_it_comp),memsizedifffilename );*/
904
905            /*
906            if ( imageToPNG(imageBase, filenamePNGbase_it_comp, it_comp) == EXIT_SUCCESS )
907            {
908            printf("<DartMeasurementFile name=\"BaselineImage_%d\" type=\"image/png\"> %s </DartMeasurementFile> \n", it_comp, filenamePNGbase_it_comp);
909            }
910
911            if ( imageToPNG(imageTest, filenamePNGtest_it_comp, it_comp) == EXIT_SUCCESS )
912            {
913            printf("<DartMeasurementFile name=\"TestImage_%d\" type=\"image/png\"> %s </DartMeasurementFile> \n", it_comp, filenamePNGtest_it_comp);
914            }
915
916            if ( imageToPNG(imageDiff, filenamePNGdiff_it_comp, it_comp) == EXIT_SUCCESS )
917            {
918            printf("<DartMeasurementFile name=\"DiffferenceImage_%d\" type=\"image/png\"> %s </DartMeasurementFile> \n", it_comp, filenamePNGdiff_it_comp);
919            }
920             */
921
922            free(filenamePNGbase_it_comp);
923            free(filenamePNGtest_it_comp);
924            free(filenamePNGdiff_it_comp);
925            }
926 #endif
927          goto cleanup;
928          }
929        }
930      } /* it_comp loop */
931
932    printf("---- TEST SUCCEED ----\n");
933    failed = 0;
934 cleanup:
935   /*-----------------------------*/
936   free(param_image_diff);
937   /* Free memory */
938   opj_image_destroy(imageBase);
939   opj_image_destroy(imageTest);
940   opj_image_destroy(imageDiff);
941
942   free(filenamePNGbase);
943   free(filenamePNGtest);
944   free(filenamePNGdiff);
945
946   free(inParam.tabMSEvalues);
947   free(inParam.tabPEAKvalues);
948   free(inParam.base_filename);
949   free(inParam.test_filename);
950
951   return failed ? EXIT_FAILURE : EXIT_SUCCESS;
952 }