FTRSuperArc.h

Go to the documentation of this file.

 
 
#pragma once
 
// local includes
#include "FTRAtomicUF.h"
#include "FTRDataTypes.h"
#include "FTRScalars.h"
#include "FTRSegmentation.h"
 
// c++ includes
#ifndef TTK_ENABLE_KAMIKAZE
#include <iostream>
#endif
 
namespace ttk {
  namespace ftr {
    class Node;
 
    class SuperArc : virtual public Debug {
      idNode upNodeId_{};
      idNode downNodeId_{};
      AtomicUF *ufProp_{};
      bool visible_{true};
      idVertex firstReg_{nullVertex}, lastReg_{nullVertex}, endV_{nullVertex};
      idSuperArc merged_{nullSuperArc};
      Segment segmentation_{};
#ifndef NDEBUG
      bool fromUp_{false};
#endif
 
    public:
      SuperArc(const idNode down = nullNode, const idNode up = nullNode)
        : upNodeId_{up}, downNodeId_{down} {
        this->setDebugMsgPrefix("SuperNode");
      }
 
      idNode getUpNodeId() const {
        // Caution. can be nullNode
        return upNodeId_;
      }
 
      void setUpNodeId(const idNode id) {
        upNodeId_ = id;
      }
 
      idNode getDownNodeId() const {
#ifndef TTK_ENABLE_KAMIKAZE
        if(downNodeId_ == nullNode) {
          this->printErr("Arc have null down node");
        }
#endif
        return downNodeId_;
      }
 
      void setDownNodeId(const idNode id) {
        downNodeId_ = id;
      }
 
      Propagation *getPropagation() const {
#ifndef TTK_ENABLE_KAMIKAZE
        if(!ufProp_) {
          this->printErr("Arc have null UF propagation");
        }
#endif
        return ufProp_->find()->getPropagation();
      }
 
      void setUfProp(AtomicUF *const UFprop) {
        ufProp_ = UFprop;
      }
 
      bool hide() {
        bool const old = visible_;
        visible_ = false;
        return old;
      }
 
      bool isVisible() const {
        return visible_;
      }
 
      // for regular vertices
      void visit(const idVertex v) {
        // firstV only set once
        if(firstReg_ == nullVertex)
          firstReg_ = v;
 
        lastReg_ = v;
      }
 
      // for all vertices:
      // used to close arc at the end if not close during construct
      void setEnd(const idVertex v) {
        // avoid an are being reclosed by a higher join
        if(endV_ == nullVertex) {
          endV_ = v;
        }
      }
 
      idVertex getEnd() const {
        return endV_;
      }
 
      bool isEmpty() const {
        return firstReg_ == nullVertex;
      }
 
      idVertex getFirstReg() const {
        return firstReg_;
      }
 
      idVertex getLastReg() const {
        return lastReg_;
      }
 
      void merge(const idSuperArc arc) {
        if(merged_ == nullSuperArc) {
          merged_ = arc;
          hide();
        }
      }
 
      bool merged() const {
        return merged_ != nullSuperArc;
      }
 
      idSuperArc mergedIn() const {
        return merged_;
      }
 
      void restore() {
        visible_ = true;
        merged_ = nullSuperArc;
      }
 
      const decltype(segmentation_) &segmentation() const {
        return segmentation_;
      }
 
      decltype(segmentation_) &segmentation() {
        return segmentation_;
      }
 
#ifndef NDEBUG
      void setFromUp(bool up) {
        fromUp_ = up;
      }
 
      bool getFromUp() const {
        return fromUp_;
      }
#endif
    };
  } // namespace ftr
} // namespace ttk