guicreateboundarylayer.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 "guicreateboundarylayer.h"
  24 #include "seedsimpleprismaticlayer.h"
  25 #include "gridsmoother.h"
  26 #include "createvolumemesh.h"
  27 #include "swaptriangles.h"
  28 #include "deletetetras.h"
  29 #include "deletecells.h"
  30
  31 GuiCreateBoundaryLayer::GuiCreateBoundaryLayer()
  32 {
  33   getSet("boundary layer", "maximal relative error", 0.01, err_max);
  34   getSet("boundary layer", "maximal number of smoothing iterations", 10, max_iter);
  35 };
  36
  37 void GuiCreateBoundaryLayer::before()
  38 {
  39   populateBoundaryCodes(ui.listWidget);
  40 };
  41
  42 void GuiCreateBoundaryLayer::operate()
  43 {
  44   getSelectedItems(ui.listWidget, boundary_codes);
  45   getSurfaceCells(boundary_codes, layer_cells, grid);
  46
  47   cout << "\n\ncreating boundary layer mesh)" << endl;
  48
  49   {
  50     EG_VTKDCN(vtkIntArray, node_status, grid, "node_status");
  51     EG_VTKDCN(vtkIntArray, node_layer,  grid, "node_layer");
  52     EG_VTKDCC(vtkIntArray, bc,          grid, "cell_code");
  53     foreach(vtkIdType id_node, nodes) {
  54       node_status->SetValue(id_node, 0);
  55       QSet<int> bcs;
  56       foreach (int i_neigh_cells, n2c[_nodes[id_node]]) {
  57         vtkIdType id_neigh_cell = cells[i_neigh_cells];
  58         if (isSurface(grid, id_neigh_cell)) {
  59           if (boundary_codes.contains(bc->GetValue(id_neigh_cell))) {
  60             bcs.insert(bc->GetValue(id_neigh_cell));
  61           };
  62         };
  63       };
  64       if (bcs.size() >= 2) {
  65         node_status->SetValue(id_node, 1);
  66       };
  67       node_layer->SetValue(id_node, -1);
  68     };
  69     foreach (vtkIdType id_cell, layer_cells) {
  70       vtkIdType N_pts, *pts;
  71       grid->GetCellPoints(id_cell, N_pts, pts);
  72       for (int i_pts = 0; i_pts < N_pts; ++i_pts) {
  73         node_layer->SetValue(pts[i_pts], 0);
  74       };
  75     };
  76   };
  77
  78   cout << "preparing prismatic layer" << endl;
  79
  80   GridSmoother smooth;
  81   smooth.setGrid(grid);
  82   smooth.setBoundaryCodes(boundary_codes);
  83   smooth.prismsOn();
  84   //smooth.setNumIterations(5);
  85
  86   SeedSimplePrismaticLayer seed_layer;
  87
  88   CreateVolumeMesh vol;
  89   vol.setGrid(grid);
  90   SwapTriangles swap;
  91   swap.setGrid(grid);
  92   swap.setBoundaryCodes(boundary_codes);
  93   DeleteTetras del;
  94   del.setGrid(grid);
  95
  96   seed_layer.setGrid(grid);
  97   del();
  98   vol();
  99   seed_layer.setAllCells();
 100   seed_layer.setLayerCells(layer_cells);
 101   seed_layer.setBoundaryCodes(boundary_codes);
 102   seed_layer();
 103   seed_layer.getLayerCells(layer_cells);
 104
 105   for (int j = 0; j < max_iter; ++j) {
 106     cout << "improving prismatic layer -> iteration " << j+1 << "/" << max_iter << endl;
 107     smooth.setAllCells();
 108     smooth();
 109     del.setAllCells();
 110     del();
 111     swap();
 112     vol.setTraceCells(layer_cells);
 113     vol();
 114     vol.getTraceCells(layer_cells);
 115     if (smooth.improvement() < err_max) break;
 116   };
 117   //smooth.setAllCells();
 118   //smooth();
 119
 120   /*
 121   del();
 122   vol();
 123   cout << "finalising prismatic layer" << endl;
 124   smooth.setAllCells();
 125   smooth();
 126   */
 127   createIndices(grid);
 128 };
 129