DOFVector.hh 49.5 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
//
// Software License for AMDiS
//
// Copyright (c) 2010 Dresden University of Technology 
// All rights reserved.
// Authors: Simon Vey, Thomas Witkowski et al.
//
// This file is part of AMDiS
//
// See also license.opensource.txt in the distribution.


13
#include <list>
Thomas Witkowski's avatar
Thomas Witkowski committed
14
#include <algorithm>
15
#include <math.h>
16

17
18
19
20
21
#include <boost/numeric/mtl/mtl.hpp>
#include <boost/numeric/mtl/utility/tag.hpp>
#include <boost/numeric/mtl/utility/category.hpp>
#include <boost/numeric/linear_algebra/identity.hpp>

22
23
24
25
26
27
28
29
30
31
32
33
#include "FixVec.h"
#include "Boundary.h"
#include "DOFAdmin.h"
#include "ElInfo.h"
#include "Error.h"
#include "FiniteElemSpace.h"
#include "Global.h"
#include "Mesh.h"
#include "Quadrature.h"
#include "AbstractFunction.h"
#include "BoundaryManager.h"
#include "Assembler.h"
34
#include "Operator.h"
35
#include "Initfile.h"
Thomas Witkowski's avatar
Thomas Witkowski committed
36
#include "Traverse.h"
37

38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// Defining the interface for MTL4
namespace mtl {

  // Let MTL4 know that DOFVector it is a column vector
  namespace traits {
    template <typename T>
    struct category< AMDiS::DOFVector<T> > 
    {
      typedef tag::dense_col_vector type;
    };
  }

  namespace ashape {
    template <typename T>
    struct ashape< AMDiS::DOFVector<T> > 
    {
      typedef cvec<typename ashape<T>::type> type;
    };
  }

  // Modelling Collection and MutableCollection
  template <typename T>
  struct Collection< AMDiS::DOFVector<T> > 
  {
    typedef T               value_type;
    typedef const T&        const_reference;
    typedef std::size_t     size_type;
  };

  template <typename T>
  struct MutableCollection< AMDiS::DOFVector<T> > 
    : public Collection< AMDiS::DOFVector<T> > 
  {
    typedef T&              reference;
  };


} // namespace mtl



79
80
81
namespace AMDiS {

  template<typename T>
82
  DOFVectorBase<T>::DOFVectorBase(const FiniteElemSpace *f, std::string n)
Thomas Witkowski's avatar
Thomas Witkowski committed
83
84
    : feSpace(f),
      name(n),
85
      elementVector(f->getBasisFcts()->getNumber()),
Thomas Witkowski's avatar
Thomas Witkowski committed
86
      boundaryManager(NULL)
87
  {    
Thomas Witkowski's avatar
Thomas Witkowski committed
88
    nBasFcts = feSpace->getBasisFcts()->getNumber();
Thomas Witkowski's avatar
Thomas Witkowski committed
89
    dim = feSpace->getMesh()->getDim();
90
91
92
93
94
  }
  

  template<typename T>
  DOFVectorBase<T>::~DOFVectorBase()
95
  {}
96

97
 
Thomas Witkowski's avatar
Thomas Witkowski committed
98
  template<typename T>
99
  DOFVector<T>::DOFVector(const FiniteElemSpace* f, std::string n)
100
    : DOFVectorBase<T>(f, n),
101
      coarsenOperation(COARSE_INTERPOL)
102
  {
103
    vec.resize(0);
104
105
106
    init(f, n);
  } 

107

108
  template<typename T>
109
  void DOFVector<T>::init(const FiniteElemSpace* f, std::string n)
110
  {
111
112
113
114
115
    this->name = n;
    this->feSpace = f;
    if (this->feSpace && this->feSpace->getAdmin())
      (this->feSpace->getAdmin())->addDOFIndexed(this);
    this->boundaryManager = new BoundaryManager(f);
116
117
118
119
120
  }

  template<typename T>
  DOFVector<T>::~DOFVector()
  {
121
122
    if (this->feSpace && this->feSpace->getAdmin())
      (this->feSpace->getAdmin())->removeDOFIndexed(this);
123
    
Thomas Witkowski's avatar
Thomas Witkowski committed
124
    if (this->boundaryManager)
125
      delete this->boundaryManager;
126
    
127
    vec.clear();
128
129
130
131
132
  }

  template<typename T>
  DOFVector<T> * DOFVector<T>::traverseVector = NULL;

Thomas Witkowski's avatar
Thomas Witkowski committed
133
134
135
136
  template<typename T>
  void DOFVectorBase<T>::addElementVector(T factor, 
					  const ElementVector &elVec, 
					  const BoundaryType *bound,
137
					  ElInfo *elInfo,
Thomas Witkowski's avatar
Thomas Witkowski committed
138
139
140
141
					  bool add)
  {
    FUNCNAME("DOFVector::addElementVector()");

142
143
144
    std::vector<DegreeOfFreedom> indices(nBasFcts);
    feSpace->getBasisFcts()->getLocalIndices(elInfo->getElement(), feSpace->getAdmin(),
					     indices);
Thomas Witkowski's avatar
Thomas Witkowski committed
145

146
    for (DegreeOfFreedom i = 0; i < nBasFcts; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
147
148
149
      BoundaryCondition *condition = 
	bound ? this->getBoundaryManager()->getBoundaryCondition(bound[i]) : NULL;

Thomas Witkowski's avatar
Thomas Witkowski committed
150
      if (!(condition && condition->isDirichlet())) {
151
	DegreeOfFreedom irow = indices[i];
Thomas Witkowski's avatar
Thomas Witkowski committed
152

153
	if (add)
Thomas Witkowski's avatar
Thomas Witkowski committed
154
	  (*this)[irow] += factor * elVec[i];
155
156
	else  
	  (*this)[irow] = factor * elVec[i];	
Thomas Witkowski's avatar
Thomas Witkowski committed
157
158
159
160
      }
    }
  }

161
162
163
164
165
  template<typename T>
  double DOFVector<T>::nrm2() const
  {
    FUNCNAME("DOFVector<T>::nrm2()");

166
    checkFeSpace(this->feSpace, vec);
Thomas Witkowski's avatar
Thomas Witkowski committed
167

168
    double nrm = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
169
170
    Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(this)), 
			 USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
171
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator)
172
173
      nrm += (*vecIterator) * (*vecIterator);

174
175
176
177
178
179
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localNrm = nrm;
    MPI::COMM_WORLD.Allreduce(&localNrm, &nrm, 1, MPI_DOUBLE, MPI_SUM);
#endif

    return sqrt(nrm);
180
181
  }

182

183
184
185
186
187
  template<typename T>
  double DOFVector<T>::squareNrm2() const
  {
    FUNCNAME("DOFVector<T>::nrm2()");

188
    checkFeSpace(this->feSpace, vec);
189
190
    
    double nrm = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
191
192
    Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(this)), 
			 USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
193
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator)
194
195
      nrm += (*vecIterator) * (*vecIterator);

196
197
198
199
200
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localNrm = nrm;
    MPI::COMM_WORLD.Allreduce(&localNrm, &nrm, 1, MPI_DOUBLE, MPI_SUM);
#endif

201
202
203
    return nrm;
  }

204

205
206
207
  template<typename T>
  T DOFVector<T>::asum() const
  {
208
    FUNCNAME("DOFVector<T>::asum()");
209

210
    checkFeSpace(this->feSpace, vec);
211

212
    double nrm = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
213
214
    Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(this)), 
			 USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
215
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator)
216
217
      nrm += abs(*vecIterator);

218
219
220
221
222
223
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localNrm = nrm;
    MPI::COMM_WORLD.Allreduce(&localNrm, &nrm, 1, MPI_DOUBLE, MPI_SUM);
#endif

    return nrm;
224
225
  }

226

227
228
229
  template<typename T>
  T DOFVector<T>::sum() const
  {
230
    FUNCNAME("DOFVector<T>::sum()");
231

232
    checkFeSpace(this->feSpace, vec);
233

234
    double nrm = 0.0;    
Thomas Witkowski's avatar
Thomas Witkowski committed
235
236
    Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(this)), 
			 USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
237
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator)
238
239
      nrm += *vecIterator;

240
241
242
243
244
245
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localNrm = nrm;
    MPI::COMM_WORLD.Allreduce(&localNrm, &nrm, 1, MPI_DOUBLE, MPI_SUM);
#endif

    return nrm;
246
247
  }

248

