Skip to content

Commit 3402224

Browse files
tasansalAltay Sansal
andauthored
Refactor compute_global_index and update filepath generation. (TGSAI#510)
Co-authored-by: Altay Sansal <[email protected]>
1 parent 52539ee commit 3402224

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

src/mdio/segy/blocked_io.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import os
77
from concurrent.futures import ProcessPoolExecutor
88
from itertools import repeat
9-
from shutil import copyfileobj
109
from typing import TYPE_CHECKING
1110

1211
import numpy as np
@@ -21,6 +20,7 @@
2120
from mdio.core.indexing import ChunkIterator
2221
from mdio.segy._workers import trace_worker
2322
from mdio.segy.creation import SegyPartRecord
23+
from mdio.segy.creation import concat_files
2424
from mdio.segy.creation import serialize_to_segy_stack
2525
from mdio.segy.utilities import find_trailing_ones_index
2626

@@ -249,11 +249,7 @@ def segy_record_concat(
249249
dest_map[record_file_path].append(block.path)
250250

251251
for dest_path, source_paths in dest_map.items():
252-
with open(dest_path, "wb") as dest_file:
253-
for src_path in source_paths:
254-
with open(src_path, "rb") as src_file:
255-
copyfileobj(src_file, dest_file)
256-
os.remove(src_path)
252+
concat_files([dest_path] + source_paths)
257253

258254
return records_metadata
259255

0 commit comments

Comments
 (0)