Teuchos - Trilinos Tools Package Version of the Day
Teuchos_MPISession.cpp
00001 // @HEADER
00002 // ***********************************************************************
00003 //
00004 //                    Teuchos: Common Tools Package
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 Michael A. Heroux (maherou@sandia.gov)
00038 //
00039 // ***********************************************************************
00040 // @HEADER
00041 
00042 #include "Teuchos_MPISession.hpp"
00043 #include "Teuchos_TestForException.hpp"
00044 
00045 using namespace Teuchos;
00046 
00047 int MPISession::rank_ = 0 ;
00048 int MPISession::nProc_ = 1 ;
00049 
00050 void MPISession::init(int* argc, void*** argv)
00051 {
00052 #ifdef HAVE_MPI
00053   /* initialize MPI */
00054   int mpiHasBeenStarted = 0;
00055   MPI_Initialized(& mpiHasBeenStarted);
00056   int mpierr = 0 ;
00057   if (!mpiHasBeenStarted)
00058     {
00059       mpierr = ::MPI_Init (argc, (char ***) argv);
00060       TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00061                          "Error code=" << mpierr 
00062                          << " detected in MPI_Init()");
00063     }
00064   
00065   /* find rank */
00066   mpierr = ::MPI_Comm_rank (MPI_COMM_WORLD, &rank_);
00067   TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00068                      "Error code=" << mpierr 
00069                      << " detected in MPI_Comm_rank()");
00070 
00071   /* find number of procs */
00072   mpierr = ::MPI_Comm_size (MPI_COMM_WORLD, &nProc_);
00073 
00074   TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00075                      "Error code=" << mpierr 
00076                      << " detected in MPI_Comm_size()");
00077 
00078   /* get machine name */
00079   int nameLen;
00080   char procName[MPI_MAX_PROCESSOR_NAME];
00081   mpierr = ::MPI_Get_processor_name(procName,&nameLen);
00082 
00083   TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00084                      "Error code=" << mpierr 
00085                      << " detected in MPI_Get_processor_name()");
00086 
00087   if (showStartupMessage())
00088     {
00089       std::cerr << "Teuchos::MPISession::init() started processor " 
00090            << procName << std::endl;
00091     }
00092   else
00093     {
00094 #else
00095   std::cerr << "Teuchos::MPISession::init() started serial run" << std::endl;
00096 #endif
00097 #ifdef HAVE_MPI
00098     }
00099 #endif
00100 }
00101 
00102 void MPISession::finalize()
00103 {
00104 #ifdef HAVE_MPI
00105   int mpierr = ::MPI_Finalize();
00106 
00107   TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00108                      "Error code=" << mpierr << " detected in MPI_Finalize()");
00109 #endif
00110 }
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines