Fix padded dimension handling in SEG-Y serialization worker (#506)

* Fix padded dimension handling in SEG-Y creation.

* Correct block file ordering by filtering live blocks when some are missing.

* update lockfile

* Fix block unwrapping by adding early return check.

---------

Co-authored-by: Altay Sansal <altay.sansal@tgs.com>

src/mdio/converters/mdio.py

@@ -174,10 +174,8 @@ def mdio_to_segy(  # noqa: C901
             else:
                 is_block_live = is_block_live.compute(num_workers=NUM_CPUS)
 
-        n_blocks = len(is_block_live)
-        ordered_files = [
-            tmp_dir.name + f"/{block}._mdiotemp" for block in range(n_blocks)
-        ]
+        live_blocks = np.where(is_block_live)[0]
+        ordered_files = [tmp_dir.name + f"/{block}._mdiotemp" for block in live_blocks]
         ordered_files = [output_segy_path] + ordered_files
 
         if client is not None:

src/mdio/segy/blocked_io.py

@@ -318,6 +318,11 @@ def to_segy(
     # to first (slowest) dimension. End result will be the blocks with the
     # size of the outermost dimension in ascending order.
     consecutive_dim_index = find_trailing_ones_index(is_block_live.numblocks)
+
+    # Shortcut if no unwrapping necessary.
+    if consecutive_dim_index == 1:
+        return is_block_live.any(axis=-1)
+
     while consecutive_dim_index != 1:
         current_chunks = is_block_live.chunks
 

src/mdio/segy/creation.py

@@ -135,16 +135,16 @@ def serialize_to_segy_stack(
     Returns:
         Live mask, as is, for combined blocks (dropped sample dimension).
     """
+    # Drop map_blocks padded dim
+    live_mask = live_mask[..., 0]
+    headers = headers[..., 0]
+
     if block_info is None:
         return live_mask
 
     if np.count_nonzero(live_mask) == 0:
         return live_mask
 
-    # Drop map_blocks padded dim
-    live_mask = live_mask[..., 0]
-    headers = headers[..., 0]
-
     # Set up chunk boundaries and coordinates to write
     global_num_blocks = block_info[0]["num-chunks"]
     block_coords = block_info[0]["array-location"]