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

9
10
11
12
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
#include "mpi.h"
#endif

13
14
namespace AMDiS {

15
  ResidualEstimator::ResidualEstimator(std::string name, int r) 
16
17
18
    : Estimator(name, r),
      C0(1.0), 
      C1(1.0), 
19
      C2(0.0), 
20
21
      C3(1.0)
  {
22
23
    FUNCNAME("ResidualEstimator::ResidualEstimator()");

24
25
26
27
28
29
30
31
32
    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;
33
34

    TEST_EXIT(C2 == 0.0)("C2 is not used! Please remove it or set it to 0.0!\n");
35
36
37
38
39
  }

  void ResidualEstimator::init(double ts)
  {
    FUNCNAME("ResidualEstimator::init()");
Thomas Witkowski's avatar
Thomas Witkowski committed
40
    
41
42
43
44
    timestep = ts;

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

Thomas Witkowski's avatar
Thomas Witkowski committed
45
46
    nSystems = static_cast<int>(uh.size());
    TEST_EXIT_DBG(nSystems > 0)("no system set\n");
47
48

    dim = mesh->getDim();
49
50
    basFcts = new const BasisFunction*[nSystems];
    quadFast = new FastQuadrature*[nSystems];
51
52

    degree = 0;
Thomas Witkowski's avatar
Thomas Witkowski committed
53
    for (int system = 0; system < nSystems; system++) {
54
      basFcts[system] = uh[system]->getFESpace()->getBasisFcts();
55
      degree = std::max(degree, basFcts[system]->getDegree());
56
57
58
59
    }
    degree *= 2;

    quad = Quadrature::provideQuadrature(dim, degree);
Thomas Witkowski's avatar
Thomas Witkowski committed
60
    nPoints = quad->getNumPoints();
61
62

    Flag flag = INIT_PHI | INIT_GRD_PHI;
63
64
    if (degree > 2)
      flag |= INIT_D2_PHI;    
65

66
    for (int system = 0; system < nSystems; system++)
67
68
      quadFast[system] = FastQuadrature::provideFastQuadrature(basFcts[system], 
							       *quad, 
69
							       flag);    
70
  
71
72
73
    uhEl = new double*[nSystems];
    uhNeigh = new double*[nSystems];
    uhOldEl = timestep ? new double*[nSystems] : NULL;
74

Thomas Witkowski's avatar
Thomas Witkowski committed
75
    for (int system = 0; system < nSystems; system++) {
76
77
      uhEl[system] = new double[basFcts[system]->getNumber()]; 
      uhNeigh[system] = new double[basFcts[system]->getNumber()];
78
      if (timestep)
79
	uhOldEl[system] = new double[basFcts[system]->getNumber()];
80
81
    }

82
83
84
    uhQP = timestep ? new double[nPoints] : NULL;
    uhOldQP = timestep ? new double[nPoints] : NULL;
    riq = new double[nPoints];
85
86
87
    grdUh_qp = NULL;
    D2uhqp = NULL;

88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
    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;
112
113
114
115

    neighInfo = mesh->createNewElInfo();

    // prepare date for computing jump residual
116
117
118
119
120
121
122
123
124
125
    if (C1 > 0.0 && dim > 1) {
      surfaceQuad = Quadrature::provideQuadrature(dim - 1, degree);
      nPointsSurface = surfaceQuad->getNumPoints();
      grdUhEl.resize(nPointsSurface);
      grdUhNeigh.resize(nPointsSurface);
      jump.resize(nPointsSurface);
      localJump.resize(nPointsSurface);
      nNeighbours = Global::getGeo(NEIGH, dim);
      lambdaNeigh = new DimVec<WorldVector<double> >(dim, NO_INIT);
      lambda = new DimVec<double>(dim, NO_INIT);
126
    }
127
128
129
130
131
132
  }

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

133
134
135
136
137
138
139
140
141
142
143
144
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double send_est_sum = est_sum;
    double send_est_max = est_max;
    double send_est_t_sum = est_t_sum;
    double send_est_t_max = est_t_max;

