| [de061d] | 1 | /* | 
|---|
|  | 2 | *    vmg - a versatile multigrid solver | 
|---|
|  | 3 | *    Copyright (C) 2012 Institute for Numerical Simulation, University of Bonn | 
|---|
|  | 4 | * | 
|---|
|  | 5 | *  vmg is free software: you can redistribute it and/or modify | 
|---|
|  | 6 | *  it under the terms of the GNU General Public License as published by | 
|---|
|  | 7 | *  the Free Software Foundation, either version 3 of the License, or | 
|---|
|  | 8 | *  (at your option) any later version. | 
|---|
|  | 9 | * | 
|---|
|  | 10 | *  vmg is distributed in the hope that it will be useful, | 
|---|
|  | 11 | *  but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|---|
|  | 12 | *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
|---|
|  | 13 | *  GNU General Public License for more details. | 
|---|
|  | 14 | * | 
|---|
|  | 15 | *  You should have received a copy of the GNU General Public License | 
|---|
|  | 16 | *  along with this program.  If not, see <http://www.gnu.org/licenses/>. | 
|---|
|  | 17 | */ | 
|---|
|  | 18 |  | 
|---|
|  | 19 | /** | 
|---|
|  | 20 | * @file   command_list.cpp | 
|---|
|  | 21 | * @author Julian Iseringhausen <isering@ins.uni-bonn.de> | 
|---|
|  | 22 | * @date   Tue Apr  5 20:15:06 2011 | 
|---|
|  | 23 | * | 
|---|
|  | 24 | */ | 
|---|
|  | 25 |  | 
|---|
|  | 26 | #ifdef HAVE_CONFIG_H | 
|---|
|  | 27 | #include <libvmg_config.h> | 
|---|
|  | 28 | #endif | 
|---|
|  | 29 |  | 
|---|
|  | 30 | #ifdef DEBUG_BARRIER | 
|---|
|  | 31 | #ifdef HAVE_MPI | 
|---|
|  | 32 | #include <mpi.h> | 
|---|
|  | 33 | #ifdef HAVE_MARMOT | 
|---|
|  | 34 | #include <enhancempicalls.h> | 
|---|
|  | 35 | #include <sourceinfompicalls.h> | 
|---|
|  | 36 | #endif | 
|---|
|  | 37 | #endif | 
|---|
|  | 38 | #endif | 
|---|
|  | 39 |  | 
|---|
|  | 40 | #include <cstdio> | 
|---|
|  | 41 |  | 
|---|
|  | 42 | #include "base/command.hpp" | 
|---|
|  | 43 | #include "base/command_factory.hpp" | 
|---|
|  | 44 | #include "base/command_list.hpp" | 
|---|
|  | 45 | #include "base/timer.hpp" | 
|---|
|  | 46 | #include "base/defs.hpp" | 
|---|
|  | 47 | #include "mg.hpp" | 
|---|
|  | 48 |  | 
|---|
|  | 49 | using namespace VMG; | 
|---|
|  | 50 |  | 
|---|
|  | 51 | Request CommandList::ExecuteList() | 
|---|
|  | 52 | { | 
|---|
|  | 53 | Request request; | 
|---|
|  | 54 | Request final_request = (commands.size() == 0 ? StopCycleNow : Continue); | 
|---|
|  | 55 |  | 
|---|
|  | 56 | for (CommandList::iterator iter=commands.begin(); iter!=commands.end(); ++iter) { | 
|---|
|  | 57 |  | 
|---|
|  | 58 | #ifdef DEBUG | 
|---|
|  | 59 | const int num_args = (iter->second.size() > 1 ? iter->second.size() : (iter->second[0] == "" ? 0 : 1)); | 
|---|
|  | 60 | MG::GetCommands().CheckNumberOfArguments(iter->first, num_args); | 
|---|
|  | 61 | #endif | 
|---|
|  | 62 |  | 
|---|
|  | 63 | #ifdef DEBUG_BARRIER | 
|---|
|  | 64 | Comm& comm = *MG::GetComm(); | 
|---|
|  | 65 | comm.Barrier(); | 
|---|
|  | 66 | comm.PrintOnce(Debug, "Command \"%s\" start", iter->first.c_str()); | 
|---|
|  | 67 | #endif | 
|---|
|  | 68 |  | 
|---|
|  | 69 | Timer::Start(iter->first); | 
|---|
|  | 70 | request = MG::GetCommands().Get(iter->first)->Run(iter->second); | 
|---|
|  | 71 | Timer::Stop(iter->first); | 
|---|
|  | 72 |  | 
|---|
|  | 73 | #ifdef DEBUG_BARRIER | 
|---|
|  | 74 | comm.Barrier(); | 
|---|
|  | 75 | comm.PrintOnce(Debug, "Command \"%s\" done", iter->first.c_str()); | 
|---|
|  | 76 | #endif | 
|---|
|  | 77 |  | 
|---|
|  | 78 | if (request == StopCycleLater) | 
|---|
|  | 79 | final_request = StopCycleNow; | 
|---|
|  | 80 | else if (request == StopCycleNow) { | 
|---|
|  | 81 | final_request = StopCycleNow; | 
|---|
|  | 82 | break; | 
|---|
|  | 83 | } | 
|---|
|  | 84 | } | 
|---|
|  | 85 |  | 
|---|
|  | 86 | return final_request; | 
|---|
|  | 87 | } | 
|---|
|  | 88 |  | 
|---|
|  | 89 | void CommandList::AddCommand(std::string command, std::string arguments) | 
|---|
|  | 90 | { | 
|---|
|  | 91 | std::vector<std::string> argument_list; | 
|---|
|  | 92 | size_t pos; | 
|---|
|  | 93 |  | 
|---|
|  | 94 | do { | 
|---|
|  | 95 | pos = arguments.find(':'); | 
|---|
|  | 96 | argument_list.push_back(arguments.substr(0, pos)); | 
|---|
|  | 97 | arguments.erase(0, pos+1); | 
|---|
|  | 98 | }while (pos != std::string::npos); | 
|---|
|  | 99 |  | 
|---|
|  | 100 | commands.push_back(std::pair<std::string, std::vector<std::string> >(command, argument_list)); | 
|---|
|  | 101 | } | 
|---|
|  | 102 |  | 
|---|
|  | 103 | void CommandList::DeleteCommand(const CommandList::iterator& iter) | 
|---|
|  | 104 | { | 
|---|
|  | 105 | commands.erase(iter); | 
|---|
|  | 106 | } | 
|---|
|  | 107 |  | 
|---|
|  | 108 | void CommandList::DeleteCommand(const std::string& _command) | 
|---|
|  | 109 | { | 
|---|
|  | 110 | CommandList::iterator iter = commands.begin(); | 
|---|
|  | 111 | for (;iter != commands.end(); ++iter) | 
|---|
|  | 112 | if (iter->first == _command) | 
|---|
|  | 113 | break; | 
|---|
|  | 114 | assert( iter != commands.end() ); | 
|---|
|  | 115 | DeleteCommand(iter); | 
|---|
|  | 116 | } | 
|---|
|  | 117 |  | 
|---|
|  | 118 | void CommandList::Print() | 
|---|
|  | 119 | { | 
|---|
|  | 120 | for (CommandList::iterator iter=commands.begin(); iter!=commands.end(); ++iter) | 
|---|
|  | 121 | printf("%s\n", (*iter).first.c_str()); | 
|---|
|  | 122 | } | 
|---|
|  | 123 |  | 
|---|
|  | 124 | void CommandList::Clear() | 
|---|
|  | 125 | { | 
|---|
|  | 126 | commands.clear(); | 
|---|
|  | 127 | } | 
|---|