defaultPartialOrderedEliminationSequenceStrategy.cpp

Go to the documentation of this file.

#include <agrum/agrum.h>
#include <agrum/core/math/math.h>

#include <agrum/graphs/algorithms/triangulations/eliminationStrategies/defaultPartialOrderedEliminationSequenceStrategy.h>
#include <agrum/graphs/undiGraph.h>

namespace gum {

  DefaultPartialOrderedEliminationSequenceStrategy::
     DefaultPartialOrderedEliminationSequenceStrategy(double theRatio,
                                                      double theThreshold) :
      __simplicial_ratio(theRatio),
      __simplicial_threshold(theThreshold) {
    // for debugging purposes
    GUM_CONSTRUCTOR(DefaultPartialOrderedEliminationSequenceStrategy);
  }

  DefaultPartialOrderedEliminationSequenceStrategy::
     DefaultPartialOrderedEliminationSequenceStrategy(
        UndiGraph*                  graph,
        const NodeProperty< Size >* dom_sizes,
        const List< NodeSet >*      subsets,
        double                      ratio,
        double                      threshold) :
      __simplicial_ratio(ratio),
      __simplicial_threshold(threshold) {
    setGraph(graph, dom_sizes);
    setPartialOrder(subsets);

    // for debugging purposes
    GUM_CONSTRUCTOR(DefaultPartialOrderedEliminationSequenceStrategy);
  }

  DefaultPartialOrderedEliminationSequenceStrategy::
     DefaultPartialOrderedEliminationSequenceStrategy(
        const DefaultPartialOrderedEliminationSequenceStrategy& from) :
      PartialOrderedEliminationSequenceStrategy(from),
      // no need to set __log_weights because the copy of the simplicial set
      // will set it properly
      __simplicial_set(new SimplicialSet(*from.__simplicial_set,
                                         _graph,
                                         &_log_domain_sizes,
                                         &__log_weights,
                                         false)),
      __simplicial_ratio(from.__simplicial_ratio),
      __simplicial_threshold(from.__simplicial_threshold),
      __provide_fill_ins(from.__provide_fill_ins) {
    // for debugging purposes
    GUM_CONS_CPY(DefaultPartialOrderedEliminationSequenceStrategy);
  }

  DefaultPartialOrderedEliminationSequenceStrategy::
     DefaultPartialOrderedEliminationSequenceStrategy(
        DefaultPartialOrderedEliminationSequenceStrategy&& from) :
      PartialOrderedEliminationSequenceStrategy(std::move(from)),
      __log_weights(std::move(from.__log_weights)),
      __simplicial_set(from.__simplicial_set),
      __simplicial_ratio(from.__simplicial_ratio),
      __simplicial_threshold(from.__simplicial_threshold),
      __provide_fill_ins(from.__provide_fill_ins) {
    __simplicial_set->replaceLogWeights(&from.__log_weights, &__log_weights);
    from.__simplicial_set = nullptr;

    // for debugging purposes
    GUM_CONS_MOV(DefaultPartialOrderedEliminationSequenceStrategy);
  }

  DefaultPartialOrderedEliminationSequenceStrategy::
     ~DefaultPartialOrderedEliminationSequenceStrategy() {
    // for debugging purposes
    GUM_DESTRUCTOR(DefaultPartialOrderedEliminationSequenceStrategy);

    if (__simplicial_set) delete __simplicial_set;
  }

  void DefaultPartialOrderedEliminationSequenceStrategy::__createSimplicialSet() {
    // remove the old simplicial set, if any
    if (__simplicial_set != nullptr) {
      delete __simplicial_set;
      __simplicial_set = nullptr;
    }

    if (_graph != nullptr) {
      // create a simplicial set suited for the graph
      __simplicial_set = new SimplicialSet(_graph,
                                           &_log_domain_sizes,
                                           &__log_weights,
                                           __simplicial_ratio,
                                           __simplicial_threshold);

      __simplicial_set->setFillIns(__provide_fill_ins);
    }
  }

  bool DefaultPartialOrderedEliminationSequenceStrategy::setGraph(
     UndiGraph* graph, const NodeProperty< Size >* domain_sizes) {
    if (PartialOrderedEliminationSequenceStrategy::setGraph(graph, domain_sizes)) {
      __createSimplicialSet();
      return true;
    }

    return false;
  }

