Skip to content

Commit 3d5cfe2

Browse files
authored
Fix crash in header parsing when trace count is parsing block_size + 1 (#477)
* handle singleton edge case when trace_range only has 1 trace. * reuse variable
1 parent eb3b35d commit 3d5cfe2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/mdio/segy/_workers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,11 @@ def header_scan_worker(
5353
non_void_fields = [(name, dtype) for name, (dtype, _) in fields.items()]
5454
new_dtype = np.dtype(non_void_fields)
5555

56-
# Allocate empty memory and assign non-void fields
57-
trace_header_filtered = np.empty_like(trace_header, dtype=new_dtype)
58-
trace_header_filtered[:] = trace_header
56+
# Copy to non-padded memory, ndmin is to handle the case where there is
57+
# 1 trace in block (singleton) so we can concat and assign stuff later.
58+
trace_header = np.array(trace_header, dtype=new_dtype, ndmin=1)
5959

60-
return cast(HeaderArray, trace_header_filtered)
60+
return cast(HeaderArray, trace_header)
6161

6262

6363
def trace_worker(

0 commit comments

Comments
 (0)