FEAT: using iterator to read frames

MannLabs · Jun 14, 2024 · d4de769 · d4de769
1 parent e2a145f
commit d4de769
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/file_readers/frame_readers/tdf_reader.rs b/src/file_readers/frame_readers/tdf_reader.rs
@@ -7,6 +7,7 @@ use crate::{
     io::readers::frame_reader::FrameReader,
     ms_data::Frame,
 };
+use rayon::iter::ParallelIterator;
 
 #[derive(Debug)]
 pub struct TDFReader {
@@ -41,14 +42,18 @@ impl ReadableFrames for TDFReader {
     }
 
     fn read_all_frames(&self) -> Vec<Frame> {
-        self.frame_reader.collect(|_| true)
+        self.frame_reader.parallel_filter(|_| true).collect()
     }
 
     fn read_all_ms1_frames(&self) -> Vec<Frame> {
-        self.frame_reader.collect(|x| x.msms_type == 0)
+        self.frame_reader
+            .parallel_filter(|x| x.msms_type == 0)
+            .collect()
     }
 
     fn read_all_ms2_frames(&self) -> Vec<Frame> {
-        self.frame_reader.collect(|x| x.msms_type != 0)
+        self.frame_reader
+            .parallel_filter(|x| x.msms_type != 0)
+            .collect()
     }
 }