NearestNeighbourSearch2D.h

Go to the documentation of this file.

#pragma once
 
#include <array>
#include <cassert>
#include <vector>
#include "hnswlib/hnswlib.h"
 
namespace krotos
{
struct Dataset2D
{
    std::vector<float> data;
    int index = 0;
    static constexpr int dim = 2;
};
 
class NearestNeighbourSearch2D
{
    // Hnswlib - fast approximate nearest neighbor search
    // https://github.com/nmslib/hnswlib
 
  public:
    NearestNeighbourSearch2D(int max_elements);
    ~NearestNeighbourSearch2D();
 
    void addDatasetItem(float x, float y);
    int knnQuery(float x, float y);
    std::vector<int> knnQuery(float x, float y, int k);
 
  private:
    Dataset2D dataset;
    std::unique_ptr<hnswlib::BruteforceSearch<float>> alg;
    hnswlib::SpaceInterface<float>* space;
 
    std::unique_ptr<hnswlib::L2Space> m_spaceInstance;
};
 
} // namespace krotos