vtkMRMLMarkupsDisplayableManagerHelper.h

Go to the documentation of this file.

/*==============================================================================

  Program: 3D Slicer

  Portions (c) Copyright Brigham and Women's Hospital (BWH) All Rights Reserved.

  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.

==============================================================================*/


#ifndef VTKMRMLMARKUPSDISPLAYABLEMANAGERHELPER_H_
#define VTKMRMLMARKUPSDISPLAYABLEMANAGERHELPER_H_

// MarkupsModule includes
#include "vtkSlicerMarkupsModuleMRMLDisplayableManagerExport.h"

// MarkupsModule/MRML includes
#include <vtkMRMLMarkupsNode.h>

// VTK includes
#include <vtkAbstractWidget.h>
#include <vtkHandleWidget.h>
#include <vtkSeedWidget.h>
#include <vtkSmartPointer.h>

// MRML includes
#include <vtkMRMLSliceNode.h>
#include <vtkMRMLInteractionNode.h>
class vtkMRMLMarkupsDisplayNode;

class VTK_SLICER_MARKUPS_MODULE_MRMLDISPLAYABLEMANAGER_EXPORT vtkMRMLMarkupsDisplayableManagerHelper :
    public vtkObject
{
public:

  static vtkMRMLMarkupsDisplayableManagerHelper *New();
  vtkTypeMacro(vtkMRMLMarkupsDisplayableManagerHelper, vtkObject);
  void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  void UpdateLockedAllWidgetsFromNodes();
  void UpdateLockedAllWidgetsFromInteractionNode(vtkMRMLInteractionNode* interactionNode);
  void UpdateLockedAllWidgets(bool locked);
  void UpdateLocked(vtkMRMLMarkupsNode* node, vtkMRMLInteractionNode *interactionNode = NULL);

  void RecordWidgetForNode(vtkAbstractWidget* widget, vtkMRMLMarkupsNode *node);

  vtkAbstractWidget * GetWidget(vtkMRMLMarkupsNode * node);
  vtkAbstractWidget * GetIntersectionWidget(vtkMRMLMarkupsNode * node);
  vtkAbstractWidget * GetPointProjectionWidget(std::string uniqueFiducialID);

  void RemoveAllWidgetsAndNodes();
  void RemoveWidgetAndNode(vtkMRMLMarkupsNode *node);


  vtkMRMLMarkupsNode * GetMarkupsNodeFromDisplayNode(vtkMRMLMarkupsDisplayNode *displayNode);

  std::vector<vtkMRMLMarkupsNode*> MarkupsNodeList;

  typedef std::vector<vtkMRMLMarkupsNode*>::iterator MarkupsNodeListIt;

  std::map<vtkMRMLMarkupsNode*, vtkAbstractWidget*> Widgets;

  typedef std::map<vtkMRMLMarkupsNode*, vtkAbstractWidget*>::iterator WidgetsIt;

  std::map<vtkMRMLMarkupsNode*, vtkAbstractWidget*> WidgetIntersections;

  typedef std::map<vtkMRMLMarkupsNode*, vtkAbstractWidget*>::iterator WidgetIntersectionsIt;

  std::map<std::string, vtkAbstractWidget*> WidgetPointProjections;

  typedef std::map<std::string, vtkAbstractWidget*>::iterator WidgetPointProjectionsIt;

  //
  // End of The Lists!!
  //
  //----------------------------------------------------------------------------------


  void PlaceSeed(double x, double y, vtkRenderWindowInteractor * interactor, vtkRenderer * renderer);

  vtkHandleWidget * GetSeed(int index);

  void RemoveSeeds();

  int GetNodeGlyphType(vtkMRMLNode *displayNode, int index);
  void SetNodeGlyphType(vtkMRMLNode *displayNode, int glyphType, int index);
  void RemoveNodeGlyphType(vtkMRMLNode *displayNode);
  void ClearNodeGlyphTypes();

protected:

  vtkMRMLMarkupsDisplayableManagerHelper();
  virtual ~vtkMRMLMarkupsDisplayableManagerHelper();

  void PrintNodeGlyphTypes();

private:

  vtkMRMLMarkupsDisplayableManagerHelper(const vtkMRMLMarkupsDisplayableManagerHelper&); 
  void operator=(const vtkMRMLMarkupsDisplayableManagerHelper&); 

  vtkSmartPointer<vtkSeedWidget> SeedWidget;
  std::vector<vtkSmartPointer<vtkHandleWidget> > HandleWidgetList;
  typedef std::vector<vtkSmartPointer<vtkHandleWidget> >::iterator HandleWidgetListIt;

  std::map<vtkMRMLNode*, std::vector<int> > NodeGlyphTypes;
};

#endif /* VTKMRMLMARKUPSDISPLAYABLEMANAGERHELPER_H_ */