from hashlib import sha256

import numpy as np
from numpy.typing import NDArray


class HashableArray[D: np.generic]:
    """A hashable wrapper for numpy arrays."""

    def __init__(self, array: NDArray[D]):
        # We copy the array because numpy arrays are mutable
        # and we cannot control their mutability outside this class.
        self.__array = np.copy(array, order="C")

    def __hash__(self) -> int:
        return int(sha256(self.__array.view(np.uint8)).hexdigest(), 16)

    def __eq__(self, other: object) -> bool:
        if isinstance(other, HashableArray):
            return np.array_equal(self.__array, other.__array)
        return False

    def unwrap(self) -> NDArray[D]:
        """Return the original numpy array.

        This method returns a copy of the original array to ensure that
        the original data is not modified outside of this class.
        """
        return self.__array.copy()