casadi_tri_project.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 "tri_project"
21 template<typename T1>
22 void casadi_tri_project(const T1* x, const casadi_int* sp_x, T1* y, casadi_int lower) {
23  casadi_int ncol_x, j, k;
24  const casadi_int *colind_x, *row_x;
25  ncol_x = sp_x[1];
26  colind_x = sp_x+2; row_x = sp_x + 2 + ncol_x+1;
27 
28  for (j=0; j<ncol_x; ++j) {
29  for (k=colind_x[j]; k<colind_x[j+1]; ++k) {
30  if (lower) {
31  if (row_x[k]>=j) *y++ = x ? x[k] : 0;
32  } else {
33  if (row_x[k]<=j) *y++ = x ? x[k] : 0;
34  }
35  }
36  }
37 }