diff options
| -rw-r--r-- | mesh_dialog.py | 29 |
1 files changed, 20 insertions, 9 deletions
diff --git a/mesh_dialog.py b/mesh_dialog.py index 5f79f14..7813489 100644 --- a/mesh_dialog.py +++ b/mesh_dialog.py @@ -125,10 +125,14 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): def tearup_mesh(self, matching=None): count = 0 + anchor, target_layer_id = self.get_anchor() for track in self.board.GetTracks(): if matching is not None and track.GetNet().GetNetname() not in matching: continue + if track.GetLayer() != target_layer_id: + continue + count += 1 self.board.Remove(track) print(f'Tore up {count} trace segments.') @@ -136,6 +140,21 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): def settings_fn(self): return path.join(path.dirname(self.board.GetFileName()), 'last_kimesh_settings.json') + def get_anchor(self): + ref = str(list(self.board.Footprints())[self.m_anchorChoice.GetSelection()].GetReference()) + footprints = [ fp for fp in self.board.Footprints() if fp.GetReference() == ref ] + if len(footprints) == 0: + wx.MessageDialog(self, f'Error: Could not find anchor footprint "{ref}".').ShowModal() + raise ValueError() + if len(footprints) > 1: + wx.MessageDialog(self, f'Error: Multiple footprints with anchor footprint reference "{ref}".').ShowModal() + raise ValueError() + anchor = footprints[0] + pad0, *_ = anchor.Pads() + lset = pad0.GetLayerSet() + target_layer_id, *_ = [l for l in lset.CuStack() if lset.Contains(l)] + return anchor, target_layer_id + def generate_mesh(self, evt): try: settings = GeneratorSettings( @@ -155,16 +174,8 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): except: wx.MessageDialog(self, "Cannot save settings: {}.".format(e), "File I/O error").ShowModal() - anchor = [ fp for fp in self.board.Footprints() if fp.GetReference() == settings.anchor ] - if len(anchor) == 0: - return wx.MessageDialog(self, f'Error: Could not find anchor footprint "{self.m_anchorInput.Value}".').ShowModal() - if len(anchor) > 1: - return wx.MessageDialog(self, f'Error: Multiple footprints with anchor footprint reference "{self.m_anchorInput.Value}".').ShowModal() - anchor, = anchor - pad0, *_ = anchor.Pads() - lset = pad0.GetLayerSet() - target_layer_id, *_ = [l for l in lset.CuStack() if lset.Contains(l)] + anchor, target_layer_id = self.get_anchor() mesh_zones = [] for drawing in self.board.GetDrawings(): |