FreeCAD / src /Mod /CAM /Path /Main /Gui /Inspect.py

Upload folder using huggingface_hub

985c397 verified 10 months ago

12 kB

# SPDX-License-Identifier: LGPL-2.1-or-later

# ***************************************************************************
# *   Copyright (c) 2015 Yorik van Havre <yorik@uncreated.net>              *
# *                                                                         *
# *   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                                                                   *
# *                                                                         *
# ***************************************************************************


from PySide import QtCore, QtGui
import FreeCAD
import FreeCADGui
import Path
from PySide.QtCore import QT_TRANSLATE_NOOP
import PathScripts.PathUtils as PathUtils
from Path.Main.Gui.Editor import CodeEditor

translate = FreeCAD.Qt.translate


class GCodeHighlighter(QtGui.QSyntaxHighlighter):
    def __init__(self, parent=None):
        def convertcolor(c):
            return QtGui.QColor(int((c >> 24) & 0xFF), int((c >> 16) & 0xFF), int((c >> 8) & 0xFF))

        super(GCodeHighlighter, self).__init__(parent)
        p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Editor")
        colors = []
        c = p.GetUnsigned("Number")
        if c:
            colors.append(convertcolor(c))
        else:
            colors.append(QtCore.Qt.red)
        c = p.GetUnsigned("Keyword")
        if c:
            colors.append(convertcolor(c))
        else:
            colors.append(QtGui.QColor(0, 170, 0))
        c = p.GetUnsigned("Define name")
        if c:
            colors.append(convertcolor(c))
        else:
            colors.append(QtGui.QColor(160, 160, 164))

        self.highlightingRules = []
        numberFormat = QtGui.QTextCharFormat()
        numberFormat.setForeground(colors[0])
        self.highlightingRules.append((QtCore.QRegularExpression("[\\-0-9\\.]"), numberFormat))
        keywordFormat = QtGui.QTextCharFormat()
        keywordFormat.setForeground(colors[1])
        keywordFormat.setFontWeight(QtGui.QFont.Bold)
        keywordPatterns = ["\\bG[0-9]+\\b", "\\bM[0-9]+\\b"]
        self.highlightingRules.extend(
            [(QtCore.QRegularExpression(pattern), keywordFormat) for pattern in keywordPatterns]
        )
        speedFormat = QtGui.QTextCharFormat()
        speedFormat.setFontWeight(QtGui.QFont.Bold)
        speedFormat.setForeground(colors[2])
        self.highlightingRules.append((QtCore.QRegularExpression("\\bF[0-9\\.]+\\b"), speedFormat))

    def highlightBlock(self, text):

        for pattern, fmt in self.highlightingRules:
            expression = QtCore.QRegularExpression(pattern)
            index = expression.match(text)
            while index.hasMatch():
                length = index.capturedLength()
                self.setFormat(index.capturedStart(), length, fmt)
                index = expression.match(text, index.capturedStart() + length)


class GCodeEditorDialog(QtGui.QDialog):
    tool = None

    def __init__(self, PathObj, parent=FreeCADGui.getMainWindow()):
        self.PathObj = PathObj
        if hasattr(PathObj, "ToolController"):
            self.tool = PathObj.ToolController.Tool
        else:
            self.tool = None

        QtGui.QDialog.__init__(self, parent)
        layout = QtGui.QVBoxLayout(self)

        p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
        c = p.GetUnsigned("DefaultHighlightPathColor", 4286382335)
        Q = QtGui.QColor(int((c >> 24) & 0xFF), int((c >> 16) & 0xFF), int((c >> 8) & 0xFF))
        highlightcolor = (
            Q.red() / 255.0,
            Q.green() / 255.0,
            Q.blue() / 255.0,
            Q.alpha() / 255.0,
        )

        self.selectionobj = FreeCAD.ActiveDocument.addObject("Path::Feature", "selection")
        self.selectionobj.ViewObject.LineWidth = 4
        self.selectionobj.ViewObject.NormalColor = highlightcolor

        # self.editor = QtGui.QTextEdit()  # without lines enumeration
        self.editor = CodeEditor()  # with lines enumeration
        font = QtGui.QFont()
        p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Editor")
        font.setFamily(p.GetString("Font", "Courier"))
        font.setFixedPitch(True)
        font.setPointSize(p.GetInt("FontSize", 10))
        self.editor.setFont(font)
        self.editor.setPlainText("G01 X55 Y4.5 F300.0")
        layout.addWidget(self.editor)

        # Note
        lab = QtGui.QLabel()
        lab.setText(
            translate(
                "CAM_Inspect",
                "<b>Note</b>: This dialog shows path commands in FreeCAD base units (mm/s)."
                "<br>Values will be converted to the desired unit during post processing.",
            )
        )
        lab.setWordWrap(True)
        layout.addWidget(lab)

        # OK and Cancel buttons
        self.buttons = QtGui.QDialogButtonBox(
            QtGui.QDialogButtonBox.Close,
            QtCore.Qt.Horizontal,
            self,
        )

        layout.addWidget(self.buttons)
        self.buttons.rejected.connect(self.reject)
        self.editor.selectionChanged.connect(self.highlightpath)
        self.finished.connect(self.cleanup)

        prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
        Xpos = int(prefs.GetString("inspecteditorX", "0"))
        Ypos = int(prefs.GetString("inspecteditorY", "0"))
        height = int(prefs.GetString("inspecteditorH", "500"))
        width = int(prefs.GetString("inspecteditorW", "600"))
        self.move(Xpos, Ypos)
        self.resize(width, height)

    def cleanup(self):
        prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
        prefs.SetString("inspecteditorX", str(self.x()))
        prefs.SetString("inspecteditorY", str(self.y()))
        prefs.SetString("inspecteditorW", str(self.width()))
        prefs.SetString("inspecteditorH", str(self.height()))
        FreeCAD.ActiveDocument.removeObject(self.selectionobj.Name)

    def highlightpath(self):
        cursor = self.editor.textCursor()
        sp = cursor.selectionStart()
        ep = cursor.selectionEnd()
        cursor.setPosition(sp)
        startrow = cursor.blockNumber()
        cursor.setPosition(ep)
        endrow = cursor.blockNumber()

        commands = PathUtils.getPathWithPlacement(self.PathObj).Commands

        # Derive the starting position for the first selected command
        prevX = prevY = prevZ = None
        prevcommands = commands[:startrow]
        prevcommands.reverse()
        for c in prevcommands:
            if prevX is None:
                if c.Parameters.get("X") is not None:
                    prevX = c.Parameters.get("X")
            if prevY is None:
                if c.Parameters.get("Y") is not None:
                    prevY = c.Parameters.get("Y")
            if prevZ is None:
                if c.Parameters.get("Z") is not None:
                    prevZ = c.Parameters.get("Z")
            if prevX is not None and prevY is not None and prevZ is not None:
                break
        if prevX is None:
            prevX = 0.0
        if prevY is None:
            prevY = 0.0
        if prevZ is None:
            prevZ = 0.0

        # Build a new path with selection
        p = Path.Path()
        firstrapid = Path.Command("G0", {"X": prevX, "Y": prevY, "Z": prevZ})

        selectionpath = [firstrapid] + commands[startrow : endrow + 1]
        p.Commands = selectionpath
        self.selectionobj.Path = p

        if self.tool is not None:
            self.tool.Placement.Base.x = prevX
            self.tool.Placement.Base.y = prevY
            self.tool.Placement.Base.z = prevZ


