casadi_trans.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 "trans"
21 template<typename T1>
22 void casadi_trans(const T1* x, const casadi_int* sp_x, T1* y,
23  const casadi_int* sp_y, casadi_int* tmp) {
24  casadi_int ncol_x, nnz_x, ncol_y, k;
25  const casadi_int* row_x, *colind_y;
26  ncol_x = sp_x[1];
27  nnz_x = sp_x[2 + ncol_x];
28  row_x = sp_x + 2 + ncol_x+1;
29  ncol_y = sp_y[1];
30  colind_y = sp_y+2;
31  for (k=0; k<ncol_y; ++k) tmp[k] = colind_y[k];
32  for (k=0; k<nnz_x; ++k) {
33  y[tmp[row_x[k]]++] = x[k];
34  }
35 }