vtkGeometryCacheWriter.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidGeometry/Rendering/vtkGeometryCacheWriter.h"
 
#include "MantidGeometry/Objects/CSGObject.h"
#include "MantidGeometry/Rendering/GeometryHandler.h"
#include "MantidKernel/Logger.h"
 
#include <Poco/DOM/AutoPtr.h>
#include <Poco/DOM/DOMWriter.h>
#include <Poco/DOM/Document.h>
#include <Poco/DOM/Element.h>
#include <Poco/DOM/Text.h>
#include <Poco/File.h>
#include <Poco/FileStream.h>
#include <Poco/Path.h>
#include <Poco/XML/XMLWriter.h>
 
#include <fstream>
#include <sstream>
#include <utility>
 
using Poco::XML::AutoPtr;
using Poco::XML::Document;
using Poco::XML::DOMWriter;
using Poco::XML::Element;
using Poco::XML::Text;
using Poco::XML::XMLWriter;
 
namespace Mantid::Geometry {
namespace {
Kernel::Logger g_log("vtkGeometryCacheWriter");
} // namespace
 
vtkGeometryCacheWriter::vtkGeometryCacheWriter(std::string filename)
    : m_doc(new Document()), m_filename(std::move(filename)) {
  Init();
}
 
vtkGeometryCacheWriter::~vtkGeometryCacheWriter() {
  m_root->release();
  m_doc->release();
}
 
void vtkGeometryCacheWriter::Init() {
  m_root = m_doc->createElement("PolyData");
  createVTKFileHeader();
}
void vtkGeometryCacheWriter::createVTKFileHeader() {
  AutoPtr<Element> pRoot = m_doc->createElement("VTKFile");
  pRoot->setAttribute("type", "PolyData");
  pRoot->setAttribute("version", "1.0");
  pRoot->setAttribute("byte_order", "LittleEndian");
  m_doc->appendChild(pRoot);
  pRoot->appendChild(m_root);
}
 
void vtkGeometryCacheWriter::addObject(CSGObject *obj) {
  // First check whether Object can be written to the file
  std::shared_ptr<GeometryHandler> handle = obj->getGeometryHandler();
  if (!(handle->canTriangulate()))
    return; // Cannot add the object to the file
  std::stringstream buf;
  // get the name of the Object
  int name = obj->getName();
  // get number of point
  auto noPts = handle->numberOfPoints();
  // get number of triangles
  auto noTris = handle->numberOfTriangles();
  // Add Piece
  AutoPtr<Element> pPiece = m_doc->createElement("Piece");
  // Add attribute name
  buf << name;
  pPiece->setAttribute("name", buf.str());
  // Add Number of Points
  buf.str("");
  buf << noPts;
  pPiece->setAttribute("NumberOfPoints", buf.str());
  // Add Number of Triangles/Polys
  buf.str("");
  buf << noTris;
  pPiece->setAttribute("NumberOfPolys", buf.str());
  // write the points
  AutoPtr<Element> pPoints = m_doc->createElement("Points");
  AutoPtr<Element> pPtsDataArray = m_doc->createElement("DataArray");
  // Add attributes to data array
  pPtsDataArray->setAttribute("type", "Float32");
  pPtsDataArray->setAttribute("NumberOfComponents", "3");
  pPtsDataArray->setAttribute("format", "ascii");
  buf.str("");
  // get the triangles info
  const auto &points = handle->getTriangleVertices();
  size_t i;
  for (i = 0; i < points.size(); i++) {
    buf << points[i] << " ";
  }
  AutoPtr<Text> pPointText = m_doc->createTextNode(buf.str());
  pPtsDataArray->appendChild(pPointText);
  pPoints->appendChild(pPtsDataArray);
  // get triangles faces info
  AutoPtr<Element> pFaces = m_doc->createElement("Polys");
  AutoPtr<Element> pTrisDataArray = m_doc->createElement("DataArray");
  // add attribute
  pTrisDataArray->setAttribute("type", "UInt32");
  pTrisDataArray->setAttribute("Name", "connectivity");
  pTrisDataArray->setAttribute("format", "ascii");
 
  buf.str("");
  const auto &faces = handle->getTriangleFaces();
  for (i = 0; i < faces.size(); i++) {
    buf << faces[i] << " ";
  }
  AutoPtr<Text> pTrisDataText = m_doc->createTextNode(buf.str());
  pTrisDataArray->appendChild(pTrisDataText);
  pFaces->appendChild(pTrisDataArray);
  // set the offsets
  AutoPtr<Element> pTrisOffsetDataArray = m_doc->createElement("DataArray");
  // add attribute
  pTrisOffsetDataArray->setAttribute("type", "UInt32");
  pTrisOffsetDataArray->setAttribute("Name", "offsets");
  pTrisOffsetDataArray->setAttribute("format", "ascii");
  buf.str("");
  for (i = 1; i < noTris + 1; i++) {
    buf << i * 3 << " ";
  }
  AutoPtr<Text> pTrisOffsetDataText = m_doc->createTextNode(buf.str());
  pTrisOffsetDataArray->appendChild(pTrisOffsetDataText);
  pFaces->appendChild(pTrisOffsetDataArray);
 
  // append all
  pPiece->appendChild(pPoints);
  pPiece->appendChild(pFaces);
 
  // add this piece to root
  m_root->appendChild(pPiece);
}
void vtkGeometryCacheWriter::write() {
  DOMWriter writer;
  writer.setNewLine("\n");
  writer.setOptions(XMLWriter::PRETTY_PRINT);
  std::ofstream file;
  try {
    g_log.information("Writing Geometry Cache file to " + m_filename);
    file.open(m_filename.c_str(), std::ios::trunc);
    writer.writeNode(file, m_doc);
    file.close();
  } catch (...) {
    g_log.error("Geometry Cache file writing exception");
  }
}
} // namespace Mantid::Geometry