deletecells.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 "deletecells.h"
  24
  25 void DeleteCells::setTraceCells(const QVector<vtkIdType> &cells)
  26 {
  27   trace_cells.resize(cells.size());
  28   qCopy(cells.begin(), cells.end(), trace_cells.begin());
  29 };
  30
  31 void DeleteCells::getTraceCells(QVector<vtkIdType> &cells)
  32 {
  33   cells.resize(trace_cells.size());
  34   qCopy(trace_cells.begin(), trace_cells.end(), cells.begin());
  35 };
  36
  37 void DeleteCells::operate()
  38 {
  39   EG_VTKSP(vtkUnstructuredGrid, new_grid);
  40   QVector<vtkIdType> new_cells;
  41   QVector<vtkIdType> new_nodes;
  42   getRestCells(grid, del_cells, new_cells);
  43   getNodesFromCells(new_cells, new_nodes, grid);
  44   allocateGrid(new_grid, new_cells.size(), new_nodes.size());
  45   QVector<vtkIdType> old2new_nodes(grid->GetNumberOfPoints(), -1);
  46   QVector<vtkIdType> old2new_cells(grid->GetNumberOfCells(), -1);
  47   {
  48     {
  49       vtkIdType id_new = 0;
  50       foreach (vtkIdType id_node, new_nodes) {
  51         vec3_t x;
  52         grid->GetPoints()->GetPoint(id_node, x.data());
  53         new_grid ->GetPoints()->SetPoint(id_new, x.data());
  54         copyNodeData(grid, id_node, new_grid, id_new);
  55         old2new_nodes[id_node] = id_new;
  56         ++id_new;
  57       };
  58     };
  59   };
  60   {
  61     foreach (vtkIdType id_cell, new_cells) {
  62       vtkIdType *pts, N_pts;
  63       grid->GetCellPoints(id_cell, N_pts, pts);
  64       QVector<vtkIdType> new_pts(N_pts);
  65       for (int i = 0; i < N_pts; ++i) {
  66         new_pts[i] = old2new_nodes[pts[i]];
  67       };
  68       vtkIdType cellType = grid->GetCellType(id_cell);
  69       vtkIdType id_new = new_grid->InsertNextCell(cellType, N_pts, new_pts.data());
  70       copyCellData(grid, id_cell, new_grid, id_new);
  71       old2new_cells[id_cell] = id_new;
  72     };
  73     QList<vtkIdType> new_trace_cells;
  74     foreach (vtkIdType id_cell, trace_cells) {
  75       if (old2new_cells[id_cell] != -1) {
  76         new_trace_cells.append(old2new_cells[id_cell]);
  77       };
  78     };
  79     trace_cells.resize(new_trace_cells.size());
  80     qCopy(new_trace_cells.begin(), new_trace_cells.end(), trace_cells.begin());
  81   };
  82   makeCopy(new_grid, grid);
  83 };
  84