ResidualEstimator.cc 14.3 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
#include "ResidualEstimator.h"
#include "Operator.h"
#include "DOFMatrix.h"
#include "DOFVector.h"
#include "Assembler.h"
#include "Traverse.h"
#include "Parameters.h"

namespace AMDiS {

11
  ResidualEstimator::ResidualEstimator(std::string name, int r) 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
    : Estimator(name, r),
      C0(1.0), 
      C1(1.0), 
      C2(1.0), 
      C3(1.0)
  {
    GET_PARAMETER(0, name + "->C0", "%f", &C0);
    GET_PARAMETER(0, name + "->C1", "%f", &C1);
    GET_PARAMETER(0, name + "->C2", "%f", &C2);
    GET_PARAMETER(0, name + "->C3", "%f", &C3);

    C0 = C0 > 1.e-25 ? sqr(C0) : 0.0;
    C1 = C1 > 1.e-25 ? sqr(C1) : 0.0;
    C2 = C2 > 1.e-25 ? sqr(C2) : 0.0;
    C3 = C3 > 1.e-25 ? sqr(C3) : 0.0;
  }

  void ResidualEstimator::init(double ts)
  {
    FUNCNAME("ResidualEstimator::init()");
Thomas Witkowski's avatar
Thomas Witkowski committed
32
    
33
34
35
36
    timestep = ts;

    mesh = uh[row == -1 ? 0 : row]->getFESpace()->getMesh();

Thomas Witkowski's avatar
Thomas Witkowski committed
37
38
    nSystems = static_cast<int>(uh.size());
    TEST_EXIT_DBG(nSystems > 0)("no system set\n");
39
40

    dim = mesh->getDim();
41
42
    basFcts = new const BasisFunction*[nSystems];
    quadFast = new FastQuadrature*[nSystems];
43
44

    degree = 0;
Thomas Witkowski's avatar
Thomas Witkowski committed
45
    for (int system = 0; system < nSystems; system++) {
46
      basFcts[system] = uh[system]->getFESpace()->getBasisFcts();
47
      degree = std::max(degree, basFcts[system]->getDegree());
48
49
50
51
52
    }

    degree *= 2;

    quad = Quadrature::provideQuadrature(dim, degree);
Thomas Witkowski's avatar
Thomas Witkowski committed
53
    nPoints = quad->getNumPoints();
54
55
56
57
58
59

    Flag flag = INIT_PHI | INIT_GRD_PHI;
    if (degree > 2) {
      flag |= INIT_D2_PHI;
    }

Thomas Witkowski's avatar
Thomas Witkowski committed
60
    for (int system = 0; system < nSystems; system++) {
61
62
63
64
65
      quadFast[system] = FastQuadrature::provideFastQuadrature(basFcts[system], 
							       *quad, 
							       flag);
    }
  
66
67
68
    uhEl = new double*[nSystems];
    uhNeigh = new double*[nSystems];
    uhOldEl = timestep ? new double*[nSystems] : NULL;
69

Thomas Witkowski's avatar
Thomas Witkowski committed
70
    for (int system = 0; system < nSystems; system++) {
71
72
      uhEl[system] = new double[basFcts[system]->getNumber()]; 
      uhNeigh[system] = new double[basFcts[system]->getNumber()];
73
      if (timestep)
74
	uhOldEl[system] = new double[basFcts[system]->getNumber()];
75
76
    }

77
78
    uhQP = timestep ? new double[nPoints] : NULL;
    uhOldQP = timestep ? new double[nPoints] : NULL;
79

80
    riq = new double[nPoints];
81

82
83
84
    grdUh_qp = NULL;
    D2uhqp = NULL;

85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
    TraverseStack stack;
    ElInfo *elInfo = NULL;

    // clear error indicators and mark elements for jumpRes
    elInfo = stack.traverseFirst(mesh, -1, Mesh::CALL_LEAF_EL);
    while (elInfo) {
      elInfo->getElement()->setEstimation(0.0, row);
      elInfo->getElement()->setMark(1);
      elInfo = stack.traverseNext(elInfo);
    }

    est_sum = 0.0;
    est_max = 0.0;
    est_t_sum = 0.0;
    est_t_max = 0.0;

    traverseFlag = 
      Mesh::FILL_NEIGH      |
      Mesh::FILL_COORDS     |
      Mesh::FILL_OPP_COORDS |
      Mesh::FILL_BOUND      |
      Mesh::FILL_GRD_LAMBDA |
      Mesh::FILL_DET        |
      Mesh::CALL_LEAF_EL;
109
110
111
112
113
114
115
116
117
118
119
120

    neighInfo = mesh->createNewElInfo();

    // prepare date for computing jump residual
    if (C1 && (dim > 1)) {
      surfaceQuad_ = Quadrature::provideQuadrature(dim - 1, degree);
      nPointsSurface_ = surfaceQuad_->getNumPoints();
      grdUhEl_.resize(nPointsSurface_);
      grdUhNeigh_.resize(nPointsSurface_);
      jump_.resize(nPointsSurface_);
      localJump_.resize(nPointsSurface_);
      neighbours_ = Global::getGeo(NEIGH, dim);
Thomas Witkowski's avatar
Thomas Witkowski committed
121
122
      lambdaNeigh_ = new DimVec<WorldVector<double> >(dim, NO_INIT);
      lambda_ = new DimVec<double>(dim, NO_INIT);
123
    }
124
125
126
127
128
129
130
131
132
  }

