GammaWriter.h

#ifndef SRC_IO_GAMMAWRITER_H_
#define SRC_IO_GAMMAWRITER_H_
 
#include "plugins/PluginBase.h"
#include <string>
#include <fstream>
 
class ParticleContainer;
class DomainDecompBase; 
class Domain;
 
class XMLfileUnits;
class GammaWriter : public PluginBase {
public:
    GammaWriter() : _gammaStream(), _writeFrequency(1), _outputPrefix("mardyn"), _Gamma() {}
    ~~GammaWriter() {}
 
    void readXML(XMLfileUnits& xmlconfig);
    void init(ParticleContainer *particleContainer,
              DomainDecompBase *domainDecomp, Domain *domain);
    void endStep(
            ParticleContainer *particleContainer,
            DomainDecompBase *domainDecomp, Domain *domain,
            unsigned long simstep
    );
    void finish(ParticleContainer *particleContainer,
                DomainDecompBase *domainDecomp, Domain *domain);
    
    std::string getPluginName() {
        return std::string("GammaWriter");
    }
    static PluginBase* createInstance() { return new GammaWriter(); }
 
private:
    void calculateGamma(ParticleContainer* particleContainer, DomainDecompBase* domainDecom);
    double getGamma(unsigned id, double globalLength[3]);
    void resetGamma();
 
    std::ofstream _gammaStream;
    unsigned long _writeFrequency;
    std::string _outputPrefix;  
    std::map<unsigned,double> _Gamma;  
};
 
#endif  // SRC_IO_GAMMAWRITER_H_