mdio-python/src/mdio/constants.py at 7b3ba70ecc508cc28fd472390254ca47e37c1d62 · dmitriyrepin/mdio-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
"""Constant values used across MDIO."""

from numpy import finfo as np_finfo
from numpy import iinfo as np_iinfo
from numpy import nan as np_nan

from mdio.schemas.dtype import ScalarType

FLOAT16_MAX = np_finfo("float16").max
FLOAT16_MIN = np_finfo("float16").min

FLOAT32_MAX = np_finfo("float32").max
FLOAT32_MIN = np_finfo("float32").min

FLOAT64_MIN = np_finfo("float64").min
FLOAT64_MAX = np_finfo("float64").max

INT8_MIN = np_iinfo("int8").min
INT8_MAX = np_iinfo("int8").max

INT16_MIN = np_iinfo("int16").min
INT16_MAX = np_iinfo("int16").max

INT32_MIN = np_iinfo("int32").min
INT32_MAX = np_iinfo("int32").max

INT64_MIN = np_iinfo("int64").min
INT64_MAX = np_iinfo("int64").max

UINT8_MIN = 0
UINT8_MAX = np_iinfo("uint8").max

UINT16_MIN = 0
UINT16_MAX = np_iinfo("uint16").max

UINT32_MIN = 0
UINT32_MAX = np_iinfo("uint32").max

UINT64_MIN = 0
UINT64_MAX = np_iinfo("uint64").max

# Zarr fill values for different scalar types
fill_value_map = {
    ScalarType.BOOL: None,
    ScalarType.FLOAT16: np_nan,
    ScalarType.FLOAT32: np_nan,
    ScalarType.FLOAT64: np_nan,
    ScalarType.UINT8: 2**8 - 1,  # Max value for uint8
    ScalarType.UINT16: 2**16 - 1,  # Max value for uint16
    ScalarType.UINT32: 2**32 - 1,  # Max value for uint32
    ScalarType.UINT64: 2**64 - 1,  # Max value for uint64
    ScalarType.INT8: 2**7 - 1,  # Max value for int8
    ScalarType.INT16: 2**15 - 1,  # Max value for int16
    ScalarType.INT32: 2**31 - 1,  # Max value for int32
    ScalarType.INT64: 2**63 - 1,  # Max value for int64
    ScalarType.COMPLEX64: complex(np_nan, np_nan),
    ScalarType.COMPLEX128: complex(np_nan, np_nan),
    ScalarType.COMPLEX256: complex(np_nan, np_nan),
}