  void ResidualEstimator::exit(bool output)
  {
    FUNCNAME("ResidualEstimator::exit()");

    est_sum = sqrt(est_sum);
    est_t_sum = sqrt(est_t_sum);

Thomas Witkowski's avatar
Thomas Witkowski committed
133
    for (int system = 0; system < nSystems; system++) {
134
135
      delete [] uhEl[system];
      delete [] uhNeigh[system];
136
      if (timestep)
137
	delete [] uhOldEl[system];
138
139
    }

140
141
    delete [] uhEl;
    delete [] uhNeigh;
142
143

    if (timestep) {
144
145
146
      delete [] uhOldEl;
      delete [] uhQP;
      delete [] uhOldQP;
147
    } else {
148
149
      if (uhQP != NULL)
	delete [] uhQP;
150
151
152
153
    }

    if (output) {
      MSG("estimate   = %.8e\n", est_sum);
154
      if (C3)
155
156
157
	MSG("time estimate   = %.8e\n", est_t_sum);
    }

158
159
160
    delete [] riq;
    delete [] basFcts;
    delete [] quadFast;
161

162
163
164
165
    if (grdUh_qp != NULL)
      delete [] grdUh_qp;
    if (D2uhqp != NULL)
      delete [] D2uhqp;
166

167
    if (C1 && (dim > 1)) {
Thomas Witkowski's avatar
Thomas Witkowski committed
168
169
      delete lambdaNeigh_;
      delete lambda_;
170
171
    }

Thomas Witkowski's avatar
Thomas Witkowski committed
172
    delete neighInfo;
173
174
175
  }

