src/libengrid/stlwriter.cpp

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