| 34 |
|
|
| 35 |
|
#ifndef IO_STATWRITER_HPP |
| 36 |
|
#define IO_STATWRITER_HPP |
| 37 |
< |
|
| 38 |
< |
#include <string.h> |
| 39 |
< |
#include <iostream> |
| 37 |
> |
#include <bitset> |
| 38 |
|
#include <fstream> |
| 39 |
+ |
#include <iostream> |
| 40 |
+ |
#include <string.h> |
| 41 |
|
|
| 42 |
|
#include "brains/Stats.hpp" |
| 43 |
|
|
| 44 |
|
namespace oopse { |
| 45 |
|
|
| 46 |
< |
typedef std::bitset<Stats::ENDINDEX-Stats:BEGININDEX> StatsBitSet; |
| 46 |
> |
typedef std::bitset<Stats::ENDINDEX-Stats::BEGININDEX> StatsBitSet; |
| 47 |
|
|
| 48 |
|
/** |
| 49 |
|
* DefaultStatsBitSet includes |
| 50 |
< |
* StatsIndex::TOTALENERGY |
| 51 |
< |
* StatsIndex::TOTALPOTENTIAL |
| 50 |
> |
* StatsIndex::TIME |
| 51 |
> |
* StatsIndex::TOTAL_ENERGY |
| 52 |
> |
* StatsIndex::POTENTIAL_ENERGY |
| 53 |
> |
* StatsIndex::KINETIC_ENERGY |
| 54 |
|
* StatsIndex::TEMPERATURE |
| 55 |
|
* StatsIndex::PRESSURE |
| 56 |
< |
* StatsIndex::CONSERVEQUAN |
| 55 |
< |
* StatsIndex::TOTALENERGY |
| 56 |
> |
* StatsIndex::CONSERVED_QUANTITY |
| 57 |
|
*/ |
| 58 |
< |
const StatsBitSet DefaultStatsBitSet(63ul); |
| 58 |
> |
const StatsBitSet DefaultStatsBitSet(127ul); |
| 59 |
|
|
| 60 |
|
|
| 61 |
|
/** |
| 64 |
|
*/ |
| 65 |
|
class StatWriter { |
| 66 |
|
public: |
| 67 |
< |
StatWriter(const std::string& filename, StatsBitSet mask = DefaultStatsBitSet); |
| 67 |
> |
StatWriter(const std::string& filename, const StatsBitSet& mask = DefaultStatsBitSet); |
| 68 |
|
~StatWriter(); |
| 69 |
|
|
| 70 |
|
void writeTitle(); |
| 71 |
< |
void writeStat(Snapshot* s); |
| 71 |
> |
void writeStat(Stats* s); |
| 72 |
|
|
| 73 |
|
StatsBitSet getMask() { |
| 74 |
|
return mask_; |
| 76 |
|
|
| 77 |
|
private: |
| 78 |
|
|
| 79 |
< |
ofstream statfile_; |
| 79 |
> |
std::ofstream statfile_; |
| 80 |
|
StatsBitSet mask_; |
| 81 |
|
|
| 82 |
|
}; |
