ContourForestsTree.h

Go to the documentation of this file.

 
 
#pragma once
 
#include <memory>
#include <queue>
#include <set>
 
// base code includes
#include <Geometry.h>
#include <Triangulation.h>
#include <Wrapper.h>
 
#include "DeprecatedDataTypes.h"
#include "ExtendedUF.h"
#include "MergeTree.h"
 
namespace ttk {
  namespace cf {
    class ContourForestsTree : public MergeTree {
      friend class ContourForests;
 
    protected:
      MergeTree jt_, st_;
 
    public:
      // -----------------
      // Constructors
      // -----------------
      // {
 
      ContourForestsTree(const std::shared_ptr<Params> &params,
                         const std::shared_ptr<Scalars> &scalars,
                         idPartition part = nullPartition);
      ~ContourForestsTree() override;
 
      // }
      // -----------------
      // INITIALIZE
      // -----------------
      // {
 
      void flush() {
        MergeTree::flush();
        jt_.flush();
        st_.flush();
      }
 
      // }
      // -----------------
      // ACCESSOR
      // -----------------
      // {
 
      inline MergeTree *getJoinTree() {
        return &jt_;
      }
 
      inline MergeTree *getSplitTree() {
        return &st_;
      }
 
      inline MergeTree *getTree(const TreeType &tt) {
        switch(tt) {
          case TreeType::JoinAndSplit:
          case TreeType::Split:
            return getSplitTree();
            break;
          case TreeType::Join:
            return getJoinTree();
            break;
          case TreeType::Contour:
            return this;
            break;
        }
        return this;
      }
 
      // }
      // -----------------
      // PROCESS
      // -----------------
      // {
 
      int combine(const SimplexId &seed0,
                  const SimplexId &seed1,
                  std::list<std::vector<std::pair<SimplexId, bool>>> &storage);
 
    private:
      // -----------------
      // PROCESS
      // -----------------
      // {
 
      template <typename scalarType>
      void initDataMT();
 
      // }
    };
  } // namespace cf
} // namespace ttk