O3ClassFactory.h

Go to the documentation of this file.

#include <algorithm>
#include <memory>
#include <string>

#include <agrum/PRM/PRM.h>
#include <agrum/PRM/PRMFactory.h>
#include <agrum/PRM/o3prm/O3NameSolver.h>
#include <agrum/PRM/o3prm/O3prm.h>
#include <agrum/core/hashTable.h>
#include <agrum/core/set.h>

#ifndef GUM_PRM_O3PRM_O3CLASS_FACTORY_H
#  define GUM_PRM_O3PRM_O3CLASS_FACTORY_H

namespace gum {
  namespace prm {
    namespace o3prm {

      template < typename GUM_SCALAR >
      class O3ClassFactory {
        public:
        O3ClassFactory(PRM< GUM_SCALAR >&          prm,
                       O3PRM&                      o3_prm,
                       O3NameSolver< GUM_SCALAR >& solver,
                       ErrorsContainer&            errors);
        O3ClassFactory(const O3ClassFactory< GUM_SCALAR >& src);
        O3ClassFactory(O3ClassFactory< GUM_SCALAR >&& src);
        ~O3ClassFactory();
        O3ClassFactory< GUM_SCALAR >&
           operator=(const O3ClassFactory< GUM_SCALAR >& src);
        O3ClassFactory< GUM_SCALAR >&
           operator=(O3ClassFactory< GUM_SCALAR >&& src);

        void buildClasses();

        void buildImplementations();

        void buildParameters();

        void buildReferenceSlots();

        void declareAggregates();

        void completeAggregates();

        void declareAttributes();

        void completeAttributes();


        private:
        PRM< GUM_SCALAR >*          __prm;
        O3PRM*                      __o3_prm;
        O3NameSolver< GUM_SCALAR >* __solver;
        ErrorsContainer*            __errors;

        HashTable< std::string, gum::NodeId > __nameMap;
        HashTable< std::string, O3Class* >    __classMap;
        HashTable< NodeId, O3Class* >         __nodeMap;
        DAG                                   __dag;
        std::vector< O3Class* >               __o3Classes;

        void __setO3ClassCreationOrder();

        bool __checkAndAddNodesToDag();

        bool __checkAndAddArcsToDag();

        bool __checkO3Classes();

        bool __checkImplementation(O3Class& c);

        bool __checkImplementation(
           O3Class&                                    c,
           O3Label&                                    i,
           HashTable< std::string, O3Attribute* >&     attrMap,
           HashTable< std::string, O3Aggregate* >&     aggMap,
           HashTable< std::string, O3ReferenceSlot* >& refMap);

        bool __checkImplementation(O3Label& o3_type, const PRMType& type);

        bool __checkImplementation(
           O3Label& o3_type, const PRMClassElementContainer< GUM_SCALAR >& type);

        void __addParameters(PRMFactory< GUM_SCALAR >& factory, O3Class& c);


        void __addReferenceSlots(O3Class& c);


        bool __checkReferenceSlot(O3Class& c, O3ReferenceSlot& ref);

        void __declareAttribute(O3Class& c);

        bool __checkAttributeForDeclaration(O3Class& o3_c, O3Attribute& attr);

        void __completeAttribute(PRMFactory< GUM_SCALAR >& factory, O3Class& c);

        bool __checkAttributeForCompletion(const O3Class& o3_c, O3Attribute& attr);

        bool __checkParent(const PRMClass< GUM_SCALAR >& c, const O3Label& prnt);

        bool __checkLocalParent(const PRMClass< GUM_SCALAR >& c,
                                const O3Label&                prnt);

        bool __checkRemoteParent(const PRMClassElementContainer< GUM_SCALAR >& c,
                                 const O3Label& prnt);

        bool __checkRawCPT(const PRMClass< GUM_SCALAR >& c, O3RawCPT& attr);

        bool __checkRuleCPT(const PRMClass< GUM_SCALAR >& c, O3RuleCPT& attr);

        bool __checkLabelsNumber(const O3RuleCPT&         attr,
                                 const O3RuleCPT::O3Rule& rule);

        bool __checkLabelsValues(const PRMClass< GUM_SCALAR >& c,
                                 const O3RuleCPT&              attr,
                                 const O3RuleCPT::O3Rule&      rule);

        void __addParamsToForms(
           const HashTable< std::string, const PRMParameter< GUM_SCALAR >* >&
                              scope,
           O3RuleCPT::O3Rule& rule);

        bool __checkRuleCPTSumsTo1(const PRMClass< GUM_SCALAR >& c,
                                   const O3RuleCPT&              attr,
                                   const O3RuleCPT::O3Rule&      rule);


        const PRMClassElement< GUM_SCALAR >*
           __resolveSlotChain(const PRMClassElementContainer< GUM_SCALAR >& c,
                              const O3Label&                                chain);

        bool __checkSlotChainLink(const PRMClassElementContainer< GUM_SCALAR >& c,
                                  const O3Label&     chain,
                                  const std::string& s);

        void __declareAggregates(O3Class& c);
        void __completeAggregates(PRMFactory< GUM_SCALAR >& factory, O3Class& c);

        bool __checkAggregateForDeclaration(O3Class& o3class, O3Aggregate& agg);
        bool __checkAggregateForCompletion(O3Class& o3class, O3Aggregate& agg);

        const PRMType* __checkAggParents(O3Class& o3class, O3Aggregate& agg);

        bool __checkAggTypeLegality(O3Class& o3class, O3Aggregate& agg);


        bool __checkAggParameters(O3Class&       o3class,
                                  O3Aggregate&   agg,
                                  const PRMType* t);

        bool __checkParametersNumber(O3Aggregate& agg, Size n);

        bool __checkParameterValue(O3Aggregate& agg, const gum::prm::PRMType& t);

        // @}
      };

    }   // namespace o3prm
  }     // namespace prm
}   // namespace gum

// always include the implementation of the templates
#  include <agrum/PRM/o3prm/O3ClassFactory_tpl.h>


#  ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
extern template class gum::prm::o3prm::O3ClassFactory< double >;
#  endif


#endif   // GUM_PRM_O3PRM_O3CLASS_FACTORY_H