152 lines
6.5 KiB
Python
152 lines
6.5 KiB
Python
import os
|
|
import wx
|
|
import pcbnew # type: ignore
|
|
|
|
from .thread import ProcessThread
|
|
from .events import StatusEvent
|
|
from .options import AUTO_FILL_OPT, AUTO_TRANSLATE_OPT, EXCLUDE_DNP_OPT, EXTEND_EDGE_CUT_OPT, ALTERNATIVE_EDGE_CUT_OPT, EXTRA_LAYERS, ALL_ACTIVE_LAYERS_OPT
|
|
from .utils import load_user_options, save_user_options, get_layer_names
|
|
|
|
|
|
# WX GUI form that show the plugin progress
|
|
class KiCadToJLCForm(wx.Frame):
|
|
def __init__(self):
|
|
wx.Dialog.__init__(
|
|
self,
|
|
None,
|
|
id=wx.ID_ANY,
|
|
title=u"Fabrication Toolkit",
|
|
pos=wx.DefaultPosition,
|
|
size=wx.DefaultSize,
|
|
style=wx.DEFAULT_DIALOG_STYLE)
|
|
|
|
# self.app = wx.PySimpleApp()
|
|
icon = wx.Icon(os.path.join(os.path.dirname(__file__), 'icon.png'))
|
|
self.SetIcon(icon)
|
|
self.SetBackgroundColour(wx.LIGHT_GREY)
|
|
|
|
self.SetSizeHints(wx.Size(600, 100), wx.DefaultSize)
|
|
|
|
userOptions = load_user_options({
|
|
EXTRA_LAYERS: "",
|
|
ALL_ACTIVE_LAYERS_OPT: False,
|
|
EXTEND_EDGE_CUT_OPT: False,
|
|
ALTERNATIVE_EDGE_CUT_OPT: False,
|
|
AUTO_TRANSLATE_OPT: True,
|
|
AUTO_FILL_OPT: True,
|
|
EXCLUDE_DNP_OPT: False
|
|
})
|
|
|
|
self.mOptionsLabel = wx.StaticText(self, label='Options:')
|
|
# self.mOptionsSeparator = wx.StaticLine(self)
|
|
|
|
layers = get_layer_names(pcbnew.GetBoard())
|
|
self.mAdditionalLayersControl = wx.TextCtrl(self, size=wx.Size(600, 50))
|
|
self.mAdditionalLayersControl.Hint = "Additional layers (comma-separated)"
|
|
self.mAdditionalLayersControl.AutoComplete(layers)
|
|
self.mAdditionalLayersControl.Enable()
|
|
self.mAdditionalLayersControl.SetValue(userOptions[EXTRA_LAYERS])
|
|
self.mAllActiveLayersCheckbox = wx.CheckBox(self, label='Plot all active layers')
|
|
self.mAllActiveLayersCheckbox.SetValue(userOptions[ALL_ACTIVE_LAYERS_OPT])
|
|
self.mExtendEdgeCutsCheckbox = wx.CheckBox(self, label='Set User.1 as V-Cut layer')
|
|
self.mExtendEdgeCutsCheckbox.SetValue(userOptions[EXTEND_EDGE_CUT_OPT])
|
|
self.mAlternativeEdgeCutsCheckbox = wx.CheckBox(self, label='Use User.2 for alternative Edge-Cut layer')
|
|
self.mAlternativeEdgeCutsCheckbox.SetValue(userOptions[ALTERNATIVE_EDGE_CUT_OPT])
|
|
self.mAutomaticTranslationCheckbox = wx.CheckBox(self, label='Apply automatic translations')
|
|
self.mAutomaticTranslationCheckbox.SetValue(userOptions[AUTO_TRANSLATE_OPT])
|
|
self.mAutomaticFillCheckbox = wx.CheckBox(self, label='Apply automatic fill for all zones')
|
|
self.mAutomaticFillCheckbox.SetValue(userOptions[AUTO_FILL_OPT])
|
|
self.mExcludeDnpCheckbox = wx.CheckBox(self, label='Exclude DNP components from BOM')
|
|
self.mExcludeDnpCheckbox.SetValue(userOptions[EXCLUDE_DNP_OPT])
|
|
|
|
self.mGaugeStatus = wx.Gauge(
|
|
self, wx.ID_ANY, 100, wx.DefaultPosition, wx.Size(600, 20), wx.GA_HORIZONTAL)
|
|
self.mGaugeStatus.SetValue(0)
|
|
self.mGaugeStatus.Hide()
|
|
|
|
self.mGenerateButton = wx.Button(self, label='Generate', size=wx.Size(600, 60))
|
|
self.mGenerateButton.Bind(wx.EVT_BUTTON, self.onGenerateButtonClick)
|
|
|
|
boxSizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
boxSizer.Add(self.mOptionsLabel, 0, wx.ALL, 5)
|
|
# boxSizer.Add(self.mOptionsSeparator, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mAdditionalLayersControl, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mAllActiveLayersCheckbox, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mExtendEdgeCutsCheckbox, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mAlternativeEdgeCutsCheckbox, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mAutomaticTranslationCheckbox, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mAutomaticFillCheckbox, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mExcludeDnpCheckbox, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mGaugeStatus, 0, wx.ALL, 5)
|
|
boxSizer.Add(self.mGenerateButton, 0, wx.ALL, 5)
|
|
|
|
self.SetSizer(boxSizer)
|
|
self.Layout()
|
|
boxSizer.Fit(self)
|
|
|
|
self.Centre(wx.BOTH)
|
|
|
|
# Bind the ESC key event to a handler
|
|
self.Bind(wx.EVT_CHAR_HOOK, self.onKey)
|
|
|
|
# Close the dialog when pressing the ESC key
|
|
def onKey(self, event):
|
|
if event.GetKeyCode() == wx.WXK_ESCAPE:
|
|
self.Close(True)
|
|
else:
|
|
event.Skip()
|
|
|
|
|
|
def onGenerateButtonClick(self, event):
|
|
options = dict()
|
|
options[EXTRA_LAYERS] = self.mAdditionalLayersControl.GetValue()
|
|
options[ALL_ACTIVE_LAYERS_OPT] = self.mAllActiveLayersCheckbox.GetValue()
|
|
options[EXTEND_EDGE_CUT_OPT] = self.mExtendEdgeCutsCheckbox.GetValue()
|
|
options[ALTERNATIVE_EDGE_CUT_OPT] = self.mAlternativeEdgeCutsCheckbox.GetValue()
|
|
options[AUTO_TRANSLATE_OPT] = self.mAutomaticTranslationCheckbox.GetValue()
|
|
options[AUTO_FILL_OPT] = self.mAutomaticFillCheckbox.GetValue()
|
|
options[EXCLUDE_DNP_OPT] = self.mExcludeDnpCheckbox.GetValue()
|
|
|
|
save_user_options(options)
|
|
|
|
self.mOptionsLabel.Hide()
|
|
self.mAdditionalLayersControl.Hide()
|
|
self.mAllActiveLayersCheckbox.Hide()
|
|
self.mExtendEdgeCutsCheckbox.Hide()
|
|
self.mAlternativeEdgeCutsCheckbox.Hide()
|
|
self.mAutomaticTranslationCheckbox.Hide()
|
|
self.mAutomaticFillCheckbox.Hide()
|
|
self.mExcludeDnpCheckbox.Hide()
|
|
self.mGenerateButton.Hide()
|
|
self.mGaugeStatus.Show()
|
|
|
|
self.Fit()
|
|
self.SetTitle('Fabrication Toolkit (Processing...)')
|
|
|
|
StatusEvent.invoke(self, self.updateDisplay)
|
|
ProcessThread(self, options)
|
|
|
|
def updateDisplay(self, status):
|
|
if status.data == -1:
|
|
self.SetTitle('Fabrication Toolkit (Done!)')
|
|
pcbnew.Refresh()
|
|
self.Destroy()
|
|
else:
|
|
self.mGaugeStatus.SetValue(int(status.data))
|
|
|
|
|
|
# Plugin definition
|
|
class Plugin(pcbnew.ActionPlugin):
|
|
def __init__(self):
|
|
self.name = "Fabrication Toolkit"
|
|
self.category = "Manufacturing"
|
|
self.description = "Toolkit for automating PCB fabrication process with KiCad and JLC PCB"
|
|
self.pcbnew_icon_support = hasattr(self, "show_toolbar_button")
|
|
self.show_toolbar_button = True
|
|
self.icon_file_name = os.path.join(os.path.dirname(__file__), 'icon.png')
|
|
self.dark_icon_file_name = os.path.join(os.path.dirname(__file__), 'icon.png')
|
|
|
|
def Run(self):
|
|
KiCadToJLCForm().Show()
|