# # FreeCAD macro ViewRotation. # This GUI allows the view to be rotated with more precision than when using # the mouse. Rotation is according to axes fixed with respect to the user and # not the objects, though the aim is that the objects rotate about their # approximate shared centre rather than the view centre. # The GUI defaults to the top right of the screen, this behaviour can be # changed by editing. # [http://forum.freecadweb.org/viewtopic.php?f=3&t=1784&hilit=View+Rotation#p12012 View+Rotation] __Name__ = 'View Rotation' __Comment__ = 'This GUI allows the view to be rotated precisely' __Author__ = 'Joe Dowsett' __Version__ = '1.0.1' __Date__ = '2024-11-22' __License__ = 'CC-BY-3.0' __Web__ = 'https://www.freecadweb.org/wiki/Macro_View_Rotation' __Wiki__ = 'https://www.freecadweb.org/wiki/Macro_View_Rotation' __Icon__ = '' __Help__ = 'Rotation is according to axes fixed with respect to the user.' __Status__ = '' __Requires__ = '' __Files__ = 'ViewRotationOut.png,ViewRotationRight.png,ViewRotationUp.png' from math import pi from functools import partial import os from PySide import QtCore # FreeCAD's PySide! from PySide import QtGui # FreeCAD's PySide! from pivy import coin import FreeCAD as app import FreeCADGui as gui def find_centre(): doc = app.activeDocument() if doc is None: return app.Vector(0.0, 0.0, 0.0) xmax = 0.0 xmin = 0.0 ymax = 0.0 ymin = 0.0 zmax = 0.0 zmin = 0.0 for obj in doc.Objects: try: if obj.TypeId[:4] == 'Mesh': box = obj.Mesh.BoundBox elif obj.TypeId[:6] == 'Points': box = obj.Points.BoundBox elif obj.TypeId[:4] == 'Part': box = obj.Shape.BoundBox else: continue except AttributeError: continue xmax = max(xmax, box.XMax) xmin = min(xmin, box.XMin) ymax = max(ymax, box.YMax) ymin = min(ymin, box.YMin) zmax = max(zmax, box.ZMax) zmin = min(zmin, box.ZMin) return app.Vector((xmax + xmin) / 2, (ymax + ymin) / 2, (zmax + zmin) / 2) class RotateGui(QtGui.QWidget): def __init__(self): super().__init__() self.init_ui() self.init_rotate() def init_ui(self): macro_dir = app.getUserMacroDir(True) self.sliders = [] self.line_edits = [] vbox = QtGui.QVBoxLayout() icons = ( 'ViewRotationRight.png', 'ViewRotationUp.png', 'ViewRotationOut.png', ) for i, icon in enumerate(icons): slider = QtGui.QSlider(QtCore.Qt.Horizontal, self) slider.setFocusPolicy(QtCore.Qt.NoFocus) slider.setSingleStep(5) slider.setPageStep(15) slider.setValue(0) slider.setMaximum(180) slider.setMinimum(-180) slider.valueChanged.connect(partial(self.on_slider_value_changed, i)) self.sliders.append(slider) line_edit = QtGui.QLineEdit(self) line_edit.setText('0') line_edit.setAlignment(QtCore.Qt.AlignRight) line_edit.returnPressed.connect(self.valueEntered) self.line_edits.append(line_edit) label = QtGui.QLabel(self) label.setPixmap(QtGui.QPixmap(os.path.join(macro_dir, icon))) hbox = QtGui.QHBoxLayout() hbox.addWidget(label, 1, QtCore.Qt.AlignCenter) hbox.addWidget(slider, 4) hbox.addWidget(line_edit, 1) vbox.addLayout(hbox) reset_button = QtGui.QPushButton('Reset') reset_button.clicked.connect(self.reset) ok_button = QtGui.QPushButton('OK') ok_button.clicked.connect(self.close) cancel_button = QtGui.QPushButton('Cancel') cancel_button.clicked.connect(self.cancel) hbox = QtGui.QHBoxLayout() hbox.addWidget(reset_button, 1) hbox.addWidget(ok_button, 1) hbox.addWidget(cancel_button, 1) vbox.addStretch(1) vbox.addLayout(hbox) self.setLayout(vbox) qapp = QtCore.QCoreApplication.instance() screen_number = QtGui.QDesktopWidget().screenNumber(self) right = qapp.screens()[screen_number].geometry().width() self.setGeometry(right - 300, 0, 300, 150) self.setWindowTitle('Rotate view') self.show() def init_rotate(self): # Index in [right, up, out] self._current_axis = 0 self.cam = gui.activeDocument().ActiveView.getCameraNode() self.centre = coin.SbVec3f(find_centre()) self.view = self.cam.orientation.getValue() self.pos = self.cam.position.getValue() # Store a copy of the original view to be restored in the case of user # selecting Reset or Cancel. self.original_view = coin.SbRotation(self.view.getValue()) self.original_pos = coin.SbVec3f(self.pos.getValue()) self.config_direction(0) def reset(self): # Reset the view to the original one. self.cam.orientation = self.original_view self.cam.position = self.original_pos for slider in self.sliders: # Deactivate callbacks. slider.blockSignals(True) slider.setValue(0) slider.blockSignals(False) for tbox in self.line_edits: tbox.setText('0') self.config_direction(0) def cancel(self): self.reset() self.close() def config_direction(self, axis_index): """ Assign `self.direction` to either the right, up or out vector. Evaluate the vectors corresponding to the three directions for the current view, and assign `self.direction` to the one at index `axis_index`. """ self.view = self.cam.orientation.getValue() self.view = coin.SbRotation(self.view.getValue()) self.pos = self.cam.position.getValue() self.pos = coin.SbVec3f(self.pos.getValue()) up = coin.SbVec3f(0, 1, 0) self.up = self.view.multVec(up) out = coin.SbVec3f(0, 0, 1) self.out = self.view.multVec(out) u = self.up.getValue() o = self.out.getValue() r = (u[1]*o[2]-u[2]*o[1], u[2]*o[0]-u[0]*o[2], u[0]*o[1]-u[1]*o[0]) self.right = coin.SbVec3f(r) self.direction = [self.right, self.up, self.out][axis_index] def check(self, axis_index): """ Check if the direction of rotation has changed. Check if the direction of rotation has changed, if so then set previous slider and textbox to zero, and setup the new direction. """ if axis_index != self._current_axis: # Deactivate callbacks. self.sliders[self._current_axis].blockSignals(True) self.sliders[self._current_axis].setValue(0) self.sliders[self._current_axis].blockSignals(False) self.line_edits[self._current_axis].blockSignals(True) self.line_edits[self._current_axis].setText("0") self.line_edits[self._current_axis].blockSignals(False) self._current_axis = axis_index self.config_direction(axis_index) def rotate(self, value): # Carry out the desired rotation about self.direction. val = value * pi / 180.0 rot = coin.SbRotation(self.direction, -val) nrot = self.view*rot prot = rot.multVec(self.pos - self.centre) + self.centre self.cam.orientation = nrot self.cam.position = prot def on_slider_value_changed(self, slider_index: int, value: int): """ Respond to the change in value of a slider. Respond to the change in value of a slider, update the corresponding text box, check for a direction change then rotate if the value was changed internally, ignore event. """ self.line_edits[slider_index].setText(str(value)) self.check(slider_index) self.rotate(value) def valueEntered(self): """ Respond to a value being entered in a text box. Respond to a value being entered in a text box, updating the corresponding slider, checking for direction change then rotate. """ sender = self.sender() for i in range(3): if sender == self.line_edits[i]: break value = round(float(self.line_edits[i].text())) self.sliders[i].blockSignals(True) self.sliders[i].setValue(value) self.sliders[i].blockSignals(False) self.line_edits[i].blockSignals(True) self.line_edits[i].setText(str(value)) self.line_edits[i].blockSignals(False) self.check(i) self.rotate(value) if __name__ == '__main__': # We need to set a variable, otherwise, the dialog doesn't appear. rotate = RotateGui()