Teko Version of the Day
Teko_InverseFactoryOperator.cpp
00001 /*
00002 // @HEADER
00003 // 
00004 // ***********************************************************************
00005 // 
00006 //      Teko: A package for block and physics based preconditioning
00007 //                  Copyright 2010 Sandia Corporation 
00008 //  
00009 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
00010 // the U.S. Government retains certain rights in this software.
00011 //  
00012 // Redistribution and use in source and binary forms, with or without
00013 // modification, are permitted provided that the following conditions are
00014 // met:
00015 //  
00016 // 1. Redistributions of source code must retain the above copyright
00017 // notice, this list of conditions and the following disclaimer.
00018 //  
00019 // 2. Redistributions in binary form must reproduce the above copyright
00020 // notice, this list of conditions and the following disclaimer in the
00021 // documentation and/or other materials provided with the distribution.
00022 //  
00023 // 3. Neither the name of the Corporation nor the names of the
00024 // contributors may be used to endorse or promote products derived from
00025 // this software without specific prior written permission. 
00026 //  
00027 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
00028 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00029 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00030 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
00031 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00032 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00033 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00034 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00035 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
00036 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00037 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00038 //  
00039 // Questions? Contact Eric C. Cyr (eccyr@sandia.gov)
00040 // 
00041 // ***********************************************************************
00042 // 
00043 // @HEADER
00044 
00045 */
00046 
00047 #include "Teko_InverseFactoryOperator.hpp"
00048 
00049 // Teko includes
00050 #include "Teko_BasicMappingStrategy.hpp"
00051 
00052 // Thyra includes
00053 #include "Thyra_EpetraLinearOp.hpp"
00054 
00055 using Teuchos::RCP;
00056 using Teuchos::rcp;
00057 using Teuchos::rcpFromRef;
00058 using Teuchos::rcp_dynamic_cast;
00059 
00060 namespace Teko {
00061 namespace Epetra {
00062 
00069 InverseFactoryOperator::InverseFactoryOperator(const Teuchos::RCP<const InverseFactory> & ifp)
00070    : inverseFactory_(ifp), firstBuildComplete_(false), setConstFwdOp_(true)
00071 {
00072 }
00073 
00086 void InverseFactoryOperator::initInverse(bool clearOld)
00087 {
00088    if(not clearOld)
00089       return;
00090    invOperator_ = Teuchos::null;
00091 }
00092 
00105 void InverseFactoryOperator::buildInverseOperator(const Teuchos::RCP<const Epetra_Operator> & A,bool clear)
00106 {
00107    Teko_DEBUG_SCOPE("InverseFactoryOperator::buildInverseOperator",10);
00108 
00109    // extract EpetraOperatorWrapper (throw on failure) and corresponding thyra operator
00110    RCP<const Thyra::LinearOpBase<double> > thyraA = extractLinearOp(A);
00111 
00112    // set the mapping strategy
00113    SetMapStrategy(rcp(new InverseMappingStrategy(extractMappingStrategy(A))));
00114 
00115    initInverse(clear);
00116 
00117    // actually build the inverse operator
00118    invOperator_ = Teko::buildInverse(*inverseFactory_,thyraA); 
00119 
00120    SetOperator(invOperator_,false);
00121 
00122    firstBuildComplete_ = true;
00123 
00124    if(setConstFwdOp_)
00125       fwdOp_ = A;
00126 
00127    setConstFwdOp_ = true;
00128 
00129    TEUCHOS_ASSERT(invOperator_!=Teuchos::null);
00130    TEUCHOS_ASSERT(getForwardOp()!=Teuchos::null);
00131    TEUCHOS_ASSERT(getThyraOp()!=Teuchos::null);
00132    TEUCHOS_ASSERT(getMapStrategy()!=Teuchos::null);
00133    TEUCHOS_ASSERT(firstBuildComplete_==true);
00134 }
00135 
00136 void InverseFactoryOperator::buildInverseOperator(const Teuchos::RCP<Epetra_Operator> & A,bool clear)
00137 {
00138    setConstFwdOp_ = false;
00139 
00140    fwdOp_.initialize(A);
00141 
00142    buildInverseOperator(A.getConst());
00143 
00144    TEUCHOS_ASSERT(setConstFwdOp_==true);
00145 }
00146 
00159 void InverseFactoryOperator::rebuildInverseOperator(const Teuchos::RCP<const Epetra_Operator> & A)
00160 {
00161    Teko_DEBUG_SCOPE("InverseFactoryOperator::rebuildPreconditioner",10);
00162 
00163    // if the inverse hasn't been built yet, rebuild from scratch
00164    if(not firstBuildComplete_) {
00165       buildInverseOperator(A,false);
00166       return;
00167    }
00168 
00169    RCP<const Thyra::LinearOpBase<double> > thyraA = extractLinearOp(A);
00170    Teko::rebuildInverse(*inverseFactory_,thyraA,invOperator_); 
00171 
00172    if(setConstFwdOp_)
00173       fwdOp_.initialize(A);
00174 
00175    SetOperator(invOperator_,false);
00176 
00177    setConstFwdOp_ = true;
00178 
00179    TEUCHOS_ASSERT(getForwardOp()!=Teuchos::null);
00180    TEUCHOS_ASSERT(invOperator_!=Teuchos::null);
00181    TEUCHOS_ASSERT(getThyraOp()!=Teuchos::null);
00182    TEUCHOS_ASSERT(firstBuildComplete_==true);
00183 }
00184 
00185 void InverseFactoryOperator::rebuildInverseOperator(const Teuchos::RCP<Epetra_Operator> & A)
00186 {
00187    setConstFwdOp_ = false;
00188 
00189    fwdOp_.initialize(A);
00190  
00191    // build from constant epetra operator
00192    rebuildInverseOperator(A.getConst());
00193 
00194    TEUCHOS_ASSERT(setConstFwdOp_==true);
00195 }
00196 
00197 Teuchos::RCP<const Thyra::LinearOpBase<double> > InverseFactoryOperator::extractLinearOp(const Teuchos::RCP<const Epetra_Operator> & A) const
00198 {
00199    // extract EpetraOperatorWrapper (throw on failure) and corresponding thyra operator
00200    const RCP<const EpetraOperatorWrapper> & eow = rcp_dynamic_cast<const EpetraOperatorWrapper>(A);
00201   
00202    // if it is an EpetraOperatorWrapper, then get the Thyra operator
00203    if(eow!=Teuchos::null)
00204       return eow->getThyraOp(); 
00205 
00206    // otherwise wrap it up as a thyra operator 
00207    return Thyra::epetraLinearOp(A);
00208 }
00209 
00210 Teuchos::RCP<const MappingStrategy> InverseFactoryOperator::extractMappingStrategy(const Teuchos::RCP<const Epetra_Operator> & A) const
00211 {
00212    // extract EpetraOperatorWrapper (throw on failure) and corresponding thyra operator
00213    const RCP<const EpetraOperatorWrapper> & eow = rcp_dynamic_cast<const EpetraOperatorWrapper>(A);
00214   
00215    // if it is an EpetraOperatorWrapper, then get the Thyra operator
00216    if(eow!=Teuchos::null)
00217       return eow->getMapStrategy(); 
00218 
00219    // otherwise wrap it up as a thyra operator 
00220    RCP<const Epetra_Map> range = rcpFromRef(A->OperatorRangeMap());
00221    RCP<const Epetra_Map> domain = rcpFromRef(A->OperatorDomainMap());
00222    return rcp(new BasicMappingStrategy(range,domain,A->Comm()));
00223 }
00224 
00225 } // end namespace Epetra
00226 } // end namespace Teko
 All Classes Files Functions Variables