load_annoy_prebuilt.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_ANNOY_LOAD_PREBUILT_HPP
#define KNNCOLLE_ANNOY_LOAD_PREBUILT_HPP
 
#include "utils.hpp"
#include "Annoy.hpp"
 
#include <cstddef>
#include <stdexcept>
#include <string>
#include <filesystem>
 
#include "knncolle/knncolle.hpp"
 
namespace knncolle_annoy {
 
struct AnnoyPrebuiltTypes {
    knncolle::NumericType index;
 
    knncolle::NumericType data;
 
    std::string distance;
};
 
inline AnnoyPrebuiltTypes load_annoy_prebuilt_types(const std::filesystem::path& dir) {
    knncolle::NumericType types[2];
    knncolle::quick_load(dir / "TYPES", types, 2);
 
    AnnoyPrebuiltTypes config;
    config.index = types[0];
    config.data = types[1];
    config.distance = knncolle::quick_load_as_string(dir / "DISTANCE");
 
    return config;
}
 
template<
    typename Index_,
    typename Data_,
    typename Distance_, 
    class AnnoyDistance_,
    typename AnnoyIndex_ = Index_,
    typename AnnoyData_ = float,
    class AnnoyRng_ = Annoy::Kiss64Random,
    class AnnoyThreadPolicy_ = Annoy::AnnoyIndexSingleThreadedBuildPolicy
>
auto load_annoy_prebuilt(const std::filesystem::path& dir) {
    std::size_t ndim;
    knncolle::quick_load(dir / "NUM_DIM", &ndim, 1);
    return new AnnoyPrebuilt<Index_, Data_, Distance_, AnnoyDistance_, AnnoyIndex_, AnnoyData_, AnnoyRng_, AnnoyThreadPolicy_>(dir, ndim);
}
 
}
 
#endif