249
250
251
  template<typename T>
  void DOFVector<T>::set(T alpha)
  {
252
    FUNCNAME("DOFVector<T>::set()");
253

254
    checkFeSpace(this->feSpace, vec);
Thomas Witkowski's avatar
Thomas Witkowski committed
255

256
    Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(this), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
257
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator)
258
259
260
261
262
263
264
      *vecIterator = alpha ; 
  }


  template<typename T>
  void DOFVector<T>::copy(const DOFVector<T>& x)
  {
265
    FUNCNAME("DOFVector<T>::copy()");
Thomas Witkowski's avatar
Thomas Witkowski committed
266

267
    checkFeSpace(this->feSpace, vec);
Thomas Witkowski's avatar
Thomas Witkowski committed
268

269
270
    TEST_EXIT_DBG(static_cast<int>(x.vec.size()) >= 
		  this->feSpace->getAdmin()->getUsedSize())
271
      ("x.size = %d too small: admin->sizeUsed = %d\n", x.vec.size(), 
272
       this->feSpace->getAdmin()->getUsedSize());
273
    
274
    Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(this), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
275
276
277
278
    Iterator xIterator(dynamic_cast<DOFVector<T>*>(const_cast<DOFVector<T>*>(&x)), 
		       USED_DOFS);
    for (vecIterator.reset(), xIterator.reset(); !vecIterator.end();
	 ++vecIterator, ++xIterator)      
279
      *vecIterator = *xIterator; 
280
281
282
283
284
285
  }


  template<typename T>
  T DOFVector<T>::min() const
  {
286
287
    FUNCNAME("DOFVector<T>::min()");
    
288
    checkFeSpace(this->feSpace, vec);
289
290

    T m;    
291
    Iterator vecIterator(const_cast<DOFIndexed<T>*>(dynamic_cast<const DOFIndexed<T>*>(this)), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
292
    for (vecIterator.reset(), m = *vecIterator; !vecIterator.end(); ++vecIterator)
293
      m = std::min(m, *vecIterator);
294

295
296
297
298
299
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localMin = m;
    MPI::COMM_WORLD.Allreduce(&localMin, &m, 1, MPI_DOUBLE, MPI_MIN);
#endif

300
301
302
    return m;
  }

303

304
305
306
  template<typename T>
  T DOFVector<T>::max() const 
  {
307
308
    FUNCNAME("DOFVector<T>::max()");
    
309
    checkFeSpace(this->feSpace, vec);
310

311
    T m;    
312
    Iterator vecIterator(const_cast<DOFIndexed<T>*>(dynamic_cast<const DOFIndexed<T>*>(this)), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
313
    for (vecIterator.reset(), m = *vecIterator; !vecIterator.end(); ++vecIterator)
314
      m = std::max(m, *vecIterator);
315

316
317
318
319
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localMax = m;
    MPI::COMM_WORLD.Allreduce(&localMax, &m, 1, MPI_DOUBLE, MPI_MAX);
#endif
320

321
    return m;
322
323
  }

324

Thomas Witkowski's avatar
Thomas Witkowski committed
325
326
327
328
329
330
  template<typename T>
  T DOFVector<T>::absMax() const
  {
    return std::max(abs(max()), abs(min()));
  }

331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346

  template<typename T>
  T DOFVector<T>::average() const
  {
    FUNCNAME("DOFVector<T>::average()");
    
    checkFeSpace(this->feSpace, vec);

    int count = 0;
    T m = 0;
    Iterator vecIterator(const_cast<DOFIndexed<T>*>(dynamic_cast<const DOFIndexed<T>*>(this)), USED_DOFS);
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator) {
      m += *vecIterator;
      count++;
    }

347
348
349
350
351
352
353
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localSum = m;
    int localCount = count;
    MPI::COMM_WORLD.Allreduce(&localSum, &m, 1, MPI_DOUBLE, MPI_SUM);
    MPI::COMM_WORLD.Allreduce(&localCount, &count, 1, MPI_INT, MPI_SUM);
#endif

354
355
356
357
    return m / count;
  }


358
359
360
  template<typename T>
  void DOFVector<T>::print() const
  {
361
    FUNCNAME("DOFVector<T>::print()");
Thomas Witkowski's avatar
Thomas Witkowski committed
362

363
    const DOFAdmin *admin = NULL;
364
    const char *format;
365

366
367
    if (this->feSpace) 
      admin = this->feSpace->getAdmin();
368

369
    MSG("Vec `%s':\n", this->name.c_str());
370
    int j = 0;
371
372
373
374
375
376
377
378
379
    if (admin) {
      if (admin->getUsedSize() > 100)
	format = "%s(%3d,%10.5le)";
      else if (admin->getUsedSize() > 10)
	format = "%s(%2d,%10.5le)";
      else
	format = "%s(%1d,%10.5le)";

      Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(this)), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
380
      for (vecIterator.reset(); !vecIterator.end(); ++vecIterator) {
381
	if ((j % 3) == 0) {
382
383
	  if (j) 
	    Msg::print("\n");
384
	  MSG(format, "", vecIterator.getDOFIndex(), *vecIterator);
Thomas Witkowski's avatar
Thomas Witkowski committed
385
	} else {
386
	  Msg::print(format, " ", vecIterator.getDOFIndex(), *vecIterator);
Thomas Witkowski's avatar
Thomas Witkowski committed
387
	}
388
	j++;
389
      }
390
      Msg::print("\n");
391
    } else {
392
393
	MSG("no DOFAdmin, print whole vector.\n");
    
394
395
396
397
398
399
	for (int i = 0; i < static_cast<int>( vec.size()); i++) {
	  if ((j % 3) == 0) {
	    if (j) 
	      Msg::print("\n");
	    MSG("(%d,%10.5e)",i,vec[i]);
	  } else {
400
	    Msg::print(" (%d,%10.5e)",i,vec[i]);
401
	  }
402
403
404
405
406
407
408
	  j++;
	}
	Msg::print("\n");
      }
    return;
  }

