Source code for phasemap._coordinate

# © 2015-2018, ETH Zurich, Institut für Theoretische Physik
# Author: Dominik Gresch <greschd@gmx.ch>

from fractions import Fraction

import numpy as np


class Coordinate(np.ndarray):
    """
    Array class describing the relative position within the calculation window.
    """

    def __new__(cls, coord):
        coord_list = [Fraction(x) for x in coord]
        self = super().__new__(cls, shape=(len(coord_list),), dtype=object)
        self[:] = coord_list
        self.flags.writeable = False
        return self

    def __hash__(self):
        return hash(tuple(self))

    def __eq__(self, other):
        return super().__eq__(other).all()