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 Data_, typename Distance_>
class Searcher {
public:
    Searcher() = default;
    Searcher(Searcher&&) = default;
    Searcher(const Searcher&) = default;
    Searcher& operator=(Searcher&&) = default;
    Searcher& operator=(const Searcher&) = default;
    virtual ~Searcher() = default;
    virtual void search(Index_ i, Index_ k, std::vector<Index_>* output_indices, std::vector<Distance_>* output_distances) = 0;
 
    virtual void search(const Data_* query, Index_ k, std::vector<Index_>* output_indices, std::vector<Distance_>* output_distances) = 0;
 
public:
    virtual bool can_search_all() const {
        return false;
    }
 
    virtual Index_ search_all([[maybe_unused]] Index_ i, [[maybe_unused]] Distance_ distance, [[maybe_unused]] std::vector<Index_>* output_indices, [[maybe_unused]] std::vector<Distance_>* output_distances) {
        throw std::runtime_error("distance-based searches not supported");
        return 0;
    }
 
    virtual Index_ search_all([[maybe_unused]] const Data_* query, [[maybe_unused]] Distance_ distance, [[maybe_unused]] std::vector<Index_>* output_indices, [[maybe_unused]] std::vector<Distance_>* output_distances) {
        throw std::runtime_error("distance-based searches not supported");
        return 0;
    }
};
 
}
 
#endif