FEAT: implemented option to only read ms1 or ms2 frames for readablef…

…rames trait
MannLabs · Oct 19, 2023 · 771e18f · 771e18f
1 parent ffa603b
commit 771e18f
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/src/file_readers/frame_readers.rs b/src/file_readers/frame_readers.rs
@@ -10,6 +10,10 @@ pub trait ReadableFrames {
     fn read_single_frame(&self, index: usize) -> Frame;
 
     fn read_all_frames(&self) -> Vec<Frame>;
+
+    fn read_ms1_frames(&self) -> Vec<Frame>;
+
+    fn read_ms2_frames(&self) -> Vec<Frame>;
 }
 
 impl FileFormat {
@@ -35,4 +39,12 @@ impl ReadableFrames for FileFormat {
     fn read_all_frames(&self) -> Vec<Frame> {
         self.unwrap_frame_reader().read_all_frames()
     }
+
+    fn read_ms1_frames(&self) -> Vec<Frame> {
+        self.unwrap_frame_reader().read_ms1_frames()
+    }
+
+    fn read_ms2_frames(&self) -> Vec<Frame> {
+        self.unwrap_frame_reader().read_ms2_frames()
+    }
 }
diff --git a/src/file_readers/frame_readers/tdf_reader.rs b/src/file_readers/frame_readers/tdf_reader.rs
@@ -88,4 +88,24 @@ impl ReadableFrames for TDFReader {
             .map(|index| self.read_single_frame(index))
             .collect()
     }
+
+    fn read_ms1_frames(&self) -> Vec<Frame> {
+        (0..self.tdf_bin_reader.size())
+            .into_par_iter()
+            .map(|index| match self.frame_types[index] {
+                FrameType::MS1 => self.read_single_frame(index),
+                _ => Frame::default(),
+            })
+            .collect()
+    }
+
+    fn read_ms2_frames(&self) -> Vec<Frame> {
+        (0..self.tdf_bin_reader.size())
+            .into_par_iter()
+            .map(|index| match self.frame_types[index] {
+                FrameType::MS2(_) => self.read_single_frame(index),
+                _ => Frame::default(),
+            })
+            .collect()
+    }
 }