409

410
411
412
413
414
415
416
417
418
  template<typename T>
  int DOFVector<T>::calcMemoryUsage() const
  {
    int result = 0;
    result += sizeof(DOFVector<T>);
    result += vec.size() * sizeof(T);

    return result;
  }
419

420

421
422
423
424
  template<typename T>
  T DOFVectorBase<T>::evalUh(const DimVec<double>& lambda, 
			     DegreeOfFreedom* dof_indices)
  {
425
    BasisFunction* phi = const_cast<BasisFunction*>(this->getFeSpace()->getBasisFcts());
426
    int nBasisFcts = phi->getNumber();
427
428
    T val = 0.0;

429
    for (int i = 0; i < nBasisFcts; i++)
430
431
      val += (*this)[dof_indices[i]]*(*phi->getPhi(i))(lambda);

432
433
434
435
436
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localVal = val;
    MPI::COMM_WORLD.Allreduce(&localVal, &val, 1, MPI_DOUBLE, MPI_SUM);
#endif

437
438
439
    return val;
  }

440

441
442
443
  template<typename T>
  void DOFVector<T>::interpol(AbstractFunction<T, WorldVector<double> > *fct)
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
444
    FUNCNAME("DOFVector::interpol()");
445

Thomas Witkowski's avatar
Thomas Witkowski committed
446
    TEST_EXIT_DBG(fct)("No function to interpolate!\n");
447
448

    interFct = fct;
449

450
    if (!this->getFeSpace()) {
451
452
453
454
      MSG("no dof admin in vec %s, skipping interpolation\n",
	  this->getName().c_str());
      return;
    }
455

456
    if (!(this->getFeSpace()->getAdmin())) {
457
      MSG("no dof admin in feSpace %s, skipping interpolation\n", 
458
	  this->getFeSpace()->getName().c_str());
459
460
      return;
    }
461
  
462
    if (!(this->getFeSpace()->getBasisFcts())) {
463
464
465
466
      MSG("no basis functions in admin of vec %s, skipping interpolation\n",
	  this->getName().c_str());
      return;
    }
467

468
469
470
471
472
    if (!(fct)) {
      MSG("function that should be interpolated only pointer to NULL, ");
      Msg::print("skipping interpolation\n");
      return;
    }
473
474
475

    traverseVector = this;

Thomas Witkowski's avatar
Thomas Witkowski committed
476
    TraverseStack stack;
477
    ElInfo *elInfo = stack.traverseFirst(this->getFeSpace()->getMesh(), -1,
Thomas Witkowski's avatar
Thomas Witkowski committed
478
479
480
481
482
					 Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS);
    while (elInfo) {
      interpolFct(elInfo);
      elInfo = stack.traverseNext(elInfo);
    }
483
484
  }

485

486
  template<typename T>
487
  void DOFVector<T>::interpolFct(ElInfo* elinfo)
488
  {
489
490
    const BasisFunction *basFct = traverseVector->getFeSpace()->getBasisFcts();
    const DOFAdmin* admin = traverseVector->getFeSpace()->getAdmin();
491
492
493
    const T *inter_val = 
      const_cast<BasisFunction*>(basFct)->interpol(elinfo, 0, NULL,
						   traverseVector->interFct, NULL);
494

495
    int nBasFcts = basFct->getNumber();
496
497
498
    std::vector<DegreeOfFreedom> myLocalIndices(nBasFcts);
    basFct->getLocalIndices(const_cast<Element*>(elinfo->getElement()), 
			    admin, myLocalIndices);
499
    for (int i = 0; i < nBasFcts; i++)
500
      (*traverseVector)[myLocalIndices[i]] = inter_val[i];
501
502
  }

503

504
505
506
  template<typename T>
  double DOFVector<T>::Int(Quadrature* q) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
507
    FUNCNAME("DOFVector::Int()");
508
  
509
    Mesh* mesh = this->feSpace->getMesh();
510

