| 1 | < | /* | 
| 1 | > | /* | 
| 2 |  | * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved. | 
| 3 |  | * | 
| 4 |  | * The University of Notre Dame grants you ("Licensee") a | 
| 46 |  | #include "utils/simError.h" | 
| 47 |  |  | 
| 48 |  | namespace oopse { | 
| 49 | < | StatWriter::StatWriter( const std::string& filename, const StatsBitSet& mask) : mask_(mask){ | 
| 49 | > | StatsBitSet parseStatFileFormat(const std::string& format) { | 
| 50 | > | StringTokenizer tokenizer(format, " ,;|\t\n\r"); | 
| 51 | > | StatsBitSet mask; | 
| 52 | > | while(tokenizer.hasMoreTokens()) { | 
| 53 | > | std::string token(tokenizer.nextToken()); | 
| 54 | > | toUpper(token); | 
| 55 | > | Stats::StatsMapType::iterator i = Stats::statsMap.find(token); | 
| 56 | > | if (i != Stats::statsMap.end()) { | 
| 57 | > | mask.set(i->second); | 
| 58 | > | } else { | 
| 59 | > | sprintf( painCave.errMsg, | 
| 60 | > | "%s is not a valid statFileFormat keyword.\n", token.c_str() ); | 
| 61 | > | painCave.isFatal = 0; | 
| 62 | > | painCave.severity = OOPSE_ERROR; | 
| 63 | > | simError(); | 
| 64 | > | } | 
| 65 | > | } | 
| 66 | > |  | 
| 67 | > | return mask; | 
| 68 | > | } | 
| 69 | > |  | 
| 70 | > | StatWriter::StatWriter( const std::string& filename, const StatsBitSet& mask) : mask_(mask){ | 
| 71 |  |  | 
| 72 |  | #ifdef IS_MPI | 
| 73 | < | if(worldRank == 0 ){ | 
| 73 | > | if(worldRank == 0 ){ | 
| 74 |  | #endif // is_mpi | 
| 75 |  |  | 
| 76 | < | statfile_.open(filename.c_str(), std::ios::out | std::ios::trunc ); | 
| 76 | > | statfile_.open(filename.c_str(), std::ios::out | std::ios::trunc ); | 
| 77 |  |  | 
| 78 | < | if( !statfile_ ){ | 
| 78 | > | if( !statfile_ ){ | 
| 79 |  |  | 
| 80 | < | sprintf( painCave.errMsg, | 
| 81 | < | "Could not open \"%s\" for stat output.\n", | 
| 82 | < | filename.c_str()); | 
| 83 | < | painCave.isFatal = 1; | 
| 84 | < | simError(); | 
| 85 | < | } | 
| 80 | > | sprintf( painCave.errMsg, | 
| 81 | > | "Could not open \"%s\" for stat output.\n", | 
| 82 | > | filename.c_str()); | 
| 83 | > | painCave.isFatal = 1; | 
| 84 | > | simError(); | 
| 85 | > | } | 
| 86 |  |  | 
| 87 | < | writeTitle(); | 
| 87 | > | writeTitle(); | 
| 88 |  |  | 
| 89 |  | #ifdef IS_MPI | 
| 90 | < | } | 
| 90 | > | } | 
| 91 |  |  | 
| 92 | < | sprintf( checkPointMsg, | 
| 93 | < | "Sucessfully opened output file for stating.\n"); | 
| 94 | < | MPIcheckPoint(); | 
| 92 | > | sprintf( checkPointMsg, | 
| 93 | > | "Sucessfully opened output file for stating.\n"); | 
| 94 | > | MPIcheckPoint(); | 
| 95 |  | #endif // is_mpi | 
| 96 |  |  | 
| 97 | < | } | 
| 97 | > | } | 
| 98 |  |  | 
| 99 | < | StatWriter::~StatWriter( ){ | 
| 99 | > | StatWriter::~StatWriter( ){ | 
| 100 |  |  | 
| 101 |  | #ifdef IS_MPI | 
| 102 | < | if(worldRank == 0 ){ | 
| 102 | > | if(worldRank == 0 ){ | 
| 103 |  | #endif // is_mpi | 
| 104 |  |  | 
| 105 | < | statfile_.close(); | 
| 105 | > | statfile_.close(); | 
| 106 |  |  | 
| 107 |  | #ifdef IS_MPI | 
| 108 | < | } | 
| 108 | > | } | 
| 109 |  | #endif // is_mpi | 
| 110 | < | } | 
| 110 | > | } | 
| 111 |  |  | 
| 112 | < | void StatWriter::writeTitle() { | 
| 112 | > | void StatWriter::writeTitle() { | 
| 113 |  |  | 
| 114 |  |  | 
| 115 |  | #ifdef IS_MPI | 
| 116 |  | if(worldRank == 0 ){ | 
| 117 |  | #endif // is_mpi | 
| 118 |  |  | 
| 119 | < | //write title | 
| 120 | < | statfile_ << "#"; | 
| 121 | < | for (int i =0; i < mask_.size(); ++i) { | 
| 122 | < | if (mask_[i]) { | 
| 123 | < | statfile_ << "\t" << Stats::getTitle(i) << "(" << Stats::getUnits(i) << ")"; | 
| 124 | < | } | 
| 125 | < | } | 
| 126 | < | statfile_ << std::endl; | 
| 119 | > | //write title | 
| 120 | > | statfile_ << "#"; | 
| 121 | > | for (int i =0; i < mask_.size(); ++i) { | 
| 122 | > | if (mask_[i]) { | 
| 123 | > | statfile_ << "\t" << Stats::getTitle(i) << "(" << Stats::getUnits(i) << ")"; | 
| 124 | > | } | 
| 125 | > | } | 
| 126 | > | statfile_ << std::endl; | 
| 127 |  |  | 
| 128 |  | #ifdef IS_MPI | 
| 129 | < | } | 
| 129 | > | } | 
| 130 |  | #endif // is_mpi | 
| 131 | < | } | 
| 131 | > | } | 
| 132 |  |  | 
| 133 | < | void StatWriter::writeStat(const Stats& s){ | 
| 133 | > | void StatWriter::writeStat(const Stats& s){ | 
| 134 |  |  | 
| 135 |  | #ifdef IS_MPI | 
| 136 |  | if(worldRank == 0 ){ | 
| 137 |  | #endif // is_mpi | 
| 138 |  |  | 
| 139 | < | statfile_.precision(8); | 
| 139 | > | statfile_.precision(8); | 
| 140 |  |  | 
| 141 | < | for (int i =0; i < mask_.size(); ++i) { | 
| 142 | < | if (mask_[i]) { | 
| 143 | < | statfile_ << "\t" << s[i]; | 
| 144 | < | } | 
| 145 | < | } | 
| 146 | < | statfile_ << std::endl; | 
| 141 | > | for (int i =0; i < mask_.size(); ++i) { | 
| 142 | > | if (mask_[i]) { | 
| 143 | > | statfile_ << "\t" << s[i]; | 
| 144 | > | } | 
| 145 | > | } | 
| 146 | > | statfile_ << std::endl; | 
| 147 |  |  | 
| 148 | < | statfile_.flush(); | 
| 148 | > | statfile_.flush(); | 
| 149 |  |  | 
| 150 |  | #ifdef IS_MPI | 
| 151 |  | } | 
| 152 |  | #endif // is_mpi | 
| 153 | < | } | 
| 153 | > | } | 
| 154 |  |  | 
| 155 |  | } |