classBayesNet.h

Go to the documentation of this file.

#ifndef GUM_CLASS_BAYESNET_H
#define GUM_CLASS_BAYESNET_H

#include <list>

#include <agrum/BN/IBayesNet.h>

#include <agrum/PRM/PRM.h>
namespace gum {
  namespace prm {

    // clang-format off
    // clang-format on
    template < typename GUM_SCALAR >
    class ClassBayesNet : public IBayesNet< GUM_SCALAR > {
      public:
      // ========================================================================
      // ========================================================================

      ClassBayesNet< GUM_SCALAR >(const PRMClass< GUM_SCALAR >& c);

      ClassBayesNet< GUM_SCALAR >(const ClassBayesNet< GUM_SCALAR >& from);

      ClassBayesNet< GUM_SCALAR >&
         operator=(const ClassBayesNet< GUM_SCALAR >& from);

      virtual ~ClassBayesNet< GUM_SCALAR >();

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

      virtual const Potential< GUM_SCALAR >& cpt(NodeId varId) const;

      virtual const VariableNodeMap& variableNodeMap() const;

      virtual const DiscreteVariable& variable(NodeId id) const;

      virtual NodeId nodeId(const DiscreteVariable& var) const;

      virtual NodeId idFromName(const std::string& name) const;

      virtual const DiscreteVariable&
         variableFromName(const std::string& name) const;

      const NodeProperty< Size >& modalities() const;

      // ===========================================================================
      // ===========================================================================
      virtual std::string toDot() const;

      private:
      HashTable< const DiscreteVariable*, const PRMClassElement< GUM_SCALAR >* >
         __varNodeMap;

      const PRMClassElement< GUM_SCALAR >& __get(NodeId id) const;

      const PRMClassElement< GUM_SCALAR >& __get(const std::string& name) const;

      const PRMClass< GUM_SCALAR >* __class;

      mutable NodeProperty< Size > __modalities;

      void __init(const PRMClass< GUM_SCALAR >& c);
    };


#ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
    extern template class ClassBayesNet< double >;
#endif


  } /* namespace prm */
} /* namespace gum */

#include <agrum/PRM/classBayesNet_tpl.h>

#endif /* GUM_CLASS_BAYESNET_H */