aboutsummaryrefslogtreecommitdiff
path: root/src/ee/tools
diff options
context:
space:
mode:
Diffstat (limited to 'src/ee/tools')
-rw-r--r--src/ee/tools/kicad_make_pos.py65
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)