ElectionFactoryRepository.h

Go to the documentation of this file.

#ifndef   _GAMS_ELECTIONS_ELECTION_FACTORY_REPOSITORY_H_
#define   _GAMS_ELECTIONS_ELECTION_FACTORY_REPOSITORY_H_

#include <vector>
#include <string>
#include <map>

#include "madara/knowledge/KnowledgeBase.h"
#include "ElectionFactory.h"

namespace gams
{
  namespace elections
  {
    typedef std::map <ElectionType, ElectionFactory *>  ElectionFactoryMap;

    class GAMS_EXPORT ElectionFactoryRepository
    {
    public:
      ElectionFactoryRepository (
        const std::string & agent_prefix,
        madara::knowledge::KnowledgeBase * knowledge = 0);

      virtual ~ElectionFactoryRepository ();

      void add (ElectionType type,
        ElectionFactory * factory);

      ElectionBase * create (const std::string & election_prefix);

      ElectionBase * create (ElectionType type);

      void set_knowledge (madara::knowledge::KnowledgeBase * knowledge);

      void set_agent_prefix (const std::string & prefix);

    protected:

      void init (void);

      madara::knowledge::KnowledgeBase * knowledge_;

      std::string agent_prefix_;

      ElectionFactoryMap factory_map_;

    };
  }
}

#include "ElectionFactoryRepository.inl"

#endif // _GAMS_ELECTIONS_ELECTION_FACTORY_REPOSITORY_H_