  void DefaultPartialOrderedEliminationSequenceStrategy::clear() {
    PartialOrderedEliminationSequenceStrategy::clear();
    __log_weights.clear();

    if (__simplicial_set != nullptr) {
      delete __simplicial_set;
      __simplicial_set = nullptr;
    }
  }

  NodeId DefaultPartialOrderedEliminationSequenceStrategy::__nodeToEliminate(
     const PriorityQueue< NodeId, double >& possibleNodes) {
    bool   found = false;
    double min_score = 0;
    NodeId best_node = 0;

    for (const auto node : _nodeset) {
      try {
        double score = possibleNodes.priority(node);

        if (!found || (score < min_score)) {
          found = true;
          min_score = score;
          best_node = node;
        }
      } catch (NotFound&) {}
    }

    if (!found) { GUM_ERROR(NotFound, "no possible node to eliminate"); }

    return best_node;
  }

  NodeId DefaultPartialOrderedEliminationSequenceStrategy::nextNodeToEliminate() {
    // if there is no simplicial set, send an exception
    if (_graph == nullptr) GUM_ERROR(NotFound, "the graph is empty");

    if (_partial_order_needed)
      GUM_ERROR(NotFound,
                "the partial order does not cover all the nodes "
                "of the graph");

    if (_nodeset.empty()) { GUM_ERROR(NotFound, "no node is admissible"); }

    // select a node to be eliminated: try simplicial nodes, then almost
    // simplicial nodes, then quasi-simplicial nodes
    // note that if _graph != nullptr, __simplicial_set has been allocated
    try {
      return __nodeToEliminate(__simplicial_set->allSimplicialNodes());
    } catch (NotFound&) {}

    try {
      return __nodeToEliminate(__simplicial_set->allAlmostSimplicialNodes());
    } catch (NotFound&) {}

    try {
      return __nodeToEliminate(__simplicial_set->allQuasiSimplicialNodes());
    } catch (NotFound&) {}

    // here: select the node through Kjaerulff's heuristic
    auto   iter = _nodeset.cbegin();
    double min_score = __log_weights[*iter];
    NodeId best_node = *iter;

    for (++iter; iter != _nodeset.cend(); ++iter) {
      double score = __log_weights[*iter];

      if (score < min_score) {
        min_score = score;
        best_node = *iter;
      }
    }

    return best_node;
  }

  void DefaultPartialOrderedEliminationSequenceStrategy::askFillIns(bool do_it) {
    __provide_fill_ins = do_it;

    if (__simplicial_set) __simplicial_set->setFillIns(__provide_fill_ins);
  }

  bool DefaultPartialOrderedEliminationSequenceStrategy::providesFillIns() const {
    return __provide_fill_ins;
  }

  bool DefaultPartialOrderedEliminationSequenceStrategy::providesGraphUpdate()
     const {
    return true;
  }

  void DefaultPartialOrderedEliminationSequenceStrategy::eliminationUpdate(
     const NodeId id) {
    // check whether we can do something
    if (__simplicial_set != nullptr) {
      __simplicial_set->makeClique(id);
      __simplicial_set->eraseClique(id);

      if (!_partial_order_needed) {
        // remove the node from _nodeset
        _nodeset.erase(id);

        if (_nodeset.empty()) {
          // go to the next non-empty subset
          for (++_subset_iter; _subset_iter != _subsets->cend(); ++_subset_iter) {
            for (const auto node : *_subset_iter) {
              if (_graph->existsNode(node)) { _nodeset.insert(node); }
            }
            if (!_nodeset.empty()) break;
          }
        }
      }
    }
  }

  const EdgeSet& DefaultPartialOrderedEliminationSequenceStrategy::fillIns() {
    if (!__provide_fill_ins || (__simplicial_set == nullptr))
      return EliminationSequenceStrategy::fillIns();
    else
      return __simplicial_set->fillIns();
  }

  DefaultPartialOrderedEliminationSequenceStrategy*
     DefaultPartialOrderedEliminationSequenceStrategy::newFactory() const {
    return new DefaultPartialOrderedEliminationSequenceStrategy(
       __simplicial_ratio, __simplicial_threshold);
  }

  DefaultPartialOrderedEliminationSequenceStrategy*
     DefaultPartialOrderedEliminationSequenceStrategy::copyFactory() const {
    return new DefaultPartialOrderedEliminationSequenceStrategy(*this);
  }

} /* namespace gum */