KDTreeND.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 KDTreeND
{
    using myVectors = std::vector<std::vector<float>>;
    using myKDTree = KDTreeVectorOfVectorsAdaptor<myVectors, float, -1, metric_L2>;
 
  public:
    void addDatasetItem(const std::vector<float>& x);
    void buildIndex();
    void clear()
    {
        m_dataset.clear();
        m_kdTree.reset(nullptr);
    }
    std::vector<std::size_t> knnQuery(const std::vector<float>& x, std::size_t k) const;
 
  private:
    myVectors m_dataset;
    std::unique_ptr<myKDTree> m_kdTree;
};
 
} // namespace krotos