VTUGridReader.hpp

/*---------------------------------------------------------------------------*\
 *
 *  mimmo
 *
 *  Copyright (C) 2015-2021 OPTIMAD engineering Srl
 *
 *  -------------------------------------------------------------------------
 *  License
 *  This file is part of mimmo.
 *
 *  mimmo is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU Lesser General Public License v3 (LGPL)
 *  as published by the Free Software Foundation.
 *
 *  mimmo is distributed in the hope that it will be useful, but WITHOUT
 *  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 *  FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
 *  License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with mimmo. If not, see <http://www.gnu.org/licenses/>.
 *
\*---------------------------------------------------------------------------*/
#ifndef __VTUGRIDREADER_HPP__
#define __VTUGRIDREADER_HPP__
 
#include "mimmoTypeDef.hpp"
#include <bitpit_patchkernel.hpp>
 
namespace mimmo{
 
class VTUAbsorbStreamer : public bitpit::VTKBaseStreamer{
public:
    VTUAbsorbStreamer();
    virtual ~VTUAbsorbStreamer();
    VTUAbsorbStreamer(const VTUAbsorbStreamer&) = default;
 
    virtual void absorbData(std::fstream &stream, const std::string & name, bitpit::VTKFormat format, uint64_t entries, uint8_t components, bitpit::VTKDataType datatype);
    virtual void decodeRawData(bitpit::PatchKernel &) = 0;
};
 
class VTUGridStreamer: public VTUAbsorbStreamer{
 
protected:
    dvecarr3E points;                            
    livector1D connectivitylist;                 
    livector1D pointsID;                         
    livector1D cellsID;                          
    livector1D pids;                              
    livector1D offsets;                           
    std::vector<bitpit::ElementType> types;      
    livector1D faces;                             
    livector1D faceoffsets;                       
#if MIMMO_ENABLE_MPI
    livector1D cellsRank;                         
    livector1D verticesRank;                       
    livector1D cellsGlobalIndex;                   
#endif
 
public:
    VTUGridStreamer();
    virtual ~VTUGridStreamer();
    VTUGridStreamer(const VTUGridStreamer&) = default;
 
    virtual void absorbData(std::fstream &stream, const std::string &name, bitpit::VTKFormat format, uint64_t entries, uint8_t components, bitpit::VTKDataType datatype);
    void decodeRawData(bitpit::PatchKernel & patch);
    void readIntegerPod(std::fstream & stream, bitpit::VTKFormat& format, bitpit::VTKDataType& datatype, long &target);
    void readDoublePod(std::fstream & stream, bitpit::VTKFormat& format, bitpit::VTKDataType& datatype, double&target);
};
 
class VTUGridReader: protected bitpit::VTKUnstructuredGrid
{
 
public:
    VTUGridReader( std::string dir, std::string name, VTUAbsorbStreamer & streamer,
                   bitpit::PatchKernel & patch, bool masterRankOnly = true, bitpit::VTKElementType eltype= bitpit::VTKElementType::UNDEFINED);
    ~VTUGridReader();
 
    void read();
 
private:
    bitpit::PatchKernel& m_patch;   
    VTUAbsorbStreamer & m_streamer; 
#if MIMMO_ENABLE_MPI
    bool m_masterRankOnly; 
#endif
};
 
 
};
 
#endif /* __VTUGRIDREADER_HPP__ */