src/operation.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 "operation.h"
  24
  25 #include "guimainwindow.h"
  26 #include "egvtkobject.h"
  27
  28 #include <vtkTriangleFilter.h>
  29 #include <vtkInformation.h>
  30 #include <vtkInformationVector.h>
  31 #include <vtkObjectFactory.h>
  32 #include <vtkPointData.h>
  33 #include <vtkPolyData.h>
  34 #include <vtkPolygon.h>
  35 #include <vtkStreamingDemandDrivenPipeline.h>
  36 #include <vtkCellArray.h>
  37 #include <vtkCellData.h>
  38 #include <vtkCellLocator.h>
  39 #include <vtkFloatArray.h>
  40 #include <vtkMath.h>
  41 #include <vtkCharArray.h>
  42
  43 #include <QApplication>
  44 #include <QTime>
  45
  46 #include "geometrytools.h"
  47 using namespace GeometryTools;
  48
  49 QSet<Operation*> Operation::garbage_operations;
  50
  51 QVector<vtkIdType>     m_static_DummyCells;
  52 QVector<int>           m_static_DummyRCells;
  53 QVector<vtkIdType>     m_static_DummyNodes;
  54 QVector<int>           m_static_DummyRNodes;
  55 QVector<QVector<int> > m_static_DummyN2N;
  56 QVector<QVector<int> > m_static_DummyN2C;
  57 QVector<QVector<int> > m_static_DummyC2C;
  58
  59
  60
  61 void Operation::collectGarbage()
  62 {
  63   QSet<Operation*> delete_operations;
  64
  65   foreach (Operation *op, garbage_operations)
  66   {
  67     if (!op->getThread().isRunning()) {
  68       delete_operations.insert(op);
  69       cout << "deleting Operation " << op << endl;
  70       delete op;
  71     }
  72   }
  73
  74   foreach (Operation *op, delete_operations)
  75   {
  76     garbage_operations.remove(op);
  77   }
  78 }
  79
  80 Operation::Operation()
  81 {
  82   grid = NULL;
  83   gui = false;
  84   m_quicksave = false;
  85   m_resetoperationcounter = false;
  86   err = NULL;
  87   autoset = true;
  88   m_TypeName = "undefined";
  89 }
  90
  91 Operation::~Operation()
  92 {
  93   if (err) {
  94     err->display();
  95     delete err;
  96   }
  97 }
  98
  99 void Operation::del()
 100 {
 101   garbage_operations.insert(this);
 102 }
 103
 104 void OperationThread::run()
 105 {
 106   try {
 107     GuiMainWindow::lock();
 108     GuiMainWindow::pointer()->setBusy();
 109     op->operate();
 110     cout << "secs. for " << qPrintable(op->getTypeName()) << ": " << op->elapsedTime() << endl;
 111   } catch (Error err) {
 112     op->err = new Error();
 113     *(op->err) = err;
 114   }
 115   GuiMainWindow::unlock();
 116   GuiMainWindow::pointer()->setIdle();
 117 }
 118
 119 void Operation::setTypeName(QString name)
 120 {
 121   int i = 0;
 122   while ((i < name.size()) && (name[i].isDigit())) {
 123     ++i;
 124   }
 125   m_TypeName = name.right(name.size() - i);
 126 }
 127
 128 void Operation::operator()()
 129 {
 130   setStartTime();
 131   if (gui) {
 132     if (GuiMainWindow::tryLock()) {
 133       checkGrid();
 134       thread.setOperation(this);
 135       GuiMainWindow::unlock();
 136       thread.start(QThread::LowPriority);
 137     } else {
 138       QMessageBox::warning(NULL, "not permitted", "Operation is not permitted while background process is running!");
 139     }
 140   } else {
 141     checkGrid();
 142     const bool gui_thread = QThread::currentThread() == QCoreApplication::instance()->thread();
 143     if (gui_thread) {
 144       try {
 145         operate();
 146         //cout << "secs. for " << qPrintable(getTypeName()) << ": " << elapsedTime() << endl;
 147       } catch (Error err) {
 148         err.display();
 149       }
 150     } else {
 151       operate();
 152     }
 153     if(m_resetoperationcounter) GuiMainWindow::pointer()->resetOperationCounter();
 154     if(m_quicksave) GuiMainWindow::pointer()->quickSave();
 155   }
 156 }
 157
 158 void Operation::setAllCells()
 159 {
 160   QVector<vtkIdType> all_cells;
 161   getAllCells(all_cells, grid);
 162   setCells(all_cells);
 163 }
 164
 165 void Operation::setAllVolumeCells()
 166 {
 167   QVector<vtkIdType> cells;
 168   getAllVolumeCells(cells, grid);
 169   setCells(cells);
 170 }
 171
 172 void Operation::setAllSurfaceCells()
 173 {
 174   QVector<vtkIdType> cells;
 175   getAllSurfaceCells(cells, grid);
 176   setCells(cells);
 177 }
 178
 179 void Operation::setVolume(QString volume_name)
 180 {
 181   m_Part.setGrid(grid);
 182   m_Part.setVolume(volume_name);
 183 }
 184
 185 void Operation::setMeshPartition(const MeshPartition &part)
 186 {
 187   m_Part.setGrid(part.getGrid());
 188   m_Part.setCells(part.getCells());
 189 }
 190
 191 void Operation::checkGrid()
 192 {
 193   if (grid == NULL) {
 194     grid = GuiMainWindow::pointer()->getGrid();
 195   }
 196   l2g_t cells = getPartCells();
 197   if ((cells.size() == 0) && autoset) {
 198     setAllCells();
 199   }
 200 }
 201
 202 void Operation::updateActors()
 203 {
 204   mainWindow()->updateActors();
 205 }
 206
 207 GuiMainWindow* Operation::mainWindow()
 208 {
 209   return GuiMainWindow::pointer();
 210 }
 211
 212 void Operation::populateBoundaryCodes(QListWidget *lw)
 213 {
 214   QSet<int> bcs;
 215   mainWindow()->getAllBoundaryCodes(bcs);
 216   foreach(int bc, bcs) {
 217     QListWidgetItem *lwi = new QListWidgetItem(lw);
 218     lwi->setCheckState(Qt::Unchecked);
 219     QString text = "";
 220     QTextStream ts(&text);
 221     ts << bc;
 222     lwi->setText(text);
 223     lwi->setFlags(Qt::ItemIsUserCheckable | Qt::ItemIsEnabled);
 224   }
 225 }
 226
 227 void Operation::populateVolumes(QListWidget *lw)
 228 {
 229   QList<VolumeDefinition> vols = mainWindow()->getAllVols();
 230   foreach (VolumeDefinition V, vols) {
 231     QListWidgetItem *lwi = new QListWidgetItem(lw);
 232     lwi->setText(V.getName());
 233   }
 234 }