EpetraExt Development
EpetraExt_LPTrans_From_MatrixTrans.cpp
Go to the documentation of this file.
00001 //@HEADER
00002 // ***********************************************************************
00003 //
00004 //     EpetraExt: Epetra Extended - Linear Algebra Services Package
00005 //                 Copyright (2011) Sandia Corporation
00006 //
00007 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
00008 // the 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 Michael A. Heroux (maherou@sandia.gov)
00038 //
00039 // ***********************************************************************
00040 //@HEADER
00041 
00042 #include <EpetraExt_LPTrans_From_MatrixTrans.h>
00043 
00044 #include <Epetra_Export.h>
00045 #include <Epetra_Import.h>
00046 #include <Epetra_LinearProblem.h>
00047 #include <Epetra_CrsGraph.h>
00048 #include <Epetra_CrsMatrix.h>
00049 #include <Epetra_MultiVector.h>
00050 #include <Epetra_Vector.h>
00051 #include <Epetra_IntVector.h>
00052 #include <Epetra_Map.h>
00053 #include <Epetra_Comm.h>
00054 
00055 namespace EpetraExt {
00056 
00057 LinearProblem_MatrixTrans::
00058 ~LinearProblem_MatrixTrans()
00059 {
00060   if( Exporter_ ) delete Exporter_;
00061   if( Importer_ ) delete Importer_;
00062 
00063   if( NewProblem_ ) delete NewProblem_;
00064   if( NewRHS_ ) delete NewRHS_;
00065   if( NewLHS_ ) delete NewLHS_;
00066 }
00067 
00068 LinearProblem_MatrixTrans::NewTypeRef
00069 LinearProblem_MatrixTrans::
00070 operator()( OriginalTypeRef orig )
00071 {
00072   OldProblem_ = &orig;
00073   OldMatrix_ = dynamic_cast<Epetra_CrsMatrix*>( orig.GetMatrix() );
00074   OldRHS_ = orig.GetRHS();
00075   OldLHS_ = orig.GetLHS();
00076   OldRowMap_ = const_cast<Epetra_Map*>(&OldMatrix_->RowMap());
00077 
00078   int ierr = 0;
00079 
00080 
00081   if( !OldMatrix_ ) ierr = -2;
00082   if( !OldRHS_ )    ierr = -3;
00083   if( !OldLHS_ )    ierr = -4;
00084 
00085   NewMatrix_ = &(matrixTrans_( *OldMatrix_ ));
00086 
00087   Epetra_BlockMap & NewRowMap = const_cast<Epetra_BlockMap&>(NewMatrix_->Graph().RowMap());
00088 
00089   NewRHS_ = new Epetra_MultiVector( NewRowMap, 1 );
00090   NewLHS_ = new Epetra_MultiVector( NewRowMap, 1 );
00091 
00092   Exporter_ = new Epetra_Export( *OldRowMap_, NewRowMap );
00093   Importer_ = new Epetra_Import( NewRowMap, *OldRowMap_ );
00094 
00095   NewProblem_ = new Epetra_LinearProblem( NewMatrix_, NewLHS_, NewRHS_ );
00096 
00097   return *NewProblem_;
00098 }
00099 
00100 bool
00101 LinearProblem_MatrixTrans::
00102 fwd()
00103 {
00104   NewLHS_->Export( *OldLHS_, *Exporter_, Insert );
00105   NewRHS_->Export( *OldRHS_, *Exporter_, Insert );
00106   NewMatrix_->Export( *OldMatrix_, *Exporter_, Insert );
00107 
00108   return true;
00109 }
00110 
00111 bool
00112 LinearProblem_MatrixTrans::
00113 rvs()
00114 {
00115   OldLHS_->Import( *NewLHS_, *Exporter_, Insert );
00116   OldRHS_->Import( *NewRHS_, *Exporter_, Insert );
00117   OldMatrix_->Import( *NewMatrix_, *Exporter_, Insert );
00118 
00119   return true;
00120 }
00121 
00122 } //namespace EpetraExt
00123 
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines