This commit is contained in:
jaseg 2020-08-13 17:27:25 +02:00
parent a21f9d909c
commit e126ace7a2
4 changed files with 1641 additions and 1016 deletions

File diff suppressed because it is too large Load diff

View file

@ -2,6 +2,7 @@ import wx
import mesh_plugin_dialog
import pcbnew
from collections import defaultdict
import pyclipper
# Implementing MainDialog
class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog):
@ -13,6 +14,10 @@ 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)
@ -21,6 +26,14 @@ 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) }
@ -72,22 +85,37 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog):
return wx.MessageDialog(self, "Error: More than two connection pads found for net {}.".format(net)).ShowModal()
eco1_id = self.board.GetLayerID('Eco1.User')
mesh_zones = {}
mesh_zones = []
for drawing in self.board.GetDrawings():
if drawing.GetLayer() == eco1_id:
mesh_zones[drawing] = []
mesh_zones.append(drawing)
if len(mesh_zones) == 0:
if not mesh_zones:
return wx.MessageDialog(self, "Error: Could not find any mesh zones on the Eco1.User layer.").ShowModal()
for zone in mesh_zones:
anchors = []
for module in self.board.GetModules():
for foo in module.GraphicalItems():
if not isinstance(foo, pcbnew.TEXTE_MODULE):
continue
if foo.GetText() == "
# find mesh area on eco1, check connection pads are within target area
if foo.GetText() == "mesh_anchor":
anchors.append(module)
break
if not anchors:
return wx.MessageDialog(self, "Error: No anchor found for mesh zone centered on {:.3f}, {:.3f} mm".format(
zone.GetCenter().x / pcbnew.IU_PER_MM, zone.GetCenter().y / pcbnew.IU_PER_MM
)).ShowModal()
if len(anchors) > 1:
return wx.MessageDialog(self, "Error: Currently, only a single anchor is supported.").ShowModal()
self.generate_mesh(zone, anchors)
def generate_mesh(self, zone, anchors):
anchor, = anchors
def update_net_label(self, evt):
self.m_netLabel.SetLabel('{} matching nets'.format(len(self.get_matching_nets())))

View file

