utils.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_ANNOY_UTILS_HPP
#define KNNCOLLE_ANNOY_UTILS_HPP
 
#include <string>
#include <cstddef>
#include <stdexcept>
#include <type_traits>
#include <functional>
#include <filesystem>
 
#include "knncolle/knncolle.hpp"
 
namespace knncolle_annoy {
 
template<typename Input_>
using I = std::remove_cv_t<std::remove_reference_t<Input_> >;
 
// Doing some SFINAE nonsense.
template<typename AnnoyDistance_, typename Other_ = int> 
struct has_name {
    static constexpr bool value = false;
};
 
template<typename AnnoyDistance_> 
struct has_name<AnnoyDistance_, decltype(AnnoyDistance_::name(), 0)> {
    static constexpr bool value = std::is_same<decltype(AnnoyDistance_::name()), const char*>::value;
};
template<typename AnnoyDistance_> 
const char* get_distance_name() { 
    if constexpr(!has_name<AnnoyDistance_>::value) {
        return "unknown";
    } else {
        return AnnoyDistance_::name();
    }
}
 
template<class AnnoyIndex_>
std::function<void(const std::filesystem::path&)>& custom_save_for_annoy_index() {
    static std::function<void(const std::filesystem::path&)> fun;
    return fun;
}
 
template<class AnnoyData_>
std::function<void(const std::filesystem::path&)>& custom_save_for_annoy_data() {
    static std::function<void(const std::filesystem::path&)> fun;
    return fun;
}
 
template<class AnnoyDistance_>
std::function<void(const std::filesystem::path&)>& custom_save_for_annoy_distance() {
    static std::function<void(const std::filesystem::path&)> fun;
    return fun;
}
 
}
 
#endif