Operator.cc 58.5 KB
Newer Older
1
2
3
4
5
6
7
8
#include "Operator.h"
#include "ElInfo.h"
#include "Assembler.h"
#include "FixVec.h"
#include "DOFVector.h"
#include "ElementMatrix.h"
#include "ElementVector.h"
#include "Quadrature.h"
9
#include "OpenMP.h"
10
11
12
13
14
15
16
17
18
19
20
21

namespace AMDiS {

  const Flag OperatorTerm::PW_CONST = 1;
  const Flag OperatorTerm::SYMMETRIC = 2;

  const Flag Operator::MATRIX_OPERATOR = 1;
  const Flag Operator::VECTOR_OPERATOR = 2;


  int Operator::getQuadratureDegree(int order, FirstOrderType firstOrderType) 
  {
22
    std::vector<OperatorTerm*>* terms = NULL;
23
    int myRank = omp_get_thread_num();
24
25
26

    switch(order) {
    case 0:
27
      terms = &zeroOrder[myRank];
28
29
30
      break;
    case 1:
      if (firstOrderType == GRD_PHI)
31
	terms = &firstOrderGrdPhi[myRank];
32
      else 
33
	terms = &firstOrderGrdPsi[myRank];
34
35
      break;
    case 2:
36
      terms = &secondOrder[myRank];
37
38
39
40
41
42
43
44
45
46
      break;
    }

    const BasisFunction *psi = rowFESpace->getBasisFcts();
    const BasisFunction *phi = colFESpace->getBasisFcts();

    int psiDegree = psi->getDegree();
    int phiDegree = phi->getDegree();
    int maxTermDegree = 0;

47
    for (int i = 0; i < static_cast<int>( terms->size()); i++) {
48
49
50
      OperatorTerm *term = (*terms)[i];
      maxTermDegree = max(maxTermDegree, term->degree);
    }
51
52
   
    return psiDegree + phiDegree - order + maxTermDegree;
53
54
55
56
57
  }


  void OperatorTerm::setSymmetric(bool symm)
  {
58
    if (symm) {
59
60
61
62
63
64
65
66
67
68
69
70
      properties.setFlag(SYMMETRIC);
    } else {
      properties.unsetFlag(SYMMETRIC);   
    }
  }

  bool OperatorTerm::isSymmetric()
  {
    return properties.isSet(SYMMETRIC);
  }

  void OperatorTerm::lalt(const DimVec<WorldVector<double> >& Lambda,
71
72
73
74
			  const WorldMatrix<double>& matrix,
			  DimMat<double>& LALt,
			  bool symm,
			  double factor)
75
  {
76
77
78
    int i, j, k, l;
    static const int dimOfWorld = Global::getGeo(WORLD);
    int dim = LALt.getNumRows() - 1;
79
80
81
  
    double val = 0.0;

82
    if (symm) {
83
      for (i = 0; i <= dim; i++) {
84
85
86
	val = 0.0;
	for (k = 0; k < dimOfWorld; k++)
	  for (l = 0; l < dimOfWorld; l++)
87
88
89
	    val += Lambda[i][k] * matrix[k][l] * Lambda[i][l];
	val *= factor;
	LALt[i][i] += val;
90
	for (j = i + 1; j <= dim; j++) {
91
92
93
	  val = 0.0;
	  for (k = 0; k < dimOfWorld; k++)
	    for (l = 0; l < dimOfWorld; l++)
94
95
96
97
98
99
100
101
102
	      val += Lambda[i][k] * matrix[k][l] * Lambda[j][l];
	  val *= factor;
	  LALt[i][j] += val;
	  LALt[j][i] += val;
	}
      }    
    } else {	
      for (i = 0; i <= dim; i++) {
	for (j = 0; j <= dim; j++) {
103
104
105
	  val = 0.0;
	  for (k = 0; k < dimOfWorld; k++)
	    for (l = 0; l < dimOfWorld; l++) 
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
	      val += Lambda[i][k] * matrix[k][l] * Lambda[j][l];
	  val *= factor;
	  LALt[i][j] += val;
	}
      }    
    }      
  }

  void OperatorTerm::lalt_kl(const DimVec<WorldVector<double> >& Lambda,
			     int k, int l,
			     DimMat<double>&            LALt,
			     double                     factor)
  {
    int i, j;
    int dim = LALt.getNumRows() - 1;

    for (i = 0; i <= dim; i++)
      for (j = 0; j <= dim; j++)
	LALt[i][j] += factor * Lambda[i][k] * Lambda[j][l];
  }

