ProblemStat.cc 52.1 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
/******************************************************************************
 *
 * AMDiS - Adaptive multidimensional simulations
 *
 * Copyright (C) 2013 Dresden University of Technology. All Rights Reserved.
 * Web: https://fusionforge.zih.tu-dresden.de/projects/amdis
 *
 * Authors: 
 * Simon Vey, Thomas Witkowski, Andreas Naumann, Simon Praetorius, et al.
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 *
 * This file is part of AMDiS
 *
 * See also license.opensource.txt in the distribution.
 * 
 ******************************************************************************/
20
21


Thomas Witkowski's avatar
Thomas Witkowski committed
22
#include <sstream>
Thomas Witkowski's avatar
Thomas Witkowski committed
23
#include <boost/lexical_cast.hpp>
24
25

#include "ProblemStat.h"
26
27
28
29
30
31
32
33
#include "Serializer.h"
#include "AbstractFunction.h"
#include "Operator.h"
#include "SystemVector.h"
#include "DOFMatrix.h"
#include "FiniteElemSpace.h"
#include "Marker.h"
#include "AdaptInfo.h"
34
#include "io/FileWriter.h"
35
36
#include "CoarseningManager.h"
#include "RefinementManager.h"
Thomas Witkowski's avatar
Thomas Witkowski committed
37
#include "DualTraverse.h"
38
#include "Mesh.h"
39
#include "solver/LinearSolverInterface.h"
40
41
42
43
#include "DirichletBC.h"
#include "RobinBC.h"
#include "PeriodicBC.h"
#include "Lagrange.h"
44
#include "Bubble.h"
Thomas Witkowski's avatar
Thomas Witkowski committed
45
#include "Flag.h"
46
#include "est/Estimator.h"
47
48
#include "io/VtkWriter.h"
#include "io/ValueReader.h"
49
#include "ProblemStatDbg.h"
Thomas Witkowski's avatar
Thomas Witkowski committed
50
#include "Debug.h"
51
52
53

namespace AMDiS {

54
  using namespace std;
55
56
  using boost::lexical_cast;

57
58
  ProblemStatSeq::ProblemStatSeq(string nameStr,
				 ProblemIterationInterface *problemIteration)
59
    : name(nameStr),
60
      nComponents(-1),
61
      nAddComponents(0),
62
63
      nMeshes(0),
      traverseInfo(0),
64
65
66
67
      solver(NULL),
      solution(NULL),
      rhs(NULL),
      systemMatrix(NULL),
68
      useGetBound(true),
69
70
      refinementManager(NULL),
      coarseningManager(NULL),
71
72
73
74
      info(10),
      deserialized(false),
      computeExactError(false),
      boundaryConditionSet(false),
75
76
77
      writeAsmInfo(false),
      solutionTime(0.0),
      buildTime(0.0)
78
79
  {
    Parameters::get(name + "->components", nComponents);
80
    Parameters::get(name + "->additional components", nAddComponents);
81
82
    TEST_EXIT(nComponents > 0)("No value set for parameter \"%s->components\"!\n",
			       name.c_str());    
83
84
    TEST_EXIT(nAddComponents >= 0)("Wrong parameter \"%s->additional components\"!\n",
				    name.c_str());
85
86
    estimator.resize(nComponents, NULL);
    marker.resize(nComponents, NULL);
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
    
    assembleMatrixOnlyOnce.resize(nComponents);
    assembledMatrix.resize(nComponents);
    for (int i = 0; i < nComponents; i++) {
      assembleMatrixOnlyOnce[i].resize(nComponents);
      assembledMatrix[i].resize(nComponents);
      for (int j = 0; j < nComponents; j++) {
	assembleMatrixOnlyOnce[i][j] = false;
	assembledMatrix[i][j] = false;
      }
    }
    
    exactSolutionFcts.resize(nComponents);


    // === Initialize name of components. ===

    componentNames.resize(nComponents, "");
    for (int i = 0; i < nComponents; i++)
      componentNames[i] = "solution[" + lexical_cast<string>(i) + "]";

    Parameters::get(name + "->name", componentNames);
109
110
    componentNames.resize(nComponents);

111
112
113
    for (int i = 0; i < nComponents; i++)
      Parameters::get(name + "->name[" + lexical_cast<string>(i) + "]",
		      componentNames[i]);
Thomas Witkowski's avatar
Thomas Witkowski committed
114
115

    Parameters::get("debug->write asm info", writeAsmInfo);
116
117
  }

Thomas Witkowski's avatar
Thomas Witkowski committed
118
119
120

