Get known size for pushed dimension

Author: BrianMichell

src/mdio/builder/dataset_builder.py

@@ -101,7 +101,7 @@ def add_dimension(self, name: str, size: int) -> "MDIODatasetBuilder":
         self._state = _BuilderState.HAS_DIMENSIONS
         return self
 
-    def push_dimension(self, dimension: NamedDimension, position: int, new_dim_chunk_size: int=1) -> "MDIODatasetBuilder":
+    def push_dimension(self, dimension: NamedDimension, position: int, new_dim_chunk_size: int=1, new_dim_size: int=1) -> "MDIODatasetBuilder":
         """Pushes a dimension to all Coordiantes and Variables.
         The position argument is the domain index of the dimension to push.
         If a Variable is within the position domain, it will be inserted at the position and all remaining dimensions will be shifted to the right.
@@ -128,8 +128,7 @@ def push_dimension(self, dimension: NamedDimension, position: int, new_dim_chunk
         # In-place insertion of the dimension to the existing list of dimensions
         self._dimensions.insert(position, dimension)
 
-        # TODO: HARDCODED FOR DEBUGGING
-        self._dimensions[-1].size = 25
+        self._dimensions[position].size = new_dim_size
 
         def propogate_dimension(variable: Variable, position: int, new_dim_chunk_size: int) -> Variable:
             """Propogates the dimension to the variable or coordinate."""

src/mdio/segy/utilities.py

@@ -52,7 +52,7 @@ def delayed_transform(builder):
 
         # Add the trace dimension
         trace_dimension = NamedDimension(name="trace", size=trace_size)
-        builder.push_dimension(trace_dimension, position=position, new_dim_chunk_size=1)
+        builder.push_dimension(trace_dimension, position=position, new_dim_chunk_size=1, new_dim_size=trace_size)
 
         # Add the corresponding coordinate for the trace dimension
         builder.add_coordinate(