diff options
Diffstat (limited to 'src/ee/tools/kicad_make_pos.py')
| -rw-r--r-- | src/ee/tools/kicad_make_pos.py | 65 |
1 files changed, 57 insertions, 8 deletions
diff --git a/src/ee/tools/kicad_make_pos.py b/src/ee/tools/kicad_make_pos.py index cd381a5..0b5e9b4 100644 --- a/src/ee/tools/kicad_make_pos.py +++ b/src/ee/tools/kicad_make_pos.py @@ -1,6 +1,7 @@ import sys import argparse import csv +import datetime as dt from . import mk_parents from ..kicad import pcb, parse_ref from .._utils import run_filters @@ -17,13 +18,24 @@ parser.add_argument("--out", metavar="FILE", help="The output file") +parser.add_argument("--format", + metavar="FORMAT", + default="plain", + help="The format of the output file. Either 'plain' or 'kicad'") + args = parser.parse_args() pcb = pcb.parse(args.pcb) +fmt = args.format + +if fmt != "plain" and fmt != "kicad": + print("Bad format: {}".format(fmt), file=sys.stderr) + def run(stream): rows = [] - for m in pcb.modules: + modules = [m for m in pcb.modules if m.has_attr("smd")] + for m in modules: row = {} ref_fp_text = next(m.filter_fp_text(kind = "reference"), None) row["ref"] = ref= ref_fp_text and ref_fp_text.value @@ -31,7 +43,9 @@ def run(stream): value_fp_text = next(m.filter_fp_text(kind = "value"), None) row["value"] = value_fp_text and value_fp_text.value (row["library"], row["footprint"]) = m.footprint.split(":") - (row["x"], row["y"], row["rot"]) = m.at + (x, y, rotation) = m.at + # The Y axis is inverted for some unknown reason. + (row["x"], row["y"], row["rotation"]) = (float(x), -float(y), float(rotation)) if m.layer == "F.Cu": row["side"] = "top" elif m.layer == "B.Cu": @@ -40,13 +54,48 @@ def run(stream): row["side"] = None rows.append(row) - rows = sorted(rows, key=lambda row: row["r"]) + if fmt == "plain": + rows = sorted(rows, key=lambda row: row["r"]) + + out = csv.writer(stream) + out.writerow(["ref", "value", "library", "footprint", "x", "y", "rotation", "side"]) + for row in rows: + del row["r"] + out.writerow(row.values()) + elif fmt == "kicad": + lines = [] + for row in rows: + row["x"] = "{:>9.4f}".format(row["x"]) + row["y"] = "{:>9.4f}".format(row["y"]) + row["rotation"] = "{:>9.4f}".format(float(row["rotation"])) + + def find_max(name, mn): + return str(max(mn, max((len(str(row[name])) for row in rows)) + 1)) + ref_sz = find_max("ref", 8) + value_sz = find_max("value", 8) + footprint_sz = find_max("footprint", 16) + x_sz = find_max("x", 10) + y_sz = find_max("y", 10) + rotation_sz = find_max("rotation", 0) + fmts = "{ref:" + ref_sz + "}" +\ + " {value:" + value_sz + "}" +\ + " {footprint:" + footprint_sz + "}" +\ + " {x} {y} {rotation}" +\ + " {side}" + #print("fmts={}".format(fmts)) + for row in rows: + #print("row={}".format(row)) + s = fmts.format(**row) + lines.append(s) - out = csv.writer(stream) - out.writerow(["ref", "value", "library", "footprint", "x", "y", "rotation", "side"]) - for row in rows: - del row["r"] - out.writerow(row.values()) + print("### Module positions - created on {:%Y-%m-%d %H:%M:%S} ###".format(dt.datetime.now()), file=stream) + print("### Printed by ee tools", file=stream) + print("## Unit = mm, Angle = deg.", file=stream) + print("## Side : All", file=stream) + print("# Ref Val Package PosX PosY Rot Side", file=stream) + for line in lines: + print(line, file=stream) + print("## End", file=stream) if args.out: mk_parents(args.out) |