  ProblemStatSeq::~ProblemStatSeq()
  {
121
122
    if (rhs)
      delete rhs;
123
    rhs = NULL;
124
125
    if (solution)
      delete solution;
126
    solution = NULL;
127
128
129
130
    
    if (solver)
      delete solver;
    solver = NULL;
Thomas Witkowski's avatar
Thomas Witkowski committed
131

132
133
134
135
136
    if (systemMatrix) {    
      for (int i = 0; i < nComponents; i++)
	for (int j = 0; j < nComponents; j++)
	  if ((*systemMatrix)[i][j]) {
	    delete (*systemMatrix)[i][j];
137
	    (*systemMatrix)[i][j] = NULL;
138
139
140
	  }

      delete systemMatrix;
141
      systemMatrix = NULL;
142
    }
Thomas Witkowski's avatar
Thomas Witkowski committed
143
144

    for (unsigned int i = 0; i < meshes.size(); i++)
145
146
      if (meshes[i]) {
// 	delete meshes[i];
147
// 	meshes[i] = NULL;
148
      }
Thomas Witkowski's avatar
Thomas Witkowski committed
149
150
    
    for (unsigned int i = 0; i < estimator.size(); i++)
151
      if (estimator[i]) {
152
	delete estimator[i];
153
	estimator[i] = NULL;
154
      }
Thomas Witkowski's avatar
Thomas Witkowski committed
155
156

    for (unsigned int i = 0; i < marker.size(); i++)
157
      if (marker[i]) {
158
	delete marker[i];
159
	marker[i] = NULL;
160
      }
Thomas Witkowski's avatar
Thomas Witkowski committed
161
  }
162
  
163
164
165
  void ProblemStatSeq::initialize(Flag initFlag,
				  ProblemStatSeq *adoptProblem,
				  Flag adoptFlag)
166
  {
167
    FUNCNAME("ProblemStat::initialize()");
168

169
    // === create meshes ===
Thomas Witkowski's avatar
Thomas Witkowski committed
170
    if (meshes.size() != 0) { 
171
172
173
      WARNING("meshes already created\n");
    } else {
      if (initFlag.isSet(CREATE_MESH) || 
Thomas Witkowski's avatar
Thomas Witkowski committed
174
	  (!adoptFlag.isSet(INIT_MESH) &&
Thomas Witkowski's avatar
Thomas Witkowski committed
175
	   (initFlag.isSet(INIT_SYSTEM) || initFlag.isSet(INIT_FE_SPACE))))
176
	createMesh();      
Thomas Witkowski's avatar
Thomas Witkowski committed
177

178
      if (adoptProblem && 
Thomas Witkowski's avatar
Thomas Witkowski committed
179
	  (adoptFlag.isSet(INIT_MESH) ||
180
181
	   adoptFlag.isSet(INIT_SYSTEM) ||
	   adoptFlag.isSet(INIT_FE_SPACE))) {
Thomas Witkowski's avatar
Thomas Witkowski committed
182
	meshes = adoptProblem->getMeshes();
Thomas Witkowski's avatar
Thomas Witkowski committed
183
	componentMeshes = adoptProblem->componentMeshes;
184
185
186
      
	TEST_EXIT(adoptProblem->getNumAddComponents() + nAddComponents > 0)
	  ("Not yet implemented.\n");
187
188
	// If the adopt problem has fewer components than this problem, but only one
	// mesh for all component, than scal up the componentMeshes array.
Thomas Witkowski's avatar
Thomas Witkowski committed
189
	if (adoptProblem->getNumComponents() < nComponents) {
Thomas Witkowski's avatar
Thomas Witkowski committed
190
	  TEST_EXIT(meshes.size() == 1)("Daran muss ich noch arbeiten!\n");
191
	  
Thomas Witkowski's avatar
Thomas Witkowski committed
192
	  componentMeshes.resize(nComponents);
193
	  for (int i = adoptProblem->getNumComponents(); i < nComponents; i++)
Thomas Witkowski's avatar
Thomas Witkowski committed
194
	    componentMeshes[i] = componentMeshes[0];
195
	}
196
197
198
      }
    }

Thomas Witkowski's avatar
Thomas Witkowski committed
199
    if (meshes.size() == 0) 
200
201
      WARNING("no mesh created\n");

202
    // === create refinement/corasening-manager ===
203
    if (refinementManager != NULL && coarseningManager != NULL) { 
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
      WARNING("refinement-/coarseningmanager already created\n");
    } else {
      if (initFlag.isSet(CREATE_MESH) || 
          (!adoptFlag.isSet(INIT_MESH) &&
           (initFlag.isSet(INIT_SYSTEM) || initFlag.isSet(INIT_FE_SPACE))))
        createRefCoarseManager();

      if (adoptProblem && 
          (adoptFlag.isSet(INIT_MESH) ||
           adoptFlag.isSet(INIT_SYSTEM) ||
           adoptFlag.isSet(INIT_FE_SPACE))) {
        refinementManager = adoptProblem->refinementManager;
        coarseningManager = adoptProblem->coarseningManager;
      }
    }

220
    if (refinementManager == NULL || coarseningManager == NULL) 
221
222
      WARNING("no refinement-/coarseningmanager created\n");

223
    // === create fespace ===
224
    if (feSpaces.size() != 0) {
225
226
227
      WARNING("feSpaces already created\n");
    } else {
      if (initFlag.isSet(INIT_FE_SPACE) || 
228
	  (initFlag.isSet(INIT_SYSTEM) && !adoptFlag.isSet(INIT_FE_SPACE)))
229
	createFeSpace(NULL);
230

231
232
      if (adoptProblem &&
	  (adoptFlag.isSet(INIT_FE_SPACE) || adoptFlag.isSet(INIT_SYSTEM))) {
233
	feSpaces = adoptProblem->getFeSpaces();
234
	componentSpaces = adoptProblem->componentSpaces;
Thomas Witkowski's avatar
Thomas Witkowski committed
235
	traverseInfo = adoptProblem->traverseInfo;
236
237
238
      
	TEST_EXIT(adoptProblem->getNumAddComponents() + nAddComponents > 0)
	  ("Not yet implemented.\n");
239
240
241

	// If the adopt problem has fewer components than this problem, but only one
	// fe space for all component, than scal up the componentSpaces array.
Thomas Witkowski's avatar
Thomas Witkowski committed
242
243
	if (adoptProblem->getNumComponents() < nComponents) {
	  TEST_EXIT(feSpaces.size() == 1)("Daran muss ich noch arbeiten!\n");
244
	  
Thomas Witkowski's avatar
Thomas Witkowski committed
245
	  componentSpaces.resize(nComponents);
Thomas Witkowski's avatar
Thomas Witkowski committed
246
	  for (int i = adoptProblem->getNumComponents(); i < nComponents; i++)
Thomas Witkowski's avatar
Thomas Witkowski committed
247
	    componentSpaces[i] = componentSpaces[0];
248
249
	}

250
251
252
      }
    }

253
    if (feSpaces.size() == 0) 
254
255
256
      WARNING("no feSpace created\n");

    // === create system ===
257
    if (initFlag.isSet(INIT_SYSTEM)) 
258
      createMatricesAndVectors();
259
    
260
    if (adoptProblem && adoptFlag.isSet(INIT_SYSTEM)) {
Thomas Witkowski's avatar
Thomas Witkowski committed
261
      solution = adoptProblem->getSolution();
262
      rhs = adoptProblem->getRhs();
Thomas Witkowski's avatar
Thomas Witkowski committed
263
      systemMatrix = adoptProblem->getSystemMatrix();
264
265
266
    }

    // === create solver ===
Thomas Witkowski's avatar
Thomas Witkowski committed
267
    if (solver) {
268
269
      WARNING("solver already created\n");
    } else {
270
      if (initFlag.isSet(INIT_SOLVER))
271
	createSolver();
272
      
273
      if (adoptProblem && adoptFlag.isSet(INIT_SOLVER)) {
Thomas Witkowski's avatar
Thomas Witkowski committed
274
275
	TEST_EXIT(!solver)("solver already created\n");
	solver = adoptProblem->getSolver();
276
277
278
      }
    }

279
// #ifndef HAVE_PARALLEL_DOMAIN_AMDIS
Thomas Witkowski's avatar
Thomas Witkowski committed
280
    if (!solver) 
281
      WARNING("no solver created\n");
282
// #endif
283
284

    // === create estimator ===
Thomas Witkowski's avatar
Thomas Witkowski committed
285
    if (initFlag.isSet(INIT_ESTIMATOR))
286
      createEstimator();
Thomas Witkowski's avatar
Thomas Witkowski committed
287
288

    if (adoptProblem && adoptFlag.isSet(INIT_ESTIMATOR))
289
      estimator = adoptProblem->getEstimators();
290
291

    // === create marker ===
Thomas Witkowski's avatar
Thomas Witkowski committed
292
    if (initFlag.isSet(INIT_MARKER))
293
      createMarker();
Thomas Witkowski's avatar
Thomas Witkowski committed
294

Thomas Witkowski's avatar
Thomas Witkowski committed
295
    if (adoptProblem && adoptFlag.isSet(INIT_MARKER))
296
      marker = adoptProblem->getMarkers();
297
298

    // === create file writer ===
Thomas Witkowski's avatar
Thomas Witkowski committed
299
    if (initFlag.isSet(INIT_FILEWRITER))
300
301
302
303
304
      createFileWriter();
    
    // === read serialization and init mesh ===
    
    // There are two possiblities where the user can define a serialization
305
306
307
308
    // to be read from disk. Either by providing the parameter -rs when 
    // executing the program or in the init file. The -rs parameter is always
    // checked first, because it can be added automatically when rescheduling
    // the program before timeout of the runqueue.
309
310

    int readSerialization = 0;
311
312
    string serializationFilename = "";
    Parameters::get("argv->rs", serializationFilename);
313
314
315
316
317
318

    // If the parameter -rs is set, we do nothing here, because the problem will be
    // deserialized in the constructor of a following AdaptInstationary initialization.
    if (!serializationFilename.compare("")) {
      int readSerializationWithAdaptInfo = 0;

319
320
321
      Parameters::get(name + "->input->read serialization", readSerialization);
      Parameters::get(name + "->input->serialization with adaptinfo",
		      readSerializationWithAdaptInfo);
322
323
324
325
326
327

      // The serialization file is only read, if the adaptInfo part should not be used.
      // If the adaptInfo part should be also read, the serialization file will be read
      // in the constructor of the AdaptInstationary problem, because we do not have here
      // the adaptInfo object.
      if (readSerialization && !readSerializationWithAdaptInfo) {
328
329
	Parameters::get(name + "->input->serialization filename", 
			serializationFilename);
330
331
	TEST_EXIT(serializationFilename != "")("no serialization file\n");

332
333
	// If AMDiS is compiled for parallel computations, the deserialization is
	// controlled by the parallel problem object.
334
#ifndef HAVE_PARALLEL_DOMAIN_AMDIS
335
	MSG("Deserialization from file: %s\n", serializationFilename.c_str());
336
	ifstream in(serializationFilename.c_str());
337
338
339
340
341
342

	// Read the revision number of the AMDiS version which was used to create 
	// the serialization file.
	int revNumber = -1;
	SerUtil::deserialize(in, revNumber);

343
344
	deserialize(in);
	in.close();
345
#endif
346
347

	deserialized = true;
348
      } else {
349
	int globalRefinements = 0;
350

351
	// If AMDiS is compiled for parallel computations, the global refinements are
352
353
354
	// ignored here. Later, each rank will add the global refinements to its 
	// private mesh.
#ifndef HAVE_PARALLEL_DOMAIN_AMDIS
355
356
	Parameters::get(meshes[0]->getName() + "->global refinements", 
			globalRefinements);
357
#endif
358

359
	bool initMesh = initFlag.isSet(INIT_MESH);
360

Thomas Witkowski's avatar
Thomas Witkowski committed
361
362
	// Initialize the meshes if there is no serialization file.
	for (int i = 0; i < static_cast<int>(meshes.size()); i++)
363
	  if (initMesh && meshes[i] && !(meshes[i]->isInitialized()))
Thomas Witkowski's avatar
Thomas Witkowski committed
364
	    meshes[i]->initialize();	    	
365
366

	// === read value file and use it for the mesh values ===
367
368
	string valueFilename("");
	Parameters::get(meshes[0]->getName() + "->value file name", valueFilename); 
369
	if (valueFilename.length())
370
	  io::ValueReader::readValue(valueFilename,
371
372
373
374
375
				 meshes[0],
				 solution->getDOFVector(0),
				 meshes[0]->getMacroFileInfo());

	// === do global refinements ===
376

Thomas Witkowski's avatar
Thomas Witkowski committed
377
	for (unsigned int i = 0; i < meshes.size(); i++)
378
	  if (initMesh && meshes[i])
Thomas Witkowski's avatar
Thomas Witkowski committed
379
	    refinementManager->globalRefine(meshes[i], globalRefinements);	
380
381
382
383
384
385
      }
    }

    doOtherStuff();
  }

386

387
  void ProblemStatSeq::createMesh() 
388
  {
389
    FUNCNAME("ProblemStat::createMesh()");
390
    
391
392
393
394
395
396
    map<int, Mesh*> meshForRefinementSet;

    string meshName("");
    Parameters::get(name + "->mesh", meshName);
    TEST_EXIT(meshName != "")("No mesh name specified for \"%s->mesh\"!\n", 
			      name.c_str());
397
398

    int dim = 0;
399
400
401
    Parameters::get(name + "->dim", dim);
    TEST_EXIT(dim)("No problem dimension specified for \"%s->dim\"!\n",
		   name.c_str());
402
403
    
    componentMeshes.resize(nComponents + nAddComponents);
404

405
    for (int i = 0; i < nComponents + nAddComponents; i++) {
406
      int refSet = -1;
407
408
      Parameters::get(name + "->refinement set[" + lexical_cast<string>(i) + "]", 
		      refSet);
409
      if (refSet < 0)
410
	refSet = 0;
411
      
412
      if (meshForRefinementSet[refSet] == NULL) {
Thomas Witkowski's avatar
Thomas Witkowski committed
413
	Mesh *newMesh = new Mesh(meshName, dim);
414
	meshForRefinementSet[refSet] = newMesh;
Thomas Witkowski's avatar
Thomas Witkowski committed
415
416
	meshes.push_back(newMesh);
	nMeshes++;
417
      }
Thomas Witkowski's avatar
Thomas Witkowski committed
418
      componentMeshes[i] = meshForRefinementSet[refSet];
419
    }
420
421
422
423
424
425
426
427
428
429
430
  }


  void ProblemStatSeq::createRefCoarseManager() 
  {
    FUNCNAME("ProblemStat::createRefCoarseManager()");

    int dim = 0;
    Parameters::get(name + "->dim", dim);
    TEST_EXIT(dim)("No problem dimension specified for \"%s->dim\"!\n",
                   name.c_str());
431
432

    switch (dim) {
433
    case 1:
Thomas Witkowski's avatar
Thomas Witkowski committed
434
      refinementManager = new RefinementManager1d();
435
      coarseningManager = new CoarseningManager1d();
436
437
      break;
    case 2:
Thomas Witkowski's avatar
Thomas Witkowski committed
438
      refinementManager = new RefinementManager2d();
439
      coarseningManager = new CoarseningManager2d();
440
441
      break;
    case 3:
Thomas Witkowski's avatar
Thomas Witkowski committed
442
      refinementManager = new RefinementManager3d();
443
      coarseningManager = new CoarseningManager3d();
444
445
446
447
448
449
      break;
    default:
      ERROR_EXIT("invalid dim!\n");
    }
  }

450

451
  void ProblemStatSeq::createFeSpace(DOFAdmin *admin)
452
  {
453
    FUNCNAME("ProblemStat::createFeSpace()");
454

455
    map<pair<Mesh*, string>, FiniteElemSpace*> feSpaceMap;
456
    int dim = -1;
457
    Parameters::get(name + "->dim", dim);
458
    TEST_EXIT(dim != -1)("no problem dimension specified!\n");
459

460
    componentSpaces.resize(nComponents + nAddComponents, NULL);
Thomas Witkowski's avatar
Thomas Witkowski committed
461
    traverseInfo.resize(nComponents);
462

463
464
465
466
467
    for (int i = 0; i < nComponents + nAddComponents; i++) {
      if (componentSpaces[i] != NULL) {
	WARNING("feSpace already created\n");
	continue;
      }
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
      string componentString = "[" + boost::lexical_cast<string>(i) + "]";
      
      string feSpaceName = "";
      string initFileStr = name + "->feSpace" + componentString;
      Parameters::get(initFileStr, feSpaceName);
      
      // synonym for "feSpace"
      if (feSpaceName.size() == 0) {
	initFileStr = name + "->finite element space" + componentString;
	Parameters::get(initFileStr, feSpaceName);
      }
      
      // for backward compatibility also accept the old syntax
      if (feSpaceName.size() == 0) {
	int degree = 1;
	initFileStr = name + "->polynomial degree" + componentString;
	Parameters::get(initFileStr, degree);
	TEST_EXIT(degree > 0)
	  ("Poynomial degree in component %d must be larger than zero!\n", i);
	  
	feSpaceName = "Lagrange" + boost::lexical_cast<string>(degree);
      }
490

491
492
493
494
495
496
497
      if (feSpaceMap[pair<Mesh*, string>(componentMeshes[i], feSpaceName)] == NULL) {      
	BasisFunctionCreator *basisFctCreator = 
	dynamic_cast<BasisFunctionCreator*>(CreatorMap<BasisFunction>::getCreator(feSpaceName, initFileStr));
	TEST_EXIT(basisFctCreator)
	  ("No valid basisfunction type found in parameter \"%s\"\n", initFileStr.c_str());
	basisFctCreator->setDim(dim);
	
498
	FiniteElemSpace *newFeSpace = 
499
500
501
502
	  FiniteElemSpace::provideFeSpace(admin, basisFctCreator->create(),
					  componentMeshes[i], "FeSpace" + componentString + " (" + feSpaceName + ")");

	feSpaceMap[pair<Mesh*, string>(componentMeshes[i], feSpaceName)] = newFeSpace;
503
	feSpaces.push_back(newFeSpace);
504
      }
505

506
      componentSpaces[i] = feSpaceMap[pair<Mesh*, string>(componentMeshes[i], feSpaceName)];
Thomas Witkowski's avatar
Thomas Witkowski committed
507
508
509
    }

    for (int i = 0; i < nComponents; i++) {
510
      for (int j = 0; j < nComponents; j++)
511
	traverseInfo.getMatrix(i, j).setFeSpace(componentSpaces[i], componentSpaces[j]);
512
      
513
      traverseInfo.getVector(i).setFeSpace(componentSpaces[i]);
514
515
516
    }

    // create dof admin for vertex dofs if neccessary
Thomas Witkowski's avatar
Thomas Witkowski committed
517
    for (int i = 0; i < static_cast<int>(meshes.size()); i++) {
518
      if (meshes[i]->getNumberOfDofs(VERTEX) == 0) {
Thomas Witkowski's avatar
Thomas Witkowski committed
519
	DimVec<int> ln_dof(meshes[i]->getDim(), DEFAULT_VALUE, 0);
520
	ln_dof[VERTEX] = 1;
Thomas Witkowski's avatar
Thomas Witkowski committed
521
	meshes[i]->createDOFAdmin("vertex dofs", ln_dof);      
522
523
524
525
      }
    }
  }

526

527
  void ProblemStatSeq::createMatricesAndVectors()
528
529
530
  {
    // === create vectors and system matrix ===

Thomas Witkowski's avatar
Thomas Witkowski committed
531
    systemMatrix = new Matrix<DOFMatrix*>(nComponents, nComponents);
532
    systemMatrix->set((DOFMatrix*)(NULL));
Thomas Witkowski's avatar
Thomas Witkowski committed
533
534
    rhs = new SystemVector("rhs", componentSpaces, nComponents);
    solution = new SystemVector("solution", componentSpaces, nComponents);
535

Thomas Witkowski's avatar
Thomas Witkowski committed
536
    for (int i = 0; i < nComponents; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
537
      (*systemMatrix)[i][i] = new DOFMatrix(componentSpaces[i], 
Thomas Witkowski's avatar
Thomas Witkowski committed
538
					    componentSpaces[i], "A_ii");
Thomas Witkowski's avatar
Thomas Witkowski committed
539
      (*systemMatrix)[i][i]->setCoupleMatrix(false);
Thomas Witkowski's avatar
Thomas Witkowski committed
540

541
      //set parallel synchronization later in ParalleProblemStat
542
      rhs->setDOFVector(i, new DOFVector<double>(componentSpaces[i], 
543
						 "rhs[" + lexical_cast<string>(i) + "]", false));
544

545
      //set parallel synchronization later in ParalleProblemStat
Thomas Witkowski's avatar
Thomas Witkowski committed
546
      solution->setDOFVector(i, new DOFVector<double>(componentSpaces[i],
547
      						      componentNames[i], false));
Thomas Witkowski's avatar
Thomas Witkowski committed
548
549
      solution->getDOFVector(i)->setCoarsenOperation(COARSE_INTERPOL);
      solution->getDOFVector(i)->set(0.0);
550
551
552
    }
  }

553

554
  void ProblemStatSeq::createSolver()
555
  {
556
    FUNCNAME("ProblemStat::createSolver()");
557
    
Praetorius, Simon's avatar
Praetorius, Simon committed
558
    // definition of standard-backends
559
560
561
562
#if defined HAVE_PARALLEL_PETSC
    string backend("p_petsc");
#elif defined HAVE_PARALLEL_MTL
    string backend("p_mtl");
563
#elif defined HAVE_SEQ_PETSC
564
565
566
567
568
569
570
571
    string backend("petsc");
#else
    string backend("mtl");
#endif
    
    // === read backend-name ===
    string initFileStr = name + "->solver";    
    Parameters::get(initFileStr + "->backend", backend);
572

573
    // === read solver-name ===
574
    string solverType("0");
575
    Parameters::get(initFileStr, solverType);
576
577
578
579
580
581
    
    if (backend != "0" && backend != "no" && backend != "")
      solverType = backend + "_" + solverType;
    
    // === create solver ===
    LinearSolverCreator *solverCreator = 
582
      dynamic_cast<LinearSolverCreator*>(CreatorMap<LinearSolverInterface>::getCreator(solverType, initFileStr));
583
584
    TEST_EXIT(solverCreator)
      ("No valid solver type found in parameter \"%s\"\n", initFileStr.c_str());
585
    solverCreator->setName(initFileStr);
Thomas Witkowski's avatar
Thomas Witkowski committed
586
    solver = solverCreator->create();
587
588
  }

589

590
  void ProblemStatSeq::createEstimator()
591
  {
592
    FUNCNAME("ProblemStat::createEstimator()");
593
594

    // create and set leaf data prototype
Thomas Witkowski's avatar
Thomas Witkowski committed
595
    for (unsigned int i = 0; i < meshes.size(); i++)
Thomas Witkowski's avatar
Thomas Witkowski committed
596
      meshes[i]->setElementDataPrototype
Thomas Witkowski's avatar
Thomas Witkowski committed
597
	(new LeafDataEstimatableVec(new LeafDataCoarsenableVec));
598

Thomas Witkowski's avatar
Thomas Witkowski committed
599
    for (int i = 0; i < nComponents; i++) {
600
      TEST_EXIT(estimator[i] == NULL)("estimator already created\n");
601
602
      string estName = 
	name + "->estimator[" + boost::lexical_cast<string>(i) + "]";
603
604

      // === create estimator ===
605
606
      string estimatorType("0");
      Parameters::get(estName, estimatorType);
607
      EstimatorCreator *estimatorCreator = 
608
	dynamic_cast<EstimatorCreator*>(CreatorMap<Estimator>::getCreator(estimatorType, estName));
Thomas Witkowski's avatar
Thomas Witkowski committed
609
      if (estimatorCreator) {
610
611
	estimatorCreator->setName(estName);
	estimatorCreator->setRow(i);
612
	estimatorCreator->setSolution(solution->getDOFVector(i));
Thomas Witkowski's avatar
Thomas Witkowski committed
613
	estimator[i] = estimatorCreator->create();
614
615
616
      }


Thomas Witkowski's avatar
Thomas Witkowski committed
617
618
619
620
      if (estimator[i])
	for (int j = 0; j < nComponents; j++)
	  estimator[i]->addSystem((*systemMatrix)[i][j], 
				  solution->getDOFVector(j), 
Praetorius, Simon's avatar
Praetorius, Simon committed
621
				  rhs->getDOFVector(j));      // TODO: hier eventuell (i) statt (j) ???
622
623
624
    }
  }

625

626
  void ProblemStatSeq::createMarker()
627
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
628
    int nMarkersCreated = 0;
629

630
    for (int i = 0; i < nComponents; i++) {
631
      marker[i] = Marker::createMarker
632
	(name + "->marker[" + boost::lexical_cast<string>(i) + "]", i);
633

634
      if (marker[i]) {
Thomas Witkowski's avatar
Thomas Witkowski committed
635
636
637
638
	nMarkersCreated++;

	// If there is more than one marker, and all components are defined
	// on the same mesh, the maximum marking has to be enabled.
639
 	if (nMarkersCreated > 1 && nMeshes == 1)
Thomas Witkowski's avatar
Thomas Witkowski committed
640
 	  marker[i]->setMaximumMarking(true);
641
642
643
644
      }
    }
  }

