diff options
Diffstat (limited to 'plugin/mesh_dialog.py')
| -rw-r--r-- | plugin/mesh_dialog.py | 142 |
1 files changed, 121 insertions, 21 deletions
diff --git a/plugin/mesh_dialog.py b/plugin/mesh_dialog.py index 97943a8..3b4e91a 100644 --- a/plugin/mesh_dialog.py +++ b/plugin/mesh_dialog.py @@ -1,14 +1,32 @@ from collections import defaultdict +from dataclasses import dataclass import wx import pcbnew import shapely +from shapely.geometry import polygon +from shapely import affinity from . import mesh_plugin_dialog -# Implementing MainDialog +class GeneratorError(ValueError): + pass + +class AbortError(SystemError): + pass + +@dataclass +class GeneratorSettings: + mesh_angle: float = 0.0 # deg + trace_width: float = 0.127 # mm + space_width: float = 0.127 # mm + anchor_exit: float = 0.0 # deg + num_traces: int = 2 + offset_x: float = 0.0 # mm + offset_y: float = 0.0 # mm + class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): def __init__(self, board): mesh_plugin_dialog.MainDialog.__init__(self, None) @@ -18,10 +36,6 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): self.m_removeButton.Bind(wx.EVT_BUTTON, self.tearup_mesh) self.m_generateButton.Bind(wx.EVT_BUTTON, self.generate_mesh) self.m_net_prefix.Bind(wx.EVT_TEXT, self.update_net_label) - self.m_traceSpin.Bind(wx.EVT_SPIN_UP, lambda evt: self.spin(self.m_traceInput, 1.0)) - self.m_traceSpin.Bind(wx.EVT_SPIN_DOWN, lambda evt: self.spin(self.m_traceInput, -1.0)) - self.m_spaceSpin.Bind(wx.EVT_SPIN_UP, lambda evt: self.spin(self.m_spaceInput, 1.0)) - self.m_spaceSpin.Bind(wx.EVT_SPIN_DOWN, lambda evt: self.spin(self.m_spaceInput, -1.0)) self.tearup_confirm_dialog = wx.MessageDialog(self, "", style=wx.YES_NO | wx.NO_DEFAULT) @@ -30,14 +44,6 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): self.SetMinSize(self.GetSize()) - def spin(self, le_input, delta): - try: - current = float(le_input.Value) - current += delta - le_input.Value = '{:.03f}'.format(current) - except ValueError: - pass - def get_matching_nets(self): prefix = self.m_net_prefix.Value return { net for net in self.nets if net.startswith(prefix) } @@ -59,16 +65,31 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): self.tearup_confirm_dialog.SetYesNoLabels("Tear up {} nets".format(len(matching)), "Close") if self.tearup_confirm_dialog.ShowModal() == wx.ID_YES: - for track in self.board.GetTracks(): - if not (track.GetStatus() & pcbnew.TRACK_AR): - continue + self.tearup_mesh() + + def tearup_mesh(self): + for track in self.board.GetTracks(): + if not (track.GetStatus() & pcbnew.TRACK_AR): + continue - if not track.GetNet().GetNetname() in matching: - continue + if not track.GetNet().GetNetname() in matching: + continue - board.Remove(track) + board.Remove(track) def generate_mesh(self, evt): + try: + settings = GeneratorSettings( + mesh_angle = float(self.m_angleSpin.Value), + trace_width = float(self.m_traceSpin.Value), + space_width = float(self.m_spaceSpin.Value), + anchor_exit = float(self.m_exitSpin.Value), + num_traces = int(self.m_traceCountSpin.Value), + offset_x = float(self.m_offsetXSpin.Value), + offset_y = float(self.m_offsetYSpin.Value)) + except ValueError as e: + return wx.MessageDialog(self, "Invalid input value: {}.".format(e), "Invalid input").ShowModal() + nets = self.get_matching_nets() pads = defaultdict(lambda: []) @@ -115,11 +136,90 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): if len(anchors) > 1: return wx.MessageDialog(self, "Error: Currently, only a single anchor is supported.").ShowModal() - self.generate_mesh(zone, anchors) + try: + def warn(msg): + dialog = wx.MessageDialog(self, msg + '\n\nDo you want to abort mesh generation?', + "Mesh Generation Warning").ShowModal() + dialog = wx.MessageDialog(self, "", style=wx.YES_NO | wx.NO_DEFAULT) + dialog.SetYesNoLabels("Abort", "Ignore and continue") + + if self.tearup_confirm_dialog.ShowModal() == wx.ID_YES: + raise AbortError() + + self.generate_mesh_backend(zone, anchors, warn=warn) + + except GeneratorError as e: + return wx.MessageDialog(self, str(e), "Mesh Generation Error").ShowModal() + except AbortError: + pass + + def poly_set_to_shapely(self, poly_set): + for i in range(poly_set.OutlineCount()): + outline = poly_set.Outline(i) + + outline_points = [] + for j in range(outline.PointCount()): + point = outline.CPoint(j) + outline_points.append((pcbnew.ToMM(point.x), pcbnew.ToMM(point.y))) + yield polygon.LinearRing(outline_points) - def generate_mesh(self, zone, anchors): + def generate_mesh_backend(self, zone, anchors, warn=lambda s: None, settings=GeneratorSettings()): anchor, = anchors + anchor_outlines = list(self.poly_set_to_shapely(anchor.GetBoundingPoly())) + if len(anchor_outlines) == 0: + raise GeneratorError('Could not find any outlines for anchor {}'.format(anchor.GetReference())) + if len(anchor_outlines) > 1: + warn('Anchor {} has multiple outlines. Using first outline for trace start.') + + zone_outlines = list(self.poly_set_to_shapely(zone.GetPolyShape())) + if len(zone_outlines) == 0: + raise GeneratorError('Could not find any outlines for mesh zone.') + if len(zone_outlines) > 1: + raise GeneratorError('Mesh zone has too many outlines (has {}, should have one).'.format(len(zone_outlines))) + zone_outline, *_rest = zone_outlines + + mesh_origin = zone_outline.centroid + width_per_trace = settings.trace_width + settings.space_width + grid_cell_width = width_per_trace * settings.num_traces + + zone_outline_rotated = affinity.rotate(zone_outline, -settings.mesh_angle, origin=mesh_origin) + bbox = zone_outline_rotated.bounds + + grid_origin = (bbox[0] + settings.offset_x - grid_cell_width, bbox[1] + settings.offset_y - grid_cell_width) + grid_rows = int((bbox[3] - grid_origin[1]) / grid_cell_width + 2) + grid_cols = int((bbox[2] - grid_origin[0]) / grid_cell_width + 2) + print(f'generating grid of size {grid_rows} * {grid_cols}') + + grid = [] + for y in range(grid_rows): + row = [] + for x in range(grid_cols): + cell = polygon.LinearRing([(0, 0), (0, 1), (1, 1), (1, 0)]) + cell = affinity.scale(cell, grid_cell_width, grid_cell_width, origin=(0, 0)) + cell = affinity.translate(cell, mesh_origin.x + x*grid_cell_width, mesh_origin.y + y*grid_cell_width) + cell = affinity.rotate(cell, settings.mesh_angle, origin=mesh_origin) + row.append(cell) + grid.append(row) + break + + for row in grid: + for cell in row: + poly = pcbnew.DRAWSEGMENT() + poly.SetLayer(self.board.GetLayerID('Eco2.User')) + poly.SetShape(pcbnew.S_POLYGON) + poly.SetWidth(0) + self.board.Add(poly) + s = poly.GetPolyShape() + s.NewOutline() + for x, y in zip(*cell.xy): + s.Append(pcbnew.FromMM(x), pcbnew.FromMM(y)) + print('OUTLINE POINT', x, y) + break + + pcbnew.Refresh() + #self.tearup_mesh() + # TODO generate def update_net_label(self, evt): self.m_netLabel.SetLabel('{} matching nets'.format(len(self.get_matching_nets()))) |