diff --git a/src/main.py b/src/main.py index 5832970..9577305 100644 --- a/src/main.py +++ b/src/main.py @@ -688,8 +688,10 @@ def build_nexus_file(pld_fabrication, output_path, rheed_osc=None, heatmap_matri heatmap.attrs["units"] = "a.u." # this is of my own initiative. good??? heatmap.attrs["interpretation"] = "spectrum" - # suggested by DeepSeek, useful? - heatmap.attrs["suggested_colormap"] = "viridis" + # suggested by DeepSeek, useful? probably not. + heatmap.attrs["suggested_colormap"] = "inferno" + heatmap.attrs["scale_min"] = 0.0 + heatmap.attrs["scale_max"] = 1.0 return @@ -731,10 +733,10 @@ if __name__ == "__main__": # if os.path.isfile("tests/LAO_16min50s_736C_STO.png"): # if PNG img = Image.open("tests/LAO_16min50s_736C_STO.bmp").convert("L") mx = np.array(img, dtype=np.uint8) - normalized_mx = mx.astype(np.float32) / 255.0 # consider deleting??? + mx = mx.astype(np.float32) / 255.0 # consider deleting??? build_nexus_file( result, output_path=f"output/sample-{sample_name}-nexus.h5", rheed_osc=rheed_osc, - heatmap_matrix=normalized_mx, + heatmap_matrix=mx, )