/
coord2.py
54 lines (36 loc) · 1.12 KB
/
coord2.py
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
from typing import Tuple, Union
import numpy as np
class Coord2:
def __init__(self, x: float, y: float) -> None:
self.point = np.array([x, y])
def __getitem__(self, i: int) -> float:
return self.point[i]
@property
def x(self) -> float:
return self.point[0]
@property
def y(self) -> float:
return self.point[1]
@property
def xy(self) -> Tuple[float, float]:
return self.x, self.y
def __add__(self, c2: Union[Tuple[float, float], "Coord2"]) -> "Coord2":
return Coord2(self[0] + c2[0], self[1] + c2[1])
def __mul__(self, a: float) -> "Coord2":
return Coord2(self[0] * a, self[1] * a)
def __rmul__(self, a: float) -> "Coord2":
return Coord2(self[0] * a, self[1] * a)
def __str__(self) -> str:
return f"Coord2({self[0]}, {self[1]})"
def __repr__(self) -> str:
return f"Coord2({self[0]}, {self[1]})"
if __name__ == "__main__":
p0 = Coord2(1.0, 1.5)
p1 = Coord2(2.0, 0.0)
p2 = p0 + p1
p3 = p0 * 2
p4 = 2 * p0
p5 = p3 + (0.0, 5.0)
print(p2)
print(p5)
print(p3.x, p3.y)