exportProject.FCMacro

# Copyright 2016, Jon Nordby <jononor@gmail.com>. Licensed LGPLv2+ & MIT
# 2020, Benjamin Richter <brichter@futuraallmende.com>

"""
exportProject:

Exports all files in document which has names like "mypart [dxf]" or "otherpart[stl,step]"
to individual files in a export directory. Default prefix: 'export/$project-'
Can be ran both as a macro from the FreeCAD UI and as a script on commandline.
"""

# TODO: support some kind of preview. Maybe using the recent camera macros?

import sys, os
import re
import time

macropath = "/home/user/.FreeCAD/Macro/FreeCAD-macros"

partTypeRegex = re.compile(r'^(.*)\[([,\w]+)\].*$')

sys.path.insert(0, '/usr/lib/freecad/lib/')
sys.path.insert(0, '/usr/lib/freecad/Mod/OpenSCAD/')
import FreeCAD
import exportCSG


def exportViewToPNG(objs, p):
    # Exports the current view to a PNG suitable for printing on A4 or Letter.
    # Reference: http://www.freecadweb.org/wiki/index.php?title=Std_ViewScreenShot

    # Create an image which fits on an A4 or Letter page at 300dpi, landscape
    width = 11 * 300    # inside letter
    height = int(8.26 * 300) # inside A4

    doc = FreeCAD.ActiveDocument

    if not doc.FileName:
        FreeCAD.Console.PrintError('Must save project first\n')
        return

    docDir, docFilename = os.path.split(doc.FileName)

    filePrefix = os.path.splitext(docFilename)[0]

    for o in doc.RootObjects:                               # hide all root objects
        FreeCADGui.ActiveDocument.getObject(o.Name).Visibility = False

    for o in objs:
        print(o.Name)
        FreeCADGui.ActiveDocument.getObject(o.Name).Visibility = True

        FreeCADGui.ActiveDocument.ActiveView.fitAll()
        Gui.updateGui()
        time.sleep(0.5)
        FreeCAD.Console.PrintMessage("Saving view to %s\n" % p)
        FreeCADGui.ActiveDocument.ActiveView.saveImage(p.encode("utf-8"), width, height, 'White')
        FreeCADGui.ActiveDocument.getObject(o.Name).Visibility = False

def exportMesh(objs, p):
    import Mesh
    Mesh.export(objs, p)
def exportDxf(objs, p):
    import importDXF
    importDXF.export(objs, p)
def exportStep(objs, p):
    # XXX: will overwrite file if called with multiple args.. Last write wins
    for o in objs:
        shape = o.Shape
        shape.exportStep(p)
def exportGif(objs, p):
    # Dummy for gif export, exports visible part to generic file
    doc = FreeCAD.ActiveDocument

    if not doc.FileName:
        FreeCAD.Console.PrintError('Must save project first\n')
        return
    for o in doc.RootObjects:                               # hide all root objects
        FreeCADGui.ActiveDocument.getObject(o.Name).Visibility = False

    for o in objs:
        print(o.Name)
        FreeCADGui.ActiveDocument.getObject(o.Name).Visibility = True

        FreeCADGui.ActiveDocument.ActiveView.fitAll()
        Gui.updateGui()
        time.sleep(0.5)
        FreeCAD.Console.PrintMessage("Saving gif")
        exec(open(macropath + "/exportGif.FCMacro").read())
        FreeCADGui.ActiveDocument.getObject(o.Name).Visibility = False

# TODO: support CAM and post-processors like: rml, spb
supportedTypes = {
    'stl': exportMesh,
    'obj': exportMesh,
    'step': exportStep,
    'dxf': exportDxf,
    'scad': exportCSG.export,
    'png': exportViewToPNG,
    'gif': exportGif,
}

def extractTypes(label):
    match = re.search(partTypeRegex, label)
    if match:
        types = match.group(2).split(',')
        part = match.group(1)
    else:
        part = label
        types = []
    return [part, types]


def exportDocument(doc, exportdir):
    exported = []
#    if not os.path.exists(exportdir):
#        os.makedirs(exportdir)

    docname = os.path.splitext(os.path.basename(doc.FileName))[0]
    for obj in doc.Objects:
        partname, types = extractTypes(obj.Label)
        #print "obj %s %s" % (obj.Label, types)
        for t in types:
          if not os.path.exists(t):
              os.makedirs(t)
          exporter = supportedTypes[t]
          filename = "%s-%s.%s" % (docname, partname, t)
          filepath = os.path.join(t, filename)
          exporter([obj], filepath)
          exported.append(filename)

    return exported


def fileListing(files):
    return '\n'.join("\t"+f for f in files)

def scriptmain():
    projectfile = sys.argv[1]
    doc = FreeCAD.openDocument(projectfile)

    if len(sys.argv) >= 3:
        exportdir = sys.argv[2]
    else:
        docdir = os.path.dirname(doc.FileName)
        exportdir = os.path.join(docdir, 'export/')

    print('Opening project', projectfile)
    files = exportDocument(doc, exportdir)
    print("Exported ", len(files), " files to '", exportdir, "'\n",fileListing(files))

def showText(title, text):
    from PySide import QtGui
    QtGui.QMessageBox.information(None, title, text)

# TODO: allow specifying export directory in UI
def macromain():
    doc = FreeCAD.ActiveDocument
    docdir = os.path.dirname(doc.FileName)
    exportdir = os.path.join(docdir, 'export/')

    files = exportDocument(doc, exportdir)
    #showText('Project export', "Exported %d files to %s\n%s" % ( len(files), exportdir, fileListing(files)))
    print(files)

if __name__ == '__main__':
    if FreeCAD.GuiUp:
        macromain()
    else:
        sys.exit(scriptmain())