niftkPolyTool.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef niftkPolyTool_h
#define niftkPolyTool_h

#include "niftkMIDASExports.h"

#include <mitkOperation.h>
#include <mitkOperationActor.h>
#include <mitkPointSet.h>

#include "niftkContourTool.h"
#include "niftkPolyToolEventInterface.h"

namespace niftk
{

class NIFTKMIDAS_EXPORT PolyTool : public ContourTool {

public:

  mitkClassMacro(PolyTool, Tool)
  itkNewMacro(PolyTool)

  
  virtual const char* GetName() const;

  virtual const char** GetXPM() const;

  static const std::string MIDAS_POLY_TOOL_ANCHOR_POINTS;

  static const std::string MIDAS_POLY_TOOL_PREVIOUS_CONTOUR;

  virtual void ExecuteOperation(mitk::Operation* operation);

  virtual void Activated();

  virtual void Deactivated();

  virtual bool AddLine(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool SelectPoint(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool MovePoint(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual bool DeselectPoint(mitk::StateMachineAction* action, mitk::InteractionEvent* event);

  virtual void ClearData();

protected:

  PolyTool(); // purposely hidden
  virtual ~PolyTool(); // purposely hidden

  virtual void InitializeStateMachine();

  virtual void ConnectActionsAndFunctions();

private:

  void SetPolyLinePointSetVisible(bool visible);

  void SetPreviousContourVisible(bool visible);

  void Disable3dRenderingOfPreviousContour();

  void DrawWholeContour(const mitk::ContourModel& contourReferencePointsInput, const mitk::PlaneGeometry* planeGeometry, mitk::ContourModel& feedbackContour, mitk::ContourModel& backgroundContour);

  void UpdateContours(mitk::StateMachineAction* action, mitk::InteractionPositionEvent* positionEvent, bool provideUndo, bool registerNewPoint);

  void UpdateFeedbackContour(bool registerNewPoint, const mitk::Point3D& closestCornerPoint, const mitk::PlaneGeometry* planeGeometry, mitk::ContourModel& contourReferencePointsInput, mitk::ContourModel& feedbackContour, mitk::ContourModel& backgroundContour, bool provideUndo);

  mitk::Point3D m_MostRecentPointInMillimetres;

  mitk::ContourModel::Pointer m_ReferencePoints;

  mitk::ContourModel::Pointer m_PreviousContourReferencePoints;

  mitk::ContourModel::Pointer  m_PreviousContour;
  mitk::DataNode::Pointer m_PreviousContourNode;
  bool                    m_PreviousContourVisible;

  mitk::PointSet::Pointer m_PolyLinePointSet;
  mitk::DataNode::Pointer m_PolyLinePointSetNode;
  bool                    m_PolyLinePointSetVisible;

  static const mitk::OperationType MIDAS_POLY_TOOL_OP_ADD_TO_FEEDBACK_CONTOUR;
  static const mitk::OperationType MIDAS_POLY_TOOL_OP_UPDATE_FEEDBACK_CONTOUR;

  PolyToolEventInterface::Pointer m_Interface;

  unsigned int m_DraggedPointIndex;

};

}

#endif