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.
238 lines
7.1 KiB
C++
238 lines
7.1 KiB
C++
/*=========================================================================
|
|
|
|
Program: Visualization Toolkit
|
|
Module: vtkImageMathematics.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 vtkImageMathematics
|
|
* @brief Add, subtract, multiply, divide, invert, sin,
|
|
* cos, exp, log.
|
|
*
|
|
* vtkImageMathematics implements basic mathematic operations SetOperation is
|
|
* used to select the filters behavior. The filter can take two or one
|
|
* input.
|
|
*/
|
|
|
|
#ifndef vtkImageMathematics_h
|
|
#define vtkImageMathematics_h
|
|
|
|
// Operation options.
|
|
#define VTK_ADD 0
|
|
#define VTK_SUBTRACT 1
|
|
#define VTK_MULTIPLY 2
|
|
#define VTK_DIVIDE 3
|
|
#define VTK_INVERT 4
|
|
#define VTK_SIN 5
|
|
#define VTK_COS 6
|
|
#define VTK_EXP 7
|
|
#define VTK_LOG 8
|
|
#define VTK_ABS 9
|
|
#define VTK_SQR 10
|
|
#define VTK_SQRT 11
|
|
#define VTK_MIN 12
|
|
#define VTK_MAX 13
|
|
#define VTK_ATAN 14
|
|
#define VTK_ATAN2 15
|
|
#define VTK_MULTIPLYBYK 16
|
|
#define VTK_ADDC 17
|
|
#define VTK_CONJUGATE 18
|
|
#define VTK_COMPLEX_MULTIPLY 19
|
|
#define VTK_REPLACECBYK 20
|
|
|
|
#include "vtkImagingMathModule.h" // For export macro
|
|
#include "vtkThreadedImageAlgorithm.h"
|
|
|
|
class VTKIMAGINGMATH_EXPORT vtkImageMathematics : public vtkThreadedImageAlgorithm
|
|
{
|
|
public:
|
|
static vtkImageMathematics* New();
|
|
vtkTypeMacro(vtkImageMathematics, vtkThreadedImageAlgorithm);
|
|
void PrintSelf(ostream& os, vtkIndent indent) override;
|
|
|
|
//@{
|
|
/**
|
|
* Set/Get the Operation to perform.
|
|
*/
|
|
vtkSetMacro(Operation, int);
|
|
vtkGetMacro(Operation, int);
|
|
//@}
|
|
|
|
/**
|
|
* Set each pixel in the output image to the sum of the corresponding pixels
|
|
* in Input1 and Input2.
|
|
*/
|
|
void SetOperationToAdd() { this->SetOperation(VTK_ADD); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the difference of the corresponding pixels
|
|
* in Input1 and Input2 (output = Input1 - Input2).
|
|
*/
|
|
void SetOperationToSubtract() { this->SetOperation(VTK_SUBTRACT); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the product of the corresponding pixels
|
|
* in Input1 and Input2.
|
|
*/
|
|
void SetOperationToMultiply() { this->SetOperation(VTK_MULTIPLY); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the quotient of the corresponding pixels
|
|
* in Input1 and Input2 (Output = Input1 / Input2).
|
|
*/
|
|
void SetOperationToDivide() { this->SetOperation(VTK_DIVIDE); }
|
|
|
|
void SetOperationToConjugate() { this->SetOperation(VTK_CONJUGATE); }
|
|
|
|
void SetOperationToComplexMultiply() { this->SetOperation(VTK_COMPLEX_MULTIPLY); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to 1 over the corresponding pixel
|
|
* in Input1 and Input2 (output = 1 / Input1). Input2 is not used.
|
|
*/
|
|
void SetOperationToInvert() { this->SetOperation(VTK_INVERT); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the sine of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToSin() { this->SetOperation(VTK_SIN); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the cosine of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToCos() { this->SetOperation(VTK_COS); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the exponential of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToExp() { this->SetOperation(VTK_EXP); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the log of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToLog() { this->SetOperation(VTK_LOG); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the absolute value of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToAbsoluteValue() { this->SetOperation(VTK_ABS); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the square of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToSquare() { this->SetOperation(VTK_SQR); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the square root of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToSquareRoot() { this->SetOperation(VTK_SQRT); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the minimum of the corresponding pixels
|
|
* in Input1 and Input2. (Output = min(Input1, Input2))
|
|
*/
|
|
void SetOperationToMin() { this->SetOperation(VTK_MIN); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the maximum of the corresponding pixels
|
|
* in Input1 and Input2. (Output = max(Input1, Input2))
|
|
*/
|
|
void SetOperationToMax() { this->SetOperation(VTK_MAX); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the arctangent of the corresponding pixel
|
|
* in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToATAN() { this->SetOperation(VTK_ATAN); }
|
|
|
|
void SetOperationToATAN2() { this->SetOperation(VTK_ATAN2); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the product of ConstantK with the
|
|
* corresponding pixel in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToMultiplyByK() { this->SetOperation(VTK_MULTIPLYBYK); }
|
|
|
|
/**
|
|
* Set each pixel in the output image to the product of ConstantC with the
|
|
* corresponding pixel in Input1. Input2 is not used.
|
|
*/
|
|
void SetOperationToAddConstant() { this->SetOperation(VTK_ADDC); }
|
|
|
|
/**
|
|
* Find every pixel in Input1 that equals ConstantC and set the corresponding pixels
|
|
* in the Output to ConstantK. Input2 is not used.
|
|
*/
|
|
void SetOperationToReplaceCByK() { this->SetOperation(VTK_REPLACECBYK); }
|
|
|
|
//@{
|
|
/**
|
|
* A constant used by some operations (typically multiplicative). Default is 1.
|
|
*/
|
|
vtkSetMacro(ConstantK, double);
|
|
vtkGetMacro(ConstantK, double);
|
|
//@}
|
|
|
|
//@{
|
|
/**
|
|
* A constant used by some operations (typically additive). Default is 0.
|
|
*/
|
|
vtkSetMacro(ConstantC, double);
|
|
vtkGetMacro(ConstantC, double);
|
|
//@}
|
|
|
|
//@{
|
|
/**
|
|
* How to handle divide by zero. Default is 0.
|
|
*/
|
|
vtkSetMacro(DivideByZeroToC, vtkTypeBool);
|
|
vtkGetMacro(DivideByZeroToC, vtkTypeBool);
|
|
vtkBooleanMacro(DivideByZeroToC, vtkTypeBool);
|
|
//@}
|
|
|
|
/**
|
|
* Set the two inputs to this filter. For some operations, the second input
|
|
* is not used.
|
|
*/
|
|
virtual void SetInput1Data(vtkDataObject* in) { this->SetInputData(0, in); }
|
|
virtual void SetInput2Data(vtkDataObject* in) { this->SetInputData(1, in); }
|
|
|
|
protected:
|
|
vtkImageMathematics();
|
|
~vtkImageMathematics() override {}
|
|
|
|
int Operation;
|
|
double ConstantK;
|
|
double ConstantC;
|
|
vtkTypeBool DivideByZeroToC;
|
|
|
|
int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
|
|
|
|
void ThreadedRequestData(vtkInformation* request, vtkInformationVector** inputVector,
|
|
vtkInformationVector* outputVector, vtkImageData*** inData, vtkImageData** outData,
|
|
int extent[6], int threadId) override;
|
|
|
|
int FillInputPortInformation(int port, vtkInformation* info) override;
|
|
|
|
private:
|
|
vtkImageMathematics(const vtkImageMathematics&) = delete;
|
|
void operator=(const vtkImageMathematics&) = delete;
|
|
};
|
|
|
|
#endif
|