/* * CXSPARSE: a Concise Sparse Matrix package - Extended. * Copyright (c) 2006-2009, Timothy A. Davis. * http://www.cise.ufl.edu/research/sparse/CXSparse * * CXSparse is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * CXSparse is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this Module; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef _CXS_H #define _CXS_H #include #include #include #include #ifdef MATLAB_MEX_FILE #include "mex.h" #endif #ifdef __cplusplus #ifndef NCOMPLEX #include typedef std::complex cs_complex_t ; #endif extern "C" { #else #ifndef NCOMPLEX #include #define cs_complex_t double _Complex #endif #endif #define CS_VER 2 /* CXSparse Version 2.2.3 */ #define CS_SUBVER 2 #define CS_SUBSUB 3 #define CS_DATE "Mar 24, 2009" /* CXSparse release date */ #define CS_COPYRIGHT "Copyright (c) Timothy A. Davis, 2006-2009" #define CXSPARSE /* define UF_long */ #include "UFconfig.h" /* -------------------------------------------------------------------------- */ /* double/int version of CXSparse */ /* -------------------------------------------------------------------------- */ /* --- primary CSparse routines and data structures ------------------------- */ typedef struct cs_di_sparse /* matrix in compressed-column or triplet form */ { int nzmax ; /* maximum number of entries */ int m ; /* number of rows */ int n ; /* number of columns */ int *p ; /* column pointers (size n+1) or col indices (size nzmax) */ int *i ; /* row indices, size nzmax */ double *x ; /* numerical values, size nzmax */ int nz ; /* # of entries in triplet matrix, -1 for compressed-col */ } cs_di ; cs_di *cs_di_add (const cs_di *A, const cs_di *B, double alpha, double beta) ; int cs_di_cholsol (int order, const cs_di *A, double *b) ; int cs_di_dupl (cs_di *A) ; int cs_di_entry (cs_di *T, int i, int j, double x) ; int cs_di_lusol (int order, const cs_di *A, double *b, double tol) ; int cs_di_gaxpy (const cs_di *A, const double *x, double *y) ; cs_di *cs_di_multiply (const cs_di *A, const cs_di *B) ; int cs_di_qrsol (int order, const cs_di *A, double *b) ; cs_di *cs_di_transpose (const cs_di *A, int values) ; cs_di *cs_di_compress (const cs_di *T) ; double cs_di_norm (const cs_di *A) ; int cs_di_print (const cs_di *A, int brief) ; cs_di *cs_di_load (FILE *f) ; /* utilities */ void *cs_di_calloc (int n, size_t size) ; void *cs_di_free (void *p) ; void *cs_di_realloc (void *p, int n, size_t size, int *ok) ; cs_di *cs_di_spalloc (int m, int n, int nzmax, int values, int t) ; cs_di *cs_di_spfree (cs_di *A) ; int cs_di_sprealloc (cs_di *A, int nzmax) ; void *cs_di_malloc (int n, size_t size) ; /* --- secondary CSparse routines and data structures ----------------------- */ typedef struct cs_di_symbolic /* symbolic Cholesky, LU, or QR analysis */ { int *pinv ; /* inverse row perm. for QR, fill red. perm for Chol */ int *q ; /* fill-reducing column permutation for LU and QR */ int *parent ; /* elimination tree for Cholesky and QR */ int *cp ; /* column pointers for Cholesky, row counts for QR */ int *leftmost ; /* leftmost[i] = min(find(A(i,:))), for QR */ int m2 ; /* # of rows for QR, after adding fictitious rows */ double lnz ; /* # entries in L for LU or Cholesky; in V for QR */ double unz ; /* # entries in U for LU; in R for QR */ } cs_dis ; typedef struct cs_di_numeric /* numeric Cholesky, LU, or QR factorization */ { cs_di *L ; /* L for LU and Cholesky, V for QR */ cs_di *U ; /* U for LU, r for QR, not used for Cholesky */ int *pinv ; /* partial pivoting for LU */ double *B ; /* beta [0..n-1] for QR */ } cs_din ; typedef struct cs_di_dmperm_results /* cs_di_dmperm or cs_di_scc output */ { int *p ; /* size m, row permutation */ int *q ; /* size n, column permutation */ int *r ; /* size nb+1, block k is rows r[k] to r[k+1]-1 in A(p,q) */ int *s ; /* size nb+1, block k is cols s[k] to s[k+1]-1 in A(p,q) */ int nb ; /* # of blocks in fine dmperm decomposition */ int rr [5] ; /* coarse row decomposition */ int cc [5] ; /* coarse column decomposition */ } cs_did ; int *cs_di_amd (int order, const cs_di *A) ; cs_din *cs_di_chol (const cs_di *A, const cs_dis *S) ; cs_did *cs_di_dmperm (const cs_di *A, int seed) ; int cs_di_droptol (cs_di *A, double tol) ; int cs_di_dropzeros (cs_di *A) ; int cs_di_happly (const cs_di *V, int i, double beta, double *x) ; int cs_di_ipvec (const int *p, const double *b, double *x, int n) ; int cs_di_lsolve (const cs_di *L, double *x) ; int cs_di_ltsolve (const cs_di *L, double *x) ; cs_din *cs_di_lu (const cs_di *A, const cs_dis *S, double tol) ; cs_di *cs_di_permute (const cs_di *A, const int *pinv, const int *q, int values) ; int *cs_di_pinv (const int *p, int n) ; int cs_di_pvec (const int *p, const double *b, double *x, int n) ; cs_din *cs_di_qr (const cs_di *A, const cs_dis *S) ; cs_dis *cs_di_schol (int order, const cs_di *A) ; cs_dis *cs_di_sqr (int order, const cs_di *A, int qr) ; cs_di *cs_di_symperm (const cs_di *A, const int *pinv, int values) ; int cs_di_usolve (const cs_di *U, double *x) ; int cs_di_utsolve (const cs_di *U, double *x) ; int cs_di_updown (cs_di *L, int sigma, const cs_di *C, const int *parent) ; /* utilities */ cs_dis *cs_di_sfree (cs_dis *S) ; cs_din *cs_di_nfree (cs_din *N) ; cs_did *cs_di_dfree (cs_did *D) ; /* --- tertiary CSparse routines -------------------------------------------- */ int *cs_di_counts (const cs_di *A, const int *parent, const int *post, int ata) ; double cs_di_cumsum (int *p, int *c, int n) ; int cs_di_dfs (int j, cs_di *G, int top, int *xi, int *pstack, const int *pinv) ; int *cs_di_etree (const cs_di *A, int ata) ; int cs_di_fkeep (cs_di *A, int (*fkeep) (int, int, double, void *), void *other) ; double cs_di_house (double *x, double *beta, int n) ; int *cs_di_maxtrans (const cs_di *A, int seed) ; int *cs_di_post (const int *parent, int n) ; cs_did *cs_di_scc (cs_di *A) ; int cs_di_scatter (const cs_di *A, int j, double beta, int *w, double *x, int mark, cs_di *C, int nz) ; int cs_di_tdfs (int j, int k, int *head, const int *next, int *post, int *stack) ; int cs_di_leaf (int i, int j, const int *first, int *maxfirst, int *prevleaf, int *ancestor, int *jleaf) ; int cs_di_reach (cs_di *G, const cs_di *B, int k, int *xi, const int *pinv) ; int cs_di_spsolve (cs_di *L, const cs_di *B, int k, int *xi, double *x, const int *pinv, int lo) ; int cs_di_ereach (const cs_di *A, int k, const int *parent, int *s, int *w) ; int *cs_di_randperm (int n, int seed) ; /* utilities */ cs_did *cs_di_dalloc (int m, int n) ; cs_di *cs_di_done (cs_di *C, void *w, void *x, int ok) ; int *cs_di_idone (int *p, cs_di *C, void *w, int ok) ; cs_din *cs_di_ndone (cs_din *N, cs_di *C, void *w, void *x, int ok) ; cs_did *cs_di_ddone (cs_did *D, cs_di *C, void *w, int ok) ; /* -------------------------------------------------------------------------- */ /* double/UF_long version of CXSparse */ /* -------------------------------------------------------------------------- */ /* --- primary CSparse routines and data structures ------------------------- */ typedef struct cs_dl_sparse /* matrix in compressed-column or triplet form */ { UF_long nzmax ; /* maximum number of entries */ UF_long m ; /* number of rows */ UF_long n ; /* number of columns */ UF_long *p ; /* column pointers (size n+1) or col indlces (size nzmax) */ UF_long *i ; /* row indices, size nzmax */ double *x ; /* numerical values, size nzmax */ UF_long nz ; /* # of entries in triplet matrix, -1 for compressed-col */ } cs_dl ; cs_dl *cs_dl_add (const cs_dl *A, const cs_dl *B, double alpha, double beta) ; UF_long cs_dl_cholsol (UF_long order, const cs_dl *A, double *b) ; UF_long cs_dl_dupl (cs_dl *A) ; UF_long cs_dl_entry (cs_dl *T, UF_long i, UF_long j, double x) ; UF_long cs_dl_lusol (UF_long order, const cs_dl *A, double *b, double tol) ; UF_long cs_dl_gaxpy (const cs_dl *A, const double *x, double *y) ; cs_dl *cs_dl_multiply (const cs_dl *A, const cs_dl *B) ; UF_long cs_dl_qrsol (UF_long order, const cs_dl *A, double *b) ; cs_dl *cs_dl_transpose (const cs_dl *A, UF_long values) ; cs_dl *cs_dl_compress (const cs_dl *T) ; double cs_dl_norm (const cs_dl *A) ; UF_long cs_dl_print (const cs_dl *A, UF_long brief) ; cs_dl *cs_dl_load (FILE *f) ; /* utilities */ void *cs_dl_calloc (UF_long n, size_t size) ; void *cs_dl_free (void *p) ; void *cs_dl_realloc (void *p, UF_long n, size_t size, UF_long *ok) ; cs_dl *cs_dl_spalloc (UF_long m, UF_long n, UF_long nzmax, UF_long values, UF_long t) ; cs_dl *cs_dl_spfree (cs_dl *A) ; UF_long cs_dl_sprealloc (cs_dl *A, UF_long nzmax) ; void *cs_dl_malloc (UF_long n, size_t size) ; /* --- secondary CSparse routines and data structures ----------------------- */ typedef struct cs_dl_symbolic /* symbolic Cholesky, LU, or QR analysis */ { UF_long *pinv ; /* inverse row perm. for QR, fill red. perm for Chol */ UF_long *q ; /* fill-reducing column permutation for LU and QR */ UF_long *parent ; /* elimination tree for Cholesky and QR */ UF_long *cp ; /* column pointers for Cholesky, row counts for QR */ UF_long *leftmost ; /* leftmost[i] = min(find(A(i,:))), for QR */ UF_long m2 ; /* # of rows for QR, after adding fictitious rows */ double lnz ; /* # entries in L for LU or Cholesky; in V for QR */ double unz ; /* # entries in U for LU; in R for QR */ } cs_dls ; typedef struct cs_dl_numeric /* numeric Cholesky, LU, or QR factorization */ { cs_dl *L ; /* L for LU and Cholesky, V for QR */ cs_dl *U ; /* U for LU, r for QR, not used for Cholesky */ UF_long *pinv ; /* partial pivoting for LU */ double *B ; /* beta [0..n-1] for QR */ } cs_dln ; typedef struct cs_dl_dmperm_results /* cs_dl_dmperm or cs_dl_scc output */ { UF_long *p ; /* size m, row permutation */ UF_long *q ; /* size n, column permutation */ UF_long *r ; /* size nb+1, block k is rows r[k] to r[k+1]-1 in A(p,q) */ UF_long *s ; /* size nb+1, block k is cols s[k] to s[k+1]-1 in A(p,q) */ UF_long nb ; /* # of blocks in fine dmperm decomposition */ UF_long rr [5] ; /* coarse row decomposition */ UF_long cc [5] ; /* coarse column decomposition */ } cs_dld ; UF_long *cs_dl_amd (UF_long order, const cs_dl *A) ; cs_dln *cs_dl_chol (const cs_dl *A, const cs_dls *S) ; cs_dld *cs_dl_dmperm (const cs_dl *A, UF_long seed) ; UF_long cs_dl_droptol (cs_dl *A, double tol) ; UF_long cs_dl_dropzeros (cs_dl *A) ; UF_long cs_dl_happly (const cs_dl *V, UF_long i, double beta, double *x) ; UF_long cs_dl_ipvec (const UF_long *p, const double *b, double *x, UF_long n) ; UF_long cs_dl_lsolve (const cs_dl *L, double *x) ; UF_long cs_dl_ltsolve (const cs_dl *L, double *x) ; cs_dln *cs_dl_lu (const cs_dl *A, const cs_dls *S, double tol) ; cs_dl *cs_dl_permute (const cs_dl *A, const UF_long *pinv, const UF_long *q, UF_long values) ; UF_long *cs_dl_pinv (const UF_long *p, UF_long n) ; UF_long cs_dl_pvec (const UF_long *p, const double *b, double *x, UF_long n) ; cs_dln *cs_dl_qr (const cs_dl *A, const cs_dls *S) ; cs_dls *cs_dl_schol (UF_long order, const cs_dl *A) ; cs_dls *cs_dl_sqr (UF_long order, const cs_dl *A, UF_long qr) ; cs_dl *cs_dl_symperm (const cs_dl *A, const UF_long *pinv, UF_long values) ; UF_long cs_dl_usolve (const cs_dl *U, double *x) ; UF_long cs_dl_utsolve (const cs_dl *U, double *x) ; UF_long cs_dl_updown (cs_dl *L, UF_long sigma, const cs_dl *C, const UF_long *parent) ; /* utilities */ cs_dls *cs_dl_sfree (cs_dls *S) ; cs_dln *cs_dl_nfree (cs_dln *N) ; cs_dld *cs_dl_dfree (cs_dld *D) ; /* --- tertiary CSparse routines -------------------------------------------- */ UF_long *cs_dl_counts (const cs_dl *A, const UF_long *parent, const UF_long *post, UF_long ata) ; double cs_dl_cumsum (UF_long *p, UF_long *c, UF_long n) ; UF_long cs_dl_dfs (UF_long j, cs_dl *G, UF_long top, UF_long *xi, UF_long *pstack, const UF_long *pinv) ; UF_long *cs_dl_etree (const cs_dl *A, UF_long ata) ; UF_long cs_dl_fkeep (cs_dl *A, UF_long (*fkeep) (UF_long, UF_long, double, void *), void *other) ; double cs_dl_house (double *x, double *beta, UF_long n) ; UF_long *cs_dl_maxtrans (const cs_dl *A, UF_long seed) ; UF_long *cs_dl_post (const UF_long *parent, UF_long n) ; cs_dld *cs_dl_scc (cs_dl *A) ; UF_long cs_dl_scatter (const cs_dl *A, UF_long j, double beta, UF_long *w, double *x, UF_long mark,cs_dl *C, UF_long nz) ; UF_long cs_dl_tdfs (UF_long j, UF_long k, UF_long *head, const UF_long *next, UF_long *post, UF_long *stack) ; UF_long cs_dl_leaf (UF_long i, UF_long j, const UF_long *first, UF_long *maxfirst, UF_long *prevleaf, UF_long *ancestor, UF_long *jleaf) ; UF_long cs_dl_reach (cs_dl *G, const cs_dl *B, UF_long k, UF_long *xi, const UF_long *pinv) ; UF_long cs_dl_spsolve (cs_dl *L, const cs_dl *B, UF_long k, UF_long *xi, double *x, const UF_long *pinv, UF_long lo) ; UF_long cs_dl_ereach (const cs_dl *A, UF_long k, const UF_long *parent, UF_long *s, UF_long *w) ; UF_long *cs_dl_randperm (UF_long n, UF_long seed) ; /* utilities */ cs_dld *cs_dl_dalloc (UF_long m, UF_long n) ; cs_dl *cs_dl_done (cs_dl *C, void *w, void *x, UF_long ok) ; UF_long *cs_dl_idone (UF_long *p, cs_dl *C, void *w, UF_long ok) ; cs_dln *cs_dl_ndone (cs_dln *N, cs_dl *C, void *w, void *x, UF_long ok) ; cs_dld *cs_dl_ddone (cs_dld *D, cs_dl *C, void *w, UF_long ok) ; /* -------------------------------------------------------------------------- */ /* complex/int version of CXSparse */ /* -------------------------------------------------------------------------- */ #ifndef NCOMPLEX /* --- primary CSparse routines and data structures ------------------------- */ typedef struct cs_ci_sparse /* matrix in compressed-column or triplet form */ { int nzmax ; /* maximum number of entries */ int m ; /* number of rows */ int n ; /* number of columns */ int *p ; /* column pointers (size n+1) or col indices (size nzmax) */ int *i ; /* row indices, size nzmax */ cs_complex_t *x ; /* numerical values, size nzmax */ int nz ; /* # of entries in triplet matrix, -1 for compressed-col */ } cs_ci ; cs_ci *cs_ci_add (const cs_ci *A, const cs_ci *B, cs_complex_t alpha, cs_complex_t beta) ; int cs_ci_cholsol (int order, const cs_ci *A, cs_complex_t *b) ; int cs_ci_dupl (cs_ci *A) ; int cs_ci_entry (cs_ci *T, int i, int j, cs_complex_t x) ; int cs_ci_lusol (int order, const cs_ci *A, cs_complex_t *b, double tol) ; int cs_ci_gaxpy (const cs_ci *A, const cs_complex_t *x, cs_complex_t *y) ; cs_ci *cs_ci_multiply (const cs_ci *A, const cs_ci *B) ; int cs_ci_qrsol (int order, const cs_ci *A, cs_complex_t *b) ; cs_ci *cs_ci_transpose (const cs_ci *A, int values) ; cs_ci *cs_ci_compress (const cs_ci *T) ; double cs_ci_norm (const cs_ci *A) ; int cs_ci_print (const cs_ci *A, int brief) ; cs_ci *cs_ci_load (FILE *f) ; /* utilities */ void *cs_ci_calloc (int n, size_t size) ; void *cs_ci_free (void *p) ; void *cs_ci_realloc (void *p, int n, size_t size, int *ok) ; cs_ci *cs_ci_spalloc (int m, int n, int nzmax, int values, int t) ; cs_ci *cs_ci_spfree (cs_ci *A) ; int cs_ci_sprealloc (cs_ci *A, int nzmax) ; void *cs_ci_malloc (int n, size_t size) ; /* --- secondary CSparse routines and data structures ----------------------- */ typedef struct cs_ci_symbolic /* symbolic Cholesky, LU, or QR analysis */ { int *pinv ; /* inverse row perm. for QR, fill red. perm for Chol */ int *q ; /* fill-reducing column permutation for LU and QR */ int *parent ; /* elimination tree for Cholesky and QR */ int *cp ; /* column pointers for Cholesky, row counts for QR */ int *leftmost ; /* leftmost[i] = min(find(A(i,:))), for QR */ int m2 ; /* # of rows for QR, after adding fictitious rows */ double lnz ; /* # entries in L for LU or Cholesky; in V for QR */ double unz ; /* # entries in U for LU; in R for QR */ } cs_cis ; typedef struct cs_ci_numeric /* numeric Cholesky, LU, or QR factorization */ { cs_ci *L ; /* L for LU and Cholesky, V for QR */ cs_ci *U ; /* U for LU, r for QR, not used for Cholesky */ int *pinv ; /* partial pivoting for LU */ double *B ; /* beta [0..n-1] for QR */ } cs_cin ; typedef struct cs_ci_dmperm_results /* cs_ci_dmperm or cs_ci_scc output */ { int *p ; /* size m, row permutation */ int *q ; /* size n, column permutation */ int *r ; /* size nb+1, block k is rows r[k] to r[k+1]-1 in A(p,q) */ int *s ; /* size nb+1, block k is cols s[k] to s[k+1]-1 in A(p,q) */ int nb ; /* # of blocks in fine dmperm decomposition */ int rr [5] ; /* coarse row decomposition */ int cc [5] ; /* coarse column decomposition */ } cs_cid ; int *cs_ci_amd (int order, const cs_ci *A) ; cs_cin *cs_ci_chol (const cs_ci *A, const cs_cis *S) ; cs_cid *cs_ci_dmperm (const cs_ci *A, int seed) ; int cs_ci_droptol (cs_ci *A, double tol) ; int cs_ci_dropzeros (cs_ci *A) ; int cs_ci_happly (const cs_ci *V, int i, double beta, cs_complex_t *x) ; int cs_ci_ipvec (const int *p, const cs_complex_t *b, cs_complex_t *x, int n) ; int cs_ci_lsolve (const cs_ci *L, cs_complex_t *x) ; int cs_ci_ltsolve (const cs_ci *L, cs_complex_t *x) ; cs_cin *cs_ci_lu (const cs_ci *A, const cs_cis *S, double tol) ; cs_ci *cs_ci_permute (const cs_ci *A, const int *pinv, const int *q, int values) ; int *cs_ci_pinv (const int *p, int n) ; int cs_ci_pvec (const int *p, const cs_complex_t *b, cs_complex_t *x, int n) ; cs_cin *cs_ci_qr (const cs_ci *A, const cs_cis *S) ; cs_cis *cs_ci_schol (int order, const cs_ci *A) ; cs_cis *cs_ci_sqr (int order, const cs_ci *A, int qr) ; cs_ci *cs_ci_symperm (const cs_ci *A, const int *pinv, int values) ; int cs_ci_usolve (const cs_ci *U, cs_complex_t *x) ; int cs_ci_utsolve (const cs_ci *U, cs_complex_t *x) ; int cs_ci_updown (cs_ci *L, int sigma, const cs_ci *C, const int *parent) ; /* utilities */ cs_cis *cs_ci_sfree (cs_cis *S) ; cs_cin *cs_ci_nfree (cs_cin *N) ; cs_cid *cs_ci_dfree (cs_cid *D) ; /* --- tertiary CSparse routines -------------------------------------------- */ int *cs_ci_counts (const cs_ci *A, const int *parent, const int *post, int ata) ; double cs_ci_cumsum (int *p, int *c, int n) ; int cs_ci_dfs (int j, cs_ci *G, int top, int *xi, int *pstack, const int *pinv) ; int *cs_ci_etree (const cs_ci *A, int ata) ; int cs_ci_fkeep (cs_ci *A, int (*fkeep) (int, int, cs_complex_t, void *), void *other) ; cs_complex_t cs_ci_house (cs_complex_t *x, double *beta, int n) ; int *cs_ci_maxtrans (const cs_ci *A, int seed) ; int *cs_ci_post (const int *parent, int n) ; cs_cid *cs_ci_scc (cs_ci *A) ; int cs_ci_scatter (const cs_ci *A, int j, cs_complex_t beta, int *w, cs_complex_t *x, int mark,cs_ci *C, int nz) ; int cs_ci_tdfs (int j, int k, int *head, const int *next, int *post, int *stack) ; int cs_ci_leaf (int i, int j, const int *first, int *maxfirst, int *prevleaf, int *ancestor, int *jleaf) ; int cs_ci_reach (cs_ci *G, const cs_ci *B, int k, int *xi, const int *pinv) ; int cs_ci_spsolve (cs_ci *L, const cs_ci *B, int k, int *xi, cs_complex_t *x, const int *pinv, int lo) ; int cs_ci_ereach (const cs_ci *A, int k, const int *parent, int *s, int *w) ; int *cs_ci_randperm (int n, int seed) ; /* utilities */ cs_cid *cs_ci_dalloc (int m, int n) ; cs_ci *cs_ci_done (cs_ci *C, void *w, void *x, int ok) ; int *cs_ci_idone (int *p, cs_ci *C, void *w, int ok) ; cs_cin *cs_ci_ndone (cs_cin *N, cs_ci *C, void *w, void *x, int ok) ; cs_cid *cs_ci_ddone (cs_cid *D, cs_ci *C, void *w, int ok) ; /* -------------------------------------------------------------------------- */ /* complex/UF_long version of CXSparse */ /* -------------------------------------------------------------------------- */ /* --- primary CSparse routines and data structures ------------------------- */ typedef struct cs_cl_sparse /* matrix in compressed-column or triplet form */ { UF_long nzmax ; /* maximum number of entries */ UF_long m ; /* number of rows */ UF_long n ; /* number of columns */ UF_long *p ; /* column pointers (size n+1) or col indlces (size nzmax) */ UF_long *i ; /* row indices, size nzmax */ cs_complex_t *x ; /* numerical values, size nzmax */ UF_long nz ; /* # of entries in triplet matrix, -1 for compressed-col */ } cs_cl ; cs_cl *cs_cl_add (const cs_cl *A, const cs_cl *B, cs_complex_t alpha, cs_complex_t beta) ; UF_long cs_cl_cholsol (UF_long order, const cs_cl *A, cs_complex_t *b) ; UF_long cs_cl_dupl (cs_cl *A) ; UF_long cs_cl_entry (cs_cl *T, UF_long i, UF_long j, cs_complex_t x) ; UF_long cs_cl_lusol (UF_long order, const cs_cl *A, cs_complex_t *b, double tol) ; UF_long cs_cl_gaxpy (const cs_cl *A, const cs_complex_t *x, cs_complex_t *y) ; cs_cl *cs_cl_multiply (const cs_cl *A, const cs_cl *B) ; UF_long cs_cl_qrsol (UF_long order, const cs_cl *A, cs_complex_t *b) ; cs_cl *cs_cl_transpose (const cs_cl *A, UF_long values) ; cs_cl *cs_cl_compress (const cs_cl *T) ; double cs_cl_norm (const cs_cl *A) ; UF_long cs_cl_print (const cs_cl *A, UF_long brief) ; cs_cl *cs_cl_load (FILE *f) ; /* utilities */ void *cs_cl_calloc (UF_long n, size_t size) ; void *cs_cl_free (void *p) ; void *cs_cl_realloc (void *p, UF_long n, size_t size, UF_long *ok) ; cs_cl *cs_cl_spalloc (UF_long m, UF_long n, UF_long nzmax, UF_long values, UF_long t) ; cs_cl *cs_cl_spfree (cs_cl *A) ; UF_long cs_cl_sprealloc (cs_cl *A, UF_long nzmax) ; void *cs_cl_malloc (UF_long n, size_t size) ; /* --- secondary CSparse routines and data structures ----------------------- */ typedef struct cs_cl_symbolic /* symbolic Cholesky, LU, or QR analysis */ { UF_long *pinv ; /* inverse row perm. for QR, fill red. perm for Chol */ UF_long *q ; /* fill-reducing column permutation for LU and QR */ UF_long *parent ; /* elimination tree for Cholesky and QR */ UF_long *cp ; /* column pointers for Cholesky, row counts for QR */ UF_long *leftmost ; /* leftmost[i] = min(find(A(i,:))), for QR */ UF_long m2 ; /* # of rows for QR, after adding fictitious rows */ double lnz ; /* # entries in L for LU or Cholesky; in V for QR */ double unz ; /* # entries in U for LU; in R for QR */ } cs_cls ; typedef struct cs_cl_numeric /* numeric Cholesky, LU, or QR factorization */ { cs_cl *L ; /* L for LU and Cholesky, V for QR */ cs_cl *U ; /* U for LU, r for QR, not used for Cholesky */ UF_long *pinv ; /* partial pivoting for LU */ double *B ; /* beta [0..n-1] for QR */ } cs_cln ; typedef struct cs_cl_dmperm_results /* cs_cl_dmperm or cs_cl_scc output */ { UF_long *p ; /* size m, row permutation */ UF_long *q ; /* size n, column permutation */ UF_long *r ; /* size nb+1, block k is rows r[k] to r[k+1]-1 in A(p,q) */ UF_long *s ; /* size nb+1, block k is cols s[k] to s[k+1]-1 in A(p,q) */ UF_long nb ; /* # of blocks in fine dmperm decomposition */ UF_long rr [5] ; /* coarse row decomposition */ UF_long cc [5] ; /* coarse column decomposition */ } cs_cld ; UF_long *cs_cl_amd (UF_long order, const cs_cl *A) ; cs_cln *cs_cl_chol (const cs_cl *A, const cs_cls *S) ; cs_cld *cs_cl_dmperm (const cs_cl *A, UF_long seed) ; UF_long cs_cl_droptol (cs_cl *A, double tol) ; UF_long cs_cl_dropzeros (cs_cl *A) ; UF_long cs_cl_happly (const cs_cl *V, UF_long i, double beta, cs_complex_t *x) ; UF_long cs_cl_ipvec (const UF_long *p, const cs_complex_t *b, cs_complex_t *x, UF_long n) ; UF_long cs_cl_lsolve (const cs_cl *L, cs_complex_t *x) ; UF_long cs_cl_ltsolve (const cs_cl *L, cs_complex_t *x) ; cs_cln *cs_cl_lu (const cs_cl *A, const cs_cls *S, double tol) ; cs_cl *cs_cl_permute (const cs_cl *A, const UF_long *pinv, const UF_long *q, UF_long values) ; UF_long *cs_cl_pinv (const UF_long *p, UF_long n) ; UF_long cs_cl_pvec (const UF_long *p, const cs_complex_t *b, cs_complex_t *x, UF_long n) ; cs_cln *cs_cl_qr (const cs_cl *A, const cs_cls *S) ; cs_cls *cs_cl_schol (UF_long order, const cs_cl *A) ; cs_cls *cs_cl_sqr (UF_long order, const cs_cl *A, UF_long qr) ; cs_cl *cs_cl_symperm (const cs_cl *A, const UF_long *pinv, UF_long values) ; UF_long cs_cl_usolve (const cs_cl *U, cs_complex_t *x) ; UF_long cs_cl_utsolve (const cs_cl *U, cs_complex_t *x) ; UF_long cs_cl_updown (cs_cl *L, UF_long sigma, const cs_cl *C, const UF_long *parent) ; /* utilities */ cs_cls *cs_cl_sfree (cs_cls *S) ; cs_cln *cs_cl_nfree (cs_cln *N) ; cs_cld *cs_cl_dfree (cs_cld *D) ; /* --- tertiary CSparse routines -------------------------------------------- */ UF_long *cs_cl_counts (const cs_cl *A, const UF_long *parent, const UF_long *post, UF_long ata) ; double cs_cl_cumsum (UF_long *p, UF_long *c, UF_long n) ; UF_long cs_cl_dfs (UF_long j, cs_cl *G, UF_long top, UF_long *xi, UF_long *pstack, const UF_long *pinv) ; UF_long *cs_cl_etree (const cs_cl *A, UF_long ata) ; UF_long cs_cl_fkeep (cs_cl *A, UF_long (*fkeep) (UF_long, UF_long, cs_complex_t, void *), void *other) ; cs_complex_t cs_cl_house (cs_complex_t *x, double *beta, UF_long n) ; UF_long *cs_cl_maxtrans (const cs_cl *A, UF_long seed) ; UF_long *cs_cl_post (const UF_long *parent, UF_long n) ; cs_cld *cs_cl_scc (cs_cl *A) ; UF_long cs_cl_scatter (const cs_cl *A, UF_long j, cs_complex_t beta, UF_long *w, cs_complex_t *x, UF_long mark,cs_cl *C, UF_long nz) ; UF_long cs_cl_tdfs (UF_long j, UF_long k, UF_long *head, const UF_long *next, UF_long *post, UF_long *stack) ; UF_long cs_cl_leaf (UF_long i, UF_long j, const UF_long *first, UF_long *maxfirst, UF_long *prevleaf, UF_long *ancestor, UF_long *jleaf) ; UF_long cs_cl_reach (cs_cl *G, const cs_cl *B, UF_long k, UF_long *xi, const UF_long *pinv) ; UF_long cs_cl_spsolve (cs_cl *L, const cs_cl *B, UF_long k, UF_long *xi, cs_complex_t *x, const UF_long *pinv, UF_long lo) ; UF_long cs_cl_ereach (const cs_cl *A, UF_long k, const UF_long *parent, UF_long *s, UF_long *w) ; UF_long *cs_cl_randperm (UF_long n, UF_long seed) ; /* utilities */ cs_cld *cs_cl_dalloc (UF_long m, UF_long n) ; cs_cl *cs_cl_done (cs_cl *C, void *w, void *x, UF_long ok) ; UF_long *cs_cl_idone (UF_long *p, cs_cl *C, void *w, UF_long ok) ; cs_cln *cs_cl_ndone (cs_cln *N, cs_cl *C, void *w, void *x, UF_long ok) ; cs_cld *cs_cl_ddone (cs_cld *D, cs_cl *C, void *w, UF_long ok) ; #endif /* -------------------------------------------------------------------------- */ /* Macros for constructing each version of CSparse */ /* -------------------------------------------------------------------------- */ #ifdef CS_LONG #define CS_INT UF_long #define CS_INT_MAX UF_long_max #define CS_ID UF_long_id #ifdef CS_COMPLEX #define CS_ENTRY cs_complex_t #define CS_NAME(nm) cs_cl ## nm #define cs cs_cl #else #define CS_ENTRY double #define CS_NAME(nm) cs_dl ## nm #define cs cs_dl #endif #else #define CS_INT int #define CS_INT_MAX INT_MAX #define CS_ID "%d" #ifdef CS_COMPLEX #define CS_ENTRY cs_complex_t #define CS_NAME(nm) cs_ci ## nm #define cs cs_ci #else #define CS_ENTRY double #define CS_NAME(nm) cs_di ## nm #define cs cs_di #endif #endif #ifdef CS_COMPLEX #define CS_REAL(x) creal(x) #define CS_IMAG(x) cimag(x) #define CS_CONJ(x) conj(x) #define CS_ABS(x) cabs(x) #else #define CS_REAL(x) (x) #define CS_IMAG(x) (0.) #define CS_CONJ(x) (x) #define CS_ABS(x) fabs(x) #endif #define CS_MAX(a,b) (((a) > (b)) ? (a) : (b)) #define CS_MIN(a,b) (((a) < (b)) ? (a) : (b)) #define CS_FLIP(i) (-(i)-2) #define CS_UNFLIP(i) (((i) < 0) ? CS_FLIP(i) : (i)) #define CS_MARKED(w,j) (w [j] < 0) #define CS_MARK(w,j) { w [j] = CS_FLIP (w [j]) ; } #define CS_CSC(A) (A && (A->nz == -1)) #define CS_TRIPLET(A) (A && (A->nz >= 0)) /* --- primary CSparse routines and data structures ------------------------- */ #define cs_add CS_NAME (_add) #define cs_cholsol CS_NAME (_cholsol) #define cs_dupl CS_NAME (_dupl) #define cs_entry CS_NAME (_entry) #define cs_lusol CS_NAME (_lusol) #define cs_gaxpy CS_NAME (_gaxpy) #define cs_multiply CS_NAME (_multiply) #define cs_qrsol CS_NAME (_qrsol) #define cs_transpose CS_NAME (_transpose) #define cs_compress CS_NAME (_compress) #define cs_norm CS_NAME (_norm) #define cs_print CS_NAME (_print) #define cs_load CS_NAME (_load) /* utilities */ #define cs_calloc CS_NAME (_calloc) #define cs_free CS_NAME (_free) #define cs_realloc CS_NAME (_realloc) #define cs_spalloc CS_NAME (_spalloc) #define cs_spfree CS_NAME (_spfree) #define cs_sprealloc CS_NAME (_sprealloc) #define cs_malloc CS_NAME (_malloc) /* --- secondary CSparse routines and data structures ----------------------- */ #define css CS_NAME (s) #define csn CS_NAME (n) #define csd CS_NAME (d) #define cs_amd CS_NAME (_amd) #define cs_chol CS_NAME (_chol) #define cs_dmperm CS_NAME (_dmperm) #define cs_droptol CS_NAME (_droptol) #define cs_dropzeros CS_NAME (_dropzeros) #define cs_happly CS_NAME (_happly) #define cs_ipvec CS_NAME (_ipvec) #define cs_lsolve CS_NAME (_lsolve) #define cs_ltsolve CS_NAME (_ltsolve) #define cs_lu CS_NAME (_lu) #define cs_permute CS_NAME (_permute) #define cs_pinv CS_NAME (_pinv) #define cs_pvec CS_NAME (_pvec) #define cs_qr CS_NAME (_qr) #define cs_schol CS_NAME (_schol) #define cs_sqr CS_NAME (_sqr) #define cs_symperm CS_NAME (_symperm) #define cs_usolve CS_NAME (_usolve) #define cs_utsolve CS_NAME (_utsolve) #define cs_updown CS_NAME (_updown) /* utilities */ #define cs_sfree CS_NAME (_sfree) #define cs_nfree CS_NAME (_nfree) #define cs_dfree CS_NAME (_dfree) /* --- tertiary CSparse routines -------------------------------------------- */ #define cs_counts CS_NAME (_counts) #define cs_cumsum CS_NAME (_cumsum) #define cs_dfs CS_NAME (_dfs) #define cs_etree CS_NAME (_etree) #define cs_fkeep CS_NAME (_fkeep) #define cs_house CS_NAME (_house) #define cs_invmatch CS_NAME (_invmatch) #define cs_maxtrans CS_NAME (_maxtrans) #define cs_post CS_NAME (_post) #define cs_scc CS_NAME (_scc) #define cs_scatter CS_NAME (_scatter) #define cs_tdfs CS_NAME (_tdfs) #define cs_reach CS_NAME (_reach) #define cs_spsolve CS_NAME (_spsolve) #define cs_ereach CS_NAME (_ereach) #define cs_randperm CS_NAME (_randperm) #define cs_leaf CS_NAME (_leaf) /* utilities */ #define cs_dalloc CS_NAME (_dalloc) #define cs_done CS_NAME (_done) #define cs_idone CS_NAME (_idone) #define cs_ndone CS_NAME (_ndone) #define cs_ddone CS_NAME (_ddone) /* -------------------------------------------------------------------------- */ /* Conversion routines */ /* -------------------------------------------------------------------------- */ #ifndef NCOMPLEX cs_di *cs_i_real (cs_ci *A, int real) ; cs_ci *cs_i_complex (cs_di *A, int real) ; cs_dl *cs_l_real (cs_cl *A, UF_long real) ; cs_cl *cs_l_complex (cs_dl *A, UF_long real) ; #endif #ifdef __cplusplus } #endif #endif