Make bone default annotation and persistent

See Issue #13729
vovythevov · Nov 23, 2012 · af462ab · af462ab
1 parent a689f0b
commit af462ab
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/Modules/Loadable/Armatures/qSlicerArmaturesModuleWidget.cxx b/Modules/Loadable/Armatures/qSlicerArmaturesModuleWidget.cxx
@@ -685,6 +685,32 @@ ::mrmlArmatureDisplayNode()
 }
 */
 
+
+//-----------------------------------------------------------------------------
+void qSlicerArmaturesModuleWidget::setMRMLScene(vtkMRMLScene* scene)
+{
+  this->Superclass::setMRMLScene(scene);
+
+  if (!this->mrmlScene())
+    {
+    return;
+    }
+
+  vtkMRMLSelectionNode* selectionNode = vtkMRMLSelectionNode::SafeDownCast(
+    this->mrmlScene()->GetNodeByID("vtkMRMLSelectionNodeSingleton"));
+  vtkMRMLInteractionNode* interactionNode =
+    vtkMRMLInteractionNode::SafeDownCast(
+      this->mrmlScene()->GetNodeByID("vtkMRMLInteractionNodeSingleton"));
+  if (!selectionNode || !interactionNode)
+    {
+    qCritical() << "Invalid scene, no interaction or selection node";
+    return;
+    }
+
+  selectionNode->SetReferenceDefaultAnnotationID("vtkMRMLBoneNode");
+  interactionNode->SetPlaceModePersistence(1);
+}
+
 //-----------------------------------------------------------------------------
 void qSlicerArmaturesModuleWidget
 ::setMRMLBoneNode(vtkMRMLBoneNode* boneNode)

diff --git a/Modules/Loadable/Armatures/qSlicerArmaturesModuleWidget.h b/Modules/Loadable/Armatures/qSlicerArmaturesModuleWidget.h
@@ -58,6 +58,10 @@ class Q_SLICER_QTMODULES_ARMATURES_EXPORT qSlicerArmaturesModuleWidget
   /// Return the display node of the current armature node if any, 0 otherwise.
   /// \sa mrmlArmatureNode()
   //vtkMRMLArmatureDisplayNode* mrmlArmatureDisplayNode()const;
+
+  /// Set the current MRML scene to the widget
+  virtual void setMRMLScene(vtkMRMLScene*);
+
 public slots:
   /// Set \a armatureNode as current.
   /// \sa setMRMLArmatureNode(vtkMRMLNode*)