KDTree.h

Go to the documentation of this file.

#pragma once
 
#include <cassert>
#include <cmath>
#include <iostream>
#include <vector>
 
#ifndef JUCE_WINDOWS
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Weverything"
#else
__pragma(warning(push)) __pragma(warning(disable : 4127 4267))
#endif
 
#include "nanoflann/nanoflann.hpp"
#include "nanoflann/KDTreeVectorOfVectorsAdaptor.h"
 
#ifdef JUCE_MAC
#pragma GCC diagnostic pop
#else
        __pragma(warning(pop))
#endif
 
using namespace nanoflann;
 
namespace krotos
{
 
class KDTree
{
    using myVectors = std::vector<std::vector<float>>;
    using myKDTree = KDTreeVectorOfVectorsAdaptor<myVectors, float, 3, metric_L2_Simple>;
 
  public:
    KDTree();
    ~KDTree();
    void addDatasetItem(float x, float y, float z = 0.f);
    void buildIndex();
    std::size_t knnQuery(float x, float y, float z);
    std::vector<std::size_t> knnQuery(float x, float y, float z, std::size_t k);
    std::vector<std::size_t> radiusSearch(float x, float y, float z, float search_radius);
    // void removePoint(std::size_t idx);//Requires KDTreeSingleIndexDynamicAdaptor
 
  private:
    myVectors dataset;
    std::unique_ptr<myKDTree> kdTree;
};
 
} // namespace krotos