first commit

bom.py

@@ -0,0 +1,138 @@
+import os
+import subprocess
+import sys
+import shutil
+
+class BOM():
+    def __init__(self):
+        self.count = 1
+        self.vitamins = {}
+        self.printed = {}
+        self.assemblies = {}
+
+    def add_part(self, s):
+        if s[-4:] == ".stl":
+            parts = self.printed
+        else:
+            parts = self.vitamins
+        if s in parts:
+            parts[s] += 1
+        else:
+            parts[s] = 1
+
+    def add_assembly(self, ass):
+        if ass in self.assemblies:
+            self.assemblies[ass].count += 1
+        else:
+            self.assemblies[ass] = BOM()
+
+    def make_name(self, ass):
+        if self.count == 1:
+            return ass
+        return ass.replace("assembly", "assemblies")
+
+    def print_bom(self, breakdown, file = None):
+        if breakdown:
+            longest = 0
+            for ass in self.assemblies:
+                name = ass.replace("_assembly","")
+                longest = max(longest, len(name))
+            for i in range(longest):
+                print >> file, " " * 14,
+                for ass in sorted(self.assemblies):
+                    name = ass.replace("_assembly","").replace("_"," ")
+                    if longest - i > len(name):
+                        print >> file, "  ",
+                    else:
+                        print >> file, " %s" % name[i - (longest - len(name))],
+                print >> file
+
+        print >> file, "Vitamins:"
+        for part in sorted(self.vitamins):
+            if ': ' in part:
+                part_no, description = part.split(': ')
+            else:
+                part_no, description = "", part
+            print >> file, "%3d %-10s" % (self.vitamins[part], part_no),
+            if breakdown:
+                for ass in sorted(self.assemblies):
+                    bom = self.assemblies[ass]
+                    if part in bom.vitamins:
+                        print >> file, "%2d" % bom.vitamins[part],
+                    else:
+                        print >> file, "  ",
+
+            print >> file, description
+
+        print >> file
+        print >> file, "Printed:"
+        for part in sorted(self.printed):
+            print >> file, "%3d" % self.printed[part],
+            if breakdown:
+                print >> file, " " * 10,
+                for ass in sorted(self.assemblies):
+                    bom = self.assemblies[ass]
+                    if part in bom.printed:
+                        print >> file, "%2d" % bom.printed[part],
+                    else:
+                        print >> file, "  ",
+
+            print >> file, part
+
+        print >> file
+        if self.assemblies:
+            print >> file, "Sub-assemblies:"
+        for ass in sorted(self.assemblies):
+            print  >> file, "%3d %s" % (self.assemblies[ass].count, self.assemblies[ass].make_name(ass))
+
+def boms(machine):
+    bom_dir = machine + "/bom"
+    if os.path.isdir(bom_dir):
+        shutil.rmtree(bom_dir)
+    os.makedirs(bom_dir)
+
+    f = open("scad/conf/machine.scad","wt")
+    f. write("include <%s_config.scad>\n" % machine);
+    f.close()
+
+    subprocess.call(["openscad_cl", "-o", "dummy.stl", "scad/bom.scad"])
+    print "Generating bom ...",
+
+    main = BOM()
+    stack = []
+
+    for line in open("openscad.log"):
+        if line[:7] == 'ECHO: "':
+            s = line[7:-2]
+            if s[-1] == '/':
+                ass = s[:-1]
+                if stack:
+                    main.assemblies[stack[-1]].add_assembly(ass)    #add to nested BOM
+                stack.append(ass)
+                main.add_assembly(ass)                              #add to flat BOM
+            else:
+                if s[0] == '/':
+                    if s[1:] != stack[-1]:
+                        raise Exception, "Mismatched assembly " + s[1:] + str(stack)
+                    stack.pop()
+                else:
+                    main.add_part(s)
+                    main.assemblies[stack[-1]].add_part(s)
+
+    main.print_bom(True, open(bom_dir + "/bom.txt","wt"))
+
+    for ass in sorted(main.assemblies):
+        f = open(bom_dir + "/" + ass + ".txt", "wt");
+        bom = main.assemblies[ass]
+        print >> f, bom.make_name(ass) + ":"
+        bom.print_bom(False, f)
+        f.close()
+
+    print " done"
+
+if __name__ == '__main__':
+    if len(sys.argv) > 1:
+        boms(sys.argv[1])
+    else:
+        print "usage: bom [mendel|sturdy|your_machine]"
+        sys.exit(1)