PyTransform/position_transform.py at master · alastaircarson/PyTransform · 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
60
61
62
63
64
65
66
67
68
69
70
71
72
from math import sin, cos


class PositionTransform:
    matrix = None

    def __init__(self):
        self.identity()

    def identity(self):
        self.matrix = [[1, 0, 0], [0, 1, 0], [0, 0, 1]]

    def print(self):
        print(self.matrix)

    def transform(self, coord):
        transformed = (
            coord[0] * self.matrix[0][0] + coord[1] * self.matrix[0][1] + self.matrix[0][2],
            coord[0] * self.matrix[1][0] + coord[1] * self.matrix[1][1] + self.matrix[1][2])
        return transformed

    def combine(self, transform):
        result = PositionTransform()

        result.matrix[0][0] = self.matrix[0][0] * transform.matrix[0][0] + self.matrix[0][1] * transform.matrix[1][0] + \
                              self.matrix[0][2] * transform.matrix[2][0]
        result.matrix[0][1] = self.matrix[0][0] * transform.matrix[0][1] + self.matrix[0][1] * transform.matrix[1][1] + \
                              self.matrix[0][2] * transform.matrix[2][1]
        result.matrix[0][2] = self.matrix[0][0] * transform.matrix[0][2] + self.matrix[0][1] * transform.matrix[1][2] + \
                              self.matrix[0][2] * transform.matrix[2][2]
        result.matrix[1][0] = self.matrix[1][0] * transform.matrix[0][0] + self.matrix[1][1] * transform.matrix[1][0] + \
                              self.matrix[1][2] * transform.matrix[2][0]
        result.matrix[1][1] = self.matrix[1][0] * transform.matrix[0][1] + self.matrix[1][1] * transform.matrix[1][1] + \
                              self.matrix[1][2] * transform.matrix[2][1]
        result.matrix[1][2] = self.matrix[1][0] * transform.matrix[0][2] + self.matrix[1][1] * transform.matrix[1][2] + \
                              self.matrix[1][2] * transform.matrix[2][2]
        result.matrix[2][0] = self.matrix[2][0] * transform.matrix[0][0] + self.matrix[2][1] * transform.matrix[1][0] + \
                              self.matrix[2][2] * transform.matrix[2][0]
        result.matrix[2][1] = self.matrix[2][0] * transform.matrix[0][1] + self.matrix[2][1] * transform.matrix[1][1] + \
                              self.matrix[2][2] * transform.matrix[2][1]
        result.matrix[2][2] = self.matrix[2][0] * transform.matrix[0][2] + self.matrix[2][1] * transform.matrix[1][2] + \
                              self.matrix[2][2] * transform.matrix[2][2]
        return result


class Shift(PositionTransform):
    def __init__(self, x_offset, y_offset):
        PositionTransform.__init__(self)
        self.matrix[0][2] = x_offset
        self.matrix[1][2] = y_offset


class Scale(PositionTransform):
    def __init__(self, scale):
        PositionTransform.__init__(self)
        self.matrix[0][0] = scale
        self.matrix[1][1] = scale


class Rotation(PositionTransform):
    def __init__(self, angle):
        PositionTransform.__init__(self)
        self.matrix[0][0] = cos(angle)
        self.matrix[0][1] = -sin(angle)
        self.matrix[1][0] = sin(angle)
        self.matrix[1][1] = cos(angle)


class HorizontalFlip(PositionTransform):
    def __init__(self):
        PositionTransform.__init__(self)
        self.matrix[1][1] = -1