17faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// This file is part of Eigen, a lightweight C++ template library 27faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// for linear algebra. 37faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// 47faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// Copyright (C) 2012 Désiré Nuentsa-Wakam <desire.nuentsa_wakam@inria.fr> 57faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// 67faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// This Source Code Form is subject to the terms of the Mozilla 77faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// Public License v. 2.0. If a copy of the MPL was not distributed 87faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 97faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez#ifndef SPARSELU_IMPL_H 107faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez#define SPARSELU_IMPL_H 117faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 127faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandeznamespace Eigen { 137faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandeznamespace internal { 147faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 157faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez/** \ingroup SparseLU_Module 167faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez * \class SparseLUImpl 177faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez * Base class for sparseLU 187faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez */ 197faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandeztemplate <typename Scalar, typename Index> 207faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandezclass SparseLUImpl 217faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez{ 227faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez public: 237faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef Matrix<Scalar,Dynamic,1> ScalarVector; 247faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef Matrix<Index,Dynamic,1> IndexVector; 257faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef typename ScalarVector::RealScalar RealScalar; 267faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef Ref<Matrix<Scalar,Dynamic,1> > BlockScalarVector; 277faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef Ref<Matrix<Index,Dynamic,1> > BlockIndexVector; 287faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef LU_GlobalLU_t<IndexVector, ScalarVector> GlobalLU_t; 297faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef SparseMatrix<Scalar,ColMajor,Index> MatrixType; 307faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 317faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez protected: 327faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez template <typename VectorType> 337faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index expand(VectorType& vec, Index& length, Index nbElts, Index keep_prev, Index& num_expansions); 347faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index memInit(Index m, Index n, Index annz, Index lwork, Index fillratio, Index panel_size, GlobalLU_t& glu); 357faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez template <typename VectorType> 367faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index memXpand(VectorType& vec, Index& maxlen, Index nbElts, MemType memtype, Index& num_expansions); 377faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez void heap_relax_snode (const Index n, IndexVector& et, const Index relax_columns, IndexVector& descendants, IndexVector& relax_end); 387faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez void relax_snode (const Index n, IndexVector& et, const Index relax_columns, IndexVector& descendants, IndexVector& relax_end); 397faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index snode_dfs(const Index jcol, const Index kcol,const MatrixType& mat, IndexVector& xprune, IndexVector& marker, GlobalLU_t& glu); 407faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index snode_bmod (const Index jcol, const Index fsupc, ScalarVector& dense, GlobalLU_t& glu); 417faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index pivotL(const Index jcol, const RealScalar& diagpivotthresh, IndexVector& perm_r, IndexVector& iperm_c, Index& pivrow, GlobalLU_t& glu); 427faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez template <typename Traits> 437faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez void dfs_kernel(const Index jj, IndexVector& perm_r, 447faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index& nseg, IndexVector& panel_lsub, IndexVector& segrep, 457faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Ref<IndexVector> repfnz_col, IndexVector& xprune, Ref<IndexVector> marker, IndexVector& parent, 467faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez IndexVector& xplore, GlobalLU_t& glu, Index& nextl_col, Index krow, Traits& traits); 477faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 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); 487faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 497faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 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); 507faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 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); 517faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index column_bmod(const Index jcol, const Index nseg, BlockScalarVector dense, ScalarVector& tempv, BlockIndexVector segrep, BlockIndexVector repfnz, Index fpanelc, GlobalLU_t& glu); 527faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index copy_to_ucol(const Index jcol, const Index nseg, IndexVector& segrep, BlockIndexVector repfnz ,IndexVector& perm_r, BlockScalarVector dense, GlobalLU_t& glu); 537faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 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); 547faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez void countnz(const Index n, Index& nnzL, Index& nnzU, GlobalLU_t& glu); 557faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez void fixupL(const Index n, const IndexVector& perm_r, GlobalLU_t& glu); 567faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 577faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez template<typename , typename > 587faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez friend struct column_dfs_traits; 597faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez}; 607faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 617faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez} // end namespace internal 627faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez} // namespace Eigen 637faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 647faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez#endif 65