  void OperatorTerm::l1lt(const DimVec<WorldVector<double> >& Lambda,
128
129
			  DimMat<double>& LALt,
			  double factor)
130
  {
131
132
    static const int dimOfWorld = Global::getGeo(WORLD);
    int dim = LALt.getNumRows() - 1;
133
134
135
    
    double val = 0.0;

136
137
138
    for (int i = 0; i <= dim; i++) {
      val = 0.0;
      for (int k = 0; k < dimOfWorld; k++)
139
140
141
	val += Lambda[i][k] * Lambda[i][k];
      val *= factor;
      LALt[i][i] += val;
142
143
144
      for (int j = i + 1; j <= dim; j++) {
	val = 0.0;
	for (int k = 0; k < dimOfWorld; k++)
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
	  val += Lambda[i][k] * Lambda[j][k];
	val *= factor;
	LALt[i][j] += val;
	LALt[j][i] += val;
      }    
    }      
  }

  void OperatorTerm::lb(const DimVec<WorldVector<double> >& Lambda,
			const WorldVector<double>&         b,
			DimVec<double>&            Lb,
			double                     factor)
  {
    int                  i, j;
    int                  dim        = Lb.getSize() - 1;
    static const int     dimOfWorld = Global::getGeo(WORLD);

    double val = 0.0;

164
165
    for (i = 0; i <= dim; i++) {
      for (val = j = 0; j < dimOfWorld; j++)
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
	val += Lambda[i][j] * b[j];
      val *= factor;
      Lb[i] += val;
    }    
  }

  Operator::Operator(Flag operatorType,
		     const FiniteElemSpace *rowFESpace_,
		     const FiniteElemSpace *colFESpace_)
    : rowFESpace(rowFESpace_), 
      colFESpace(colFESpace_ ? colFESpace_ : rowFESpace_),
      type(operatorType), 
      fillFlag(Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS | 
	       Mesh::FILL_DET | Mesh::FILL_GRD_LAMBDA),
      uhOld(NULL),
      optimized(true)
  {
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
    int maxThreads = omp_get_max_threads();

    assembler.resize(maxThreads);
    secondOrder.resize(maxThreads);
    firstOrderGrdPsi.resize(maxThreads);
    firstOrderGrdPhi.resize(maxThreads);
    zeroOrder.resize(maxThreads);

    for (int i = 0; i < maxThreads; i++) {
      assembler[i] = NULL;
      secondOrder[i].resize(0);
      firstOrderGrdPsi[i].resize(0);
      firstOrderGrdPhi[i].resize(0);
      zeroOrder[i].resize(0);
    }

199
200
201
202
203
204
205
206
207
208
209
210
211
    nRow = rowFESpace->getBasisFcts()->getNumber();
    nCol = colFESpace->getBasisFcts()->getNumber();
  };

  void Operator::setUhOld(const DOFVectorBase<double> *uhOld_)
  {
    uhOld = uhOld_;
  }

  void Operator::getElementMatrix(const ElInfo *elInfo, 
				  ElementMatrix *userMat, 
				  double factor)
  {
212
213
214
215
    int myRank = omp_get_thread_num();

    if (!assembler[myRank]) {
      initAssembler(myRank, NULL, NULL, NULL, NULL);
216
217
    }

218
    assembler[myRank]->calculateElementMatrix(elInfo, userMat, factor);
219
220
  }

221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
  void Operator::getElementMatrix(const ElInfo *rowElInfo, 
				  const ElInfo *colElInfo,
				  ElementMatrix *userMat, 
				  double factor)
  {
    int myRank = omp_get_thread_num();

    if (!assembler[myRank]) {
      initAssembler(myRank, NULL, NULL, NULL, NULL);
    }

    assembler[myRank]->calculateElementMatrix(rowElInfo, colElInfo, 
    					      userMat, factor);
  }

236
237
238
239
  void Operator::getElementVector(const ElInfo *elInfo, 
				  ElementVector *userVec, 
				  double factor)
  {
240
241
242
243
    int myRank = omp_get_thread_num();

    if (!assembler[myRank]) {
      initAssembler(myRank, NULL, NULL, NULL, NULL);
244
245
    }

246
    assembler[myRank]->calculateElementVector(elInfo, userVec, factor);
247
248
  }

249
250
  void Operator::initAssembler(int rank,
			       Quadrature *quad2,
251
252
253
254
			       Quadrature *quad1GrdPsi,
			       Quadrature *quad1GrdPhi,
			       Quadrature *quad0) 
  {
255
    if (!assembler[rank])
Thomas Witkowski's avatar
Thomas Witkowski committed
256
      if (optimized) {
257
	assembler[rank] = 
258
259
260
261
	  NEW OptimizedAssembler(this,
				 quad2, quad1GrdPsi, quad1GrdPhi, quad0,
				 rowFESpace, colFESpace);
      } else {
262
	assembler[rank] = 
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
	  NEW StandardAssembler(this,
				quad2, quad1GrdPsi, quad1GrdPhi, quad0,
				rowFESpace, colFESpace);
      }
  }


  void MatrixFct_SOT::initElement(const ElInfo* elInfo, 
				  SubAssembler* subAssembler,
				  Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
  }
 
  void VecAtQP_SOT::initElement(const ElInfo* elInfo, 
				SubAssembler* subAssembler,
Thomas Witkowski's avatar
Thomas Witkowski committed
279
				Quadrature* quad) 
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
  }
 
  void CoordsAtQP_SOT::initElement(const ElInfo* elInfo, 
				   SubAssembler* subAssembler,
				   Quadrature *quad) 
  {
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }

  void MatrixGradient_SOT::initElement(const ElInfo* elInfo, 
				       SubAssembler* subAssembler,
				       Quadrature *quad) 
  {
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
  }

  void FctGradient_SOT::initElement(const ElInfo* elInfo, 
				    SubAssembler* subAssembler,
				    Quadrature *quad) 
  {
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
  }

  void VecAndGradientAtQP_SOT::initElement(const ElInfo* elInfo, 
					   SubAssembler* subAssembler,
					   Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
  }

  void VecAndCoordsAtQP_SOT::initElement(const ElInfo* elInfo, 
					 SubAssembler* subAssembler,
					 Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }
 
  void MatrixGradientAndCoords_SOT::initElement(const ElInfo* elInfo, 
						SubAssembler* subAssembler,
						Quadrature *quad) 
  {
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);  
  }

  void VecGradCoordsAtQP_SOT::initElement(const ElInfo* elInfo, 
					  SubAssembler* subAssembler,
					  Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);  
  }

  void VecAtQP_FOT::initElement(const ElInfo* elInfo, 
				SubAssembler* subAssembler,
				Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
  }

  void CoordsAtQP_FOT::initElement(const ElInfo* elInfo, 
				   SubAssembler* subAssembler,
				   Quadrature *quad) 
  {
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }

  void VecCoordsAtQP_FOT::initElement(const ElInfo* elInfo, 
				      SubAssembler* subAssembler,
				      Quadrature *quad) 
  {
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }

  void VectorGradient_FOT::initElement(const ElInfo* elInfo, 
				       SubAssembler* subAssembler,
				       Quadrature *quad) 
  {
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
  }

  void VectorFct_FOT::initElement(const ElInfo* elInfo, 
				  SubAssembler* subAssembler,
				  Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
  }
 
  void VecFctAtQP_FOT::initElement(const ElInfo* elInfo, 
				   SubAssembler* subAssembler,
				   Quadrature *quad) 
  {
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }

  void VecAtQP_ZOT::initElement(const ElInfo* elInfo, 
				SubAssembler* subAssembler,
				Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
  }

  void MultVecAtQP_ZOT::initElement(const ElInfo* elInfo, 
				    SubAssembler* subAssembler,
				    Quadrature *quad) 
  {
    vecAtQPs1 = subAssembler->getVectorAtQPs(vec1, elInfo, quad);
    vecAtQPs2 = subAssembler->getVectorAtQPs(vec2, elInfo, quad);
  }

  void Vec2AtQP_ZOT::initElement(const ElInfo* elInfo, 
				 SubAssembler* subAssembler,
				 Quadrature *quad) 
  {
    vecAtQPs1 = subAssembler->getVectorAtQPs(vec1, elInfo, quad);
    vecAtQPs2 = subAssembler->getVectorAtQPs(vec2, elInfo, quad);
  }

  void VecAndCoordsAtQP_ZOT::initElement(const ElInfo* elInfo, 
					 SubAssembler* subAssembler,
					 Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }
 
  void FctGradientCoords_ZOT::initElement(const ElInfo* elInfo, 
					  SubAssembler* subAssembler,
					  Quadrature *quad) 
  {
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }
 
  void VecGradCoordsAtQP_ZOT::initElement(const ElInfo* elInfo, 
					  SubAssembler* subAssembler,
					  Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }
 
  void VecAndGradAtQP_ZOT::initElement(const ElInfo* elInfo, 
				       SubAssembler* subAssembler,
				       Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
  }
 
  void VecAndGradVecAtQP_ZOT::initElement(const ElInfo* elInfo, 
					  SubAssembler* subAssembler,
					  Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    gradAtQPs = subAssembler->getGradientsAtQPs(vecGrd, elInfo, quad);
  }

  void VecAndGradVec2AtQP_ZOT::initElement(const ElInfo* elInfo, 
					   SubAssembler* subAssembler,
					   Quadrature *quad) 
  {
    vecAtQPs = subAssembler->getVectorAtQPs(vec, elInfo, quad);
    grad1AtQPs = subAssembler->getGradientsAtQPs(vecGrd1, elInfo, quad);
    grad2AtQPs = subAssembler->getGradientsAtQPs(vecGrd2, elInfo, quad);
  }
 
  void FctGradient_ZOT::initElement(const ElInfo* elInfo, 
				    SubAssembler* subAssembler,
				    Quadrature *quad) 
  {
    gradAtQPs = subAssembler->getGradientsAtQPs(vec, elInfo, quad);
  }
 
  void CoordsAtQP_ZOT::initElement(const ElInfo* elInfo, 
				   SubAssembler* subAssembler,
				   Quadrature *quad) {
    coordsAtQPs = subAssembler->getCoordsAtQPs(elInfo, quad);
  }

  void MatrixFct_SOT::getLALt(const ElInfo *elInfo, int numPoints, DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
468
    for (int iq = 0; iq < numPoints; iq++) {
469
470
471
472
473
474
      lalt(Lambda, (*matrixFct)(vecAtQPs[iq]), *(LALt[iq]), symmetric, 1.0);
    }
  }

  void VecAtQP_SOT::getLALt(const ElInfo *elInfo, int numPoints, DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
475
    for (int iq = 0; iq < numPoints; iq++) {
476
477
478
479
480
481
      l1lt(Lambda, *(LALt[iq]), (*f)(vecAtQPs[iq]));
    }
  }
  
  void CoordsAtQP_SOT::getLALt(const ElInfo *elInfo, int numPoints, DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
482
    for (int iq = 0; iq < numPoints; iq++) {
483
484
485
486
487
488
      l1lt(Lambda, (*LALt[iq]), (*g)(coordsAtQPs[iq]));
    }
  }

  void MatrixGradient_SOT::getLALt(const ElInfo *elInfo, int numPoints, DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
489
    for (int iq = 0; iq < numPoints; iq++) {
490
491
492
493
494
495
      lalt(Lambda, (*f)(gradAtQPs[iq]), (*LALt[iq]), symmetric, 1.0);
    }
  }

  void FctGradient_SOT::getLALt(const ElInfo *elInfo, int numPoints, DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
496
    for (int iq = 0; iq < numPoints; iq++) {
497
498
499
500
501
502
503
      l1lt(Lambda, *(LALt[iq]), (*f)(gradAtQPs[iq]));
    }
  }
  
  void VecAndGradientAtQP_SOT::getLALt(const ElInfo *elInfo, int numPoints,
				       DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
504
    for (int iq = 0; iq < numPoints; iq++) {
505
506
507
508
509
510
      l1lt(Lambda, *(LALt[iq]), (*f)(vecAtQPs[iq], gradAtQPs[iq]));
    }
  }

  void VecAndCoordsAtQP_SOT::getLALt(const ElInfo *elInfo, int numPoints, DimMat<double> **LALt)
    const { const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
511
    for (int iq = 0; iq < numPoints; iq++) {
512
513
514
515
516
517
518
      l1lt(Lambda, *(LALt[iq]), (*f)(vecAtQPs[iq], coordsAtQPs[iq]));
    }
  }
  
  void MatrixGradientAndCoords_SOT::getLALt(const ElInfo *elInfo, int numPoints,
					    DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
519
    for (int iq = 0; iq < numPoints; iq++) {
520
521
522
523
524
525
526
      lalt(Lambda, (*f)(gradAtQPs[iq], coordsAtQPs[iq]), (*LALt[iq]), symmetric, 1.0);
    }
  }

  void VecGradCoordsAtQP_SOT::getLALt(const ElInfo *elInfo, int numPoints,
				      DimMat<double> **LALt) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
527
    for (int iq = 0; iq < numPoints; iq++) {
528
529
530
531
532
533
      l1lt(Lambda, *(LALt[iq]), (*f)(vecAtQPs[iq], gradAtQPs[iq], coordsAtQPs[iq]));
    }
  }

  void VecAtQP_FOT::getLb(const ElInfo *elInfo, int numPoints, VectorOfFixVecs<DimVec<double> >& Lb) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
534
535
    for (int iq = 0; iq < numPoints; iq++) {
      if (b)
536
537
538
539
540
541
542
543
	lb(Lambda, *b, Lb[iq], (*f)(vecAtQPs[iq]));
      else
	l1(Lambda, Lb[iq], (*f)(vecAtQPs[iq]));
    }
  }

  void CoordsAtQP_FOT::getLb(const ElInfo *elInfo, int numPoints, VectorOfFixVecs<DimVec<double> >& Lb) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
544
    for (int iq = 0; iq < numPoints; iq++) {
545
546
547
548
549
550
551
552
      l1(Lambda, Lb[iq], (*g)(coordsAtQPs[iq]));
    }
  }

  void VecCoordsAtQP_FOT::getLb(const ElInfo *elInfo, int numPoints,
				VectorOfFixVecs<DimVec<double> >& Lb) const
  {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
553
554
555
    for (int iq = 0; iq < numPoints; iq++) {
      lb(Lambda, b, Lb[iq], (*g)(coordsAtQPs[iq]));
    }
556
557
558
559
  }

  void VectorGradient_FOT::getLb(const ElInfo *elInfo, int numPoints, VectorOfFixVecs<DimVec<double> >& Lb) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