    MPI::COMM_WORLD.Allreduce(&send_est_sum, &est_sum, 1, MPI_DOUBLE, MPI_SUM);
    MPI::COMM_WORLD.Allreduce(&send_est_max, &est_max, 1, MPI_DOUBLE, MPI_MAX);
    MPI::COMM_WORLD.Allreduce(&send_est_t_sum, &est_t_sum, 1, MPI_DOUBLE, MPI_SUM);
    MPI::COMM_WORLD.Allreduce(&send_est_t_max, &est_t_max, 1, MPI_DOUBLE, MPI_MAX);
#endif

145
146
147
    est_sum = sqrt(est_sum);
    est_t_sum = sqrt(est_t_sum);

Thomas Witkowski's avatar
Thomas Witkowski committed
148
    for (int system = 0; system < nSystems; system++) {
149
150
      delete [] uhEl[system];
      delete [] uhNeigh[system];
151
      if (timestep)
152
	delete [] uhOldEl[system];
153
154
    }

155
156
    delete [] uhEl;
    delete [] uhNeigh;
157
158

    if (timestep) {
159
160
161
      delete [] uhOldEl;
      delete [] uhQP;
      delete [] uhOldQP;
162
    } else {
163
164
      if (uhQP != NULL)
	delete [] uhQP;
165
166
167
168
    }

    if (output) {
      MSG("estimate   = %.8e\n", est_sum);
169
      if (C3)
170
171
172
	MSG("time estimate   = %.8e\n", est_t_sum);
    }

173
174
175
    delete [] riq;
    delete [] basFcts;
    delete [] quadFast;
176

177
178
179
180
    if (grdUh_qp != NULL)
      delete [] grdUh_qp;
    if (D2uhqp != NULL)
      delete [] D2uhqp;
181

182
    if (C1 && (dim > 1)) {
183
184
      delete lambdaNeigh;
      delete lambda;
185
186
    }

Thomas Witkowski's avatar
Thomas Witkowski committed
187
    delete neighInfo;
188
189
190
  }

  void ResidualEstimator::estimateElement(ElInfo *elInfo)
