WallVTKWriter.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 WALL_VTKWRITER_H
#define WALL_VTKWRITER_H
 
#include "VTKWriter/BaseVTKWriter.h"
#include "WallHandler.h"
 
class WallVTKWriter final : public BaseVTKWriter<WallHandler>
{
 
public:
    
    explicit WallVTKWriter(WallHandler& wallHandler) : BaseVTKWriter(wallHandler)
    {}
    
    WallVTKWriter(const WallVTKWriter&) = default;
    
    void getVTKData(VTKContainer& vtk) const;
    
    void writeVTK() const override;
    
    std::string getName() const
    { return "WallVTKWriter"; }
 
    void setWriteWallSurfaceAreaVTK(bool writeWallSurfaceAreaVTK);
    bool getWriteWallSurfaceAreaVTK() const;
 
    // set function for cg fields (set cgWidth to zero to turn off writing cg fields)
    void setCGWidth(double cgWidth) {
        cgWidth_ = std::max(0.0,cgWidth);
        if (cgWidth==0.0) {
            logger(INFO, "CG Width set to zero; no CG data is written to the VTU files");
        }
    }
 
protected:
    
    void write(std::fstream& file, std::string name, std::function<double(BaseWall*)> f) const;
    
    void writeVTKPoints(std::fstream& file, VTKContainer& vtk) const;
    
    void writeVTKCells(std::fstream& file, VTKContainer& vtk) const;
    
    void writeVTKCellData(std::fstream& file, VTKContainer& vtk) const;
 
    void writeVTKPointData(std::fstream& file, VTKContainer& vtk) const;
 
    void writeCGFields(std::fstream& file, VTKContainer& vtk) const;
 
    void writeVTKSurfaceArea(std::fstream& file, VTKContainer& vtk) const;
 
    // determines whether wall surface area is written to the vtu file
    bool writeWallSurfaceAreaVTK_ { false };
 
    // stores the cg width used in writeCGFields (0 means no cg data is written)
    double cgWidth_ = 0.0;
};
 
 
#endif