560
561
    if (f) {
      for (int iq = 0; iq < numPoints; iq++) {
562
563
564
	lb(Lambda, (*f)(gradAtQPs[iq]), Lb[iq], 1.0);
      }
    } else {
Thomas Witkowski's avatar
Thomas Witkowski committed
565
      for (int iq = 0; iq < numPoints; iq++) {
566
567
568
569
570
571
572
	lb(Lambda, gradAtQPs[iq], Lb[iq], 1.0);
      }
    }
  }

  void VectorFct_FOT::getLb(const ElInfo *elInfo, int numPoints, VectorOfFixVecs<DimVec<double> >& Lb) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
573
    for (int iq = 0; iq < numPoints; iq++) {
574
575
576
577
578
579
      lb(Lambda, (*vecFct)(vecAtQPs[iq]), Lb[iq], 1.0);
    }
  }

  void VecFctAtQP_FOT::getLb(const ElInfo *elInfo, int numPoints, VectorOfFixVecs<DimVec<double> >& Lb) const {
    const DimVec<WorldVector<double> > &Lambda = elInfo->getGrdLambda();
Thomas Witkowski's avatar
Thomas Witkowski committed
580
    for (int iq = 0; iq < numPoints; iq++) {
581
582
583
584
585
      lb(Lambda, (*g)(coordsAtQPs[iq]), Lb[iq], 1.0);
    }
  }

  void VecAtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
586
587
    if (f) {
      for (int iq = 0; iq < numPoints; iq++) {
588
589
590
	C[iq] += (*f)(vecAtQPs[iq]);
      }
    } else {
Thomas Witkowski's avatar
Thomas Witkowski committed
591
      for (int iq = 0; iq < numPoints; iq++) {
592
593
594
595
596
597
598
599
600
601
602
603
	C[iq] += vecAtQPs[iq];
      }
    }
  }

  void VecAtQP_ZOT::eval(int numPoints,
			 const double              *uhAtQP,
			 const WorldVector<double> *grdUhAtQP,
			 const WorldMatrix<double> *D2UhAtQP,
			 double *result,
			 double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
604
    for (int iq = 0; iq < numPoints; iq++) {
605
606
607
608
609
      result[iq] += fac * (*f)(vecAtQPs[iq]) * uhAtQP[iq];
    }
  }

  void MultVecAtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
610
    for (int iq = 0; iq < numPoints; iq++) {
611
612
613
614
615
616
617
618
619
620
621
      C[iq] += (*f1)(vecAtQPs1[iq]) * (*f2)(vecAtQPs2[iq]);
    }
  }

  void MultVecAtQP_ZOT::eval(int numPoints,
			     const double              *uhAtQP,
			     const WorldVector<double> *grdUhAtQP,
			     const WorldMatrix<double> *D2UhAtQP,
			     double *result,
			     double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
622
623
624
    for (int iq = 0; iq < numPoints; iq++) {
      result[iq] += fac * (*f1)(vecAtQPs1[iq]) * 
	(*f2)(vecAtQPs2[iq]) * uhAtQP[iq];
625
626
627
628
    }
  }

  void Vec2AtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
629
    for (int iq = 0; iq < numPoints; iq++) {
630
631
632
633
634
635
636
637
638
639
640
      C[iq] += (*f)(vecAtQPs1[iq], vecAtQPs2[iq]);
    }
  }

  void Vec2AtQP_ZOT::eval(int numPoints,
			  const double              *uhAtQP,
			  const WorldVector<double> *grdUhAtQP,
			  const WorldMatrix<double> *D2UhAtQP,
			  double *result,
			  double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
641
642
    for (int iq = 0; iq < numPoints; iq++) {
      result[iq] += fac * (*f)(vecAtQPs1[iq], vecAtQPs2[iq]) * 
643
644
645
646
647
	uhAtQP[iq];
    }
  }

  void VecAndCoordsAtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
