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&) VTK_DELETE_FUNCTION;
  void operator=(const vtkImageMathematics&) VTK_DELETE_FUNCTION;
};

#endif