padding -0 instead of 0 when saving dref with mua_float medium

fangq · Aug 5, 2023 · da0beda · da0beda
1 parent b781036
commit da0beda
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/mcx_utils.c b/src/mcx_utils.c
@@ -3206,7 +3206,7 @@ void mcx_loadvolume(char* filename, Config* cfg, int isbuf) {
                 f2i.f = EPS;
             }
 
-            if (val[i] != val[i]) { /*if input is nan in continuous medium, convert to 0-voxel*/
+            if (val[i] != val[i] || f2i.i == SIGN_BIT) { /*if input is nan in continuous medium, convert to 0-voxel*/
                 f2i.i = 0;
             }
 
@@ -3271,6 +3271,10 @@ void mcx_loadvolume(char* filename, Config* cfg, int isbuf) {
                 f2h.i[0] = 0x00010000;
             }
 
+            if (f2h.i[0] == SIGN_BIT) { /*avoid being detected as a 0-label voxel, setting mus=EPS_fp16*/
+                f2h.i[0] = 0;
+            }
+
             cfg->vol[i] = f2h.i[0];
         }
     } else if (cfg->mediabyte == MEDIA_2LABEL_SPLIT) {

diff --git a/src/mcxlab.cpp b/src/mcxlab.cpp
@@ -665,7 +665,7 @@ void mcx_set_field(const mxArray* root, const mxArray* item, int idx, Config* cf
                         f2i.f = EPS;
                     }
 
-                    if (val[i] != val[i]) { /*if input is nan in continuous medium, convert to 0-voxel*/
+                    if (val[i] != val[i] || f2i.i == SIGN_BIT) { /*if input is nan in continuous medium, convert to 0-voxel*/
                         f2i.i = 0;
                     }
 
@@ -730,6 +730,10 @@ void mcx_set_field(const mxArray* root, const mxArray* item, int idx, Config* cf
                         f2h.i[0] = 0x00010000;
                     }
 
+                    if (f2h.i[0] == SIGN_BIT) { /*avoid being detected as a 0-label voxel, setting mus=EPS_fp16*/
+                        f2h.i[0] = 0;
+                    }
+
                     cfg->vol[i] = f2h.i[0];
                 }
             } else if (cfg->mediabyte == MEDIA_LABEL_HALF) {