multipleInferenceEngine.h

Go to the documentation of this file.

#ifndef __MULTIPLE_INFERENCE_ENGINES__H__
#define __MULTIPLE_INFERENCE_ENGINES__H__


#include <agrum/BN/inference/lazyPropagation.h>
#include <agrum/CN/inferenceEngine.h>

namespace gum {
  namespace credal {

    template < typename GUM_SCALAR, class BNInferenceEngine >
    class MultipleInferenceEngine : public InferenceEngine< GUM_SCALAR > {
      private:
      typedef InferenceEngine< GUM_SCALAR > __infE;

      typedef NodeProperty< std::vector< NodeId > >                    __cluster;
      typedef NodeProperty< std::vector< std::vector< GUM_SCALAR > > > __credalSet;
      typedef NodeProperty< std::vector< GUM_SCALAR > >                __margi;
      typedef NodeProperty< GUM_SCALAR >                               __expe;

      typedef IBayesNet< GUM_SCALAR >                 __bnet;
      typedef std::vector< __margi >                  __margis;
      typedef std::vector< __expe >                   __expes;
      typedef std::vector< __credalSet >              __credalSets;
      typedef std::vector< std::vector< __cluster > > __clusters;

      typedef typename std::vector<
         HashTable< std::string, std::vector< GUM_SCALAR > > >
         __modals;

      inline void __updateThreadCredalSets(const NodeId&                    id,
                                           const std::vector< GUM_SCALAR >& vertex,
                                           const bool& elimRedund);

      protected:
      __margis _l_marginalMin;
      __margis _l_marginalMax;
      __expes _l_expectationMin;
      __expes _l_expectationMax;
      __modals _l_modal;
      __credalSets _l_marginalSets;
      __margis _l_evidence;
      __clusters _l_clusters;

      typename std::vector< __bnet* > _workingSet;
      typename std::vector< List< const Potential< GUM_SCALAR >* >* > _workingSetE;

      typename std::vector< BNInferenceEngine* > _l_inferenceEngine;
      std::vector< VarMod2BNsMap< GUM_SCALAR >* > _l_optimalNet;
      // OptBN< GUM_SCALAR > _threadFusion; // we should use this OptBN if omp
      // is
      // disabled (avoid creating 2 objects when only one is necessary)
      // it should also avoid calling thread fusion operations


      void _initThreadsData(const Size& num_threads,
                            const bool  __storeVertices,
                            const bool  __storeBNOpt);



      inline bool _updateThread(const NodeId&                    id,
                                const std::vector< GUM_SCALAR >& vertex,
                                const bool& elimRedund = false);

      inline void _updateMarginals();

      inline const GUM_SCALAR _computeEpsilon();

      void _updateOldMarginals();



      void _optFusion();
      void _expFusion();
      void _verticesFusion();   // called ?? not done yet


      public:

      explicit MultipleInferenceEngine(const CredalNet< GUM_SCALAR >& credalNet);

      virtual ~MultipleInferenceEngine();



      virtual void eraseAllEvidence();


      virtual void makeInference() = 0;
    };


#ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
    extern template class MultipleInferenceEngine<
       double,
       gum::LazyPropagation< double > >;

    extern template class MultipleInferenceEngine< float,
                                                   gum::LazyPropagation< float > >;
#endif

  }   // namespace credal
}   // namespace gum

#include <agrum/CN/multipleInferenceEngine_tpl.h>

#endif