vtkReader.h

//  Copyright (c) 2019 Prashant K. Jha
//  Copyright (c) 2019 Patrick Diehl
//
//  Distributed under the Boost Software License, Version 1.0. (See accompanying
//  file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
 
#ifndef RW_VTKREADER_H
#define RW_VTKREADER_H
 
#include <util/matrixBlaze.h>
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>
#include <vtkXMLUnstructuredGridReader.h>
 
#include "util/matrix.h" // definition of matrices
#include "util/matrixBlaze.h"
#include "util/point.h" // definition of Point3
 
namespace rw {
 
namespace reader {
 
class VtkReader {
 
public:
  explicit VtkReader(const std::string &filename);
 
  bool vtuHasPointData(const std::string &data_tag);
 
  bool vtuHasCellData(const std::string &data_tag);
 
  std::vector<std::string> readVtuFilePointTags();
 
  std::vector<std::string> readVtuFileCellTags();
 
  void readMesh(size_t dim, std::vector<util::Point3> *nodes,
                size_t &element_type, size_t &num_elem,
                std::vector<size_t> *enc, std::vector<std::vector<size_t>> *nec,
                std::vector<double> *volumes, bool is_fd = false);
 
  void readNodes(std::vector<util::Point3> *nodes);
 
  void readCells(size_t dim, size_t &element_type, size_t &num_elem,
                 std::vector<size_t> *enc,
                 std::vector<std::vector<size_t>> *nec);
 
  bool readPointData(const std::string &name, std::vector<uint8_t> *data);
 
  bool readPointData(const std::string &name, std::vector<size_t> *data);
 
  bool readPointData(const std::string &name, std::vector<int> *data);
 
  bool readPointData(const std::string &name, std::vector<float> *data);
 
  bool readPointData(const std::string &name, std::vector<double> *data);
 
  bool readPointData(const std::string &name, std::vector<util::Point3> *data);
 
  bool readPointData(const std::string &name,
                     std::vector<util::SymMatrix3> *data);
 
  bool readPointData(const std::string &name,
                     std::vector<util::Matrix33> *data);
 
  bool readCellData(const std::string &name, std::vector<float> *data);
 
  bool readCellData(const std::string &name, std::vector<double> *data);
 
  bool readCellData(const std::string &name, std::vector<util::Point3> *data);
 
  bool readCellData(const std::string &name,
                    std::vector<util::SymMatrix3> *data);
 
  bool readCellData(const std::string &name, std::vector<util::Matrix33> *data);
 
  void close();
 
private:
  static size_t d_count;
 
  vtkSmartPointer<vtkXMLUnstructuredGridReader> d_reader_p;
 
  vtkSmartPointer<vtkUnstructuredGrid> d_grid_p;
};
 
} // namespace reader
 
} // namespace rw
 
#endif // RW_VTKREADER_H