Add primitive definitions and bounding box calcs for DRC

1 files changed, 193 insertions, 0 deletions

diff --git a/gerber/primitives.py b/gerber/primitives.py
new file mode 100644
index 0000000..366c397
--- /dev/null
+++ b/gerber/primitives.py
@@ -0,0 +1,193 @@
+#! /usr/bin/env python

+# -*- coding: utf-8 -*-

+

+# copyright 2014 Hamilton Kibbe <ham@hamiltonkib.be>

+

+# Licensed under the Apache License, Version 2.0 (the "License");

+# you may not use this file except in compliance with the License.

+# You may obtain a copy of the License at

+

+#     http://www.apache.org/licenses/LICENSE-2.0

+

+# Unless required by applicable law or agreed to in writing, software

+# distributed under the License is distributed on an "AS IS" BASIS,

+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

+# See the License for the specific language governing permissions and

+# limitations under the License.

+import math

+from operator import sub

+

+

+class Primitive(object):

+    def bounding_box(self):

+        """ Calculate bounding box

+

+        will be helpful for sweep & prune during DRC clearance checks.

+        

+        Return ((min x, max x), (min y, max y))

+        """

+        pass

+

+

+class Line(Primitive):

+    """

+    """

+    def __init__(self, start, end, width):

+        self.start = start

+        self.end = end

+        self.width = width

+        

+    @property

+    def angle(self):

+        dx, dy = tuple(map(sub, end, start))

+        angle = degrees(math.tan(dy/dx))

+

+    def bounding_box(self):

+        width_2 = self.width / 2.

+        min_x = min(self.start[0], self.end[0]) - width_2

+        max_x = max(self.start[0], self.end[0]) + width_2

+        min_y = min(self.start[1], self.end[1]) - width_2

+        max_y = max(self.start[1], self.end[1]) + width_2

+        return ((min_x, max_x), (min_y, max_y))

+

+

+class Arc(Primitive):

+    """

+    """

+    def __init__(self, start, end, center, direction, width):

+        self.start = start

+        self.end = end

+        self.center = center

+        self.direction = direction

+        self.width = width

+

+    @property

+    def start_angle(self):

+        dy, dx = map(sub, self.start, self.center)

+        return math.atan2(dy, dx)

+

+    @property

+    def end_angle(self):

+        dy, dx = map(sub, self.end, self.center)

+        return math.atan2(dy, dx)

+

+    def bounding_box(self):

+        pass

+

+class Circle(Primitive):

+    """

+    """

+    def __init__(self, position, diameter):

+        self.position = position

+        self.diameter = diameter

+        self.radius = diameter / 2.

+

+    def bounding_box(self):

+        min_x = self.position[0] - self.radius

+        max_x = self.position[0] + self.radius

+        min_y = self.position[1] - self.radius

+        max_y = self.position[1] + self.radius

+        return ((min_x, max_x), (min_y, max_y))

+

+

+class Rectangle(Primitive):

+    """

+    """

+    def __init__(self, position, width, height):

+        self.position = position

+        self.width = width

+        self.height = height

+

+    @property

+    def lower_left(self):

+        return (self.position[0] - (self.width / 2.), 

+                self.position[1] - (self.height / 2.))

+

+    @property

+    def upper_right(self):

+        return (self.position[0] + (self.width / 2.), 

+                self.position[1] + (self.height / 2.))

+

+    def bounding_box(self):

+        min_x = self.lower_left[0]

+        max_x = self.upper_right[0]

+        min_y = self.lower_left[1]

+        max_y = self.upper_right[1]

+        return ((min_x, max_x), (min_y, max_y))

+

+

+class Obround(Primitive):

+    """

+    """

+    def __init__(self, position, width, height)

+        self.position = position

+        self.width = width

+        self.height = height

+

+    @property

+    def orientation(self):

+        return 'vertical' if self.height > self.width else 'horizontal'

+

+    @property

+    def lower_left(self):

+        return (self.position[0] - (self.width / 2.), 

+                self.position[1] - (self.height / 2.))

+

+    @property

+    def upper_right(self):

+        return (self.position[0] + (self.width / 2.), 

+                self.position[1] + (self.height / 2.))

+

+    def bounding_box(self):

+        min_x = self.lower_left[0]

+        max_x = self.upper_right[0]

+        min_y = self.lower_left[1]

+        max_y = self.upper_right[1]

+        return ((min_x, max_x), (min_y, max_y))

+

+

+class Polygon(Primitive):

+    """

+    """

+    def __init__(self, position, sides, radius):

+        self.position = position

+        self.sides = sides

+        self.radius = radius

+        

+    def bounding_box(self):

+        min_x = self.position[0] - self.radius

+        max_x = self.position[0] + self.radius

+        min_y = self.position[1] - self.radius

+        max_y = self.position[1] + self.radius

+        return ((min_x, max_x), (min_y, max_y))

+

+

+class Region(Primitive):

+    """

+    """

+    def __init__(self, points):

+        self.points = points

+        

+    def bounding_box(self):

+        x_list, y_list = zip(*self.points)

+        min_x = min(x_list)

+        max_x = max(x_list)

+        min_y = min(y_list)

+        max_y = max(y_list)

+        return ((min_x, max_x), (min_y, max_y))

+

+

+class Drill(Primitive):

+    """

+    """

+    def __init__(self, position, diameter):

+        self.position = position

+        self.diameter = diameter

+        self.radius = diameter / 2.

+        

+    def bounding_box(self):

+        min_x = self.position[0] - self.radius

+        max_x = self.position[0] + self.radius

+        min_y = self.position[1] - self.radius

+        max_y = self.position[1] + self.radius

+        return ((min_x, max_x), (min_y, max_y))