TrackingFromPersistenceDiagrams.h

Go to the documentation of this file.

 
#pragma once
 
// base code includes
#include <BottleneckDistance.h>
#include <Geometry.h>
#include <set>
 
namespace ttk {
 
  using trackingTuple = std::tuple<int, int, std::vector<SimplexId>>;
 
  class TrackingFromPersistenceDiagrams : virtual public Debug {
 
  public:
    TrackingFromPersistenceDiagrams();
 
    int execute();
 
    int performSingleMatching(
      int i,
      std::vector<ttk::DiagramType> &inputPersistenceDiagrams,
      std::vector<std::vector<MatchingType>> &outputMatchings,
      const std::string &algorithm,
      const std::string &wasserstein,
      double tolerance,
      double px,
      double py,
      double pz,
      double ps,
      double pe);
 
    int
      performMatchings(int numInputs,
                       std::vector<ttk::DiagramType> &inputPersistenceDiagrams,
                       std::vector<std::vector<MatchingType>> &outputMatchings,
                       const std::string &algorithm,
                       const std::string &wasserstein,
                       double tolerance,
                       double px,
                       double py,
                       double pz,
                       double ps,
                       double pe);
 
    int performTracking(std::vector<ttk::DiagramType> &allDiagrams,
                        std::vector<std::vector<MatchingType>> &allMatchings,
                        std::vector<trackingTuple> &trackings);
 
    int performPostProcess(const std::vector<ttk::DiagramType> &allDiagrams,
                           const std::vector<trackingTuple> &trackings,
                           std::vector<std::set<int>> &trackingTupleToMerged,
                           const double postProcThresh);
 
    inline int setInputDataPointer(int idx, void *data) {
      if(idx < numberOfInputs_)
        inputData_[idx] = data;
      else
        return -1;
      return 0;
    }
 
    inline void setNumberOfInputs(int numberOfInputs) {
      numberOfInputs_ = numberOfInputs;
    }
 
  protected:
    int numberOfInputs_{};
    void **inputData_{};
  };
} // namespace ttk