mostly complete tests

tests/tools/SEGYImport/pytests/test_prestack_gather_spacing.py

@@ -29,8 +29,8 @@ def output_vds() -> TempVDSGuard:
 
 @pytest.fixture
 def prestack_segy() -> str:
-    return os.path.join(test_data_dir, "Plugins", "ImportPlugins", "SEGYUnittest", "Mutes",
-                        "")
+    return os.path.join(test_data_dir, "HeadwavePlatform", "PlatformIntegration", "Teleport_Trim", "3D_Prestack",
+                        "ST0202R08_Gather_Time.segy")
 
 
 @pytest.fixture
@@ -68,6 +68,81 @@ def invalid_executor(prestack_segy, output_vds) -> Tuple[ImportExecutor, TempVDS
     return ex, output_vds
 
 
+def get_gathers_stats(vds_filename: str) -> Tuple[int, int, int, int, int, int]:
+    """
+    Read all the gathers in a single inline and return stats on how many gathers have dead traces in various
+    regions of the gather.
+
+    :return tuple of (
+      total gathers,
+      number of gathers with dead traces only at front of gather,
+      number of gathers with dead traces only within the gather,
+      number of gathers with dead traces only at end of gather,
+      number of gathers with dead traces in multiple places,
+      number of gathers with no dead traces
+      )
+    """
+    with openvds.open(output_vds.filename, "") as handle:
+        layout = openvds.getLayout(handle)
+        access_manager = openvds.getAccessManager(handle)
+
+        trace_channel = layout.getChannelIndex("Trace")
+        assert trace_channel > 0, "Trace channel not found"
+
+        # only read one inline; make it the middle inline
+        inline_index = layout.getDimensionNumSamples(3) // 2
+        dim2_size = layout.getDimensionNumSamples(2)
+        dim1_size = layout.getDimensionNumSamples(1)
+
+        # stats counters
+        front_only = 0
+        middle_only = 0
+        end_only = 0
+        mixed = 0
+        no_dead = 0
+
+        for crossline_index in range(dim2_size):
+            trace_flag_request = access_manager.requestVolumeSubset((0, 0, crossline_index, inline_index, 0, 0),
+                                                                    (1, dim1_size, crossline_index + 1, inline_index + 1, 1,
+                                                                     1),
+                                                                    channel=trace_channel,
+                                                                    format=openvds.VolumeDataChannelDescriptor.Format.Format_U8,
+                                                                    dimensionsND=openvds.DimensionsND.Dimensions_023)
+
+            assert dim1_size == trace_flag_request.data.shape[0]
+
+            # analyze trace_flag_request.data to figure out which stats counter to bump
+
+            dead_trace_ranges = []
+            i = 0
+            while i < dim1_size:
+                if trace_flag_request.data[i] == 0:
+                    start = i
+                    i += 1
+                    while i < dim1_size and trace_flag_request.data[i] == 0:
+                        i += 1
+                    end = i - 1
+                    dead_trace_ranges.append((start, end))
+                else:
+                    i += 1
+
+            if len(dead_trace_ranges) == 0:
+                no_dead += 1
+            elif len(dead_trace_ranges) == 1:
+                # either front, middle, end
+                start, stop = dead_trace_ranges[0]
+                if start == 0:
+                    front_only += 1
+                elif stop == 0:
+                    end_only += 1
+                else:
+                    middle_only += 1
+            else:
+                mixed += 1
+
+        return dim2_size, front_only, middle_only, end_only, mixed, no_dead
+
+
 def test_gather_spacing_invalid_arg(invalid_executor):
     ex, output_vds = invalid_executor
 
@@ -86,14 +161,14 @@ def test_gather_spacing_default(default_executor):
     assert result == 0, ex.output()
     assert Path(output_vds.filename).exists()
 
-    with openvds.open(output_vds.filename, "") as handle:
-        layout = openvds.getLayout(handle)
+    total, leading_only, middle_only, trailing_only, mixed, no_dead = get_gathers_stats(output_vds.filename)
 
-        # TODO read an inline (or all inlines)
-        # TODO for each gather, check location of dead traces
-        # TODO assert dead traces occur elsewhere besides the end of the gather
-
-    assert False, "not implemented"
+    assert total == 60
+    assert middle_only > 0
+    assert mixed > 0
+    assert trailing_only == 0
+    assert leading_only > 0
+    assert no_dead == 0
 
 
 def test_gather_spacing_auto(auto_executor):
@@ -104,14 +179,14 @@ def test_gather_spacing_auto(auto_executor):
     assert result == 0, ex.output()
     assert Path(output_vds.filename).exists()
 
-    with openvds.open(output_vds.filename, "") as handle:
-        layout = openvds.getLayout(handle)
-
-        # TODO read an inline (or all inlines)
-        # TODO for each gather, check location of dead traces
-        # TODO assert dead traces occur elsewhere besides the end of the gather
+    total, leading_only, middle_only, trailing_only, mixed, no_dead = get_gathers_stats(output_vds.filename)
 
-    assert False, "not implemented"
+    assert total == 60
+    assert middle_only > 0
+    assert mixed > 0
+    assert trailing_only == 0
+    assert leading_only > 0
+    assert no_dead == 0
 
 
 def test_gather_spacing_on(on_executor):
@@ -122,14 +197,14 @@ def test_gather_spacing_on(on_executor):
     assert result == 0, ex.output()
     assert Path(output_vds.filename).exists()
 
-    with openvds.open(output_vds.filename, "") as handle:
-        layout = openvds.getLayout(handle)
-
-        # TODO read an inline (or all inlines)
-        # TODO for each gather, check location of dead traces
-        # TODO assert dead traces occur elsewhere besides the end of the gather
+    total, leading_only, middle_only, trailing_only, mixed, no_dead = get_gathers_stats(output_vds.filename)
 
-    assert False, "not implemented"
+    assert total == 60
+    assert middle_only > 0
+    assert mixed > 0
+    assert trailing_only == 0
+    assert leading_only > 0
+    assert no_dead == 0
 
 
 def test_gather_spacing_off(off_executor):
@@ -140,11 +215,11 @@ def test_gather_spacing_off(off_executor):
     assert result == 0, ex.output()
     assert Path(output_vds.filename).exists()
 
-    with openvds.open(output_vds.filename, "") as handle:
-        layout = openvds.getLayout(handle)
-
-        # TODO read an inline (or all inlines)
-        # TODO for each gather, check location of dead traces
-        # TODO assert dead traces occur only at the end of the gather
+    total, leading_only, middle_only, trailing_only, mixed, no_dead = get_gathers_stats(output_vds.filename)
 
-    assert False, "not implemented"
+    assert total == 60
+    assert trailing_only == total
+    assert middle_only == 0
+    assert mixed == 0
+    assert leading_only == 0
+    assert no_dead == 0