#!/usr/bin/env python3

# Export-import test
#
#
# Usage: <script> <inputfile> [--openscad=<executable-path>] --format=<format> --require-manifold [<openscad args>] file.png
#
# step 1. If the input file is _not_ an .scad file, create a temporary .scad file importing the input file.
# step 2. Run OpenSCAD on the .scad file, output an export format (csg, stl, off, dxf, svg, amf, 3mf)
# step 3. If the export format is _not_ .csg, create a temporary new .scad file importing the exported file
# step 4. Run OpenSCAD on the .csg or .scad file, export to the given .png file
# step 5. (done in CTest) - compare the generated .png file to expected output
#         of the original .scad file. they should be the same!
#
# All the optional openscad args are passed on to OpenSCAD both in step 2 and 4.
# Exception: In any --render arguments are passed, the first pass (step 2) will always
# be run with --render=force while the second pass (step 4) will use the passed --render
# argument.
#
# This script should return 0 on success, not-0 on error.
#
# The CSG file tests do not include the use<fontfile> statements, so to allow the
# export tests to find the font files in the test data directory, the OPENSCAD_FONT_PATH
# is set to the tests/data/ttf directory.
#
# Authors: Torsten Paul, Don Bright, Marius Kintel


import sys, os, re, subprocess, argparse
from validatestl import validateSTL

def failquit(*args):
    if len(args)!=0: print(args, file=sys.stderr)
    print('export_import_pngtest args:',str(sys.argv), file=sys.stderr)
    print('exiting export_import_pngtest.py with failure', file=sys.stderr)
    sys.exit(1)

def createImport(inputfile, scadfile):
        inputfilename = os.path.split(inputfile)[1]
        print ('createImport: ' + inputfile + " " + scadfile, file=sys.stderr)
        outputdir = os.path.dirname(scadfile)
        try:
            if outputdir and not os.path.exists(outputdir): os.mkdir(outputdir)
            f = open(scadfile,'w')
            f.write('import("'+inputfilename+'");'+os.linesep)
            f.close()
        except:
            failquit('failure while opening/writing ' + scadfile + ': ' + str(sys.exc_info()))

#
# Parse arguments
#
formats = ['csg', 'asciistl', 'binstl', 'stl', 'off', 'amf', '3mf', 'obj', 'dxf', 'svg']
parser = argparse.ArgumentParser()
parser.add_argument('--openscad', required=False, default=os.environ["OPENSCAD_BINARY"],
    help='Specify OpenSCAD executable, default to env["OPENSCAD_BINARY"] if absent.')
parser.add_argument('--format', required=True, choices=[item for sublist in [(f,f.upper()) for f in formats] for item in sublist], help='Specify 3d export format')
parser.add_argument('--require-manifold', dest='requiremanifold', action='store_true', help='Require STL output to be manifold')
parser.set_defaults(requiremanifold=False)
args,remaining_args = parser.parse_known_args()

args.format = args.format.lower()

export_format = None
if args.format == 'asciistl':
    export_format = 'asciistl'
    args.format = 'stl'
elif args.format == 'binstl':
    export_format = 'binstl'
    args.format = 'stl'

inputfile = remaining_args[0]         # Can be .scad file or a file to be imported
pngfile = remaining_args[-1]
remaining_args = remaining_args[1:-1] # Passed on to the OpenSCAD executable

if not os.path.exists(inputfile):
    failquit('cant find input file named: ' + inputfile)
if not os.path.exists(args.openscad):
    failquit('cant find openscad executable named: ' + args.openscad)

outputdir = os.path.dirname(pngfile)
inputpath, inputfilename = os.path.split(inputfile)
inputbasename,inputsuffix = os.path.splitext(inputfilename)

if args.format == 'csg':
    # Must export to same folder for include/use/import to work
    exportfile = os.path.abspath(inputfile + '.' + args.format)
else:
    exportfile = os.path.join(outputdir, inputfilename)
    if args.format != inputsuffix[1:]: exportfile += '.' + args.format

# If we're not reading an .scad or .csg file, we need to import it.
if inputsuffix != '.scad' and inputsuffix != '.csg':
    # FIXME: Remove tempfile if created
    tempfile = os.path.join(outputdir, inputfilename + '.scad')
    createImport(inputfile, tempfile)
    inputfile = tempfile

#
# First run: Just export the given filetype
# For any --render arguments to --render=force
#
tmpargs =  ['--render=force' if arg.startswith('--render') else arg for arg in remaining_args]

if export_format is not None:
    tmpargs.extend(['--export-format', export_format])

export_cmd = [args.openscad, inputfile, '-o', exportfile] + tmpargs
print('Running OpenSCAD #1:', file=sys.stderr)
print(' '.join(export_cmd), file=sys.stderr)
sys.stderr.flush()
result = subprocess.call(export_cmd)
stat = os.stat(exportfile)
print(stat, file=sys.stderr)
if result != 0:
    failquit('OpenSCAD #1 failed with return code ' + str(result))

if args.format == 'stl' and args.requiremanifold:
    if not validateSTL(exportfile):
        failquit("Error: Non-manifold STL file exported from OpenSCAD")


#
# Second run: Import the exported file and render as png
#
newscadfile = exportfile
# If we didn't export a .csg file, we need to import it
if args.format != 'csg':
    newscadfile += '.scad'
    createImport(exportfile, newscadfile)

create_png_cmd = [args.openscad, newscadfile, '-o', pngfile] + remaining_args
print('Running OpenSCAD #2:', file=sys.stderr)
print(' '.join(create_png_cmd), file=sys.stderr)
fontdir = os.path.abspath(os.path.join(os.path.dirname(__file__), "data/ttf"))
fontenv = os.environ.copy()
fontenv["OPENSCAD_FONT_PATH"] = fontdir
sys.stderr.flush()
result = subprocess.call(create_png_cmd, env = fontenv)
stat = os.stat(pngfile)
print(stat, file=sys.stderr)
if result != 0:
    failquit('OpenSCAD #2 failed with return code ' + str(result))

try:    os.remove(exportfile)
except: failquit('failure at os.remove('+exportfile+')')
if newscadfile != exportfile:
    try: os.remove(newscadfile)
    except: failquit('failure at os.remove('+newscadfile+')')
