niftkBaseSegmentorController.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 niftkBaseSegmentorController_h
#define niftkBaseSegmentorController_h

#include <niftkMIDASGuiExports.h>

#include <QColor>
#include <QList>

#include <mitkDataNode.h>
#include <mitkToolManager.h>

#include <niftkBaseController.h>
#include <niftkStateMachineEventFilter.h>
#include <niftkImageOrientationUtils.h>

class QWidget;

namespace niftk
{

class BaseSegmentorGUI;


class NIFTKMIDASGUI_EXPORT BaseSegmentorController : public BaseController, public StateMachineEventFilter
{

  Q_OBJECT

public:

  BaseSegmentorController(IBaseView* view);

  virtual ~BaseSegmentorController();

  virtual void SetupGUI(QWidget* parent) override;

  mitk::ToolManager* GetToolManager() const;

  template <class ToolType>
  ToolType* GetToolByType();

  virtual bool EventFilter(const mitk::StateEvent* stateEvent) const override;

  virtual bool EventFilter(mitk::InteractionEvent* event) const override;

  const QColor& GetDefaultSegmentationColour() const;

  void SetDefaultSegmentationColour(const QColor& defaultSegmentationColour);

  virtual void OnViewGetsActivated() override;

protected:

  virtual void OnActiveToolChanged();

  std::vector<mitk::DataNode*> GetWorkingData();

  mitk::Image* GetWorkingImage(int index);

  mitk::DataNode* GetReferenceNode();

  mitk::Image* GetReferenceImage();

  mitk::DataNode* FindReferenceNodeFromSegmentationNode(const mitk::DataNode::Pointer segmentationNode);

  void SetReferenceImageSelected();

  virtual bool IsAReferenceImage(const mitk::DataNode::Pointer node);

  virtual bool IsASegmentationImage(const mitk::DataNode::Pointer node);

  virtual bool IsAWorkingImage(const mitk::DataNode::Pointer node);

  virtual std::vector<mitk::DataNode*> GetWorkingDataFromSegmentationNode(const mitk::DataNode::Pointer node);

  virtual mitk::DataNode* GetSegmentationNodeFromWorkingData(const mitk::DataNode::Pointer node);

  virtual bool CanStartSegmentationForBinaryNode(const mitk::DataNode::Pointer node) = 0;

  virtual void ApplyDisplayOptions(mitk::DataNode* node);

  int GetReferenceImageSliceAxis();

  int GetReferenceImageSliceAxis(ImageOrientation orientation);

  int GetReferenceImageSliceIndex();

  int GetReferenceImageSliceUpDirection();

  virtual mitk::DataNode* CreateNewSegmentation();

  BaseSegmentorGUI* GetSegmentorGUI() const;

  bool HasInitialisedWorkingData();

  virtual void OnDataManagerSelectionChanged(const QList<mitk::DataNode::Pointer>& nodes);

protected slots:

  virtual void OnNewSegmentationButtonClicked() = 0;

private:

  virtual void SetToolManagerSelection(const mitk::DataNode* referenceData, const std::vector<mitk::DataNode*>& workingDataNodes);

  mitk::ToolManager::Pointer m_ToolManager;

  BaseSegmentorGUI* m_SegmentorGUI;

  QColor m_DefaultSegmentationColour;

  int m_ActiveToolID;

  bool m_CursorIsVisibleWhenToolsAreOff;

friend class BaseSegmentorView;

};


//-----------------------------------------------------------------------------
template <class ToolType>
ToolType* BaseSegmentorController::GetToolByType()
{
  int toolId = m_ToolManager->GetToolIdByToolType<ToolType>();
  return dynamic_cast<ToolType*>(m_ToolManager->GetToolById(toolId));
}

}

#endif