FEI Version of the Day
fei_LinProbMgr_EpetraBasic.hpp
00001 /*
00002 // @HEADER
00003 // ************************************************************************
00004 //             FEI: Finite Element Interface to Linear Solvers
00005 //                  Copyright (2005) Sandia Corporation.
00006 //
00007 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation, the
00008 // U.S. Government retains certain rights in this software.
00009 //
00010 // Redistribution and use in source and binary forms, with or without
00011 // modification, are permitted provided that the following conditions are
00012 // met:
00013 //
00014 // 1. Redistributions of source code must retain the above copyright
00015 // notice, this list of conditions and the following disclaimer.
00016 //
00017 // 2. Redistributions in binary form must reproduce the above copyright
00018 // notice, this list of conditions and the following disclaimer in the
00019 // documentation and/or other materials provided with the distribution.
00020 //
00021 // 3. Neither the name of the Corporation nor the names of the
00022 // contributors may be used to endorse or promote products derived from
00023 // this software without specific prior written permission.
00024 //
00025 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
00026 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00027 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00028 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
00029 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00030 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00031 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00032 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00033 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00034 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00035 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00036 //
00037 // Questions? Contact Alan Williams (william@sandia.gov) 
00038 //
00039 // ************************************************************************
00040 // @HEADER
00041 */
00042 
00043 
00044 #ifndef _fei_LinProbMgr_EpetraBasic_hpp_
00045 #define _fei_LinProbMgr_EpetraBasic_hpp_
00046 
00047 #include <fei_LinearProblemManager.hpp>
00048 
00049 #include <fei_Include_Trilinos.hpp>
00050 #include <fei_SharedPtr.hpp>
00051 
00052 #include <vector>
00053 
00054 class LinProbMgr_EpetraBasic : public fei::LinearProblemManager {
00055  public:
00056   LinProbMgr_EpetraBasic(MPI_Comm comm);
00057   virtual ~LinProbMgr_EpetraBasic();
00058 
00063   void setRowDistribution(const std::vector<int>& ownedGlobalRows);
00064 
00068   void setMatrixGraph(fei::SharedPtr<fei::SparseRowGraph> matrixGraph);
00069 
00072   void setMatrixValues(double scalar);
00073 
00081   void setVectorValues(double scalar, bool soln_vector);
00082 
00086   int getLocalNumRows();
00087 
00091   int getRowLength(int row);
00092 
00105   int copyOutMatrixRow(int row, int len,
00106                        double* coefs, int* indices);
00107 
00115   int insertMatrixValues(int numRows, const int* rows,
00116                          int numCols, const int* cols,
00117                          const double* const* values,
00118                          bool sum_into);
00119 
00142   int insertVectorValues(int numValues,
00143                          const int* globalIndices,
00144                          const double* values,
00145                          bool sum_into,
00146                          bool soln_vector,
00147                          int vectorIndex=0);
00148 
00152   int copyOutVectorValues(int numValues,
00153                            const int* globalIndices,
00154                            double* values,
00155                            bool soln_vector,
00156                            int vectorIndex=0);
00157 
00163   double* getLocalVectorValuesPtr(bool soln_vector,
00164                                   int vectorIndex=0);
00165 
00171   int globalAssemble();
00172 
00175   fei::SharedPtr<Epetra_CrsMatrix> get_A_matrix();
00176 
00179   fei::SharedPtr<Epetra_MultiVector> get_rhs_vector();
00180 
00183   fei::SharedPtr<Epetra_MultiVector> get_solution_vector();
00184 
00185  private:
00186   MPI_Comm comm_;
00187   std::vector<int> ownedRows_;
00188   fei::SharedPtr<Epetra_Comm> epetra_comm_;
00189   fei::SharedPtr<Epetra_Map> epetra_rowmap_;
00190   fei::SharedPtr<fei::SparseRowGraph> fei_srgraph_;
00191   fei::SharedPtr<Epetra_CrsGraph> crsgraph_;
00192   fei::SharedPtr<Epetra_CrsMatrix> A_;
00193   int numVectors_;
00194   fei::SharedPtr<Epetra_MultiVector> x_;
00195   fei::SharedPtr<Epetra_MultiVector> b_;
00196 };
00197 
00198 #endif // _LinProbMgr_EpetraBasic_hpp_
00199 
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Friends