| 1 | #include <cstring> | 
| 2 | #include <iostream> | 
| 3 | #include <fstream> | 
| 4 |  | 
| 5 | #include "ReadWrite.hpp" | 
| 6 | #include "simError.h" | 
| 7 |  | 
| 8 |  | 
| 9 | StatWriter::StatWriter( SimInfo* the_entry_plug ){ | 
| 10 |  | 
| 11 | entry_plug = the_entry_plug; | 
| 12 |  | 
| 13 | #ifdef IS_MPI | 
| 14 | if(worldRank == 0 ){ | 
| 15 | #endif // is_mpi | 
| 16 |  | 
| 17 | strcpy( outName, entry_plug->statusName ); | 
| 18 |  | 
| 19 | std::cerr << "Opening " << outName << " for stat\n"; | 
| 20 |  | 
| 21 | outFile.open(outName, ios::out | ios::trunc ); | 
| 22 |  | 
| 23 | if( !outFile ){ | 
| 24 |  | 
| 25 | sprintf( painCave.errMsg, | 
| 26 | "Could not open \"%s\" for stat output.\n", | 
| 27 | outName); | 
| 28 | painCave.isFatal = 1; | 
| 29 | simError(); | 
| 30 | } | 
| 31 |  | 
| 32 | //outFile.setf( ios::scientific ); | 
| 33 | outFile << "#time(fs)\ttot_E\tpotential\tkinetic\ttemperature\tpressure\n"; | 
| 34 |  | 
| 35 |  | 
| 36 |  | 
| 37 | #ifdef IS_MPI | 
| 38 | } | 
| 39 |  | 
| 40 | sprintf( checkPointMsg, | 
| 41 | "Sucessfully opened output file for stating.\n"); | 
| 42 | MPIcheckPoint(); | 
| 43 | #endif // is_mpi | 
| 44 |  | 
| 45 | tStats = new Thermo( entry_plug ); | 
| 46 | } | 
| 47 |  | 
| 48 | StatWriter::~StatWriter( ){ | 
| 49 |  | 
| 50 | #ifdef IS_MPI | 
| 51 | if(worldRank == 0 ){ | 
| 52 | #endif // is_mpi | 
| 53 |  | 
| 54 | outFile.close(); | 
| 55 | delete tStats; | 
| 56 |  | 
| 57 | #ifdef IS_MPI | 
| 58 | } | 
| 59 | #endif // is_mpi | 
| 60 | } | 
| 61 |  | 
| 62 | void StatWriter::writeStat( double currentTime ){ | 
| 63 |  | 
| 64 | const double p_convert =  1.63882576e8; | 
| 65 | double totE, potE, kinE, temp, press; | 
| 66 |  | 
| 67 | totE = tStats->getTotalE(); | 
| 68 | potE = tStats->getPotential(); | 
| 69 | kinE = tStats->getKinetic(); | 
| 70 | temp = tStats->getTemperature(); | 
| 71 | press = tStats->getPressure()*p_convert; | 
| 72 |  | 
| 73 | #ifdef IS_MPI | 
| 74 | if(worldRank == 0 ){ | 
| 75 | #endif // is_mpi | 
| 76 |  | 
| 77 | outFile.precision(8); | 
| 78 | outFile | 
| 79 | << currentTime << "\t" | 
| 80 | << totE << "\t" | 
| 81 | << potE << "\t" | 
| 82 | << kinE << "\t" | 
| 83 | << temp << "\t" | 
| 84 | << press << "\n"; | 
| 85 | outFile.flush(); | 
| 86 |  | 
| 87 | #ifdef IS_MPI | 
| 88 | } | 
| 89 | #endif // is_mpi | 
| 90 | } |