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
|
// CXSparse/MATLAB/CSparse/cs_permute_mex: permute a sparse matrix
// CXSparse, Copyright (c) 2006-2022, Timothy A. Davis. All Rights Reserved.
// SPDX-License-Identifier: LGPL-2.1+
#include "cs_mex.h"
/* cs_permute: permute a sparse matrix */
void mexFunction
(
int nargout,
mxArray *pargout [ ],
int nargin,
const mxArray *pargin [ ]
)
{
int64_t ignore, *P, *Q, *Pinv, m, n ;
if (nargout > 1 || nargin != 3)
{
mexErrMsgTxt ("Usage: C = cs_permute(A,p,q)") ;
}
m = mxGetM (pargin [0]) ;
n = mxGetN (pargin [0]) ;
P = cs_dl_mex_get_int (m, pargin [1], &ignore, 1) ; /* get P */
Q = cs_dl_mex_get_int (n, pargin [2], &ignore, 1) ; /* get Q */
Pinv = cs_dl_pinv (P, m) ; /* P = Pinv' */
if (mxIsComplex (pargin [0]))
{
#ifndef NCOMPLEX
cs_cl Amatrix, *A, *C, *D ;
A = cs_cl_mex_get_sparse (&Amatrix, 0, pargin [0]) ; /* get A */
C = cs_cl_permute (A, Pinv, Q, 1) ; /* C = A(p,q) */
cs_cl_free (A->x) ;
D = cs_cl_transpose (C, 1) ; /* sort C via double transpose */
cs_cl_spfree (C) ;
C = cs_cl_transpose (D, 1) ;
cs_cl_spfree (D) ;
pargout [0] = cs_cl_mex_put_sparse (&C) ; /* return C */
#else
mexErrMsgTxt ("complex matrices not supported") ;
#endif
}
else
{
cs_dl Amatrix, *A, *C, *D ;
A = cs_dl_mex_get_sparse (&Amatrix, 0, 1, pargin [0]) ; /* get A */
C = cs_dl_permute (A, Pinv, Q, 1) ; /* C = A(p,q) */
D = cs_dl_transpose (C, 1) ; /* sort C via double transpose */
cs_dl_spfree (C) ;
C = cs_dl_transpose (D, 1) ;
cs_dl_spfree (D) ;
pargout [0] = cs_dl_mex_put_sparse (&C) ; /* return C */
}
cs_dl_free (Pinv) ;
cs_dl_free (P) ;
cs_dl_free (Q) ;
}
|
