update shot_index dtype to uint32 and add calculated_dims property

Author: tasansal

src/mdio/builder/templates/base.py

@@ -39,6 +39,7 @@ def __init__(self, data_domain: SeismicDataDomain) -> None:
             raise ValueError(msg)
 
         self._dim_names: tuple[str, ...] = ()
+        self._calculated_dims: tuple[str, ...] = ()
         self._physical_coord_names: tuple[str, ...] = ()
         self._logical_coord_names: tuple[str, ...] = ()
         self._var_chunk_shape: tuple[int, ...] = ()
@@ -130,6 +131,11 @@ def dimension_names(self) -> tuple[str, ...]:
         """Returns the names of the dimensions."""
         return copy.deepcopy(self._dim_names)
 
+    @property
+    def calculated_dimension_names(self) -> tuple[str, ...]:
+        """Returns the names of the dimensions."""
+        return copy.deepcopy(self._calculated_dims)
+
     @property
     def physical_coordinate_names(self) -> tuple[str, ...]:
         """Returns the names of the physical (world) coordinates."""

src/mdio/builder/templates/seismic_3d_streamer_field.py

@@ -25,6 +25,7 @@ def __init__(self, data_domain: SeismicDataDomain = "time"):
         super().__init__(data_domain=data_domain)
 
         self._spatial_dim_names = ("sail_line", "gun", "shot_index", "cable", "channel")
+        self._calculated_dims = ("shot_index",)
         self._dim_names = (*self._spatial_dim_names, self._data_domain)
         self._physical_coord_names = ("source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y")
         self._logical_coord_names = ("shot_point", "orig_field_record_num")  # ffid

src/mdio/converters/segy.py

@@ -488,6 +488,7 @@ def _validate_spec_in_template(segy_spec: SegySpec, mdio_template: AbstractDatas
     header_fields = {field.name for field in segy_spec.trace.header.fields}
 
     required_fields = set(mdio_template.spatial_dimension_names) | set(mdio_template.coordinate_names)
+    required_fields = required_fields - set(mdio_template.calculated_dimension_names)  # remove to be calculated
     required_fields = required_fields | {"coordinate_scalar"}  # ensure coordinate scalar is always present
     missing_fields = required_fields - header_fields
 

src/mdio/segy/geometry.py

@@ -487,7 +487,7 @@ def transform(
 
         # This might be slow and potentially could be improved with a rewrite
         # to prevent so many lookups
-        shot_index = np.empty(len(index_headers), dtype="uint8")
+        shot_index = np.empty(len(index_headers), dtype="uint32")
         index_headers = rfn.append_fields(index_headers, "shot_index", shot_index)
         if geom_type == ShotGunGeometryType.B:
             for sail_line in unique_sail_lines: