From 6f876edd09d9b81649691e529f85653f14b8fd1c Mon Sep 17 00:00:00 2001 From: Hamilton Kibbe Date: Tue, 22 Dec 2015 02:45:48 -0500 Subject: Add PCB interface this incorporates some of @chintal's layers.py changes PCB.from_directory() simplifies loading of multiple gerbers the PCB() class should be pretty helpful going forward... the context classes could use some cleaning up, although I'd like to wait until the freecad stuff gets merged, that way we can try to refactor the context base to support more use cases --- gerber/layers.py | 246 ++++++++++++++++++++++++++++++++++++++++++++++--------- 1 file changed, 209 insertions(+), 37 deletions(-) (limited to 'gerber/layers.py') diff --git a/gerber/layers.py b/gerber/layers.py index b10cf16..c6a5bf7 100644 --- a/gerber/layers.py +++ b/gerber/layers.py @@ -15,40 +15,212 @@ # See the License for the specific language governing permissions and # limitations under the License. -top_copper_ext = ['gtl', 'cmp', 'top', ] -top_copper_name = ['art01', 'top', 'GTL', 'layer1', 'soldcom', 'comp', ] - -bottom_copper_ext = ['gbl', 'sld', 'bot', 'sol', ] -bottom_coppper_name = ['art02', 'bottom', 'bot', 'GBL', 'layer2', 'soldsold', ] - -internal_layer_ext = ['in', 'gt1', 'gt2', 'gt3', 'gt4', 'gt5', 'gt6', 'g1', - 'g2', 'g3', 'g4', 'g5', 'g6', ] -internal_layer_name = ['art', 'internal'] - -power_plane_name = ['pgp', 'pwr', ] -ground_plane_name = ['gp1', 'gp2', 'gp3', 'gp4', 'gt5', 'gp6', 'gnd', - 'ground', ] - -top_silk_ext = ['gto', 'sst', 'plc', 'ts', 'skt', ] -top_silk_name = ['sst01', 'topsilk', 'silk', 'slk', 'sst', ] - -bottom_silk_ext = ['gbo', 'ssb', 'pls', 'bs', 'skb', ] -bottom_silk_name = ['sst', 'bsilk', 'ssb', 'botsilk', ] - -top_mask_ext = ['gts', 'stc', 'tmk', 'smt', 'tr', ] -top_mask_name = ['sm01', 'cmask', 'tmask', 'mask1', 'maskcom', 'topmask', - 'mst', ] - -bottom_mask_ext = ['gbs', 'sts', 'bmk', 'smb', 'br', ] -bottom_mask_name = ['sm', 'bmask', 'mask2', 'masksold', 'botmask', 'msb', ] - -top_paste_ext = ['gtp', 'tm'] -top_paste_name = ['sp01', 'toppaste', 'pst'] - -bottom_paste_ext = ['gbp', 'bm'] -bottom_paste_name = ['sp02', 'botpaste', 'psb'] - -board_outline_ext = ['gko'] -board_outline_name = ['BDR', 'border', 'out', ] - - +import os +import re +from collections import namedtuple + +from .excellon import ExcellonFile +from .ipc356 import IPC_D_356 +from .render.render import Renderable + +Hint = namedtuple('Hint', 'layer ext name') + +hints = [ + Hint(layer='top', + ext=['gtl', 'cmp', 'top', ], + name=['art01', 'top', 'GTL', 'layer1', 'soldcom', 'comp', ] + ), + Hint(layer='bottom', + ext=['gbl', 'sld', 'bot', 'sol', 'bottom', ], + name=['art02', 'bottom', 'bot', 'GBL', 'layer2', 'soldsold', ] + ), + Hint(layer='internal', + ext=['in', 'gt1', 'gt2', 'gt3', 'gt4', 'gt5', 'gt6', 'g1', + 'g2', 'g3', 'g4', 'g5', 'g6', ], + name=['art', 'internal', 'pgp', 'pwr', 'gp1', 'gp2', 'gp3', 'gp4', + 'gt5', 'gp6', 'gnd', 'ground', ] + ), + Hint(layer='topsilk', + ext=['gto', 'sst', 'plc', 'ts', 'skt', 'topsilk', ], + name=['sst01', 'topsilk', 'silk', 'slk', 'sst', ] + ), + Hint(layer='bottomsilk', + ext=['gbo', 'ssb', 'pls', 'bs', 'skb', 'bottomsilk', ], + name=['bsilk', 'ssb', 'botsilk', ] + ), + Hint(layer='topmask', + ext=['gts', 'stc', 'tmk', 'smt', 'tr', 'topmask', ], + name=['sm01', 'cmask', 'tmask', 'mask1', 'maskcom', 'topmask', + 'mst', ] + ), + Hint(layer='bottommask', + ext=['gbs', 'sts', 'bmk', 'smb', 'br', 'bottommask', ], + name=['sm', 'bmask', 'mask2', 'masksold', 'botmask', 'msb', ] + ), + Hint(layer='toppaste', + ext=['gtp', 'tm', 'toppaste', ], + name=['sp01', 'toppaste', 'pst'] + ), + Hint(layer='bottompaste', + ext=['gbp', 'bm', 'bottompaste', ], + name=['sp02', 'botpaste', 'psb'] + ), + Hint(layer='outline', + ext=['gko', 'outline', ], + name=['BDR', 'border', 'out', ] + ), + Hint(layer='ipc_netlist', + ext=['ipc'], + name=[], + ), +] + + +def guess_layer_class(filename): + try: + directory, name = os.path.split(filename) + name, ext = os.path.splitext(name.lower()) + for hint in hints: + patterns = [r'^(\w*[.-])*{}([.-]\w*)?$'.format(x) for x in hint.name] + if ext[1:] in hint.ext or any(re.findall(p, name, re.IGNORECASE) for p in patterns): + return hint.layer + except: + pass + return 'unknown' + + +def sort_layers(layers): + layer_order = ['outline', 'toppaste', 'topsilk', 'topmask', 'top', + 'internal', 'bottom', 'bottommask', 'bottomsilk', + 'bottompaste', 'drill', ] + output = [] + drill_layers = [layer for layer in layers if layer.layer_class == 'drill'] + internal_layers = list(sorted([layer for layer in layers if layer.layer_class == 'internal'])) + + for layer_class in layer_order: + if layer_class == 'internal': + output += internal_layers + elif layer_class == 'drill': + output += drill_layers + else: + for layer in layers: + if layer.layer_class == layer_class: + output.append(layer) + return output + + +class PCBLayer(Renderable): + """ Base class for PCB Layers + + Parameters + ---------- + source : CAMFile + CAMFile representing the layer + + + Attributes + ---------- + filename : string + Source Filename + + """ + @classmethod + def from_gerber(cls, camfile): + filename = camfile.filename + layer_class = guess_layer_class(filename) + if isinstance(camfile, ExcellonFile) or (layer_class == 'drill'): + return DrillLayer.from_gerber(camfile) + elif layer_class == 'internal': + return InternalLayer.from_gerber(camfile) + if isinstance(camfile, IPC_D_356): + layer_class = 'ipc_netlist' + return cls(filename, layer_class, camfile) + + def __init__(self, filename=None, layer_class=None, cam_source=None, **kwargs): + super(PCBLayer, self).__init__(**kwargs) + self.filename = filename + self.layer_class = layer_class + self.cam_source = cam_source + self.surface = None + self.primitives = cam_source.primitives if cam_source is not None else [] + + @property + def bounds(self): + if self.cam_source is not None: + return self.cam_source.bounds + else: + return None + + +class DrillLayer(PCBLayer): + @classmethod + def from_gerber(cls, camfile): + return cls(camfile.filename, camfile) + + def __init__(self, filename=None, cam_source=None, layers=None, **kwargs): + super(DrillLayer, self).__init__(filename, 'drill', cam_source, **kwargs) + self.layers = layers if layers is not None else ['top', 'bottom'] + + +class InternalLayer(PCBLayer): + @classmethod + def from_gerber(cls, camfile): + filename = camfile.filename + try: + order = int(re.search(r'\d+', filename).group()) + except: + order = 0 + return cls(filename, camfile, order) + + def __init__(self, filename=None, cam_source=None, order=0, **kwargs): + super(InternalLayer, self).__init__(filename, 'internal', cam_source, **kwargs) + self.order = order + + def __eq__(self, other): + if not hasattr(other, 'order'): + raise TypeError() + return (self.order == other.order) + + def __ne__(self, other): + if not hasattr(other, 'order'): + raise TypeError() + return (self.order != other.order) + + def __gt__(self, other): + if not hasattr(other, 'order'): + raise TypeError() + return (self.order > other.order) + + def __lt__(self, other): + if not hasattr(other, 'order'): + raise TypeError() + return (self.order < other.order) + + def __ge__(self, other): + if not hasattr(other, 'order'): + raise TypeError() + return (self.order >= other.order) + + def __le__(self, other): + if not hasattr(other, 'order'): + raise TypeError() + return (self.order <= other.order) + + +class LayerSet(Renderable): + def __init__(self, name, layers, **kwargs): + super(LayerSet, self).__init__(**kwargs) + self.name = name + self.layers = list(layers) + + def __len__(self): + return len(self.layers) + + def __getitem__(self, item): + return self.layers[item] + + def to_render(self): + return self.layers + + def apply_theme(self, theme): + pass -- cgit From 6a005436b475e3517fd6a583473b60e601bcc661 Mon Sep 17 00:00:00 2001 From: Hamilton Kibbe Date: Fri, 1 Jan 2016 12:25:38 -0500 Subject: Refactor a little pulled all rendering stuff out of the pcb/layer objects --- gerber/layers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'gerber/layers.py') diff --git a/gerber/layers.py b/gerber/layers.py index c6a5bf7..2b73893 100644 --- a/gerber/layers.py +++ b/gerber/layers.py @@ -21,7 +21,7 @@ from collections import namedtuple from .excellon import ExcellonFile from .ipc356 import IPC_D_356 -from .render.render import Renderable + Hint = namedtuple('Hint', 'layer ext name') @@ -109,7 +109,7 @@ def sort_layers(layers): return output -class PCBLayer(Renderable): +class PCBLayer(object): """ Base class for PCB Layers Parameters @@ -207,7 +207,7 @@ class InternalLayer(PCBLayer): return (self.order <= other.order) -class LayerSet(Renderable): +class LayerSet(object): def __init__(self, name, layers, **kwargs): super(LayerSet, self).__init__(**kwargs) self.name = name -- cgit From 8cd842a41a55ab3d8f558a2e3e198beba7da58a1 Mon Sep 17 00:00:00 2001 From: Hamilton Kibbe Date: Thu, 21 Jan 2016 03:57:44 -0500 Subject: Manually mere rendering changes --- gerber/layers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'gerber/layers.py') diff --git a/gerber/layers.py b/gerber/layers.py index 2b73893..29e452b 100644 --- a/gerber/layers.py +++ b/gerber/layers.py @@ -95,7 +95,8 @@ def sort_layers(layers): 'bottompaste', 'drill', ] output = [] drill_layers = [layer for layer in layers if layer.layer_class == 'drill'] - internal_layers = list(sorted([layer for layer in layers if layer.layer_class == 'internal'])) + internal_layers = list(sorted([layer for layer in layers + if layer.layer_class == 'internal'])) for layer_class in layer_order: if layer_class == 'internal': @@ -151,6 +152,8 @@ class PCBLayer(object): else: return None + def __repr__(self): + return ''.format(self.layer_class) class DrillLayer(PCBLayer): @classmethod @@ -163,6 +166,7 @@ class DrillLayer(PCBLayer): class InternalLayer(PCBLayer): + @classmethod def from_gerber(cls, camfile): filename = camfile.filename @@ -208,6 +212,7 @@ class InternalLayer(PCBLayer): class LayerSet(object): + def __init__(self, name, layers, **kwargs): super(LayerSet, self).__init__(**kwargs) self.name = name -- cgit