From 560f172a203782c8b091c866e1e7fce3afc7533a Mon Sep 17 00:00:00 2001 From: Axel Huebl Date: Fri, 4 Mar 2022 13:55:04 -0800 Subject: [PATCH] MR Reads: Zero->NaN Padding Fill invalid regions with `np.nan` instead of zeros. --- .gitignore | 1 + .../openpmd_timeseries/data_reader/io_reader/utilities.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/.gitignore b/.gitignore index 3df2b513..ab25bf08 100644 --- a/.gitignore +++ b/.gitignore @@ -7,3 +7,4 @@ openpmd_viewer.egg-info/ .cache/ .eggs/ .vs/ +__pycache__ diff --git a/openpmd_viewer/openpmd_timeseries/data_reader/io_reader/utilities.py b/openpmd_viewer/openpmd_timeseries/data_reader/io_reader/utilities.py index 7b21cea5..c5d8bf85 100644 --- a/openpmd_viewer/openpmd_timeseries/data_reader/io_reader/utilities.py +++ b/openpmd_viewer/openpmd_timeseries/data_reader/io_reader/utilities.py @@ -59,8 +59,8 @@ def get_data(series, record_component, i_slice=None, pos_slice=None, chunks = record_component.available_chunks() if pos_slice is None: - # mask invalid regions with zero - data = np.zeros_like(record_component) + # mask invalid regions with NaN + data = np.full_like(record_component, np.nan) for chunk in chunks: chunk_slice = chunk_to_slice(chunk) # read only valid region @@ -76,8 +76,8 @@ def get_data(series, record_component, i_slice=None, pos_slice=None, for dir_index in pos_slice_sorted: # remove indices in list del slice_shape[dir_index] - # mask invalid regions with zero - data = np.zeros(slice_shape, dtype=output_type) + # mask invalid regions with NaN + data = np.full(slice_shape, np.nan, dtype=output_type) # build requested ND slice with respect to full data s = []