PyEFIE-2D/primitive_objects.py at main · eattardo/PyEFIE-2D · 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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import numpy as np
from object import Object
from point2D import Point2D, compute_distance


class Rectangle(Object):
    """ Create a rectangle object
    """

    def __init__(self, idObj=-1, pMin=Point2D(), pMax=Point2D()):

        self.pMin = pMin
        self.pMax = pMax
        self.type = "Rectangle"
        Object.__init__(self, idObj)

    def isInternal(self, aPoint):
        """Check if a point is within this rectangle
        """
        if (aPoint.x >= self.pMin.x and aPoint.x <= self.pMax.x) \
                and (aPoint.y >= self.pMin.y and aPoint.y <= self.pMax.y):
            return True
        else:
            return False


# Print the class Rectangle


    def __str__(self):
        s0 = 'Object is: '
        s1 = self.type
        s11 = 'idObj: {}'.format(self.getId())
        s2 = 'pMin: {}'.format(self.pMin)
        s3 = 'pMax: {}'.format(self.pMax)
        return s0 + s1 + "\n" + s11 + "\n" + s2 + "\n" + s3


class Circle(Object):
    """ Create a circle object
    """

    def __init__(self, idObj=-1, radius=0.0, center=Point2D()):
        self.radius = radius
        self.center = center
        self.type = "Circle"
        Object.__init__(self, idObj)

    def isInternal(self, aPoint):
        """Check if a point is within this circle
        """
        if compute_distance(self.center, aPoint) <= self.radius:
            return True
        else:
            return False

# Print the class Rectangle
    def __str__(self):
        s0 = 'Object is: '
        s1 = self.type
        s11 = 'idObj: {}'.format(self.getId())
        s2 = 'center: {}'.format(self.center)
        s3 = 'radius: {}'.format(self.radius)
        return s0 + s1 + "\n" + s11 + "\n" + s2 + "\n" + s3


if __name__ == '__main__':
    rec1 = Rectangle()

    pMin = Point2D(1, 2)
    pMax = Point2D(10, 20)
    rec2 = Rectangle(pMin=pMin, pMax=pMax)
    print(rec2.isInternal(pMax))

    circ1 = Circle()

    circ2 = Circle(radius=10.0, center=Point2D(0.0, 0.0))
    print(circ2.isInternal(pMax))
    print(circ2.isInternal(Point2D()))

    # append a couple of mix objs
    print("\n")
    vPrObj = []
    vPrObj.append(rec1)
    vPrObj.append(rec2)
    vPrObj.append(circ1)
    vPrObj.append(circ2)
    for obj in vPrObj:
        print(obj)
        print("\n")