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
|
// CXSparse/MATLAB/CSparse/cs_etree_mex: elimination tree
// CXSparse, Copyright (c) 2006-2022, Timothy A. Davis. All Rights Reserved.
// SPDX-License-Identifier: LGPL-2.1+
#include "cs_mex.h"
/* cs_etree: elimination tree of A or A'*A */
void mexFunction
(
int nargout,
mxArray *pargout [ ],
int nargin,
const mxArray *pargin [ ]
)
{
cs_dl Amatrix, *A ;
int64_t n, *parent, *post ;
int ata ;
char mode [20] ;
if (nargout > 2 || nargin < 1 || nargin > 2)
{
mexErrMsgTxt ("Usage: [parent,post] = cs_etree(A,mode)") ;
}
ata = 0 ; /* get mode */
if (nargin > 1 && mxIsChar (pargin [1]))
{
mxGetString (pargin [1], mode, 8) ;
ata = (mode [0] == 'c') ;
}
A = cs_dl_mex_get_sparse (&Amatrix, !ata, 0, pargin [0]) ; /* get A */
n = A->n ;
parent = cs_dl_etree (A, ata) ; /* compute etree */
if (nargout > 1)
{
post = cs_dl_post (parent, n) ; /* postorder the etree*/
pargout [1] = cs_dl_mex_put_int (post, n, 1, 1) ; /* return post */
}
pargout [0] = cs_dl_mex_put_int (parent, n, 1, 1) ; /* return parent */
}
|
