load_hnsw_prebuilt.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_HNSW_LOAD_PREBUILT_HPP
#define KNNCOLLE_HNSW_LOAD_PREBUILT_HPP
 
#include "utils.hpp"
#include "Hnsw.hpp"
 
#include <cstddef>
#include <stdexcept>
#include <string>
 
#include "knncolle/knncolle.hpp"
 
namespace knncolle_hnsw {
 
struct HnswPrebuiltTypes {
    knncolle::NumericType data;
};
 
inline HnswPrebuiltTypes load_hnsw_prebuilt_types(const std::filesystem::path& dir) {
    knncolle::NumericType type;
    knncolle::quick_load(dir / "TYPE", &type, 1);
    HnswPrebuiltTypes config;
    config.data = type;
    return config;
}
 
template<typename Index_, typename Data_, typename Distance_, typename HnswData_ = float>
auto load_hnsw_prebuilt(const std::filesystem::path& dir) {
    return new HnswPrebuilt<Index_, Data_, Distance_, HnswData_>(dir);
}
 
}
 
#endif