Cleanup prints

Author: BrianMichell

src/mdio/converters/mdio.py

@@ -91,20 +91,10 @@ def mdio_to_segy(  # noqa: PLR0912, PLR0913
 
     output_segy_path = Path(output_segy_path)
 
-    # mdio = MDIOReader(
-    #     mdio_path_or_buffer=mdio_path_or_buffer,
-    #     access_pattern=access_pattern,
-    #     storage_options=storage_options,
-    # )
-
     mdio = MDIO.open(mdio_path_or_buffer)
 
     seismic_chunks = mdio.seismic.encoding.get("chunks", mdio.seismic.shape)
 
-    print(f"mdio.seismic.chunks: {seismic_chunks}")
-    print(f"mdio.seismic.shape: {mdio.seismic.shape}")
-    print(f"mdio.seismic.dtype: {mdio.seismic.dtype}")
-
     if new_chunks is None:
         new_chunks = segy_export_rechunker(seismic_chunks, mdio.seismic.shape, mdio.seismic.dtype)
 
@@ -129,9 +119,6 @@ def mdio_to_segy(  # noqa: PLR0912, PLR0913
     else:
         mdio, segy_factory = mdio_spec_to_segy(*creation_args)
 
-    print(f"segy_factory: {segy_factory}")
-
-    # live_mask = mdio.live_mask.compute()
     live_mask = mdio.trace_mask.compute().to_numpy()
 
     if selection_mask is not None:
@@ -152,17 +139,11 @@ def mdio_to_segy(  # noqa: PLR0912, PLR0913
         dim_slices += (slice(start, stop),)
 
     # Lazily pull the data with limits now, and limit mask so its the same shape.
-    print(f"dim_slices: {dim_slices}")
     indexer = {}
     for dim, i in zip(mdio.trace_mask.dims, dim_slices):
         indexer[dim] = i
 
-    print(f"indexer: {indexer}")
-    # live_mask = mdio.trace_mask.isel(dim_slices[:-1])
-    # headers = mdio.headers.isel(dim_slices[:-1])
-    # samples = mdio.seismic.isel(dim_slices[:-1])
     live_mask = mdio.trace_mask.isel(indexer)
-    print(f"live_mask {live_mask}")
 
     headers = mdio.headers.isel(indexer)
     samples = mdio.seismic.isel(indexer)

src/mdio/converters/segy.py

@@ -693,9 +693,6 @@ def segy_to_mdio_schematized(
     mdio_spec = mdio_segy_spec()
     mdio_spec.trace.header = get_mdio_header_defined_fields(ds)
 
-    # print(f"mdio_spec: {mdio_spec}")
-    # raise ValueError("Stop here")
-
     segy_settings = SegySettings(storage_options=storage_options_input)
     segy = SegyFile(url=segy_schema["path"], spec=mdio_spec, settings=segy_settings)
 
@@ -725,9 +722,6 @@ def segy_to_mdio_schematized(
 
     mdio_spec_grid = mdio_spec.customize(trace_header_fields=index_fields)
 
-    print(f"mdio_spec_grid: {mdio_spec_grid}")
-    # raise ValueError("Stop here")
-
     segy_grid = SegyFile(url=segy_schema["path"], spec=mdio_spec_grid, settings=segy_settings)
     dimensions, chunksize, index_headers = get_grid_plan(
         segy_file=segy_grid,
@@ -746,10 +740,6 @@ def segy_to_mdio_schematized(
     grid_density_qc(grid, num_traces)
     grid.build_map(index_headers)
 
-    # print(f"dimensions: {dimensions}")
-    # print(f"chunksize: {chunksize}")
-    # print(f"index_headers: {index_headers}")
-
     # Set dimension coordinates
     new_coords = {dim.name: dim.coords for dim in dimensions}
     ds = ds.assign_coords(new_coords)

src/mdio/segy/_workers.py

@@ -128,14 +128,6 @@ def trace_worker(
     if nonzero_count == 0:
         return None
 
-    # Flush data to Zarr
-    # data_array.set_basic_selection(selection=chunk_indices, value=tmp_data)
-
-    # print("Writing data to the underlying array...")
-    # print(f"Chunk indices: {chunk_indices}")
-    # print(f"tmp_data shape: {tmp_data.shape}")
-    # print(f"data_array shape: {data_array.shape}")
-
     # Direct assignment to underlying data array
     data_array.data[chunk_indices] = tmp_data
     data_array.to_mdio(store=mdio_path_or_buffer, mode="r+")

src/mdio/segy/compat.py

@@ -36,13 +36,11 @@ def get_mdio_header_defined_fields(mdio: MDIO) -> HeaderSpec:
     fields = []
     offset = 1
     for field_name, (field_dtype, _) in mdio.headers.dtype.fields.items():
-        # print(f"field_name: {field_name}, field_dtype: {field_dtype}, offset: {offset}")
         # Convert NumPy dtype to string format
         format_str = str(field_dtype).replace('dtype(', '').replace(')', '')
         fields.append(HeaderField(name=field_name, byte=offset, format=format_str))
         offset += field_dtype.itemsize
 
-    print(f"Custom header fields offset: {offset}")
     return HeaderSpec(fields=fields, item_size=240)
 
 

src/mdio/segy/creation.py

@@ -107,23 +107,9 @@ def mdio_spec_to_segy(  # noqa: PLR0913
 
     # from segy.schema.HeaderSpec import HeaderField, HeaderSpec
     from segy.schema import HeaderField, HeaderSpec
-    # fields = []
-    # offset = 1
-    # for field_name, (field_dtype, _) in mdio.headers.dtype.fields.items():
-    #     print(f"field_name: {field_name}, field_dtype: {field_dtype}, offset: {offset}")
-    #     # Convert NumPy dtype to string format
-    #     format_str = str(field_dtype).replace('dtype(', '').replace(')', '')
-    #     fields.append(HeaderField(name=field_name, byte=offset, format=format_str))
-    #     offset += field_dtype.itemsize
-
     fields = get_mdio_header_defined_fields(mdio)
-
-    print(f"Before spec: {spec.trace}")
-    # mdio_header_mapping = HeaderSpec(fields=fields)
     spec.trace.header = fields
 
-    print(f"After spec: {spec.trace}")
-
     spec.endianness = Endianness(output_endian)
     factory = make_segy_factory(mdio, spec=spec)