648
    for (int iq = 0; iq < numPoints; iq++) {
649
650
651
652
653
654
655
656
657
658
659
      C[iq] += (*f)(vecAtQPs[iq], coordsAtQPs[iq]);
    }
  }

  void VecAndCoordsAtQP_ZOT::eval(int numPoints,
				  const double              *uhAtQP,
				  const WorldVector<double> *grdUhAtQP,
				  const WorldMatrix<double> *D2UhAtQP,
				  double *result,
				  double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
660
661
    for (int iq = 0; iq < numPoints; iq++) {
      result[iq] += fac * (*f)(vecAtQPs[iq], coordsAtQPs[iq]) * 
662
663
664
665
666
667
	uhAtQP[iq];
    }
  }

 
  void FctGradientCoords_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
668
    for (int iq = 0; iq < numPoints; iq++) {
669
670
671
672
673
674
675
676
677
678
679
      C[iq] += (*f)(gradAtQPs[iq], coordsAtQPs[iq]);
    }
  }
 
  void FctGradientCoords_ZOT::eval(int numPoints,
				   const double              *uhAtQP,
				   const WorldVector<double> *grdUhAtQP,
				   const WorldMatrix<double> *D2UhAtQP,
				   double *result,
				   double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
680
681
    for (int iq = 0; iq < numPoints; iq++) {
      result[iq] += fac * (*f)(gradAtQPs[iq], coordsAtQPs[iq]) * 
682
683
684
685
686
	uhAtQP[iq];
    }
  }

  void VecGradCoordsAtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
687
    for (int iq = 0; iq < numPoints; iq++) {
688
689
690
691
692
693
694
695
696
697
698
      C[iq] += (*f)(vecAtQPs[iq], gradAtQPs[iq], coordsAtQPs[iq]);
    }
  }
 
  void VecGradCoordsAtQP_ZOT::eval(int numPoints,
				   const double              *uhAtQP,
				   const WorldVector<double> *grdUhAtQP,
				   const WorldMatrix<double> *D2UhAtQP,
				   double *result,
				   double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
699
    for (int iq = 0; iq < numPoints; iq++) {
700
701
702
703
704
705
706
707
      result[iq] += 
	fac * 
	(*f)(vecAtQPs[iq], gradAtQPs[iq], coordsAtQPs[iq]) * 
	uhAtQP[iq];
    }
  }

  void VecAndGradAtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
708
    for (int iq = 0; iq < numPoints; iq++) {
709
710
711
712
713
714
715
716
717
718
719
      C[iq] += (*f)(vecAtQPs[iq], gradAtQPs[iq]);
    }
  }
 
  void VecAndGradAtQP_ZOT::eval(int numPoints,
				const double              *uhAtQP,
				const WorldVector<double> *grdUhAtQP,
				const WorldMatrix<double> *D2UhAtQP,
				double *result,
				double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
720
    for (int iq = 0; iq < numPoints; iq++) {
721
722
723
724
725
726
727
728
729
      result[iq] += 
	fac * 
	(*f)(vecAtQPs[iq], gradAtQPs[iq]) * 
	uhAtQP[iq];
    }
  }

  void VecAndGradVecAtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const 
  { 
Thomas Witkowski's avatar
Thomas Witkowski committed
730
    for (int iq = 0; iq < numPoints; iq++) {
731
732
733
734
735
736
737
738
739
740
741
      C[iq] += (*f)(vecAtQPs[iq], gradAtQPs[iq]);
    }
  }
 
  void VecAndGradVecAtQP_ZOT::eval(int numPoints,
				   const double              *uhAtQP,
				   const WorldVector<double> *grdUhAtQP,
				   const WorldMatrix<double> *D2UhAtQP,
				   double *result,
				   double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
742
    for (int iq = 0; iq < numPoints; iq++) {
743
744
745
746
747
748
749
750
      result[iq] += 
	fac * 
	(*f)(vecAtQPs[iq], gradAtQPs[iq]) * 
	uhAtQP[iq];
    }
  }

  void VecAndGradVec2AtQP_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
