144 lines
6.0 KiB
Python
144 lines
6.0 KiB
Python
# ***************************************************************************
|
|
# * (c) 2019 Eliud Cabrera Castillo <e.cabrera-castillo@tum.de> *
|
|
# * *
|
|
# * 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 Library 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 *
|
|
# * *
|
|
# ***************************************************************************
|
|
"""Provides GUI tools to create polar Array objects."""
|
|
## @package gui_polararray
|
|
# \ingroup draftguitools
|
|
# \brief Provides GUI tools to create polar Array objects.
|
|
|
|
## \addtogroup draftguitools
|
|
# @{
|
|
from pivy import coin
|
|
from PySide.QtCore import QT_TRANSLATE_NOOP
|
|
|
|
import FreeCAD as App
|
|
import FreeCADGui as Gui
|
|
import Draft
|
|
import Draft_rc # include resources, icons, ui files
|
|
from draftguitools import gui_base
|
|
from draftutils import gui_utils
|
|
from draftutils import todo
|
|
from draftutils.messages import _log
|
|
from draftutils.translate import translate
|
|
from drafttaskpanels import task_polararray
|
|
|
|
# The module is used to prevent complaints from code checkers (flake8)
|
|
bool(Draft_rc.__name__)
|
|
|
|
|
|
class PolarArray(gui_base.GuiCommandBase):
|
|
"""Gui command for the PolarArray tool."""
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.command_name = "Polar array"
|
|
self.location = None
|
|
self.mouse_event = None
|
|
self.view = None
|
|
self.callback_move = None
|
|
self.callback_click = None
|
|
self.ui = None
|
|
self.point = App.Vector()
|
|
|
|
def GetResources(self):
|
|
"""Set icon, menu and tooltip."""
|
|
return {'Pixmap': 'Draft_PolarArray',
|
|
'MenuText': QT_TRANSLATE_NOOP("Draft_PolarArray", "Polar array"),
|
|
'ToolTip': QT_TRANSLATE_NOOP("Draft_PolarArray", "Creates copies of the selected object, and places the copies in a polar pattern\ndefined by a center of rotation and its angle.\n\nThe array can be turned into an orthogonal or a circular array by changing its type.")}
|
|
|
|
def Activated(self):
|
|
"""Execute when the command is called.
|
|
|
|
We add callbacks that connect the 3D view with
|
|
the widgets of the task panel.
|
|
"""
|
|
_log("GuiCommand: {}".format(self.command_name))
|
|
|
|
self.location = coin.SoLocation2Event.getClassTypeId()
|
|
self.mouse_event = coin.SoMouseButtonEvent.getClassTypeId()
|
|
self.view = Draft.get3DView()
|
|
self.callback_move = \
|
|
self.view.addEventCallbackPivy(self.location, self.move)
|
|
self.callback_click = \
|
|
self.view.addEventCallbackPivy(self.mouse_event, self.click)
|
|
|
|
self.ui = task_polararray.TaskPanelPolarArray()
|
|
# The calling class (this one) is saved in the object
|
|
# of the interface, to be able to call a function from within it.
|
|
self.ui.source_command = self
|
|
# Gui.Control.showDialog(self.ui)
|
|
todo.ToDo.delay(Gui.Control.showDialog, self.ui)
|
|
|
|
def move(self, event_cb):
|
|
"""Execute as a callback when the pointer moves in the 3D view.
|
|
|
|
It should automatically update the coordinates in the widgets
|
|
of the task panel.
|
|
"""
|
|
event = event_cb.getEvent()
|
|
mousepos = event.getPosition().getValue()
|
|
ctrl = event.wasCtrlDown()
|
|
self.point = Gui.Snapper.snap(mousepos, active=ctrl)
|
|
if self.ui:
|
|
self.ui.display_point(self.point)
|
|
|
|
def click(self, event_cb=None):
|
|
"""Execute as a callback when the pointer clicks on the 3D view.
|
|
|
|
It should act as if the Enter key was pressed, or the OK button
|
|
was pressed in the task panel.
|
|
"""
|
|
if event_cb:
|
|
event = event_cb.getEvent()
|
|
if (event.getState() != coin.SoMouseButtonEvent.DOWN
|
|
or event.getButton() != coin.SoMouseButtonEvent.BUTTON1):
|
|
return
|
|
if self.ui and self.point:
|
|
# The accept function of the interface
|
|
# should call the completed function
|
|
# of the calling class (this one).
|
|
self.ui.accept()
|
|
|
|
def completed(self):
|
|
"""Execute when the command is terminated.
|
|
|
|
We should remove the callbacks that were added to the 3D view
|
|
and then close the task panel.
|
|
"""
|
|
try:
|
|
self.view.removeEventCallbackPivy(self.location, self.callback_move)
|
|
self.view.removeEventCallbackPivy(self.mouse_event, self.callback_click)
|
|
gui_utils.end_all_events()
|
|
except RuntimeError:
|
|
# the view has been deleted already
|
|
pass
|
|
self.callback_move = None
|
|
self.callback_click = None
|
|
if Gui.Control.activeDialog():
|
|
Gui.Control.closeDialog()
|
|
self.finish()
|
|
|
|
|
|
Gui.addCommand('Draft_PolarArray', PolarArray())
|
|
|
|
## @}
|