191
  {    
192
193
    FUNCNAME("ResidualEstimator::estimateElement()");

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

Thomas Witkowski's avatar
Thomas Witkowski committed
196
    double val = 0.0;
197
    std::vector<Operator*>::iterator it;
198
    std::vector<double*>::iterator itfac;
Thomas Witkowski's avatar
Thomas Witkowski committed
199
    Element *el = elInfo->getElement();
200
    double det = elInfo->getDet();
201
    const DimVec<WorldVector<double> > &grdLambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
202
    double est_el = el->getEstimation(row);
203
204
    double h2 = h2_from_det(det, dim);

205
    for (int iq = 0; iq < nPoints; iq++)
206
207
      riq[iq] = 0.0;

Thomas Witkowski's avatar
Thomas Witkowski committed
208
    for (int system = 0; system < nSystems; system++) {
209
210
211
212

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

213
214
      // === init assemblers ===

215
216
      for (it = const_cast<DOFMatrix*>(matrix[system])->getOperatorsBegin(),
	   itfac = const_cast<DOFMatrix*>(matrix[system])->getOperatorEstFactorBegin();
217
	   it != const_cast<DOFMatrix*>(matrix[system])->getOperatorsEnd(); 
218
219
	   ++it, ++itfac)
	if (*itfac == NULL || **itfac != 0.0)
220
	  (*it)->getAssembler(omp_get_thread_num())->initElement(elInfo, NULL, quad);
221

222
      if (C0 > 0.0)
Thomas Witkowski's avatar
Thomas Witkowski committed
223
224
	for (it = const_cast<DOFVector<double>*>(fh[system])->getOperatorsBegin();
	     it != const_cast<DOFVector<double>*>(fh[system])->getOperatorsEnd(); 
225
	     ++it)
226
	  (*it)->getAssembler(omp_get_thread_num())->initElement(elInfo, NULL, quad);	
Naumann, Andreas's avatar
Naumann, Andreas committed
227
	
228
      if (timestep && uhOld[system]) {
229
230
231
	TEST_EXIT_DBG(uhOld[system])("no uhOld\n");
	uhOld[system]->getLocalVector(el, uhOldEl[system]);
  
232
233
234
	// === Compute time error. ===

	if (C0 > 0.0 || C3 > 0.0) {   
235
236
237
	  uh[system]->getVecAtQPs(elInfo, NULL, quadFast[system], uhQP);
	  uhOld[system]->getVecAtQPs(elInfo, NULL, quadFast[system], uhOldQP);
	  
238
	  if (C3 > 0.0 && uhOldQP && system == std::max(row, 0)) {
239
	    val = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
240
	    for (int iq = 0; iq < nPoints; iq++) {
241
242
243
244
245
246
247
248
249
	      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);
	  }
	}
      }
250
           
251
252
253
      // === Compute element residual. ===

      if (C0 > 0.0) {  
254
255
	for (it = const_cast<DOFMatrix*>(matrix[system])->getOperatorsBegin(),
	     itfac = const_cast<DOFMatrix*>(matrix[system])->getOperatorEstFactorBegin(); 
256
	     it != const_cast<DOFMatrix*>(matrix[system])->getOperatorsEnd(); 
257
	     ++it, ++itfac) {
258
	  if (*itfac == NULL || **itfac != 0.0) {
259
	    if (uhQP == NULL && (*it)->zeroOrderTerms()) {
260
	      uhQP = new double[nPoints];
261
262
	      uh[system]->getVecAtQPs(elInfo, NULL, quadFast[system], uhQP);
	    }
263
264
	    if (grdUh_qp == NULL && 
		((*it)->firstOrderTermsGrdPsi() || (*it)->firstOrderTermsGrdPhi())) {
265
266
267
	      grdUh_qp = new WorldVector<double>[nPoints];
	      uh[system]->getGrdAtQPs(elInfo, NULL, quadFast[system], grdUh_qp);
	    }
268
	    if (D2uhqp == NULL && degree > 2 && (*it)->secondOrderTerms()) { 
269
270
271
	      D2uhqp = new WorldMatrix<double>[nPoints];
	      uh[system]->getD2AtQPs(elInfo, NULL, quadFast[system], D2uhqp);	    
	    }
272
273
274
	  }
	}
	
275
276
	// === Compute the element residual and store it in irq. ===

277
	r(elInfo,
Thomas Witkowski's avatar
Thomas Witkowski committed
278
	  nPoints, 
279
280
281
282
283
284
285
286
287
	  uhQP,
	  grdUh_qp,
	  D2uhqp,
	  uhOldQP,
	  NULL,  // grdUhOldQP 
	  NULL,  // D2UhOldQP
	  matrix[system], 
	  fh[system],
	  quad,
288
	  riq);
289
290
291
292
      }     
    }

    // add integral over r square
293
    val = 0.0;
294
    for (int iq = 0; iq < nPoints; iq++)
295
      val += quad->getWeight(iq) * riq[iq] * riq[iq];
296
   
297
    if (timestep != 0.0 || norm == NO_NORM || norm == L2_NORM)
298
      val = C0 * h2 * h2 * det * val;
299
    else
300
      val = C0 * h2 * det * val;
301
    
302
303
    est_el += val;

304
305
306

    // === Compute jump residuals. ===

307
    if (C1 && (dim > 1)) {
308
      int dow = Global::getGeo(WORLD);
309

310
      for (int face = 0; face < nNeighbours; face++) {  
Thomas Witkowski's avatar
Thomas Witkowski committed
311
	Element *neigh = const_cast<Element*>(elInfo->getNeighbour(face));
312
313
314
	if (neigh && neigh->getMark()) {      
	  int oppV = elInfo->getOppVertex(face);
	      
315
316
	  el->sortFaceIndices(face, &faceIndEl);
	  neigh->sortFaceIndices(oppV, &faceIndNeigh);
317
318
319
	    
	  neighInfo->setElement(const_cast<Element*>(neigh));
	  neighInfo->setFillFlag(Mesh::FILL_COORDS);
320
	      	
321
322
323
324
325
326
327
328
329
	  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) {
330
331
	    std::list<LeafDataPeriodic::PeriodicInfo>::iterator it;
	    std::list<LeafDataPeriodic::PeriodicInfo>& infoList = 
332
		dynamic_cast<LeafDataPeriodic*>(ldp)->getInfoList();
333

334
	    for (it = infoList.begin(); it != infoList.end(); ++it) {
335
336
	      if (it->elementSide == face) {
		for (int i = 0; i < dim; i++) {
337
338
		  int i1 = faceIndEl[i];
		  int i2 = faceIndNeigh[i];
339

Thomas Witkowski's avatar
Thomas Witkowski committed
340
341
		  int j = 0;
		  for (; j < dim; j++) {
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
		    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++) {
362
363
	      int i1 = faceIndEl[i];
	      int i2 = faceIndNeigh[i];
Thomas Witkowski's avatar
Thomas Witkowski committed
364
	      for (int j = 0; j < dow; j++)
365
366
367
368
369
		neighInfo->getCoord(i2)[j] = elInfo->getCoord(i1)[j];
	    }
	  }
	      
	  Parametric *parametric = mesh->getParametric();
370
371
	  if (parametric)
	    neighInfo = parametric->addParametricInfo(neighInfo);	  
372
	      
373
	  double detNeigh = abs(neighInfo->calcGrdLambda(*lambdaNeigh));
374
	      
375
376
	  for (int iq = 0; iq < nPointsSurface; iq++)
	    jump[iq].set(0.0);     
377

Thomas Witkowski's avatar
Thomas Witkowski committed
378
	  for (int system = 0; system < nSystems; system++) {	
379
380
381
	    if (matrix[system] == NULL) 
	      continue;
	      
382
383
	    uh[system]->getLocalVector(el, uhEl[system]);	
	    uh[system]->getLocalVector(neigh, uhNeigh[system]);
384
			
385
386
387
388
	    for (int iq = 0; iq < nPointsSurface; iq++) {
	      (*lambda)[face] = 0.0;
	      for (int i = 0; i < dim; i++)
		(*lambda)[faceIndEl[i]] = surfaceQuad->getLambda(iq, i);
389
		  
390
	      basFcts[system]->evalGrdUh(*lambda, 
391
					 grdLambda, 
392
					 uhEl[system], 
393
					 &grdUhEl[iq]);
394
		  
395
396
397
	      (*lambda)[oppV] = 0.0;
	      for (int i = 0; i < dim; i++)
		(*lambda)[faceIndNeigh[i]] = surfaceQuad->getLambda(iq, i);
398
		  
399
400
	      basFcts[system]->evalGrdUh(*lambda, 
					 *lambdaNeigh, 
401
					 uhNeigh[system], 
402
					 &grdUhNeigh[iq]);
403
		  
404
	      grdUhEl[iq] -= grdUhNeigh[iq];
405
406
	    }				

407
	    std::vector<double*>::iterator fac;
408
409
410
411
412

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

414
	      if (*fac == NULL || **fac != 0.0) {
415
416
		for (int iq = 0; iq < nPointsSurface; iq++)
		  localJump[iq].set(0.0);
417
		
418
419
420
		(*it)->weakEvalSecondOrder(nPointsSurface,
					   grdUhEl.getValArray(),
					   localJump.getValArray());
421
		double factor = *fac ? **fac : 1.0;
422
423
424
		if (factor != 1.0)
		  for (int i = 0; i < nPointsSurface; i++)
		    localJump[i] *= factor;
425
		
426
427
		for (int i = 0; i < nPointsSurface; i++)
		  jump[i] += localJump[i];
428
429
	      }		
	    }
430
431
	  }
	      
432
	  val = 0.0;
433
434
	  for (int iq = 0; iq < nPointsSurface; iq++)
	    val += surfaceQuad->getWeight(iq) * (jump[iq] * jump[iq]);
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;
	      
443
	  if (parametric)
444
445
446
447
448
449
450
	    neighInfo = parametric->removeParametricInfo(neighInfo);

	  neigh->setEstimation(neigh->getEstimation(row) + val, row);
	  est_el += val;
	} 
      } 
       
451
      val = fh[std::max(row, 0)]->
452
	getBoundaryManager()->
453
	boundResidual(elInfo, matrix[std::max(row, 0)], uh[std::max(row, 0)]);
454
455
456
457
458
459
460
461
462
463
464
465
466
467
      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);

468
    elInfo->getElement()->setMark(0);  
469
470
  }

471
  void r(const ElInfo *elInfo,
Thomas Witkowski's avatar
Thomas Witkowski committed
472
	 int nPoints,
473
	 const double *uhIq,
474
475
	 const WorldVector<double> *grdUhIq,
	 const WorldMatrix<double> *D2UhIq,
476
	 const double *uhOldIq,
477
478
479
480
481
482
483
	 const WorldVector<double> *grdUhOldIq,
	 const WorldMatrix<double> *D2UhOldIq,
	 DOFMatrix *A, 
	 DOFVector<double> *fh,
	 Quadrature *quad,
	 double *result)
  {
484
485
    std::vector<Operator*>::iterator it;
    std::vector<double*>::iterator fac;
486
487
488
489
490
491

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

495
496
      if (factor) {
	if (D2UhIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
497
	  (*it)->evalSecondOrder(nPoints, uhIq, grdUhIq, D2UhIq, result, -factor);
498
499
500
	}

	if (grdUhIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
501
502
	  (*it)->evalFirstOrderGrdPsi(nPoints, uhIq, grdUhIq, D2UhIq, result, factor);
	  (*it)->evalFirstOrderGrdPhi(nPoints, uhIq, grdUhIq, D2UhIq, result, factor);
503
504
505
	}
	
	if (uhIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
506
	  (*it)->evalZeroOrder(nPoints, uhIq, grdUhIq, D2UhIq, result, factor);
507
508
509
510
511
512
513
514
515
516
	}
      }
    }
    
    // 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) {

517
518
      double factor = *fac ? **fac : 1.0;

519
520
521
      if (factor) {
	if ((*it)->getUhOld()) {
	  if (D2UhOldIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
522
	    (*it)->evalSecondOrder(nPoints, 
523
524
525
526
				   uhOldIq, grdUhOldIq, D2UhOldIq, 
				   result, factor);
	  }
	  if (grdUhOldIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
527
	    (*it)->evalFirstOrderGrdPsi(nPoints, 
528
529
					uhOldIq, grdUhOldIq, D2UhOldIq, 
					result, -factor);
Thomas Witkowski's avatar
Thomas Witkowski committed
530
	    (*it)->evalFirstOrderGrdPhi(nPoints, 
531
532
533
534
					uhOldIq, grdUhOldIq, D2UhOldIq, 
					result, -factor);
	  }
	  if (uhOldIq) {
Thomas Witkowski's avatar
Thomas Witkowski committed
535
	    (*it)->evalZeroOrder(nPoints, 
536
537
538
539
				 uhOldIq, grdUhOldIq, D2UhOldIq, 
				 result, -factor);
	  }
	} else {
540
	  std::vector<double> fx(nPoints, 0.0);
Thomas Witkowski's avatar
Thomas Witkowski committed
541
	  (*it)->getC(elInfo, nPoints, fx);
542

543
	  for (int iq = 0; iq < nPoints; iq++)
544
545
546
547
548
549
550
551
	    result[iq] -= factor * fx[iq];
	}
      }
    }    
  }


}