src/libengrid/plywriter.cpp

   1 //
   2 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   3 // +                                                                      +
   4 // + This file is part of enGrid.                                         +
   5 // +                                                                      +
   6 // + Copyright 2008-2013 enGits GmbH                                      +
   7 // +                                                                      +
   8 // + enGrid is free software: you can redistribute it and/or modify       +
   9 // + it under the terms of the GNU General Public License as published by +
  10 // + the Free Software Foundation, either version 3 of the License, or    +
  11 // + (at your option) any later version.                                  +
  12 // +                                                                      +
  13 // + enGrid is distributed in the hope that it will be useful,            +
  14 // + but WITHOUT ANY WARRANTY; without even the implied warranty of       +
  15 // + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        +
  16 // + GNU General Public License for more details.                         +
  17 // +                                                                      +
  18 // + You should have received a copy of the GNU General Public License    +
  19 // + along with enGrid. If not, see <http://www.gnu.org/licenses/>.       +
  20 // +                                                                      +
  21 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  22 //
  23 #include "plywriter.h"
  24
  25 #include <vtkPLYWriter.h>
  26 #include <vtkGeometryFilter.h>
  27 #include <vtkTriangleFilter.h>
  28
  29 #include <QFileInfo>
  30 #include "guimainwindow.h"
  31
  32 PlyWriter::PlyWriter()
  33 {
  34   setFormat("Stanford PLY files (*.ply *.PLY)");
  35   m_AsciiFileType = true;
  36 };
  37
  38 void PlyWriter::operate()
  39 {
  40   try {
  41     QFileInfo file_info(GuiMainWindow::pointer()->getFilename());
  42     readOutputFileName(file_info.completeBaseName() + ".ply");
  43     if (isValid()) {
  44       EG_VTKSP(vtkGeometryFilter, geometry);
  45       geometry->SetInput(m_Grid);
  46       EG_VTKSP(vtkTriangleFilter, triangle);
  47       triangle->SetInput(geometry->GetOutput());
  48
  49       EG_VTKSP(vtkPLYWriter, write_ply);
  50       write_ply->SetInput(triangle->GetOutput());
  51       write_ply->SetFileName(qPrintable(getFileName()));
  52       if(m_AsciiFileType) {
  53         write_ply->SetFileTypeToASCII();
  54       }
  55       else {
  56         write_ply->SetFileTypeToBinary();
  57       }
  58       write_ply->SetDataByteOrderToLittleEndian();
  59       write_ply->SetColorModeToUniformCellColor();
  60       write_ply->SetColor(255,0,0);
  61       write_ply->Write();
  62
  63     };
  64   } catch (Error err) {
  65     err.display();
  66   };
  67 };