BaseProblem.hh 9.66 KB
Newer Older
1
2
3
using namespace AMDiS;

template<typename ProblemType>
4
BaseProblem<ProblemType>::BaseProblem(const std::string &name_, bool createProblem) 
5
6
7
8
9
  : ProblemInstatBase(name_,NULL),
    prob(NULL),
    secureIteration(false),
    dim(1),
    dow(1),
10
11
    nTimesteps(-1),
    oldMeshChangeIdx(0),
12
    oldTimestep(0.0)
13
14
{
  // create basic problems
15
16
17
  if (createProblem)
    prob = new ProblemType(name + "->space");
  
18
  dow = Global::getGeo(WORLD);
19
  
20
  Initfile::get(name + "->secure iteration", secureIteration);
21
}
22
23
24
25


template<typename ProblemType>
void BaseProblem<ProblemType>::initialize(Flag initFlag,
26
27
					  ProblemStat *adoptProblem,
					  Flag adoptFlag)
28
{ FUNCNAME("BaseProblem::initialize()");
29
30

  TEST_EXIT(prob)("Problem must be created! Add flag createProblem=true to constructor!\n");
31
  
32
33
  prob->initialize(initFlag, adoptProblem, adoptFlag);
  dim = getMesh()->getDim();
34
}
35
36
37
38


