utils.hpp

#ifndef KNNCOLLE_HNSW_UTILS_HPP
#define KNNCOLLE_HNSW_UTILS_HPP
 
#include "hnswlib/hnswalg.h"
 
#include <functional>
#include <string>
#include <type_traits>
#include <filesystem>
 
namespace knncolle_hnsw {
 
template<typename Input_>
using I = std::remove_cv_t<std::remove_reference_t<Input_> >;
template<class HnswData_>
std::function<void(const std::filesystem::path&)>& custom_save_for_hnsw_data() {
    static std::function<void(const std::filesystem::path&)> fun;
    return fun;
}
 
template<typename HnswData_>
std::function<void(const std::filesystem::path&, const hnswlib::SpaceInterface<HnswData_>*)>& custom_save_for_hnsw_distance() {
    static std::function<void(const std::filesystem::path&, const hnswlib::SpaceInterface<HnswData_>*)> fun;
    return fun;
}
 
template<typename Distance_>
std::function<void(const std::filesystem::path&, const std::function<Distance_(Distance_)>&)>& custom_save_for_hnsw_normalize() {
    static std::function<void(const std::filesystem::path&, const std::function<Distance_(Distance_)>&)> fun;
    return fun;
}
 
template<typename HnswData_>
std::function<hnswlib::SpaceInterface<HnswData_>*(const std::filesystem::path&, std::size_t)>& custom_load_for_hnsw_distance() {
    static std::function<hnswlib::SpaceInterface<HnswData_>*(const std::filesystem::path&, std::size_t)> fun;
    return fun;
}
 
template<typename Distance_>
std::function<std::function<Distance_(Distance_)>(const std::filesystem::path&)>& custom_load_for_hnsw_normalize() {
    static std::function<std::function<Distance_(Distance_)>(const std::filesystem::path&)> fun;
    return fun;
}
 
}
 
#endif