ProblemInstat.cc 5.17 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
#include "ProblemInstat.h"
#include "FileWriter.h"
#include "AdaptStationary.h"
#include "AdaptInstationary.h"
#include "Estimator.h"
#include "ProblemScal.h"
#include "StandardProblemIteration.h"

namespace AMDiS {

  ProblemInstat::~ProblemInstat()
12
  {}
13
14
15
16

  void ProblemInstat::initialize(Flag initFlag,
				 ProblemInstat *adoptProblem/* = NULL*/,
				 Flag adoptFlag /* = INIT_NOTHING*/) 
17
  {}
18
19
20
21

  void ProblemInstat::solveInitialProblem(AdaptInfo *adaptInfo)
  {
    AdaptStationary initialAdapt((name + "->initial->adapt").c_str(),
Thomas Witkowski's avatar
Thomas Witkowski committed
22
				 new StandardProblemIteration(initialProblem),
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
				 adaptInfo);

    initialAdapt.adapt();
  }

  void ProblemInstatScal::transferInitialSolution(AdaptInfo *adaptInfo)
  {
    TEST_EXIT(adaptInfo->getTime() == 
	      adaptInfo->getStartTime())
      ("after initial solution: time != start time\n");
    problemStat->writeFiles(adaptInfo, true);
  }

  void ProblemInstatVec::transferInitialSolution(AdaptInfo *adaptInfo)
  {
    TEST_EXIT(adaptInfo->getTime() == 
	      adaptInfo->getStartTime())
      ("after initial solution: time != start time\n");
    problemStat->writeFiles(adaptInfo, true);
  }


45
  ProblemInstatScal::ProblemInstatScal(std::string name_, 
46
47
48
				       ProblemScal *prob,
				       ProblemStatBase *initialProb)
    : ProblemInstat(name_, initialProb), 
49
50
      problemStat(prob),
      oldSolution(NULL)
51
  {}
52
53
54
55


  ProblemInstatScal::~ProblemInstatScal()
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
56
    delete oldSolution;
57
58
59
60
61
62
63
64
65
66
67
68
69
70
  }

  void ProblemInstatScal::initialize(Flag initFlag,
				     ProblemInstat *adoptProblem/* = NULL*/,
				     Flag adoptFlag /* = INIT_NOTHING*/) 
  {
    FUNCNAME("ProblemInstat::initialize()");
  
    ProblemInstat::initialize(initFlag, adoptProblem, adoptFlag);

    // === create vector for old solution ===
    if (oldSolution) {
      WARNING("oldSolution already created\n");
    } else {
71
      if (initFlag.isSet(INIT_UH_OLD))
72
	createUhOld();
73
      
74
      if (adoptProblem && adoptFlag.isSet(INIT_UH_OLD)) {
75
76
77
78
	ProblemInstatScal* _adoptProblem 
	  = dynamic_cast<ProblemInstatScal*>(adoptProblem);
	TEST_EXIT(_adoptProblem)
	  ("can't adopt oldSolution from problem which is not instationary and scalar");
79
80
81
82
83
84
85
86
87
	TEST_EXIT(!oldSolution)("oldSolution already created");
	oldSolution = _adoptProblem->getOldSolution();
      }
    }

    if (!oldSolution) 
      WARNING("no oldSolution created\n");
  }

Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
88
89
  void ProblemInstatScal::createUhOld() 
  {
90
91
92
93
    if (oldSolution) {
      WARNING("oldSolution already created\n");
    } else {
      // create oldSolution
Thomas Witkowski's avatar
Thomas Witkowski committed
94
      oldSolution = new DOFVector<double>(problemStat->getFESpace(), name + "->uOld");
95
      oldSolution->setCoarsenOperation(COARSE_INTERPOL);
Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
96
      if (problemStat->getEstimator())
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
	dynamic_cast<Estimator*>(problemStat->getEstimator())
	  ->addUhOldToSystem(0, oldSolution);
    }
  }


  void ProblemInstatScal::closeTimestep(AdaptInfo *adaptInfo) 
  {
    bool force = (adaptInfo->getTime() >= adaptInfo->getEndTime());
    problemStat->writeFiles(adaptInfo, force);
  }

  void ProblemInstatVec::closeTimestep(AdaptInfo *adaptInfo) 
  {
    bool force = (adaptInfo->getTime() >= adaptInfo->getEndTime());
    problemStat->writeFiles(adaptInfo, force);
  }

