Searcher.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_SEARCHER_HPP
#define KNNCOLLE_SEARCHER_HPP
 
#include <vector>
#include <stdexcept>
 
namespace knncolle {
 
template<typename Index_, typename Float_>
class Searcher {
public:
    virtual ~Searcher() = default;
    virtual void search(Index_ i, Index_ k, std::vector<Index_>* output_indices, std::vector<Float_>* output_distances) = 0;
 
    virtual void search(const Float_* query, Index_ k, std::vector<Index_>* output_indices, std::vector<Float_>* output_distances) = 0;
 
public:
    virtual bool can_search_all() const {
        return false;
    }
 
    virtual Index_ search_all([[maybe_unused]] Index_ i, [[maybe_unused]] Float_ distance, [[maybe_unused]] std::vector<Index_>* output_indices, [[maybe_unused]] std::vector<Float_>* output_distances) {
        throw std::runtime_error("distance-based searches not supported");
        return 0;
    }
 
    virtual Index_ search_all([[maybe_unused]] const Float_* query, [[maybe_unused]] Float_ distance, [[maybe_unused]] std::vector<Index_>* output_indices, [[maybe_unused]] std::vector<Float_>* output_distances) {
        throw std::runtime_error("distance-based searches not supported");
        return 0;
    }
};
 
}
 
#endif