updates to visualize script

DLR-MF-DAS · Dec 5, 2024 · 3792fd1 · 3792fd1
1 parent 59bd425
commit 3792fd1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/methane_super_emitters/show.py b/src/methane_super_emitters/show.py
@@ -4,10 +4,15 @@
 
 @click.command()
 @click.option('-i', '--input-file', help='Input NPZ file')
-def main(input_file):
+@click.option('--vmin', help='Minimum value for methane concentration')
+@click.option('--vmax', help='Maximum value for methane concentration')
+@click.option('--histogram', help='Show histogram', is_flag=True)
+def main(input_file, vmin, vmax, histogram):
     data = np.load(input_file)
-    import pdb; pdb.set_trace()
-    plt.imshow(data['methane'])
+    if histogram:
+        plt.hist(data['methane'], bins=50)
+        plt.show()
+    plt.imshow(data['methane'], vmin=int(vmin), vmax=int(vmax))
     plt.show()
 
 if __name__ == '__main__':