171 lines
6.7 KiB
Python
171 lines
6.7 KiB
Python
#***************************************************************************
|
|
#* (c) Yorik van Havre (yorik@uncreated.net) 2014 *
|
|
#* *
|
|
#* This file is part of the FreeCAD CAx development system. *
|
|
#* *
|
|
#* 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. *
|
|
#* *
|
|
#* FreeCAD 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 Lesser General Public License for more details. *
|
|
#* *
|
|
#* You should have received a copy of the GNU Library General Public *
|
|
#* License along with FreeCAD; if not, write to the Free Software *
|
|
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
#* USA *
|
|
#* *
|
|
#***************************************************************************/
|
|
|
|
|
|
'''
|
|
These are a common functions and classes for creating custom post processors.
|
|
'''
|
|
|
|
from PySide import QtCore, QtGui
|
|
import FreeCAD
|
|
|
|
FreeCADGui = None
|
|
if FreeCAD.GuiUp:
|
|
import FreeCADGui
|
|
|
|
|
|
class GCodeHighlighter(QtGui.QSyntaxHighlighter):
|
|
def __init__(self, parent=None):
|
|
super(GCodeHighlighter, self).__init__(parent)
|
|
|
|
keywordFormat = QtGui.QTextCharFormat()
|
|
keywordFormat.setForeground(QtCore.Qt.cyan)
|
|
keywordFormat.setFontWeight(QtGui.QFont.Bold)
|
|
keywordPatterns = ["\\bG[0-9]+\\b",
|
|
"\\bM[0-9]+\\b"]
|
|
|
|
self.highlightingRules = [(QtCore.QRegExp(pattern), keywordFormat) for pattern in keywordPatterns]
|
|
|
|
speedFormat = QtGui.QTextCharFormat()
|
|
speedFormat.setFontWeight(QtGui.QFont.Bold)
|
|
speedFormat.setForeground(QtCore.Qt.green)
|
|
self.highlightingRules.append((QtCore.QRegExp("\\bF[0-9\\.]+\\b"),speedFormat))
|
|
|
|
def highlightBlock(self, text):
|
|
for pattern, hlFormat in self.highlightingRules:
|
|
expression = QtCore.QRegExp(pattern)
|
|
index = expression.indexIn(text)
|
|
while index >= 0:
|
|
length = expression.matchedLength()
|
|
self.setFormat(index, length, hlFormat)
|
|
index = expression.indexIn(text, index + length)
|
|
|
|
|
|
|
|
class GCodeEditorDialog(QtGui.QDialog):
|
|
def __init__(self, parent = None):
|
|
if parent is None:
|
|
parent = FreeCADGui.getMainWindow()
|
|
QtGui.QDialog.__init__(self,parent)
|
|
|
|
layout = QtGui.QVBoxLayout(self)
|
|
|
|
# nice text editor widget for editing the gcode
|
|
self.editor = QtGui.QTextEdit()
|
|
font = QtGui.QFont()
|
|
font.setFamily("Courier")
|
|
font.setFixedPitch(True)
|
|
font.setPointSize(10)
|
|
self.editor.setFont(font)
|
|
self.editor.setText("G01 X55 Y4.5 F300.0")
|
|
layout.addWidget(self.editor)
|
|
|
|
# OK and Cancel buttons
|
|
self.buttons = QtGui.QDialogButtonBox(
|
|
QtGui.QDialogButtonBox.Ok | QtGui.QDialogButtonBox.Cancel,
|
|
QtCore.Qt.Horizontal, self)
|
|
layout.addWidget(self.buttons)
|
|
|
|
# restore placement and size
|
|
self.paramKey = "User parameter:BaseApp/Values/Mod/Path/GCodeEditor/"
|
|
params = FreeCAD.ParamGet(self.paramKey)
|
|
posX = params.GetInt("posX")
|
|
posY = params.GetInt("posY")
|
|
if posX > 0 and posY > 0:
|
|
self.move(posX, posY)
|
|
width = params.GetInt("width")
|
|
height = params.GetInt("height")
|
|
if width > 0 and height > 0:
|
|
self.resize(width, height)
|
|
|
|
self.buttons.accepted.connect(self.accept)
|
|
self.buttons.rejected.connect(self.reject)
|
|
|
|
def done(self, *args, **kwargs):
|
|
params = FreeCAD.ParamGet(self.paramKey)
|
|
params.SetInt("posX", self.x())
|
|
params.SetInt("posY", self.y())
|
|
params.SetInt("width", self.size().width())
|
|
params.SetInt("height", self.size().height())
|
|
return QtGui.QDialog.done(self, *args, **kwargs)
|
|
|
|
|
|
def stringsplit(commandline):
|
|
returndict = {'command':None, 'X':None, 'Y':None, 'Z':None, 'A':None, 'B':None, 'F':None, 'T':None, 'S':None, 'I':None, 'J':None,'K':None, 'txt': None}
|
|
wordlist = [a.strip() for a in commandline.split(" ")]
|
|
if wordlist[0][0] == '(':
|
|
returndict['command'] = 'message'
|
|
returndict['txt'] = wordlist[0]
|
|
else:
|
|
returndict['command'] = wordlist[0]
|
|
for word in wordlist[1:]:
|
|
returndict[word[0]] = word[1:]
|
|
|
|
return returndict
|
|
|
|
def fmt(num,dec,units):
|
|
''' used to format axis moves, feedrate, etc for decimal places and units'''
|
|
if units == 'G21': #metric
|
|
fnum = '%.*f' % (dec, num)
|
|
else: #inch
|
|
fnum = '%.*f' % (dec, num/25.4) #since FreeCAD uses metric units internally
|
|
return fnum
|
|
|
|
def editor(gcode):
|
|
'''pops up a handy little editor to look at the code output '''
|
|
|
|
prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Path")
|
|
# default Max Highlighter Size = 512 Ko
|
|
defaultMHS = 512 * 1024
|
|
mhs = prefs.GetUnsigned('inspecteditorMaxHighlighterSize', defaultMHS)
|
|
|
|
dia = GCodeEditorDialog()
|
|
dia.editor.setText(gcode)
|
|
gcodeSize = len(dia.editor.toPlainText())
|
|
if (gcodeSize <= mhs):
|
|
# because of poor performance, syntax highlighting is
|
|
# limited to mhs octets (default 512 KB).
|
|
# It seems than the response time curve has an inflexion near 500 KB
|
|
# beyond 500 KB, the response time increases exponentially.
|
|
dia.highlighter = GCodeHighlighter(dia.editor.document())
|
|
else:
|
|
FreeCAD.Console.PrintMessage(translate("Path", "GCode size too big ({} o), disabling syntax highlighter.".format(gcodeSize)))
|
|
result = dia.exec_()
|
|
if result: # If user selected 'OK' get modified G Code
|
|
final = dia.editor.toPlainText()
|
|
else:
|
|
final = gcode
|
|
return final
|
|
|
|
def fcoms(string,commentsym):
|
|
''' filter and rebuild comments with user preferred comment symbol'''
|
|
if len(commentsym)==1:
|
|
s1 = string.replace('(', commentsym)
|
|
comment = s1.replace(')', '')
|
|
else:
|
|
return string
|
|
return comment
|
|
|
|
|
|
|