NopSCADlib/scripts/plateup.py

#!/usr/bin/env python

#
# NopSCADlib Copyright Chris Palmer 2018
# nop.head@gmail.com
# hydraraptor.blogspot.com
#
# This file is part of NopSCADlib.
#
# NopSCADlib is free software: you can redistribute it and/or modify it under the terms of the
# GNU General Public License as published by the Free Software Foundation, either version 3 of
# the License, or (at your option) any later version.
#
# NopSCADlib is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with NopSCADlib.
# If not, see <https://www.gnu.org/licenses/>.
#

from __future__ import print_function
import os
import openscad
import sys
import c14n_stl
from set_config import *
from deps import *
import shutil
import re
import time
import times

source_dirs = { "stl" : "platters", "dxf" : "panels" }
target_dirs = { "stl" : "printed",  "dxf" : "routed" }

def plateup(target, part_type, usage = None):
    #
    # Make the target directory
    #
    top_dir = set_config(target, usage)
    parts_dir = top_dir + part_type + 's'
    target_dir = parts_dir + '/' + target_dirs[part_type]
    source_dir1 = source_dirs[part_type]
    source_dir2 = top_dir + source_dirs[part_type]

    #
    # Loop through source directories
    #
    all_used = []
    all_sources = []
    all_parts = []
    read_times = False
    for dir in [source_dir1, source_dir2]:
        if not os.path.isdir(dir):
            continue
        if not os.path.isdir(target_dir):
            os.makedirs(target_dir)

        if not read_times:
            times.read_times(target_dir)
            read_times = True
        #
        # Make the deps dir
        #
        deps_dir = parts_dir + "/deps"
        if not os.path.isdir(deps_dir):
            os.makedirs(deps_dir)

        if os.path.isdir(dir + '/deps'): #old deps
            shutil.rmtree(dir + '/deps')
        #
        # Decide which files to make
        #
        sources = [file for file in os.listdir(dir) if file.endswith('.scad')]
        all_sources += sources
        #
        # Run OpenSCAD on the source files to make the targets
        #
        target_def = ['-D$target="%s"' % target] if target else []
        cwd_def = ['-D$cwd="%s"' % os.getcwd().replace('\\', '/')]
        for src in sources:
            src_file = dir + '/' + src
            part =  src[:-4] + part_type
            all_parts.append(part)
            part_file = target_dir + '/' + part
            uses_file = deps_dir + '/' + src[:-4] + 'txt'
            dname = deps_name(deps_dir, src)
            oldest = part_file if mtime(part_file) < mtime(uses_file) else uses_file
            changed = check_deps(oldest, dname)
            used = []
            if changed:
                print(changed)
                t = time.time()
                openscad.run_list(["-D$bom=1"] + target_def + cwd_def + ["-d", dname, "-o", part_file, src_file])
                if part_type == 'stl':
                    c14n_stl.canonicalise(part_file)
                times.add_time(part, t)
                log_name = 'openscad.log'
                #
                # Add the files on the BOM to the used list
                #
                with open(log_name) as file:
                    for line in file.readlines():
                        match = re.match(r'^ECHO: "~(.*?\.' + part_type + r').*"$', line)
                        if match:
                            used.append(match.group(1))
                with open(uses_file, "wt") as file:
                    for part in used:
                        print(part, file = file)
            else:
                with open(uses_file, "rt") as file:
                    for line in file:
                        used.append(line[:-1])
            all_used += used

    copied = []
    if all_sources:
        #
        # Copy files that are not included
        #
        for file in os.listdir(parts_dir):
            if file.endswith('.' + part_type) and not file in all_used:
                src = parts_dir + '/' + file
                dst = target_dir + '/' + file
                if mtime(src) > mtime(dst):
                    print("Copying %s to %s" % (src, dst))
                    shutil.copyfile(src, dst)
                copied.append(file)
        #
        # Remove any cruft
        #
        targets = [file[:-4] + part_type for file in all_sources]
        for file in os.listdir(target_dir):
            if file.endswith('.' + part_type):
                if not file in targets and not file in copied:
                    print("Removing %s" % file)
                    os.remove(target_dir + '/' + file)

        targets = [file[:-4] + 'txt' for file in all_sources]
        for file in os.listdir(deps_dir):
            if file.endswith('.' + 'txt'):
                if not file in targets:
                    print("Removing %s" % file)
                    os.remove(deps_dir + '/' + file)

        times.print_times(all_parts)