FlopRateWriter.h

/*
 * FlopRateWriter.h
 *
 *  Created on: 8 May 2017
 *      Author: tchipevn
 */
 
#ifndef SRC_IO_FLOPRATEWRITER_H_
#define SRC_IO_FLOPRATEWRITER_H_
 
#include "plugins/PluginBase.h"
 
#include <fstream>
 
class FlopCounter;
 
class FlopRateWriter: public PluginBase {
public:
    FlopRateWriter() : PluginBase(), _flopCounter(nullptr) {}
    ~~FlopRateWriter() {}
 
    void init(ParticleContainer *particleContainer,
              DomainDecompBase *domainDecomp, Domain *domain);
 
    void readXML(XMLfileUnits& /*xmlconfig*/);
 
 
    void afterForces(
            ParticleContainer* particleContainer, DomainDecompBase* domainDecomp,
            unsigned long simstep);
 
    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("FlopRateWriter");
    }
    static PluginBase* createInstance() { return new FlopRateWriter(); }
 
    void measureFLOPS(ParticleContainer* particleContainer, unsigned long simstep);
 
private:
    void setPrefix(double f_in, double& f_out, char& prefix) const;
 
    FlopCounter * _flopCounter;
    bool _writeToStdout, _writeToFile;
    std::ofstream _fileStream;
    unsigned long _writeFrequency;
    std::string _outputPrefix;
};
 
#endif /* SRC_IO_FLOPRATEWRITER_H_ */