Merge pull request #398 from RemiLehe/fix_warning

Close #396 warning when reading particle data
openPMD · Oct 4, 2023 · 88ab6e1 · 88ab6e1
2 parents 3b38a54 + 964bd16
commit 88ab6e1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/openpmd_viewer/openpmd_timeseries/data_reader/io_reader/utilities.py b/openpmd_viewer/openpmd_timeseries/data_reader/io_reader/utilities.py
@@ -59,11 +59,16 @@ def get_data(series, record_component, i_slice=None, pos_slice=None,
     # ADIOS2: Actual chunks, all other: one chunk
     chunks = record_component.available_chunks()
 
+    # mask invalid regions with NaN: fill value
+    # note: NaN is only defined for floating point types
+    NaN_value = np.nan if np.issubdtype(record_component.dtype, np.floating) or np.issubdtype(record_component.dtype, np.complexfloating) else 0
+
     # read whole data set
     if pos_slice is None:
         # mask invalid regions with NaN
         #   note: full_like triggers a full read, thus we avoid it #340
-        data = np.full(record_component.shape, np.nan, record_component.dtype)
+        data = np.full(record_component.shape, NaN_value, record_component.dtype)
+
         for chunk in chunks:
             chunk_slice = chunk_to_slice(chunk)
 
@@ -90,7 +95,7 @@ def get_data(series, record_component, i_slice=None, pos_slice=None,
             del slice_shape[dir_index]
 
         # mask invalid regions with NaN
-        data = np.full(slice_shape, np.nan, dtype=record_component.dtype)
+        data = np.full(slice_shape, NaN_value, dtype=record_component.dtype)
 
         # build requested ND slice with respect to full data
         s = []