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

#include <vtkParticleWriter.h>

Collaboration diagram for rw::writer::VtkParticleWriter:

Public Member Functions
	VtkParticleWriter (const std::string &filename, const std::string &compress_type="")
	Constructor.

void	addTimeStep (const double &timestep)
	Writes the time step to the file.

void	close ()
	Closes the file and store it to the hard disk.

Mesh data
void	appendNodes (const model::ModelData *model, const std::vector< std::string > &tags)
	Writes the nodes to the file.

void	appendMesh (const model::ModelData *model, const std::vector< std::string > &tags)
	Writes the nodes to the file.

void	appendContactData (const model::ModelData model, const std::vector< size_t > processed_nodes, const std::vector< std::pair< size_t, size_t > > *processed_elems)
	Prepares contact data that is set of nodes in contact and line-element connecting two contacting nodes.

void	appendStrainStress (const model::ModelData *model)
	Writes strain/stress.

Private Attributes
vtkSmartPointer< vtkXMLUnstructuredGridWriter >	d_writer_p
	XML unstructured grid writer.

vtkSmartPointer< vtkUnstructuredGrid >	d_grid_p
	Unstructured grid.

std::string	d_compressType
	compression_type Specify the compressor (if any)

Detailed Description

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

Definition at line 28 of file vtkParticleWriter.h.

Constructor & Destructor Documentation

◆ VtkParticleWriter()

rw::writer::VtkParticleWriter::VtkParticleWriter	(	const std::string &	filename,
		const std::string &	compress_type = `""`
	)

explicit

Constructor.

Creates and opens .vtu file of name given by filename. The file remains open till the close() function is invoked.

Parameters

filename	Name of file which will be created
compress_type	Compression method (optional)

Definition at line 29 of file vtkParticleWriter.cpp.

    : d_compressType(compress_type) {
 
  std::string f = filename + ".vtu";
 
  d_writer_p = vtkSmartPointer<vtkXMLUnstructuredGridWriter>::New();
  d_writer_p->SetFileName(const_cast<char *>(f.c_str()));
}

References d_writer_p.

Member Function Documentation

◆ addTimeStep()

void rw::writer::VtkParticleWriter::addTimeStep ( const double & timestep )

Writes the time step to the file.

Parameters

timestep Current time step of the simulation

Definition at line 341 of file vtkParticleWriter.cpp.

                                                                  {
 
  auto t = vtkDoubleArray::New();
  t->SetName("TIME");
  t->SetNumberOfTuples(1);
  t->SetTuple1(0, timestep);
  d_grid_p->GetFieldData()->AddArray(t);
}

◆ appendContactData()

void rw::writer::VtkParticleWriter::appendContactData	(	const model::ModelData *	model,
		const std::vector< size_t > *	processed_nodes,
		const std::vector< std::pair< size_t, size_t > > *	processed_elems
	)

Prepares contact data that is set of nodes in contact and line-element connecting two contacting nodes.

Parameters

model	ModelData class object
processed_nodes	Nodes in the list for which we are writing contact data
processed_elems	Number of line elements (two nodes)

Definition at line 361 of file vtkParticleWriter.cpp.

                                                   {
 
  if (processed_nodes->size() == 0)
    return;
 
  // write point data
  auto points = vtkSmartPointer<vtkPoints>::New();
 
 
  const size_t num_nodes = processed_nodes->size();
  const size_t num_elems = processed_elems->size();
 
  // get all the nodes first
  for (const auto &i : *processed_nodes) {
    const auto &x = model->d_x[i];
    points->InsertNextPoint(x.d_x, x.d_y, x.d_z);
  }
 
  // write point data
  d_grid_p = vtkSmartPointer<vtkUnstructuredGrid>::New();
  d_grid_p->SetPoints(points);
 
  // now wrtie element data
  const size_t vtk_element_type = 3; // line element
  const size_t num_vertex = 2;
  // element node connectivity
  auto cells = vtkSmartPointer<vtkCellArray>::New();
  cells->Allocate(num_vertex, num_elems);
 
  // element type
  int cell_types[num_elems];
 
  vtkIdType ids[num_vertex];
  for (size_t i = 0; i < num_elems; i++) {
 
    ids[0] = (*processed_elems)[i].first;
    ids[1] = (*processed_elems)[i].second;
 
    cells->InsertNextCell(num_vertex, ids);
    cell_types[i] = vtk_element_type;
  }
 
  // element node connectivity
  d_grid_p->SetCells(cell_types, cells);
 
  // write cell data (normal direction)
  {
    auto array = vtkSmartPointer < vtkDoubleArray > ::New();
    array->SetNumberOfComponents(3);
    array->SetName("Normal");
    array->SetComponentName(0, "x");
    array->SetComponentName(1, "y");
    array->SetComponentName(2, "z");
 
    double value[3];
    for (size_t i = 0; i < num_elems; i++) {
 
      ids[0] = (*processed_elems)[i].first;
      ids[1] = (*processed_elems)[i].second;
 
      auto glob_id1 = (*processed_nodes)[ids[0]];
      auto glob_id2 = (*processed_nodes)[ids[1]];
 
      const auto &x1 = model->d_x[glob_id1];
      const auto &x2 = model->d_x[glob_id2];
 
      auto xd = (x1 - x2)/((x2 - x1).length());
 
      value[0] = xd[0];
      value[1] = xd[1];
      value[2] = xd[2];
      array->InsertNextTuple(value);
    }
 
    d_grid_p->GetCellData()->AddArray(array);
  }
}

