wums 0.1.4__tar.gz → 0.1.5__tar.gz

{wums-0.1.4 → wums-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wums
-Version: 0.1.4
+Version: 0.1.5
 Summary: .
 Author-email: David Walter <david.walter@cern.ch>, Josh Bendavid <josh.bendavid@cern.ch>, Kenneth Long <kenneth.long@cern.ch>
 License: MIT

{wums-0.1.4 → wums-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wums"
-version = "0.1.4"
+version = "0.1.5"
 description = "."
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { text = "MIT" }

{wums-0.1.4 → wums-0.1.5}/wums/ioutils.py

@@ -1,5 +1,9 @@
 import copyreg
 import pickle
+import sys
+
+import wums
+sys.modules['narf.ioutils'] = sys.modules['wums.ioutils'] # backwards compatibility to use old files
 
 import boost_histogram as bh
 import h5py

{wums-0.1.4 → wums-0.1.5}/wums/plot_tools.py

@@ -6,6 +6,7 @@ import shutil
 import socket
 import sys
 import textwrap
+import importlib
 
 import hist
 import matplotlib as mpl
@@ -1780,3 +1781,44 @@ def make_summary_plot(
     ax1.xaxis.set_major_locator(ticker.LinearLocator(numticks=5))
 
     return fig
+
+
+def load_config(config_path):
+    if config_path is None:
+        return {}
+    # load a python module
+    spec = importlib.util.spec_from_file_location("config", config_path)
+    config = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(config)
+    return config
+
+
+def read_axis_label(x, labels, with_unit=True):
+    if x in labels:
+        label = labels[x]
+        if isinstance(label, str):
+            return label
+        elif with_unit:
+            return f'{label["label"]} ({label["unit"]})'
+        else:
+            return label["label"]
+    else:
+        return x
+
+
+def get_axis_label(config, default_keys=None, label=None, is_bin=False):
+    if label is not None:
+        return label
+
+    if default_keys is None:
+        return "Bin index"
+
+    labels = getattr(config, "axis_labels", {})
+
+    if len(default_keys) == 1:        
+        if is_bin:
+            return f"{read_axis_label(default_keys[0], labels, False)} bin"
+        else:
+            return read_axis_label(default_keys[0], labels)
+    else:
+        return f"({', '.join([read_axis_label(a, labels, False) for a in default_keys])}) bin"

{wums-0.1.4 → wums-0.1.5}/wums.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wums
-Version: 0.1.4
+Version: 0.1.5
 Summary: .
 Author-email: David Walter <david.walter@cern.ch>, Josh Bendavid <josh.bendavid@cern.ch>, Kenneth Long <kenneth.long@cern.ch>
 License: MIT