KDTreeND.cpp

Go to the documentation of this file.

namespace krotos
{
 
void KDTreeND::addDatasetItem(const std::vector<float>& x) { m_dataset.push_back(x); }
 
void KDTreeND::buildIndex()
{
    if (m_dataset.empty())
    {
        jassertfalse;
        return;
    }
 
    auto dims = m_dataset.front().size();
    m_kdTree = std::make_unique<myKDTree>(dims, m_dataset, 10);
}
 
std::vector<std::size_t> KDTreeND::knnQuery(const std::vector<float>& x, std::size_t k) const
{
    jassert(m_kdTree);
 
    std::vector<std::size_t> ret_indexes(k);
    std::vector<float> out_dists_sqr(k);
 
    nanoflann::KNNResultSet<float> resultSet(k);
 
    resultSet.init(&ret_indexes[0], &out_dists_sqr[0]);
    m_kdTree->index->findNeighbors(resultSet, &x[0]);
 
    return ret_indexes;
}
 
} // namespace krotos