src/vtkEgGridFilter.cxx

   1 //
   2 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   3 // +                                                                      +
   4 // + This file is part of enGrid.                                         +
   5 // +                                                                      +
   6 // + Copyright 2008 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 "vtkEgGridFilter.h"
  24 #include "vtkInformation.h"
  25
  26 #include "guimainwindow.h"
  27
  28 vtkEgGridFilter::vtkEgGridFilter()
  29 {
  30   m_BoundaryCodes.clear();
  31   m_Input = NULL;
  32   m_Output = NULL;
  33   m_LastOutput = vtkUnstructuredGrid::New();
  34   m_LastRun = GetMTime();
  35 }
  36
  37 vtkEgGridFilter::~vtkEgGridFilter()
  38 {
  39   m_LastOutput->Delete();
  40 }
  41
  42 int vtkEgGridFilter::FillInputPortInformation
  43 (
  44   int,
  45   vtkInformation* info
  46 )
  47 {
  48   info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(),"vtkUnstructuredGrid");
  49   return 1;
  50 }
  51
  52 int vtkEgGridFilter::FillOutputPortInformation
  53 (
  54   int,
  55   vtkInformation* info
  56 )
  57 {
  58   info->Set(vtkDataObject::DATA_TYPE_NAME(),"vtkUnstructuredGrid");
  59   return 1;
  60 }
  61
  62 int vtkEgGridFilter::RequestData
  63 (
  64   vtkInformation*,
  65   vtkInformationVector** m_InputVector,
  66   vtkInformationVector*  m_OutputVector
  67 )
  68 {
  69   // Get m_Input and m_Output data
  70   m_Input  = vtkUnstructuredGrid::GetData(m_InputVector[0]);
  71   m_Output = vtkUnstructuredGrid::GetData(m_OutputVector);
  72
  73   // check if we have a proper m_InputVector
  74   if (!m_Input)                                      { vtkDebugMacro("No input!"); return 1; }
  75   if (!m_Input->GetPoints())                         { vtkDebugMacro("No input!"); return 1; }
  76   if (m_Input->GetPoints()->GetNumberOfPoints() < 1) { vtkDebugMacro("No input!"); return 1; }
  77
  78   // call the enGrid filter method
  79   try {
  80     bool m_Input_changed = (m_Input->GetMTime() > m_LastRun);
  81     bool filter_changed = (GetMTime() > m_LastRun);
  82     if (m_Input_changed || filter_changed) {
  83       if (GuiMainWindow::tryLock()) {
  84         ExecuteEg();
  85         makeCopy(m_Output, m_LastOutput);
  86         Modified();
  87         m_LastRun = GetMTime();
  88         GuiMainWindow::unlock();
  89       } else {
  90         makeCopy(m_LastOutput, m_Output);
  91       }
  92     }
  93   } catch (Error err) {
  94     GuiMainWindow::unlock();
  95     err.display();
  96     m_Output->DeepCopy(m_Input);
  97   }
  98
  99   return 1;
 100 }
 101
 102 void vtkEgGridFilter::SetBoundaryCodes(const QSet<int> &bc)
 103 {
 104   bool update = false;
 105   if (m_BoundaryCodes.size() != bc.size()) {
 106     update = true;
 107   } else {
 108     QSet<int> bc_inters = m_BoundaryCodes.intersect(bc);
 109     if (bc_inters.size() != bc.size()) {
 110       update = true;
 111     }
 112   }
 113   if (update) {
 114     m_BoundaryCodes = bc;
 115     Modified();
 116   }
 117 }
 118
 119 void vtkEgGridFilter::ExtractBoundary(QVector<vtkIdType>  &cells, QVector<vtkIdType>  &nodes, QSet<int> &bc, vtkUnstructuredGrid *grid)
 120 {
 121   cells.clear();
 122   nodes.clear();
 123   QSet<vtkIdType> ex_nodes, ex_cells;
 124   EG_VTKDCC(vtkIntArray, cell_code, grid, "cell_code");
 125   for (vtkIdType i = 0; i < grid->GetNumberOfCells(); ++i) {
 126     if ((grid->GetCellType(i) == VTK_TRIANGLE) || (grid->GetCellType(i) == VTK_QUAD)){
 127       if (bc.contains(cell_code->GetValue(i))) {
 128         ex_cells.insert(i);
 129         vtkIdType *pts;
 130         vtkIdType npts;
 131         m_Input->GetCellPoints(i,npts,pts);
 132         for (int j = 0; j < npts; ++j) {
 133           ex_nodes.insert(pts[j]);
 134         }
 135       }
 136     }
 137   }
 138   cells.resize(ex_cells.size());
 139   nodes.resize(ex_nodes.size());
 140   {
 141     int j = 0;
 142     vtkIdType i;
 143     foreach(i,ex_cells) {
 144       cells[j] = i;
 145       ++j;
 146     }
 147   }
 148   {
 149     int j = 0;
 150     vtkIdType i;
 151     foreach(i,ex_nodes) {
 152       nodes[j] = i;
 153       ++j;
 154     }
 155   }
 156 }
 157
 158
 159
 160
 161
 162
 163
 164
 165
 166
 167
 168
 169
 170
 171
 172
 173
 174
 175
 176
 177
 178
 179
 180