support reading from csv

wilhelm-lab · Oct 2, 2023 · 1b4e47c · 1b4e47c
1 parent 0467c63
commit 1b4e47c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/oktoberfest/data/spectra.py b/oktoberfest/data/spectra.py
@@ -211,9 +211,22 @@ def from_hdf5(cls: Type[SpectraT], input_file: Union[str, Path]) -> SpectraT:
         """
         input_file = str(input_file)
         spectra = cls()
-        spectra.spectra_data
         spectra.add_columns(hdf5.read_file(input_file, hdf5.META_DATA_KEY))
         spectra.add_matrix_from_hdf5(hdf5.read_file(input_file, f"sparse_{hdf5.INTENSITY_RAW_KEY}"), FragmentType.RAW)
         spectra.add_matrix_from_hdf5(hdf5.read_file(input_file, f"sparse_{hdf5.MZ_RAW_KEY}"), FragmentType.MZ)
 
         return spectra
+
+    @classmethod
+    def from_csv(cls: Type[SpectraT], input_file: Union[str, Path]) -> SpectraT:
+        """
+        Read from hdf5 file.
+
+        :param input_file: path to input file
+        :return: a spectra instance
+        """
+        input_file = str(input_file)
+        spectra = cls()
+        spectra.spectra_data = pd.read_csv(input_file)
+
+        return spectra