H5Writer.h

Go to the documentation of this file.

#ifndef OPAL_H5_WRITER_H
#define OPAL_H5_WRITER_H

#include "Structure/H5PartWrapper.h"
#include "Algorithms/PartBunchBase.h"
#include "H5hut.h"

class H5Writer {
    
public:
    
    H5Writer(H5PartWrapper* h5wrapper, bool restart);

    void close();
    
    void changeH5Wrapper(H5PartWrapper *h5wrapper);
    
    
    void storeCavityInformation();
    
    double getLastPosition();
    
    void writePhaseSpace(PartBunchBase<double, 3> *beam, Vector_t FDext[]);

    int writePhaseSpace(PartBunchBase<double, 3> *beam, Vector_t FDext[], double E,
                        double refPr, double refPt, double refPz,
                        double refR, double refTheta, double refZ,
                        double azimuth, double elevation, bool local);

    void writePhaseSpace(EnvelopeBunch &beam, Vector_t FDext[], double sposHead, double sposRef, double sposTail);
//     void stashPhaseSpaceEnvelope(EnvelopeBunch &beam, Vector_t FDext[], double sposHead, double sposRef, double sposTail);
//     void dumpStashedPhaseSpaceEnvelope();
    
private:
    IpplTimings::TimerRef H5PartTimer_m;
    
    H5PartWrapper *h5wrapper_m;
    
    int H5call_m;
};


inline
void H5Writer::close() {
    h5wrapper_m->close();
}


inline
void H5Writer::changeH5Wrapper(H5PartWrapper *h5wrapper) {
    h5wrapper_m = h5wrapper;
}


inline
void H5Writer::storeCavityInformation() {
    h5wrapper_m->storeCavityInformation();
}


inline
double H5Writer::getLastPosition() {
    return h5wrapper_m->getLastPosition();
}

#endif