vtkImageMathematics.h

Go to the documentation of this file.

/*=========================================================================
 
  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.
 
=========================================================================*/
#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);
 
 
  vtkSetMacro(Operation,int);
  vtkGetMacro(Operation,int);
 
  void SetOperationToAdd() {this->SetOperation(VTK_ADD);};
 
  void SetOperationToSubtract() {this->SetOperation(VTK_SUBTRACT);};
 
  void SetOperationToMultiply() {this->SetOperation(VTK_MULTIPLY);};
 
  void SetOperationToDivide() {this->SetOperation(VTK_DIVIDE);};
 
  void SetOperationToConjugate() {this->SetOperation(VTK_CONJUGATE);};
 
  void SetOperationToComplexMultiply()
    {this->SetOperation(VTK_COMPLEX_MULTIPLY);};
 
  void SetOperationToInvert() {this->SetOperation(VTK_INVERT);};
 
  void SetOperationToSin() {this->SetOperation(VTK_SIN);};
 
  void SetOperationToCos() {this->SetOperation(VTK_COS);};
 
  void SetOperationToExp() {this->SetOperation(VTK_EXP);};
 
  void SetOperationToLog() {this->SetOperation(VTK_LOG);};
 
  void SetOperationToAbsoluteValue() {this->SetOperation(VTK_ABS);};
 
  void SetOperationToSquare() {this->SetOperation(VTK_SQR);};
 
  void SetOperationToSquareRoot() {this->SetOperation(VTK_SQRT);};
 
  void SetOperationToMin() {this->SetOperation(VTK_MIN);};
 
  void SetOperationToMax() {this->SetOperation(VTK_MAX);};
 
  void SetOperationToATAN() {this->SetOperation(VTK_ATAN);};
 
  void SetOperationToATAN2() {this->SetOperation(VTK_ATAN2);};
 
  void SetOperationToMultiplyByK() {this->SetOperation(VTK_MULTIPLYBYK);};
 
  void SetOperationToAddConstant() {this->SetOperation(VTK_ADDC);};
 
  void SetOperationToReplaceCByK() {this->SetOperation(VTK_REPLACECBYK);};
 
 
  vtkSetMacro(ConstantK,double);
  vtkGetMacro(ConstantK,double);
 
 
  vtkSetMacro(ConstantC,double);
  vtkGetMacro(ConstantC,double);
 
 
  vtkSetMacro(DivideByZeroToC,int);
  vtkGetMacro(DivideByZeroToC,int);
  vtkBooleanMacro(DivideByZeroToC,int);
 
 
  virtual void SetInput1Data(vtkDataObject *in) { this->SetInputData(0,in); }
  virtual void SetInput2Data(vtkDataObject *in) { this->SetInputData(1,in); }
 
protected:
  vtkImageMathematics();
  ~vtkImageMathematics() {}
 
  int Operation;
  double ConstantK;
  double ConstantC;
  int DivideByZeroToC;
 
  virtual int RequestInformation (vtkInformation *,
                                  vtkInformationVector **,
                                  vtkInformationVector *);
 
  virtual void ThreadedRequestData(vtkInformation *request,
                                   vtkInformationVector **inputVector,
                                   vtkInformationVector *outputVector,
                                   vtkImageData ***inData,
                                   vtkImageData **outData,
                                   int extent[6], int threadId);
 
  virtual int FillInputPortInformation(int port, vtkInformation* info);
 
private:
  vtkImageMathematics(const vtkImageMathematics&);  // Not implemented.
  void operator=(const vtkImageMathematics&);  // Not implemented.
};
 
#endif