◆ appendMesh()

void rw::writer::VtkParticleWriter::appendMesh	(	const model::ModelData *	model,
		const std::vector< std::string > &	tags
	)

Writes the nodes to the file.

Parameters

model	ModelData class object
tags	Vector of tags (name of data, e.g., 'Displacement', 'Velocity') to append to the file

Definition at line 275 of file vtkParticleWriter.cpp.

                                      {
 
  if (model->d_x.size() == 0)
    return;
 
  // write point data
  appendNodes(model, tags);
 
  //
  // process elements data
  //
 
  // get total number of elements and maximum number of vertex in any element
  size_t num_elems = 0;
  size_t num_vertex = 0;
 
  // count number of elements in all particles
  for (const auto &p : model->d_particlesListTypeAll) {
    //const auto &rp = p->d_rp_p;
    num_elems += p->getMeshP()->getNumElements();
    auto n =
        util::vtk_map_element_to_num_nodes[p->getMeshP()->getElementType()];
    if (num_vertex < n)
      num_vertex = n;
  }
 
  // element node connectivity
  auto cells = vtkSmartPointer<vtkCellArray>::New();
  cells->Allocate(num_vertex, num_elems);
 
  // element type
  int cell_types[num_elems];
 
  // loop over particles
  size_t global_elem_counter = 0;
  for (const auto &p : model->d_particlesListTypeAll) {
    // get mesh of reference particle in this zone
    const auto &mesh = p->getMeshP();
 
    // get element type
    size_t element_type = mesh->getElementType();
 
    // loop over elements of this particle
    size_t num_vertex_p = util::vtk_map_element_to_num_nodes[element_type];
    vtkIdType ids[num_vertex_p];
    for (size_t e =0; e < mesh->getNumElements(); e++) {
      auto elem = mesh->getElementConnectivity(e);
 
      // assign global ids to the nodes
      for (size_t n=0; n<elem.size(); n++)
        ids[n] = elem[n] + p->d_globStart;
 
      cells->InsertNextCell(num_vertex_p, ids);
      cell_types[global_elem_counter] = element_type;
 
      // increment global element counter
      global_elem_counter++;
    }
  }
 
  // element node connectivity
  d_grid_p->SetCells(cell_types, cells);
}

References util::vtk_map_element_to_num_nodes.

◆ appendNodes()

void rw::writer::VtkParticleWriter::appendNodes	(	const model::ModelData *	model,
		const std::vector< std::string > &	tags
	)

Writes the nodes to the file.

Parameters

model	ModelData class object
tags	Vector of tags (name of data, e.g., 'Displacement', 'Velocity') to append to the file