645

646
  void ProblemStatSeq::createFileWriter()
647
  {
648
    FUNCNAME("ProblemStat::createFileWriter()");
649
650
  
    // Create one filewriter for all components of the problem
651
652
653
    string numberedName  = name + "->output";
    string filename = "";
    Parameters::get(numberedName + "->filename", filename);
654
655

    if (filename != "") {
656
      vector< DOFVector<double>* > solutionList(nComponents);
657

658
      for (int i = 0; i < nComponents; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
659
	TEST_EXIT(componentMeshes[0] == componentMeshes[i])
660
661
	  ("All Meshes have to be equal to write a vector file.\n");

Thomas Witkowski's avatar
Thomas Witkowski committed
662
	solutionList[i] = solution->getDOFVector(i);
663
664
      }

Thomas Witkowski's avatar
Thomas Witkowski committed
665
      fileWriters.push_back(new FileWriter(numberedName,
666
667
					   componentMeshes[0],
					   solutionList));
668
669
    }

Praetorius, Simon's avatar
Praetorius, Simon committed
670
    // Create own filewriters for each component of the problem
671
    for (int i = 0; i < nComponents; i++) {
672
      numberedName = name + "->output[" + boost::lexical_cast<string>(i) + "]";
673
      filename = "";
674
      Parameters::get(numberedName + "->filename", filename);
675

676
      if (filename != "")
Thomas Witkowski's avatar
Thomas Witkowski committed
677
	fileWriters.push_back(new FileWriter(numberedName, 
678
					     componentMeshes[i], 
679
					     solution->getDOFVector(i)));      
680
    }
Praetorius, Simon's avatar
Praetorius, Simon committed
681
    
682
    // create a filewrite for groups of components to write vector-valued output
Praetorius, Simon's avatar
Praetorius, Simon committed
683
684
685
686
687
688
689
690
691
692
693
    int nVectors = 0;
    Parameters::get(name + "->output->num vectors", nVectors);
    if (nVectors > 0) {
      for (int j = 0; j < nVectors; j++) {
	numberedName = name + "->output->vector[" + boost::lexical_cast<string>(j) + "]";
	
	filename = "";
	Parameters::get(numberedName + "->filename", filename);
	std::string componentName = "";
	Parameters::get(numberedName + "->name", componentName);
	std::vector<std::string> names; 
694
	if (componentName != "")
Praetorius, Simon's avatar
Praetorius, Simon committed
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
	  names.push_back(componentName);
	std::vector<int> comp;
	Parameters::get(numberedName + "->components", comp);
	
	if (filename != "" && comp.size() > 0) {
	  // Create own filewriters for each component of the problem
	  std::vector<DOFVector<double>*> vectors;
	  for (size_t i = 0; i < comp.size(); i++)
	    vectors.push_back(solution->getDOFVector(comp[i]));
	  
	  fileWriters.push_back(new FileWriter(numberedName, 
					      componentMeshes[comp[0]], 
					      vectors,
					      names
					      ));      
	  
	}
      }
    }
714
715

    int writeSerialization = 0;
716
    Parameters::get(name + "->output->write serialization", writeSerialization);
717
    if (writeSerialization)
718
      fileWriters.push_back(new Serializer<ProblemStatSeq>(this));
719
720
  }