Thomas Witkowski's avatar
Thomas Witkowski committed
511
    if (!q) {
512
      int deg = 2 * this->feSpace->getBasisFcts()->getDegree();
Thomas Witkowski's avatar
Thomas Witkowski committed
513
514
      q = Quadrature::provideQuadrature(this->dim, deg);
    }
515

516
    FastQuadrature *quadFast = 
517
      FastQuadrature::provideFastQuadrature(this->feSpace->getBasisFcts(), *q, INIT_PHI);
518

519
520
    double result = 0.0;
    int nPoints = quadFast->getNumPoints();
521
    mtl::dense_vector<T> uh_vec(nPoints);
522
    TraverseStack stack;
523
524
    ElInfo *elInfo = stack.traverseFirst(mesh, -1, 
      Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS | Mesh::FILL_DET);
525
526
527
    while (elInfo) {
      double det = elInfo->getDet();
      double normT = 0.0;
528
      this->getVecAtQPs(elInfo, NULL, quadFast, uh_vec);
529
530
531
      for (int iq = 0; iq < nPoints; iq++)
	normT += quadFast->getWeight(iq) * (uh_vec[iq]);
      result += det * normT;
532

533
534
      elInfo = stack.traverseNext(elInfo);
    }
535

536
537
538
539
540
541
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localResult = result;
    MPI::COMM_WORLD.Allreduce(&localResult, &result, 1, MPI_DOUBLE, MPI_SUM);
#endif
    
    return result;
542
543
  }

544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
  template<typename T>
  T integrate_VecAndCoords(const DOFVector<double> &vec,
			   BinaryAbstractFunction<T, double, WorldVector<double> > *fct)
  {
    FUNCNAME("integrate_VecAndCoords()");

    TEST_EXIT(fct)("No function defined!\n");

    int deg = std::max(fct->getDegree(),
		       2 * vec.getFeSpace()->getBasisFcts()->getDegree());
    Quadrature* quad =
      Quadrature::provideQuadrature(vec.getFeSpace()->getMesh()->getDim(), deg);
    FastQuadrature *fastQuad =
      FastQuadrature::provideFastQuadrature(vec.getFeSpace()->getBasisFcts(), *quad, INIT_PHI);

    mtl::dense_vector<double> qp(fastQuad->getNumPoints());
    WorldVector<double> coordsAtQP;

    T value;
    nullify(value);
    
    Flag traverseFlag = Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS | Mesh::FILL_DET;
    TraverseStack stack;
    ElInfo *elInfo = stack.traverseFirst(vec.getFeSpace()->getMesh(), -1, traverseFlag);
    while (elInfo) {
      vec.getVecAtQPs(elInfo, quad, fastQuad, qp);
      T tmp;
      nullify(tmp);
      for (int iq = 0; iq < fastQuad->getNumPoints(); iq++) {
	elInfo->coordToWorld(fastQuad->getLambda(iq), coordsAtQP);
 	tmp += fastQuad->getWeight(iq) * (*fct)(qp[iq], coordsAtQP);
      }
      value += tmp * elInfo->getDet();

      elInfo = stack.traverseNext(elInfo);
    }

// #ifdef HAVE_PARALLEL_DOMAIN_AMDIS
//     double localValue = value;
//     MPI::COMM_WORLD.Allreduce(&localValue, &value, 1, MPI_DOUBLE, MPI_SUM);
// #endif
585

586
587
588
    return value;
  }
  
589
590
591
  template<typename T>
  double DOFVector<T>::L1Norm(Quadrature* q) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
592
    FUNCNAME("DOFVector::L1Norm()");
593
  
594
595
    Mesh* mesh = this->feSpace->getMesh();

Thomas Witkowski's avatar
Thomas Witkowski committed
596
    if (!q) {
597
      int deg = 2 * this->feSpace->getBasisFcts()->getDegree();
Thomas Witkowski's avatar
Thomas Witkowski committed
598
599
      q = Quadrature::provideQuadrature(this->dim, deg);
    }
600

601
602
    FastQuadrature *quadFast = 
      FastQuadrature::provideFastQuadrature(this->feSpace->getBasisFcts(), *q, INIT_PHI);
603

604
605
    double result = 0.0;
    int nPoints = quadFast->getNumPoints();
606
    mtl::dense_vector<T> uh_vec(nPoints);
607
608
609
    TraverseStack stack;
    ElInfo *elInfo = 
      stack.traverseFirst(mesh, -1, 
610
			  Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS | Mesh::FILL_DET);
611
612
613
    while (elInfo) {
      double det = elInfo->getDet();
      double normT = 0.0;
614
      this->getVecAtQPs(elInfo, NULL, quadFast, uh_vec);
615
616
617
      for (int iq = 0; iq < nPoints; iq++)
	normT += quadFast->getWeight(iq) * abs(uh_vec[iq]);
      result += det * normT;
618

619
620
      elInfo = stack.traverseNext(elInfo);
    }
