Move _DATATYPES_TO_FILL_VALUES back to particlefile.py

Author: VeckoTheGecko

parcels/_constants.py

@@ -14,7 +14,6 @@
 from zarr.storage import DirectoryStore
 
 import parcels
-from parcels._constants import DATATYPES_TO_FILL_VALUES
 from parcels.particle import _SAME_AS_FIELDSET_TIME_INTERVAL, ParticleClass
 from parcels.tools._helpers import timedelta_to_float
 
@@ -25,6 +24,21 @@
 
 __all__ = ["ParticleFile"]
 
+_DATATYPES_TO_FILL_VALUES = {
+    np.dtype(np.float16): np.nan,
+    np.dtype(np.float32): np.nan,
+    np.dtype(np.float64): np.nan,
+    np.dtype(np.bool_): np.iinfo(np.int8).max,
+    np.dtype(np.int8): np.iinfo(np.int8).max,
+    np.dtype(np.int16): np.iinfo(np.int16).max,
+    np.dtype(np.int32): np.iinfo(np.int32).max,
+    np.dtype(np.int64): np.iinfo(np.int64).max,
+    np.dtype(np.uint8): np.iinfo(np.uint8).max,
+    np.dtype(np.uint16): np.iinfo(np.uint16).max,
+    np.dtype(np.uint32): np.iinfo(np.uint32).max,
+    np.dtype(np.uint64): np.iinfo(np.uint64).max,
+}
+
 
 class ParticleFile:
     """Initialise trajectory output.
@@ -109,16 +123,16 @@ def _convert_varout_name(self, var):
 
     def _extend_zarr_dims(self, Z, store, dtype, axis):
         if axis == 1:
-            a = np.full((Z.shape[0], self.chunks[1]), DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
+            a = np.full((Z.shape[0], self.chunks[1]), _DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
             obs = zarr.group(store=store, overwrite=False)["obs"]
             if len(obs) == Z.shape[1]:
                 obs.append(np.arange(self.chunks[1]) + obs[-1] + 1)
         else:
             extra_trajs = self._maxids - Z.shape[0]
             if len(Z.shape) == 2:
-                a = np.full((extra_trajs, Z.shape[1]), DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
+                a = np.full((extra_trajs, Z.shape[1]), _DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
             else:
-                a = np.full((extra_trajs,), DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
+                a = np.full((extra_trajs,), _DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
         Z.append(a, axis=axis)
         zarr.consolidate_metadata(store)
 
@@ -194,13 +208,13 @@ def _write_particle_data(self, *, particle_data, pclass, time_interval, time, in
                     if var.to_write == "once":
                         data = np.full(
                             (arrsize[0],),
-                            DATATYPES_TO_FILL_VALUES[dtype],
+                            _DATATYPES_TO_FILL_VALUES[dtype],
                             dtype=dtype,
                         )
                         data[ids_once] = particle_data[var.name][indices_to_write_once]
                         dims = ["trajectory"]
                     else:
-                        data = np.full(arrsize, DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
+                        data = np.full(arrsize, _DATATYPES_TO_FILL_VALUES[dtype], dtype=dtype)
                         data[ids, 0] = particle_data[var.name][indices_to_write]
                         dims = ["trajectory", "obs"]
                     ds[varout] = xr.DataArray(data=data, dims=dims, attrs=attrs[var.name])
@@ -269,7 +283,7 @@ def _create_variables_attribute_dict(particle: ParticleClass, time_interval: Tim
     for var in vars:
         fill_value = {}
         if var.dtype is not _SAME_AS_FIELDSET_TIME_INTERVAL.VALUE:
-            fill_value = {"_FillValue": DATATYPES_TO_FILL_VALUES[var.dtype]}
+            fill_value = {"_FillValue": _DATATYPES_TO_FILL_VALUES[var.dtype]}
 
         attrs[var.name] = {**var.attrs, **fill_value}