EpetraExt Development
EpetraExt_BlockCrsMatrix.h
Go to the documentation of this file.
00001 //@HEADER
00002 /*
00003 ************************************************************************
00004 
00005               EpetraExt: Extended Linear Algebra Services Package 
00006                 Copyright (2001) Sandia Corporation
00007 
00008 Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
00009 license for use of this work by or on behalf of the U.S. Government.
00010 
00011 This library is free software; you can redistribute it and/or modify
00012 it under the terms of the GNU Lesser General Public License as
00013 published by the Free Software Foundation; either version 2.1 of the
00014 License, or (at your option) any later version.
00015  
00016 This library is distributed in the hope that it will be useful, but
00017 WITHOUT ANY WARRANTY; without even the implied warranty of
00018 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00019 Lesser General Public License for more details.
00020  
00021 You should have received a copy of the GNU Lesser General Public
00022 License along with this library; if not, write to the Free Software
00023 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
00024 USA
00025 Questions? Contact Michael A. Heroux (maherou@sandia.gov) 
00026 
00027 ************************************************************************
00028 */
00029 //@HEADER
00030 
00031 #ifndef EPETRAEXT_BLOCKCRSMATRIX_H
00032 #define EPETRAEXT_BLOCKCRSMATRIX_H
00033 
00034 #include <vector>
00035 
00036 #include "Epetra_CrsMatrix.h" 
00037 
00039 
00046 namespace EpetraExt {
00047 
00048 class BlockCrsMatrix: public Epetra_CrsMatrix {
00049  public:
00050 
00052 
00053 
00062   BlockCrsMatrix( const Epetra_CrsGraph & BaseGraph, const std::vector<int> & RowStencil, int RowIndex, const Epetra_Comm & GlobalComm );
00063   
00065 
00074   BlockCrsMatrix( const Epetra_CrsGraph & BaseGraph, const std::vector< std::vector<int> > & RowStencil, const std::vector<int> & RowIndices, const Epetra_Comm & GlobalComm );
00075 
00076   BlockCrsMatrix( const Epetra_RowMatrix & BaseMatrix, const std::vector< std::vector<int> > & RowStencil, const std::vector<int> & RowIndices, const Epetra_Comm & GlobalComm );
00077   
00079   BlockCrsMatrix( const BlockCrsMatrix & Matrix );
00080 
00082   virtual ~BlockCrsMatrix();
00084   
00086   const std::vector<int> & Stencil( int i = 0 ) { return RowStencil_[i]; }
00087 
00089   int RowIndex( int i = 0 ) { return RowIndices_[i]; }
00090   
00093   void LoadBlock(const Epetra_RowMatrix & BaseMatrix, const int Row, const int Col);
00094 
00097   void SumIntoBlock(double alpha, const Epetra_RowMatrix & BaseMatrix, const int Row, const int Col);
00098 
00101   void BlockSumIntoGlobalValues(const int BaseRow, int NumIndices,
00102      double* Values, const int* Indices, const int Row, const int Col);
00103   void BlockReplaceGlobalValues(const int BaseRow, int NumIndices,
00104      double* Values, const int* Indices, const int Row, const int Col);
00105   void BlockExtractGlobalRowView(const int BaseRow, int& NumEntries, 
00106      double*& Values, const int Row, const int Col);
00107 
00108   void ExtractBlock(Epetra_CrsMatrix & BaseMatrix, const int Row, const int Col);
00109 
00110  protected:
00111 
00112   Epetra_CrsGraph BaseGraph_;
00113 
00114   std::vector< std::vector<int> > RowStencil_;
00115 
00116   std::vector<int> RowIndices_; 
00117 
00118   int Offset_;
00119 
00120 };
00121 
00122 } //namespace EpetraExt
00123 
00124 #endif /* EPETRA_CRSMATRIX_H */
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines