AmrYtWriter.h

Go to the documentation of this file.

#ifndef AMR_YT_WRITER_H
#define AMR_YT_WRITER_H

#include "Amr/AbstractAmrWriter.h"

#include <boost/filesystem.hpp>

#include <vector>

class AmrYtWriter : public AbstractAmrWriter {
    
public:
    
    AmrYtWriter(int step, int bin = 0);
    
    void writeFields(const amr::AmrScalarFieldContainer_t& rho,
                     const amr::AmrScalarFieldContainer_t& phi,
                     const amr::AmrVectorFieldContainer_t& efield,
                     const amr::AmrIntArray_t& refRatio,
                     const amr::AmrGeomContainer_t& geom,
                     const int& nLevel,
                     const double& time,
                     const double& scale);
    
    
    void writeBunch(const AmrPartBunch* bunch_p,
                    const double& time,
                    const double& gamma);
    
private:
    /* Copied and slightely modified version of
     * AMReX_ParticleContainerI.H
     * 
     * @param level to write
     * @param ofs out stream
     * @param fnum file number
     * @param which file
     * @param count how many particles on this grid
     * @param where file offset
     * @param bunch_p to get data from
     * @param gamma is the Lorentz factor
     */
    void writeParticles_m(int level,
                          std::ofstream& ofs,
                          int fnum,
                          amrex::Vector<int>& which,
                          amrex::Vector<int>& count,
                          amrex::Vector<long>& where,
                          const AmrPartBunch* bunch_p,
                          const double gamma) const;
    
private:
    std::string dir_m;                      
    
    std::vector<std::string> intData_m;     
    std::vector<std::string> realData_m;    
};

#endif