VTKGridWriter.h

/*
 * VTKGridWriter.h
 *
 * @Date: 24.08.2010
 * @Author: eckhardw
 */
 
#pragma once
 
#ifndef MARDYN_AUTOPAS
 
#include "plugins/PluginBase.h"
#include "io/vtk/VTKGridCell.h"
#include "io/vtk/VTKGridVertex.h"
 
class LinkedCells;
class VTKGridWriterImplementation;
 
class VTKGridWriter : public PluginBase {
 
private:
 
    unsigned int _writeFrequency;
 
    std::string _fileName;
 
    VTKGridCell* _cells;
 
    VTKGridVertex* _vertices;
 
    int _numCells;
 
    int _numVertices;
 
public:
 
    VTKGridWriter();
 
    VTKGridWriter(unsigned int frequency, std::string name);
 
    virtual ~~VTKGridWriter();
 
    virtual void init(ParticleContainer *particleContainer,
                      DomainDecompBase *domainDecomp, Domain *domain);
 
    virtual void endStep(
            ParticleContainer *particleContainer, DomainDecompBase *domainDecomp,
            Domain *domain, unsigned long simstep
    );
 
    virtual void finish(ParticleContainer *particleContainer,
                        DomainDecompBase *domainDecomp, Domain *domain);
 
    std::string getPluginName() {
        return std::string("VTKGridWriter");
    }
    static PluginBase* createInstance() { return new VTKGridWriter(); }
 
    void readXML(XMLfileUnits& xmlconfig);
 
private:
 
    void setupVTKGrid(ParticleContainer* particleContainer);
 
    void releaseVTKGrid();
 
    void getCellData(LinkedCells* container, VTKGridCell& cell);
 
    void outputParallelVTKFile(unsigned int numProcs, unsigned long simstep,
            VTKGridWriterImplementation& impl);
 
};
 
#endif /* MARDYN_AUTOPAS */