621

622
623
624
625
626
627
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localResult = result;
    MPI::COMM_WORLD.Allreduce(&localResult, &result, 1, MPI_DOUBLE, MPI_SUM);
#endif
    
    return result;
628
629
  }

630

631
632
633
  template<typename T>
  double DOFVector<T>::L2NormSquare(Quadrature* q) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
634
    FUNCNAME("DOFVector::L2NormSquare()");
635

636
637
    Mesh* mesh = this->feSpace->getMesh();

Thomas Witkowski's avatar
Thomas Witkowski committed
638
    if (!q) {
639
      int deg = 2 * this->feSpace->getBasisFcts()->getDegree();
Thomas Witkowski's avatar
Thomas Witkowski committed
640
641
      q = Quadrature::provideQuadrature(this->dim, deg);
    }
642

643
644
    FastQuadrature *quadFast = 
      FastQuadrature::provideFastQuadrature(this->feSpace->getBasisFcts(), *q, INIT_PHI);
645

646
647
    double result = 0.0;
    int nPoints = quadFast->getNumPoints();
648
    mtl::dense_vector<T> uh_vec(nPoints);
649
650
651
    TraverseStack stack;
    ElInfo *elInfo = 
      stack.traverseFirst(mesh, -1, 
652
			  Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS | Mesh::FILL_DET);
653
654
655
    while (elInfo) {
      double det = elInfo->getDet();
      double normT = 0.0;
656
      this->getVecAtQPs(elInfo, NULL, quadFast, uh_vec);
657
658
659
      for (int iq = 0; iq < nPoints; iq++)
	normT += quadFast->getWeight(iq) * sqr(uh_vec[iq]);
      result += det * normT;
660

661
      elInfo = stack.traverseNext(elInfo);
Thomas Witkowski's avatar
Thomas Witkowski committed
662
    }
663

664
665
666
667
668
669
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localResult = result;
    MPI::COMM_WORLD.Allreduce(&localResult, &result, 1, MPI_DOUBLE, MPI_SUM);
#endif
    
    return result;
670
671
  }

672

673
  template<typename T>  
674
675
  double DOFVector<T>::H1NormSquare(Quadrature *q) const
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
676
    FUNCNAME("DOFVector::H1NormSquare()");
677

678
679
    Mesh* mesh = this->feSpace->getMesh();

Thomas Witkowski's avatar
Thomas Witkowski committed
680
    if (!q) {
681
      int deg = 2 * this->feSpace->getBasisFcts()->getDegree() - 2;
Thomas Witkowski's avatar
Thomas Witkowski committed
682
683
684
      q = Quadrature::provideQuadrature(this->dim, deg);
    }

685
686
687
688
689
690
    FastQuadrature *quadFast = 
      FastQuadrature::provideFastQuadrature(this->feSpace->getBasisFcts(), *q, INIT_GRD_PHI);

    double result = 0.0;
    int nPoints = quadFast->getNumPoints();
    int dimOfWorld = Global::getGeo(WORLD);
691
    mtl::dense_vector<WorldVector<T> > grduh_vec(nPoints);
692
693
694
    TraverseStack stack;
    ElInfo *elInfo = 
      stack.traverseFirst(mesh, -1, 
695
696
			  Mesh::CALL_LEAF_EL | Mesh::FILL_COORDS | 
			  Mesh::FILL_DET | Mesh::FILL_GRD_LAMBDA);
697
698
699
    while (elInfo) {
      double det = elInfo->getDet();
      double normT = 0.0;
700
      this->getGrdAtQPs(elInfo, NULL, quadFast, grduh_vec);
701
702
703
704
705
706
707

      for (int iq = 0; iq < nPoints; iq++) {
	double norm2 = 0.0;
	for (int j = 0; j < dimOfWorld; j++)
	  norm2 += sqr(grduh_vec[iq][j]);
	normT += quadFast->getWeight(iq) * norm2;
      }
708

709
      result += det * normT;
710

711
712
      elInfo = stack.traverseNext(elInfo);
    }
713

714
715
716
717
718
719
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    double localResult = result;
    MPI::COMM_WORLD.Allreduce(&localResult, &result, 1, MPI_DOUBLE, MPI_SUM);
#endif
    
    return result;
720
721
  }

722

723
724

  template<typename T>
Praetorius, Simon's avatar
Praetorius, Simon committed
725
  const bool DOFVector<T>::getDOFidxAtPoint(WorldVector<double> &p, DegreeOfFreedom &idx, ElInfo *oldElInfo, bool useOldElInfo) const
