Thyra Package Browser (Single Doxygen Collection) Version of the Day
Thyra_TpetraVectorSpace_def.hpp
Go to the documentation of this file.
00001 // @HEADER
00002 // ***********************************************************************
00003 // 
00004 //    Thyra: Interfaces and Support for Abstract Numerical Algorithms
00005 //                 Copyright (2004) Sandia Corporation
00006 // 
00007 // Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
00008 // license for use of this work by or on behalf of the U.S. Government.
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 Roscoe A. Bartlett (bartlettra@ornl.gov) 
00038 // 
00039 // ***********************************************************************
00040 // @HEADER
00041 
00042 
00043 #ifndef THYRA_TPETRA_VECTOR_SPACE_HPP
00044 #define THYRA_TPETRA_VECTOR_SPACE_HPP
00045 
00046 
00047 #include "Thyra_TpetraVectorSpace_decl.hpp"
00048 #include "Thyra_TpetraThyraWrappers.hpp"
00049 #include "Thyra_TpetraVector.hpp"
00050 #include "Thyra_TpetraMultiVector.hpp"
00051 
00052 
00053 namespace Thyra {
00054 
00055 
00056 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00057 RCP<TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node> >
00058 TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::create()
00059 {
00060   const RCP<this_t> vs(new this_t);
00061   vs->weakSelfPtr_ = vs.create_weak();
00062   return vs;
00063 }
00064 
00065 
00066 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00067 void TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::initialize(
00068   const RCP<const Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node> > &tpetraMap
00069   )
00070 {
00071   comm_ = convertTpetraToThyraComm(tpetraMap->getComm());
00072   tpetraMap_ = tpetraMap;
00073   localSubDim_ = tpetraMap->getNodeNumElements();
00074   numProc_ = comm_->getSize();
00075   procRank_ = comm_->getRank();
00076   this->updateState(tpetraMap->getGlobalNumElements());
00077 }
00078 
00079 
00080 // Overridden from VectorSpace
00081 
00082 
00083 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00084 RCP<VectorBase<Scalar> >
00085 TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::createMember() const
00086 {
00087   return tpetraVector<Scalar>(
00088     weakSelfPtr_.create_strong().getConst(),
00089     Teuchos::rcp(
00090       new Tpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node>(tpetraMap_, false)
00091       )
00092     );
00093 }
00094 
00095 
00096 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00097 RCP< MultiVectorBase<Scalar> >
00098 TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::createMembers(int numMembers) const
00099 {
00100   return tpetraMultiVector<Scalar>(
00101     weakSelfPtr_.create_strong().getConst(),
00102     tpetraVectorSpace<Scalar>(
00103       Tpetra::createLocalMapWithNode<LocalOrdinal, GlobalOrdinal, Node>(
00104         numMembers, tpetraMap_->getComm(), tpetraMap_->getNode()
00105         )
00106       ),
00107     Teuchos::rcp(
00108       new Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>(
00109         tpetraMap_, numMembers, false)
00110       )
00111     );
00112   // ToDo: Create wrapper function to create locally replicated vector space
00113   // and use it.
00114 }
00115 
00116 
00117 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00118 bool TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::hasInCoreView(
00119   const Range1D& rng_in, const EViewType viewType, const EStrideType strideType
00120   ) const
00121 {
00122   const Range1D rng = full_range(rng_in,0,this->dim()-1);
00123   const Ordinal l_localOffset = this->localOffset();
00124   return ( l_localOffset<=rng.lbound() && rng.ubound()<l_localOffset+localSubDim_ );
00125 }
00126 
00127 
00128 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00129 RCP< const VectorSpaceBase<Scalar> >
00130 TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::clone() const
00131 {
00132   return tpetraVectorSpace<Scalar>(tpetraMap_);
00133 }
00134 
00135 
00136 // Overridden from SpmdVectorSpaceDefaultBase
00137 
00138 
00139 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00140 RCP<const Teuchos::Comm<Ordinal> >
00141 TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::getComm() const
00142 {
00143   return comm_;
00144 }
00145 
00146 
00147 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00148 Ordinal TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::localSubDim() const
00149 {
00150   return localSubDim_;
00151 }
00152 
00153 
00154 // private
00155 
00156 
00157 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
00158 TpetraVectorSpace<Scalar,LocalOrdinal,GlobalOrdinal,Node>::TpetraVectorSpace()
00159   :localSubDim_(-1), numProc_(-1), procRank_(-1)
00160 {
00161   // The base classes should automatically default initialize to a safe
00162   // uninitialized state.
00163 }
00164 
00165 
00166 } // end namespace Thyra
00167 
00168 
00169 #endif // THYRA_TPETRA_VECTOR_SPACE_HPP
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines