Use new segy API calls

Author: BrianMichell

pyproject.toml

@@ -26,7 +26,7 @@ dependencies = [
     "psutil>=7.0.0",
     "pydantic>=2.11.9",
     "rich>=14.1.0",
-    "segy>=0.5.0",
+    "segy @ git+https://github.com/TGSAI/segy.git@easier-access-to-raw-bytes",
     "tqdm>=4.67.1",
     "universal-pathlib>=0.2.6",
     "xarray>=2025.9.1",

src/mdio/segy/_disaster_recovery_wrapper.py

@@ -2,11 +2,8 @@
 
 from __future__ import annotations
 
-from copy import deepcopy
 from typing import TYPE_CHECKING
 
-import numpy as np
-
 if TYPE_CHECKING:
     from numpy.typing import NDArray
     from segy import SegyFile
@@ -16,19 +13,21 @@ class SegyFileTraceDataWrapper:
     def __init__(self, segy_file: SegyFile, indices: int | list[int] | NDArray | slice):
         self.segy_file = segy_file
         self.indices = indices
-        self._header_pipeline = deepcopy(segy_file.accessors.header_decode_pipeline)
-        segy_file.accessors.header_decode_pipeline.transforms = []
-        self.traces = segy_file.trace[indices]
+        self.idx = self.segy_file.trace.normalize_and_validate_query(self.indices)
+        self.traces = self.segy_file.trace.fetch(self.idx, raw=True)
+
+        self.raw_view = self.traces.view(self.segy_file.spec.trace.dtype)
+        self.decoded_traces = self.segy_file.accessors.trace_decode_pipeline.apply(self.raw_view.copy())
 
-    @property
-    def header(self) -> NDArray:
-        # The copy is necessary to avoid applying the pipeline to the original header.
-        return self._header_pipeline.apply(self.traces.header.copy())
 
     @property
     def raw_header(self) -> NDArray:
-        return np.ascontiguousarray(self.traces.header.copy()).view("|V240")
+        return self.raw_view.header.view("|V240")
+
+    @property
+    def header(self) -> NDArray:
+        return self.decoded_traces.header
 
     @property
     def sample(self) -> NDArray:
-        return self.traces.sample
+        return self.decoded_traces.sample