StatWriter.h

Go to the documentation of this file.

#ifndef OPAL_STAT_WRITER_H
#define OPAL_STAT_WRITER_H

#include "StatBaseWriter.h"

#include "Algorithms/bet/EnvelopeBunch.h"

class StatWriter : public StatBaseWriter {

public:
    typedef std::vector<std::pair<std::string, unsigned int> > losses_t;

    StatWriter(const std::string& fname, bool restart);

    using SDDSWriter::write;
    void write(PartBunchBase<double, 3> *beam, Vector_t FDext[],
               const losses_t &losses = losses_t(), const double& azimuth = -1);

    void write(EnvelopeBunch &beam, Vector_t FDext[],
               double sposHead, double sposRef, double sposTail);

private:
    void fillHeader(const losses_t &losses = losses_t());
};

#endif