  void ResidualEstimator::estimateElement(ElInfo *elInfo)
176
  {    
177
178
    FUNCNAME("ResidualEstimator::estimateElement()");

Thomas Witkowski's avatar
Thomas Witkowski committed
179
    TEST_EXIT_DBG(nSystems > 0)("no system set\n");
180

Thomas Witkowski's avatar
Thomas Witkowski committed
181
    double val = 0.0;
182
    std::vector<Operator*>::iterator it;
183
    std::vector<double*>::iterator itfac;
Thomas Witkowski's avatar
Thomas Witkowski committed
184
    Element *el = elInfo->getElement();
185
    double det = elInfo->getDet();
186
    const DimVec<WorldVector<double> > &grdLambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
187
    double est_el = el->getEstimation(row);
188
189
    double h2 = h2_from_det(det, dim);

Thomas Witkowski's avatar
Thomas Witkowski committed
190
    for (int iq = 0; iq < nPoints; iq++) {
191
192
193
      riq[iq] = 0.0;
    }

Thomas Witkowski's avatar
Thomas Witkowski committed
194
    for (int system = 0; system < nSystems; system++) {
195
196
197
198
199

      if (matrix[system] == NULL) 
	continue;

      // init assemblers
200
201
      for (it = const_cast<DOFMatrix*>(matrix[system])->getOperatorsBegin(),
	   itfac = const_cast<DOFMatrix*>(matrix[system])->getOperatorEstFactorBegin();
202
	   it != const_cast<DOFMatrix*>(matrix[system])->getOperatorsEnd(); 
203
	   ++it, ++itfac) {
204
	if (*itfac == NULL || **itfac != 0.0) {
205
206
	  (*it)->getAssembler(omp_get_thread_num())->initElement(elInfo, NULL, quad);
	}
207
208
      }

Thomas Witkowski's avatar
Thomas Witkowski committed
209
210
211
212
      if (C0) {
	for (it = const_cast<DOFVector<double>*>(fh[system])->getOperatorsBegin();
	     it != const_cast<DOFVector<double>*>(fh[system])->getOperatorsEnd(); 
	     ++it) {
213
	  (*it)->getAssembler(omp_get_thread_num())->initElement(elInfo, NULL, quad);	
Thomas Witkowski's avatar
Thomas Witkowski committed
214
	}
215
      }
Naumann, Andreas's avatar
Naumann, Andreas committed
216
	
217
      if (timestep && uhOld[system]) {
218
219
220
221
222
223
224
225
	TEST_EXIT_DBG(uhOld[system])("no uhOld\n");
	uhOld[system]->getLocalVector(el, uhOldEl[system]);
  
	// ===== time and element residuals       
	if (C0 || C3) {   
	  uh[system]->getVecAtQPs(elInfo, NULL, quadFast[system], uhQP);
	  uhOld[system]->getVecAtQPs(elInfo, NULL, quadFast[system], uhOldQP);
	  
226
227
	  if (C3 && uhOldQP && system == std::max(row, 0)) {
	    val = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
228
	    for (int iq = 0; iq < nPoints; iq++) {
229
230
231
232
233
234
235
236
237
	      double tiq = (uhQP[iq] - uhOldQP[iq]);
	      val += quad->getWeight(iq) * tiq * tiq;
	    }
	    double v = C3 * det * val;
	    est_t_sum += v;
	    est_t_max = max(est_t_max, v);
	  }
	}
      }
238
           
239
      if (C0) {  
240
241
	for (it = const_cast<DOFMatrix*>(matrix[system])->getOperatorsBegin(),
	     itfac = const_cast<DOFMatrix*>(matrix[system])->getOperatorEstFactorBegin(); 
242
	     it != const_cast<DOFMatrix*>(matrix[system])->getOperatorsEnd(); 
243
	     ++it, ++itfac) {
244
	  if (*itfac == NULL || **itfac != 0.0) {
245
	    if ((uhQP == NULL) && (*it)->zeroOrderTerms()) {
246
	      uhQP = new double[nPoints];
247
248
249
250
251
252
253
254
255
256
	      uh[system]->getVecAtQPs(elInfo, NULL, quadFast[system], uhQP);
	    }
	    if ((grdUh_qp == NULL) && ((*it)->firstOrderTermsGrdPsi() || (*it)->firstOrderTermsGrdPhi())) {
	      grdUh_qp = new WorldVector<double>[nPoints];
	      uh[system]->getGrdAtQPs(elInfo, NULL, quadFast[system], grdUh_qp);
	    }
	    if ((D2uhqp == NULL) && (degree > 2) && (*it)->secondOrderTerms()) { 
	      D2uhqp = new WorldMatrix<double>[nPoints];
	      uh[system]->getD2AtQPs(elInfo, NULL, quadFast[system], D2uhqp);	    
	    }
257
258
259
260
	  }
	}
	
	r(elInfo,
Thomas Witkowski's avatar
Thomas Witkowski committed
261
	  nPoints, 
262
263
264
265
266
267
268
269
270
	  uhQP,
	  grdUh_qp,
	  D2uhqp,
	  uhOldQP,
	  NULL,  // grdUhOldQP 
	  NULL,  // D2UhOldQP
	  matrix[system], 
	  fh[system],
	  quad,
271
	  riq);
272
273
274
275
      }     
    }

    // add integral over r square
276
    val = 0.0;
277
    for (int iq = 0; iq < nPoints; iq++)
278
      val += quad->getWeight(iq) * riq[iq] * riq[iq];
279
   
280
    if (timestep != 0.0 || norm == NO_NORM || norm == L2_NORM)
281
      val = C0 * h2 * h2 * det * val;
282
    else
283
      val = C0 * h2 * det * val;
284
    
285
286
287
288
    est_el += val;

    // ===== jump residuals 
    if (C1 && (dim > 1)) {
289
      int dow = Global::getGeo(WORLD);
290

291
      for (int face = 0; face < neighbours_; face++) {  
Thomas Witkowski's avatar
Thomas Witkowski committed
292
	Element *neigh = const_cast<Element*>(elInfo->getNeighbour(face));
293
294
295
	if (neigh && neigh->getMark()) {      
	  int oppV = elInfo->getOppVertex(face);
	      
296
297
	  el->sortFaceIndices(face, &faceIndEl_);
	  neigh->sortFaceIndices(oppV, &faceIndNeigh_);
298
299
300
	    
	  neighInfo->setElement(const_cast<Element*>(neigh));
	  neighInfo->setFillFlag(Mesh::FILL_COORDS);
301
	      	
302
303
304
305
306
307
308
309
310
	  for (int i = 0; i < dow; i++)
	    neighInfo->getCoord(oppV)[i] = elInfo->getOppCoord(face)[i];
		
	  // periodic leaf data ?
	  ElementData *ldp = el->getElementData()->getElementData(PERIODIC);

	  bool periodicCoords = false;

	  if (ldp) {
311
312
	    std::list<LeafDataPeriodic::PeriodicInfo>::iterator it;
	    std::list<LeafDataPeriodic::PeriodicInfo>& infoList = 
313
		dynamic_cast<LeafDataPeriodic*>(ldp)->getInfoList();
314

315
	    for (it = infoList.begin(); it != infoList.end(); ++it) {
316
317
	      if (it->elementSide == face) {
		for (int i = 0; i < dim; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
318
319
		  int i1 = faceIndEl_[i];
		  int i2 = faceIndNeigh_[i];
320

Thomas Witkowski's avatar
Thomas Witkowski committed
321
322
		  int j = 0;
		  for (; j < dim; j++) {
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
		    if (i1 == el->getVertexOfPosition(INDEX_OF_DIM(dim - 1, 
								   dim),
						      face,
						      j)) {
		      break;
		    }
		  }

		  TEST_EXIT_DBG(j != dim)("vertex i1 not on face ???\n");
		      
		  neighInfo->getCoord(i2) = (*(it->periodicCoords))[j];
		}
		periodicCoords = true;
		break;
	      }
	    }
	  }
      
	  if (!periodicCoords) {
	    for (int i = 0; i < dim; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
343
344
345
	      int i1 = faceIndEl_[i];
	      int i2 = faceIndNeigh_[i];
	      for (int j = 0; j < dow; j++)
346
347
348
349
350
351
352
353
354
		neighInfo->getCoord(i2)[j] = elInfo->getCoord(i1)[j];
	    }
	  }
	      
	  Parametric *parametric = mesh->getParametric();
	  if (parametric) {
	    neighInfo = parametric->addParametricInfo(neighInfo);
	  }
	      
355
	  double detNeigh = abs(neighInfo->calcGrdLambda(*lambdaNeigh_));
356
	      
357
	  for (int iq = 0; iq < nPointsSurface_; iq++) {
358
	    jump_[iq].set(0.0);
359
360
361
	  }
	     

Thomas Witkowski's avatar
Thomas Witkowski committed
362
	  for (int system = 0; system < nSystems; system++) {	
363
364
365
	    if (matrix[system] == NULL) 
	      continue;
	      
366
367
	    uh[system]->getLocalVector(el, uhEl[system]);	
	    uh[system]->getLocalVector(neigh, uhNeigh[system]);
368
			
369
	    for (int iq = 0; iq < nPointsSurface_; iq++) {
370
	      (*lambda_)[face] = 0.0;
371
	      for (int i = 0; i < dim; i++) {
372
		(*lambda_)[faceIndEl_[i]] = surfaceQuad_->getLambda(iq, i);
373
374
	      }
		  
375
376
	      basFcts[system]->evalGrdUh(*lambda_, 
					 grdLambda, 
377
					 uhEl[system], 
378
					 &grdUhEl_[iq]);
379
		  
380
	      (*lambda_)[oppV] = 0.0;
381
	      for (int i = 0; i < dim; i++) {
382
		(*lambda_)[faceIndNeigh_[i]] = surfaceQuad_->getLambda(iq, i);
383
384
	      }
		  
385
386
	      basFcts[system]->evalGrdUh(*lambda_, 
					 *lambdaNeigh_, 
387
388
					 uhNeigh[system], 
					 &grdUhNeigh_[iq]);
389
		  
390
	      grdUhEl_[iq] -= grdUhNeigh_[iq];
391
392
	    }				

393
	    std::vector<double*>::iterator fac;
394
395
396
397
398

	    for (it = const_cast<DOFMatrix*>(matrix[system])->getOperatorsBegin(),
		   fac = const_cast<DOFMatrix*>(matrix[system])->getOperatorEstFactorBegin(); 
		 it != const_cast<DOFMatrix*>(matrix[system])->getOperatorsEnd(); 
		 ++it, ++fac) {
399

400
	      if (*fac == NULL || **fac != 0.0) {
401
402
403
404
405
406
407
408
409
410
411
412
413
414
		for (int iq = 0; iq < nPointsSurface_; iq++) {
		  localJump_[iq].set(0.0);
		}
		
		(*it)->weakEvalSecondOrder(nPointsSurface_,
					   grdUhEl_.getValArray(),
					   localJump_.getValArray());
		double factor = *fac ? **fac : 1.0;
		if (factor != 1.0) {
		  for (int i = 0; i < nPointsSurface_; i++) {
		    localJump_[i] *= factor;
		  }
		}
		
415
		for (int i = 0; i < nPointsSurface_; i++) {
416
		  jump_[i] += localJump_[i];
417
		}
418
419
	      }		
	    }
420
421
	  }
	      
422
	  val = 0.0;
423
	  for (int iq = 0; iq < nPointsSurface_; iq++) {
424
	    val += surfaceQuad_->getWeight(iq) * (jump_[iq] * jump_[iq]);
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
	  }
	      
	  double d = 0.5 * (det + detNeigh);

	  if (norm == NO_NORM || norm == L2_NORM)
	    val *= C1 * h2_from_det(d, dim) * d;
	  else
	    val *= C1 * d;
	      
	  if (parametric) {
	    neighInfo = parametric->removeParametricInfo(neighInfo);
	  }

	  neigh->setEstimation(neigh->getEstimation(row) + val, row);
	  est_el += val;
	} 
      } 
       
443
      val = fh[std::max(row, 0)]->
444
	getBoundaryManager()->
445
	boundResidual(elInfo, matrix[std::max(row, 0)], uh[std::max(row, 0)]);
446
447
448
449
450
451
452
453
454
455
456
457
458
459
      if (norm == NO_NORM || norm == L2_NORM)
	val *= C1 * h2;
      else
	val *= C1;
	
      est_el += val;
    } 
  

    el->setEstimation(est_el, row);

    est_sum += est_el;
    est_max = max(est_max, est_el);

460
    elInfo->getElement()->setMark(0);  
461
462
  }

463
  void r(const ElInfo *elInfo,
Thomas Witkowski's avatar
Thomas Witkowski committed
464
	 int nPoints,
465
	 const double *uhIq,
466
467
	 const WorldVector<double> *grdUhIq,
	 const WorldMatrix<double> *D2UhIq,
468
	 const double *uhOldIq,
469
470
471
472
473
474
475
	 const WorldVector<double> *grdUhOldIq,
	 const WorldMatrix<double> *D2UhOldIq,
	 DOFMatrix *A, 
	 DOFVector<double> *fh,
	 Quadrature *quad,
	 double *result)
  {
476
477
    std::vector<Operator*>::iterator it;
    std::vector<double*>::iterator fac;
478
479
480
481
482
483

    // lhs
    for (it = const_cast<DOFMatrix*>(A)->getOperatorsBegin(),
	   fac = const_cast<DOFMatrix*>(A)->getOperatorEstFactorBegin(); 
	 it != const_cast<DOFMatrix*>(A)->getOperatorsEnd(); 
	 ++it, ++fac) {
484
485
486
     
      double factor = *fac ? **fac : 1.0;

487
488
      if (factor) {
	if (D2UhIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
489
	  (*it)->evalSecondOrder(nPoints, uhIq, grdUhIq, D2UhIq, result, -factor);
490
491
492
	}

	if (grdUhIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
493
494
	  (*it)->evalFirstOrderGrdPsi(nPoints, uhIq, grdUhIq, D2UhIq, result, factor);
	  (*it)->evalFirstOrderGrdPhi(nPoints, uhIq, grdUhIq, D2UhIq, result, factor);
495
496
497
	}
	
	if (uhIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
498
	  (*it)->evalZeroOrder(nPoints, uhIq, grdUhIq, D2UhIq, result, factor);
499
500
501
502
503
504
505
506
507
508
	}
      }
    }
    
    // rhs
    for (it = const_cast<DOFVector<double>*>(fh)->getOperatorsBegin(),
	 fac = const_cast<DOFVector<double>*>(fh)->getOperatorEstFactorBegin(); 
	 it != const_cast<DOFVector<double>*>(fh)->getOperatorsEnd(); 
	 ++it, ++fac) {

509
510
      double factor = *fac ? **fac : 1.0;

511
512
513
      if (factor) {
	if ((*it)->getUhOld()) {
	  if (D2UhOldIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
514
	    (*it)->evalSecondOrder(nPoints, 
515
516
517
518
				   uhOldIq, grdUhOldIq, D2UhOldIq, 
				   result, factor);
	  }
	  if (grdUhOldIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
519
	    (*it)->evalFirstOrderGrdPsi(nPoints, 
520
521
					uhOldIq, grdUhOldIq, D2UhOldIq, 
					result, -factor);
Thomas Witkowski's avatar
Thomas Witkowski committed
522
	    (*it)->evalFirstOrderGrdPhi(nPoints, 
523
524
525
526
					uhOldIq, grdUhOldIq, D2UhOldIq, 
					result, -factor);
	  }
	  if (uhOldIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
527
	    (*it)->evalZeroOrder(nPoints, 
528
529
530
531
				 uhOldIq, grdUhOldIq, D2UhOldIq, 
				 result, -factor);
	  }
	} else {
532
	  std::vector<double> fx(nPoints, 0.0);
Thomas Witkowski's avatar
Thomas Witkowski committed
533
	  (*it)->getC(elInfo, nPoints, fx);
534

535
	  for (int iq = 0; iq < nPoints; iq++)
536
537
538
539
540
541
542
543
	    result[iq] -= factor * fx[iq];
	}
      }
    }    
  }


}