vtkImageBinaryThreshold.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkImageBinaryThreshold_h
#define vtkImageBinaryThreshold_h

#include "vtkImagingCoreModule.h" // For export macro
#include "vtkThreadedImageAlgorithm.h"

VTK_ABI_NAMESPACE_BEGIN

class VTKIMAGINGCORE_EXPORT vtkImageBinaryThreshold : public vtkThreadedImageAlgorithm
{
public:
  static vtkImageBinaryThreshold* New();
  vtkTypeMacro(vtkImageBinaryThreshold, vtkThreadedImageAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetMacro(ReplaceIn, bool);
  vtkGetMacro(ReplaceIn, bool);
  vtkBooleanMacro(ReplaceIn, bool);


  vtkSetMacro(InValue, double);
  vtkGetMacro(InValue, double);


  vtkSetMacro(ReplaceOut, bool);
  vtkGetMacro(ReplaceOut, bool);
  vtkBooleanMacro(ReplaceOut, bool);


  vtkSetMacro(OutValue, double);
  vtkGetMacro(OutValue, double);


  vtkSetMacro(UpperThreshold, double);
  vtkGetMacro(UpperThreshold, double);


  vtkSetMacro(LowerThreshold, double);
  vtkGetMacro(LowerThreshold, double);

  enum ThresholdFunction
  {
    THRESHOLD_BETWEEN = 0,
    THRESHOLD_LOWER,
    THRESHOLD_UPPER,
  };


  vtkSetMacro(ThresholdFunction, int);
  vtkGetMacro(ThresholdFunction, int);


  vtkSetMacro(OutputScalarType, int);
  vtkGetMacro(OutputScalarType, int);
  void SetOutputScalarTypeToDouble() { this->SetOutputScalarType(VTK_DOUBLE); }
  void SetOutputScalarTypeToFloat() { this->SetOutputScalarType(VTK_FLOAT); }
  void SetOutputScalarTypeToLong() { this->SetOutputScalarType(VTK_LONG); }
  void SetOutputScalarTypeToUnsignedLong() { this->SetOutputScalarType(VTK_UNSIGNED_LONG); }
  void SetOutputScalarTypeToInt() { this->SetOutputScalarType(VTK_INT); }
  void SetOutputScalarTypeToUnsignedInt() { this->SetOutputScalarType(VTK_UNSIGNED_INT); }
  void SetOutputScalarTypeToShort() { this->SetOutputScalarType(VTK_SHORT); }
  void SetOutputScalarTypeToUnsignedShort() { this->SetOutputScalarType(VTK_UNSIGNED_SHORT); }
  void SetOutputScalarTypeToChar() { this->SetOutputScalarType(VTK_CHAR); }
  void SetOutputScalarTypeToSignedChar() { this->SetOutputScalarType(VTK_SIGNED_CHAR); }
  void SetOutputScalarTypeToUnsignedChar() { this->SetOutputScalarType(VTK_UNSIGNED_CHAR); }

protected:
  vtkImageBinaryThreshold() = default;
  ~vtkImageBinaryThreshold() override = default;

  int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

  void ThreadedRequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector, vtkImageData*** inData, vtkImageData** outData,
    VTK_FUTURE_CONST int outExt[6], int id) override;

private:
  vtkImageBinaryThreshold(const vtkImageBinaryThreshold&) = delete;
  void operator=(const vtkImageBinaryThreshold&) = delete;

  int ThresholdFunction = ThresholdFunction::THRESHOLD_BETWEEN;
  double UpperThreshold = VTK_FLOAT_MAX;
  double LowerThreshold = VTK_FLOAT_MIN;
  bool ReplaceIn = false;
  double InValue = 0.0;
  bool ReplaceOut = false;
  double OutValue = 0.0;

  int OutputScalarType = -1; // invalid; output same as input
};

VTK_ABI_NAMESPACE_END
#endif // vtkImageBinaryThreshold_h