Merge pull request #191 from h2020charisma/spc-multimap

Add support for SPC multidimentional maps

src/ramanchada2/spectrum/__init__.py

@@ -14,5 +14,6 @@
 from .creators.from_theoretical_lines import from_theoretical_lines
 from .creators.hdr_from_multi_exposure import hdr_from_multi_exposure
 from .filters import *  # noqa
+from .multimap import *  # noqa
 from .peaks import *  # noqa
 from .spectrum import Spectrum  # noqa

src/ramanchada2/spectrum/multimap/__init__.py

@@ -0,0 +1 @@
+from .spc import read_map_spc

src/ramanchada2/spectrum/multimap/spc.py

@@ -0,0 +1,24 @@
+from collections import namedtuple
+from typing import Dict
+
+import spc_io
+
+from ...misc.types import SpeMetadataModel
+from ..spectrum import Spectrum
+
+SPCMapCoordinates = namedtuple('SPCMapCoordinates', ['z', 'w'])
+
+
+def read_map_spc(filename: str) -> Dict[SPCMapCoordinates, Spectrum]:
+    spc = spc_io.SPC.from_bytes_io(open(filename, 'rb'))
+
+    ret = dict()
+
+    spc_meta = {k.strip(): v.strip() for k, v in spc.log_book.text.items()}
+    for meas in spc:
+        spe_meta = {}
+        spe_meta.update(spc_meta)
+        spe_meta.update(dict(w=meas.w, z=meas.z))
+        ret[SPCMapCoordinates(w=meas.w, z=meas.z)] = Spectrum(x=meas.xarray, y=meas.yarray,
+                                                              metadata=SpeMetadataModel.model_validate(spe_meta))
+    return ret