Skip to content

Commit

Permalink
Merge pull request #340 from ax3l/fix-chunkingOnlyForADIOS2
Browse files Browse the repository at this point in the history
Fix: Avoid `_like` function in Chunking
  • Loading branch information
RemiLehe authored Apr 14, 2022
2 parents 5511da8 + fd41324 commit 80eeb5e
Showing 1 changed file with 5 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -56,17 +56,21 @@ def get_data(series, record_component, i_slice=None, pos_slice=None,
if i_slice is not None and not isinstance(i_slice, list):
i_slice = [i_slice]

# ADIOS2: Actual chunks, all other: one chunk
chunks = record_component.available_chunks()

# read whole data set
if pos_slice is None:
# mask invalid regions with NaN
data = np.full_like(record_component, np.nan)
# note: full_like triggers a full read, thus we avoid it #340
data = np.full(record_component.shape, np.nan, record_component.dtype)
for chunk in chunks:
chunk_slice = chunk_to_slice(chunk)
# read only valid region
x = record_component[chunk_slice]
series.flush()
data[chunk_slice] = x
# slice: read only part of the data set
else:
full_shape = record_component.shape

Expand Down

0 comments on commit 80eeb5e

Please sign in to comment.