ShaferShenoyInference.h

Go to the documentation of this file.

#ifndef GUM_SHAFER_SHENOY_INFERENCE_H
#define GUM_SHAFER_SHENOY_INFERENCE_H

#include <utility>

#include <agrum/core/math/math.h>
#include <agrum/BN/algorithms/barrenNodesFinder.h>
#include <agrum/BN/inference/tools/evidenceInference.h>
#include <agrum/BN/inference/tools/jointTargetedInference.h>
#include <agrum/agrum.h>
#include <agrum/graphs/algorithms/triangulations/defaultTriangulation.h>

namespace gum {


  // the function used to combine two tables
  template < typename GUM_SCALAR >
  INLINE static Potential< GUM_SCALAR >*
     SSNewmultiPotential(const Potential< GUM_SCALAR >& t1,
                         const Potential< GUM_SCALAR >& t2) {
    return new Potential< GUM_SCALAR >(t1 * t2);
  }

  // the function used to combine two tables
  template < typename GUM_SCALAR >
  INLINE static Potential< GUM_SCALAR >*
     SSNewprojPotential(const Potential< GUM_SCALAR >&        t1,
                        const Set< const DiscreteVariable* >& del_vars) {
    return new Potential< GUM_SCALAR >(t1.margSumOut(del_vars));
  }


  template < typename GUM_SCALAR >
  class ShaferShenoyInference
      : public JointTargetedInference< GUM_SCALAR >
      , public EvidenceInference< GUM_SCALAR > {
    public:
    // ############################################################################
    // ############################################################################

    explicit ShaferShenoyInference(
       const IBayesNet< GUM_SCALAR >* BN,
       FindBarrenNodesType barren_type = FindBarrenNodesType::FIND_BARREN_NODES,
       bool                use_binary_join_tree = true);

    ~ShaferShenoyInference() final;



    // ############################################################################
    // ############################################################################

    void setTriangulation(const Triangulation& new_triangulation);


    void setFindBarrenNodesType(FindBarrenNodesType type);


    const JoinTree* joinTree();


    const JunctionTree* junctionTree();

    GUM_SCALAR evidenceProbability();



    protected:
    void _onStateChanged() final{};

    void _onEvidenceAdded(const NodeId id, bool isHardEvidence) final;

    void _onEvidenceErased(const NodeId id, bool isHardEvidence) final;

    void _onAllEvidenceErased(bool contains_hard_evidence) final;

    void _onEvidenceChanged(const NodeId id, bool hasChangedSoftHard) final;


    void _onMarginalTargetAdded(const NodeId id) final;


    void _onMarginalTargetErased(const NodeId id) final;

    virtual void _onBayesNetChanged(const IBayesNet< GUM_SCALAR >* bn) final;


    void _onJointTargetAdded(const NodeSet& set) final;


    void _onJointTargetErased(const NodeSet& set) final;

    void _onAllMarginalTargetsAdded() final;

    void _onAllMarginalTargetsErased() final;

    void _onAllJointTargetsErased() final;

    void _onAllTargetsErased() final;


    void _updateOutdatedBNStructure() final;


    void _updateOutdatedBNPotentials() final;


    void _makeInference() final;



    const Potential< GUM_SCALAR >& _posterior(NodeId id) final;


    const Potential< GUM_SCALAR >& _jointPosterior(const NodeSet& set) final;

    const Potential< GUM_SCALAR >&
       _jointPosterior(const NodeSet& wanted_target,
                       const NodeSet& declared_target) final;

    Potential< GUM_SCALAR >* _unnormalizedJointPosterior(NodeId id) final;

    Potential< GUM_SCALAR >* _unnormalizedJointPosterior(const NodeSet& set) final;


    private:
    typedef Set< const Potential< GUM_SCALAR >* > __PotentialSet;
    typedef SetIteratorSafe< const Potential< GUM_SCALAR >* >
       __PotentialSetIterator;


    FindBarrenNodesType __barren_nodes_type;

    Potential< GUM_SCALAR >* (*__projection_op)(
       const Potential< GUM_SCALAR >&,
       const Set< const DiscreteVariable* >&){SSNewprojPotential};

    Potential< GUM_SCALAR >* (*__combination_op)(const Potential< GUM_SCALAR >&,
                                                 const Potential< GUM_SCALAR >&){
       SSNewmultiPotential};

    Triangulation* __triangulation;

    bool __use_binary_join_tree{true};


    UndiGraph __graph;

    JoinTree* __JT{nullptr};

    JunctionTree* __junctionTree{nullptr};


    bool __is_new_jt_needed{true};


    NodeSet __roots;

    HashTable< NodeId, NodeId > __node_to_clique;

    HashTable< NodeSet, NodeId > __joint_target_to_clique;


    NodeProperty< __PotentialSet > __clique_potentials;


    NodeProperty< const Potential< GUM_SCALAR >* > __clique_ss_potential;


    ArcProperty< __PotentialSet > __separator_potentials;


    ArcProperty< __PotentialSet > __created_potentials;


    NodeProperty< const Potential< GUM_SCALAR >* > __target_posteriors;


    HashTable< NodeSet, const Potential< GUM_SCALAR >* > __joint_target_posteriors;

    NodeProperty< GUM_SCALAR > __constants;


    ArcProperty< bool > __messages_computed;


    NodeProperty< const Potential< GUM_SCALAR >* > __node_to_soft_evidence;


    NodeProperty< const Potential< GUM_SCALAR >* > __hard_ev_projected_CPTs;

    NodeSet __hard_ev_nodes;

    enum EvidenceChangeType { EVIDENCE_ADDED, EVIDENCE_ERASED, EVIDENCE_MODIFIED };

    NodeProperty< EvidenceChangeType > __evidence_changes;

    const GUM_SCALAR __1_minus_epsilon{GUM_SCALAR(1.0 - 1e-6)};


    bool __isNewJTNeeded() const;

    void __createNewJT();
    void __setProjectionFunction(Potential< GUM_SCALAR >* (*proj)(
       const Potential< GUM_SCALAR >&, const Set< const DiscreteVariable* >&));

    void __setCombinationFunction(Potential< GUM_SCALAR >* (*comb)(
       const Potential< GUM_SCALAR >&, const Potential< GUM_SCALAR >&));

    void __diffuseMessageInvalidations(NodeId   from,
                                       NodeId   to,
                                       NodeSet& cliques_invalidated);

    void __invalidateAllMessages();

    void __computeJoinTreeRoots();

    // remove barren variables and return the newly created projected potentials
    __PotentialSet
       __removeBarrenVariables(__PotentialSet&                 pot_list,
                               Set< const DiscreteVariable* >& del_vars);

    __PotentialSet __marginalizeOut(__PotentialSet                  pot_list,
                                    Set< const DiscreteVariable* >& del_vars,
                                    Set< const DiscreteVariable* >& kept_vars);

    void __produceMessage(NodeId from_id, NodeId to_id);

    void __collectMessage(NodeId id, NodeId from);

    ShaferShenoyInference(const ShaferShenoyInference< GUM_SCALAR >&);

    ShaferShenoyInference< GUM_SCALAR >&
       operator=(const ShaferShenoyInference< GUM_SCALAR >&);
  };


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


} /* namespace gum */

#include <agrum/BN/inference/ShaferShenoyInference_tpl.h>

#endif /* SHAFER_SHENOY_INFERENCE_H */