QFP58

#!/usr/bin/env python3
"""QFP-58 one-shot generator:  python3 generate.py [outdir]   (default: the project root)
   -> KiCad boards (showcase RHCP feed in <outdir>, the rest in kicad/<board>/), docs/ (report, SVGs, previews), native KiCad boards (+ footprint lib), fab Gerbers/drills (+ zips).
   Needs: numpy scipy scikit-rf shapely matplotlib; KiCad 7+ (pcbnew python module + kicad-cli) for boards/Gerbers."""
import os, sys, json, shutil, subprocess, warnings
warnings.filterwarnings('ignore')
from qfp_geom import P, build
from qfp_layers import write_svgs, preview

ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
out = os.path.abspath(sys.argv[1] if len(sys.argv) > 1 else ROOT)
docs = f'{out}/docs'
os.makedirs(docs, exist_ok=True)
rep = {'parameters': P}
for hand in ('RHCP', 'LHCP'):
    rep[hand] = build(hand)[1]
    preview(f'{docs}/preview_{hand}.png', hand)
json.dump(rep, open(f'{docs}/design_report.json', 'w'), indent=1, default=float)
write_svgs(f'{docs}/svg')

try:
    import pcbnew  # noqa
except ImportError:
    sys.exit('pcbnew not found: SVGs/report written, KiCad boards skipped')
import kicad_build
kdir = out
sys.argv = ['kicad_build.py', kdir]
exec(open(kicad_build.__file__).read().split("if __name__ == '__main__':")[1].replace('\n    ', '\n'),
     vars(kicad_build))
layers = 'F.Cu,B.Cu,F.Mask,B.Mask,F.SilkS,B.SilkS,Edge.Cuts'
for n in kicad_build.board_names():
    g = f'{out}/gerbers/{n}'
    os.makedirs(g, exist_ok=True)
    pcb = kicad_build.board_path(out, n)
    subprocess.run(['kicad-cli', 'pcb', 'export', 'gerbers', '-l', layers, '--subtract-soldermask', '-o', g + '/', pcb],
                   check=True, capture_output=True)
    subprocess.run(['kicad-cli', 'pcb', 'export', 'drill', '--excellon-separate-th', '-o', g + '/', pcb],
                   check=True, capture_output=True)
    shutil.make_archive(f'{out}/gerbers/{n}-fab', 'zip', g)
print('done ->', out)