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:
00049       typedef Node NodeType;
00051 
00052 
00054       explicit MpiPlatform(Teuchos::RCP<Node> node);
00055 
00057       MpiPlatform(Teuchos::RCP<Node> node, const Teuchos::RCP<const Teuchos::OpaqueWrapper<MPI_Comm> > &rawMpiComm);
00058 
00060       ~MpiPlatform();
00061 
00063 
00065 
00066 
00068       Teuchos::RCP< const Teuchos::Comm<int> > getComm() const;
00069 
00071       Teuchos::RCP<Node> getNode() const;
00072 
00074 
00075     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 

Generated on Wed May 12 21:40:14 2010 for Tpetra Matrix/Vector Services by  doxygen 1.4.7