VtkMeshReader.hpp

Go to the documentation of this file.

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   This file is part of CEPS.
 
   CEPS is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   CEPS 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 General Public License for more details.
 
   You should have received a copy of the GNU General Public License
   along with CEPS (see file LICENSE at root of project).
   If not, see <https://www.gnu.org/licenses/>.
 
 
   Copyright 2019-2024 Inria, Universite de Bordeaux
 
   Authors, in alphabetical order:
 
     Pierre-Elliott BECUE, Florian CARO, Yves COUDIERE(*), Andjela DAVIDOVIC, 
     Charlie DOUANLA-LONTSI, Marc FUENTES, Mehdi JUHOOR, Michael LEGUEBE(*), 
     Pauline MIGERDITICHAN, Valentin PANNETIER(*), Nejib ZEMZEMI.
     * : currently active authors
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#pragma once
 
#include "geometry/readers/MeshReader.hpp"
 
#include <vtkCellData.h>
#include <vtkDataSet.h>
#include <vtkDataSetReader.h>
#include <vtkGenericCell.h>
#include <vtkIdList.h>
#include <vtkPointData.h>
#include <vtkPointSet.h>
#include <vtkPoints.h>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>
#include <vtkXMLPUnstructuredGridReader.h>
#include <vtkCommand.h>
#include <vtkNew.h>
 
class VtkMeshReader : public MeshReader
{
 
  public:
 
    VtkMeshReader(const CepsString &fileName, const CepsUInt &dim, CepsBool quads = false);
  
    ~VtkMeshReader() override;
  
    void
    initialize() override;
  
    CepsBool
    isVtkFile(const CepsString &fileName) const;
 
    void
    readNodes(
      CepsUInt                   indexStart,
      CepsUInt                   indexEnd,
      CepsUInt                   nodeOffset,
      CepsVector<CepsReal>&      coords,
      CepsVector<CepsAttribute>& attr,
      CepsVector<CepsUInt>&      attrPtr
    ) override;
  
    void   
    readCells(
      CepsUInt                         indexStart,
      CepsUInt                         indexEnd,
      CepsUInt                         nodeOffset,
      CepsUInt                         cellOffset,
      CepsVector<CepsNodeGlobalIndex>& nodeIndices,
      CepsVector<CepsUInt>&            nodeIndicesPtr,
      CepsVector<CepsAttribute>&       cellAttr,
      CepsVector<CepsUInt>&            cellAttrPtr,
      CepsVector<CepsChar>&            isBoundaryCell
    ) override;
 
 
  private:
  
    void
    getNextCell(
      CepsUInt                         cellId,
      CepsUInt                         fileCellId,
      CepsUInt                         nodeOffset,
      CepsVector<CepsNodeGlobalIndex>& nodeIndices,
      CepsVector<CepsUInt>&            nodeIndicesPtr,
      CepsVector<CepsAttribute>&       cellAttr,
      CepsVector<CepsUInt>&            cellAttrPtr,
      CepsVector<CepsChar>&            isBoundaryCell,
      CepsBool                         boundaryFlag
    ) override;
 
    void
    initializePvtu(const CepsString &fileName);
  
    void
    initializeLegacy(const CepsString &fileName);
  
    void
    determineCellDataToRead();
  
    void
    determinePointDataToRead();
  
  
  protected:
  
    vtkSmartPointer<vtkDataSet> m_mesh;             
  
    CepsVector<vtkDataArray *> m_cellArrayToRead;   
    CepsVector<vtkDataArray *> m_pointArrayToRead;  
};