#! /usr/bin/env python # -*- coding: utf-8 -*- # copyright 2014 Hamilton Kibbe # 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))