Kernel/VTKWriter/VTKData.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_VTKDATA_H
#define MERCURYDPM_VTKDATA_H
 
#include <string>
#include <vector>
#include <fstream>
#include "Math/Vector.h"
#include <unordered_map>
 
class VTKData
{
public:
    VTKData() = default;
    ~VTKData() = default;
    
    void addToPoints(Vec3D point);
    
    void addToPointData(const std::string& key, Mdouble value);
    
    void addToConnectivity(const std::vector<size_t>& indices);
    
    void addToTypes(int type);
    
    std::vector<Vec3D> getPoints() const;
    
    std::unordered_map<std::string, std::vector<Mdouble>> getPointData() const;
    
    std::vector<std::vector<size_t>> getConnectivity() const;
    
    std::vector<int> getTypes() const;
    
    void reservePoints(unsigned int n, const std::vector<std::string>& keys = {});
    
    void reserveCells(unsigned int n);
    
    void writeVTKData(std::string fileName) const;
 
    void writeVTKDataFromVtkContainer(std::string fileName, const std::vector<Vec3D>& points, const std::vector<std::vector<double>>& triangleStrips);
    
private:
    std::fstream makeVTKFileWithHeader(std::string& fileName) const;
    void writePoints(std::fstream& file) const;
    void writePointData(std::fstream& file) const;
    void writeConnectivity(std::fstream& file) const;
    void writeOffsets(std::fstream& file) const;
    void writeTypes(std::fstream& file) const;
    
    std::vector<Vec3D> points_;
    
    std::unordered_map<std::string, std::vector<Mdouble>> pointData_;
    
    std::vector<std::vector<size_t>> connectivity_;
    
    std::vector<int> types_;
};
 
 
#endif //MERCURYDPM_VTKDATA_H