Reduce possibility of integer overflow

Author: BrianMichell

src/mdio/constants.py

@@ -21,6 +21,9 @@
 INT32_MIN = -0x80000000
 INT32_MAX = 0x7FFFFFFF
 
+INT64_MIN = -0x8000000000000000
+INT64_MAX = 0x7FFFFFFFFFFFFFFF
+
 UINT8_MIN = 0x0
 UINT8_MAX = 0xFF
 
@@ -29,3 +32,6 @@
 
 UINT32_MIN = 0x0
 UINT32_MAX = 0xFFFFFFFF
+
+UINT64_MIN = 0x0
+UINT64_MAX = 0xFFFFFFFFFFFFFFFF

src/mdio/core/grid.py

@@ -8,7 +8,7 @@
 import numpy as np
 import zarr
 
-from mdio.constants import UINT32_MAX
+from mdio.constants import UINT64_MAX
 from mdio.core import Dimension
 from mdio.core.serialization import Serializer
 
@@ -88,8 +88,8 @@ def build_map(self, index_headers):
             dim_hdr = index_headers[dim.name]
             live_dim_indices += (np.searchsorted(dim, dim_hdr),)
 
-        # We set dead traces to uint32 max. Should be far away from actual trace counts.
-        self.map = zarr.full(self.shape[:-1], dtype="uint32", fill_value=UINT32_MAX)
+        # We set dead traces to uint64 max. Should be far away from actual trace counts.
+        self.map = zarr.full(self.shape[:-1], dtype="uint64", fill_value=UINT64_MAX)
         self.map.vindex[live_dim_indices] = range(len(live_dim_indices[0]))
 
         self.live_mask = zarr.zeros(self.shape[:-1], dtype="bool")