115
  ProblemInstatVec::ProblemInstatVec(std::string name_, 
116
117
118
				     ProblemVec *prob,
				     ProblemStatBase *initialProb)
    : ProblemInstat(name_, initialProb), 
119
120
      problemStat(prob),
      oldSolution(NULL)
Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
121
  {}
122
123
124

  ProblemInstatVec::~ProblemInstatVec()
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
125
    delete oldSolution;
126
127
128
129
130
131
132
133
  }

  void ProblemInstatVec::initialize(Flag initFlag,
				    ProblemInstat *adoptProblem/* = NULL*/,
				    Flag adoptFlag /* = INIT_NOTHING*/) 
  {
    FUNCNAME("ProblemInstatVec::initialize()");
  
134
    ProblemInstat::initialize(initFlag, adoptProblem, adoptFlag);
135
136

    // === create vector for old solution ===
Thomas Witkowski's avatar
Thomas Witkowski committed
137
    if (oldSolution) {
138
139
      WARNING("oldSolution already created\n");
    } else {
Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
140
      if (initFlag.isSet(INIT_UH_OLD))
141
	createUhOld();
Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
142

Thomas Witkowski's avatar
Thomas Witkowski committed
143
144
      if (adoptProblem && adoptFlag.isSet(INIT_UH_OLD)) {
	ProblemInstatVec* _adoptProblem = dynamic_cast<ProblemInstatVec*>(adoptProblem);
145
146
	TEST_EXIT(_adoptProblem)
	  ("can't adopt oldSolution from problem which is not instationary and vectorial");
147
148
149
150
151
	TEST_EXIT(!oldSolution)("oldSolution already created");
	oldSolution = _adoptProblem->getOldSolution();
      }
    }

152
153
    if (!oldSolution) 
      WARNING("no oldSolution created\n");
154
155
  }

Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
156
157
  void ProblemInstatVec::createUhOld() 
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
158
159
160
    if (oldSolution) {
      WARNING("oldSolution already created\n");
    } else {
161
162
      int size = problemStat->getNumComponents();
      // create oldSolution
Thomas Witkowski's avatar
Thomas Witkowski committed
163
      oldSolution = new SystemVector("old solution",
164
165
				     problemStat->getFESpaces(), 
				     size);
Thomas Witkowski's avatar
Thomas Witkowski committed
166
      for (int i = 0; i < size; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
167
	oldSolution->setDOFVector(i, new DOFVector<double>(problemStat->getFESpace(i), 
168
169
170
							   name + "->uOld"));
	oldSolution->getDOFVector(i)->setCoarsenOperation(COARSE_INTERPOL);
      
Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
171
172
173
	if (problemStat->getEstimator(i))
	  problemStat->getEstimator(i)->
	    addUhOldToSystem(i, oldSolution->getDOFVector(i));
174
175
176
177
      }
    }
  }

Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
178
179
  void ProblemInstatScal::initTimestep(AdaptInfo *adaptInfo) 
  {
180
181
182
    oldSolution->copy(*(problemStat->getSolution())); 
  }

Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
183
184
  void ProblemInstatVec::initTimestep(AdaptInfo *adaptInfo) 
  {
185
186
    oldSolution->copy(*(problemStat->getSolution())); 
  }
Thomas Witkowski's avatar
nix.  
Thomas Witkowski committed
187

188
}