| /* Copyright 2015. Los Alamos National Security, LLC. This material was produced |
| * under U.S. Government contract DE-AC52-06NA25396 for Los Alamos National |
| * Laboratory (LANL), which is operated by Los Alamos National Security, LLC |
| * for the U.S. Department of Energy. The U.S. Government has rights to use, |
| * reproduce, and distribute this software. NEITHER THE GOVERNMENT NOR LOS |
| * ALAMOS NATIONAL SECURITY, LLC MAKES ANY WARRANTY, EXPRESS OR IMPLIED, OR |
| * ASSUMES ANY LIABILITY FOR THE USE OF THIS SOFTWARE. If software is modified |
| * to produce derivative works, such modified software should be clearly marked, |
| * so as not to confuse it with the version available from LANL. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not |
| * use this file except in compliance with the License. You may obtain a copy |
| * of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software distributed |
| * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR |
| * CONDITIONS OF ANY KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations under the License. |
| * |
| * Under this license, it is required to include a reference to this work. We |
| * request that each derivative work contain a reference to LANL Copyright |
| * Disclosure C15076/LA-CC-15-054 so that this work's impact can be roughly |
| * measured. |
| * |
| * This is LANL Copyright Disclosure C15076/LA-CC-15-054 |
| */ |
| |
| /* |
| * PowerParser is a general purpose input file parser for software applications. |
| * |
| * Authors: Chuck Wingate XCP-2 caw@lanl.gov |
| * Robert Robey XCP-2 brobey@lanl.gov |
| */ |
| |
| #include <cstdlib> |
| #include <assert.h> |
| #include "Comm.hh" |
| |
| namespace PP { |
| |
| |
| // =========================================================================== |
| // Constructor |
| // =========================================================================== |
| Comm::Comm() |
| { |
| npes = 1; |
| mype = 0; |
| iope = 0; |
| |
| #ifdef HAVE_MPI |
| int argc = 1; |
| char **argv = NULL; |
| |
| int init_check; |
| MPI_Initialized(&init_check); |
| //printf("DEBUG -- mpi initialized %d\n",init_check); |
| |
| init_flag = 0; |
| if (! init_check) { |
| // Only way for init_flag to be true is here; must be false otherwise |
| init_flag = 1; |
| MPI_Init(&argc, &argv); |
| } |
| //printf("DEBUG -- comm constructor -- init_flag %d\n",init_flag); |
| |
| MPI_Comm_set_errhandler(MPI_COMM_WORLD, MPI_ERRORS_RETURN); |
| MPI_Comm_size(MPI_COMM_WORLD, &npes ); |
| MPI_Comm_rank(MPI_COMM_WORLD, &mype ); |
| #endif |
| } |
| |
| // =========================================================================== |
| // Destructor |
| // =========================================================================== |
| Comm::~Comm() |
| { |
| //printf("DEBUG -- comm destructor -- init_flag %d\n",init_flag); |
| #ifdef HAVE_MPI |
| if (init_flag) { |
| init_flag = 0; |
| MPI_Finalize(); |
| } |
| #endif |
| } |
| |
| // =========================================================================== |
| // Broadcast |
| // =========================================================================== |
| void Comm::broadcast(char *buffer, int count) |
| { |
| // To suppress compiler warnings of unused parameters |
| assert(buffer == buffer); |
| assert(count == count); |
| |
| if (npes == 1) return; |
| #ifdef HAVE_MPI |
| MPI_Bcast(buffer, count, MPI_CHAR, 0, MPI_COMM_WORLD); |
| #endif |
| } |
| |
| // =========================================================================== |
| // Error handling |
| // =========================================================================== |
| void Comm::global_abort_parser() |
| { |
| #ifdef HAVE_MPI |
| MPI_Abort(MPI_COMM_WORLD, 1); |
| #endif |
| exit(1); |
| } |
| // =========================================================================== |
| } // End of PP namespace |
| |
| |