def show(obj):
    "show(obj): shows the G-code data of the given Path object in a dialog"

    prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
    # default Max Highlighter Size = 256 Ko
    defaultMHS = 256 * 1024
    mhs = prefs.GetUnsigned("inspecteditorMaxHighlighterSize", defaultMHS)

    if hasattr(obj, "Path"):
        if obj.Path:
            dia = GCodeEditorDialog(obj)
            dia.editor.setPlainText(obj.Path.toGCode())
            gcodeSize = len(dia.editor.toPlainText())
            if gcodeSize <= mhs:
                # because of poor performance, syntax highlighting is
                # limited to mhs octets (default 256 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_()
            # exec_() returns 0 or 1 depending on the button pressed (Ok or Cancel)
            if result:
                p = Path.Path(dia.editor.toPlainText())
                FreeCAD.ActiveDocument.openTransaction("Edit Path")
                obj.Path = p
                FreeCAD.ActiveDocument.commitTransaction()
                FreeCAD.ActiveDocument.recompute()


class CommandPathInspect:
    def GetResources(self):
        return {
            "Pixmap": "CAM_Inspect",
            "MenuText": QT_TRANSLATE_NOOP("CAM_Inspect", "Inspect Toolpath"),
            "Accel": "P, I",
            "ToolTip": QT_TRANSLATE_NOOP(
                "CAM_Inspect", "Inspects the contents of a toolpath object"
            ),
        }

    def IsActive(self):
        selection = FreeCADGui.Selection.getSelection()
        if len(selection) == 0:
            return False
        obj = selection[0]
        return hasattr(obj, "Path") and len(obj.Path.Commands) > 0

    def Activated(self):
        # check that the selection contains exactly what we want
        selection = FreeCADGui.Selection.getSelection()
        if len(selection) != 1:
            FreeCAD.Console.PrintError(
                translate("CAM_Inspect", "Select exactly one path object") + "\n"
            )
            return
        if not (selection[0].isDerivedFrom("Path::Feature")):
            FreeCAD.Console.PrintError(
                translate("CAM_Inspect", "Select exactly one path object") + "\n"
            )
            return

        # if everything is ok, execute
        FreeCADGui.addModule("Path.Main.Gui.Inspect")
        FreeCADGui.doCommand(
            "Path.Main.Gui.Inspect.show(FreeCAD.ActiveDocument." + selection[0].Name + ")"
        )


