load_prebuilt.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_LOAD_PREBUILT_HPP
#define KNNCOLLE_LOAD_PREBUILT_HPP
 
#include <string>
#include <unordered_map>
#include <functional>
#include <fstream>
#include <memory>
 
#include "Prebuilt.hpp"
#include "Bruteforce.hpp"
#include "Vptree.hpp"
#include "L2Normalized.hpp"
 
namespace knncolle {
 
template<typename Index_, typename Data_, typename Distance_>
using LoadPrebuiltFunction = std::function<Prebuilt<Index_, Data_, Distance_>* (const std::string&)>;
 
template<typename Index_, typename Data_, typename Distance_>
auto default_prebuilt_registry() {
    std::unordered_map<std::string, LoadPrebuiltFunction<Index_, Data_, Distance_> > registry;
    registry["knncolle::Bruteforce"] = [](const std::string& prefix) -> Prebuilt<Index_, Data_, Distance_>* {
        return new BruteforcePrebuilt<Index_, Data_, Distance_, DistanceMetric<Data_, Distance_> >(prefix);
    };
    registry["knncolle::Vptree"] = [](const std::string& prefix) -> Prebuilt<Index_, Data_, Distance_>* {
        return new VptreePrebuilt<Index_, Data_, Distance_, DistanceMetric<Data_, Distance_> >(prefix);
    };
    return registry;
}
template<typename Index_, typename Data_, typename Distance_>
inline std::unordered_map<std::string, LoadPrebuiltFunction<Index_, Data_, Distance_> >& load_prebuilt_registry() {
    static std::unordered_map<std::string, LoadPrebuiltFunction<Index_, Data_, Distance_> > registry = default_prebuilt_registry<Index_, Data_, Distance_>();
    return registry;
}
 
template<typename Index_, typename Data_, typename Distance_>
Prebuilt<Index_, Data_, Distance_>* load_prebuilt_raw(const std::string& prefix) {
    const auto meth_path = prefix + "ALGORITHM";
    auto method = quick_load_as_string(meth_path);
 
    const auto& reg = load_prebuilt_registry<Index_, Data_, Distance_>(); 
    auto it = reg.find(method);
    if (it == reg.end()) {
        throw std::runtime_error("cannot find load_prebuilt method for '" + method + "' at '" + meth_path + "'");
    }
 
    return (it->second)(prefix);
}
 
template<typename Index_, typename Data_, typename Distance_>
std::unique_ptr<Prebuilt<Index_, Data_, Distance_> > load_prebuilt_unique(const std::string& prefix) {
    return std::unique_ptr<Prebuilt<Index_, Data_, Distance_> >(load_prebuilt_raw<Index_, Data_, Distance_>(prefix));
}
 
template<typename Index_, typename Data_, typename Distance_>
std::shared_ptr<Prebuilt<Index_, Data_, Distance_> > load_prebuilt_shared(const std::string& prefix) {
    return std::shared_ptr<Prebuilt<Index_, Data_, Distance_> >(load_prebuilt_raw<Index_, Data_, Distance_>(prefix));
}
 
struct L2NormalizedPrebuiltTypes {
    knncolle::NumericType normalized;
};
 
inline L2NormalizedPrebuiltTypes load_l2normalized_prebuilt_types(const std::string& prefix) {
    L2NormalizedPrebuiltTypes config;
    quick_load(prefix + "normalized", &(config.normalized), 1);
    return config;
}
 
template<typename Index_, typename Data_, typename Distance_, typename Normalized_>
Prebuilt<Index_, Data_, Distance_>* load_l2normalized_prebuilt(const std::string& prefix) {
    return new L2NormalizedPrebuilt<Index_, Data_, Distance_, Normalized_>(prefix);
}
 
}
 
#endif