featurizer_factory.h

Go to the documentation of this file.

#ifndef META_ANALYZERS_TREE_FEATURIZER_FACTORY_H_
#define META_ANALYZERS_TREE_FEATURIZER_FACTORY_H_

#include "parser/analyzers/featurizers/tree_featurizer.h"
#include "util/factory.h"
#include "util/shim.h"

namespace cpptoml
{
class table;
}

namespace meta
{
namespace analyzers
{

class featurizer_factory
    : public util::factory<featurizer_factory, tree_featurizer>
{
    friend base_factory;

  private:
    featurizer_factory();

    template <class Analyzer>
    void register_featurizer();

    std::unordered_map<std::string, factory_method> methods_;
};

template <class Featurizer>
std::unique_ptr<tree_featurizer> make_featurizer()
{
    return make_unique<Featurizer>();
}

template <class Featurizer>
void register_featurizer()
{
    featurizer_factory::get().add(Featurizer::id, make_featurizer<Featurizer>);
}
}
}
#endif