utils.hpp

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