721

722
  void ProblemStatSeq::doOtherStuff()
723
  {}
724

725

726
727
728
  void ProblemStatSeq::solve(AdaptInfo *adaptInfo, 	       
			     bool createMatrixData,
			     bool storeMatrixData)
729
730
731
  {
    FUNCNAME("Problem::solve()");

Thomas Witkowski's avatar
Thomas Witkowski committed
732
    if (!solver) {
733
734
735
736
      WARNING("no solver\n");
      return;
    }

Praetorius, Simon's avatar
Praetorius, Simon committed
737
    Timer t;
738
739
    solver->solveSystem(solverMatrix, *solution, *rhs, 
			createMatrixData, storeMatrixData);
740

Thomas Witkowski's avatar
Thomas Witkowski committed
741
    INFO(info, 8)("solution of discrete system needed %.5f seconds\n", 
Praetorius, Simon's avatar
Praetorius, Simon committed
742
		  t.elapsed());
743
    solutionTime = t.elapsed();
744

745
    adaptInfo->setSolverIterations(solver->getIterations());
Thomas Witkowski's avatar
Thomas Witkowski committed
746
747
748
    adaptInfo->setMaxSolverIterations(solver->getMaxIterations());
    adaptInfo->setSolverTolerance(solver->getTolerance());
    adaptInfo->setSolverResidual(solver->getResidual());
749
750
  }

