![]() |
Eigen
3.3.3
|
00001 // This file is part of Eigen, a lightweight C++ template library 00002 // for linear algebra. 00003 // 00004 // Copyright (C) 2012 Désiré Nuentsa-Wakam <desire.nuentsa_wakam@inria.fr> 00005 // 00006 // This Source Code Form is subject to the terms of the Mozilla 00007 // Public License v. 2.0. If a copy of the MPL was not distributed 00008 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 00009 #ifndef SPARSELU_IMPL_H 00010 #define SPARSELU_IMPL_H 00011 00012 namespace Eigen { 00013 namespace internal { 00014 00019 template <typename Scalar, typename StorageIndex> 00020 class SparseLUImpl 00021 { 00022 public: 00023 typedef Matrix<Scalar,Dynamic,1> ScalarVector; 00024 typedef Matrix<StorageIndex,Dynamic,1> IndexVector; 00025 typedef Matrix<Scalar,Dynamic,Dynamic,ColMajor> ScalarMatrix; 00026 typedef Map<ScalarMatrix, 0, OuterStride<> > MappedMatrixBlock; 00027 typedef typename ScalarVector::RealScalar RealScalar; 00028 typedef Ref<Matrix<Scalar,Dynamic,1> > BlockScalarVector; 00029 typedef Ref<Matrix<StorageIndex,Dynamic,1> > BlockIndexVector; 00030 typedef LU_GlobalLU_t<IndexVector, ScalarVector> GlobalLU_t; 00031 typedef SparseMatrix<Scalar,ColMajor,StorageIndex> MatrixType; 00032 00033 protected: 00034 template <typename VectorType> 00035 Index expand(VectorType& vec, Index& length, Index nbElts, Index keep_prev, Index& num_expansions); 00036 Index memInit(Index m, Index n, Index annz, Index lwork, Index fillratio, Index panel_size, GlobalLU_t& glu); 00037 template <typename VectorType> 00038 Index memXpand(VectorType& vec, Index& maxlen, Index nbElts, MemType memtype, Index& num_expansions); 00039 void heap_relax_snode (const Index n, IndexVector& et, const Index relax_columns, IndexVector& descendants, IndexVector& relax_end); 00040 void relax_snode (const Index n, IndexVector& et, const Index relax_columns, IndexVector& descendants, IndexVector& relax_end); 00041 Index snode_dfs(const Index jcol, const Index kcol,const MatrixType& mat, IndexVector& xprune, IndexVector& marker, GlobalLU_t& glu); 00042 Index snode_bmod (const Index jcol, const Index fsupc, ScalarVector& dense, GlobalLU_t& glu); 00043 Index pivotL(const Index jcol, const RealScalar& diagpivotthresh, IndexVector& perm_r, IndexVector& iperm_c, Index& pivrow, GlobalLU_t& glu); 00044 template <typename Traits> 00045 void dfs_kernel(const StorageIndex jj, IndexVector& perm_r, 00046 Index& nseg, IndexVector& panel_lsub, IndexVector& segrep, 00047 Ref<IndexVector> repfnz_col, IndexVector& xprune, Ref<IndexVector> marker, IndexVector& parent, 00048 IndexVector& xplore, GlobalLU_t& glu, Index& nextl_col, Index krow, Traits& traits); 00049 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); 00050 00051 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); 00052 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); 00053 Index column_bmod(const Index jcol, const Index nseg, BlockScalarVector dense, ScalarVector& tempv, BlockIndexVector segrep, BlockIndexVector repfnz, Index fpanelc, GlobalLU_t& glu); 00054 Index copy_to_ucol(const Index jcol, const Index nseg, IndexVector& segrep, BlockIndexVector repfnz ,IndexVector& perm_r, BlockScalarVector dense, GlobalLU_t& glu); 00055 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); 00056 void countnz(const Index n, Index& nnzL, Index& nnzU, GlobalLU_t& glu); 00057 void fixupL(const Index n, const IndexVector& perm_r, GlobalLU_t& glu); 00058 00059 template<typename , typename > 00060 friend struct column_dfs_traits; 00061 }; 00062 00063 } // end namespace internal 00064 } // namespace Eigen 00065 00066 #endif