@ -1,17 +1,60 @@
import os.path
from os import path
import subprocess
import sys
import wx
import pcbnew
from .mesh_dialog import show_dialog
def check_requirements(*packages):
missing = []
for pkg in packages:
try:
__import__(pkg)
except ImportError:
missing.append(pkg)
if missing:
if 'win' not in sys.platform:
wx.MessageDialog(None, "Error: Missing python dependencies {}.".format(', '.join(missing)),
"Missing Dependencies").ShowModal()
return False
else:
msg = 'The following python dependencies are missing:\n\n' + '\n'.join(missing) +\
'\n\nShould we go ahead and install these missing dependencies into the plugin directory?'
dialog = wx.MessageDialog(None, msg, caption='Error: Missing dependencies', style=wx.YES_NO | wx.NO_DEFAULT)
dialog.SetYesNoLabels("Install missing dependencies", "Cancel")
if dialog.ShowModal() == wx.ID_YES:
for dep in packages:
proc = subprocess.Popen(
"pip install --target deps {} --no-use-pep517 --only-binary :all: --platform win_amd64"\
.format(dep).split(),
cwd=path.dirname(__file__),
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
(stdout, stderr) = proc.communicate()
if proc.returncode != 0:
wx.MessageDialog(None, "Error installing dependencies:\n\n{}\n{}".format(stdout, stderr),
"Installation Error").ShowModal()
return False
sys.path.append(path.abspath(path.join(path.dirname(__file__), 'deps')))
else:
return True
class MeshPlugin(pcbnew.ActionPlugin):
def defaults(self):
self.name = 'Mesh generator'
self.category = 'Modify PCB'
self.description = 'Creates security mesh traces on a PCB'
self.icon_file_name = os.path.join(os.path.dirname(__file__), 'mesh_plugin_icon.png')
self.icon_file_name = path.join(path.dirname(__file__), 'mesh_plugin_icon.png')
self.show_toolbar_button = True
def Run(self):
import pcbnew
if not check_requirements('pyclipper'):
return
from .mesh_dialog import show_dialog
show_dialog(pcbnew.GetBoard())

View file

@ -1,149 +1,133 @@
# -*- coding: utf-8 -*-
###########################################################################
## Python code generated with wxFormBuilder (version Oct 26 2018)
## http://www.wxformbuilder.org/
##
## PLEASE DO *NOT* EDIT THIS FILE!
###########################################################################
import wx
import wx.xrc
###########################################################################
## Class MainDialog
###########################################################################
class MainDialog ( wx.Dialog ):
def __init__( self, parent ):
wx.Dialog.__init__ ( self, parent, id = wx.ID_ANY, title = u"Security Mesh Generator Plugin", pos = wx.DefaultPosition, size = wx.Size( 455,283 ), style = wx.DEFAULT_DIALOG_STYLE )
self.SetSizeHints( wx.DefaultSize, wx.DefaultSize )
bSizer1 = wx.BoxSizer( wx.VERTICAL )
fgSizer1 = wx.FlexGridSizer( 3, 2, 0, 0 )
fgSizer1.AddGrowableCol( 1 )
fgSizer1.AddGrowableRow( 0 )
fgSizer1.AddGrowableRow( 4 )
fgSizer1.SetFlexibleDirection( wx.VERTICAL )
fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
fgSizer1.Add( ( 0, 0), 1, wx.EXPAND, 5 )
fgSizer1.Add( ( 0, 0), 1, wx.EXPAND, 5 )
self.m_staticText1 = wx.StaticText( self, wx.ID_ANY, u"Mesh net name prefix", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText1.Wrap( -1 )
fgSizer1.Add( self.m_staticText1, 1, wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer3 = wx.BoxSizer( wx.VERTICAL )
self.m_net_prefix = wx.TextCtrl( self, wx.ID_ANY, u"mesh-", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer3.Add( self.m_net_prefix, 2, wx.ALIGN_CENTER_VERTICAL|wx.ALL|wx.EXPAND, 5 )
self.m_netLabel = wx.StaticText( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_netLabel.Wrap( -1 )
bSizer3.Add( self.m_netLabel, 0, wx.ALL, 5 )
fgSizer1.Add( bSizer3, 1, wx.EXPAND, 5 )
self.m_staticText3 = wx.StaticText( self, wx.ID_ANY, u"Mesh angle", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText3.Wrap( -1 )
fgSizer1.Add( self.m_staticText3, 0, wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer4 = wx.BoxSizer( wx.HORIZONTAL )
self.m_angleSpin = wx.SpinCtrlDouble( self, wx.ID_ANY, u"deg", wx.DefaultPosition, wx.DefaultSize, wx.SP_ARROW_KEYS|wx.SP_WRAP, 0, 360, 0.000000, 1 )
self.m_angleSpin.SetDigits( 2 )
bSizer4.Add( self.m_angleSpin, 0, wx.ALL, 5 )
self.m_staticText5 = wx.StaticText( self, wx.ID_ANY, u"° (deg)", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText5.Wrap( -1 )
bSizer4.Add( self.m_staticText5, 0, wx.ALL, 5 )
fgSizer1.Add( bSizer4, 1, wx.EXPAND, 5 )
self.m_staticText4 = wx.StaticText( self, wx.ID_ANY, u"Trace width", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText4.Wrap( -1 )
fgSizer1.Add( self.m_staticText4, 0, wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer5 = wx.BoxSizer( wx.HORIZONTAL )
self.m_traceSpin = wx.SpinCtrlDouble( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, wx.SP_ARROW_KEYS, 0, 100, 0.127, 1 )
self.m_traceSpin.SetDigits( 3 )
bSizer5.Add( self.m_traceSpin, 0, wx.ALL, 5 )
self.m_staticText6 = wx.StaticText( self, wx.ID_ANY, u"mm", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText6.Wrap( -1 )
bSizer5.Add( self.m_staticText6, 0, wx.ALL, 5 )
fgSizer1.Add( bSizer5, 1, wx.EXPAND, 5 )
self.m_staticText7 = wx.StaticText( self, wx.ID_ANY, u"Space width", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText7.Wrap( -1 )
fgSizer1.Add( self.m_staticText7, 0, wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer6 = wx.BoxSizer( wx.HORIZONTAL )
self.m_spaceSpin = wx.SpinCtrlDouble( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, wx.SP_ARROW_KEYS, 0, 10000, 0.127, 1 )
self.m_spaceSpin.SetDigits( 3 )
bSizer6.Add( self.m_spaceSpin, 0, wx.ALL, 5 )
self.m_staticText8 = wx.StaticText( self, wx.ID_ANY, u"mm", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText8.Wrap( -1 )
bSizer6.Add( self.m_staticText8, 0, wx.ALL, 5 )
fgSizer1.Add( bSizer6, 1, wx.EXPAND, 5 )
fgSizer1.Add( ( 0, 0), 1, wx.EXPAND, 5 )
fgSizer1.Add( ( 0, 0), 1, wx.EXPAND, 5 )
bSizer1.Add( fgSizer1, 1, wx.ALL|wx.EXPAND, 15 )
wSizer1 = wx.WrapSizer( wx.HORIZONTAL, wx.WRAPSIZER_DEFAULT_FLAGS )
self.m_cancelButton = wx.Button( self, wx.ID_ANY, u"Cancel", wx.DefaultPosition, wx.DefaultSize, 0 )
wSizer1.Add( self.m_cancelButton, 0, wx.ALL, 5 )
wSizer1.Add( ( 0, 0), 1, wx.EXPAND, 5 )
self.m_removeButton = wx.Button( self, wx.ID_ANY, u"Remove Mesh Traces", wx.DefaultPosition, wx.DefaultSize, 0 )
wSizer1.Add( self.m_removeButton, 0, wx.ALL, 5 )
self.m_generateButton = wx.Button( self, wx.ID_ANY, u"Generate", wx.DefaultPosition, wx.DefaultSize, 0 )
wSizer1.Add( self.m_generateButton, 0, wx.ALL, 5 )
bSizer1.Add( wSizer1, 0, wx.ALL|wx.EXPAND, 3 )
self.SetSizer( bSizer1 )
self.Layout()
self.Centre( wx.BOTH )
def __del__( self ):
pass
# -*- coding: utf-8 -*-
###########################################################################
## Python code generated with wxFormBuilder (version Jun 17 2015)
## http://www.wxformbuilder.org/
##
## PLEASE DO "NOT" EDIT THIS FILE!
###########################################################################
import wx
import wx.xrc
###########################################################################
## Class MainDialog
###########################################################################
class MainDialog ( wx.Dialog ):
def __init__( self, parent ):
wx.Dialog.__init__ ( self, parent, id = wx.ID_ANY, title = u"Security Mesh Generator Plugin", pos = wx.DefaultPosition, size = wx.Size( 496,258 ), style = wx.CLOSE_BOX|wx.DEFAULT_DIALOG_STYLE|wx.MINIMIZE_BOX|wx.RESIZE_BORDER|wx.STAY_ON_TOP )
self.SetSizeHintsSz( wx.DefaultSize, wx.DefaultSize )
bSizer1 = wx.BoxSizer( wx.VERTICAL )
self.m_scrolledWindow1 = wx.ScrolledWindow( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.HSCROLL|wx.VSCROLL )
self.m_scrolledWindow1.SetScrollRate( 5, 5 )
fgSizer1 = wx.FlexGridSizer( 3, 2, 0, 0 )
fgSizer1.AddGrowableCol( 1 )
fgSizer1.SetFlexibleDirection( wx.VERTICAL )
fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
self.m_staticText1 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"Mesh net name prefix", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText1.Wrap( -1 )
fgSizer1.Add( self.m_staticText1, 1, wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_RIGHT|wx.ALL, 5 )
self.m_net_prefix = wx.TextCtrl( self.m_scrolledWindow1, wx.ID_ANY, u"mesh-", wx.DefaultPosition, wx.DefaultSize, 0 )
fgSizer1.Add( self.m_net_prefix, 2, wx.ALIGN_CENTER_VERTICAL|wx.EXPAND, 5 )
fgSizer1.AddSpacer( ( 0, 0), 1, wx.EXPAND, 5 )
self.m_netLabel = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"0 matching nets", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_netLabel.Wrap( -1 )
fgSizer1.Add( self.m_netLabel, 0, wx.ALL, 5 )
self.m_staticText3 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"Mesh angle", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText3.Wrap( -1 )
fgSizer1.Add( self.m_staticText3, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer4 = wx.BoxSizer( wx.HORIZONTAL )
self.m_angleInput = wx.TextCtrl( self.m_scrolledWindow1, wx.ID_ANY, u"0.00", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer4.Add( self.m_angleInput, 0, wx.ALIGN_CENTER_VERTICAL, 5 )
self.m_staticText5 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"° (deg)", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText5.Wrap( -1 )
bSizer4.Add( self.m_staticText5, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
fgSizer1.Add( bSizer4, 1, wx.EXPAND, 5 )
self.m_staticText4 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"Trace width", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText4.Wrap( -1 )
fgSizer1.Add( self.m_staticText4, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer5 = wx.BoxSizer( wx.HORIZONTAL )
self.m_traceInput = wx.TextCtrl( self.m_scrolledWindow1, wx.ID_ANY, u"0.127", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer5.Add( self.m_traceInput, 0, wx.ALIGN_CENTER_VERTICAL, 5 )
self.m_spinBtn1 = wx.SpinButton( self.m_scrolledWindow1, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.SP_VERTICAL )
bSizer5.Add( self.m_spinBtn1, 0, wx.ALL, 5 )
self.m_staticText6 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"mm", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText6.Wrap( -1 )
bSizer5.Add( self.m_staticText6, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
fgSizer1.Add( bSizer5, 1, wx.EXPAND, 5 )
self.m_staticText7 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"Space width", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText7.Wrap( -1 )
fgSizer1.Add( self.m_staticText7, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_RIGHT|wx.ALL, 5 )
bSizer6 = wx.BoxSizer( wx.HORIZONTAL )
self.m_spaceInput = wx.TextCtrl( self.m_scrolledWindow1, wx.ID_ANY, u"0.127", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer6.Add( self.m_spaceInput, 0, wx.ALIGN_CENTER_VERTICAL, 5 )
self.m_spinBtn2 = wx.SpinButton( self.m_scrolledWindow1, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.SP_VERTICAL )
bSizer6.Add( self.m_spinBtn2, 0, wx.ALL, 5 )
self.m_staticText8 = wx.StaticText( self.m_scrolledWindow1, wx.ID_ANY, u"mm", wx.DefaultPosition, wx.DefaultSize, 0 )
self.m_staticText8.Wrap( -1 )
bSizer6.Add( self.m_staticText8, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
fgSizer1.Add( bSizer6, 1, wx.EXPAND, 5 )
self.m_scrolledWindow1.SetSizer( fgSizer1 )
self.m_scrolledWindow1.Layout()
fgSizer1.Fit( self.m_scrolledWindow1 )
bSizer1.Add( self.m_scrolledWindow1, 1, wx.EXPAND |wx.ALL, 5 )
bSizer99 = wx.BoxSizer( wx.HORIZONTAL )
self.m_cancelButton = wx.Button( self, wx.ID_ANY, u"Cancel", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer99.Add( self.m_cancelButton, 0, wx.ALL, 5 )
bSizer99.AddSpacer( ( 0, 0), 1, wx.EXPAND, 5 )
self.m_removeButton = wx.Button( self, wx.ID_ANY, u"Remove Mesh Traces", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer99.Add( self.m_removeButton, 0, wx.ALL, 5 )
self.m_generateButton = wx.Button( self, wx.ID_ANY, u"Generate", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer99.Add( self.m_generateButton, 0, wx.ALL, 5 )
bSizer1.Add( bSizer99, 0, wx.ALL|wx.EXPAND, 3 )
self.SetSizer( bSizer1 )
self.Layout()
self.Centre( wx.BOTH )
def __del__( self ):
pass