1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#pragma once
#include "../common.hpp"
#include "../iterator.hpp"
namespace kel {
namespace lbm {
template<typename FieldSchema, typename Encode, typename T, uint64_t D>
saw::data<typename FieldSchema::StoredValueSchema> n_closest_read(const saw::data<sch::Ptr<FieldSchema>,Encode>& f, const saw::data<sch::Vector<T,D>>& frac_ind){
auto shift_frac_ind = frac_ind;
for(uint64_t i{0u}; i < D; ++i){
shift_frac_ind.at({{i}}) = shift_frac_ind.at({{i}}) + saw::data<T>{0.5};
if(shift_frac_ind.at({{i}}).get() < 0){
shift_frac_ind.at({{i}}) = {};
}
}
saw::data<sch::FixedArray<sch::UInt64,D>> shift_ind;
for(uint64_t i{0u}; i < D; ++i){
shift_ind.at({i}) = frac_ind.at({{i}}).template cast_to<sch::UInt64>();
}
return f.at(shift_ind);
}
template<typename FieldSchema, typename Encode, typename T, uint64_t D>
void n_closest_add(const saw::data<sch::Ptr<FieldSchema>,Encode>& f, const saw::data<sch::Vector<T,D>>& frac_ind, const saw::data<typename FieldSchema::StoredValueSchema>& val){
auto shift_frac_ind = frac_ind;
for(uint64_t i{0u}; i < D; ++i){
shift_frac_ind.at({{i}}) = shift_frac_ind.at({{i}}) + saw::data<T>{0.5};
if(shift_frac_ind.at({{i}}).get() < 0){
shift_frac_ind.at({{i}}) = {};
}
}
auto f_meta = f.meta();
saw::data<sch::FixedArray<sch::UInt64,D>> shift_ind;
for(uint64_t i{0u}; i < D; ++i){
shift_ind.at({i}) = frac_ind.at({{i}}).template cast_to<sch::UInt64>();
if(shift_ind.at({i}) < f_meta.at({i})){
shift_ind.at({i}) = f_meta.at({i}) - 1u;
}
}
auto& f_i = f.at(shift_ind);
f_i = f_i + val;
}
}
}
|
