treeOperator.h

Go to the documentation of this file.

#ifndef GUM_TREE_OPERATOR_H
#define GUM_TREE_OPERATOR_H

#include <functional>

#include <agrum/multidim/implementations/multiDimFunctionGraph.h>
#include <agrum/multidim/utils/FunctionGraphUtilities/terminalNodePolicies/ExactTerminalNodePolicy.h>

namespace gum {

  template < typename GUM_SCALAR,
             template < typename >
             class COMBINEOPERATOR,
             template < typename > class TerminalNodePolicy =
                ExactTerminalNodePolicy >
  class TreeOperator {
    public:
    // ============================================================================
    // ============================================================================

    TreeOperator(
       const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* dt1,
       const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* dt2);

    TreeOperator(
       const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* dt1,
       const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* dt2,
       const HashTable< const DiscreteVariable*, Idx > givenContext);

    ~TreeOperator();

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

    MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* compute();


    private:
    NodeId __xPloreDT1(NodeId currentNodeId);

    NodeId __xPloreDT2(NodeId currentNodeId);

    NodeId __checkRedundancy(const DiscreteVariable*, NodeId*);

    const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* __dt1;
    const MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* __dt2;

    MultiDimFunctionGraph< GUM_SCALAR, TerminalNodePolicy >* __rd;

    const COMBINEOPERATOR< GUM_SCALAR > __combine;

    HashTable< const DiscreteVariable*, Idx > __context;
    NodeId                                    __curDT1Leaf;
  };

#ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
  extern template class TreeOperator< double, std::plus >;
#endif

}   // namespace gum

#include <agrum/multidim/utils/FunctionGraphUtilities/operators/treeOperator_tpl.h>

#endif   // GUM_OPERATOR_H