751
    for (int iq = 0; iq < numPoints; iq++) {
752
753
754
755
756
757
758
759
760
761
762
      C[iq] += (*f)(vecAtQPs[iq], grad1AtQPs[iq], grad2AtQPs[iq]);
    }
  }
 
  void VecAndGradVec2AtQP_ZOT::eval(int numPoints,
				    const double              *uhAtQP,
				    const WorldVector<double> *grdUhAtQP,
				    const WorldMatrix<double> *D2UhAtQP,
				    double *result,
				    double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
763
    for (int iq = 0; iq < numPoints; iq++) {
764
765
766
767
768
769
770
771
      result[iq] += 
	fac * 
	(*f)(vecAtQPs[iq], grad1AtQPs[iq], grad2AtQPs[iq]) * 
	uhAtQP[iq];
    }
  }

  void FctGradient_ZOT::getC(const ElInfo *, int numPoints, double *C) const { 
Thomas Witkowski's avatar
Thomas Witkowski committed
772
    for (int iq = 0; iq < numPoints; iq++) {
773
774
775
776
777
778
779
780
781
782
783
      C[iq] += (*f)(gradAtQPs[iq]);
    }
  }
 
  void FctGradient_ZOT::eval(int numPoints,
			     const double              *uhAtQP,
			     const WorldVector<double> *grdUhAtQP,
			     const WorldMatrix<double> *D2UhAtQP,
			     double *result,
			     double fac) const
  {
784
785
    for (int iq = 0; iq < numPoints; iq++) {
      result[iq] += fac * (*f)(gradAtQPs[iq]) * uhAtQP[iq];
786
787
788
    }
  }

789
  void CoordsAtQP_ZOT::getC(const ElInfo *elInfo, int numPoints, double *C) const { 
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
    for (int iq = 0; iq < numPoints; iq++) {
      C[iq] += (*g)(coordsAtQPs[iq]);
    }
  }

  void CoordsAtQP_ZOT::eval(int numPoints,
			    const double              *uhAtQP,
			    const WorldVector<double> *grdUhAtQP,
			    const WorldMatrix<double> *D2UhAtQP,
			    double *result,
			    double fac) const
  {
    for (int iq = 0; iq < numPoints; iq++) {
      result[iq] += fac * (*g)(coordsAtQPs[iq]) * uhAtQP[iq];
    }
  }

  void MatrixFct_SOT::eval(int numPoints,
			   const double              *uhAtQP,
			   const WorldVector<double> *grdUhAtQP,
			   const WorldMatrix<double> *D2UhAtQP,
			   double *result,
			   double factor) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
814
    int dow = Global::getGeo(WORLD);
815

Thomas Witkowski's avatar
Thomas Witkowski committed
816
    for (int iq = 0; iq < numPoints; iq++) {
817
818
819
820
      double resultQP = 0.0;

      WorldMatrix<double> A = (*matrixFct)(vecAtQPs[iq]);

Thomas Witkowski's avatar
Thomas Witkowski committed
821
822
823
      if (D2UhAtQP) {
	for (int i = 0; i < dow; i++) {
	  for (int j = 0; j < dow; j++) {
824
825
826
827
828
	    resultQP += A[i][j] * D2UhAtQP[iq][j][i];
	  }
	}
      }

Thomas Witkowski's avatar
Thomas Witkowski committed
829
      if (grdUhAtQP) {
830
831
832
833
834
835
836
837
838
839
840
	resultQP += (*divFct)(A) * grdUhAtQP[iq];
      }

      result[iq] += resultQP * factor;
    }
  }

  void MatrixFct_SOT::weakEval(int numPoints,
			       const WorldVector<double> *grdUhAtQP,
			       WorldVector<double> *result) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
841
    if (grdUhAtQP) {
842
      WorldMatrix<double> A;
Thomas Witkowski's avatar
Thomas Witkowski committed
843
      for (int iq = 0; iq < numPoints; iq++) {
844
845
846
847
848
849
850
851
852
853
854
855
856
857
	A = (*matrixFct)(vecAtQPs[iq]);
	result[iq] += A * grdUhAtQP[iq];
      }
    }
  }


  void Matrix_SOT::eval(int numPoints,
			const double              *uhAtQP,
			const WorldVector<double> *grdUhAtQP,
			const WorldMatrix<double> *D2UhAtQP,
			double *result,
			double factor) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
858
    int dow = Global::getGeo(WORLD);
859
  
Thomas Witkowski's avatar
Thomas Witkowski committed
860
861
    if (D2UhAtQP) {
      for (int iq = 0; iq < numPoints; iq++) {
862
	double resultQP = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
863
864
	for (int i = 0; i < dow; i++) {
	  for (int j = 0; j < dow; j++) {
865
866
867
868
869
870
871
872
873
874
875
876
	    resultQP += matrix[i][j] * D2UhAtQP[iq][j][i];
	  }
	}
	result[iq] += resultQP * factor;
      }
    }
  }

  void Matrix_SOT::weakEval(int numPoints,
			    const WorldVector<double> *grdUhAtQP,
			    WorldVector<double> *result) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
