vtkImageGrowCutSegment.h

Go to the documentation of this file.

#ifndef FASTGROWCUT_H
#define FASTGROWCUT_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