if FreeCAD.GuiUp:
    # register the FreeCAD command
    FreeCADGui.addCommand("CAM_Inspect", CommandPathInspect())

	# SPDX-License-Identifier: LGPL-2.1-or-later

	# ***************************************************************************
	# * Copyright (c) 2015 Yorik van Havre <yorik@uncreated.net> *
	# * *
	# * 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 *
	# * *
	# ***************************************************************************


	from PySide import QtCore, QtGui
	import FreeCAD
	import FreeCADGui
	import Path
	from PySide.QtCore import QT_TRANSLATE_NOOP
	import PathScripts.PathUtils as PathUtils
	from Path.Main.Gui.Editor import CodeEditor

	translate = FreeCAD.Qt.translate


	class GCodeHighlighter(QtGui.QSyntaxHighlighter):
	def __init__(self, parent=None):
	def convertcolor(c):
	return QtGui.QColor(int((c >> 24) & 0xFF), int((c >> 16) & 0xFF), int((c >> 8) & 0xFF))

	super(GCodeHighlighter, self).__init__(parent)
	p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Editor")
	colors = []
	c = p.GetUnsigned("Number")
	if c:
	colors.append(convertcolor(c))
	else:
	colors.append(QtCore.Qt.red)
	c = p.GetUnsigned("Keyword")
	if c:
	colors.append(convertcolor(c))
	else:
	colors.append(QtGui.QColor(0, 170, 0))
	c = p.GetUnsigned("Define name")
	if c:
	colors.append(convertcolor(c))
	else:
	colors.append(QtGui.QColor(160, 160, 164))

	self.highlightingRules = []
	numberFormat = QtGui.QTextCharFormat()
	numberFormat.setForeground(colors[0])
	self.highlightingRules.append((QtCore.QRegularExpression("[\\-0-9\\.]"), numberFormat))
	keywordFormat = QtGui.QTextCharFormat()
	keywordFormat.setForeground(colors[1])
	keywordFormat.setFontWeight(QtGui.QFont.Bold)
	keywordPatterns = ["\\bG[0-9]+\\b", "\\bM[0-9]+\\b"]
	self.highlightingRules.extend(
	[(QtCore.QRegularExpression(pattern), keywordFormat) for pattern in keywordPatterns]
	)
	speedFormat = QtGui.QTextCharFormat()
	speedFormat.setFontWeight(QtGui.QFont.Bold)
	speedFormat.setForeground(colors[2])
	self.highlightingRules.append((QtCore.QRegularExpression("\\bF[0-9\\.]+\\b"), speedFormat))

	def highlightBlock(self, text):

	for pattern, fmt in self.highlightingRules:
	expression = QtCore.QRegularExpression(pattern)
	index = expression.match(text)
	while index.hasMatch():
	length = index.capturedLength()
	self.setFormat(index.capturedStart(), length, fmt)
	index = expression.match(text, index.capturedStart() + length)


	class GCodeEditorDialog(QtGui.QDialog):
	tool = None

	def __init__(self, PathObj, parent=FreeCADGui.getMainWindow()):
	self.PathObj = PathObj
	if hasattr(PathObj, "ToolController"):
	self.tool = PathObj.ToolController.Tool
	else:
	self.tool = None

	QtGui.QDialog.__init__(self, parent)
	layout = QtGui.QVBoxLayout(self)

	p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
	c = p.GetUnsigned("DefaultHighlightPathColor", 4286382335)
	Q = QtGui.QColor(int((c >> 24) & 0xFF), int((c >> 16) & 0xFF), int((c >> 8) & 0xFF))
	highlightcolor = (
	Q.red() / 255.0,
	Q.green() / 255.0,
	Q.blue() / 255.0,
	Q.alpha() / 255.0,
	)

	self.selectionobj = FreeCAD.ActiveDocument.addObject("Path::Feature", "selection")
	self.selectionobj.ViewObject.LineWidth = 4
	self.selectionobj.ViewObject.NormalColor = highlightcolor

	# self.editor = QtGui.QTextEdit() # without lines enumeration
	self.editor = CodeEditor() # with lines enumeration
	font = QtGui.QFont()
	p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Editor")
	font.setFamily(p.GetString("Font", "Courier"))
	font.setFixedPitch(True)
	font.setPointSize(p.GetInt("FontSize", 10))
	self.editor.setFont(font)
	self.editor.setPlainText("G01 X55 Y4.5 F300.0")
	layout.addWidget(self.editor)

	# Note
	lab = QtGui.QLabel()
	lab.setText(
	translate(
	"CAM_Inspect",
	"<b>Note</b>: This dialog shows path commands in FreeCAD base units (mm/s)."
	"<br>Values will be converted to the desired unit during post processing.",
	)
	)
	lab.setWordWrap(True)
	layout.addWidget(lab)

	# OK and Cancel buttons
	self.buttons = QtGui.QDialogButtonBox(
	QtGui.QDialogButtonBox.Close,
	QtCore.Qt.Horizontal,
	self,
	)

	layout.addWidget(self.buttons)
	self.buttons.rejected.connect(self.reject)
	self.editor.selectionChanged.connect(self.highlightpath)
	self.finished.connect(self.cleanup)

	prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
	Xpos = int(prefs.GetString("inspecteditorX", "0"))
	Ypos = int(prefs.GetString("inspecteditorY", "0"))
	height = int(prefs.GetString("inspecteditorH", "500"))
	width = int(prefs.GetString("inspecteditorW", "600"))
	self.move(Xpos, Ypos)
	self.resize(width, height)

	def cleanup(self):
	prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
	prefs.SetString("inspecteditorX", str(self.x()))
	prefs.SetString("inspecteditorY", str(self.y()))
	prefs.SetString("inspecteditorW", str(self.width()))
	prefs.SetString("inspecteditorH", str(self.height()))
	FreeCAD.ActiveDocument.removeObject(self.selectionobj.Name)

	def highlightpath(self):
	cursor = self.editor.textCursor()
	sp = cursor.selectionStart()
	ep = cursor.selectionEnd()
	cursor.setPosition(sp)
	startrow = cursor.blockNumber()
	cursor.setPosition(ep)
	endrow = cursor.blockNumber()

	commands = PathUtils.getPathWithPlacement(self.PathObj).Commands

	# Derive the starting position for the first selected command
	prevX = prevY = prevZ = None
	prevcommands = commands[:startrow]
	prevcommands.reverse()
	for c in prevcommands:
	if prevX is None:
	if c.Parameters.get("X") is not None:
	prevX = c.Parameters.get("X")
	if prevY is None:
	if c.Parameters.get("Y") is not None:
	prevY = c.Parameters.get("Y")
	if prevZ is None:
	if c.Parameters.get("Z") is not None:
	prevZ = c.Parameters.get("Z")
	if prevX is not None and prevY is not None and prevZ is not None:
	break
	if prevX is None:
	prevX = 0.0
	if prevY is None:
	prevY = 0.0
	if prevZ is None:
	prevZ = 0.0

	# Build a new path with selection
	p = Path.Path()
	firstrapid = Path.Command("G0", {"X": prevX, "Y": prevY, "Z": prevZ})

	selectionpath = [firstrapid] + commands[startrow : endrow + 1]
	p.Commands = selectionpath
	self.selectionobj.Path = p

	if self.tool is not None:
	self.tool.Placement.Base.x = prevX
	self.tool.Placement.Base.y = prevY
	self.tool.Placement.Base.z = prevZ


	def show(obj):
	"show(obj): shows the G-code data of the given Path object in a dialog"

	prefs = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/CAM")
	# default Max Highlighter Size = 256 Ko
	defaultMHS = 256 * 1024
	mhs = prefs.GetUnsigned("inspecteditorMaxHighlighterSize", defaultMHS)

	if hasattr(obj, "Path"):
	if obj.Path:
	dia = GCodeEditorDialog(obj)
	dia.editor.setPlainText(obj.Path.toGCode())
	gcodeSize = len(dia.editor.toPlainText())
	if gcodeSize <= mhs:
	# because of poor performance, syntax highlighting is
	# limited to mhs octets (default 256 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_()
	# exec_() returns 0 or 1 depending on the button pressed (Ok or Cancel)
	if result:
	p = Path.Path(dia.editor.toPlainText())
	FreeCAD.ActiveDocument.openTransaction("Edit Path")
	obj.Path = p
	FreeCAD.ActiveDocument.commitTransaction()
	FreeCAD.ActiveDocument.recompute()


	class CommandPathInspect:
	def GetResources(self):
	return {
	"Pixmap": "CAM_Inspect",
	"MenuText": QT_TRANSLATE_NOOP("CAM_Inspect", "Inspect Toolpath"),
	"Accel": "P, I",
	"ToolTip": QT_TRANSLATE_NOOP(
	"CAM_Inspect", "Inspects the contents of a toolpath object"
	),
	}

	def IsActive(self):
	selection = FreeCADGui.Selection.getSelection()
	if len(selection) == 0:
	return False
	obj = selection[0]
	return hasattr(obj, "Path") and len(obj.Path.Commands) > 0

	def Activated(self):
	# check that the selection contains exactly what we want
	selection = FreeCADGui.Selection.getSelection()
	if len(selection) != 1:
	FreeCAD.Console.PrintError(
	translate("CAM_Inspect", "Select exactly one path object") + "\n"
	)
	return
	if not (selection[0].isDerivedFrom("Path::Feature")):
	FreeCAD.Console.PrintError(
	translate("CAM_Inspect", "Select exactly one path object") + "\n"
	)
	return

	# if everything is ok, execute
	FreeCADGui.addModule("Path.Main.Gui.Inspect")
	FreeCADGui.doCommand(
	"Path.Main.Gui.Inspect.show(FreeCAD.ActiveDocument." + selection[0].Name + ")"
	)


	if FreeCAD.GuiUp:
	# register the FreeCAD command
	FreeCADGui.addCommand("CAM_Inspect", CommandPathInspect())