OpenSCADMachineDesignFramework/scripts/assemblies.py at dc95c4e3585cb14d74c0041934af3b40271da6ec · Axford/OpenSCADMachineDesignFramework · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/usr/bin/env python

# Renders views for assembly steps

import config
import os
import openscad
import shutil
import sys
import c14n_stl
import re
import json
import jsontools
from types import *

from views import polish;
from views import render_view;
from views import render_view_using_file;
from animate import animateAssembly;

def machine_dir(s):
    s = s.replace(" ","")
    return re.sub(r"\W+|\s+", "", s, re.I)


def assemblies():
    print("Assemblies")
    print("----------")

    # load hardware.json
    jf = open(config.paths['json'],"r")
    jso = json.load(jf)
    jf.close()

    # for each machine
    for m in jso:
        if type(m) is dict and m['type'] == 'machine':
            print((m['title']))

            al = m['assemblies']

            # make target directory
            view_dir = config.paths['assemblies']+"/"+machine_dir(m['title'])
            if not os.path.isdir(view_dir):
                os.makedirs(view_dir)

            # for each assembly
            for a in al:
                print(("  "+a['title']))
                fn = config.paths['root'] + a['file']
                if (os.path.isfile(fn)):

                    print("    Checking csg hash")
                    h = openscad.get_csg_hash(config.paths['tempscad'], a['call']);
                    os.remove(config.paths['tempscad']);

                    hashchanged = ('hash' in a and h != a['hash']) or (not 'hash' in a)

                    # update hash in json
                    a['hash'] = h

                    # Steps
                    for step in a['steps']:
                        # Generate step file
                        f = open(config.paths['tempscad'], "w")
                        f.write("include <../config/config.scad>\n")
                        f.write("DebugConnectors = false;\n");
                        f.write("DebugCoordinateFrames = false;\n");
                        f.write("UseSTL=true;\n");
                        f.write("$Explode = true;\n");
                        f.write("$ShowStep = "+ str(step['num']) +";\n");
                        f.write(a['call'] + ";\n");
                        f.close()

                        # Views
                        print(("      Step "+str(step['num'])))
                        for view in step['views']:
                            render_view_using_file(a['title']+'_step'+str(step['num']), config.paths['tempscad'], view_dir, view, hashchanged, h)

                    # for each animation
                    for anim in a['animations']:
                        print(("    Animation: "+anim['title']+", framesPerStep="+str(anim['framesPerStep'])))

                        animateAssembly(m['title'], a['title'], anim['title'], anim['framesPerStep'])

                else:
                    print(("    Error: scad file not found: "+a['file']))


    # Save changes to json
    with open(config.paths['json'], 'w') as f:
        f.write(json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))

    return 0


if __name__ == '__main__':
    assemblies()