template<typename ProblemType>
Flag BaseProblem<ProblemType>::initDataFromFile(AdaptInfo *adaptInfo)
39
40
41
{ 
  FUNCNAME("BaseProblem::initDataFromFile()");
  
42
  Flag initFlag;
43
  bool readDataFromFile = false;
44
45
46
  Initfile::get(name + "->read data from file", readDataFromFile, 2);
  if (!readDataFromFile)
    return initFlag;
47
  
48
49
  std::string readFormat = "arh";
  Initfile::get(name + "->read format", readFormat, 2);
Praetorius, Simon's avatar
Praetorius, Simon committed
50
  if (readFormat != "arh" && readFormat != "dat" && readFormat != "vtu" && readFormat != "multi-vtu") {
51
52
    WARNING("You can not read data from formats other than .arh, .dat or .vtu! The .arh-format is selected.\n");
  }
53
  
54
55
  // read data and mesh from arh-files/dat-files
  MSG("read data from file...\n");
56
57
58
59
60
61
  if (readFormat == "arh") {
    std::string filename = "";
    Initfile::get(name + "->value file", filename);
    if (filename.size() == 0) return initFlag;
    if (!file_exists(filename))
      throw(std::runtime_error("The file '" + filename + "' does not exist!"));
62
63
    
    std::vector<DOFVector<double>*> solutions;
64
    for (int i = 0; i < prob->getNumComponents(); i++)
65
      solutions.push_back(prob->getSolution()->getDOFVector(i));
66
67
68
    ArhReader::read(filename, prob->getMesh(), solutions);
  }
  else if (readFormat == "dat") {
69
70
71
72
73
74
    bool preserveMacroFileInfo = false;
    Parameters::get(prob->getMesh()->getName() + "->preserve macroFileInfo", preserveMacroFileInfo);
    if (prob->getMesh()->getMacroFileInfo() == NULL || !preserveMacroFileInfo)
      throw(std::runtime_error("Yout have to set the flag 'mesh_name->preserve macroFileInfo' to 'true' ("+boost::lexical_cast<std::string>(preserveMacroFileInfo)+"), in order to read .dat-files"));
    
    std::string filename = "";
75
    for (int i = 0; i < prob->getNumComponents(); i++) {
76
77
78
79
80
      Parameters::get(name + "->value file["+boost::lexical_cast<std::string>(i)+"]",filename);
      if (!file_exists(filename))
	throw(std::runtime_error("The file '" + filename + "'does not exist!"));
      ValueReader::readValue(filename,prob->getMesh(),prob->getSolution()->getDOFVector(i),prob->getMesh()->getMacroFileInfo());
    }
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
  }
  else if (readFormat == "vtu") {
    std::vector<std::string> filenames;
    Initfile::get(name + "->value file", filenames);
    if (filenames.size() == 0)
      return initFlag;
    int arh_idx = -1, vtu_idx = -1;
    for (size_t i = 0; i < filenames.size(); i++) {
      if (!file_exists(filenames[i]))
	throw(std::runtime_error("The file '" + filenames[i] + "' does not exist!"));
      if (filenames[i].find(".vtu") != std::string::npos)
	vtu_idx = i;
      else if (filenames[i].find(".arh") != std::string::npos)
	arh_idx = i;
      else
	throw(std::runtime_error("The file '" + filenames[i] + "' must have the extension .arh or .vtu!"));
    }
    if (arh_idx >= 0) {
      ArhReader::read(filenames[arh_idx], prob->getMesh());
    }
Praetorius, Simon's avatar
Praetorius, Simon committed
101
102
103
104
    if (vtu_idx >= 0) {      
      std::vector<int> readComponents;
      Initfile::get(name + "->read components", readComponents);
      if (readComponents.size() == 0)
105
	for (int i = 0; i < prob->getNumComponents(); i++)
Praetorius, Simon's avatar
Praetorius, Simon committed
106
107
	  readComponents.push_back(i);
	
108
109
      std::vector<DOFVector<double>*> solutions;
      std::vector<std::string> names;
Praetorius, Simon's avatar
Praetorius, Simon committed
110
111
112
      for (size_t i = 0; i < readComponents.size(); i++) {
	solutions.push_back(prob->getSolution()->getDOFVector(readComponents[i]));
	names.push_back(prob->getComponentName(readComponents[i]));
113
114
115
116
117
      }
      VtuReader::readValue(filenames[vtu_idx], prob->getMesh(), solutions, names);
    } else
      throw(std::runtime_error("You have to specify a .vtu file!"));
  }
Praetorius, Simon's avatar
Praetorius, Simon committed
118
119
120
  else if (readFormat == "multi-vtu") {
    size_t numFiles = 0;
    Initfile::get(name + "->number of files", numFiles);
121
    for (size_t n = 0; n < numFiles; n++) {
Praetorius, Simon's avatar
Praetorius, Simon committed
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
      std::vector<std::string> filenames;
      Initfile::get(name + "->value file[" + boost::lexical_cast<std::string>(n) + "]", filenames);
      if (filenames.size() == 0)
	return initFlag;
      int arh_idx = -1, vtu_idx = -1;
      for (size_t i = 0; i < filenames.size(); i++) {
	if (!file_exists(filenames[i]))
	  throw(std::runtime_error("The file '" + filenames[i] + "' does not exist!"));
	if (filenames[i].find(".vtu") != std::string::npos)
	  vtu_idx = i;
	else if (filenames[i].find(".arh") != std::string::npos)
	  arh_idx = i;
	else
	  throw(std::runtime_error("The file '" + filenames[i] + "' must have the extension .arh or .vtu!"));
      }
      if (arh_idx >= 0) {
	ArhReader::read(filenames[arh_idx], prob->getMesh());
      }
      if (vtu_idx >= 0) {      
	std::vector<int> readComponents;
	Initfile::get(name + "->read components[" + boost::lexical_cast<std::string>(n) + "]", readComponents);
	if (readComponents.size() == 0)
144
	  for (int i = 0; i < prob->getNumComponents(); i++)
Praetorius, Simon's avatar
Praetorius, Simon committed
145
146
147
148
149
150
151
152
153
154
155
156
157
	    readComponents.push_back(i);
	  
	std::vector<DOFVector<double>*> solutions;
	std::vector<std::string> names;
	for (size_t i = 0; i < readComponents.size(); i++) {
	  solutions.push_back(prob->getSolution()->getDOFVector(readComponents[i]));
	  names.push_back(prob->getComponentName(readComponents[i]));
	}
	VtuReader::readValue(filenames[vtu_idx], prob->getMesh(), solutions, names);
      } else
	throw(std::runtime_error("You have to specify a .vtu file!"));
    }
  }
158
159
160
  else {
    throw(std::runtime_error("Parameter 'read data from file' set to 'true', but no parameter 'read format' specified!"));
  }
161
162
163
164
165
  
  initFlag.setFlag(DATA_ADOPTED);
  initFlag.setFlag(MESH_ADOPTED);

  return initFlag;
166
}
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189


template<typename ProblemType>
void BaseProblem<ProblemType>::beginIteration(AdaptInfo *adaptInfo) 
{ FUNCNAME("BaseProblem::beginIteration()");
	
  MSG("\n");
  MSG(("[[ <"+name+"> iteration ]]\n").c_str());
    
  // assemble some blocks only once, if some conditions are fullfilled
  #ifdef HAVE_PARALLEL_DOMAIN_AMDIS
  bool fixedMatrixCell = false;
  #else
  bool fixedMatrixCell = adaptInfo->getTimestepNumber() > 0
    && std::abs(adaptInfo->getTimestep() - oldTimestep) < DBL_TOL
    && oldMeshChangeIdx == getMesh()->getChangeIndex();
  #endif
  for(size_t i = 0; i < fixedMatrixTimestep.size(); ++i) {
    prob->setAssembleMatrixOnlyOnce(
      fixedMatrixTimestep[i].first, 
      fixedMatrixTimestep[i].second,
      fixedMatrixCell);
  }
190
}
191
192
193
194
195


