brendanhaines/kicad
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

black

Browse Source
This commit is contained in:
Brendan Haines 2022-10-01 14:57:23 -06:00
parent a433fa8304
commit 610e37f107
2 changed files with 57 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
scripting/plugins/OutputGeneration/FabOutputs.py
View File

@ -1,8 +1,9 @@
import pcbnew
import os import os
import zipfile
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
import zipfile
import pcbnew
__all__ = ["FabOutputs"] __all__ = ["FabOutputs"]
@ -45,33 +46,36 @@ class FabOutputs(pcbnew.ActionPlugin):
("Front Silkscreen", "gto", pcbnew.F_SilkS), ("Front Silkscreen", "gto", pcbnew.F_SilkS),
("Front Mask", "gts", pcbnew.F_Mask), ("Front Mask", "gts", pcbnew.F_Mask),
("Front Copper", "gtl", pcbnew.F_Cu), ("Front Copper", "gtl", pcbnew.F_Cu),
*[(f'Inner Layer {layer} Copper', f'g{layer}', layer) for layer in range(1, layer_count-1)], *[
('Back Copper', 'gbl', pcbnew.B_Cu), (f"Inner Layer {layer} Copper", f"g{layer}", layer)
('Back Mask', 'gbs', pcbnew.B_Mask), for layer in range(1, layer_count - 1)
('Back SilkScreen', 'gbo', pcbnew.B_SilkS), ],
('Back Paste', 'gbp', pcbnew.B_Paste), ("Back Copper", "gbl", pcbnew.B_Cu),
('Edges Cuts', 'gm1', pcbnew.Edge_Cuts), ("Back Mask", "gbs", pcbnew.B_Mask),
('Drill', 'drl', None), ("Back SilkScreen", "gbo", pcbnew.B_SilkS),
("Back Paste", "gbp", pcbnew.B_Paste),
("Edges Cuts", "gm1", pcbnew.Edge_Cuts),
("Drill", "drl", None),
] ]
stackup = [ stackup = [
# [pcbnew layer, file extension, thickness, comment] # [pcbnew layer, file extension, thickness, comment]
[pcbnew.F_Paste, 'gtp', None, "SN63/PB37"], [pcbnew.F_Paste, "gtp", None, "SN63/PB37"],
[pcbnew.F_SilkS, 'gto', None, "White"], [pcbnew.F_SilkS, "gto", None, "White"],
[pcbnew.F_Mask, 'gts', 1, "Explicit mask material"], [pcbnew.F_Mask, "gts", 1, "Explicit mask material"],
[None, None, None, "ENIG"], [None, None, None, "ENIG"],
[pcbnew.F_Cu, 'gtl', 2.1, "copper roughness"], [pcbnew.F_Cu, "gtl", 2.1, "copper roughness"],
[None, None, 10, "Dielectric stuff"], [None, None, 10, "Dielectric stuff"],
[1, 'g1', 0.7, "Copper roughness"], [1, "g1", 0.7, "Copper roughness"],
[None, None, 24, "Dielectric stuff"], [None, None, 24, "Dielectric stuff"],
[None, None, 12, "Dielectric stuff"], [None, None, 12, "Dielectric stuff"],
[2, 'g2', 0.7, "Copper roughness"], [2, "g2", 0.7, "Copper roughness"],
[None, None, 10, "Dielectric stuff"], [None, None, 10, "Dielectric stuff"],
[pcbnew.B_Cu, 'gbl', 2.1, "copper roughness"], [pcbnew.B_Cu, "gbl", 2.1, "copper roughness"],
[None, None, None, "ENIG"], [None, None, None, "ENIG"],
[pcbnew.B_Mask, 'gbs', 1, "Explicit mask material"], [pcbnew.B_Mask, "gbs", 1, "Explicit mask material"],
[pcbnew.B_SilkS, 'gbo', None, "White"], [pcbnew.B_SilkS, "gbo", None, "White"],
[pcbnew.B_Paste, 'gbp', None, "SN63/PB37"], [pcbnew.B_Paste, "gbp", None, "SN63/PB37"],
] ]
board_features = { board_features = {
@ -81,7 +85,7 @@ class FabOutputs(pcbnew.ActionPlugin):
"copper finish": "ENIG", "copper finish": "ENIG",
"hard gold": False, "hard gold": False,
"bevelled edge": False, "bevelled edge": False,
"soldermask defined": None, # TODO: how do I want to determine this? "soldermask defined": None, # TODO: how do I want to determine this?
} }
dir_fab.mkdir(parents=True, exist_ok=True) dir_fab.mkdir(parents=True, exist_ok=True)
@ -99,7 +103,7 @@ class FabOutputs(pcbnew.ActionPlugin):
plot_controller = pcbnew.PLOT_CONTROLLER(pcb) plot_controller = pcbnew.PLOT_CONTROLLER(pcb)
plot_options = plot_controller.GetPlotOptions() plot_options = plot_controller.GetPlotOptions()
# Set General Options: # Set General Options:
# plot_options.Format() # plot_options.Format()
plot_options.SetOutputDirectory(dir_fab) plot_options.SetOutputDirectory(dir_fab)
@ -114,8 +118,8 @@ class FabOutputs(pcbnew.ActionPlugin):
plot_options.SetNegative(False) plot_options.SetNegative(False)
plot_options.SetScale(1) plot_options.SetScale(1)
# plot_options.SetAutoScale(True) # plot_options.SetAutoScale(True)
#plot_options.SetPlotMode(PLOT_MODE) # plot_options.SetPlotMode(PLOT_MODE)
#plot_options.SetLineWidth(pcbnew.FromMM(PLOT_LINE_WIDTH)) # plot_options.SetLineWidth(pcbnew.FromMM(PLOT_LINE_WIDTH))
plot_options.SetUseGerberAttributes(True) plot_options.SetUseGerberAttributes(True)
plot_options.SetUseGerberProtelExtensions(False) plot_options.SetUseGerberProtelExtensions(False)
plot_options.SetCreateGerberJobFile(False) plot_options.SetCreateGerberJobFile(False)
@ -126,28 +130,30 @@ class FabOutputs(pcbnew.ActionPlugin):
plot_plan = [ plot_plan = [
# ( layer ID, file extension, description) # ( layer ID, file extension, description)
( pcbnew.F_Paste, 'gtp', 'Front Paste' ), (pcbnew.F_Paste, "gtp", "Front Paste"),
( pcbnew.F_SilkS, 'gto', 'Front SilkScreen' ), (pcbnew.F_SilkS, "gto", "Front SilkScreen"),
( pcbnew.F_Mask, 'gts', 'Front Mask' ), (pcbnew.F_Mask, "gts", "Front Mask"),
( pcbnew.F_Cu, 'gtl', 'Front Copper' ), (pcbnew.F_Cu, "gtl", "Front Copper"),
*[(layer, f'g{layer}', f'Inner Layer {layer} Copper') for layer in range(1, layer_count-1)], *[
( pcbnew.B_Cu, 'gbl', 'Back Copper' ), (layer, f"g{layer}", f"Inner Layer {layer} Copper")
( pcbnew.B_Mask, 'gbs', 'Back Mask' ), for layer in range(1, layer_count - 1)
( pcbnew.B_SilkS, 'gbo', 'Back SilkScreen' ), ],
( pcbnew.B_Paste, 'gbp', 'Back Paste' ), (pcbnew.B_Cu, "gbl", "Back Copper"),
( pcbnew.Edge_Cuts, 'gm1', 'Edges Cuts' ), (pcbnew.B_Mask, "gbs", "Back Mask"),
(pcbnew.B_SilkS, "gbo", "Back SilkScreen"),
(pcbnew.B_Paste, "gbp", "Back Paste"),
(pcbnew.Edge_Cuts, "gm1", "Edges Cuts"),
] ]
for layer_info in plot_plan: for layer_info in plot_plan:
plot_controller.SetLayer(layer_info[0]) plot_controller.SetLayer(layer_info[0])
plot_controller.OpenPlotfile('', pcbnew.PLOT_FORMAT_GERBER, layer_info[2]) plot_controller.OpenPlotfile("", pcbnew.PLOT_FORMAT_GERBER, layer_info[2])
plot_controller.PlotLayer() plot_controller.PlotLayer()
fname = f"{project_name}{suffix}.{layer_info[1]}" fname = f"{project_name}{suffix}.{layer_info[1]}"
os.rename(dir_fab / f"{project_name}.gbr", dir_fab / fname) os.rename(dir_fab / f"{project_name}.gbr", dir_fab / fname)
files_fab.append(fname) files_fab.append(fname)
plot_controller.ClosePlot() plot_controller.ClosePlot()
# ================ # ================
@ -160,7 +166,7 @@ class FabOutputs(pcbnew.ActionPlugin):
MANTISSA_DIGITS = 3 MANTISSA_DIGITS = 3
MIRROR_Y_AXIS = False MIRROR_Y_AXIS = False
HEADER = True HEADER = True
OFFSET = pcbnew.wxPoint(0,0) OFFSET = pcbnew.wxPoint(0, 0)
MERGE_PTH_NPTH = True MERGE_PTH_NPTH = True
DRILL_FILE = True DRILL_FILE = True
MAP_FILE = False MAP_FILE = False
@ -169,7 +175,9 @@ class FabOutputs(pcbnew.ActionPlugin):
drill_writer = pcbnew.EXCELLON_WRITER(pcb) drill_writer = pcbnew.EXCELLON_WRITER(pcb)
drill_writer.SetFormat(METRIC, ZERO_FORMAT, INTEGER_DIGITS, MANTISSA_DIGITS) drill_writer.SetFormat(METRIC, ZERO_FORMAT, INTEGER_DIGITS, MANTISSA_DIGITS)
drill_writer.SetOptions(MIRROR_Y_AXIS, HEADER, OFFSET, MERGE_PTH_NPTH) drill_writer.SetOptions(MIRROR_Y_AXIS, HEADER, OFFSET, MERGE_PTH_NPTH)
drill_writer.CreateDrillandMapFilesSet(str(dir_fab), DRILL_FILE, MAP_FILE, REPORTER) drill_writer.CreateDrillandMapFilesSet(
str(dir_fab), DRILL_FILE, MAP_FILE, REPORTER
)
fname = f"{project_name}{suffix}.drl" fname = f"{project_name}{suffix}.drl"
os.rename(dir_fab / f"{project_name}.drl", dir_fab / fname) os.rename(dir_fab / f"{project_name}.drl", dir_fab / fname)
@ -200,7 +208,7 @@ class FabOutputs(pcbnew.ActionPlugin):
# f.write(f"Layer Order\n") # f.write(f"Layer Order\n")
# # for layer in plot_plan: # # for layer in plot_plan:
# files_fab.append(fname) # files_fab.append(fname)
# ================ # ================
# Assembly Notes # Assembly Notes
# ================ # ================
@ -214,20 +222,24 @@ class FabOutputs(pcbnew.ActionPlugin):
# Zip # Zip
# ================ # ================
with zipfile.ZipFile(dir_fab / f"{project_name}{suffix}_fabrication.zip", "w") as z: with zipfile.ZipFile(
dir_fab / f"{project_name}{suffix}_fabrication.zip", "w"
) as z:
for fname in files_fab: for fname in files_fab:
z.write(dir_fab / fname, arcname=fname) z.write(dir_fab / fname, arcname=fname)
with zipfile.ZipFile(dir_asy / f"{project_name}{suffix}_assembly.zip", "w") as z: with zipfile.ZipFile(
dir_asy / f"{project_name}{suffix}_assembly.zip", "w"
) as z:
for fname in files_fab: for fname in files_fab:
z.write(dir_fab / fname, arcname=Path("fabrication") / fname) z.write(dir_fab / fname, arcname=Path("fabrication") / fname)
for fname in files_asy: for fname in files_asy:
z.write(dir_asy / fname, arcname=fname) z.write(dir_asy / fname, arcname=fname)
# dir_archive = dir_pcb / "Archive" # dir_archive = dir_pcb / "Archive"
# with zipfile.ZipFile(dir_archive / f"{project_name}{suffix}_archive.zip", "w") as z: # with zipfile.ZipFile(dir_archive / f"{project_name}{suffix}_archive.zip", "w") as z:
# for fname in files_fab: # for fname in files_fab:
# z.write(dir_fab / fname, arcname=Path("fabrication") / fname) # z.write(dir_fab / fname, arcname=Path("fabrication") / fname)
# for fname in files_asy: # for fname in files_asy:
# z.write(dir_asy / fname, arcname=Path("assembly") / fname) # z.write(dir_asy / fname, arcname=Path("assembly") / fname)
# # TODO: archive project here # # TODO: archive project here

3
scripting/plugins/OutputGeneration/__init__.py
View File

@ -1,2 +1,3 @@
from .FabOutputs import FabOutputs from .FabOutputs import FabOutputs
FabOutputs().register() # Instantiate and register to Pcbnew
FabOutputs().register() # Instantiate and register to Pcbnew