def _createCurve(self):
curve = CurveModel(self._model)
curve_id = self._model.getNextCurveIdentifier()
self._model.insertCurve(curve_id, curve)
self._setCurveIdentifier(curve_id)
return curve
def redo(self):
region = self._model.getRegion()
scene = region.getScene()
scene.beginChange()
selection_node_ids = []
for curve_identifier in self._curves:
node_statuses = self._node_statuses[curve_identifier]
node_ids = self._model.createNodes(
node_statuses, group=self._model.getCurveGroup())
_updateNodeIdentifiers(node_statuses, node_ids)
curve = CurveModel(self._model)
next_curve_identifier = self._model.getNextCurveIdentifier()
self._model.insertCurve(next_curve_identifier, curve)
curve.setInterpolationCount(
self._interpolation_counts[curve_identifier])
curve.setNodes(node_ids)
self._curves[curve_identifier] = next_curve_identifier
if len(curve) > 1:
locations = curve.calculate()
self._scene.setInterpolationPoints(next_curve_identifier,
locations)
selection_node_ids += node_ids
self._model.setSelection(selection_node_ids)
self._set_rotation_point_method(self._rotation_point_end)
self._set_normal_method(self._normal)
scene.endChange()
def deserialize(self, str_rep):
scene = self._region.getScene()
scene.beginChange()
master_nodeset = self._point_cloud_group.getMasterNodeset(
) # removeAllNodes()
master_nodeset.destroyAllNodes()
master_nodeset = self._interpolation_point_group.getMasterNodeset()
master_nodeset.destroyAllNodes()
self.setSelection([])
d = json.loads(str_rep)
self._deserializeNodeset(self._point_cloud_group, d['_basic_points'])
del d['_basic_points']
self._deserializeNodeset(self._curve_group, d['_curve_points'])
del d['_curve_points']
self._plane.deserialize(json.dumps(d['_plane']))
del d['_plane']
self._curves = {}
curves = d['_curves']
for curve_index in curves:
c = CurveModel(self)
c.deserialize(json.dumps(curves[curve_index]))
self.insertCurve(int(curve_index), c)
del d['_curves']
self._plane_attitude_store = []
plane_attitude_store = d['_plane_attitude_store']
for plane_attitude in plane_attitude_store:
p = PlaneAttitude(None, None)
p.deserialize(json.dumps(plane_attitude))
self._plane_attitude_store.append(p)
del d['_plane_attitude_store']
selection = d['_selection']
selection_field = scene.getSelectionField()
if not selection_field.isValid():
scene.setSelectionField(self._selection_group_field)
del d['_selection']
self.__dict__.update(d)
self.setSelection(selection)
scene.endChange()
def deserialize(self, str_rep):
scene = self._region.getScene()
scene.beginChange()
master_nodeset = self._point_cloud_group.getMasterNodeset() # removeAllNodes()
master_nodeset.destroyAllNodes()
master_nodeset = self._interpolation_point_group.getMasterNodeset()
master_nodeset.destroyAllNodes()
self.setSelection([])
d = json.loads(str_rep)
self._deserializeNodeset(self._point_cloud_group, d['_basic_points'])
del d['_basic_points']
self._deserializeNodeset(self._curve_group, d['_curve_points'])
del d['_curve_points']
self._plane.deserialize(json.dumps(d['_plane']))
del d['_plane']
self._curves = {}
curves = d['_curves']
for curve_index in curves:
c = CurveModel(self)
c.deserialize(json.dumps(curves[curve_index]))
self.insertCurve(int(curve_index), c)
del d['_curves']
self._plane_attitude_store = []
plane_attitude_store = d['_plane_attitude_store']
for plane_attitude in plane_attitude_store:
p = PlaneAttitude(None, None)
p.deserialize(json.dumps(plane_attitude))
self._plane_attitude_store.append(p)
del d['_plane_attitude_store']
selection = d['_selection']
selection_field = scene.getSelectionField()
if not selection_field.isValid():
scene.setSelectionField(self._selection_group_field)
del d['_selection']
self.__dict__.update(d)
self.setSelection(selection)
scene.endChange()
def redo(self):
region = self._model.getRegion()
scene = region.getScene()
scene.beginChange()
selection_node_ids = []
for curve_identifier in self._curves:
node_statuses = self._node_statuses[curve_identifier]
node_ids = self._model.createNodes(node_statuses, group=self._model.getCurveGroup())
_updateNodeIdentifiers(node_statuses, node_ids)
curve = CurveModel(self._model)
next_curve_identifier = self._model.getNextCurveIdentifier()
self._model.insertCurve(next_curve_identifier, curve)
curve.setInterpolationCount(self._interpolation_counts[curve_identifier])
curve.setNodes(node_ids)
self._curves[curve_identifier] = next_curve_identifier
if len(curve) > 1:
locations = curve.calculate()
self._scene.setInterpolationPoints(next_curve_identifier, locations)
selection_node_ids += node_ids
self._model.setSelection(selection_node_ids)
self._set_rotation_point_method(self._rotation_point_end)
self._set_normal_method(self._normal)
scene.endChange()
def undo(self):
region = self._model.getRegion()
scene = region.getScene()
scene.beginChange()
for curve_identifier in self._curves:
if self._curves[curve_identifier] is None:
curve = CurveModel(self._model)
self._model.insertCurve(curve_identifier, curve)
curve.setInterpolationCount(self._interpolation_counts[curve_identifier])
node_ids = self._model.createNodes(self._node_statuses[curve_identifier], group=self._model.getCurveGroup())
curve.setNodes(node_ids)
self._curves[curve_identifier] = curve
if len(curve) > 1:
locations = curve.calculate()
self._scene.setInterpolationPoints(curve_identifier, locations)
self._model.setSelection(self._selected)
scene.endChange()
def undo(self):
region = self._model.getRegion()
scene = region.getScene()
scene.beginChange()
for curve_identifier in self._curves:
if self._curves[curve_identifier] is None:
curve = CurveModel(self._model)
self._model.insertCurve(curve_identifier, curve)
curve.setInterpolationCount(
self._interpolation_counts[curve_identifier])
node_ids = self._model.createNodes(
self._node_statuses[curve_identifier],
group=self._model.getCurveGroup())
curve.setNodes(node_ids)
self._curves[curve_identifier] = curve
if len(curve) > 1:
locations = curve.calculate()
self._scene.setInterpolationPoints(curve_identifier,
locations)
self._model.setSelection(self._selected)
scene.endChange()
def mousePressEvent(self, event):
if self._active_button != QtCore.Qt.NoButton:
return
self._active_button = event.button()
self._start_plane_attitude = None
self._finshing_curve = False
self._adding_to_curve = False
self._modifying_curve = False
if self._node_status:
if (event.modifiers() & QtCore.Qt.CTRL) and event.button() == QtCore.Qt.RightButton:
node_id = self._node_status.getNodeIdentifier()
self._active_curve.removeNode(node_id)
curve_index = self._model.getCurveIdentifier(self._active_curve)
if len(self._active_curve) > 1:
locations = self._active_curve.calculate()
self._scene.setInterpolationPoints(curve_index, locations)
else:
self._scene.clearInterpolationPoints(curve_index)
self._model.removeNode(node_id)
self._node_status = None
self._zinc_view.setMouseTracking(False)
self._finshing_curve = True
elif (event.modifiers() & QtCore.Qt.CTRL) and event.button() == QtCore.Qt.LeftButton:
node = self._zinc_view.getNearestNode(event.x(), event.y())
if node and node.isValid():
self._closing_curve_node_id = self._node_status.getNodeIdentifier()
self._node_status.setNodeIdentifier(node.getIdentifier())
self._adding_to_curve = True
else:
super(Curve, self).mousePressEvent(event)
elif (event.modifiers() & QtCore.Qt.CTRL) and event.button() == QtCore.Qt.LeftButton:
# The start of a new curve
self._active_curve = None
node = self._zinc_view.getNearestNode(event.x(), event.y())
if node and node.isValid():
# node exists at this location so select it
group = self._model.getSelectionGroup()
group.removeAllNodes()
group.addNode(node)
node_id = node.getIdentifier()
self._active_curve = self._model.getCurveForNode(node_id)
node_location = self._model.getNodeLocation(node)
plane_attitude = self._model.getNodePlaneAttitude(node_id)
self._start_plane_attitude = self._plane.getAttitude()
self._modifying_curve = True
else:
# The start of a new curve
self._active_curve = CurveModel(self._model)
self._model.insertCurve(self._model.getNextCurveIdentifier(), self._active_curve)
self._active_curve.setInterpolationCount(self._interpolation_count)
node_location = None
plane_attitude = None
point_on_plane = self._calculatePointOnPlane(event.x(), event.y())
region = self._model.getRegion()
fieldmodule = region.getFieldmodule()
fieldmodule.beginChange()
node = self._model.createNode()
self._model.setNodeLocation(node, point_on_plane)
group = self._model.getCurveGroup()
group.addNode(node)
node_id = node.getIdentifier()
fieldmodule.endChange()
self._adding_to_curve = True
self._active_curve.addNode(node_id)
self._node_status = ControlPointStatus(node_id, node_location, plane_attitude)
curve_index = self._model.getCurveIdentifier(self._active_curve)
self._node_status.setCurveIdentifier(curve_index)
elif self._node_status is None:
super(Curve, self).mousePressEvent(event)
class Curve(AbstractSelection):
def __init__(self, plane, undo_redo_stack):
super(Curve, self).__init__(plane, undo_redo_stack)
self._mode_type = ViewMode.SEGMENT_CURVE
self._model = None
self._scene = None
self._node_status = None
self._active_curve = None
self._interpolation_count = DEFAULT_INTERPOLATION_COUNT
def setModel(self, model):
self._model = model
def setScene(self, scene):
self._scene = scene
def setInterpolationCount(self, count):
self._interpolation_count = count
def enter(self):
super(Curve, self).enter()
def leave(self):
super(Curve, self).leave()
def mousePressEvent(self, event):
if self._active_button != QtCore.Qt.NoButton:
return
self._active_button = event.button()
self._start_plane_attitude = None
self._finshing_curve = False
self._adding_to_curve = False
self._modifying_curve = False
if self._node_status:
if (event.modifiers() & QtCore.Qt.CTRL) and event.button() == QtCore.Qt.RightButton:
node_id = self._node_status.getNodeIdentifier()
self._active_curve.removeNode(node_id)
curve_index = self._model.getCurveIdentifier(self._active_curve)
if len(self._active_curve) > 1:
locations = self._active_curve.calculate()
self._scene.setInterpolationPoints(curve_index, locations)
else:
self._scene.clearInterpolationPoints(curve_index)
self._model.removeNode(node_id)
self._node_status = None
self._zinc_view.setMouseTracking(False)
self._finshing_curve = True
elif (event.modifiers() & QtCore.Qt.CTRL) and event.button() == QtCore.Qt.LeftButton:
node = self._zinc_view.getNearestNode(event.x(), event.y())
if node and node.isValid():
self._closing_curve_node_id = self._node_status.getNodeIdentifier()
self._node_status.setNodeIdentifier(node.getIdentifier())
self._adding_to_curve = True
else:
super(Curve, self).mousePressEvent(event)
elif (event.modifiers() & QtCore.Qt.CTRL) and event.button() == QtCore.Qt.LeftButton:
# The start of a new curve
self._active_curve = None
node = self._zinc_view.getNearestNode(event.x(), event.y())
if node and node.isValid():
# node exists at this location so select it
group = self._model.getSelectionGroup()
group.removeAllNodes()
group.addNode(node)
node_id = node.getIdentifier()
self._active_curve = self._model.getCurveForNode(node_id)
node_location = self._model.getNodeLocation(node)
plane_attitude = self._model.getNodePlaneAttitude(node_id)
self._start_plane_attitude = self._plane.getAttitude()
self._modifying_curve = True
else:
# The start of a new curve
self._active_curve = CurveModel(self._model)
self._model.insertCurve(self._model.getNextCurveIdentifier(), self._active_curve)
self._active_curve.setInterpolationCount(self._interpolation_count)
node_location = None
plane_attitude = None
point_on_plane = self._calculatePointOnPlane(event.x(), event.y())
region = self._model.getRegion()
fieldmodule = region.getFieldmodule()
fieldmodule.beginChange()
node = self._model.createNode()
self._model.setNodeLocation(node, point_on_plane)
group = self._model.getCurveGroup()
group.addNode(node)
node_id = node.getIdentifier()
fieldmodule.endChange()
self._adding_to_curve = True
self._active_curve.addNode(node_id)
self._node_status = ControlPointStatus(node_id, node_location, plane_attitude)
curve_index = self._model.getCurveIdentifier(self._active_curve)
self._node_status.setCurveIdentifier(curve_index)
elif self._node_status is None:
super(Curve, self).mousePressEvent(event)
def mouseMoveEvent(self, event):
if self._node_status is not None:
self._start_plane_attitude = None
node = self._model.getNodeByIdentifier(self._node_status.getNodeIdentifier())
point_on_plane = self._calculatePointOnPlane(event.x(), event.y())
self._model.setNodeLocation(node, point_on_plane)
curve_index = self._model.getCurveIdentifier(self._active_curve)
if len(self._active_curve) > 1:
locations = self._active_curve.calculate()
self._scene.setInterpolationPoints(curve_index, locations)
if self._modifying_curve:
pass
elif not self._adding_to_curve or not self._finshing_curve:
super(Curve, self).mouseMoveEvent(event)
else:
super(Curve, self).mouseMoveEvent(event)
def mouseReleaseEvent(self, event):
if self._active_button != event.button():
return
if self._start_plane_attitude is not None and self._node_status is not None:
group = self._model.getSelectionGroup()
group.removeAllNodes()
target_plane_attitude = self._node_status.getPlaneAttitude()
if target_plane_attitude != self._start_plane_attitude:
c = CommandMovePlane(self._plane, self._start_plane_attitude, target_plane_attitude)
self._undo_redo_stack.push(c)
self._node_status = None
elif self._node_status is not None and self._adding_to_curve:
node_id = self._node_status.getNodeIdentifier()
node1 = self._model.getNodeByIdentifier(node_id)
group = self._model.getSelectionGroup()
group.removeNode(node1)
node_location = self._model.getNodeLocation(node1)
plane_attitude = self._plane.getAttitude()
node_status = ControlPointStatus(node_id, node_location, plane_attitude)
c = CommandCurveNode(self._model, self._node_status, node_status)
c.setScene(self._scene)
if self._active_curve.closes(node_id):
node2 = self._model.getNodeByIdentifier(self._closing_curve_node_id)
else:
node2 = self._model.createNode()
self._model.setNodeLocation(node2, node_location)
group = self._model.getCurveGroup()
group.addNode(node2)
node_id = node2.getIdentifier()
curve_index = self._model.getCurveIdentifier(self._active_curve)
self._scene.clearInterpolationPoints(curve_index)
self._active_curve.addNode(node_id)
node_status.setCurveIdentifier(curve_index)
self._node_status.setCurveIdentifier(curve_index)
self._undo_redo_stack.push(c)
locations = self._active_curve.calculate()
self._scene.setInterpolationPoints(curve_index, locations)
self._node_status = ControlPointStatus(node_id, None, None)
self._node_status.setCurveIdentifier(curve_index)
self._zinc_view.setMouseTracking(True)
elif self._finshing_curve:
self._active_curve = None
elif self._modifying_curve:
node_id = self._node_status.getNodeIdentifier()
node = self._model.getNodeByIdentifier(node_id)
node_location = self._model.getNodeLocation(node)
plane_attitude = self._plane.getAttitude()
node_status = ControlPointStatus(node_id, node_location, plane_attitude)
curve_index = self._model.getCurveIdentifier(self._active_curve)
node_status.setCurveIdentifier(curve_index)
c = CommandCurveNode(self._model, self._node_status, node_status)
c.setScene(self._scene)
group = self._model.getSelectionGroup()
group.removeNode(node)
self._node_status = None
self._undo_redo_stack.push(c)
elif not self._adding_to_curve:
super(Curve, self).mouseReleaseEvent(event)
self._active_button = QtCore.Qt.NoButton
def _calculatePointOnPlane(self, x, y):
far_plane_point = self._zinc_view.unproject(x, -y, -1.0)
near_plane_point = self._zinc_view.unproject(x, -y, 1.0)
point_on_plane = calculateLinePlaneIntersection(near_plane_point, far_plane_point, self._plane.getRotationPoint(), self._plane.getNormal())
return point_on_plane