casadi_interpn.hpp
1 //
2 // MIT No Attribution
3 //
4 // Copyright (C) 2010-2023 Joel Andersson, Joris Gillis, Moritz Diehl, KU Leuven.
5 //
6 // Permission is hereby granted, free of charge, to any person obtaining a copy of this
7 // software and associated documentation files (the "Software"), to deal in the Software
8 // without restriction, including without limitation the rights to use, copy, modify,
9 // merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
10 // permit persons to whom the Software is furnished to do so.
11 //
12 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
13 // INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
14 // PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
15 // HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
16 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
17 // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
18 //
19 
20 // SYMBOL "interpn"
21 template<typename T1>
22 void casadi_interpn(T1* res, casadi_int ndim, const T1* grid, const casadi_int* offset, const T1* values, const T1* x, const casadi_int* lookup_mode, casadi_int m, casadi_int* iw, T1* w) { // NOLINT(whitespace/line_length)
23  // Work vectors
24  T1* alpha;
25  casadi_int *index, *corner;
26  alpha = w; w += ndim;
27  index = iw; iw += ndim;
28  corner = iw; iw += ndim;
29  // Left index and fraction of interval
30  casadi_interpn_weights(ndim, grid, offset, x, alpha, index, lookup_mode);
31  // Loop over all corners, add contribution to output
32  casadi_clear_casadi_int(corner, ndim);
33  casadi_clear(res, m);
34  do {
35  T1* coeff = 0;
36  casadi_interpn_interpolate(res, ndim, offset, values,
37  alpha, index, corner, coeff, m);
38  } while (casadi_flip(corner, ndim));
39 }