rw::reader::VtkParticleReader Class Reference

A vtk writer for simple point data and complex fem mesh data. More...

#include <vtkParticleReader.h>

Collaboration diagram for rw::reader::VtkParticleReader:

Public Member Functions
	VtkParticleReader (const std::string &filename)
	Constructor.

Mesh data
vtkSmartPointer< vtkXMLUnstructuredGridReader >	d_reader_p
	XML unstructured grid writer.
vtkSmartPointer< vtkUnstructuredGrid >	d_grid_p
	Unstructured grid.
void	readNodes (model::ModelData *model)
	Writes the nodes to the file.
void	close ()
	Closes the file and store it to the hard disk.

Detailed Description

A vtk writer for simple point data and complex fem mesh data.

Definition at line 31 of file vtkParticleReader.h.

Constructor & Destructor Documentation

VtkParticleReader()

rw::reader::VtkParticleReader::VtkParticleReader ( const std::string &  filename )

explicit

Constructor.

Parameters

filename

Name of the vtk file (with/without .vtu extension)

Definition at line 32 of file vtkParticleReader.cpp.

                                                                      {
 
  auto f = util::io::checkAndCreateNewFilename(filename, "vtu");
 
  d_reader_p = vtkSmartPointer<vtkXMLUnstructuredGridReader>::New();
  d_reader_p->SetFileName(const_cast<char *>(f.c_str()));
  d_reader_p->Update();
}

References util::io::checkAndCreateNewFilename(), and d_reader_p.

Here is the call graph for this function:

Member Function Documentation

close()

void rw::reader::VtkParticleReader::close

(

)

Closes the file and store it to the hard disk.

Definition at line 113 of file vtkParticleReader.cpp.

                                      {
  // delete d_reader_p;
  // delete d_grid_p;
}

readNodes()

void rw::reader::VtkParticleReader::readNodes

(

model::ModelData *

model

)

Writes the nodes to the file.

Parameters

model

ModelData class object

Definition at line 41 of file vtkParticleReader.cpp.

                           {
 
  if (model->d_x.size() == 0)
    return;
 
  //
  // For nodes, we read following data
  // 1. nodes current configuration
  // 2. nodes displacement (if not then compute from current and reference
  // config)
  // 3. nodes velocity (if not then produce error)
  //
  d_grid_p = d_reader_p->GetOutput();
  vtkIdType num_nodes = d_grid_p->GetNumberOfPoints();
 
  if (num_nodes != model->d_x.size()) {
    std::cerr << "Error: Number of points data = " << num_nodes <<
                 " in file does not match the number of points = "
              << model->d_x.size() << ". Aborting reading.\n";
    exit(1);
  }
 
  // read current configuration
  for (size_t i = 0; i < num_nodes; i++) {
    vtkIdType id = i;
 
    double x[3];
    d_grid_p->GetPoint(id, x);
 
    util::Point i_node = util::Point(x[0], x[1], x[2]);
    model->setX(i, i_node);
  }
 
  // read point field data
  vtkPointData *p_field = d_grid_p->GetPointData();
 
  std::vector<std::string> tags = {"Displacement", "Velocity"};
  for (auto tag: tags) {
 
    if (p_field->HasArray(tag.c_str())) {
      vtkDataArray *array = p_field->GetArray(tag.c_str());
 
      auto data_a = vtkSmartPointer<vtkDoubleArray>::New();
      data_a->SetNumberOfComponents(3);
      data_a->Allocate(3, 1); // allocate memory
 
      for (size_t i = 0; i < array->GetNumberOfTuples(); i++) {
        array->GetTuples(i, i, data_a);
        auto xi = util::Point(data_a->GetValue(0), data_a->GetValue(1),
                               data_a->GetValue(2));
        if (tag == "Displacement")
          model->setU(i, xi);
        else if (tag == "Velocity")
          model->setV(i, xi);
      }
    } else {
 
      if (tag == "Velocity") {
        std::cerr << "Error: Restart file does not have <Velocity> data. "
                     "Aborting reading.\n";
        exit(1);
      } else if (tag == "Displacement") {
        // compute from current and reference configuration
        for (size_t i=0; i<model->d_x.size(); i++)
          model->setU(i, model->getX(i) - model->getXRef(i));
      }
    }
  }
}

Field Documentation

d_grid_p

vtkSmartPointer<vtkUnstructuredGrid> rw::reader::VtkParticleReader::d_grid_p

private

Unstructured grid.

Definition at line 59 of file vtkParticleReader.h.

d_reader_p

vtkSmartPointer<vtkXMLUnstructuredGridReader> rw::reader::VtkParticleReader::d_reader_p

private

XML unstructured grid writer.

Definition at line 56 of file vtkParticleReader.h.

Referenced by VtkParticleReader().

---

The documentation for this class was generated from the following files:

• /home/user/project/src/rw/vtkParticleReader.h
• /home/user/project/src/rw/vtkParticleReader.cpp