Add spectrum_axis configuration for SpectrumTemplateSource

alea/examples/configs/unbinned_wimp_statistical_model_template_source_test.yaml

@@ -126,4 +126,5 @@ likelihood_config:
           - signal_efficiency
         template_filename: wimp50gev_template.ii.h5
         spectrum_name: test_cs1_spectrum.json
+        spectrum_axis: 1
         efficiency_name: signal_efficiency

alea/template_source.py

@@ -373,14 +373,14 @@ def build_histogram(self):
 
         if "spectrum_name" not in self.config:
             raise ValueError("spectrum_name not in config")
+        if "spectrum_axis" not in self.config:
+            raise ValueError("spectrum_axis not in config")
         spectrum = self._get_json_spectrum(
             self.config["spectrum_name"].format(**self.format_named_parameters)
         )
 
-        # Perform scaling, the first axis is assumed to be reweighted
         # The spectrum is assumed to be probability density (in per the unit of first axis).
-        axis = 0
-        # h = h.normalize(axis=axis)
+        axis = self.config["spectrum_axis"]
         bin_edges = h.bin_edges[axis]
         bin_centers = h.bin_centers(axis=axis)
         slices = [None] * h.histogram.ndim

alea/utils.py

@@ -206,6 +206,12 @@ def dump_yaml(file_name: str, data: dict):
         yaml.safe_dump(data, file)
 
 
+def dump_json(file_name: str, data: dict):
+    """Dump data to a json file."""
+    with open(file_name, "w") as file:
+        json.dump(data, file, indent=4)
+
+
 def _get_abspath(file_name):
     """Get the abspath of the file.