add from hdf method to spectrum class

.mailmap

@@ -119,6 +119,8 @@ Josh Shields <shields.joshua.v@gmail.com>
 Karan Desai <karandesai_96@live.com>
 Karan Desai <karandesai_96@live.com> karandesai-96 <karandesai_96@live.com>
 
+Karthik Rishinarada <karthikrk11135@gmail.com>
+
 Kaushik Varanasi <kaushik.varanasi1@gmail.com>
 Kaushik Varanasi <kaushik.varanasi1@gmail.com> kaushik94 <kaushik.varanasi1@gmail.com>
 

tardis/spectrum/spectrum.py

@@ -1,6 +1,7 @@
 import warnings
 
 import numpy as np
+import pandas as pd
 from astropy import units as u
 
 from tardis.io.util import HDFWriterMixin
@@ -151,6 +152,20 @@ def plot(self, ax=None, mode="wavelength", **kwargs):
         else:
             warnings.warn(f"Did not find plotting mode {mode}, doing nothing.")
 
+    def from_hdf_to_class(self,hdf_path):
+        try:
+            with pd.HDFStore(hdf_path, mode="r") as store:
+                _freq = store['/tardis_spectrum/_frequency']
+                _f = np.array(_freq) * u.Hz
+
+                _lumi = store['/tardis_spectrum/luminosity']
+                _l = np.array(_lumi) * (u.erg/u.s)
+
+                return TARDISSpectrum(_f, _l)
+
+        except FileNotFoundError as e:
+            print(f"File not found: {e}")
+
     def to_ascii(self, fname, mode="luminosity_density"):
         if mode == "luminosity_density":
             np.savetxt(

tardis/spectrum/tests/test_spectrum.py

@@ -135,6 +135,14 @@ def test_f_nu_to_f_lambda(spectrum):
         spectrum.luminosity_density_lambda.value, expected.value
     )
 
+
+def test_from_hdf_to_class(spectrum):
+    actual = TARDISSpectrum(spectrum._frequency, spectrum.luminosity)
+    actual.to_hdf("test.hdf", overwrite=True)
+    expected = actual.from_hdf_to_class("test.hdf")
+
+    compare_spectra(actual, expected)
+
 
 ###
 # Save and Load