vtkImageWeightedSum.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 vtkImageWeightedSum_h
#define vtkImageWeightedSum_h
 
#include "vtkImagingMathModule.h" // For export macro
#include "vtkThreadedImageAlgorithm.h"
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDoubleArray;
class VTKIMAGINGMATH_EXPORT vtkImageWeightedSum : public vtkThreadedImageAlgorithm
{
public:
  static vtkImageWeightedSum* New();
  vtkTypeMacro(vtkImageWeightedSum, vtkThreadedImageAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  virtual void SetWeights(vtkDoubleArray*);
  vtkGetObjectMacro(Weights, vtkDoubleArray);

  virtual void SetWeight(vtkIdType id, double weight);


  vtkGetMacro(NormalizeByWeight, vtkTypeBool);
  vtkSetClampMacro(NormalizeByWeight, vtkTypeBool, 0, 1);
  vtkBooleanMacro(NormalizeByWeight, vtkTypeBool);

  double CalculateTotalWeight();
 
protected:
  vtkImageWeightedSum();
  ~vtkImageWeightedSum() override;
 
  // Array to hold all the weights
  vtkDoubleArray* Weights;
 
  // Boolean flag to divide by sum or not
  vtkTypeBool NormalizeByWeight;
 
  int RequestInformation(vtkInformation* vtkNotUsed(request),
    vtkInformationVector** vtkNotUsed(inputVector), vtkInformationVector* outputVector) override;
 
  void ThreadedRequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector, vtkImageData*** inData, vtkImageData** outData,
    VTK_FUTURE_CONST int ext[6], int id) override;
  int FillInputPortInformation(int i, vtkInformation* info) override;
 
private:
  vtkImageWeightedSum(const vtkImageWeightedSum&) = delete;
  void operator=(const vtkImageWeightedSum&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif