ParticleVtkWriter.h

Go to the documentation of this file.

// This file is part of the MercuryDPM project (https://www.mercurydpm.org).
// Copyright (c), The MercuryDPM Developers Team. All rights reserved.
// License: BSD 3-Clause License; see the LICENSE file in the root directory.
 
#ifndef MERCURYDPM_VTKWRITER_H
#define MERCURYDPM_VTKWRITER_H
 
#include "VTKWriter/BaseVTKWriter.h"
#include "ParticleHandler.h"
 
class ParticleVtkWriter : public BaseVTKWriter<ParticleHandler>
{
 
public:
    
    explicit ParticleVtkWriter(ParticleHandler& particleHandler) : BaseVTKWriter(particleHandler)
    {}
    
    virtual ~ParticleVtkWriter() = default;
    
    virtual std::string getName() const = 0;
 
protected:
 
    void writeExtraFields(std::fstream& file) const;
 
    void writeVTKIndSpecies(std::fstream& file) const;
    
    void writeVTKPositions(std::fstream& file) const;
    
    bool particleMustBeWritten(BaseParticle* particle) const
    {
        return !(particle->isMPIParticle() || particle->isPeriodicGhostParticle());
    }
};
 
 
#endif //MERCURYDPM_VTKWRITER_H