views.py

#!/usr/bin/env python

from __future__ import print_function
import os
import sys
import shutil
import openscad
import re
from set_machine import *
from time import *

def views(machine, clean = False):
    scad_dir = "views"
    render_dir = machine + os.sep + "views"

    if os.path.isdir(render_dir):
        shutil.rmtree(render_dir)
        sleep(0.1)
    os.makedirs(render_dir)
    #
    # Set the target machine
    #
    set_machine(machine)

    #
    # List of individual part files
    #
    scads = [i for i in os.listdir(scad_dir) if i[-5:] == ".scad"]

    for scad in scads:
        scad_name = scad_dir + os.sep + scad
        png_name = render_dir + os.sep + scad[:-4] + "png"

        dx = None
        rx = None
        d = None
        for line in open(scad_name, "r").readlines():
            m = re.match(r'\$vpt *= *\[ *(.*) *, *(.*) *, *(.*) *\].*', line[:-1])
            if m:
                dx = float(m.group(1))
                dy = float(m.group(2))
                dz = float(m.group(3))
            m = re.match(r'\$vpr *= *\[ *(.*) *, *(.*) *, *(.*) *\].*', line[:-1])
            if m:
                rx = float(m.group(1))
                ry = float(m.group(2))
                rz = float(m.group(3))
            m = re.match(r'\$vpd *= * *(.*) *;.*', line[:-1])
            if m:
                d = float(m.group(1))
            words = line.split()
            if len(words) > 3 and words[0] == "//":
                cmd = words[1]
                if cmd == "view" or cmd == "assembled" or cmd == "assembly":
                    w = int(words[2]) * 2
                    h = int(words[3]) * 2

                    if len(words) > 10:
                        dx = float(words[4])
                        dy = float(words[5])
                        dz = float(words[6])

                        rx = float(words[7])
                        ry = float(words[8])
                        rz = float(words[9])

                        d = float(words[10])

                    if dx == None or rx == None or d == None:
                        print("Missing camera data in " + scad_name)
                        sys.exit(1)

                    camera = "%.2f,%.2f,%.2f,%.2f,%.2f,%.2f,%.2f" % (dx, dy, dz, rx, ry, rz, d)

                    exploded = "0"
                    if cmd == "assembly":
                        exploded = "1"

                    if cmd == "assembled":
                        png_name = png_name.replace("assembly","assembled")

                    if not os.path.isfile(png_name) or os.path.getmtime(png_name) < os.path.getmtime(scad_name):
                        openscad.run("--projection=p",
                                    ("--imgsize=%d,%d" % (w, h)),
                                     "--camera=" + camera,
                                      "-D$exploded=" + exploded,
                                      "-o", png_name, scad_name)
                        print()

if __name__ == '__main__':
    if len(sys.argv) > 1:
        views(sys.argv[1])
    else:
        print("usage: views dibond|mendel|sturdy|huxley|your_machine")
        sys.exit(1)