WIP: added novisim as metadata flag for h5 files

src/lib/py/esrf_read.py

@@ -36,7 +36,8 @@ def esrf_edf_metadata(filename):
             if (len(kv) >= 2):
                 meta[kv[0].strip()] = kv[1].strip()
 
-        assert meta["ByteOrder"] == "LowByteFirst"
+        # removing " ;"
+        assert meta["ByteOrder"].split()[0] == "LowByteFirst"
 
         if (meta["DataType"] == "UnsignedShort"):
             meta["NumpyType"] = np.uint16

src/processing_steps/0200_generate_byte_hdf5.py

@@ -71,6 +71,8 @@
 
     # Store metadata in both files for each subvolume scan
     for h5file in [h5file_msb,h5file_lsb]:
+        # never a novisim tomogram at this point, but added to all *.h5 files for later checks
+        h5file.attrs["novisim"] = 0
         grp_meta = h5file.create_group("metadata")
         for i in range(len(subvolume_metadata)):
             subvolume_info = subvolume_metadata[i]

src/processing_steps/0225_generate_hdf5_novisim.py

@@ -112,6 +112,7 @@
     h5_msb = h5py.File(f'{output_dir}/msb/{args.sample}.h5', 'w')
 
     for h5, tomo in [(h5_lsb, tomo_lsb), (h5_msb, tomo_msb)]:
+        h5.attrs["novisim"] = 1 # set flag to mark file as novisim tomogram
         h5.create_dataset('subvolume_dimensions', (1,3), data=[[nz, ny, nx]], dtype=np.uint16)
         h5.create_dataset('subvolume_range', (1,2,), dtype=np.float32, data=np.array([0, 65535]))
         h5_tomo = h5.create_dataset('voxels', (nz, ny, nx), dtype=np.uint8, data=tomo)

src/processing_steps/0600_segment_implant_cc.py

@@ -42,7 +42,9 @@
     global_vmax = np.max(h5meta['subvolume_range'][:,1])
     values      = np.linspace(global_vmin,global_vmax,2**16)
     implant_threshold_u16 = np.argmin(np.abs(values-implant_threshold))
-    if 'novisim' in args.sample:
+
+    # check if chosen sample has novisim flag set in metadata
+    if h5meta["novisim"]:
         implant_threshold_u16 = implant_threshold_u16_novisim
 
     if args.verbose >= 2: print(f"""