niftkMorphologicalSegmentorPipelineManager.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 niftkMorphologicalSegmentorPipelineManager_h
#define niftkMorphologicalSegmentorPipelineManager_h

#include "niftkMIDASExports.h"

#include <itkLightObject.h>

#include <mitkDataStorage.h>
#include <mitkImage.h>
#include <mitkToolManager.h>

#include <MorphologicalSegmentorPipeline.h>
#include <MorphologicalSegmentorPipelineInterface.h>
#include <MorphologicalSegmentorPipelineParams.h>

namespace niftk
{

class NIFTKMIDAS_EXPORT MorphologicalSegmentorPipelineManager : public itk::LightObject
{

public:

  static const std::string PROPERTY_MIDAS_MORPH_SEGMENTATION_FINISHED;

  static const std::string SEGMENTATION_OF_LAST_STAGE_NAME;


  mitkClassMacroItkParent(MorphologicalSegmentorPipelineManager, itk::Object)
  itkNewMacro(MorphologicalSegmentorPipelineManager)

  void SetDataStorage(mitk::DataStorage::Pointer dataStorage);

  mitk::DataStorage::Pointer GetDataStorage() const;

  void SetToolManager(mitk::ToolManager::Pointer toolManager);

  mitk::ToolManager::Pointer GetToolManager() const;

//  void SetErosionSubtractionsInput(mitk::Image::ConstPointer erosionSubtractions, mitk::Image::Pointer segmentation);
//  void SetDilationSubtractionsInput(mitk::Image::ConstPointer dilationSubtractions, mitk::Image::Pointer segmentation);

  void OnThresholdingValuesChanged(double lowerThreshold, double upperThreshold, int axialSliceNumber);

  void OnErosionsValuesChanged(double upperThreshold, int numberOfErosions);

  void OnDilationsValuesChanged(double lowerPercentage, double upperPercentage, int numberOfDilations);

  void OnRethresholdingValuesChanged(int boxSize);

  void OnTabChanged(int tabIndex);

  bool HasSegmentationNode() const;

  mitk::Image::ConstPointer GetReferenceImage() const;

  mitk::Image::Pointer GetWorkingImage(unsigned int dataIndex) const;

  mitk::DataNode::Pointer GetSegmentationNode() const;

  mitk::Image::Pointer GetSegmentationImage() const;

  void GetPipelineParamsFromSegmentationNode(MorphologicalSegmentorPipelineParams& params) const;

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

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

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

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

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

  void SetSegmentationNodePropsFromReferenceImage();

  void UpdateSegmentation();

  void FinalizeSegmentation();

  void ClearWorkingData();

  void RemoveWorkingData();

  void DestroyPipeline(mitk::Image::Pointer segmentation);

protected:

  MorphologicalSegmentorPipelineManager();
  virtual ~MorphologicalSegmentorPipelineManager();

  MorphologicalSegmentorPipelineManager(const MorphologicalSegmentorPipelineManager&); // Purposefully not implemented.
  MorphologicalSegmentorPipelineManager& operator=(const MorphologicalSegmentorPipelineManager&); // Purposefully not implemented.

private:

  template<typename TPixel, unsigned int VImageDimension>
  void InvokeITKPipeline(
      const itk::Image<TPixel, VImageDimension>* referenceImage,
      const std::vector<typename itk::Image<unsigned char, VImageDimension>::ConstPointer>& workingImages,
      MorphologicalSegmentorPipelineParams& params,
      const std::vector<int>& editingRegion,
      const std::vector<bool>& editingFlags,
      bool isRestarting,
      mitk::Image::Pointer outputImage
      );

  template<typename TPixel, unsigned int VImageDimension>
  void FinalizeITKPipeline(
      const itk::Image<TPixel, VImageDimension>* referenceImage,
      mitk::Image::Pointer segmentation
      );

  template<typename TPixel, unsigned int VImageDimension>
  void ClearITKImage(
      itk::Image<TPixel, VImageDimension>* itkImage
      );

  std::map<mitk::Image::Pointer, MorphologicalSegmentorPipelineInterface*> m_Pipelines;

  mitk::DataStorage::Pointer m_DataStorage;

  mitk::ToolManager::Pointer m_ToolManager;

};

}

#endif