Fixes graphicatorx

npazosmendez · Nov 23, 2018 · 923909d · 923909d
1 parent b53d740
commit 923909d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/apps/inpainter.cpp b/apps/inpainter.cpp
@@ -302,7 +302,7 @@ void paint_debug(Mat img) {
         debug_data data = ((debug_data *) debug)[LINEAR(i, j)];
 
         // Nt
-        if (data.normal_t.x * data.normal_t.x + data.normal_t.y * data.normal_t.y > 1e-3) {
+        if (data.normal_t.x != FLT_MAX) {
             line(
                     img,                                                     // mat to draw into
                     Point(j, i),                                             // origin position
@@ -319,7 +319,7 @@ void paint_debug(Mat img) {
         debug_data data = ((debug_data *) debug)[LINEAR(i, j)];
 
         // Gt
-        if (data.gradient_t.x * data.gradient_t.x + data.gradient_t.y * data.gradient_t.y > 1e-3) {
+        if (data.gradient_t.x != FLT_MAX) {
             line(
                     img,                                                     // mat to draw into
                     Point(j, i),                                             // origin position
@@ -334,18 +334,17 @@ void paint_debug(Mat img) {
     forn(i, height) forn(j, width) {
         debug_data data = ((debug_data *) debug)[LINEAR(i, j)];
 
-        if (abs(data.confidence - 1) > 1e-3) {
-            img.at<Vec3b>(Point(j, i)) = Vec3b(0, (int)data.confidence, (int)data.confidence);
+        if (data.confidence != FLT_MAX) {
+            img.at<Vec3b>(Point(j, i)) = Vec3b(0, (int)(data.confidence * 255), (int)(data.confidence * 255));
         }
     }
 
     /*
     forn(i, height) forn(j, width) {
         debug_data data = ((debug_data *) debug)[LINEAR(i, j)];
 
-        int scaled_data = ((int) (data.data * 255));
-
-        if (data.data != 0) {
+        if (data.data != FLT_MAX) {
+            int scaled_data = ((int) (data.data * 255));
             img.at<Vec3b>(Point(j, i)) = Vec3b(scaled_data, scaled_data, 0);
         }
     }

diff --git a/c/inpainting.c b/c/inpainting.c
@@ -163,6 +163,11 @@ bool inpaint_step(int width, int height, char * img, bool * mask, int * debug) {
         for (int j = 0; j < width; j++) {
 
             if (!contour_mask[LINEAR(i,j)]) {
+                debug_data * casted_debug = (debug_data *) debug;
+                casted_debug[LINEAR(i,j)].confidence = FLT_MAX;
+                casted_debug[LINEAR(i,j)].data = FLT_MAX;
+                casted_debug[LINEAR(i,j)].gradient_t = (point) {FLT_MAX, FLT_MAX};
+                casted_debug[LINEAR(i,j)].normal_t = (point) {FLT_MAX, FLT_MAX};
                 continue;
             }