Tpetra Matrix/Vector Services Version of the Day
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_Describable.hpp>
00034 #include <Kokkos_DefaultNode.hpp>
00035 
00036 namespace Tpetra {
00037 
00039 
00044   template<class Node=Kokkos::DefaultNode::DefaultNodeType>
00045   class MpiPlatform : public Teuchos::Describable {
00046     public:
00048       typedef Node NodeType;
00050 
00051 
00053       explicit MpiPlatform(Teuchos::RCP<Node> node);
00054 
00056       MpiPlatform(Teuchos::RCP<Node> node, const Teuchos::RCP<const Teuchos::OpaqueWrapper<MPI_Comm> > &rawMpiComm);
00057 
00059       ~MpiPlatform();
00060 
00062 
00064 
00065 
00067       Teuchos::RCP< const Teuchos::Comm<int> > getComm() const;
00068 
00070       Teuchos::RCP<Node> getNode() const;
00071 
00073 
00074     protected: 
00076       Teuchos::RCP<Node> node_;
00077 
00078     private:
00079       Teuchos::RCP<Teuchos::MpiComm<int> > comm_;
00080       MpiPlatform(const MpiPlatform<Node> &platform);
00081   };
00082 
00083   template <class Node>
00084   MpiPlatform<Node>::MpiPlatform(Teuchos::RCP<Node> node, const Teuchos::RCP<const Teuchos::OpaqueWrapper<MPI_Comm> > &rawMpiComm)
00085   : node_(node) {
00086     comm_ = Teuchos::createMpiComm<int>(rawMpiComm);
00087   }
00088 
00089   template <class Node>
00090   MpiPlatform<Node>::MpiPlatform(Teuchos::RCP<Node> node)
00091   : node_(node) {
00092     comm_ = Teuchos::createMpiComm<int>(Teuchos::opaqueWrapper<MPI_Comm>(MPI_COMM_WORLD));
00093   } 
00094 
00095   template <class Node>
00096   MpiPlatform<Node>::~MpiPlatform() {}
00097 
00098   template <class Node>
00099   MpiPlatform<Node>::MpiPlatform(const MpiPlatform<Node> &platform) {
00100     comm_ = platform.comm_;
00101   }
00102 
00103   template <class Node>
00104   Teuchos::RCP< const Teuchos::Comm<int> > 
00105   MpiPlatform<Node>::getComm() const {
00106     return comm_;
00107   }
00108 
00109   template <class Node>
00110   Teuchos::RCP<Node> MpiPlatform<Node>::getNode() const 
00111   { return node_; }
00112 
00113 } // namespace Tpetra
00114 
00115 #endif // TPETRA_MPIPLATFORM_HPP
00116 
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines