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 <vtkSlicerMarkupsWidget.h>
#include <vtkSmartPointer.h>

// MRML includes
#include <vtkMRMLSliceNode.h>

// STL includes
#include <set>

class vtkMRMLMarkupsDisplayableManager;
class vtkMRMLMarkupsDisplayNode;
class vtkMRMLInteractionNode;

class VTK_SLICER_MARKUPS_MODULE_MRMLDISPLAYABLEMANAGER_EXPORT vtkMRMLMarkupsDisplayableManagerHelper :
    public vtkObject
{
public:

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

  vtkGetObjectMacro(DisplayableManager, vtkMRMLMarkupsDisplayableManager);
  void SetDisplayableManager(vtkMRMLMarkupsDisplayableManager*);

  //void SetAllWidgetsToManipulate();

  vtkSlicerMarkupsWidget * GetWidget(vtkMRMLMarkupsDisplayNode * markupsDisplayNode);
  vtkSlicerMarkupsWidget * GetWidget(vtkMRMLMarkupsNode * markupsNode);

  void RemoveAllWidgetsAndNodes();

  typedef std::map < vtkSmartPointer<vtkMRMLMarkupsDisplayNode>, vtkSlicerMarkupsWidget* > DisplayNodeToWidgetType;
  typedef std::map < vtkSmartPointer<vtkMRMLMarkupsDisplayNode>, vtkSlicerMarkupsWidget* >::iterator DisplayNodeToWidgetIt;
  DisplayNodeToWidgetType MarkupsDisplayNodesToWidgets;  // display nodes with widgets assigned

  typedef std::set < vtkSmartPointer<vtkMRMLMarkupsNode> > MarkupsNodesType;
  typedef std::set < vtkSmartPointer<vtkMRMLMarkupsNode> >::iterator MarkupsNodesIt;
  MarkupsNodesType MarkupsNodes; // observed markups nodes

  void AddMarkupsNode(vtkMRMLMarkupsNode* node);
  void RemoveMarkupsNode(vtkMRMLMarkupsNode* node);
  void AddDisplayNode(vtkMRMLMarkupsDisplayNode* displayNode);
  void RemoveDisplayNode(vtkMRMLMarkupsDisplayNode* displayNode);

  void DeleteWidget(vtkSlicerMarkupsWidget* widget);

  void AddObservations(vtkMRMLMarkupsNode* node);
  void RemoveObservations(vtkMRMLMarkupsNode* node);

protected:

  vtkMRMLMarkupsDisplayableManagerHelper();
  ~vtkMRMLMarkupsDisplayableManagerHelper() override;

private:

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

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

  bool AddingMarkupsNode;

  std::vector<unsigned long> ObservedMarkupNodeEvents;

  vtkMRMLMarkupsDisplayableManager* DisplayableManager;
};

#endif /* VTKMRMLMARKUPSDISPLAYABLEMANAGERHELPER_H_ */