726
727
  { FUNCNAME("DOFVector::getDOFidxAtPoint()");

Praetorius, Simon's avatar
Praetorius, Simon committed
728
729
    Mesh *mesh = this->feSpace->getMesh();
    const BasisFunction *basFcts = this->feSpace->getBasisFcts();
730
731
732
733
734
735
736
737
738

    int dim = mesh->getDim();
    int numBasFcts = basFcts->getNumber();

    std::vector<DegreeOfFreedom> localIndices(numBasFcts);
    DimVec<double> lambda(dim, NO_INIT);

    ElInfo *elInfo = mesh->createNewElInfo();
    idx = 0;
Praetorius, Simon's avatar
Praetorius, Simon committed
739
    bool inside = false;
740
741
742
743
744
745
746
747
748
749
750
751
752
753

    if (oldElInfo && useOldElInfo && oldElInfo->getMacroElement()) {
      inside = mesh->findElInfoAtPoint(p, elInfo, lambda, oldElInfo->getMacroElement(), NULL, NULL);
      delete oldElInfo;
    } else {
      inside = mesh->findElInfoAtPoint(p, elInfo, lambda, NULL, NULL, NULL);
    }
    
    if (oldElInfo)
      oldElInfo = elInfo;
    
    if (!inside)
      return false;
    
Praetorius, Simon's avatar
Praetorius, Simon committed
754
    basFcts->getLocalIndices(elInfo->getElement(), this->feSpace->getAdmin(), localIndices);
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
    FixVec<WorldVector<double>, VERTEX> coords = elInfo->getCoords();
    int minIdx = -1;
    double minDist = 1.e15;
    
    for (int i = 0; i < coords.size(); i++) {
      WorldVector<double> dist = coords[i] - p;
      double newDist = norm(dist);
      if (newDist < minDist) {
	minIdx = i;
	minDist = newDist;
      }
    }
    
    if (minIdx >= 0)
      idx = localIndices[minIdx];
    
    if(!oldElInfo) delete elInfo;
    return inside;
  };


776
777
  template<typename T>
  void DOFVector<T>::compressDOFIndexed(int first, int last, 
778
					std::vector<DegreeOfFreedom> &newDOF)
779
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
780
781
    for (int i = first; i <= last; i++)
      if (newDOF[i] >= 0)
Thomas Witkowski's avatar
Thomas Witkowski committed
782
	vec[newDOF[i]] = vec[i];
783
784
  }

785

786
787
788
789
  template<typename T>
  Flag DOFVectorBase<T>::getAssembleFlag()
  {
    Flag fillFlag(0);
Thomas Witkowski's avatar
Thomas Witkowski committed
790
791
    
    for (std::vector<Operator*>::iterator op = this->operators.begin(); 
Thomas Witkowski's avatar
Thomas Witkowski committed
792
	 op != this->operators.end(); ++op)
793
      fillFlag |= (*op)->getFillFlag();
Thomas Witkowski's avatar
Thomas Witkowski committed
794

795
796
797
    return fillFlag;
  }

798

Thomas Witkowski's avatar
Thomas Witkowski committed
799
800
801
802
803
  template<typename T>
  void DOFVectorBase<T>::finishAssembling()
  {
    // call the operatos cleanup procedures
    for (std::vector<Operator*>::iterator it = this->operators.begin();
Thomas Witkowski's avatar
Thomas Witkowski committed
804
	 it != this->operators.end(); ++it)
Thomas Witkowski's avatar
Thomas Witkowski committed
805
806
807
      (*it)->finishAssembling();
  }

808

809
  template<typename T>
Thomas Witkowski's avatar
Thomas Witkowski committed
810
  DOFVector<T>& DOFVector<T>::operator=(const DOFVector<T>& rhs)
811
  {
812
    this->feSpace = rhs.feSpace;
813
    this->dim = rhs.dim;
814
    this->nBasFcts = rhs.nBasFcts;
815
    vec = rhs.vec;
816
    this->elementVector.change_dim(this->nBasFcts);
817
818
819
820
    interFct = rhs.interFct;
    coarsenOperation = rhs.coarsenOperation;
    this->operators = rhs.operators;
    this->operatorFactor = rhs.operatorFactor;
821
    
822
    if (rhs.boundaryManager) {
823
824
825
      if (this->boundaryManager) 
	delete this->boundaryManager; 

826
      this->boundaryManager = new BoundaryManager(*rhs.boundaryManager);
827
    } else {
828
      this->boundaryManager = NULL;
829
830
    }

831
832
833
834
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
    this->rankDofs = rhs.rankDofs;
#endif

835
836
837
    return *this;
  }

