align signature size

[libucw.git] / images / image-sim-test.c

diff --git a/images/image-sim-test.c b/images/image-sim-test.c
index c58c9d08ed8ed1d399ec63c9e52af3440a780f87..bac384ec28a673113effe453b46621e08fda9664 100644 (file)
--- a/images/image-sim-test.c
+++ b/images/image-sim-test.c
@@ -31,13 +31,13 @@ Usage: image-sim-test [options] image1 [image2] \n\
 -f --format-1        image1 format (jpeg, gif, png)\n\
 -F --format-2        image2 format\n\
 -g --background      background color (hexadecimal RRGGBB)\n\
--s --segmentation-1  writes image1 segmentation to given file\n\
--S --segmentation-2  writes image2 segmentation to given file\n\
+-r --segmentation-1  writes image1 segmentation to given file\n\
+-R --segmentation-2  writes image2 segmentation to given file\n\
 ", stderr);
   exit(1);
 }
 
-static char *shortopts = "qf:F:g:t:s:S:" CF_SHORT_OPTS;
+static char *shortopts = "qf:F:g:t:r:R:" CF_SHORT_OPTS;
 static struct option longopts[] =
 {
   CF_LONG_OPTS
@@ -45,8 +45,8 @@ static struct option longopts[] =
   { "format-1",                0, 0, 'f' },
   { "format-2",                0, 0, 'F' },
   { "background",      0, 0, 'g' },
-  { "segmentation-1",  0, 0, 's' },
-  { "segmentation-2",  0, 0, 'S' },
+  { "segmentation-1",  0, 0, 'r' },
+  { "segmentation-2",  0, 0, 'R' },
   { NULL,              0, 0, 0 }
 };
 
@@ -62,6 +62,12 @@ static byte *segmentation_name_2;
 #define MSG(x...) do{ if (verbose) log(L_INFO, ##x); }while(0)
 #define TRY(x) do{ if (!(x)) die("Error: %s", it.err_msg); }while(0)
 
+static void
+msg_str(byte *s, void *param UNUSED)
+{
+  MSG("%s", s);
+}
+
 static void
 dump_signature(struct image_signature *sig)
 {
@@ -91,10 +97,15 @@ write_segmentation(struct image_sig_data *data, byte *fn)
   for (uns i = 0; i < data->regions_count; i++)
     {
       byte c[3];
-      // FIXME: convert from Luv to RGB
-      c[0] = data->regions[i].a[0];
-      c[1] = data->regions[i].a[1];
-      c[2] = data->regions[i].a[2];
+      double luv[3], xyz[3], srgb[3];
+      luv[0] = data->regions[i].a[0] * (4 / 2.55);
+      luv[1] = ((int)data->regions[i].a[1] - 128) * (4 / 2.55);
+      luv[2] = ((int)data->regions[i].a[2] - 128) * (4 / 2.55);
+      luv_to_xyz_slow(xyz, luv);
+      xyz_to_srgb_slow(srgb, xyz);
+      c[0] = CLAMP(srgb[0] * 255, 0, 255); 
+      c[1] = CLAMP(srgb[1] * 255, 0, 255); 
+      c[2] = CLAMP(srgb[2] * 255, 0, 255); 
       for (struct image_sig_block *block = data->regions[i].blocks; block; block = block->next)
         {
          uns x1 = block->x * 4;
@@ -156,10 +167,10 @@ main(int argc, char **argv)
            color_make_rgb(&background_color, (v >> 16) & 255, (v >> 8) & 255, v & 255);
          }
          break;
-       case 's':
+       case 'r':
          segmentation_name_1 = optarg;
          break;
-       case 'S':
+       case 'R':
          segmentation_name_2 = optarg;
          break;
        default:
@@ -230,7 +241,10 @@ main(int argc, char **argv)
       TRY(image_sig_init(&it, &data, img1));
       image_sig_preprocess(&data);
       if (data.valid)
-       image_sig_segmentation(&data);
+        {
+         image_sig_segmentation(&data);
+         image_sig_detect_textured(&data);
+       }
       if (segmentation_name_1)
        write_segmentation(&data, segmentation_name_1);
       image_sig_finish(&data, &sig1);
@@ -243,7 +257,10 @@ main(int argc, char **argv)
       TRY(image_sig_init(&it, &data, img2));
       image_sig_preprocess(&data);
       if (data.valid)
-       image_sig_segmentation(&data);
+        {
+         image_sig_segmentation(&data);
+         image_sig_detect_textured(&data);
+       }
       if (segmentation_name_2)
        write_segmentation(&data, segmentation_name_2);
       image_sig_finish(&data, &sig2);
@@ -253,8 +270,16 @@ main(int argc, char **argv)
 
   if (img1 && img2)
     {
-      uns dist = image_signatures_dist(&sig1, &sig2);
-      MSG("dist=%.6f", dist / (double)(1 << IMAGE_SIG_DIST_SCALE));
+      uns dist;
+      if (verbose)
+        {
+          struct fastbuf *fb = bfdopen(0, 4096);
+          dist = image_signatures_dist_explain(&sig1, &sig2, msg_str, NULL);
+          bclose(fb);
+       }
+      else
+       dist = image_signatures_dist(&sig1, &sig2);
+      MSG("dist=%u", dist);
     }
 
   if (img1)