271 lines
11 KiB
Python
271 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2015 Dan Falck <ddfalck@gmail.com> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program 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 Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
''' Post Process command that will make use of the Output File and Post Processor entries in PathJob '''
|
|
|
|
from __future__ import print_function
|
|
|
|
import FreeCAD
|
|
import FreeCADGui
|
|
import os
|
|
import PathScripts.PathLog as PathLog
|
|
|
|
from PathScripts import PathJob
|
|
from PathScripts import PathLoadTool
|
|
from PathScripts import PathUtils
|
|
from PathScripts.PathPostProcessor import PostProcessor
|
|
from PathScripts.PathPreferences import PathPreferences
|
|
from PySide import QtCore, QtGui
|
|
|
|
|
|
LOG_MODULE = PathLog.thisModule()
|
|
|
|
PathLog.setLevel(PathLog.Level.DEBUG, LOG_MODULE)
|
|
PathLog.trackModule(LOG_MODULE)
|
|
|
|
# Qt tanslation handling
|
|
def translate(text, context = "Path_Post", disambig=None):
|
|
return QtCore.QCoreApplication.translate(context, text, disambig)
|
|
|
|
class DlgSelectPostProcessor:
|
|
|
|
def __init__(self, parent=None):
|
|
self.dialog = FreeCADGui.PySideUic.loadUi(":/panels/DlgSelectPostProcessor.ui")
|
|
firstItem = None
|
|
for post in PathPreferences.allEnabledPostProcessors():
|
|
item = QtGui.QListWidgetItem(post)
|
|
item.setFlags(QtCore.Qt.ItemFlag.ItemIsSelectable | QtCore.Qt.ItemFlag.ItemIsEnabled)
|
|
self.dialog.lwPostProcessor.addItem(item)
|
|
if not firstItem:
|
|
firstItem = item
|
|
if firstItem:
|
|
self.dialog.lwPostProcessor.setCurrentItem(firstItem)
|
|
else:
|
|
self.dialog.buttonBox.button(QtGui.QDialogButtonBox.Ok).setEnabled(False)
|
|
self.tooltips = {}
|
|
self.dialog.lwPostProcessor.itemDoubleClicked.connect(self.dialog.accept)
|
|
self.dialog.lwPostProcessor.setMouseTracking(True)
|
|
self.dialog.lwPostProcessor.itemEntered.connect(self.updateTooltip)
|
|
|
|
def updateTooltip(self, item):
|
|
if item.text() in self.tooltips.keys():
|
|
tooltip = self.tooltips[item.text()]
|
|
else:
|
|
processor = PostProcessor.load(item.text())
|
|
self.tooltips[item.text()] = processor.tooltip
|
|
tooltip = processor.tooltip
|
|
self.dialog.lwPostProcessor.setToolTip(tooltip)
|
|
|
|
def exec_(self):
|
|
if self.dialog.exec_() == 1:
|
|
posts = self.dialog.lwPostProcessor.selectedItems()
|
|
return posts[0].text()
|
|
return None
|
|
|
|
class CommandPathPost:
|
|
|
|
def resolveFileName(self, job):
|
|
path = PathPreferences.defaultOutputFile()
|
|
if job.PostProcessorOutputFile:
|
|
path = job.PostProcessorOutputFile
|
|
filename = path
|
|
if '%D' in filename:
|
|
D = FreeCAD.ActiveDocument.FileName
|
|
if D:
|
|
D = os.path.dirname(D)
|
|
# in case the document is in the current working directory
|
|
if not D:
|
|
D = '.'
|
|
else:
|
|
FreeCAD.Console.PrintError("Please save document in order to resolve output path!\n")
|
|
return None
|
|
filename = filename.replace('%D', D)
|
|
|
|
if '%d' in filename:
|
|
d = FreeCAD.ActiveDocument.Label
|
|
filename = filename.replace('%d', d)
|
|
|
|
if '%j' in filename:
|
|
j = job.Label
|
|
filename = filename.replace('%j', j)
|
|
|
|
if '%M' in filename:
|
|
pref = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Macro")
|
|
M = pref.GetString("MacroPath", FreeCAD.getUserAppDataDir())
|
|
filename = filename.replace('%M', M)
|
|
|
|
policy = PathPreferences.defaultOutputPolicy()
|
|
|
|
openDialog = policy == 'Open File Dialog'
|
|
if os.path.isdir(filename) or not os.path.isdir(os.path.dirname(filename)):
|
|
# Either the entire filename resolves into a directory or the parent directory doesn't exist.
|
|
# Either way I don't know what to do - ask for help
|
|
openDialog = True
|
|
|
|
if os.path.isfile(filename) and not openDialog:
|
|
if policy == 'Open File Dialog on conflict':
|
|
openDialog = True
|
|
elif policy == 'Append Unique ID on conflict':
|
|
fn, ext = os.path.splitext(filename)
|
|
nr = fn[-3:]
|
|
n = 1
|
|
if nr.isdigit():
|
|
n = int(nr)
|
|
while os.path.isfile("%s%03d%s" % (fn, n, ext)):
|
|
n = n + 1
|
|
filename = "%s%03d%s" % (fn, n, ext)
|
|
|
|
if openDialog:
|
|
foo = QtGui.QFileDialog.getSaveFileName(QtGui.qApp.activeWindow(), "Output File", filename)
|
|
if foo:
|
|
filename = foo[0]
|
|
else:
|
|
filename = None
|
|
|
|
return filename
|
|
|
|
def resolvePostProcessor(self, job):
|
|
if hasattr(job, "PostProcessor"):
|
|
post = PathPreferences.defaultPostProcessor()
|
|
if job.PostProcessor:
|
|
post = job.PostProcessor
|
|
if post and PostProcessor.exists(post):
|
|
return post
|
|
dlg = DlgSelectPostProcessor()
|
|
return dlg.exec_()
|
|
|
|
|
|
def GetResources(self):
|
|
return {'Pixmap': 'Path-Post',
|
|
'MenuText': QtCore.QT_TRANSLATE_NOOP("Path_Post", "Post Process"),
|
|
'Accel': "P, P",
|
|
'ToolTip': QtCore.QT_TRANSLATE_NOOP("Path_Post", "Post Process the selected Job")}
|
|
|
|
def IsActive(self):
|
|
if FreeCAD.ActiveDocument is not None:
|
|
if FreeCADGui.Selection.getCompleteSelection():
|
|
for o in FreeCAD.ActiveDocument.Objects:
|
|
if o.Name[:3] == "Job":
|
|
return True
|
|
|
|
return False
|
|
|
|
def exportObjectsWith(self, objs, job, needFilename = True):
|
|
PathLog.track()
|
|
# check if the user has a project and has set the default post and
|
|
# output filename
|
|
postArgs = PathPreferences.defaultPostProcessorArgs()
|
|
if hasattr(job, "PostProcessorArgs") and job.PostProcessorArgs:
|
|
postArgs = job.PostProcessorArgs
|
|
elif hasattr(job, "PostProcessor") and job.PostProcessor:
|
|
postArgs = ''
|
|
|
|
postname = self.resolvePostProcessor(job)
|
|
filename = '-'
|
|
if postname and needFilename:
|
|
filename = self.resolveFileName(job)
|
|
|
|
if postname and filename:
|
|
print("post: %s(%s, %s)" % (postname, filename, postArgs))
|
|
processor = PostProcessor.load(postname)
|
|
gcode = processor.export(objs, filename, postArgs)
|
|
return (False, gcode)
|
|
else:
|
|
return (True, '')
|
|
|
|
def Activated(self):
|
|
PathLog.track()
|
|
FreeCAD.ActiveDocument.openTransaction(
|
|
translate("Post Process the Selected path(s)"))
|
|
FreeCADGui.addModule("PathScripts.PathPost")
|
|
|
|
# Attempt to figure out what the user wants to post-process
|
|
# If a job is selected, post that.
|
|
# If there's only one job in a document, post it.
|
|
# If a user has selected a subobject of a job, post the job.
|
|
# If multiple jobs and can't guess, ask them.
|
|
|
|
selected = FreeCADGui.Selection.getSelectionEx()
|
|
if len(selected) > 1:
|
|
FreeCAD.Console.PrintError("Please select a single job or other path object\n")
|
|
return
|
|
elif len(selected) == 1:
|
|
sel = selected[0].Object
|
|
if sel.Name[:3] == "Job":
|
|
job = sel
|
|
elif hasattr(sel, "Path"):
|
|
try:
|
|
job = PathUtils.findParentJob(sel)
|
|
except:
|
|
job = None
|
|
else:
|
|
job = None
|
|
if job is None:
|
|
targetlist = []
|
|
for o in FreeCAD.ActiveDocument.Objects:
|
|
if hasattr(o, "Proxy"):
|
|
if isinstance(o.Proxy, PathJob.ObjectPathJob):
|
|
targetlist.append(o.Label)
|
|
PathLog.debug("Possible post objects: {}".format(targetlist))
|
|
if len(targetlist) > 1:
|
|
form = FreeCADGui.PySideUic.loadUi(":/panels/DlgJobChooser.ui")
|
|
form.cboProject.addItems(targetlist)
|
|
r = form.exec_()
|
|
if r is False:
|
|
return
|
|
else:
|
|
jobname = form.cboProject.currentText()
|
|
else:
|
|
jobname = targetlist[0]
|
|
job = FreeCAD.ActiveDocument.getObject(jobname)
|
|
|
|
PathLog.debug("about to postprocess job: {}".format(job.Name))
|
|
|
|
# Build up an ordered list of operations and tool changes.
|
|
# Then post-the ordered list
|
|
postlist = []
|
|
currTool = None
|
|
for obj in job.Group:
|
|
PathLog.debug("obj: {}".format(obj.Name))
|
|
if not isinstance(obj.Proxy, PathLoadTool.LoadTool):
|
|
if obj.ToolController.ToolNumber != currTool:
|
|
postlist.append(obj.ToolController)
|
|
postlist.append(obj)
|
|
|
|
fail = True
|
|
rc = ''
|
|
(fail, rc) = self.exportObjectsWith(postlist, job)
|
|
|
|
if fail:
|
|
FreeCAD.ActiveDocument.abortTransaction()
|
|
else:
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
|
|
if FreeCAD.GuiUp:
|
|
# register the FreeCAD command
|
|
FreeCADGui.addCommand('Path_Post', CommandPathPost())
|
|
|
|
FreeCAD.Console.PrintLog("Loading PathPost... done\n")
|