neighbors.h 701 Bytes
Newer Older
Alexander Liao's avatar
Alexander Liao committed
1
2
3
4
5
6


#include "cloud.h"
#include "nanoflann.hpp"
#include <set>
#include <cstdint>
7
#include <thread>
Alexander Liao's avatar
Alexander Liao committed
8
9
10
11
12
13

using namespace std;


template<typename scalar_t>
int nanoflann_neighbors(vector<scalar_t>& queries, vector<scalar_t>& supports,
14
			vector<long>& neighbors_indices, double radius, int dim, int64_t max_num);
15
16
17
18
19
20
21

template<typename scalar_t>
int batch_nanoflann_neighbors (vector<scalar_t>& queries,
                               vector<scalar_t>& supports,
                               vector<long>& q_batches,
                               vector<long>& s_batches,
                               vector<long>& neighbors_indices,
22
                               double radius, int dim, int64_t max_num);