ttkMergeTreeTemporalReduction.h

Go to the documentation of this file.

 
#pragma once
 
// VTK Module
#include <ttkMergeTreeTemporalReductionModule.h>
 
// VTK Includes
#include <ttkAlgorithm.h>
#include <vtkMultiBlockDataSet.h>
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>
 
// TTK Base Includes
#include <MergeTreeTemporalReduction.h>
 
class TTKMERGETREETEMPORALREDUCTION_EXPORT ttkMergeTreeTemporalReduction
  : public ttkAlgorithm // we inherit from the generic ttkAlgorithm class
  ,
    protected ttk::MergeTreeTemporalReduction // and we inherit from the
                                              // base class
{
private:
  // Execution options
  std::string TimeVariableName{""};
 
  // Input Options
  bool DoResampleToImage = false;
 
  // ----------------------
  // Data for visualization
  // ----------------------
  // Trees
  std::vector<vtkUnstructuredGrid *> treesNodes;
  std::vector<vtkUnstructuredGrid *> treesArcs;
  std::vector<vtkDataSet *> treesSegmentation;
  // Output
  std::vector<std::vector<int>> treesNodeCorrMesh;
  std::vector<double> emptyTreeDistances;
  std::vector<ttk::ftm::MergeTree<double>> keyFrames;
  std::vector<int> removed;
 
  void setDataVisualization(int numInputs) {
    // Trees
    treesNodes = std::vector<vtkUnstructuredGrid *>(numInputs);
    treesArcs = std::vector<vtkUnstructuredGrid *>(numInputs);
    treesSegmentation = std::vector<vtkDataSet *>(numInputs);
  }
 
  void resetDataVisualization() {
    setDataVisualization(0);
    treesNodeCorrMesh = std::vector<std::vector<int>>();
    emptyTreeDistances = std::vector<double>();
    keyFrames = std::vector<ttk::ftm::MergeTree<double>>();
    removed = std::vector<int>();
  }
 
  bool isDataVisualizationFilled() {
    return treesNodeCorrMesh.size() != 0 and keyFrames.size() != 0
           and emptyTreeDistances.size() != 0 and removed.size() != 0;
  }
 
public:
  // Input Options
  void SetEpsilon1UseFarthestSaddle(bool epsilon1UseFarthestSaddle) {
    epsilon1UseFarthestSaddle_ = epsilon1UseFarthestSaddle;
    Modified();
    resetDataVisualization();
  }
  bool GetEpsilon1UseFarthestSaddle() {
    return epsilon1UseFarthestSaddle_;
  }
 
  void SetEpsilonTree1(double epsilonTree1) {
    epsilonTree1_ = epsilonTree1;
    Modified();
    resetDataVisualization();
  }
  double SetEpsilonTree1() {
    return epsilonTree1_;
  }
 
  void SetEpsilon2Tree1(double epsilon2Tree1) {
    epsilon2Tree1_ = epsilon2Tree1;
    Modified();
    resetDataVisualization();
  }
  double SetEpsilon2Tree1() {
    return epsilon2Tree1_;
  }
 
  void SetEpsilon3Tree1(double epsilon3Tree1) {
    epsilon3Tree1_ = epsilon3Tree1;
    Modified();
    resetDataVisualization();
  }
  double SetEpsilon3Tree1() {
    return epsilon3Tree1_;
  }
 
  void SetPersistenceThreshold(double persistenceThreshold) {
    persistenceThreshold_ = persistenceThreshold;
    Modified();
    resetDataVisualization();
  }
  double SetPersistenceThreshold() {
    return persistenceThreshold_;
  }
 
  void SetUseMinMaxPair(bool useMinMaxPair) {
    useMinMaxPair_ = useMinMaxPair;
    Modified();
    resetDataVisualization();
  }
  bool SetUseMinMaxPair() {
    return useMinMaxPair_;
  }
 
  void SetDeleteMultiPersPairs(bool doDelete) {
    deleteMultiPersPairs_ = doDelete;
    Modified();
    resetDataVisualization();
  }
  bool SetDeleteMultiPersPairs() {
    return deleteMultiPersPairs_;
  }
 
  // Execution Options
  void SetAssignmentSolver(int assignmentSolver) {
    assignmentSolverID_ = assignmentSolver;
    Modified();
    resetDataVisualization();
  }
  int GetAssignmentSolver() {
    return assignmentSolverID_;
  }
 
  void SetRemovalPercentage(double removePerc) {
    removalPercentage_ = removePerc;
    Modified();
    resetDataVisualization();
  }
  double GetRemovalPercentage() {
    return removalPercentage_;
  }
 
  void SetUseL2Distance(double useL2) {
    useL2Distance_ = useL2;
    Modified();
    resetDataVisualization();
  }
  double GetUseL2Distance() {
    return useL2Distance_;
  }
 
  void SetUseCustomTimeVariable(bool useCustomTime) {
    useCustomTimeVariable_ = useCustomTime;
    Modified();
    resetDataVisualization();
  }
  bool GetUseCustomTimeVariable() {
    return useCustomTimeVariable_;
  }
 
  void SetTimeVariableName(
    int idx, int port, int connection, int fieldAssociation, const char *name) {
    SetInputArrayToProcess(idx, port, connection, fieldAssociation, name);
    TimeVariableName = std::string(name);
    Modified();
    resetDataVisualization();
  }
  vtkGetMacro(TimeVariableName, std::string);

  static ttkMergeTreeTemporalReduction *New();
  vtkTypeMacro(ttkMergeTreeTemporalReduction, ttkAlgorithm);
 
protected:
  ttkMergeTreeTemporalReduction();
  ~ttkMergeTreeTemporalReduction() override;

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

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

  int RequestData(vtkInformation *request,
                  vtkInformationVector **inputVector,
                  vtkInformationVector *outputVector) override;
 
  template <class dataType>
  int run(vtkInformationVector *outputVector,
          std::vector<vtkSmartPointer<vtkMultiBlockDataSet>> &inputTrees);
 
  template <class dataType>
  int runCompute(
    std::vector<vtkSmartPointer<vtkMultiBlockDataSet>> &inputTrees);
 
  template <class dataType>
  int runOutput(vtkInformationVector *outputVector,
                std::vector<vtkSmartPointer<vtkMultiBlockDataSet>> &inputTrees);
};