9 #ifndef SPARSELU_IMPL_H 10 #define SPARSELU_IMPL_H 19 template <
typename Scalar,
typename Index>
23 typedef Matrix<Scalar,Dynamic,1> ScalarVector;
24 typedef Matrix<Index,Dynamic,1> IndexVector;
25 typedef typename ScalarVector::RealScalar RealScalar;
26 typedef Ref<Matrix<Scalar,Dynamic,1> > BlockScalarVector;
27 typedef Ref<Matrix<Index,Dynamic,1> > BlockIndexVector;
28 typedef LU_GlobalLU_t<IndexVector, ScalarVector> GlobalLU_t;
29 typedef SparseMatrix<Scalar,ColMajor,Index> MatrixType;
32 template <
typename VectorType>
33 Index expand(VectorType& vec, Index& length, Index nbElts, Index keep_prev, Index& num_expansions);
34 Index memInit(Index m, Index n, Index annz, Index lwork, Index fillratio, Index panel_size, GlobalLU_t& glu);
35 template <
typename VectorType>
36 Index memXpand(VectorType& vec, Index& maxlen, Index nbElts, MemType memtype, Index& num_expansions);
37 void heap_relax_snode (
const Index n, IndexVector& et,
const Index relax_columns, IndexVector& descendants, IndexVector& relax_end);
38 void relax_snode (
const Index n, IndexVector& et,
const Index relax_columns, IndexVector& descendants, IndexVector& relax_end);
39 Index snode_dfs(
const Index jcol,
const Index kcol,
const MatrixType& mat, IndexVector& xprune, IndexVector& marker, GlobalLU_t& glu);
40 Index snode_bmod (
const Index jcol,
const Index fsupc, ScalarVector& dense, GlobalLU_t& glu);
41 Index pivotL(
const Index jcol,
const RealScalar& diagpivotthresh, IndexVector& perm_r, IndexVector& iperm_c, Index& pivrow, GlobalLU_t& glu);
42 template <
typename Traits>
43 void dfs_kernel(
const Index jj, IndexVector& perm_r,
44 Index& nseg, IndexVector& panel_lsub, IndexVector& segrep,
45 Ref<IndexVector> repfnz_col, IndexVector& xprune, Ref<IndexVector> marker, IndexVector& parent,
46 IndexVector& xplore, GlobalLU_t& glu, Index& nextl_col, Index krow, Traits& traits);
47 void panel_dfs(
const Index m,
const Index w,
const Index jcol, MatrixType& A, IndexVector& perm_r, Index& nseg, ScalarVector& dense, IndexVector& panel_lsub, IndexVector& segrep, IndexVector& repfnz, IndexVector& xprune, IndexVector& marker, IndexVector& parent, IndexVector& xplore, GlobalLU_t& glu);
49 void panel_bmod(
const Index m,
const Index w,
const Index jcol,
const Index nseg, ScalarVector& dense, ScalarVector& tempv, IndexVector& segrep, IndexVector& repfnz, GlobalLU_t& glu);
50 Index column_dfs(
const Index m,
const Index jcol, IndexVector& perm_r, Index maxsuper, Index& nseg, BlockIndexVector lsub_col, IndexVector& segrep, BlockIndexVector repfnz, IndexVector& xprune, IndexVector& marker, IndexVector& parent, IndexVector& xplore, GlobalLU_t& glu);
51 Index column_bmod(
const Index jcol,
const Index nseg, BlockScalarVector dense, ScalarVector& tempv, BlockIndexVector segrep, BlockIndexVector repfnz, Index fpanelc, GlobalLU_t& glu);
52 Index copy_to_ucol(
const Index jcol,
const Index nseg, IndexVector& segrep, BlockIndexVector repfnz ,IndexVector& perm_r, BlockScalarVector dense, GlobalLU_t& glu);
53 void pruneL(
const Index jcol,
const IndexVector& perm_r,
const Index pivrow,
const Index nseg,
const IndexVector& segrep, BlockIndexVector repfnz, IndexVector& xprune, GlobalLU_t& glu);
54 void countnz(
const Index n, Index& nnzL, Index& nnzU, GlobalLU_t& glu);
55 void fixupL(
const Index n,
const IndexVector& perm_r, GlobalLU_t& glu);
57 template<
typename ,
typename >
58 friend struct column_dfs_traits;
Definition: Eigen_Colamd.h:54