# -*- coding: utf8 -*- #*************************************************************************** #* Copyright (c) 2009 Yorik van Havre * #* * #* 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 * #* * #*************************************************************************** __title__ = "FreeCAD Draft Workbench - GUI part" __author__ = "Yorik van Havre " __url__ = "https://www.freecadweb.org" ## @package DraftGui # \ingroup DRAFT # \brief GUI elements and utilities of the Draft workbench # # This module provides GUI tools for the Draft workbench, such as # toolbars and task panels, and Qt-dependent utilities such as # a delayed (todo) commit system """This is the GUI part of the Draft module. Report to Draft.py for info """ import os import sys import math import PySide.QtCore as QtCore import PySide.QtGui as QtGui import FreeCAD import FreeCADGui import Draft import DraftVecUtils from draftutils.translate import translate from draftutils.utils import utf8_decode # in-command shortcut definitions: Shortcut / Translation / related UI control inCommandShortcuts = { "Relative": [ Draft.getParam("inCommandShortcutRelative", "R"), translate("draft","Relative"), "isRelative" ], "Global": [Draft.getParam("inCommandShortcutGlobal", "G"), translate("draft","Global"), "isGlobal" ], "Continue": [ Draft.getParam("inCommandShortcutContinue", "T"), translate("draft","Continue"), "continueCmd" ], "Close": [ Draft.getParam("inCommandShortcutClose", "O"), translate("draft","Close"), "closeButton" ], "Copy": [ Draft.getParam("inCommandShortcutCopy", "P"), translate("draft","Copy"), "isCopy" ], "SubelementMode": [ Draft.getParam("inCommandShortcutSubelementMode", "D"), translate("draft","Subelement mode"), "isSubelementMode" ], "Fill": [ Draft.getParam("inCommandShortcutFill", "L"), translate("draft","Fill"), "hasFill" ], "Exit": [ Draft.getParam("inCommandShortcutExit", "A"), translate("draft","Exit"), "finishButton" ], "Snap": [ Draft.getParam("inCommandShortcutSnap", "S"), translate("draft","Snap On/Off"), None ], "Increase": [ Draft.getParam("inCommandShortcutIncreaseRadius", "["), translate("draft","Increase snap radius"), None ], "Decrease": [ Draft.getParam("inCommandShortcutDecreaseRadius", "]"), translate("draft","Decrease snap radius"), None ], "RestrictX": [ Draft.getParam("inCommandShortcutRestrictX", "X"), translate("draft","Restrict X"), None ], "RestrictY": [ Draft.getParam("inCommandShortcutRestrictY", "Y"), translate("draft","Restrict Y"), None ], "RestrictZ": [ Draft.getParam("inCommandShortcutRestrictZ", "Z"), translate("draft","Restrict Z"), None ], "SelectEdge": [ Draft.getParam("inCommandShortcutSelectEdge", "E"), translate("draft","Select edge"), "selectButton" ], "AddHold": [ Draft.getParam("inCommandShortcutAddHold", "Q"), translate("draft","Add custom snap point"), None ], "Length": [ Draft.getParam("inCommandShortcutLength", "H"), translate("draft","Length mode"), "lengthValue" ], "Wipe": [ Draft.getParam("inCommandShortcutWipe", "W"), translate("draft","Wipe"), "wipeButton" ], "SetWP": [ Draft.getParam("inCommandShortcutSetWP", "U"), translate("draft","Set Working Plane"), "orientWPButton" ], "CycleSnap": [ Draft.getParam("inCommandShortcutCycleSnap", "`"), translate("draft","Cycle snap object"), None ], "Undo": [ Draft.getParam("inCommandShortcutUndo", "/"), translate("draft","Undo last segment"), None ], } from draftutils.todo import todo #--------------------------------------------------------------------------- # UNITS handling #--------------------------------------------------------------------------- from draftutils.units import (getDefaultUnit, makeFormatSpec, displayExternal) #--------------------------------------------------------------------------- # Customized widgets #--------------------------------------------------------------------------- class DraftBaseWidget(QtGui.QWidget): def __init__(self,parent = None): super().__init__(parent) def eventFilter(self, widget, event): if (event.type() == QtCore.QEvent.KeyPress and event.text().upper() == inCommandShortcuts["CycleSnap"][0]): if hasattr(FreeCADGui,"Snapper"): FreeCADGui.Snapper.cycleSnapObject() return True return super().eventFilter(widget, event) class DraftDockWidget(DraftBaseWidget): """custom Widget that emits a resized() signal when resized""" def __init__(self,parent = None): super().__init__(parent) def resizeEvent(self,event): self.emit(QtCore.SIGNAL("resized()")) def changeEvent(self, event): if event.type() == QtCore.QEvent.LanguageChange: self.emit(QtCore.SIGNAL("retranslate()")) else: super().changeEvent(event) class DraftLineEdit(QtGui.QLineEdit): """custom QLineEdit widget that has the power to catch Escape keypress""" def __init__(self, parent=None): super().__init__(parent) def keyPressEvent(self, event): if event.key() == QtCore.Qt.Key_Escape: self.emit(QtCore.SIGNAL("escaped()")) elif event.key() == QtCore.Qt.Key_Up: self.emit(QtCore.SIGNAL("up()")) elif event.key() == QtCore.Qt.Key_Down: self.emit(QtCore.SIGNAL("down()")) else: super().keyPressEvent(event) class DraftTaskPanel: def __init__(self,widget,extra=None): if extra: if isinstance(extra,list): self.form = [widget] + extra else: self.form = [widget,extra] else: self.form = widget def getStandardButtons(self): return int(QtGui.QDialogButtonBox.Close) def accept(self): if hasattr(FreeCADGui,"draftToolBar"): return FreeCADGui.draftToolBar.validatePoint() else: if FreeCADGui.ActiveDocument: FreeCADGui.ActiveDocument.resetEdit() return True def reject(self): FreeCADGui.draftToolBar.isTaskOn = False FreeCADGui.draftToolBar.escape() if FreeCADGui.ActiveDocument: FreeCADGui.ActiveDocument.resetEdit() return True def isAllowedAlterDocument(self): return False class DraftToolBar: """The Draft Task panel UI Draft Toolbar is the main ui of the Draft Module. Once displayed as a toolbar, now it define the ui of the Task Panel. Toolbar become obsolete due to lack of manteinence and was disabled by default in February 2020. Draft Ui Commands call and get information such as point coordinates, subcommands activation, continue mode, etc. from Task Panel Ui """ def __init__(self): self.tray = None self.sourceCmd = None self.cancel = None self.pointcallback = None # OBSOLETE BUT STILL USED BY SOME ADDONS AND MACROS self.paramcolor = Draft.getParam("color",255)>>8 self.color = QtGui.QColor(self.paramcolor) self.facecolor = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/View")\ .GetUnsigned("DefaultShapeColor",4294967295)>>8 self.linewidth = Draft.getParam("linewidth",2) self.fontsize = Draft.getParam("textheight",0.20) self.paramconstr = Draft.getParam("constructioncolor",746455039)>>8 self.constrMode = False self.continueMode = False self.relativeMode = True self.globalMode = False self.state = None self.textbuffer = [] self.crossedViews = [] self.isTaskOn = False self.fillmode = Draft.getParam("fillmode", True) self.mask = None self.alock = False self.x = 0 self.y = 0 self.z = 0 self.lvalue = 0 self.pvalue = 90 self.avalue = 0 self.angle = None self.radius = 0 self.offset = 0 self.uiloader = FreeCADGui.UiLoader() self.autogroup = None self.isCenterPlane = False self.input_fields = { "xValue":{"value":"x","unit":"Length"}, "yValue":{"value":"y","unit":"Length"}, "zValue":{"value":"z","unit":"Length"}, "lengthValue":{"value":"lvalue","unit":"Length"}, "radiusValue":{"value":"radius","unit":"Length"}, "angleValue":{"value":"avalue","unit":"Angle"} } # add only a dummy widget, since widgets are created on demand self.baseWidget = DraftBaseWidget() self.tray = QtGui.QToolBar(None) self.tray.setObjectName("Draft tray") self.tray.setWindowTitle("Draft tray") self.toptray = self.tray self.bottomtray = self.tray self.setupTray() self.setupStyle() mw = FreeCADGui.getMainWindow() mw.addToolBar(self.tray) self.tray.setParent(mw) self.tray.hide() #--------------------------------------------------------------------------- # General UI setup #--------------------------------------------------------------------------- def _pushbutton(self,name, layout, hide=True, icon=None, width=None, checkable=False, square=False): p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/General") bsize = p.GetInt("ToolbarIconSize",24)+2 isize = p.GetInt("ToolbarIconSize",24)/3*2 button = QtGui.QPushButton(self.baseWidget) button.setObjectName(name) if square: button.setMaximumSize(QtCore.QSize(button.height(), button.height())) button.setFlat(True) #button.setMaximumSize(QtCore.QSize(width,bsize)) if hide: button.hide() if icon: if icon.endswith(".svg"): button.setIcon(QtGui.QIcon(icon)) else: button.setIcon(QtGui.QIcon.fromTheme( icon, QtGui.QIcon(':/icons/'+icon+'.svg'))) #button.setIconSize(QtCore.QSize(isize, isize)) if checkable: button.setCheckable(True) button.setChecked(False) layout.addWidget(button) return button def _label (self,name, layout, hide=True, wrap=False): label = QtGui.QLabel(self.baseWidget) label.setObjectName(name) if wrap: label.setWordWrap(True) if hide: label.hide() layout.addWidget(label) return label def _lineedit (self,name, layout, hide=True, width=None): p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/General") bsize = p.GetInt("ToolbarIconSize",24)-2 lineedit = DraftLineEdit(self.baseWidget) lineedit.setObjectName(name) if hide: lineedit.hide() #if not width: width = 800 #lineedit.setMaximumSize(QtCore.QSize(width,bsize)) layout.addWidget(lineedit) return lineedit def _inputfield (self,name, layout, hide=True, width=None): inputfield = self.uiloader.createWidget("Gui::InputField") inputfield.setObjectName(name) if hide: inputfield.hide() if not width: sizePolicy = QtGui.QSizePolicy( QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Preferred) inputfield.setSizePolicy(sizePolicy) inputfield.setMinimumWidth(110) else: inputfield.setMaximumWidth(width) layout.addWidget(inputfield) return inputfield def _spinbox (self,name, layout, val=None, vmax=None, hide=True, double=False, size=None): if double: sbox = QtGui.QDoubleSpinBox(self.baseWidget) sbox.setDecimals(FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units")\ .GetInt("Decimals",2)) else: sbox = QtGui.QSpinBox(self.baseWidget) sbox.setObjectName(name) if vmax: sbox.setMaximum(vmax) if val: sbox.setValue(val) #if size: sbox.setMaximumSize(QtCore.QSize(size[0],size[1])) if hide: sbox.hide() layout.addWidget(sbox) return sbox def _checkbox (self,name, layout, checked=True, hide=True): chk = QtGui.QCheckBox(self.baseWidget) chk.setChecked(checked) chk.setObjectName(name) if hide: chk.hide() layout.addWidget(chk) return chk def _combo (self,name,layout,hide=True): cb = QtGui.QComboBox(self.baseWidget) cb.setObjectName(name) if hide: cb.hide() layout.addWidget(cb) def setupToolBar(self,task=False): """sets the draft toolbar up""" # command self.promptlabel = self._label("promptlabel", self.layout, hide=task) self.cmdlabel = self._label("cmdlabel", self.layout, hide=task) boldtxt = QtGui.QFont() boldtxt.setWeight(75) boldtxt.setBold(True) self.cmdlabel.setFont(boldtxt) # point xl = QtGui.QHBoxLayout() yl = QtGui.QHBoxLayout() zl = QtGui.QHBoxLayout() bl = QtGui.QHBoxLayout() self.layout.addLayout(xl) self.layout.addLayout(yl) self.layout.addLayout(zl) self.layout.addLayout(bl) self.labelx = self._label("labelx", xl) self.xValue = self._inputfield("xValue", xl) #width=60 self.xValue.installEventFilter(self.baseWidget) self.xValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.labely = self._label("labely", yl) self.yValue = self._inputfield("yValue", yl) self.yValue.installEventFilter(self.baseWidget) # Required to detect snap cycling in case of Y constraining. self.yValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.labelz = self._label("labelz", zl) self.zValue = self._inputfield("zValue", zl) self.zValue.installEventFilter(self.baseWidget) # Required to detect snap cycling in case of Z constraining. self.zValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.pointButton = self._pushbutton("addButton", bl, icon="Draft_AddPoint") # text self.textValue = QtGui.QTextEdit(self.baseWidget) self.textValue.setObjectName("textValue") self.textValue.setTabChangesFocus(True) self.layout.addWidget(self.textValue) self.textValue.hide() tl = QtGui.QHBoxLayout() self.layout.addLayout(tl) self.textOkButton = self._pushbutton("textButton", tl, icon="button_valid") # additional line controls ll = QtGui.QHBoxLayout() al = QtGui.QHBoxLayout() self.layout.addLayout(ll) self.layout.addLayout(al) self.labellength = self._label("labellength", ll) self.lengthValue = self._inputfield("lengthValue", ll) self.lengthValue.installEventFilter(self.baseWidget) # Required to detect snap cycling if focusOnLength is True. self.lengthValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.labelangle = self._label("labelangle", al) self.angleLock = self._checkbox("angleLock",al,checked=self.alock) self.angleValue = self._inputfield("angleValue", al) self.angleValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Angle).UserString) # shapestring self.labelSSize = self._label("labelSize", self.layout) self.SSizeValue = self._inputfield("SSizeValue", self.layout) #, width=60) self.SSizeValue.setText(FreeCAD.Units.Quantity(1,FreeCAD.Units.Length).UserString) self.labelSTrack = self._label("labelTracking", self.layout) self.STrackValue = self._inputfield("STrackValue", self.layout) #, width=60) self.STrackValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.labelSString = self._label("labelString", self.layout) self.SStringValue = self._lineedit("SStringValue", self.layout) self.SStringValue.setText("") self.labelFFile = self._label("labelFFile", self.layout) self.FFileValue = self._lineedit("FFileValue", self.layout) self.chooserButton = self._pushbutton("chooserButton", self.layout, width=26) self.chooserButton.setText("...") self.SSize = 1 self.STrack = 0 # options fl = QtGui.QHBoxLayout() self.layout.addLayout(fl) self.numFacesLabel = self._label("numfaceslabel", fl) self.numFaces = self._spinbox("numFaces", fl, 3) ol = QtGui.QHBoxLayout() self.layout.addLayout(ol) rl = QtGui.QHBoxLayout() self.layout.addLayout(rl) self.labelRadius = self._label("labelRadius", rl) self.radiusValue = self._inputfield("radiusValue", rl) self.radiusValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.isRelative = self._checkbox("isRelative",self.layout,checked=self.relativeMode) self.isGlobal = self._checkbox("isGlobal",self.layout,checked=self.globalMode) self.hasFill = self._checkbox("hasFill",self.layout,checked=self.fillmode) self.continueCmd = self._checkbox("continueCmd",self.layout,checked=self.continueMode) self.occOffset = self._checkbox("occOffset",self.layout,checked=False) bl = QtGui.QHBoxLayout() self.layout.addLayout(bl) self.undoButton = self._pushbutton("undoButton", bl, icon='Draft_Rotate') bl = QtGui.QHBoxLayout() self.layout.addLayout(bl) self.finishButton = self._pushbutton("finishButton", bl, icon='Draft_Finish') bl = QtGui.QHBoxLayout() self.layout.addLayout(bl) self.closeButton = self._pushbutton("closeButton", bl, icon='Draft_Lock') bl = QtGui.QHBoxLayout() self.layout.addLayout(bl) self.wipeButton = self._pushbutton("wipeButton", bl, icon='Draft_Wipe') bl = QtGui.QHBoxLayout() self.layout.addLayout(bl) self.orientWPButton = self._pushbutton("orientWPButton", bl, icon='Draft_SelectPlane') bl = QtGui.QHBoxLayout() self.layout.addLayout(bl) self.selectButton = self._pushbutton("selectButton", bl, icon='view-select') self.isCopy = self._checkbox("isCopy",self.layout,checked=False) self.isSubelementMode = self._checkbox("isSubelementMode",self.layout,checked=False) # spacer spacerItem = QtGui.QSpacerItem(20, 40, QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Expanding) self.layout.addItem(spacerItem) QtCore.QObject.connect(self.xValue,QtCore.SIGNAL("valueChanged(double)"),self.changeXValue) QtCore.QObject.connect(self.yValue,QtCore.SIGNAL("valueChanged(double)"),self.changeYValue) QtCore.QObject.connect(self.zValue,QtCore.SIGNAL("valueChanged(double)"),self.changeZValue) QtCore.QObject.connect(self.lengthValue,QtCore.SIGNAL("valueChanged(double)"),self.changeLengthValue) QtCore.QObject.connect(self.angleValue,QtCore.SIGNAL("valueChanged(double)"),self.changeAngleValue) QtCore.QObject.connect(self.angleLock,QtCore.SIGNAL("stateChanged(int)"),self.toggleAngle) QtCore.QObject.connect(self.radiusValue,QtCore.SIGNAL("valueChanged(double)"),self.changeRadiusValue) QtCore.QObject.connect(self.xValue,QtCore.SIGNAL("returnPressed()"),self.checkx) QtCore.QObject.connect(self.yValue,QtCore.SIGNAL("returnPressed()"),self.checky) QtCore.QObject.connect(self.lengthValue,QtCore.SIGNAL("returnPressed()"),self.checklength) QtCore.QObject.connect(self.xValue,QtCore.SIGNAL("textEdited(QString)"),self.checkSpecialChars) QtCore.QObject.connect(self.yValue,QtCore.SIGNAL("textEdited(QString)"),self.checkSpecialChars) QtCore.QObject.connect(self.zValue,QtCore.SIGNAL("textEdited(QString)"),self.checkSpecialChars) QtCore.QObject.connect(self.lengthValue,QtCore.SIGNAL("textEdited(QString)"),self.checkSpecialChars) QtCore.QObject.connect(self.radiusValue,QtCore.SIGNAL("textEdited(QString)"),self.checkSpecialChars) QtCore.QObject.connect(self.angleValue,QtCore.SIGNAL("textEdited(QString)"),self.checkSpecialChars) QtCore.QObject.connect(self.zValue,QtCore.SIGNAL("returnPressed()"),self.validatePoint) QtCore.QObject.connect(self.pointButton,QtCore.SIGNAL("clicked()"),self.validatePoint) QtCore.QObject.connect(self.radiusValue,QtCore.SIGNAL("returnPressed()"),self.validatePoint) QtCore.QObject.connect(self.angleValue,QtCore.SIGNAL("returnPressed()"),self.validatePoint) QtCore.QObject.connect(self.textValue,QtCore.SIGNAL("textChanged()"),self.checkEnterText) QtCore.QObject.connect(self.textOkButton,QtCore.SIGNAL("clicked()"),self.sendText) QtCore.QObject.connect(self.zValue,QtCore.SIGNAL("returnPressed()"),self.setFocus) QtCore.QObject.connect(self.finishButton,QtCore.SIGNAL("pressed()"),self.finish) QtCore.QObject.connect(self.closeButton,QtCore.SIGNAL("pressed()"),self.closeLine) QtCore.QObject.connect(self.wipeButton,QtCore.SIGNAL("pressed()"),self.wipeLine) QtCore.QObject.connect(self.orientWPButton,QtCore.SIGNAL("pressed()"),self.orientWP) QtCore.QObject.connect(self.undoButton,QtCore.SIGNAL("pressed()"),self.undoSegment) QtCore.QObject.connect(self.selectButton,QtCore.SIGNAL("pressed()"),self.selectEdge) QtCore.QObject.connect(self.continueCmd,QtCore.SIGNAL("stateChanged(int)"),self.setContinue) QtCore.QObject.connect(self.isCopy,QtCore.SIGNAL("stateChanged(int)"),self.setCopymode) QtCore.QObject.connect(self.isSubelementMode, QtCore.SIGNAL("stateChanged(int)"), self.setSubelementMode) QtCore.QObject.connect(self.isRelative,QtCore.SIGNAL("stateChanged(int)"),self.setRelative) QtCore.QObject.connect(self.isGlobal,QtCore.SIGNAL("stateChanged(int)"),self.setGlobal) QtCore.QObject.connect(self.hasFill,QtCore.SIGNAL("stateChanged(int)"),self.setFill) QtCore.QObject.connect(self.baseWidget,QtCore.SIGNAL("resized()"),self.relocate) QtCore.QObject.connect(self.baseWidget,QtCore.SIGNAL("retranslate()"),self.retranslateUi) QtCore.QObject.connect(self.SSizeValue,QtCore.SIGNAL("valueChanged(double)"),self.changeSSizeValue) QtCore.QObject.connect(self.SSizeValue,QtCore.SIGNAL("returnPressed()"),self.validateSNumeric) QtCore.QObject.connect(self.STrackValue,QtCore.SIGNAL("valueChanged(double)"),self.changeSTrackValue) QtCore.QObject.connect(self.STrackValue,QtCore.SIGNAL("returnPressed()"),self.validateSNumeric) QtCore.QObject.connect(self.SStringValue,QtCore.SIGNAL("returnPressed()"),self.validateSString) QtCore.QObject.connect(self.chooserButton,QtCore.SIGNAL("pressed()"),self.pickFile) QtCore.QObject.connect(self.FFileValue,QtCore.SIGNAL("returnPressed()"),self.validateFile) def setupTray(self): """sets draft tray buttons up""" self.wplabel = self._pushbutton( "wplabel", self.toptray, icon='Draft_SelectPlane', hide=False,width=120) defaultWP = Draft.getParam("defaultWP",0) if defaultWP == 1: self.wplabel.setText(translate("draft","Top")) elif defaultWP == 2: self.wplabel.setText(translate("draft","Front")) elif defaultWP == 3: self.wplabel.setText(translate("draft","Side")) else: self.wplabel.setText(translate("draft","Auto")) p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/General") bsize = p.GetInt("ToolbarIconSize",24)+2 self.styleButton = self._pushbutton( "stylebutton", self.toptray, icon='Draft_Apply', hide=False, width=120) self.setStyleButton() self.constrButton = self._pushbutton( "constrButton", self.toptray, hide=False, icon='Draft_Construction', checkable=True, square=True) self.constrColor = QtGui.QColor(self.paramconstr) self.autoGroupButton = self._pushbutton( "autoGroup", self.bottomtray,icon=":/icons/button_invalid.svg", hide=False, width=120) self.autoGroupButton.setText(translate("draft", "None")) self.autoGroupButton.setFlat(True) QtCore.QObject.connect(self.wplabel,QtCore.SIGNAL("pressed()"),self.selectplane) QtCore.QObject.connect(self.styleButton,QtCore.SIGNAL("pressed()"),self.setstyle) QtCore.QObject.connect(self.constrButton,QtCore.SIGNAL("toggled(bool)"),self.toggleConstrMode) QtCore.QObject.connect(self.autoGroupButton,QtCore.SIGNAL("pressed()"),self.runAutoGroup) QtCore.QTimer.singleShot(2000,self.retranslateTray) # delay so translations get a chance to load def setupStyle(self): style = "#constrButton:Checked {background-color: " style += self.getDefaultColor("constr",rgb=True)+" } " style += "#addButton:Checked, #delButton:checked, " style += "#sharpButton:Checked, " style += "#tangentButton:Checked, #symmetricButton:checked {" style += "background-color: rgb(20,100,250) }" self.baseWidget.setStyleSheet(style) #if hasattr(self,"tray"): # self.tray.setStyleSheet(style) #--------------------------------------------------------------------------- # language tools #--------------------------------------------------------------------------- def retranslateUi(self, widget=None): self.promptlabel.setText(translate("draft", "active command:")) self.cmdlabel.setText(translate("draft", "None")) self.cmdlabel.setToolTip(translate("draft", "Active Draft command")) self.xValue.setToolTip(translate("draft", "X coordinate of next point")) self.labelx.setText(translate("draft", "X")) self.labely.setText(translate("draft", "Y")) self.labelz.setText(translate("draft", "Z")) self.yValue.setToolTip(translate("draft", "Y coordinate of next point")) self.zValue.setToolTip(translate("draft", "Z coordinate of next point")) self.pointButton.setText(translate("draft", "Enter point")) self.pointButton.setToolTip(translate( "draft","Enter a new point with the given coordinates")) self.labellength.setText(translate("draft", "Length")) self.labelangle.setText(translate("draft", "Angle")) self.lengthValue.setToolTip(translate("draft", "Length of current segment")) self.angleValue.setToolTip(translate("draft", "Angle of current segment")) #self.angleLock.setText(translate("draft", "&Lock")) self.angleLock.setToolTip(translate( "draft", "Check this to lock the current angle")\ + " (" + inCommandShortcuts["Length"][0] + ")") self.labelRadius.setText(translate("draft", "Radius")) self.radiusValue.setToolTip(translate("draft", "Radius of Circle")) self.isRelative.setText(translate( "draft", "Relative") + " (" + inCommandShortcuts["Relative"][0] + ")") self.isRelative.setToolTip(translate( "draft", "Coordinates relative to last point or to coordinate system " + "origin\nif is the first point to set")) self.isGlobal.setText(translate( "draft", "Global") + " (" + inCommandShortcuts["Global"][0] + ")") self.isGlobal.setToolTip(translate( "draft", "Coordinates relative to global coordinate system." + "\nUncheck to use working plane coordinate system")) self.hasFill.setText(translate( "draft", "Filled")+" ("+inCommandShortcuts["Fill"][0]+")") self.hasFill.setToolTip(translate( "draft", "Check this if the object should appear as filled, " + "otherwise it will appear as wireframe.\nNot available if " + "Draft preference option 'Use Part Primitives' is enabled")) self.finishButton.setText(translate( "draft", "Finish")+" ("+inCommandShortcuts["Exit"][0]+")") self.finishButton.setToolTip(translate( "draft", "Finishes the current drawing or editing operation")) self.continueCmd.setToolTip(translate( "draft", "If checked, command will not finish until you press " + "the command button again")) self.continueCmd.setText(translate( "draft", "Continue") + " (" + inCommandShortcuts["Continue"][0] + ")") self.occOffset.setToolTip(translate( "draft", "If checked, an OCC-style offset will be performed" + " instead of the classic offset")) self.occOffset.setText(translate("draft", "&OCC-style offset")) self.undoButton.setText(translate("draft", "&Undo")+" ("+inCommandShortcuts["Undo"][0]+")") self.undoButton.setToolTip(translate("draft", "Undo the last segment")) self.closeButton.setText(translate("draft", "Close")+" ("+inCommandShortcuts["Close"][0]+")") self.closeButton.setToolTip(translate("draft", "Finishes and closes the current line")) self.wipeButton.setText(translate("draft", "Wipe")+" ("+inCommandShortcuts["Wipe"][0]+")") self.wipeButton.setToolTip(translate("draft", "Wipes the existing segments of this line and starts again from the last point")) self.orientWPButton.setText(translate("draft", "Set WP")+" ("+inCommandShortcuts["SetWP"][0]+")") self.orientWPButton.setToolTip(translate("draft", "Reorients the working plane on the last segment")) self.selectButton.setText(translate("draft", "Select edge")+" ("+inCommandShortcuts["SelectEdge"][0]+")") self.selectButton.setToolTip(translate("draft", "Selects an existing edge to be measured by this dimension")) self.numFacesLabel.setText(translate("draft", "Sides")) self.numFaces.setToolTip(translate("draft", "Number of sides")) self.isCopy.setText(translate("draft", "Copy")+" ("+inCommandShortcuts["Copy"][0]+")") self.isCopy.setToolTip(translate("draft", "If checked, objects will be copied instead of moved. Preferences -> Draft -> Global copy mode to keep this mode in next commands")) self.isSubelementMode.setText(translate("draft", "Modify subelements")+" ("+inCommandShortcuts["SubelementMode"][0]+")") self.isSubelementMode.setToolTip(translate("draft", "If checked, subelements will be modified instead of entire objects")) self.SStringValue.setToolTip(translate("draft", "Text string to draw")) self.labelSString.setText(translate("draft", "String")) self.SSizeValue.setToolTip(translate("draft", "Height of text")) self.labelSSize.setText(translate("draft", "Height")) self.STrackValue.setToolTip(translate("draft", "Intercharacter spacing")) self.labelSTrack.setText(translate("draft", "Tracking")) self.labelFFile.setText(translate("draft", "Full path to font file:")) self.chooserButton.setToolTip(translate("draft", "Open a FileChooser for font file")) self.textOkButton.setText(translate("draft", "Create text")) self.textOkButton.setToolTip(translate("draft", "Press this button to create the text object, or finish your text with two blank lines")) self.retranslateTray(widget) # Update the maximum width of the push buttons maxwidth = 66 # that's the default pb = [] for i in range(self.layout.count()): w = self.layout.itemAt(i).widget() if w is not None and w.inherits('QPushButton'): pb.append(w) for i in pb: fm = QtGui.QFontMetrics(i.font()) fw = fm.width(i.text()) fw = max(fw, maxwidth) maxwidth = maxwidth + 16 +10 # add icon width and a margin for i in pb: i.setMaximumWidth(maxwidth) def retranslateTray(self,widget=None): self.wplabel.setToolTip(translate("draft", "Current working plane")+":"+self.wplabel.text()) self.styleButton.setToolTip(translate("draft", "Change default style for new objects")) self.constrButton.setToolTip(translate("draft", "Toggle construction mode")) self.autoGroupButton.setToolTip(translate("draft", "Autogroup off")) #--------------------------------------------------------------------------- # Interface modes #--------------------------------------------------------------------------- def taskUi(self,title="Draft",extra=None,icon="Draft_Draft"): # reset InputField values self.reset_ui_values() self.isTaskOn = True todo.delay(FreeCADGui.Control.closeDialog,None) self.baseWidget = DraftBaseWidget() self.layout = QtGui.QVBoxLayout(self.baseWidget) self.setupToolBar(task=True) self.retranslateUi(self.baseWidget) self.panel = DraftTaskPanel(self.baseWidget,extra) todo.delay(FreeCADGui.Control.showDialog,self.panel) self.setTitle(title,icon) def redraw(self): """utility function that is performed after each clicked point""" self.checkLocal() def setFocus(self,f=None): p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft") if p.GetBool("focusOnLength",False) and self.lengthValue.isVisible(): self.lengthValue.setFocus() self.lengthValue.setSelection(0,self.number_length(self.lengthValue.text())) elif self.angleLock.isVisible() and self.angleLock.isChecked(): self.lengthValue.setFocus() self.lengthValue.setSelection(0,self.number_length(self.lengthValue.text())) elif (f is None) or (f == "x"): self.xValue.setFocus() self.xValue.setSelection(0,self.number_length(self.xValue.text())) elif f == "y": self.yValue.setFocus() self.yValue.setSelection(0,self.number_length(self.yValue.text())) elif f == "z": self.zValue.setFocus() self.zValue.setSelection(0,self.number_length(self.zValue.text())) elif f == "radius": self.radiusValue.setFocus() self.radiusValue.setSelection(0,self.number_length(self.radiusValue.text())) def number_length(self, str): nl = 0 for char in str: if char in "0123456789.,-": nl += 1 return nl def extraLineUi(self): '''shows length and angle controls''' self.labellength.show() self.lengthValue.show() self.labelangle.show() self.angleValue.show() self.angleLock.show() self.angleLock.setChecked(False) def hideXYZ(self): ''' turn off all the point entry widgets ''' self.labelx.hide() self.labely.hide() self.labelz.hide() self.labellength.hide() self.labelangle.hide() self.xValue.hide() self.yValue.hide() self.zValue.hide() self.pointButton.hide() self.lengthValue.hide() self.angleValue.hide() self.angleLock.hide() self.isRelative.hide() self.isGlobal.hide() def lineUi(self, title=translate("draft", "Line"), cancel=None, extra=None, getcoords=None,rel=False,icon="Draft_Line"): self.pointUi(title, cancel, extra, getcoords, rel, icon) self.extraLineUi() self.xValue.setEnabled(True) self.yValue.setEnabled(True) self.continueCmd.show() def wireUi(self, title=translate("draft", "DWire"), cancel=None, extra=None, getcoords=None, rel=False, icon="Draft_Wire"): self.pointUi(title, cancel, extra, getcoords, rel, icon) self.xValue.setEnabled(True) self.yValue.setEnabled(True) if Draft.getParam("UsePartPrimitives",False): self.hasFill.setEnabled(False) else: self.hasFill.setEnabled(True) self.hasFill.show() self.finishButton.show() self.closeButton.show() self.wipeButton.show() self.orientWPButton.show() self.undoButton.show() self.continueCmd.show() def circleUi(self): self.pointUi(translate("draft", "Circle"),icon="Draft_Circle") self.extUi() self.isRelative.hide() def arcUi(self): self.pointUi(translate("draft", "Arc"),icon="Draft_Arc") self.continueCmd.show() self.isRelative.hide() def rotateSetCenterUi(self): self.pointUi(translate("draft", "Rotate"),icon="Draft_Rotate") self.modUi() self.isRelative.hide() def pointUi(self, title=translate("draft","Point"), cancel=None, extra=None, getcoords=None, rel=False, icon="Draft_Draft"): if cancel: self.cancel = cancel if getcoords: self.pointcallback = getcoords self.taskUi(title,extra,icon) self.xValue.setEnabled(True) self.yValue.setEnabled(True) self.isRelative.show() self.isGlobal.show() self.checkLocal() self.labelx.show() self.labely.show() self.labelz.show() self.xValue.show() self.yValue.show() self.zValue.show() # reset UI to (0,0,0) on start self.xValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.yValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.zValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.x = 0 self.y = 0 self.z = 0 self.pointButton.show() if rel: self.isRelative.show() todo.delay(self.setFocus,None) self.xValue.selectAll() def labelUi(self,title=translate("draft","Label"),callback=None): w = QtGui.QWidget() w.setWindowTitle(translate("draft","Label type", utf8_decode=True)) l = QtGui.QVBoxLayout(w) combo = QtGui.QComboBox() from draftobjects.label import get_label_types types = get_label_types() for s in types: combo.addItem(s) combo.setCurrentIndex(types.index(Draft.getParam("labeltype","Custom"))) l.addWidget(combo) QtCore.QObject.connect(combo,QtCore.SIGNAL("currentIndexChanged(int)"),callback) self.pointUi(title=title, extra=w, icon="Draft_Label") def extraUi(self): pass def offsetUi(self): self.taskUi(translate("draft","Offset"), icon="Draft_Offset") self.radiusUi() self.isCopy.show() p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft") self.isCopy.setChecked(p.GetBool("OffsetCopyMode",False)) self.occOffset.show() self.labelRadius.setText(translate("draft","Distance")) self.radiusValue.setToolTip(translate("draft", "Offset distance")) self.radiusValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) todo.delay(self.radiusValue.setFocus,None) self.radiusValue.selectAll() def offUi(self): todo.delay(FreeCADGui.Control.closeDialog,None) self.cancel = None self.sourceCmd = None self.pointcallback = None self.mask = None self.isTaskOn = False self.baseWidget = QtGui.QWidget() def trimUi(self,title=translate("draft","Trimex")): self.taskUi(title, icon="Draft_Trimex") self.radiusUi() self.labelRadius.setText(translate("draft","Distance")) self.radiusValue.setToolTip(translate("draft", "Offset distance")) self.radiusValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) todo.delay(self.radiusValue.setFocus,None) self.radiusValue.selectAll() def radiusUi(self): self.hideXYZ() self.labelRadius.setText(translate("draft", "Radius")) self.radiusValue.setToolTip(translate("draft", "Radius of Circle")) self.labelRadius.show() self.radiusValue.setText(FreeCAD.Units.Quantity(0,FreeCAD.Units.Length).UserString) self.radiusValue.show() todo.delay(self.radiusValue.setFocus,None) self.radiusValue.selectAll() def textUi(self): self.hideXYZ() self.textValue.show() self.textOkButton.show() self.textValue.setText('') todo.delay(self.textValue.setFocus,None) self.textbuffer=[] self.textline=0 self.continueCmd.show() # Change the checkbox label as the in-command shortcut cannot be used: self.continueCmd.setText(translate("draft", "Continue")) def switchUi(self,store=True): if store: self.state = [] self.state.append(self.labelx.isVisible()) self.state.append(self.labely.isVisible()) self.state.append(self.labelz.isVisible()) self.state.append(self.xValue.isVisible()) self.state.append(self.yValue.isVisible()) self.state.append(self.zValue.isVisible()) self.hideXYZ() else: if self.state: if self.state[0]:self.labelx.show() if self.state[1]:self.labely.show() if self.state[2]:self.labelz.show() if self.state[3]:self.xValue.show() if self.state[4]:self.yValue.show() if self.state[5]:self.zValue.show() self.state = None def setTitle(self,title,icon="Draft_Draft"): self.baseWidget.setWindowTitle(title) self.baseWidget.setWindowIcon(QtGui.QIcon(":/icons/"+icon+".svg")) def selectUi(self,extra=None, on_close_call=None): self.makeDumbTask(extra, on_close_call) def editUi(self): self.makeDumbTask(on_close_call=self.finish) def extUi(self): if Draft.getParam("UsePartPrimitives",False): self.hasFill.setEnabled(False) else: self.hasFill.setEnabled(True) self.hasFill.show() self.continueCmd.show() def modUi(self): self.isCopy.show() self.isSubelementMode.show() p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft") if p.GetBool("copymode",False): self.isCopy.setChecked(p.GetBool("copymodeValue",False)) self.continueCmd.show() def checkLocal(self): """checks if x,y,z coords must be displayed as local or global""" if not self.globalMode and self.relativeMode: self.labelx.setText(translate("draft", "Local \u0394X")) self.labely.setText(translate("draft", "Local \u0394Y")) self.labelz.setText(translate("draft", "Local \u0394Z")) elif not self.globalMode and not self.relativeMode: self.labelx.setText(translate("draft", "Local X")) self.labely.setText(translate("draft", "Local Y")) self.labelz.setText(translate("draft", "Local Z")) elif self.globalMode and self.relativeMode: self.labelx.setText(translate("draft", "Global \u0394X")) self.labely.setText(translate("draft", "Global \u0394Y")) self.labelz.setText(translate("draft", "Global \u0394Z")) else: self.labelx.setText(translate("draft", "Global X")) self.labely.setText(translate("draft", "Global Y")) self.labelz.setText(translate("draft", "Global Z")) def setNextFocus(self): def isThere(widget): if widget.isEnabled() and widget.isVisible(): return True else: return False if self.isTaskOn: if isThere(self.xValue): self.setFocus() elif isThere(self.yValue): self.yValue.setFocus() self.yValue.selectAll() elif isThere(self.zValue): self.zValue.setFocus() self.zValue.selectAll() elif isThere(self.radiusValue): self.radiusValue.setFocus() self.radiusValue.selectAll() def setRelative(self,val=1): self.relativeMode = bool(val) self.checkLocal() def setGlobal(self,val=0): self.globalMode = bool(val) self.checkLocal() def setCopymode(self,val=0): p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft") p.SetBool("copymodeValue",bool(val)) # special value for offset command if self.sourceCmd: if self.sourceCmd.featureName == "Offset": p.SetBool("OffsetCopyMode",bool(val)) def setSubelementMode(self): self.sourceCmd.set_ghosts() def relocate(self): """relocates the right-aligned buttons depending on the toolbar size""" if self.baseWidget.geometry().width() < 400: self.layout.setDirection(QtGui.QBoxLayout.TopToBottom) else: self.layout.setDirection(QtGui.QBoxLayout.LeftToRight) def makeDumbTask(self, extra=None, on_close_call=None): """create a dumb taskdialog to prevent deleting the temp object""" class TaskPanel: def __init__(self, extra=None, callback=None): if extra: self.form = [extra] self.callback = callback def getStandardButtons(self): return int(QtGui.QDialogButtonBox.Close) def reject(self): if self.callback: self.callback() return True todo.delay(FreeCADGui.Control.closeDialog,None) panel = TaskPanel(extra, on_close_call) todo.delay(FreeCADGui.Control.showDialog,panel) #--------------------------------------------------------------------------- # Processing functions #--------------------------------------------------------------------------- def setContinue(self,val): if self.continueCmd.isVisible(): self.continueMode = bool(val) def setFill(self,val): if self.hasFill.isVisible(): self.fillmode = bool(val) def checkx(self): if self.yValue.isEnabled(): self.yValue.setFocus() self.yValue.setSelection(0,self.number_length(self.yValue.text())) self.updateSnapper() else: self.checky() def checky(self): if self.zValue.isEnabled(): self.zValue.setFocus() self.zValue.setSelection(0,self.number_length(self.zValue.text())) self.updateSnapper() else: self.validatePoint() def checklength(self): if self.angleValue.isEnabled(): self.angleValue.setFocus() self.angleValue.setSelection(0,self.number_length(self.angleValue.text())) self.updateSnapper() else: self.validatePoint() def validatePoint(self): """function for checking and sending numbers entered manually""" if self.sourceCmd or self.pointcallback: if (self.labelRadius.isVisible()): try: #rad=float(self.radiusValue.text()) rad = self.radius except (ValueError, AttributeError): print("debug: DraftGui.validatePoint: AttributeError") else: self.sourceCmd.numericRadius(rad) elif (self.labelx.isVisible()): try: #numx=float(self.xValue.text()) numx = self.x #numy=float(self.yValue.text()) numy = self.y #numz=float(self.zValue.text()) numz = self.z except (ValueError, AttributeError): print("debug: DraftGui.validatePoint: AttributeError") else: num_vec = FreeCAD.Vector(numx, numy, numz) ref_vec = FreeCAD.Vector(0,0,0) if self.pointcallback: self.pointcallback(num_vec, self.relativeMode) else: if FreeCAD.DraftWorkingPlane and not self.globalMode: num_vec = FreeCAD.DraftWorkingPlane.getGlobalRot(num_vec) ref_vec = FreeCAD.DraftWorkingPlane.getGlobalCoords(ref_vec) if self.relativeMode: if self.sourceCmd.node: ref_vec = self.sourceCmd.node[-1] numx, numy, numz = num_vec + ref_vec self.sourceCmd.numericInput(numx,numy,numz) elif (self.textValue.isVisible() or self.SStringValue.isVisible() or self.SSizeValue.isVisible() or self.STrackValue.isVisible() or self.FFileValue.isVisible()): return False else: FreeCADGui.ActiveDocument.resetEdit() return True def validateSNumeric(self): ''' send valid numeric parameters to ShapeString ''' if self.sourceCmd: if (self.labelSSize.isVisible()): try: SSize=float(self.SSize) except ValueError: FreeCAD.Console.PrintMessage(translate("draft", "Invalid Size value. Using 200.0.")) self.sourceCmd.numericSSize(200.0) else: self.sourceCmd.numericSSize(SSize) elif (self.labelSTrack.isVisible()): try: track=int(self.STrack) except ValueError: FreeCAD.Console.PrintMessage(translate("draft", "Invalid Tracking value. Using 0.")) self.sourceCmd.numericSTrack(0) else: self.sourceCmd.numericSTrack(track) def validateSString(self): ''' send a valid text string to ShapeString as unicode ''' if self.sourceCmd: if (self.labelSString.isVisible()): if self.SStringValue.text(): #print("debug: D_G DraftToolBar.validateSString type(SStringValue.text): " str(type(self.SStringValue.text))) #self.sourceCmd.validSString(str(self.SStringValue.text())) # QString to QByteArray to PyString self.sourceCmd.validSString(self.SStringValue.text()) # PySide returns Unicode from QString else: FreeCAD.Console.PrintMessage(translate("draft", "Please enter a text string.")) def pickFile(self): ''' invoke a font file chooser dialog and send result to ShapeString to''' if self.sourceCmd: if (self.chooserButton.isVisible()): try: dialogCaption = translate("draft", "Select a Font file") dialogDir = os.path.dirname(Draft.getParam("FontFile",)) # reasonable default? dialogFilter = "Fonts (*.ttf *.pfb *.otf);;All files (*.*)" fname = QtGui.QFileDialog.getOpenFileName(self.baseWidget, dialogCaption, dialogDir, dialogFilter) # fname = utf8_decode(fname[0]) # 1947: utf8_decode fails ('ascii' codec can't encode character) # when fname[0] contains accented chars fname = fname[0].encode('utf8') #TODO: this needs changing for Py3?? # accented chars cause "UnicodeEncodeError" failure in DraftGui.todo without # .encode('utf8') except Exception as e: FreeCAD.Console.PrintMessage("DraftGui.pickFile: unable to select a font file.") print(type(e)) print(e.args) else: if fname[0]: self.FFileValue.setText(fname) self.sourceCmd.validFFile(fname) else: FreeCAD.Console.PrintMessage("DraftGui.pickFile: no file selected.") # can this happen? def validateFile(self): ''' check and send font file parameter to ShapeString as unicode''' if self.sourceCmd: if (self.labelFFile.isVisible()): if self.FFileValue.text(): # QString to PyString self.sourceCmd.validFFile( utf8_decode(self.FFileValue.text())) else: FreeCAD.Console.PrintMessage( translate("draft", "Please enter a font file.")) def finish(self, cont=None): """finish button action""" if self.sourceCmd: if cont is None: cont = self.continueMode self.sourceCmd.finish(cont=cont) if self.cancel: self.cancel() self.cancel = None if FreeCADGui.ActiveDocument: FreeCADGui.ActiveDocument.resetEdit() def escape(self): """escapes the current command""" self.finish(cont=False) def closeLine(self): """close button action""" self.sourceCmd.finish(cont=self.continueMode, closed=True) FreeCADGui.ActiveDocument.resetEdit() def wipeLine(self): """wipes existing segments of a line""" self.sourceCmd.wipe() def orientWP(self): """reorients the current working plane""" self.sourceCmd.orientWP() def selectEdge(self): """allows the dimension command to select an edge""" if hasattr(self.sourceCmd,"selectEdge"): self.sourceCmd.selectEdge() def undoSegment(self): """undo last line segment""" if hasattr(self.sourceCmd,"undolast"): self.sourceCmd.undolast() def checkSpecialChars(self,txt): """checks for special characters in the entered coords that must be treated as shortcuts """ # in-command shortcut definitions #Relative #Continue #Close #Copy #Fill #Exit #Snap #Increase #Decrease #RestrictX #RestrictY #RestrictZ #SelectEdge #AddHold #Length #Wipe #SetWP #Undo spec = False if txt.upper().startswith(inCommandShortcuts["Relative"][0]): if self.isRelative.isVisible(): self.isRelative.setChecked(not self.isRelative.isChecked()) self.relativeMode = self.isRelative.isChecked() spec = True if txt.upper().startswith(inCommandShortcuts["Global"][0]): if self.isGlobal.isVisible(): self.isGlobal.setChecked(not self.isGlobal.isChecked()) self.globalMode = self.isGlobal.isChecked() spec = True elif txt.upper().startswith(inCommandShortcuts["Fill"][0]): if self.hasFill.isVisible(): self.hasFill.setChecked(not self.hasFill.isChecked()) spec = True elif txt.upper().startswith(inCommandShortcuts["Exit"][0]): if self.finishButton.isVisible(): self.finish() elif txt.upper().startswith(inCommandShortcuts["Continue"][0]): if self.continueCmd.isVisible(): self.toggleContinue() spec = True elif txt.upper().startswith(inCommandShortcuts["Wipe"][0]): if self.wipeButton.isVisible(): self.wipeLine() spec = True elif txt.upper().startswith(inCommandShortcuts["SelectEdge"][0]): self.selectEdge() spec = True elif txt.upper().startswith(inCommandShortcuts["Snap"][0]): self.togglesnap() spec = True elif txt.upper().startswith(inCommandShortcuts["Increase"][0]): self.toggleradius(1) spec = True elif txt.upper().startswith(inCommandShortcuts["Decrease"][0]): self.toggleradius(-1) spec = True elif txt.upper().startswith(inCommandShortcuts["AddHold"][0]): if hasattr(FreeCADGui,"Snapper"): FreeCADGui.Snapper.addHoldPoint() spec = True elif txt.upper().startswith(inCommandShortcuts["RestrictX"][0]): self.constrain("x") self.displayPoint() spec = True elif txt.upper().startswith(inCommandShortcuts["RestrictY"][0]): self.constrain("y") self.displayPoint() spec = True elif txt.upper().startswith(inCommandShortcuts["RestrictZ"][0]): self.constrain("z") self.displayPoint() spec = True elif txt.upper().startswith(inCommandShortcuts["Length"][0]): self.constrain("angle") self.displayPoint() spec = True elif txt.upper().startswith(inCommandShortcuts["Close"][0]): if self.closeButton.isVisible(): self.closeLine() elif txt.upper().startswith(inCommandShortcuts["SetWP"][0]): if self.orientWPButton.isVisible(): self.orientWP() spec = True elif txt.upper().startswith(inCommandShortcuts["Copy"][0]): if self.isCopy.isVisible(): self.isCopy.setChecked(not self.isCopy.isChecked()) spec = True elif txt.upper().startswith(inCommandShortcuts["Undo"][0]): self.undoSegment() spec = True elif txt.upper().startswith(inCommandShortcuts["SubelementMode"][0]): if self.isSubelementMode.isVisible(): self.isSubelementMode.setChecked(not self.isSubelementMode.isChecked()) spec = True if spec: widget = self.baseWidget.focusWidget() field = self.input_fields[widget.objectName()] value = getattr(self, field["value"]) unit = getattr(FreeCAD.Units, field["unit"]) v = FreeCAD.Units.Quantity(value, unit).getUserPreferred()[0] widget.setProperty("text",v) widget.setFocus() widget.selectAll() self.updateSnapper() def updateSnapper(self): """updates the snapper track line if applicable""" if hasattr(FreeCADGui,"Snapper"): if FreeCADGui.Snapper.trackLine: if FreeCADGui.Snapper.trackLine.Visible: last = FreeCAD.Vector(0,0,0) if not self.xValue.isVisible(): return if self.isRelative.isChecked(): if self.sourceCmd: if hasattr(self.sourceCmd,"node"): if self.sourceCmd.node: last = self.sourceCmd.node[-1] delta = FreeCAD.DraftWorkingPlane.getGlobalCoords( FreeCAD.Vector(self.x,self.y,self.z)) FreeCADGui.Snapper.trackLine.p2(last.add(delta)) def checkEnterText(self): """this function checks if the entered text ends with two blank lines""" t = self.textValue.toPlainText() if t.endswith("\n\n"): self.sendText() def sendText(self): """this function sends the entered text to the active draft command if enter has been pressed twice. Otherwise it blanks the line. """ self.sourceCmd.text = self.textValue.toPlainText()\ .replace("\\","\\\\")\ .replace("\"","\\\"")\ .replace("\'","\\\'")\ .splitlines() self.sourceCmd.createObject() def displayPoint(self, point=None, last=None, plane=None, mask=None): """this function displays the passed coords in the x, y, and z widgets""" if not self.isTaskOn: return if not plane: plane = FreeCAD.DraftWorkingPlane # get coords to display if not last: if self.globalMode: last = FreeCAD.Vector(0,0,0) else: last = plane.getPlacement().Base dp = None if point: dp = point if self.relativeMode: # and (last is not None): if self.globalMode: dp = point - last else: dp = plane.getLocalRot(point - last) else: if self.globalMode: dp = point else: dp = plane.getLocalCoords(point) # set widgets if dp: if self.mask in ['y','z']: self.xValue.setText(displayExternal(dp.x,None,'Length')) else: self.xValue.setText(displayExternal(dp.x,None,'Length')) if self.mask in ['x','z']: self.yValue.setText(displayExternal(dp.y,None,'Length')) else: self.yValue.setText(displayExternal(dp.y,None,'Length')) if self.mask in ['x','y']: self.zValue.setText(displayExternal(dp.z,None,'Length')) else: self.zValue.setText(displayExternal(dp.z,None,'Length')) # set length and angle if last and dp and plane: length, theta, phi = DraftVecUtils.get_spherical_coords(*dp) theta = math.degrees(theta) phi = math.degrees(phi) self.lengthValue.setText(displayExternal(length,None,'Length')) #if not self.angleLock.isChecked(): self.angleValue.setText(displayExternal(phi,None,'Angle')) if not mask: # automask, phi is rounded to identify one of the below cases phi = round(phi, Draft.getParam("precision")) if phi in [0,180,-180]: mask = "x" elif phi in [90,270,-90,-270]: mask = "y" # set masks if (mask == "x") or (self.mask == "x"): self.xValue.setEnabled(True) self.yValue.setEnabled(False) self.zValue.setEnabled(False) self.angleValue.setEnabled(False) self.setFocus() elif (mask == "y") or (self.mask == "y"): self.xValue.setEnabled(False) self.yValue.setEnabled(True) self.zValue.setEnabled(False) self.angleValue.setEnabled(False) self.setFocus("y") elif (mask == "z") or (self.mask == "z"): self.xValue.setEnabled(False) self.yValue.setEnabled(False) self.zValue.setEnabled(True) self.angleValue.setEnabled(False) self.setFocus("z") else: self.xValue.setEnabled(True) self.yValue.setEnabled(True) self.zValue.setEnabled(True) self.angleValue.setEnabled(True) self.setFocus() def getDefaultColor(self,type,rgb=False): """gets color from the preferences or toolbar""" r = 0 g = 0 b = 0 if type == "snap": color = Draft.getParam("snapcolor",4294967295) r = ((color>>24)&0xFF)/255 g = ((color>>16)&0xFF)/255 b = ((color>>8)&0xFF)/255 elif type == "ui": print("draft: deprecation warning: Do not use getDefaultColor(\"ui\") anymore - use getDefaultColor(\"line\") instead.") r = float(self.color.red()/255.0) g = float(self.color.green()/255.0) b = float(self.color.blue()/255.0) elif type == "line": color = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/View")\ .GetUnsigned("DefaultShapeLineColor",255) r = ((color>>24)&0xFF)/255 g = ((color>>16)&0xFF)/255 b = ((color>>8)&0xFF)/255 elif type == "text": color = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft")\ .GetUnsigned("DefaultTextColor",255) r = ((color>>24)&0xFF)/255 g = ((color>>16)&0xFF)/255 b = ((color>>8)&0xFF)/255 elif type == "face": color = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/View")\ .GetUnsigned("DefaultShapeColor",4294967295) r = ((color>>24)&0xFF)/255 g = ((color>>16)&0xFF)/255 b = ((color>>8)&0xFF)/255 elif type == "constr": color = Draft.getParam("constructioncolor",746455039) r = ((color>>24)&0xFF)/255 g = ((color>>16)&0xFF)/255 b = ((color>>8)&0xFF)/255 else: print("draft: error: couldn't get a color for ",type," type.") if rgb: return("rgb("+str(int(r*255))+","+str(int(g*255))+","+str(int(b*255))+")") else: return (r,g,b) def cross(self,on=True): """deprecated""" pass def toggleConstrMode(self,checked): self.baseWidget.setStyleSheet( "#constrButton:Checked {background-color: " + self.getDefaultColor("constr",rgb=True)+" }") self.constrMode = checked def toggleContinue(self): FreeCAD.Console.PrintMessage("toggle continue\n") self.continueMode = not self.continueMode try: if hasattr(self,"continueCmd"): if self.continueCmd.isVisible(): self.continueCmd.toggle() if hasattr(self,"panel"): if hasattr(self.panel,"form"): if isinstance(self.panel.form,list): for w in self.panel.form: c = w.findChild(QtGui.QCheckBox,"ContinueCmd") if c: c.toggle() else: c = self.panel.form.findChild(QtGui.QCheckBox,"ContinueCmd") if c: c.toggle() except Exception: pass def isConstructionMode(self): return self.tray is not None and self.constrButton.isChecked() def drawPage(self): self.sourceCmd.draw() def changePage(self,index): pagename = str(self.pageBox.itemText(index)) vobj = FreeCADGui.ActiveDocument.getObject(pagename) if vobj: self.scaleBox.setEditText(str(vobj.HintScale)) self.marginXValue.setValue(float(vobj.HintOffsetX)) self.marginYValue.setValue(float(vobj.HintOffsetY)) def selectplane(self): FreeCADGui.runCommand("Draft_SelectPlane") def setstyle(self): FreeCADGui.runCommand("Draft_SetStyle") def setStyleButton(self): "sets icon and text on the style button" linecolor = QtGui.QColor(Draft.getParam("color",255)>>8) facecolor = QtGui.QColor( FreeCAD.ParamGet("User parameter:BaseApp/Preferences/View") .GetUnsigned( "DefaultShapeColor",4294967295)>>8) im = QtGui.QImage(32,32,QtGui.QImage.Format_ARGB32) im.fill(QtCore.Qt.transparent) pt = QtGui.QPainter(im) pt.setPen(QtGui.QPen(QtCore.Qt.black, 1, QtCore.Qt.SolidLine, QtCore.Qt.FlatCap)) pt.setBrush(QtGui.QBrush(linecolor, QtCore.Qt.SolidPattern)) pts = [QtCore.QPointF(4.0,4.0), QtCore.QPointF(4.0,26.0), QtCore.QPointF(26.0,4.0)] pt.drawPolygon(pts, QtCore.Qt.OddEvenFill) pt.setBrush(QtGui.QBrush(facecolor, QtCore.Qt.SolidPattern)) pts = [QtCore.QPointF(28.0,28.0),QtCore.QPointF(8.0,28.0),QtCore.QPointF(28.0,8.0)] pt.drawPolygon(pts,QtCore.Qt.OddEvenFill) pt.end() icon = QtGui.QIcon(QtGui.QPixmap.fromImage(im)) linewidth = Draft.getParam("linewidth",2) fontsize = Draft.getParam("textheight",0.20) txt = str(linewidth) + "px | "\ + FreeCAD.Units.Quantity(fontsize,FreeCAD.Units.Length).UserString self.styleButton.setIcon(icon) self.styleButton.setText(txt) # FOR BACKWARDS COMPATIBILITY self.color = linecolor self.facecolor = facecolor self.linewidth = linewidth self.fontsize = fontsize def popupMenu(self,llist,ilist=None,pos=None): """pops up a menu filled with the given list""" self.groupmenu = QtGui.QMenu() for i,l in enumerate(llist): if ilist: self.groupmenu.addAction(ilist[i],l) else: self.groupmenu.addAction(l) if not pos: pos = FreeCADGui.getMainWindow().cursor().pos() self.groupmenu.popup(pos) QtCore.QObject.connect(self.groupmenu,QtCore.SIGNAL("triggered(QAction *)"),self.popupTriggered) def getIcon(self,iconpath): return QtGui.QIcon(iconpath) def popupTriggered(self,action): self.sourceCmd.proceed(str(action.text())) def setRadiusValue(self,val,unit=None): #print("DEBUG: setRadiusValue val: ", val, " unit: ", unit) if not isinstance(val, (int, float)): #??some code passes strings or ??? t = val elif unit: t= displayExternal(val,None, unit) else: print("Error: setRadiusValue called for number without Dimension") t = displayExternal(val,None, None) self.radiusValue.setText(t) self.radiusValue.setFocus() def runAutoGroup(self): FreeCADGui.runCommand("Draft_AutoGroup") def setAutoGroup(self,value=None): if value is None: self.autogroup = None self.autoGroupButton.setText(translate("draft", "None")) self.autoGroupButton.setIcon(QtGui.QIcon.fromTheme('Draft_AutoGroup_off', QtGui.QIcon(':/icons/button_invalid.svg'))) self.autoGroupButton.setToolTip(translate("draft", "Autogroup off")) self.autoGroupButton.setDown(False) else: obj = FreeCAD.ActiveDocument.getObject(value) if obj: self.autogroup = value self.autoGroupButton.setText(obj.Label) self.autoGroupButton.setIcon(obj.ViewObject.Icon) self.autoGroupButton.setToolTip(translate("draft", "Autogroup:") + " " + obj.Label) self.autoGroupButton.setDown(False) else: self.autogroup = None self.autoGroupButton.setText(translate("draft", "None")) self.autoGroupButton.setIcon(QtGui.QIcon.fromTheme('Draft_AutoGroup_off', QtGui.QIcon(':/icons/button_invalid.svg'))) self.autoGroupButton.setToolTip(translate("draft", "Autogroup off")) self.autoGroupButton.setDown(False) def getXPM(self,iconname,size=16): i = QtGui.QIcon(":/icons/"+iconname+".svg") p = i.pixmap(size,size) a = QtCore.QByteArray() b = QtCore.QBuffer(a) b.open(QtCore.QIODevice.WriteOnly) p.save(b,"XPM") b.close() return str(a) def togglesnap(self): FreeCADGui.doCommand('FreeCADGui.runCommand("Draft_Snap_Lock")') def toggleradius(self,val): if hasattr(FreeCADGui,"Snapper"): par = Draft.getParam("snapRange", 8) Draft.setParam("snapRange", max(0, par+val)) FreeCADGui.Snapper.showradius() def constrain(self,val): if val == "angle": self.alock = not(self.alock) self.angleLock.setChecked(self.alock) elif self.mask == val: self.mask = None if hasattr(FreeCADGui,"Snapper"): FreeCADGui.Snapper.mask = None else: self.mask = val if hasattr(FreeCADGui,"Snapper"): FreeCADGui.Snapper.mask = val def changeXValue(self,d): self.x = d if not self.xValue.hasFocus(): return None self.update_spherical_coords() def changeYValue(self,d): self.y = d if not self.yValue.hasFocus(): return None self.update_spherical_coords() def changeZValue(self,d): self.z = d if not self.zValue.hasFocus(): return None self.update_spherical_coords() def changeRadiusValue(self,d): self.radius = d def changeSSizeValue(self,d): self.SSize = d def changeSTrackValue(self,d): self.STrack = d def changeLengthValue(self,d): self.lvalue = d if not self.lengthValue.hasFocus(): return None self.update_cartesian_coords() def changeAngleValue(self,d): self.avalue = d if not self.angleValue.hasFocus(): return None self.update_cartesian_coords() if self.angleLock.isChecked(): if not self.globalMode: angle_vec = FreeCAD.DraftWorkingPlane.getGlobalRot(self.angle) else: angle_vec = self.angle FreeCADGui.Snapper.setAngle(angle_vec) def toggleAngle(self,b): self.alock = self.angleLock.isChecked() self.update_cartesian_coords() if self.alock: if not self.globalMode: angle_vec = FreeCAD.DraftWorkingPlane.getGlobalRot(self.angle) else: angle_vec = self.angle FreeCADGui.Snapper.setAngle(angle_vec) else: FreeCADGui.Snapper.setAngle() self.angle = None def update_spherical_coords(self): length, theta, phi = DraftVecUtils.get_spherical_coords( self.x,self.y,self.z) self.lvalue = length self.pvalue = math.degrees(theta) self.avalue = math.degrees(phi) self.angle = FreeCAD.Vector(DraftVecUtils.get_cartesian_coords( 1, theta, phi)) self.lengthValue.setText(displayExternal(self.lvalue,None,'Length')) self.angleValue.setText(displayExternal(self.avalue,None,'Angle')) def update_cartesian_coords(self): self.x, self.y, self.z = DraftVecUtils.get_cartesian_coords( self.lvalue,math.radians(self.pvalue),math.radians(self.avalue)) self.angle = FreeCAD.Vector(DraftVecUtils.get_cartesian_coords( 1, math.radians(self.pvalue), math.radians(self.avalue))) self.xValue.setText(displayExternal(self.x,None,'Length')) self.yValue.setText(displayExternal(self.y,None,'Length')) self.zValue.setText(displayExternal(self.z,None,'Length')) #--------------------------------------------------------------------------- # TaskView operations #--------------------------------------------------------------------------- def setWatchers(self): class DraftCreateWatcher: def __init__(self): self.commands = ["Draft_Line","Draft_Wire", "Draft_Rectangle","Draft_Arc", "Draft_Circle","Draft_BSpline", "Draft_Text","Draft_Dimension", "Draft_ShapeString","Draft_BezCurve"] self.title = "Create objects" def shouldShow(self): return (FreeCAD.ActiveDocument is not None) and (not FreeCADGui.Selection.getSelection()) class DraftModifyWatcher: def __init__(self): self.commands = ["Draft_Move","Draft_Rotate", "Draft_Scale","Draft_Offset", "Draft_Trimex","Draft_Upgrade", "Draft_Downgrade","Draft_Edit", "Draft_Drawing"] self.title = "Modify objects" def shouldShow(self): return (FreeCAD.ActiveDocument is not None) and (FreeCADGui.Selection.getSelection() != []) FreeCADGui.Control.addTaskWatcher([DraftCreateWatcher(),DraftModifyWatcher()]) def changeEvent(self, event): if event.type() == QtCore.QEvent.LanguageChange: #print("Language changed!") self.ui.retranslateUi(self) def Activated(self): self.setWatchers() if hasattr(self,"tray"): self.tray.show() def Deactivated(self): if (FreeCAD.activeDraftCommand is not None): self.continueMode = False FreeCAD.activeDraftCommand.finish() FreeCADGui.Control.clearTaskWatcher() #self.tray = None if hasattr(self,"tray"): self.tray.hide() def reset_ui_values(self): """Method to reset task panel values""" self.x = 0 self.y = 0 self.z = 0 self.lvalue = 0 self.pvalue = 90 self.avalue = 0 self.angle = None self.radius = 0 self.offset = 0 class FacebinderTaskPanel: '''A TaskPanel for the facebinder''' def __init__(self): self.obj = None self.form = QtGui.QWidget() self.form.setObjectName("FacebinderTaskPanel") self.grid = QtGui.QGridLayout(self.form) self.grid.setObjectName("grid") self.title = QtGui.QLabel(self.form) self.grid.addWidget(self.title, 0, 0, 1, 2) # tree self.tree = QtGui.QTreeWidget(self.form) self.grid.addWidget(self.tree, 1, 0, 1, 2) self.tree.setColumnCount(2) self.tree.setHeaderLabels(["Name","Subelement"]) # buttons self.addButton = QtGui.QPushButton(self.form) self.addButton.setObjectName("addButton") self.addButton.setIcon(QtGui.QIcon(":/icons/Arch_Add.svg")) self.grid.addWidget(self.addButton, 3, 0, 1, 1) self.delButton = QtGui.QPushButton(self.form) self.delButton.setObjectName("delButton") self.delButton.setIcon(QtGui.QIcon(":/icons/Arch_Remove.svg")) self.grid.addWidget(self.delButton, 3, 1, 1, 1) QtCore.QObject.connect(self.addButton, QtCore.SIGNAL("clicked()"), self.addElement) QtCore.QObject.connect(self.delButton, QtCore.SIGNAL("clicked()"), self.removeElement) self.update() def isAllowedAlterSelection(self): return True def isAllowedAlterView(self): return True def getStandardButtons(self): return int(QtGui.QDialogButtonBox.Ok) def update(self): """fills the treewidget""" self.tree.clear() if self.obj: for f in self.obj.Faces: if isinstance(f[1],tuple): for subf in f[1]: item = QtGui.QTreeWidgetItem(self.tree) item.setText(0,f[0].Name) item.setIcon(0, QtGui.QIcon(":/icons/Part_3D_object.svg")) item.setText(1,subf) else: item = QtGui.QTreeWidgetItem(self.tree) item.setText(0,f[0].Name) item.setIcon(0, QtGui.QIcon(":/icons/Part_3D_object.svg")) item.setText(1,f[1]) self.retranslateUi(self.form) def addElement(self): if self.obj: for sel in FreeCADGui.Selection.getSelectionEx(): if sel.HasSubObjects: obj = sel.Object for elt in sel.SubElementNames: if "Face" in elt: flist = self.obj.Faces found = False for face in flist: if (face[0] == obj.Name): if isinstance(face[1],tuple): for subf in face[1]: if subf == elt: found = True else: if (face[1] == elt): found = True if not found: flist.append((obj,elt)) self.obj.Faces = flist FreeCAD.ActiveDocument.recompute() self.update() def removeElement(self): if self.obj: it = self.tree.currentItem() if it: obj = FreeCAD.ActiveDocument.getObject(str(it.text(0))) elt = str(it.text(1)) flist = [] for face in self.obj.Faces: if (face[0].Name != obj.Name): flist.append(face) else: if isinstance(face[1],tuple): for subf in face[1]: if subf != elt: flist.append((obj,subf)) else: if (face[1] != elt): flist.append(face) self.obj.Faces = flist FreeCAD.ActiveDocument.recompute() self.update() def accept(self): FreeCAD.ActiveDocument.recompute() FreeCADGui.ActiveDocument.resetEdit() return True def retranslateUi(self, TaskPanel): TaskPanel.setWindowTitle(QtGui.QApplication.translate("draft", "Faces", None)) self.delButton.setText(QtGui.QApplication.translate("draft", "Remove", None)) self.addButton.setText(QtGui.QApplication.translate("draft", "Add", None)) self.title.setText(QtGui.QApplication.translate("draft", "Facebinder elements", None)) #def translateWidget(w, context=None, disAmb=None): # '''translator for items where retranslateUi() is unavailable. # translates widget w and children.''' # #handle w itself # if w.metaObject().className() == "QWidget": # origText = None # origText = w.windowTitle() # if origText: # newText = translate(context, str(origText)) # if newText: # w.setWindowTitle(newText) # #handle children # wKids = w.findChildren(QtGui.QWidget) # for i in wKids: # className = i.metaObject().className() # if hasattr(i,"text") and hasattr(i,"setText"): # origText = i.text() # newText = translate(context, str(origText)) # if newText: # i.setText(newText) # elif hasattr(i,"title") and hasattr(i,"setTitle"): # origText = i.title() # newText = translate(context, str(origText)) # if newText: # i.setTitle(newText) # elif hasattr(i,"itemText") and hasattr(i,"setItemText"): # for item in range(i.count()): # oldText = i.itemText(item) # newText = translate(context, str(origText)) # if newText: # i.setItemText(item,newText) ##for debugging: ## else: ## msg = "TranslateWidget: Can not translate widget: {0} type: {1}\n".format(w.objectName(),w.metaObject().className()) ## FreeCAD.Console.PrintMessage(msg) if not hasattr(FreeCADGui,"draftToolBar"): FreeCADGui.draftToolBar = DraftToolBar() #----End of Python Features Definitions----#