gerbonara/gerber/primitives.py

#! /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))