Kernel/VTKWriter/VTKData.h

Go to the documentation of this file.

//Copyright (c) 2013-2023, The MercuryDPM Developers Team. All rights reserved.
//For the list of developers, see <http://www.MercuryDPM.org/Team>.
//
//Redistribution and use in source and binary forms, with or without
//modification, are permitted provided that the following conditions are met:
//  * Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//  * Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//  * Neither the name MercuryDPM nor the
//    names of its contributors may be used to endorse or promote products
//    derived from this software without specific prior written permission.
//
//THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
//ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
//WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
//DISCLAIMED. IN NO EVENT SHALL THE MERCURYDPM DEVELOPERS TEAM BE LIABLE FOR ANY
//DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
//(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
//LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
//ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
//(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
//SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
#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