From 5a6ab34565fa3d783337a68e07e3500361c8fcf5 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Mon, 4 Mar 2019 00:06:32 +0100
Subject: o PartDB: don't die if the directory exist *and* is empty. Ninja
 creates   empty parent directories. o PartDB: escape some path names. o new
 comamnd: "ninja". Generates a build.ninja from a sch. Not done!

---
 src/ee/tools/__init__.py              |  3 +-
 src/ee/tools/ninja.py                 | 80 +++++++++++++++++++++++++++++++++++
 src/ee/tools/templates/build.ninja.j2 | 53 +++++++++++++++++++++++
 3 files changed, 135 insertions(+), 1 deletion(-)
 create mode 100644 src/ee/tools/ninja.py
 create mode 100644 src/ee/tools/templates/build.ninja.j2

(limited to 'src/ee/tools')

diff --git a/src/ee/tools/__init__.py b/src/ee/tools/__init__.py
index 68e3adc..12916ca 100644
--- a/src/ee/tools/__init__.py
+++ b/src/ee/tools/__init__.py
@@ -1,4 +1,5 @@
 import os.path
+
 from colors import color
 
 
@@ -13,7 +14,7 @@ def _mkdir_and_open(path):
 
 
 def mk_parents(path: str):
-    dirname = os.path.dirname(path)
+    dirname = os.path.diŒrname(path)
 
     if len(dirname) == 0:
         return
diff --git a/src/ee/tools/ninja.py b/src/ee/tools/ninja.py
new file mode 100644
index 0000000..cc090a0
--- /dev/null
+++ b/src/ee/tools/ninja.py
@@ -0,0 +1,80 @@
+import argparse
+import sys
+from pathlib import Path
+from typing import List, Union, Optional
+
+from jinja2 import Environment, PackageLoader, select_autoescape
+
+from ee.kicad import read_schematics
+
+
+def ninja_path_filter(s: Union[str, List[str]]) -> str:
+    if isinstance(s, str):
+        return s. \
+            replace("$", "$$"). \
+            replace(" ", "$ ")
+    elif isinstance(s, list):
+
+        if len(s) == 1:
+            return ninja_path_filter(s[0])
+
+        return "$\n" + " $\n".join(["  " + ninja_path_filter(path) for path in s])
+    else:
+        raise Exception("Unsupported argument type: {}".format(type(s)))
+
+
+def parent_dir_filter(s: str) -> str:
+    return str(Path(s).parent)
+
+
+def generate(sch_path: Path, kicad_bom_strategy: Optional[str]):
+    def _create_env():
+        e = Environment(
+            loader=PackageLoader(__name__, "templates"),
+            autoescape=select_autoescape(["html", "xml"]),
+            keep_trailing_newline=True,
+        )
+        e.filters["ninja_path"] = ninja_path_filter
+        e.filters["parent_dir"] = parent_dir_filter
+        return e
+
+    gerber_zip = "prod/gerber.zip"
+
+    sch = read_schematics(str(sch_path))
+
+    sch_files = sorted([s.path for s in sch.schematics])
+
+    params = {}
+    import os.path
+    params["ee"] = "{} -m ee".format(os.path.relpath(sys.executable, Path(".")))
+    params["sch"] = sch_path
+    params["sch_files"] = sch_files
+
+    params["kicad_bom_strategy"] = kicad_bom_strategy
+
+    params["pcb"] = str(sch_path).replace(".sch", ".kicad_pcb")
+
+    if gerber_zip is not None:
+        params["gerber_zip"] = gerber_zip
+
+    build_ninja = Path("build.ninja")
+
+    with build_ninja.open("w") as f:
+        env = _create_env()
+        template = env.get_template("build.ninja.j2")
+        f.write(template.render(**params))
+
+
+parser = argparse.ArgumentParser()
+
+parser.add_argument("--sch",
+                    required=True,
+                    metavar="FILE")
+
+parser.add_argument("--kicad-bom-strategy",
+                    required=False,
+                    metavar="PY CALLABLE")
+
+args = parser.parse_args()
+
+generate(args.sch, args.kicad_bom_strategy)
diff --git a/src/ee/tools/templates/build.ninja.j2 b/src/ee/tools/templates/build.ninja.j2
new file mode 100644
index 0000000..79a2bc1
--- /dev/null
+++ b/src/ee/tools/templates/build.ninja.j2
@@ -0,0 +1,53 @@
+ee = {{ ee }}
+sch = {{ sch | ninja_path }}
+sch_files = {{ sch_files | ninja_path }}
+pcb = {{ pcb | ninja_path }}
+
+rule kicad-gerber
+    description = kicad-gerber
+    command = $ee kicad-gerber $
+      --output-dir $gerber_dir $
+      --pcb $pcb
+#    mkdir -p $(
+#    (cd $(GERBER_DIR); zip tmp.zip $(foreach GBR,$(GERBERS),$(notdir $(GBR))))
+#    mv $(GERBER_DIR)/tmp.zip $@
+
+rule kicad-make-bom
+    description = kicad-make-bom $out_dir
+    command = $ee kicad-make-bom --sch $sch --out $out_dir $strategy
+
+rule part-create-distributor-search-list
+    description = part-create-distributor-search-list distributor=$distributor $in_dir => $out_dir
+    command = $ee part-create-distributor-search-list --in $in_dir --out $out_dir
+
+rule digikey-search-parts
+    description = digikey-search-parts
+    command = $ee digikey-search-parts --in $in_dir --out $out_dir
+
+rule digikey-normalize-facts
+    description = digikey-normalize-facts
+    command = $ee digikey-normalize-facts --in $in_dir --out $out_dir
+
+{% if gerber_zip is defined %}
+build gerbers: phony {{ gerber_zip }}
+build {{ gerber_zip }}: kicad-gerber $pcb
+    gerber_dir = {{ gerber_zip | parent_dir }}
+{%- endif %}
+
+build ee/sch/index.xml: kicad-make-bom $sch
+    out_dir = ee/sch
+    strategy = {{ "--strategy " + kicad_bom_strategy if kicad_bom_strategy else "" }}
+
+build ee/digikey/search-list/index.xml: part-create-distributor-search-list ee/sch/index.xml
+    in_dir = ee/sch
+    out_dir = ee/digikey/search-list
+
+build ee/digikey/downloaded/index.xml: digikey-search-parts ee/digikey/search-list/index.xml
+    in_dir = ee/digikey/search-list
+    out_dir = ee/digikey/downloaded
+
+build ee/digikey/normalized/index.xml: digikey-normalize-facts ee/digikey/downloaded/index.xml
+    in_dir = ee/digikey/downloaded
+    out_dir = ee/digikey/normalized
+
+default ee/digikey/normalized/index.xml
-- 
cgit v1.2.3