You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

101 lines
2.6 KiB
C++

/*=========================================================================
Program: Visualization Toolkit
Module: vtkImageIslandRemoval2D.h
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
/**
* @class vtkImageIslandRemoval2D
* @brief Removes small clusters in masks.
*
* vtkImageIslandRemoval2D computes the area of separate islands in
* a mask image. It removes any island that has less than AreaThreshold
* pixels. Output has the same ScalarType as input. It generates
* the whole 2D output image for any output request.
*/
#ifndef vtkImageIslandRemoval2D_h
#define vtkImageIslandRemoval2D_h
#include "vtkImageAlgorithm.h"
#include "vtkImagingMorphologicalModule.h" // For export macro
typedef struct
{
void* inPtr;
void* outPtr;
int idx0;
int idx1;
} vtkImage2DIslandPixel;
class VTKIMAGINGMORPHOLOGICAL_EXPORT vtkImageIslandRemoval2D : public vtkImageAlgorithm
{
public:
//@{
/**
* Constructor: Sets default filter to be identity.
*/
static vtkImageIslandRemoval2D* New();
vtkTypeMacro(vtkImageIslandRemoval2D, vtkImageAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent) override;
//@}
//@{
/**
* Set/Get the cutoff area for removal
*/
vtkSetMacro(AreaThreshold, int);
vtkGetMacro(AreaThreshold, int);
//@}
//@{
/**
* Set/Get whether to use 4 or 8 neighbors
*/
vtkSetMacro(SquareNeighborhood, vtkTypeBool);
vtkGetMacro(SquareNeighborhood, vtkTypeBool);
vtkBooleanMacro(SquareNeighborhood, vtkTypeBool);
//@}
//@{
/**
* Set/Get the value to remove.
*/
vtkSetMacro(IslandValue, double);
vtkGetMacro(IslandValue, double);
//@}
//@{
/**
* Set/Get the value to put in the place of removed pixels.
*/
vtkSetMacro(ReplaceValue, double);
vtkGetMacro(ReplaceValue, double);
//@}
protected:
vtkImageIslandRemoval2D();
~vtkImageIslandRemoval2D() override {}
int AreaThreshold;
vtkTypeBool SquareNeighborhood;
double IslandValue;
double ReplaceValue;
int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
private:
vtkImageIslandRemoval2D(const vtkImageIslandRemoval2D&) = delete;
void operator=(const vtkImageIslandRemoval2D&) = delete;
};
#endif