Skip to content

coordinate_3d

Bases: coordinate

Source code in tinybig/koala/geometry/coordinate.py 
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
class coordinate_3d(coordinate):
    def __init__(self, h: int, w: int, d: int, *args, **kwargs):
        super().__init__(coords=(h, w, d), *args, **kwargs)

    def __add__(self, other):
        if isinstance(other, self.__class__) and len(self.coords) == len(other.coords):
            return self.__class__(*tuple(a + b for a, b in zip(self.coords, other.coords)))
        raise TypeError("Operands must be of type Coordinate and have the same dimensions")

    # Redefine the - operator
    def __sub__(self, other):
        if isinstance(other, self.__class__) and len(self.coords) == len(other.coords):
            return self.__class__(*tuple(a - b for a, b in zip(self.coords, other.coords)))
        raise TypeError("Operands must be of type Coordinate and have the same dimensions")

    @property
    def h(self):
        return self.coords[0] if self.dimension() > 0 else None

    @h.setter
    def h(self, value):
        if self.dimension() > 0:
            self.coords = (value,) + self.coords[1:]

    @property
    def w(self):
        return self.coords[1] if self.dimension() > 1 else None

    @w.setter
    def w(self, value):
        if self.dimension() > 1:
            self.coords = self.coords[0:1] + (value,) + self.coords[2:]

    @property
    def d(self):
        return self.coords[2] if self.dimension() > 2 else None

    @d.setter
    def d(self, value):
        if self.dimension() > 2:
            self.coords = self.coords[0:2] + (value,)

    # Aliases for x, y, z: h, w, d
    @property
    def x(self):
        return self.h

    @x.setter
    def x(self, value):
        self.h = value

    @property
    def y(self):
        return self.w

    @y.setter
    def y(self, value):
        self.w = value

    @property
    def z(self):
        return self.d

    @z.setter
    def z(self, value):
        self.d = value