ParallelDomainVec.cc 2.42 KB
Newer Older
1
2
3
#include "ParallelDomainVec.h"
#include "ProblemVec.h"
#include "ProblemInstat.h"
Thomas Witkowski's avatar
Thomas Witkowski committed
4
#include "SystemVector.h"
5
#include "Serializer.h"
6
7
8

namespace AMDiS {

9
  ParallelDomainVec::ParallelDomainVec(ProblemVec *problem,
10
				       ProblemInstatVec *problemInstat)
11
    : ParallelDomainBase(problem, 
12
13
14
15
16
17
18
19
20
21
			 problemInstat, 
			 problem->getFESpace(0),
			 problem->getRefinementManager(0)),
      probVec(problem)
  {
    FUNCNAME("ParallelDomainVec::ParallelDomainVec()");

    info = probVec->getInfo();
    nComponents = probVec->getNumComponents();

22
23
    // Test if all fe spaces are equal. Yet, different fe spaces are not supported for
    // domain parallelization.
24
25
26
27
    const FiniteElemSpace *fe = probVec->getFESpace(0);
    for (int i = 0; i < nComponents; i++)
      TEST_EXIT(fe == probVec->getFESpace(i))
	("Parallelization does not supported different FE spaces!\n");
28
29
30
31
32
33

    // Create parallel serialization file writer, if needed.
    int writeSerialization = 0;
    GET_PARAMETER(0, name + "->output->write serialization", "%d", &writeSerialization);
    if (writeSerialization)
      problem->getFileWriterList().push_back(new Serializer<ParallelDomainVec>(this));
34
35
36
37
38
39
40
41
42
  }


  void ParallelDomainVec::initParallelization(AdaptInfo *adaptInfo)
  {
    FUNCNAME("ParallelDomainVec::initParallelization()");

    ParallelDomainBase::initParallelization(adaptInfo);

Thomas Witkowski's avatar
Thomas Witkowski committed
43
    for (int i = 0; i < nComponents; i++) {
44
      for (int j = 0; j < nComponents; j++)
45
46
47
48
49
 	if (probVec->getSystemMatrix(i, j))
 	  probVec->getSystemMatrix(i, j)->setRankDofs(isRankDof);

      TEST_EXIT_DBG(probVec->getRHS()->getDOFVector(i))("No rhs vector!\n");
      TEST_EXIT_DBG(probVec->getSolution()->getDOFVector(i))("No solution vector!\n");
Thomas Witkowski's avatar
Thomas Witkowski committed
50
51
52
53

      probVec->getRHS()->getDOFVector(i)->setRankDofs(isRankDof);
      probVec->getSolution()->getDOFVector(i)->setRankDofs(isRankDof);
    }
54
55
  }

56

57
58
59
60
61
62
63
64
65
66
  void ParallelDomainVec::solve()
  {
    FUNCNAME("ParallelDomainVec::solve()");

#ifdef _OPENMP
    double wtime = omp_get_wtime();
#endif
    clock_t first = clock();

    fillPetscMatrix(probVec->getSystemMatrix(), probVec->getRHS());      
67
    solvePetscMatrix(*(probVec->getSolution()));
68
69
70
71
72
73
74
75

#ifdef _OPENMP
    INFO(info, 8)("solution of discrete system needed %.5f seconds system time / %.5f seconds wallclock time\n",
		   TIME_USED(first, clock()),
		   omp_get_wtime() - wtime);
#else
    INFO(info, 8)("solution of discrete system needed %.5f seconds\n",
		   TIME_USED(first, clock()));
76
#endif    
77
78
79
  }

}