vtkImageGrowCutSegment.h

Go to the documentation of this file.

#ifndef vtkImageGrowCutSegment_h
#define vtkImageGrowCutSegment_h
 
#include "vtkSlicerSegmentationsModuleLogicExport.h"
 
#include <vtkImageAlgorithm.h>
#include <vtkImageData.h>
#include <vtkInformation.h>
 
class VTK_SLICER_SEGMENTATIONS_LOGIC_EXPORT vtkImageGrowCutSegment : public vtkImageAlgorithm
{
public:
  static vtkImageGrowCutSegment* New();
  vtkTypeMacro(vtkImageGrowCutSegment, vtkImageAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  void SetIntensityVolume(vtkImageData* grayscaleImage) { this->SetInputData(0, grayscaleImage); }

  void SetSeedLabelVolume(vtkImageData* labelImage) { this->SetInputData(1, labelImage); }

  void SetMaskVolume(vtkImageData* labelImage) { this->SetInputData(2, labelImage); }

  void Reset();

  vtkGetMacro(DistancePenalty, double);
  vtkSetMacro(DistancePenalty, double);
 
protected:
  vtkImageGrowCutSegment();
  ~vtkImageGrowCutSegment() override;
 
  void ExecuteDataWithInformation(vtkDataObject* outData, vtkInformation* outInfo) override;
  int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  int FillInputPortInformation(int port, vtkInformation* info) override;
 
private:
  vtkImageGrowCutSegment(const vtkImageGrowCutSegment&) = delete;
  void operator=(const vtkImageGrowCutSegment&) = delete;
 
  class vtkInternal;
  vtkInternal* Internal;
  double DistancePenalty;
};
 
#endif