Tpetra_MpiPlatform.hpp

00001 // @HEADER
00002 // ***********************************************************************
00003 // 
00004 //          Tpetra: Templated Linear Algebra Services Package
00005 //                 Copyright (2008) 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 // This library is free software; you can redistribute it and/or modify
00011 // it under the terms of the GNU Lesser General Public License as
00012 // published by the Free Software Foundation; either version 2.1 of the
00013 // License, or (at your option) any later version.
00014 //  
00015 // This library is distributed in the hope that it will be useful, but
00016 // WITHOUT ANY WARRANTY; without even the implied warranty of
00017 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00018 // Lesser General Public License for more details.
00019 //  
00020 // You should have received a copy of the GNU Lesser General Public
00021 // License along with this library; if not, write to the Free Software
00022 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
00023 // USA
00024 // Questions? Contact Michael A. Heroux (maherou@sandia.gov) 
00025 // 
00026 // ***********************************************************************
00027 // @HEADER
00028 
00029 #ifndef TPETRA_MPIPLATFORM_HPP
00030 #define TPETRA_MPIPLATFORM_HPP
00031 
00032 #include <Teuchos_DefaultMpiComm.hpp>
00033 #include <Teuchos_RCP.hpp>
00034 #include <Teuchos_Comm.hpp>
00035 #include <Teuchos_Describable.hpp>
00036 #include <Kokkos_DefaultNode.hpp>
00037 
00038 namespace Tpetra {
00039 
00041 
00046   template<class Node=Kokkos::DefaultNode::DefaultNodeType>
00047   class MpiPlatform : public Teuchos::Describable {
00048     public:
00050       typedef Node NodeType;
00052 
00053 
00055       explicit MpiPlatform(Teuchos::RCP<Node> node);
00056 
00058       MpiPlatform(Teuchos::RCP<Node> node, const Teuchos::RCP<const Teuchos::OpaqueWrapper<MPI_Comm> > &rawMpiComm);
00059 
00061       ~MpiPlatform();
00062 
00064 
00066 
00067 
00069       Teuchos::RCP< const Teuchos::Comm<int> > getComm() const;
00070 
00072       Teuchos::RCP<Node> getNode() const;
00073 
00075 
00076     protected: 
00078       Teuchos::RCP<Node> node_;
00079 
00080     private:
00081       Teuchos::RCP<Teuchos::MpiComm<int> > comm_;
00082       MpiPlatform(const MpiPlatform<Node> &platform);
00083   };
00084 
00085   template <class Node>
00086   MpiPlatform<Node>::MpiPlatform(Teuchos::RCP<Node> node, const Teuchos::RCP<const Teuchos::OpaqueWrapper<MPI_Comm> > &rawMpiComm)
00087   : node_(node) {
00088     comm_ = Teuchos::createMpiComm<int>(rawMpiComm);
00089   }
00090 
00091   template <class Node>
00092   MpiPlatform<Node>::MpiPlatform(Teuchos::RCP<Node> node)
00093   : node_(node) {
00094     comm_ = Teuchos::createMpiComm<int>(Teuchos::opaqueWrapper<MPI_Comm>(MPI_COMM_WORLD));
00095   } 
00096 
00097   template <class Node>
00098   MpiPlatform<Node>::~MpiPlatform() {}
00099 
00100   template <class Node>
00101   MpiPlatform<Node>::MpiPlatform(const MpiPlatform<Node> &platform) {
00102     comm_ = platform.comm_;
00103   }
00104 
00105   template <class Node>
00106   Teuchos::RCP< const Teuchos::Comm<int> > 
00107   MpiPlatform<Node>::getComm() const {
00108     return comm_;
00109   }
00110 
00111   template <class Node>
00112   Teuchos::RCP<Node> MpiPlatform<Node>::getNode() const 
00113   { return node_; }
00114 
00115 } // namespace Tpetra
00116 
00117 #endif // TPETRA_MPIPLATFORM_HPP
00118 
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines
Generated on Wed Apr 13 10:21:41 2011 for Tpetra Matrix/Vector Services by  doxygen 1.6.3