diff options
Diffstat (limited to 'gerber/primitives.py')
| -rw-r--r-- | gerber/primitives.py | 228 |
1 files changed, 228 insertions, 0 deletions
diff --git a/gerber/primitives.py b/gerber/primitives.py new file mode 100644 index 0000000..670b758 --- /dev/null +++ b/gerber/primitives.py @@ -0,0 +1,228 @@ +#! /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 __init__(self, level_polarity='dark'):
+ self.level_polarity = level_polarity
+
+ 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, level_polarity='dark'):
+ super(Line, self).__init__(level_polarity)
+ self.start = start
+ self.end = end
+ self.width = width
+
+ @property
+ def angle(self):
+ delta_x, delta_y = tuple(map(sub, end, start))
+ angle = degrees(math.tan(delta_y/delta_x))
+ return angle
+
+ @property
+ 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, level_polarity='dark'):
+ super(Arc, self).__init__(level_polarity)
+ 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)
+
+ @property
+ def bounding_box(self):
+ pass
+
+class Circle(Primitive):
+ """
+ """
+ def __init__(self, position, diameter, level_polarity='dark'):
+ super(Circle, self).__init__(level_polarity)
+ self.position = position
+ self.diameter = diameter
+
+ @property
+ def radius(self):
+ return self.diameter / 2.
+
+ @property
+ 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))
+
+ @property
+ def stroke_width(self):
+ return self.diameter
+
+
+class Rectangle(Primitive):
+ """
+ """
+ def __init__(self, position, width, height, level_polarity='dark'):
+ super(Rectangle, self).__init__(level_polarity)
+ 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.))
+
+ @property
+ 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))
+
+ @property
+ def stroke_width(self):
+ return max((self.width, self.height))
+
+
+class Obround(Primitive):
+ """
+ """
+ def __init__(self, position, width, height, level_polarity='dark'):
+ super(Obround, self).__init__(level_polarity)
+ 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.))
+
+ @property
+ 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, level_polarity='dark'):
+ super(Polygon, self).__init__(level_polarity)
+ self.position = position
+ self.sides = sides
+ self.radius = radius
+
+ @property
+ 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, level_polarity='dark'):
+ super(Region, self).__init__(level_polarity)
+ self.points = points
+
+ @property
+ 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):
+ super(Drill, self).__init__('dark')
+ self.position = position
+ self.diameter = diameter
+
+ @property
+ def radius(self):
+ return self.diameter / 2.
+
+ @property
+ 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))
|