vtkPVExtractHistogram2D.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkPVExtractHistogram2D_h
#define vtkPVExtractHistogram2D_h

// VTK includes
#include "vtkImageAlgorithm.h"
#include "vtkPVVTKExtensionsMiscModule.h" // needed for exports

// Forward declarations
class vtkDataArray;
class vtkMultiProcessController;
class vtkUnsignedCharArray;

class VTKPVVTKEXTENSIONSMISC_EXPORT vtkPVExtractHistogram2D : public vtkImageAlgorithm
{
public:
  static vtkPVExtractHistogram2D* New();


  vtkTypeMacro(vtkPVExtractHistogram2D, vtkImageAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  virtual void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);


  vtkSetMacro(Component0, int);
  vtkGetMacro(Component0, int);
  vtkSetMacro(Component1, int);
  vtkGetMacro(Component1, int);
  void SetComponent(int id, int val) { (id == 0) ? SetComponent0(val) : SetComponent1(val); }
  int GetComponent(int id) { return ((id == 0) ? GetComponent0() : GetComponent1()); }


  vtkSetVector2Macro(NumberOfBins, int);
  vtkGetVector2Macro(NumberOfBins, int);


  vtkSetVector2Macro(CustomBinRanges0, double);
  vtkGetVector2Macro(CustomBinRanges0, double);
  vtkSetVector2Macro(CustomBinRanges1, double);
  vtkGetVector2Macro(CustomBinRanges1, double);
  vtkSetMacro(UseCustomBinRanges0, bool);
  vtkGetMacro(UseCustomBinRanges0, bool);
  vtkBooleanMacro(UseCustomBinRanges0, bool);
  vtkSetMacro(UseCustomBinRanges1, bool);
  vtkGetMacro(UseCustomBinRanges1, bool);
  vtkBooleanMacro(UseCustomBinRanges1, bool);


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


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


  vtkSetVector2Macro(OutputOrigin, double);
  vtkGetVector2Macro(OutputOrigin, double);
  vtkSetVector2Macro(OutputSpacing, double);
  vtkGetVector2Macro(OutputSpacing, double);

protected:
  vtkPVExtractHistogram2D();
  ~vtkPVExtractHistogram2D() override;

  // Helper members
  int RequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
  int RequestInformation(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
  int RequestUpdateExtent(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;

  int FillInputPortInformation(int port, vtkInformation* info) override;

  void ComputeHistogram2D(vtkImageData* histogram);
  void ComputeGradient(vtkDataObject* input);

  int Component0 = 0;
  int Component1 = 0;
  int NumberOfBins[2] = { 256, 256 };
  bool UseGradientForYAxis = false;

  double CustomBinRanges0[2];
  double CustomBinRanges1[2];
  bool UseCustomBinRanges0 = false;
  bool UseCustomBinRanges1 = false;
  vtkMultiProcessController* Controller = nullptr;
  bool UseInputRangesForOutputBounds = true;
  double OutputOrigin[2] = { 0.0, 0.0 };
  double OutputSpacing[2] = { 1.0, 1.0 };

  // Cache of internal array and range
  int ComponentIndexCache[2];
  vtkDataArray* ComponentArrayCache[2];
  vtkUnsignedCharArray* GhostArray;
  unsigned char GhostsToSkip;
  double ComponentRangeCache[2][2];

  void InitializeCache();
  void GetInputArrays(vtkInformationVector**);
  void ComputeVectorMagnitude(vtkDataArray*, vtkDataArray*&);
  void ComputeComponentRange();

private:
  vtkPVExtractHistogram2D(const vtkPVExtractHistogram2D&) = delete;
  void operator=(const vtkPVExtractHistogram2D) = delete;
};

#endif // vtkPVExtractHistogram2D_h