classifier_factory.h

Go to the documentation of this file.

#ifndef META_CLASSIFIER_FACTORY_H_
#define META_CLASSIFIER_FACTORY_H_

#include "classify/classifier/classifier.h"
#include "util/factory.h"
#include "util/shim.h"

namespace cpptoml
{
class table;
}

namespace meta
{
namespace classify
{

class classifier_factory
    : public util::factory<classifier_factory, classifier,
                           const cpptoml::table&,
                           std::shared_ptr<index::forward_index>,
                           std::shared_ptr<index::inverted_index>>
{
    friend base_factory;

  private:
    classifier_factory();

    template <class Classifier>
    void reg();

    template <class Classifier>
    void reg_mi();
};

std::unique_ptr<classifier>
    make_classifier(const cpptoml::table& config,
                    std::shared_ptr<index::forward_index> idx,
                    std::shared_ptr<index::inverted_index> inv_idx = nullptr);

template <class Classifier>
std::unique_ptr<classifier>
    make_classifier(const cpptoml::table&,
                    std::shared_ptr<index::forward_index> idx)
{
    return make_unique<Classifier>(idx);
}

template <class Classifier>
std::unique_ptr<classifier>
    make_multi_index_classifier(const cpptoml::table&,
                                std::shared_ptr<index::forward_index> idx,
                                std::shared_ptr<index::inverted_index> inv_idx)
{
    return make_unique<Classifier>(idx, inv_idx);
}

template <class Classifier>
void register_classifier()
{
    // wrap the make_classifier function to make it appear like it takes
    // two indexes, when we really only care about one.
    classifier_factory::get().add(Classifier::id,
                                  [](const cpptoml::table& config,
                                     std::shared_ptr<index::forward_index> idx,
                                     std::shared_ptr<index::inverted_index>)
                                  {
        return make_classifier<Classifier>(config, std::move(idx));
    });
}

template <class Classifier>
void register_multi_index_classifier()
{
    classifier_factory::get().add(Classifier::id,
                                  make_multi_index_classifier<Classifier>);
}
}
}
#endif