factory.h

Go to the documentation of this file.

 
#if !defined(VITA_PRIMITIVE_FACTORY_H)
#define      VITA_PRIMITIVE_FACTORY_H
 
#include <map>
 
#include "kernel/gp/symbol.h"
 
namespace vita
{
class symbol_factory
{
public:
  symbol_factory();
 
  std::unique_ptr<symbol> make(const std::string &, cvect = {0});
  std::unique_ptr<symbol> make(domain_t, int, int, category_t = 0);
 
  template<class> bool register_symbol(const std::string &, std::size_t);
  bool unregister_symbol(const std::string &);
 
  std::size_t args(const std::string &) const;
 
private:
  using build_func = std::unique_ptr<symbol> (*)(const cvect &);
 
  template<class T> static std::unique_ptr<symbol> build(const cvect &c)
  { return std::make_unique<T>(c); }
 
  // Private data members.
  struct build_info
  {
    build_func  make;
    std::size_t args;
  };
 
  std::map<std::string, build_info> factory_;
};
 
template<class T>
bool symbol_factory::register_symbol(const std::string &name, std::size_t n)
{
  return factory_.insert({name, build_info{build<T>, n}}).second;
}
 
}  // namespace vita
 
#endif  // include guard