838

839
840
841
  template<typename T>
  const DOFVector<T>& operator*=(DOFVector<T>& x, T scal)
  {
842
    FUNCNAME("DOFVector<T>::operator*=(DOFVector<T>& x, T scal)");
843

844
845
    TEST_EXIT_DBG(x.getFeSpace() && x.getFeSpace()->getAdmin())
      ("pointer is NULL: %8X, %8X\n", x.getFeSpace(), x.getFeSpace()->getAdmin());
846

Thomas Witkowski's avatar
Thomas Witkowski committed
847
848
    typename DOFVector<T>::Iterator vecIterator(dynamic_cast<DOFIndexed<T>*>(&x), 
						USED_DOFS);
849
    for (vecIterator.reset(); !vecIterator.end(); ++vecIterator)
850
      (*vecIterator) *= scal; 
851

852
853
854
855
856
857
858
    return x;
  }


  template<typename T>
  const DOFVector<T>& operator+=(DOFVector<T>& x, const DOFVector<T>& y)
  {
859
860
    FUNCNAME("DOFVector<T>::operator+=(DOFVector<T>& x, const DOFVector<T>& y)");
    
861
862
863
864
    TEST_EXIT_DBG(x.getFeSpace() && y.getFeSpace())
      ("feSpace is NULL: %8X, %8X\n", x.getFeSpace(), y.getFeSpace());
    TEST_EXIT_DBG(x.getFeSpace()->getAdmin() &&
	      (x.getFeSpace()->getAdmin() == y.getFeSpace()->getAdmin()))
865
      ("no admin or different admins: %8X, %8X\n",
866
       x.getFeSpace()->getAdmin(), y.getFeSpace()->getAdmin());
867
868
    TEST_EXIT_DBG(x.getSize() == y.getSize())("different sizes\n");
    
869
870
    typename DOFVector<T>::Iterator xIterator(dynamic_cast<DOFIndexed<T>*>(&x), USED_DOFS);
    typename DOFVector<T>::Iterator yIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(&y)), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
871
872
    for (xIterator.reset(), yIterator.reset(); !xIterator.end();
	 ++xIterator, ++yIterator)
873
874
      *xIterator += *yIterator; 

875
876
877
    return x;
  }

878

879
880
881
  template<typename T>
  const DOFVector<T>& operator-=(DOFVector<T>& x, const DOFVector<T>& y)
  {
882
    FUNCNAME("DOFVector<T>::operator-=(DOFVector<T>& x, const DOFVector<T>& y)");
883

884
885
886
887
    TEST_EXIT_DBG(x.getFeSpace() && y.getFeSpace())
      ("feSpace is NULL: %8X, %8X\n", x.getFeSpace(), y.getFeSpace());
    TEST_EXIT_DBG(x.getFeSpace()->getAdmin() &&
	      (x.getFeSpace()->getAdmin() == y.getFeSpace()->getAdmin()))
888
      ("no admin or different admins: %8X, %8X\n",
889
       x.getFeSpace()->getAdmin(), y.getFeSpace()->getAdmin());
890
891
    TEST_EXIT_DBG(x.getSize() == y.getSize())("different sizes\n");
    
892
893
    typename DOFVector<T>::Iterator xIterator(dynamic_cast<DOFIndexed<T>*>(&x), USED_DOFS);
    typename DOFVector<T>::Iterator yIterator(dynamic_cast<DOFIndexed<T>*>(const_cast<DOFVector<T>*>(&y)), USED_DOFS);
Thomas Witkowski's avatar
Thomas Witkowski committed
894
895
    for (xIterator.reset(), yIterator.reset(); !xIterator.end();
	 ++xIterator, ++yIterator)
896
      *xIterator -= *yIterator; 
Thomas Witkowski's avatar
Thomas Witkowski committed
897

898
899
900
    return x;
  }

901

902
903
904
  template<typename T>
  const DOFVector<T>& operator*=(DOFVector<T>& x, const DOFVector<T>& y)
  {
905
906
    FUNCNAME("DOFVector<T>::operator*=(DOFVector<T>& x, const DOFVector<T>& y)");
    
907
908
909
910
    TEST_EXIT_DBG(x.getFeSpace() && y.getFeSpace())
      ("feSpace is NULL: %8X, %8X\n", x.getFeSpace(), y.getFeSpace());
    TEST_EXIT_DBG(x.getFeSpace()->getAdmin() &&
	      (x.getFeSpace()->getAdmin() == y.getFeSpace()->getAdmin()))
911
      ("no admin or different admins: %8X, %8X\n",