SetTerminalNodePolicy.h

Go to the documentation of this file.

#ifndef GUM_MULTI_DIM_FUNCTION_GRAPH_SET_TERMINAL_NODE_POLICY_H
#define GUM_MULTI_DIM_FUNCTION_GRAPH_SET_TERMINAL_NODE_POLICY_H

#include <agrum/graphs/parts/nodeGraphPart.h>

namespace gum {

  // clang-format off
  // clang-format on
  template < typename GUM_SCALAR >
  class SetTerminalNodePolicy {
    public:
    ~SetTerminalNodePolicy() { clearAllTerminalNodes(); }

    // ============================================================================
    // ============================================================================

    void addTerminalNode(const NodeId& n, const GUM_SCALAR& v) {
      if (__map.exists(n)) *(__map[n]) += v;
      __map.insert(n, new GUM_SCALAR(v));
    }

    void eraseTerminalNode(const NodeId& n) {
      if (__map.exists(n)) __map.erase(n);
    }

    void clearAllTerminalNodes() {
      for (auto nodeIter = __map.beginSafe(); nodeIter != __map.endSafe();
           ++nodeIter)
        delete nodeIter.val();
      __map.clear();
    }

    // ============================================================================
    // ============================================================================

    bool existsTerminalNodeWithId(const NodeId& n) const {
      return __map.exists(n);
    }

    bool existsTerminalNodeWithValue(const GUM_SCALAR& v) const {
      return terminalNodeId(v) != 0;
    }

    // ============================================================================
    // ============================================================================

    const GUM_SCALAR& terminalNodeValue(const NodeId& n) const {
      return *(__map[n]);
    }

    const NodeId& terminalNodeId(const GUM_SCALAR& v) const {
      for (auto nodeIter = __map.beginSafe(); nodeIter != __map.endSafe();
           ++nodeIter)
        if (*(nodeIter.val()) == v) return nodeIter.key();
      return jocker;
    }

    // ============================================================================
    // ============================================================================

    void beginValues() const { __mappy = __map.cbeginSafe(); }

    bool hasValue() const { return __mappy != __map.cendSafe(); }

    void nextValue() const { ++__mappy; }

    const GUM_SCALAR& value() const { return *(__mappy.val()); }

    const NodeId& id() const { return __mappy.key(); }

    private:
    HashTable< NodeId, GUM_SCALAR* >                          __map;
    mutable HashTableConstIteratorSafe< NodeId, GUM_SCALAR* > __mappy;

    const NodeId jocker = 0;
  };

}   // End of namespace gum

#endif /* GUM_MULTI_DIM_FUNCTION_GRAPH_SET_TERMINAL_NODE_POLICY_H */