187 lines
8.5 KiB
Python
187 lines
8.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2017 sliptonic <shopinthewoods@gmail.com> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program is distributed in the hope that it will be useful, *
|
|
# * but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
# * GNU Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
import FreeCAD
|
|
import FreeCADGui
|
|
import PathScripts.PathLog as PathLog
|
|
import PathScripts.PathOpGui as PathOpGui
|
|
|
|
from PySide import QtCore, QtGui
|
|
|
|
__title__ = "Base for Circular Hole based operations' UI"
|
|
__author__ = "sliptonic (Brad Collette)"
|
|
__url__ = "http://www.freecadweb.org"
|
|
__doc__ = "Implementation of circular hole specific base geometry page controller."
|
|
|
|
LOGLEVEL = False
|
|
|
|
if LOGLEVEL:
|
|
PathLog.setLevel(PathLog.Level.DEBUG, PathLog.thisModule())
|
|
PathLog.trackModule(PathLog.thisModule())
|
|
else:
|
|
PathLog.setLevel(PathLog.Level.NOTICE, PathLog.thisModule())
|
|
|
|
class TaskPanelHoleGeometryPage(PathOpGui.TaskPanelBaseGeometryPage):
|
|
'''Controller class to be used for the BaseGeomtery page.
|
|
Circular holes don't just display the feature, they also add a column
|
|
displaying the radius the feature describes. This page provides that
|
|
UI and functionality for all circular hole based operations.'''
|
|
|
|
DataFeatureName = QtCore.Qt.ItemDataRole.UserRole
|
|
DataObject = QtCore.Qt.ItemDataRole.UserRole + 1
|
|
DataObjectSub = QtCore.Qt.ItemDataRole.UserRole + 2
|
|
|
|
def getForm(self):
|
|
'''getForm() ... load and return page'''
|
|
return FreeCADGui.PySideUic.loadUi(":/panels/PageBaseHoleGeometryEdit.ui")
|
|
|
|
def initPage(self, obj):
|
|
self.updating = False # pylint: disable=attribute-defined-outside-init
|
|
|
|
def setFields(self, obj):
|
|
'''setFields(obj) ... fill form with values from obj'''
|
|
PathLog.track()
|
|
self.form.baseList.blockSignals(True)
|
|
self.form.baseList.clearContents()
|
|
self.form.baseList.setRowCount(0)
|
|
for (base, subs) in obj.Base:
|
|
for sub in subs:
|
|
self.form.baseList.insertRow(self.form.baseList.rowCount())
|
|
|
|
item = QtGui.QTableWidgetItem("%s.%s" % (base.Label, sub))
|
|
item.setFlags(item.flags() | QtCore.Qt.ItemIsUserCheckable)
|
|
if obj.Proxy.isHoleEnabled(obj, base, sub):
|
|
item.setCheckState(QtCore.Qt.Checked)
|
|
else:
|
|
item.setCheckState(QtCore.Qt.Unchecked)
|
|
name = "%s.%s" % (base.Name, sub)
|
|
item.setData(self.DataFeatureName, name)
|
|
item.setData(self.DataObject, base)
|
|
item.setData(self.DataObjectSub, sub)
|
|
self.form.baseList.setItem(self.form.baseList.rowCount()-1, 0, item)
|
|
|
|
dia = obj.Proxy.holeDiameter(obj, base, sub)
|
|
item = QtGui.QTableWidgetItem("{:.3f}".format(dia))
|
|
item.setData(self.DataFeatureName, name)
|
|
item.setData(self.DataObject, base)
|
|
item.setData(self.DataObjectSub, sub)
|
|
item.setTextAlignment(QtCore.Qt.AlignHCenter)
|
|
self.form.baseList.setItem(self.form.baseList.rowCount()-1, 1, item)
|
|
|
|
self.form.baseList.resizeColumnToContents(0)
|
|
self.form.baseList.blockSignals(False)
|
|
self.form.baseList.setSortingEnabled(True)
|
|
self.itemActivated()
|
|
|
|
def itemActivated(self):
|
|
'''itemActivated() ... callback when item in table is selected'''
|
|
PathLog.track()
|
|
if self.form.baseList.selectedItems():
|
|
self.form.deleteBase.setEnabled(True)
|
|
FreeCADGui.Selection.clearSelection()
|
|
activatedRows = []
|
|
for item in self.form.baseList.selectedItems():
|
|
row = item.row()
|
|
if not row in activatedRows:
|
|
activatedRows.append(row)
|
|
obj = item.data(self.DataObject)
|
|
sub = str(item.data(self.DataObjectSub))
|
|
PathLog.debug("itemActivated() -> %s.%s" % (obj.Label, sub))
|
|
if sub:
|
|
FreeCADGui.Selection.addSelection(obj, sub)
|
|
else:
|
|
FreeCADGui.Selection.addSelection(obj)
|
|
else:
|
|
self.form.deleteBase.setEnabled(False)
|
|
|
|
def deleteBase(self):
|
|
'''deleteBase() ... callback for push button'''
|
|
PathLog.track()
|
|
selected = [self.form.baseList.row(item) for item in self.form.baseList.selectedItems()]
|
|
self.form.baseList.blockSignals(True)
|
|
for row in sorted(list(set(selected)), key=lambda row: -row):
|
|
self.form.baseList.removeRow(row)
|
|
self.updateBase()
|
|
self.form.baseList.resizeColumnToContents(0)
|
|
self.form.baseList.blockSignals(False)
|
|
#self.obj.Proxy.execute(self.obj)
|
|
FreeCAD.ActiveDocument.recompute()
|
|
self.setFields(self.obj)
|
|
|
|
def updateBase(self):
|
|
'''updateBase() ... helper function to transfer current table to obj'''
|
|
PathLog.track()
|
|
newlist = []
|
|
for i in range(self.form.baseList.rowCount()):
|
|
item = self.form.baseList.item(i, 0)
|
|
obj = item.data(self.DataObject)
|
|
sub = str(item.data(self.DataObjectSub))
|
|
base = (obj, sub)
|
|
PathLog.debug("keeping (%s.%s)" % (obj.Label, sub))
|
|
newlist.append(base)
|
|
PathLog.debug("obj.Base=%s newlist=%s" % (self.obj.Base, newlist))
|
|
self.updating = True # pylint: disable=attribute-defined-outside-init
|
|
self.obj.Base = newlist
|
|
self.updating = False # pylint: disable=attribute-defined-outside-init
|
|
|
|
def checkedChanged(self):
|
|
'''checkeChanged() ... callback when checked status of a base feature changed'''
|
|
PathLog.track()
|
|
disabled = []
|
|
for i in range(0, self.form.baseList.rowCount()):
|
|
item = self.form.baseList.item(i, 0)
|
|
if item.checkState() != QtCore.Qt.Checked:
|
|
disabled.append(item.data(self.DataFeatureName))
|
|
self.obj.Disabled = disabled
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
def registerSignalHandlers(self, obj):
|
|
'''registerSignalHandlers(obj) ... setup signal handlers'''
|
|
self.form.baseList.itemSelectionChanged.connect(self.itemActivated)
|
|
self.form.addBase.clicked.connect(self.addBase)
|
|
self.form.deleteBase.clicked.connect(self.deleteBase)
|
|
self.form.resetBase.clicked.connect(self.resetBase)
|
|
self.form.baseList.itemChanged.connect(self.checkedChanged)
|
|
|
|
def resetBase(self):
|
|
'''resetBase() ... push button callback'''
|
|
self.obj.Base = []
|
|
self.obj.Disabled = []
|
|
self.obj.Proxy.findAllHoles(self.obj)
|
|
|
|
self.obj.Proxy.execute(self.obj)
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
def updateData(self, obj, prop):
|
|
'''updateData(obj, prop) ... callback whenever a property of the model changed'''
|
|
if not self.updating and prop in ['Base', 'Disabled']:
|
|
self.setFields(obj)
|
|
|
|
class TaskPanelOpPage(PathOpGui.TaskPanelPage):
|
|
'''Base class for circular hole based operation's page controller.'''
|
|
|
|
def taskPanelBaseGeometryPage(self, obj, features):
|
|
'''taskPanelBaseGeometryPage(obj, features) ... Return circular hole specific page controller for Base Geometry.'''
|
|
return TaskPanelHoleGeometryPage(obj, features)
|