Definition at line 39 of file vtkParticleWriter.cpp.

                                      {
 
  if (model->d_x.size() == 0)
    return;
 
  // write point data
  auto points = vtkSmartPointer<vtkPoints>::New();
 
  // get all the nodes first
  for (const auto &x : model->d_x)
    points->InsertNextPoint(x.d_x, x.d_y, x.d_z);
 
  // write point data
  d_grid_p = vtkSmartPointer<vtkUnstructuredGrid>::New();
  d_grid_p->SetPoints(points);
 
  // now write data associated to nodes in both particle and wall
  double value[3];
  value[0] = 0;
  value[1] = 0;
  value[2] = 0;
  double p_tag[1];
  p_tag[0] = 0;
 
  // handle displacement
  if (util::methods::isTagInList("Displacement", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(3);
    array->SetName("Displacement");
 
    array->SetComponentName(0, "x");
    array->SetComponentName(1, "y");
    array->SetComponentName(2, "z");
 
    for (const auto &ui : model->d_u) {
      value[0] = ui.d_x;
      value[1] = ui.d_y;
      value[2] = ui.d_z;
      array->InsertNextTuple(value);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // displacement
 
  // handle velocity
  if (util::methods::isTagInList("Velocity", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(3);
    array->SetName("Velocity");
 
    array->SetComponentName(0, "x");
    array->SetComponentName(1, "y");
    array->SetComponentName(2, "z");
 
    for (const auto &ui : model->d_v) {
      value[0] = ui.d_x;
      value[1] = ui.d_y;
      value[2] = ui.d_z;
      array->InsertNextTuple(value);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // velocity
 
  // handle force
  if (util::methods::isTagInList("Force_Density", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(3);
    array->SetName("Force_Density");
 
    array->SetComponentName(0, "x");
    array->SetComponentName(1, "y");
    array->SetComponentName(2, "z");
 
    for (const auto &ui : model->d_f) {
      value[0] = ui.d_x;
      value[1] = ui.d_y;
      value[2] = ui.d_z;
      array->InsertNextTuple(value);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // force
 
  // handle force
  if (util::methods::isTagInList("Force", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(3);
    array->SetName("Force");
 
    array->SetComponentName(0, "x");
    array->SetComponentName(1, "y");
    array->SetComponentName(2, "z");
 
    size_t i_count = 0;
    for (const auto &ui : model->d_f) {
      const auto &voli = model->d_vol[i_count];
      value[0] = ui.d_x * voli;
      value[1] = ui.d_y * voli;
      value[2] = ui.d_z * voli;
      array->InsertNextTuple(value);
 
      i_count++;
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // force
 
  // handle fixity
  if (util::methods::isTagInList("Fixity", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(1);
    array->SetName("Fixity");
 
    for (const auto &n : model->d_fix) {
      p_tag[0] = double(n);
      array->InsertNextTuple(p_tag);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // fixity
 
  // handle Particle ID
  if (util::methods::isTagInList("Particle_ID", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(1);
    array->SetName("Particle_ID");
 
    for (size_t i = 0; i<model->d_x.size(); i++) {
      auto pi = model->getPtId(i);
      p_tag[0] = double(model->getParticleFromAllList(pi)->getId());
      array->InsertNextTuple(p_tag);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // Particle ID
 
  // handle Zone ID
  if (util::methods::isTagInList("Zone_ID", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(1);
    array->SetName("Zone_ID");
 
    for (size_t i = 0; i<model->d_x.size(); i++) {
      auto pi = model->getPtId(i);
      p_tag[0] = double(model->getParticleFromAllList(pi)->d_zoneId);
      array->InsertNextTuple(p_tag);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // Zone ID
 
  // handle force fixity
  if (util::methods::isTagInList("Force_Fixity", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(1);
    array->SetName("Force_Fixity");
 
    for (const auto &n : model->d_forceFixity) {
      p_tag[0] = double(n);
      array->InsertNextTuple(p_tag);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // force fixity
 
  // handle nodal volume
  if (util::methods::isTagInList("Nodal_Volume", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(1);
    array->SetName("Nodal_Volume");
 
    for (const auto &n : model->d_vol) {
      p_tag[0] = double(n);
      array->InsertNextTuple(p_tag);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // nodal volume
 
  // handle damage_Z
  if (util::methods::isTagInList("Damage_Z", tags)) {
 
    auto array = vtkSmartPointer<vtkDoubleArray>::New();
    array->SetNumberOfComponents(1);
    array->SetName("Damage_Z");
 
    for (const auto &n : model->d_Z) {
      p_tag[0] = double(n);
      array->InsertNextTuple(p_tag);
    }
 
    // write
    d_grid_p->GetPointData()->AddArray(array);
  } // damage_Z
 
  // handle theta
  if (util::methods::isTagInList("Theta", tags)) {
 
    if (model->getParticleFromAllList(0)->d_material_p->isStateActive()) {
 
      auto array = vtkSmartPointer<vtkDoubleArray>::New();
      array->SetNumberOfComponents(1);
      array->SetName("Theta");
 
      for (const auto &n : model->d_thetaX) {
        p_tag[0] = double(n);
        array->InsertNextTuple(p_tag);
      }
 
      // write
      d_grid_p->GetPointData()->AddArray(array);
    }
  } // Theta
}

References util::methods::isTagInList().

Here is the call graph for this function:

◆ appendStrainStress()

void rw::writer::VtkParticleWriter::appendStrainStress ( const model::ModelData * model )

Writes strain/stress.

Parameters

model ModelData class object

Definition at line 444 of file vtkParticleWriter.cpp.

                                     {
 
  if (model->d_xQuadCur.size() == 0) {
    std::cout << "VtkParticleWriter::appendStrainStress: Nothing to write.\n";
    return;
  }
 
  // write point data
  auto points = vtkSmartPointer<vtkPoints>::New();
 
  // get all the quadrature points first
  for (const auto &x : model->d_xQuadCur)
    points->InsertNextPoint(x.d_x, x.d_y, x.d_z);
 
  // write point data
  d_grid_p = vtkSmartPointer<vtkUnstructuredGrid>::New();
  d_grid_p->SetPoints(points);
 
  // now write data associated to nodes (in this case, quad points are nodes)
  double value[3] = {0., 0., 0.};
  double value_s[6] = {0., 0., 0., 0., 0., 0.};
  double p_tag[1] = {0.};
 
  auto array_strain = vtkSmartPointer<vtkDoubleArray>::New();
  array_strain->SetNumberOfComponents(6);
  array_strain->SetName("Strain");
 
  auto array_stress = vtkSmartPointer<vtkDoubleArray>::New();
  array_stress->SetNumberOfComponents(6);
  array_stress->SetName("Stress");
 
  std::vector<std::string> coord_strings = {"xx", "yy", "zz", "yz", "xz", "xy"};
  for (size_t i =0; i<6; i++) {
    array_strain->SetComponentName(i, coord_strings[i].c_str());
    array_stress->SetComponentName(i, coord_strings[i].c_str());
  }
 
  for (size_t i=0; i<model->d_strain.size(); i++) {
 
    model->d_strain[i].copy(value_s);
    array_strain->InsertNextTuple(value_s);
 
    model->d_stress[i].copy(value_s);
    array_stress->InsertNextTuple(value_s);
  }
 
 
  // write
  d_grid_p->GetPointData()->AddArray(array_strain);
  d_grid_p->GetPointData()->AddArray(array_stress);
}

◆ close()

void rw::writer::VtkParticleWriter::close ( )

Closes the file and store it to the hard disk.

Definition at line 350 of file vtkParticleWriter.cpp.

                                      {
  d_writer_p->SetInputData(d_grid_p);
  d_writer_p->SetDataModeToAppended();
  d_writer_p->EncodeAppendedDataOn();
  if (d_compressType == "zlib")
    d_writer_p->SetCompressorTypeToZLib();
  else
    d_writer_p->SetCompressor(0);
  d_writer_p->Write();
}

Field Documentation

◆ d_compressType

std::string rw::writer::VtkParticleWriter::d_compressType

private

compression_type Specify the compressor (if any)

Definition at line 103 of file vtkParticleWriter.h.

◆ d_grid_p

vtkSmartPointer<vtkUnstructuredGrid> rw::writer::VtkParticleWriter::d_grid_p

private

Unstructured grid.

Definition at line 100 of file vtkParticleWriter.h.

◆ d_writer_p

vtkSmartPointer<vtkXMLUnstructuredGridWriter> rw::writer::VtkParticleWriter::d_writer_p

private

XML unstructured grid writer.

Definition at line 97 of file vtkParticleWriter.h.

Referenced by VtkParticleWriter().

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

/home/user/project/src/rw/vtkParticleWriter.h
/home/user/project/src/rw/vtkParticleWriter.cpp

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ VtkParticleWriter()

Member Function Documentation

◆ addTimeStep()

◆ appendContactData()

◆ appendMesh()

◆ appendNodes()

◆ appendStrainStress()

◆ close()

Field Documentation

◆ d_compressType

◆ d_grid_p

◆ d_writer_p