751

752
  void ProblemStatSeq::estimate(AdaptInfo *adaptInfo) 
753
  {
754
    FUNCNAME("ProblemStat::estimate()");
755

Praetorius, Simon's avatar
Praetorius, Simon committed
756
    Timer t;
757

758
    if (computeExactError) {
Thomas Witkowski's avatar
Thomas Witkowski committed
759
      computeError(adaptInfo);
760
761
    } else {
      for (int i = 0; i < nComponents; i++) {
Thomas Witkowski's avatar
Thomas Witkowski committed
762
	Estimator *scalEstimator = estimator[i];
763
764
	
	if (scalEstimator) {
765
766
	  traverseInfo.updateStatus();
	  scalEstimator->setTraverseInfo(traverseInfo);
767
	  scalEstimator->estimate(adaptInfo->getTimestep());
768

769
770
	  adaptInfo->setEstSum(scalEstimator->getErrorSum(), i);
	  adaptInfo->setEstMax(scalEstimator->getErrorMax(), i);
771
772
773
774
775

	  if (adaptInfo->getRosenbrockMode() == false) {
	    adaptInfo->setTimeEstSum(scalEstimator->getTimeEst(), i);
	    adaptInfo->setTimeEstMax(scalEstimator->getTimeEstMax(), i);
	  }
776
	}
777
778
779
      }
    }

780
#ifdef HAVE_PARALLEL_DOMAIN_AMDIS
781
    MPI::COMM_WORLD.Barrier();
782
#endif
Praetorius, Simon's avatar
Praetorius, Simon committed
783
784
    INFO(info, 8)("estimation of the error needed %.5f seconds\n", 
		  t.elapsed());
785
786
  }

787

788
  Flag ProblemStatSeq::markElements(AdaptInfo *adaptInfo) 
789
  {
790
    FUNCNAME_DBG("ProblemStat::markElements()");
791

792
    TEST_EXIT_DBG(static_cast<unsigned int>(nComponents) == marker.size())
Thomas Witkowski's avatar
Thomas Witkowski committed
793
794
      ("Wrong number of markers!\n");

795
    Flag markFlag = 0;
Thomas Witkowski's avatar
Thomas Witkowski committed
796
    for (int i = 0; i < nComponents; i++)
797
      if (marker[i])
Thomas Witkowski's avatar
Thomas Witkowski committed
798
	markFlag |= marker[i]->markMesh(adaptInfo, componentMeshes[i]);
799
    
800
801