ttkMergeTree.h

Go to the documentation of this file.

 
 
#pragma once
 
// VTK includes
#include <vtkSmartPointer.h>
 
// VTK module
#include <ttkMergeTreeModule.h>
 
// ttk code includes
#include <PathCompression.h>
#include <ttkAlgorithm.h>
#include <ttkMergeTreeBase.h>
#include <ttkMergeTreeStructures.h>
 
class vtkDataSet;
 
class TTKMERGETREE_EXPORT ttkMergeTree : public ttkAlgorithm,
                                         public ttkMergeTreeBase {
 
public:
  enum class BACKEND {
    FTM = 0,
    EXTREEM = 1,
  };
 
  static ttkMergeTree *New();
 
  vtkTypeMacro(ttkMergeTree, ttkAlgorithm);
 
  vtkGetMacro(ForceInputOffsetScalarField, bool);
  vtkSetMacro(ForceInputOffsetScalarField, bool);
 
  vtkGetMacro(Backend, int);
  vtkSetMacro(Backend, int);
 
  // Parameters uses a structure, we can't use vtkMacro on them
 
  void SetTreeType(const int type) {
    params_.treeType = (ttk::ftm::TreeType)type;
    Modified();
  }
 
  void SetWithSegmentation(const bool segm) {
    params_.segm = segm;
    Modified();
  }
  bool GetWithSegmentation() const {
    return params_.segm;
  }
 
  void SetWithNormalize(const bool norm) {
    params_.normalize = norm;
    Modified();
  }
  bool GetWithNormalize() const {
    return params_.normalize;
  }
 
  void SetWithAdvStats(const bool adv) {
    params_.advStats = adv;
    Modified();
  }
  bool GetWithAdvStats() const {
    return params_.advStats;
  }
 
  void SetSuperArcSamplingLevel(int lvl) {
    params_.samplingLvl = lvl;
    Modified();
  }
  int GetSuperArcSamplingLevel() const {
    return params_.samplingLvl;
  }
 
protected:
  ttkMergeTree();
 
  int getOffsets();
  int getScalars();
 
  int preconditionTriangulation();
 
  // vtkDataSetAlgorithm methods
  int FillInputPortInformation(int port, vtkInformation *info) override;
  int FillOutputPortInformation(int port, vtkInformation *info) override;
  int RequestData(vtkInformation *request,
                  vtkInformationVector **inputVector,
                  vtkInformationVector *outputVector) override;
 
private:
  int Backend{(int)BACKEND::FTM};
};