sections len

Author: SyMind

src/cached_source.rs

@@ -252,6 +252,16 @@ impl Stream for CachedSourceStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    if let Some(index_map) = self.cache.columns_index_map.get() {
+      index_map.as_ref().map(|index_map| index_map.sections().len()).unwrap_or(0)
+    } else if let Some(index_map) = self.cache.line_only_index_map.get() {
+      index_map.as_ref().map(|index_map| index_map.sections().len()).unwrap_or(0)
+    } else {
+      self.stream.sections_len()
+    }
+  }
+
   fn sections<'a>(
     &'a self,
     object_pool: &'a ObjectPool,
@@ -304,6 +314,7 @@ impl Stream for CachedSourceStream<'_> {
 }
 
 impl ToStream for CachedSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(CachedSourceStream::new(self))
   }

src/concat_source.rs

@@ -220,8 +220,9 @@ impl Source for ConcatSource {
     object_pool: &ObjectPool,
     options: &MapOptions,
   ) -> Option<IndexSourceMap> {
-    let mut sections = Vec::new();
-    self.to_stream().sections(
+    let stream = self.to_stream();
+    let mut sections = Vec::with_capacity(stream.sections_len());
+    stream.sections(
       object_pool,
       options.columns,
       &mut |offset, map| {
@@ -451,6 +452,10 @@ impl Stream for ConcatSourceStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    self.children_streams.iter().map(|child_stream| child_stream.sections_len()).sum()
+  }
+
   fn sections<'a>(
     &'a self,
     object_pool: &'a ObjectPool,
@@ -481,6 +486,7 @@ impl Stream for ConcatSourceStream<'_> {
 }
 
 impl ToStream for ConcatSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(ConcatSourceStream::new(self))
   }

src/helpers.rs

@@ -90,6 +90,8 @@ pub trait Stream {
     on_name: crate::helpers::OnName<'_, 'a>,
   ) -> crate::helpers::GeneratedInfo;
 
+  fn sections_len(&self) -> usize;
+
   fn sections<'a>(
     &'a self,
     object_pool: &'a ObjectPool,

src/original_source.rs

@@ -248,6 +248,10 @@ impl Stream for OriginalSourceStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    1
+  }
+
   fn sections<'a>(
     &'a self,
     object_pool: &'a ObjectPool,
@@ -268,6 +272,7 @@ impl Stream for OriginalSourceStream<'_> {
 }
 
 impl ToStream for OriginalSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(OriginalSourceStream::new(self))
   }

src/raw_source.rs

@@ -131,6 +131,10 @@ impl Stream for RawStringStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    0
+  }
+
   fn sections<'a>(
     &'a self,
     _object_pool: &'a ObjectPool,
@@ -144,6 +148,7 @@ impl Stream for RawStringStream<'_> {
 }
 
 impl ToStream for RawStringSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(RawStringStream::new(self))
   }
@@ -283,6 +288,10 @@ impl Stream for RawBufferSourceStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    0
+  }
+
   fn sections<'a>(
     &'a self,
     _object_pool: &'a ObjectPool,
@@ -297,6 +306,7 @@ impl Stream for RawBufferSourceStream<'_> {
 }
 
 impl ToStream for RawBufferSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(RawBufferSourceStream(self))
   }

src/replace_source.rs

@@ -861,6 +861,10 @@ impl Stream for ReplaceSourceStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    0
+  }
+
   fn sections<'a>(
     &'a self,
     object_pool: &'a ObjectPool,
@@ -881,6 +885,7 @@ impl Stream for ReplaceSourceStream<'_> {
 }
 
 impl ToStream for ReplaceSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(ReplaceSourceStream::new(self))
   }

src/source.rs

@@ -206,6 +206,7 @@ impl Source for BoxSource {
 dyn_clone::clone_trait_object!(Source);
 
 impl ToStream for BoxSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     self.as_ref().to_stream()
   }

src/source_map_source.rs

@@ -226,6 +226,10 @@ impl Stream for SourceMapSourceStream<'_> {
     }
   }
 
+  fn sections_len(&self) -> usize {
+    1
+  }
+
   fn sections<'a>(
     &'a self,
     object_pool: &'a ObjectPool,

tests/compat_source.rs

@@ -72,6 +72,10 @@ impl Stream for CompatSourceStream<'_> {
     )
   }
 
+  fn sections_len(&self) -> usize {
+    1
+  }
+
   fn sections<'a>(
     &'a self,
     _object_pool: &'a ObjectPool,
@@ -85,6 +89,7 @@ impl Stream for CompatSourceStream<'_> {
 }
 
 impl ToStream for CompatSource {
+  #[inline]
   fn to_stream<'a>(&'a self) -> Box<dyn Stream + 'a> {
     Box::new(CompatSourceStream::new(self))
   }