877
878
    if (grdUhAtQP) {
      for (int iq = 0; iq < numPoints; iq++) {
879
880
881
882
883
884
885
886
887
888
889
890
891
892
	result[iq] += matrix * grdUhAtQP[iq];
      }
    }
  }



  void MatrixGradient_SOT::eval(int numPoints,
				const double              *uhAtQP,
				const WorldVector<double> *grdUhAtQP,
				const WorldMatrix<double> *D2UhAtQP,
				double *result,
				double factor) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
893
    int dow = Global::getGeo(WORLD);
894

Thomas Witkowski's avatar
Thomas Witkowski committed
895
    for (int iq = 0; iq < numPoints; iq++) {
896
897
898
899
      double resultQP = 0.0;

      WorldMatrix<double> A = (*f)(gradAtQPs[iq]);

Thomas Witkowski's avatar
Thomas Witkowski committed
900
901
902
      if (D2UhAtQP) {
	for (int i = 0; i < dow; i++) {
	  for (int j = 0; j < dow; j++) {
903
904
905
906
907
	    resultQP += A[i][j] * D2UhAtQP[iq][j][i];
	  }
	}
      }

Thomas Witkowski's avatar
Thomas Witkowski committed
908
      if (grdUhAtQP) {
909
910
911
912
913
914
915
916
917
918
919
	resultQP += (*divFct)(A) * grdUhAtQP[iq];
      }

      result[iq] += resultQP * factor;
    }
  };

  void MatrixGradient_SOT::weakEval(int numPoints,
				    const WorldVector<double> *grdUhAtQP,
				    WorldVector<double> *result) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
920
    if (grdUhAtQP) {
921
      WorldMatrix<double> A;
Thomas Witkowski's avatar
Thomas Witkowski committed
922
      for (int iq = 0; iq < numPoints; iq++) {
923
924
925
926
927
928
929
930
931
932
933
934
935
	A = (*f)(gradAtQPs[iq]);
	result[iq] += A * grdUhAtQP[iq];
      }
    }
  }

  void VecAtQP_SOT::eval(int numPoints,
			 const double              *uhAtQP,
			 const WorldVector<double> *grdUhAtQP,
			 const WorldMatrix<double> *D2UhAtQP,
			 double *result,
			 double fac) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
936
    int dow = Global::getGeo(WORLD);
937

Thomas Witkowski's avatar
Thomas Witkowski committed
938
939
    if (D2UhAtQP) {
      for (int iq = 0; iq < numPoints; iq++) {
940
941
	double factor = (*f)(vecAtQPs[iq]);
	double resultQP = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
942
	for (int i = 0; i < dow; i++) {
943
944
945
946
947
948
949
950
951
952
953
	  resultQP += D2UhAtQP[iq][i][i];
	}
	result[iq] += fac * factor * resultQP;
      }
    }
  }

  void VecAtQP_SOT::weakEval(int numPoints,
			     const WorldVector<double> *grdUhAtQP,
			     WorldVector<double> *result) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
954
955
    if (grdUhAtQP) {
      for (int iq = 0; iq < numPoints; iq++) {
956
957
958
959
960
961
962
963
964
965
966
967
968
	double factor = (*f)(vecAtQPs[iq]);  
	axpy(factor, grdUhAtQP[iq], result[iq]);
      }
    }
  }

  void CoordsAtQP_SOT::eval(int numPoints,
			    const double              *uhAtQP,
			    const WorldVector<double> *grdUhAtQP,
			    const WorldMatrix<double> *D2UhAtQP,
			    double *result,
			    double f) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
969
    int dow = Global::getGeo(WORLD);
970

Thomas Witkowski's avatar
Thomas Witkowski committed
971
972
    if (D2UhAtQP) {
      for (int iq = 0; iq < numPoints; iq++) {
973
974
	double factor = (*g)(coordsAtQPs[iq]);
	double resultQP = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
975
	for (int i = 0; i < dow; i++) {
976
977
978
979
980
981
982
983
984
985
986
	  resultQP += D2UhAtQP[iq][i][i];
	}
	result[iq] += factor * f * resultQP;
      }
    }
  }

  void CoordsAtQP_SOT::weakEval(int numPoints,
				const WorldVector<double> *grdUhAtQP,
				WorldVector<double> *result) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
987
988
    if (grdUhAtQP) {
      for (int iq = 0; iq < numPoints; iq++) {
989
990
991
992
993
994
995
996
997
998
999
1000
	double factor = (*g)(coordsAtQPs[iq]);  
	axpy(factor, grdUhAtQP[iq], result[iq]);
      }
    }
  }

  void FctGradient_SOT::eval(int numPoints,
			     const double              *uhAtQP,
			     const WorldVector<double> *grdUhAtQP,
			     const WorldMatrix<double> *D2UhAtQP,
			     double *result,
			     double fac) const
For faster browsing, not all history is shown. View entire blame