src/polydatareader.cpp

   1 //
   2 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   3 // +                                                                      +
   4 // + This file is part of enGrid.                                         +
   5 // +                                                                      +
   6 // + Copyright 2008,2009 Oliver Gloth                                     +
   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 "polydatareader.h"
  24 #include "vtkEgPolyDataToUnstructuredGridFilter.h"
  25
  26 #include <vtkXMLPolyDataReader.h>
  27
  28 #include "guimainwindow.h"
  29
  30 PolyDataReader::PolyDataReader()
  31 {
  32   setFormat("VTK poly data files(*.vtp)");
  33   setExtension(".vtp");
  34 }
  35
  36 void PolyDataReader::operate()
  37 {
  38   try {
  39     QFileInfo file_info(GuiMainWindow::pointer()->getFilename());
  40     readInputFileName(file_info.completeBaseName() + ".vtp");
  41     if (isValid()) {
  42       EG_VTKSP(vtkXMLPolyDataReader,vtp);
  43       EG_VTKSP(vtkEgPolyDataToUnstructuredGridFilter,pd2ug);
  44       vtp->SetFileName(qPrintable(getFileName()));
  45       pd2ug->SetInput(vtp->GetOutput());
  46       pd2ug->Update();
  47       m_Grid->DeepCopy(pd2ug->GetOutput());
  48       createBasicFields(m_Grid, m_Grid->GetNumberOfCells(), m_Grid->GetNumberOfPoints());
  49       UpdateNodeIndex(m_Grid);
  50       UpdateCellIndex(m_Grid);
  51       EG_VTKDCC(vtkIntArray, bc, m_Grid, "cell_code");
  52       for (vtkIdType id_cell = 0; id_cell < m_Grid->GetNumberOfPoints(); ++id_cell) {
  53         bc->SetValue(id_cell,99);
  54       }
  55     }
  56   } catch (Error err) {
  57     err.display();
  58   }
  59 }