template<typename ProblemType>
Flag BaseProblem<ProblemType>::oneIteration(AdaptInfo *adaptInfo, 
                                Flag toDo) 
196
197
{ 
  FUNCNAME("BaseProblem::oneIteration()");
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216

  Flag flag, markFlag;
  
  if (toDo.isSet(MARK))
    markFlag = prob->markElements(adaptInfo);
  else
    markFlag = 3;

  // refine
  if (toDo.isSet(ADAPT) && markFlag.isSet(MESH_REFINED))
    flag = prob->refineMesh(adaptInfo);  
  // coarsen
  if (toDo.isSet(ADAPT) && markFlag.isSet(MESH_COARSENED))
    flag |= prob->coarsenMesh(adaptInfo);

  #ifdef HAVE_PARALLEL_DOMAIN_AMDIS
  MPI::COMM_WORLD.Barrier();
  #endif

217
218
#ifndef NONLIN_PROBLEM
  if (toDo.isSet(BUILD))
219
    prob->buildAfterCoarsen(adaptInfo, markFlag, true, true);
220
#endif
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249

  if (toDo.isSet(SOLVE)) {
    if (secureIteration) {
      try {
	prob->solve(adaptInfo);
      } catch(std::exception &e) {
	WARNING(("Could not solve system. Simulation will be stoped here! ERROR: "
	  + std::string(e.what()) + "\n").c_str());
	adaptInfo->setTime(adaptInfo->getEndTime());
	adaptInfo->setTimestepNumber(adaptInfo->getNumberOfTimesteps());
	return flag;
      } catch(...) {
	WARNING("Could not solve system. Simulation will be stoped here! Unknown ERROR\n");
	adaptInfo->setTime(adaptInfo->getEndTime());
	adaptInfo->setTimestepNumber(adaptInfo->getNumberOfTimesteps());
	return flag;
      }
    } else {
      prob->solve(adaptInfo);
    }
  }

  if (toDo.isSet(ESTIMATE))
    prob->estimate(adaptInfo);

  oldTimestep = adaptInfo->getTimestep();
  oldMeshChangeIdx = getMesh()->getChangeIndex();

  return flag;
250
}
251
252
253
254
255
256
257
258


template<typename ProblemType>
void BaseProblem<ProblemType>::endIteration(AdaptInfo *adaptInfo) 
{ FUNCNAME("BaseProblem::endIteration()");
	
  MSG("\n");
  MSG(("[[ end of <"+name+"> iteration ]]\n").c_str());
259
}
260
261
262
263


template<typename ProblemType>
void BaseProblem<ProblemType>::closeTimestep(AdaptInfo *adaptInfo)
264
{
265
266
267
268
269
270
  int outputPeriod = 1;
  Initfile::get("user parameter->write every ith timestep", outputPeriod);
  if (adaptInfo->getTimestepNumber() % outputPeriod == 0
    && adaptInfo->getStartTime() < adaptInfo->getEndTime()) {
    writeFiles(adaptInfo, false);
  }
271
}
272
273
274
275


template<typename ProblemType>
void BaseProblem<ProblemType>::addTimeOperator(ProblemStat *prob, int i, int j)
276
{
277
278
279
280
281
282
283
  Operator *op = new Operator(prob->getFeSpace(i), prob->getFeSpace(j));
    op->addZeroOrderTerm(new Simple_ZOT);
  Operator *opRhs = new Operator(prob->getFeSpace(i));
    opRhs->addZeroOrderTerm(new VecAtQP_ZOT(prob->getSolution()->getDOFVector(j), NULL));

  prob->addMatrixOperator(op, i, j, getInvTau(), getInvTau());
  prob->addVectorOperator(opRhs, i, getInvTau(), getInvTau());
284
}
285
286
287
288


template<typename ProblemType>
void BaseProblem<ProblemType>::addTimeOperator(RosenbrockStationary *prob, int i, int j)
289
{
290
  prob->addTimeOperator(i,j);
291
}