Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
neighbors.h 678 Bytes
Newer Older
Alexander Liao's avatar
python tests passing  
Alexander Liao committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 


#include "cloud.h"
#include "nanoflann.hpp"
#include <set>
#include <cstdint>

using namespace std;


template<typename scalar_t>
int nanoflann_neighbors(vector<scalar_t>& queries, vector<scalar_t>& supports,
			vector<long>& neighbors_indices, float radius, int dim, int max_num, int mode);


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,
			        float radius, int dim, int max_num, int mode);