HL_SignedDistTraverse.cc 7.9 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
/******************************************************************************
 *
 * AMDiS - Adaptive multidimensional simulations
 *
 * Copyright (C) 2013 Dresden University of Technology. All Rights Reserved.
 * Web: https://fusionforge.zih.tu-dresden.de/projects/amdis
 *
 * Authors: 
 * Simon Vey, Thomas Witkowski, Andreas Naumann, Simon Praetorius, et al.
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 *
 * This file is part of AMDiS
 *
 * See also license.opensource.txt in the distribution.
 * 
 ******************************************************************************/
20
21


22
23
24
#include "HL_SignedDistTraverse.h"
#include "VelocityExtFromVelocityField.h"

25
26
namespace reinit {

27
void HL_SignedDistTraverse::initializeBoundary() 
28
29
{
  FUNCNAME("HL_SignedDistTraverse::initializeBoundary()");
Thomas Witkowski's avatar
Thomas Witkowski committed
30
 
31
32
33
34
35
36
37
38
39
  // ===== All non-boundary vertices are initialized with "infinity". =====
  sD_DOF->set(inftyValue);
  
  // ===== Traverse mesh and initialize boundary elements. =====
  TraverseStack stack;
  FixVec<double, VERTEX> distVec(dim, NO_INIT);
  int elStatus;
  
  const int nBasFcts = feSpace->getBasisFcts()->getNumber();
40
  if (static_cast<int>(locInd.size()) < nBasFcts)
41
    locInd.resize(nBasFcts);
42
43
44
45
46
47
48
49
50
  
  ElInfo *elInfo;
  if (velExt && velExtType.isSet(VEL_EXT_FROM_VEL_FIELD)) {
    elInfo = stack.traverseFirst(feSpace->getMesh(),
				 -1, 
				 Mesh::CALL_LEAF_EL | 
				 Mesh::FILL_BOUND |
				 Mesh::FILL_COORDS |
				 Mesh::FILL_GRD_LAMBDA);
51
  } else {
52
53
54
55
56
57
    elInfo = stack.traverseFirst(feSpace->getMesh(),
				 -1, 
				 Mesh::CALL_LEAF_EL | 
				 Mesh::FILL_BOUND |
				 Mesh::FILL_COORDS);
  }
58

59
60
61
  while(elInfo) {

    // Set elInfo in case velocity extension from velocity field is used.
62
63
    if (velExt && velExtType.isSet(VEL_EXT_FROM_VEL_FIELD))
      ((VelocityExtFromVelocityField *)velExt)->setElInfo(elInfo);    
64
65
    
    // Get local indices of vertices.
Thomas Witkowski's avatar
Blub    
Thomas Witkowski committed
66
67
68
    feSpace->getBasisFcts()->getLocalIndices(const_cast<Element*>(elInfo->getElement()),
					     const_cast<DOFAdmin*>(feSpace->getAdmin()),
					     locInd); 
69
70
71
72
73
74
75
76
    
    // Get element status.
    elStatus = elLS->createElementLevelSet(elInfo);
    
    // Is element cut by the interface ?
    if (elStatus == ElementLevelSet::LEVEL_SET_BOUNDARY) {
      
      // Reset element distance vector.
77
78
      for (int i=0; i <= dim; i++)
	distVec[i] = inftyValue;      
79
80
      
      // Mark all vertices as boundary vertices.
81
82
      for (int i = 0; i <= dim; i++)
	(*bound_DOF)[locInd[i]] = 1.0;      
83
84
85
86
87
      
      // Calculate distance for all vertices.
      if (bndElDist->calcDistOnBoundaryElement(elInfo, distVec) !=
	  ElementLevelSet::LEVEL_SET_BOUNDARY) {
	ERROR_EXIT("error in distance calculation !\n");
88
      } else {
89
90
	
	// If distance is smaller, correct to new distance.
91
	for (int i = 0; i <= dim; i++) {
92
93
	  
	  // --> for test purposes:
Thomas Witkowski's avatar
blub    
Thomas Witkowski committed
94
95
96
97
98
99
	  if (distVec[i] > 1000) {
	    MSG("Element %d  Knoten %d  keine Randwertinitialisierung!\n",
		elInfo->getElement()->getIndex(), i);
	  }
		  
	  // --> end: for test purposes	  
100
101
102
	  if ((*sD_DOF)[locInd[i]] > distVec[i]) {
	    (*sD_DOF)[locInd[i]] = distVec[i];
	    //If Distance is corrected, calculate new velocity.
103
	    if (velExt != NULL)	      
104
	      velExt->calcVelocityBoundary(&(locInd[0]), i);	      
105
106
107
108
109
110
	  }
	}
      }
    }  // end of: elStatus == ElementLevelSet::LEVEL_SET_BOUNDARY
    
    elInfo = stack.traverseNext(elInfo);
111
  }  // end of: mesh traverse 
112
113
}

114

115
void HL_SignedDistTraverse::HL_updateIteration()
116
{
Thomas Witkowski's avatar
blub    
Thomas Witkowski committed
117
118
  FUNCNAME("HL_SignedDistTraverse::HL_updateIteration()");

119
120
  // ===== Create DOF vector for the last iteration step. =====
  if (sDOld_DOF)
121
122
123
    delete sDOld_DOF;

  sDOld_DOF = new DOFVector<double>(feSpace, "sDOld_DOF");
124
125
126
  sDOld_DOF->copy(const_cast<DOFVector<double> &>(*sD_DOF));
  
  // ===== Gauss-Seidel or Jacobi iteration ? =====
127
128
129
130
  if (GaussSeidelFlag)
    update_DOF = sD_DOF;  
  else
    update_DOF = sDOld_DOF;  
131
132
133
134
135
136
  
  // ===== Iteration loop: proceed until tolerance is reached. =====
  TraverseStack stack;
  ElInfo *elInfo;
  tol_reached = false;
  int itCntr = 0;
137
  while (!tol_reached && itCntr != maxIt) {   
138
139
140
    ++itCntr;
    tol_reached = true;
    
141
142
143
144
145
    // ===== Traverse mesh: perform Hopf-Lax element update on each element. =====
    elInfo = 
      stack.traverseFirst(feSpace->getMesh(), -1, 
			  Mesh::CALL_LEAF_EL | Mesh::FILL_BOUND | Mesh::FILL_COORDS);
    while (elInfo) {
146
147
148
149
150
151
152
153
154
155
      HL_elementUpdate(elInfo);
      elInfo = stack.traverseNext(elInfo);
    }
    
    // ===== Is tolerance reached ? =====
    tol_reached = checkTol();
    
    sDOld_DOF->copy(const_cast<DOFVector<double> &>(*sD_DOF));
  }
  
156

Thomas Witkowski's avatar
blub    
Thomas Witkowski committed
157
  MSG("Calculation of signed distance function via mesh traverse iteration:\n");
158
  if (GaussSeidelFlag)
Thomas Witkowski's avatar
blub    
Thomas Witkowski committed
159
    MSG("\tGauss-Seidel iteration\n");
160
  else
Thomas Witkowski's avatar
blub    
Thomas Witkowski committed
161
    MSG("\tJacobi iteration\n");
162
  
Thomas Witkowski's avatar
blub    
Thomas Witkowski committed
163
  MSG("\tnumber of iterations needed: %d\n", itCntr);
164
165
}

166

167
void HL_SignedDistTraverse::HL_elementUpdate(ElInfo *elInfo) 
168
169
170
{
  // ===== Get global indices of vertices of element. =====
  const int nBasFcts = feSpace->getBasisFcts()->getNumber();
171
  if (static_cast<int>(locInd.size()) < nBasFcts)
172
    locInd.resize(nBasFcts);
173
  
Thomas Witkowski's avatar
Blub    
Thomas Witkowski committed
174
175
176
  feSpace->getBasisFcts()->getLocalIndices(const_cast<Element *>(elInfo->getElement()),
					   const_cast<DOFAdmin *>(feSpace->getAdmin()),
					   locInd);
177
178
  
  // ===== Hopf-Lax element update for each vertex of element. =====
179
  for (int i = 0; i <= dim; i++) {
180
181
182
183
    
    // ===== Calculate update for non-boundary vertex. =====
    if ((*bound_DOF)[locInd[i]] < 1.e-15) {
      //save permutation of vertexes for calculation of the velocity
184
      if (velExt != NULL)
185
186
187
	velExt->setPermutation(i, 1);
	
      double update = calcElementUpdate(elInfo, i, &(locInd[0]));
188
189
190
191
192
193
194
195
      // ---> for test purposes: count number of calculated updates
      ++calcUpdate_Cntr;
      // ---> end: for test purposes
      
      // Calculates minimum of all element updates on elements
      // containing vertex i.
      if (update < (*sD_DOF)[locInd[i]]) {
	(*sD_DOF)[locInd[i]] = update;
196
197

	// If Distance is corrected, calculate new velocity.
198
	if(velExt != NULL)
199
200
	  velExt->calcVelocity(&(locInd[0]), i);
	
201
202
203
204
205
	// ---> for test purposes: count number of calculated updates
	++setUpdate_Cntr;
	// ---> end: for test purposes
      }
    }
206
  } 
207
208
}

209

210
211
212
double HL_SignedDistTraverse::calcElementUpdate(ElInfo *elInfo, 
						int nXh,
						const DegreeOfFreedom *locInd) 
213
214
{
  // ===== Get local indices of element vertices (except xh). =====
215
216
217
218
219
  int nWh = 0;  
  int nYh = (nXh + 1) % (dim+1);
  int nZh = (nXh + 2) % (dim+1);
  if (dim == 3) 
    nWh = (nXh + 3) % (dim+1);
220
221
222
223
224
225
  
  // ===== Get world coordinates of vertices of element and their values
  //       of uh. 
  //       The coordinates of the vertex the update is calculated for
  //       are stored at the end of the vector elVert. =====
  switch (dim) {
226
227
  case 2: 
    elVert[0] = &(elInfo->getCoord(nYh));
228
229
230
231
232
233
234
235
    elVert[1] = &(elInfo->getCoord(nZh));
    elVert[2] = &(elInfo->getCoord(nXh));
    
    uhVal[0] = (*update_DOF)[locInd[nYh]];
    uhVal[1] = (*update_DOF)[locInd[nZh]];
    uhVal[2] = (*update_DOF)[locInd[nXh]];
    
    break;
236
237
  case 3: 
    elVert[0] = &(elInfo->getCoord(nYh));
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
    elVert[1] = &(elInfo->getCoord(nZh));
    elVert[2] = &(elInfo->getCoord(nWh));
    elVert[3] = &(elInfo->getCoord(nXh));

    uhVal[0] = (*update_DOF)[locInd[nYh]];
    uhVal[1] = (*update_DOF)[locInd[nZh]];
    uhVal[2] = (*update_DOF)[locInd[nWh]];
    uhVal[3] = (*update_DOF)[locInd[nXh]];
    
    break;
  default: ERROR_EXIT("illegal dimension !\n");
    break;
  }
  
  // ===== Calculate Hopf-Lax element update for vertex. =====
253
  return elUpdate->calcElementUpdate(elVert, uhVal);
254
255
}

256

257
bool HL_SignedDistTraverse::checkTol()
258
259
260
261
{
  DOFVector<double>::Iterator it_sD(sD_DOF, USED_DOFS);
  DOFVector<double>::Iterator it_sDOld(sDOld_DOF, USED_DOFS);
  
262
263
  for (it_sD.reset(), it_sDOld.reset(); !it_sD.end(); ++it_sD, ++it_sDOld)
    if ((*it_sDOld) - (*it_sD) > tol || (*it_sDOld) - (*it_sD) < 0)
264
265
266
267
      return false;
  
  return true;
}
268
269

}