DOFMatrix.cc 20.8 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#include "DOFMatrix.h"
#include <algorithm>
#include <png.h>
#include "QPsiPhi.h"
#include "BasisFunction.h"
#include "Boundary.h"
#include "DOFAdmin.h"
#include "ElInfo.h"
#include "FiniteElemSpace.h"
#include "Mesh.h"
#include "DOFVector.h"
#include "Operator.h"
#include "BoundaryCondition.h"
#include "BoundaryManager.h"
#include "ElementMatrix.h"
#include "Assembler.h"

namespace AMDiS {

  DOFMatrix *DOFMatrix::traversePtr = NULL;

  DOFMatrix::DOFMatrix()
  {
    rowFESpace = NULL;
    colFESpace = NULL;
  }

  DOFMatrix::DOFMatrix(const FiniteElemSpace* rowFESpace_,
		       const FiniteElemSpace* colFESpace_,
30
		       std::string name_)
31
32
33
34
35
36
37
    : rowFESpace(rowFESpace_),
      colFESpace(colFESpace_),
      name(name_), 
      coupleMatrix(false)
  {
    TEST_EXIT(rowFESpace)("no rowFESpace\n");
  
38
    if (!colFESpace) {
39
40
41
42
      colFESpace = rowFESpace;
    }

    if (rowFESpace && rowFESpace->getAdmin())
Thomas Witkowski's avatar
Thomas Witkowski committed
43
      (const_cast<DOFAdmin*>(rowFESpace->getAdmin()))->addDOFIndexed(this);
44
45

    boundaryManager = NEW BoundaryManager;
Thomas Witkowski's avatar
Thomas Witkowski committed
46
47
    elementMatrix = NEW ElementMatrix(rowFESpace->getBasisFcts()->getNumber(),
				      colFESpace->getBasisFcts()->getNumber());
48
49
50
  }

  DOFMatrix::DOFMatrix(const DOFMatrix& rhs)
Thomas Witkowski's avatar
Thomas Witkowski committed
51
    : name(rhs.name + "copy")
52
  {
53
    *this = rhs;
54
55
56
57
58
59
    if (rowFESpace && rowFESpace->getAdmin())
      (const_cast<DOFAdmin*>( rowFESpace->getAdmin()))->addDOFIndexed(this);
  }

  DOFMatrix::~DOFMatrix()
  {
60
61
    FUNCNAME("DOFMatrix::~DOFMatrix()");

62
    if (rowFESpace && rowFESpace->getAdmin()) {
63
64
      (const_cast<DOFAdmin*>(rowFESpace->getAdmin()))->removeDOFIndexed(this);
    }  
Thomas Witkowski's avatar
Thomas Witkowski committed
65
66
67

    if (elementMatrix)
      DELETE elementMatrix;
68
69
70
71
72
73
74
75
76
77
78
79
80
  }

  void DOFMatrix::print() const
  {
    FUNCNAME("DOFMatrix::print()");

    int sizeUsed = rowFESpace->getAdmin()->getUsedSize();

    if (static_cast<int>(matrix.size()) < sizeUsed) {
      WARNING("DOFMatrix not yet initialized\n");
      return;
    }

Thomas Witkowski's avatar
Thomas Witkowski committed
81
82
    for (int i = 0; i < sizeUsed; i++) { 
      DOFMatrix::MatrixRow row = matrix[i];
83
84
      MSG("row %3d:",i);
      int rowSize = static_cast<int>( row.size());
Thomas Witkowski's avatar
Thomas Witkowski committed
85
86
87
      for (int j = 0; j < rowSize; j++) {
	int jcol = row[j].col;
	if (entryUsed(i, j)) {
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
	  Msg::print(" (%3d,%20.17lf)", jcol, row[j].entry);
	}
      }
      Msg::print("\n");
    }
  }

  void DOFMatrix::printRow(int i) const
  {
    FUNCNAME("DOFMatrix::printRow()");

    int sizeUsed = rowFESpace->getAdmin()->getUsedSize();

    if (static_cast<int>(matrix.size()) < sizeUsed) {
      WARNING("DOFMatrix not yet initialized\n");
      return;
    }

    DOFMatrix::MatrixRow row = matrix[i];
    MSG("row %3d:",i);
    int rowSize = static_cast<int>( row.size());
    for (int j = 0; j < rowSize; j++) {
      int jcol = row[j].col;
      if (entryUsed(i,j)) {
	Msg::print(" (%3d,%20.17lf)", jcol, row[j].entry);
      }
    }
    Msg::print("\n");   
  }

  /****************************************************************************/
  /*  clear: remove all entries from dof_matrix                               */
  /****************************************************************************/

  void DOFMatrix::clear()
  {
124
125
    int mSize = static_cast<int>(matrix.size());
    for (int i = 0; i < mSize; i++) {
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
      matrix[i].resize(0);
    }
  }

  bool DOFMatrix::symmetric()
  {
    FUNCNAME("DOFMatrix::symmetric()");

    DegreeOfFreedom row, col;
    double entry, tol = 1e-5;

    DOFMatrix::Iterator matrixRow(this, USED_DOFS);

    for (matrixRow.reset(); !matrixRow.end(); ++matrixRow) {
      row = matrixRow.getDOFIndex();
      int rowSize = matrixRow->size();
      for (int i = 0; i < rowSize; i++) {
	col = (*matrixRow)[i].col;
	entry = (*matrixRow)[i].entry;
	if (abs(entry - logAcc(col, row)) > tol) {
	  MSG("matrix[%d][%d] = %e, matrix[%d][%d] = %e\n",
	      row, col, entry, col, row, logAcc(col, row));

	  return false;
	}
      }
    }

    return true;
  }

  void DOFMatrix::test()
  {
    FUNCNAME("DOFMatrix::test()");

    int non_symmetric = 0, found = 0;

    /* test symmetry */
    for (int i = 0; i < static_cast<int>(matrix.size()); i++) { 
      double sum = 0.0;
      DOFMatrix::MatrixRow *row = &matrix[i];
      for (int j = 0; j < static_cast<int>(row->size()); j++) {
	int jcol = (*row)[j].col;
	if (entryUsed(i, j)) {
	  found = 0;
	  if ((*row)[j].entry != (*row)[j].entry) {
	    MSG("mat[%d,%d]=%10.5e ???\n", i, jcol, (*row)[j].entry);
	    WAIT;
	  }
	  DOFMatrix::MatrixRow *row2 = &matrix[jcol];
	  for (int k = 0; k < static_cast<int>(row->size()); k++) {
	    int kcol = (*row)[k].col;
	    if (entryUsed(jcol, k)) {
	      if (kcol == i) {
		found = 1;
		if (abs((*row2)[k].entry - (*row)[j].entry) > 1.E-5) {
		  non_symmetric = 1;
		  MSG("mat[%d,%d]=%10.5e != mat[%d,%d]=%10.5e\n",
		      i, jcol, (*row)[j].entry, jcol, i, (*row2)[k].entry);
		}
		row2 = NULL;
		break;
	      }
	    }
	  }
	  if (!found) {
	    non_symmetric = 1;
	    MSG("mat[%d,%d] not found\n", jcol, i);
	  }
	}
      }
      if (abs(sum) > 1.E-5) {
	MSG("Zeilensumme[%d] = %10.5e\n", i, sum);
      }
    }

    if (non_symmetric) {
      MSG("matrix `%s' not symmetric.\n", name.data());
    } else {
      MSG("matrix `%s' is symmetric.\n", name.data());
    }
  }


  DOFMatrix& DOFMatrix::operator=(const DOFMatrix& rhs)
  {
    rowFESpace = rhs.rowFESpace;
    colFESpace = rhs.colFESpace;
    operators = rhs.operators;
    operatorFactor = rhs.operatorFactor;
    matrix = rhs.matrix;
    if (rhs.boundaryManager) {
      boundaryManager = new BoundaryManager(*rhs.boundaryManager);
    } else {
Thomas Witkowski's avatar
Thomas Witkowski committed
220
221
222
223
224
225
226
      boundaryManager = NULL;
    }
    if (rhs.elementMatrix) {
      elementMatrix = NEW ElementMatrix(rowFESpace->getBasisFcts()->getNumber(),
					colFESpace->getBasisFcts()->getNumber());
    } else {
      elementMatrix = NULL;
227
228
229
230
231
232
233
234
235
236
237
238
239
    }

    return *this;
  }


  void DOFMatrix::addElementMatrix(double sign, 
				   const ElementMatrix &elMat, 
				   const BoundaryType *bound,
				   bool add)
  {
    FUNCNAME("DOFMatrix::addElementMatrix");

Thomas Witkowski's avatar
Thomas Witkowski committed
240
    DegreeOfFreedom row;
241

Thomas Witkowski's avatar
Thomas Witkowski committed
242
243
    int nRow = elMat.rowIndices.getSize();
    int nCol = elMat.colIndices.getSize();
244

Thomas Witkowski's avatar
Thomas Witkowski committed
245
    for (int i = 0; i < nRow; i++)  {   // for all rows of element matrix
246
247
248
249
250
251
      row = elMat.rowIndices[i];
      BoundaryCondition *condition = 
	bound ? boundaryManager->getBoundaryCondition(bound[i]) : NULL;

      if (condition && condition->isDirichlet()) {
	MatrixRow *matrixRow = &(matrix[row]);
252
	if (coupleMatrix) {
253
254
255
256
257
258
	  matrixRow->resize(0);
	} else {
	  matrixRow->resize(1);
	  ((*matrixRow)[0]).col = row;
	  ((*matrixRow)[0]).entry = 1.0;
	}     
Thomas Witkowski's avatar
Thomas Witkowski committed
259
      } else {     
Thomas Witkowski's avatar
Thomas Witkowski committed
260
261
262
	for (int j = 0; j < nCol; j++) {  // for all columns
	  addSparseDOFEntry(sign, row, elMat.colIndices[j], 
			    elMat[i][j], add); 
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
	}
      }
    }
  }

  double DOFMatrix::logAcc(DegreeOfFreedom a,DegreeOfFreedom b) const
  {
    int j;

    for (j = 0; j < static_cast<int>(matrix[a].size()); j++) 
      if (b == matrix[a][j].col) 
	break;

    return (j == static_cast<int>(matrix[a].size())) ? 0.0 : matrix[a][j].entry;
  }

  void DOFMatrix::changeColOfEntry(DegreeOfFreedom a,
				   DegreeOfFreedom b,
				   DegreeOfFreedom c)
  {
    int j;

Thomas Witkowski's avatar
Thomas Witkowski committed
285
286
287
288
289
290
    for (j = 0; j<static_cast<int>(matrix[a].size());j++) 
      if (b == matrix[a][j].col) 
	break;

    if (j != static_cast<int>(matrix[a].size())) 
      matrix[a][j].col = c;
291
292
  }

293
294
295
  void DOFMatrix::addSparseDOFEntry(double sign, int irow, 
				    int jcol, double entry,
				    bool add)
296
  {
297
    FUNCNAME("DOFMatrix::addSparseDOFEntry()");
298
299
300

    MatrixRow *row = &(matrix[irow]);

301
    if (add && !entry) 
302
      return;
303

304
    int freeCol = -1;
Thomas Witkowski's avatar
Thomas Witkowski committed
305
    int rowSize = static_cast<int>(row->size());
306

307
308
309
    TEST_EXIT_DBG(jcol >= 0 && 
		  jcol < colFESpace->getAdmin()->getUsedSize())
      ("Column index %d out of range 0-%d\n", jcol, colFESpace->getAdmin()->getUsedSize() - 1);
310
311

    // first entry is diagonal entry
Thomas Witkowski's avatar
Thomas Witkowski committed
312
313
314
315
316
    if ((rowSize == 0) && (rowFESpace == colFESpace)) {
      MatEntry newEntry = {irow, 0.0};
      row->push_back(newEntry);
      rowSize = 1;
    }
317

318
    int i;
319
    // search jcol
Thomas Witkowski's avatar
Thomas Witkowski committed
320
    for (i = 0; i < rowSize; i++) {
321
      // jcol found ?
Thomas Witkowski's avatar
Thomas Witkowski committed
322
      if ((*row)[i].col == jcol) {
323
324
325
	break;
      }
      // remember free entry
Thomas Witkowski's avatar
Thomas Witkowski committed
326
      if ((*row)[i].col == UNUSED_ENTRY) {
327
328
329
	freeCol = i;
      }
      // no more entries
Thomas Witkowski's avatar
Thomas Witkowski committed
330
      if ((*row)[i].col == NO_MORE_ENTRIES) {
331
	freeCol = i;
Thomas Witkowski's avatar
Thomas Witkowski committed
332
	if (rowSize > i+1) {
Thomas Witkowski's avatar
Thomas Witkowski committed
333
	  (*row)[i + 1].entry = NO_MORE_ENTRIES;
334
335
336
337
338
339
	}
	break;
      }
    }

    // jcol found?
Thomas Witkowski's avatar
Thomas Witkowski committed
340
    if (i < rowSize) {
Thomas Witkowski's avatar
Thomas Witkowski committed
341
      if (!add) {
Thomas Witkowski's avatar
Thomas Witkowski committed
342
	(*row)[i].entry = 0.0;
Thomas Witkowski's avatar
Thomas Witkowski committed
343
344
      }

345
346
      (*row)[i].entry += sign * entry;
    } else {
347
      if (freeCol == -1) {
348
349
350
351
	MatEntry newEntry = {jcol, sign * entry};
	row->push_back(newEntry);
      } else {
	(*row)[freeCol].col = jcol;
Thomas Witkowski's avatar
Thomas Witkowski committed
352
353
	if (!add) 
	  (*row)[freeCol].entry = 0.0;
354
355
	(*row)[freeCol].entry += sign * entry;
      }
356
    } 
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
  }

  void DOFMatrix::addMatEntry(int row, MatEntry entry)
  {
    matrix[row].push_back(entry);
  }

  void DOFMatrix::addMatEntry(int row,  DegreeOfFreedom col, double value)
  {
    MatEntry entry;
    entry.col = col;
    entry.entry = value;
    matrix[row].push_back(entry);
  }

372
  void DOFMatrix::addRow(std::vector<MatEntry> row)
373
374
375
376
377
  {
    matrix.push_back(row);
  }

  void DOFMatrix::compressDOFIndexed(int first, int last, 
378
				     std::vector<DegreeOfFreedom> &newDOF)
379
380
  {
    int i, j, k, col;
381
    std::vector<MatEntry> *row;
382
383
384
385
386
387
388
389

    for(i = first; i <= last; i++) {
      if((k = newDOF[i]) >= 0) {
	matrix[k].swap(matrix[i]);
	matrix[i].resize(0);
      }
    }
    int usedSize = rowFESpace->getAdmin()->getUsedSize();
Thomas Witkowski's avatar
Thomas Witkowski committed
390
    for (i = 0; i < usedSize; i++) {
391
      row = reinterpret_cast< std::vector<MatEntry>*>(&(matrix[i])); 
392
      int rowSize = static_cast<int>(row->size());
Thomas Witkowski's avatar
Thomas Witkowski committed
393
      for (j = 0; j < rowSize; j++) {
394
	col = (*row)[j].col;
Thomas Witkowski's avatar
Thomas Witkowski committed
395
396
	if (entryUsed(i,j)) 
	  (*row)[j].col = newDOF[col];
397
398
399
400
401
402
      }    
    }
  }

  void DOFMatrix::freeDOFContent(int index)
  {
403
    int col = 0;
404
405
406
407

    if (0 < matrix[index].size()) {
      // for all columns in this row
      int size = static_cast<int>(matrix[index].size());
408
      for (int i = 0; i < size; i++) {
409
	// if entry is used
410
	if (entryUsed(index, i)) {
411
412
413
414
	  // get column of this entry
	  col = matrix[index][i].col;
	  if (col != index) {  // remove symmetric entry if exists
	    int colsize = static_cast<int>(matrix[col].size());
415
416
	    for (int j = 0; j < colsize; j++) {
	      int col2 = matrix[col][j].col;
417
418
	      if (col2 == index) {
		matrix[col][j].col = DOFMatrix::UNUSED_ENTRY;
419
	      } else if (col2 == DOFMatrix::NO_MORE_ENTRIES) {
420
421
422
423
		break;
	      }
	    }
	  }
424
	} else if (col == DOFMatrix::NO_MORE_ENTRIES) {
425
426
427
428
429
430
431
432
	  break;
	}
      }
      matrix[index].resize(0);	
    }
  }


433
434
  void DOFMatrix::assemble(double factor, ElInfo *elInfo, 
			   const BoundaryType *bound, Operator *op)
435
  {
436
    FUNCNAME("DOFMatrix::assemble()");
437

438
    if (!op && operators.size() == 0) {
439
      return;
440
    }
Thomas Witkowski's avatar
Thomas Witkowski committed
441
    
442
    Operator *operat = op ? op : operators[0];
Thomas Witkowski's avatar
Thomas Witkowski committed
443
    operat->getAssembler(omp_get_thread_num())->initElementMatrix(elementMatrix, elInfo);
Thomas Witkowski's avatar
Thomas Witkowski committed
444
    
445
    if (op) {
446
447
      op->getElementMatrix(elInfo, elementMatrix);
    } else {
448
449
      std::vector<Operator*>::iterator it;
      std::vector<double*>::iterator factorIt;
450
      for (it = operators.begin(), factorIt = operatorFactor.begin();	
451
	   it != operators.end(); 
452
453
454
455
456
	   ++it, ++factorIt) {
	(*it)->getElementMatrix(elInfo, 
				elementMatrix, 
				*factorIt ? **factorIt : 1.0);
      }      
457
    }
458
459
   
    addElementMatrix(factor, *elementMatrix, bound);   
460
461
  }

462
463
464
465
466
467
468
469
470
471
  void DOFMatrix::assemble(double factor, ElInfo *rowElInfo, ElInfo *colElInfo,
			   const BoundaryType *bound, Operator *op)
  {
    FUNCNAME("DOFMatrix::assemble()");

    if (!op && operators.size() == 0) {
      return;
    }

    Operator *operat = op ? op : operators[0];
472
473
    operat->getAssembler(omp_get_thread_num())->
      initElementMatrix(elementMatrix, rowElInfo, colElInfo);
474
475
476
477
478
479
480
481
482
483
484
485
486
487
    
    if (op) {
      op->getElementMatrix(rowElInfo, colElInfo, elementMatrix);
    } else {
      std::vector<Operator*>::iterator it;
      std::vector<double*>::iterator factorIt;
      for (it = operators.begin(), factorIt = operatorFactor.begin();	
	   it != operators.end(); 
	   ++it, ++factorIt) {
	(*it)->getElementMatrix(rowElInfo, colElInfo,
				elementMatrix, 
				*factorIt ? **factorIt : 1.0);
      }      
    }
488
      
489
490
491
492
    addElementMatrix(factor, *elementMatrix, bound);       
  }


493
494
495
  Flag DOFMatrix::getAssembleFlag()
  {
    Flag fillFlag(0);
496
    std::vector<Operator*>::iterator op;
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
    for(op = operators.begin(); op != operators.end(); ++op) {
      fillFlag |= (*op)->getFillFlag();
    }
    return fillFlag;
  }

  void DOFMatrix::mm(MatrixTranspose aTranspose,
		     DOFMatrix& a, 
		     MatrixTranspose bTranspose,
		     DOFMatrix& b)
  {
    FUNCNAME("DOFMatrix::mm()");

    WARNING("implementation not finished!!!\n");

512
    TEST_EXIT_DBG(a.getColFESpace() == b.getRowFESpace())
513
      ("a.colFESpace != b.rowFESpace\n");
514
    TEST_EXIT_DBG(rowFESpace == a.getRowFESpace())
515
      ("rowFESpace != a.rowFESpace\n");  
516
    TEST_EXIT_DBG(colFESpace == b.getColFESpace())
517
518
519
520
      ("colFESpace != b.colFESpace\n");  

    clear();

521
    int i, j;
522

523
    if (aTranspose == NoTranspose && bTranspose == NoTranspose) {
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
      int cols  = b.getColFESpace()->getAdmin()->getUsedSize();
      DOFMatrix::Iterator rowIterator(this, USED_DOFS);

      // for every row ...
      for(rowIterator.reset(); !rowIterator.end(); ++rowIterator) {
	int rowIndex = rowIterator.getDOFIndex();
	// and every column of result   
	for(i=0; i < cols; i++) {
	  double entry = 0.0;
	  // for every entry in a[i] ...
	  for(j=0; j < static_cast<int>( a[rowIndex].size()); j++) {
	    int logIndex = a[rowIndex][j].col;
	    int physIndex = b.logToPhysIndex(logIndex, i);
	    if(physIndex != -1) {
	      entry += a[rowIndex][j].entry * b[logIndex][physIndex].entry;
	    }
	  }
541
	  if (entry != 0.0) {
542
543
544
545
546
547
548
549
550
551
552
	    addSparseDOFEntry(1.0, rowIndex, i, entry);
	  }
	}
      }
    } else if(aTranspose == Transpose && bTranspose == NoTranspose) {
      DOFMatrix::Iterator aIterator(&a, USED_DOFS);
      DOFMatrix::Iterator bIterator(&b, USED_DOFS);
      for(aIterator.reset(), bIterator.reset(); 
	  !aIterator.end(); 
	  ++aIterator, ++bIterator) 
	{
553
554
	  std::vector<MatEntry>::const_iterator aRowIt;
	  std::vector<MatEntry>::const_iterator bRowIt;
555
556
557
558
559
560
561
562
563
564
565
	  for(aRowIt = aIterator->begin(); aRowIt != aIterator->end(); ++aRowIt) {
	    int aCol = aRowIt->col;
	    if(aCol == UNUSED_ENTRY) continue;
	    if(aCol == NO_MORE_ENTRIES) break;
	    for(bRowIt = bIterator->begin(); bRowIt !=bIterator->end(); ++bRowIt) {
	      int bCol = bRowIt->col;
	      if(bCol == UNUSED_ENTRY) continue;
	      if(bCol == NO_MORE_ENTRIES) break;

	      double entry = aRowIt->entry * bRowIt->entry;

566
	      if (entry != 0.0) {
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
		addSparseDOFEntry(1.0, aCol, bCol, entry);	    
	      }
	    }
	  }
	}
    } else if(aTranspose == NoTranspose && bTranspose == Transpose) {
      ERROR_EXIT("not yet\n");
    } else if(aTranspose == Transpose && bTranspose == Transpose) {
      ERROR_EXIT("not yet\n");
    }
  }

  void DOFMatrix::axpy(double a, 
		       const DOFMatrix& x,
		       const DOFMatrix& y)
  {
583
584
585
586
    FUNCNAME("DOFMatrix::axpy()");

    TEST_EXIT_DBG(x.getRowFESpace() == y.getRowFESpace() &&
		  rowFESpace == x.getRowFESpace())
587
      ("row fe-spaces not equal\n");
588
589
    TEST_EXIT_DBG(x.getColFESpace() == y.getColFESpace() &&
		  colFESpace == x.getColFESpace())
590
      ("col fe-spaces not equal\n");
591
    
592
593
594
595
596
597
598
599
600
601
602
603
604
605
    DOFMatrix::Iterator rowIterator(this, USED_DOFS);
    DOFMatrix::Iterator xIterator(const_cast<DOFMatrix*>(&x), USED_DOFS);
    DOFMatrix::Iterator yIterator(const_cast<DOFMatrix*>(&y), USED_DOFS);

    int i, rowIndex, colIndex;

    for(rowIterator.reset(), xIterator.reset(), yIterator.reset();
	!rowIterator.end();
	++rowIterator, ++xIterator, ++yIterator)
      {
	rowIndex = rowIterator.getDOFIndex();
	// add x contributions to this row
	for(i=0; i < static_cast<int>((*xIterator).size()); i++) {
	  colIndex = (*xIterator)[i].col;
606
	  if (colIndex >= 0) {
607
608
609
610
611
612
	    addSparseDOFEntry(a, rowIndex, colIndex, (*xIterator)[i].entry);
	  }
	}
	// add y contributions to this row
	for(i=0; i < static_cast<int>((*yIterator).size()); i++) {
	  colIndex = (*yIterator)[i].col;
613
	  if (colIndex >= 0) {
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
	    addSparseDOFEntry(1.0, rowIndex, colIndex, (*yIterator)[i].entry);
	  }
	}
      }
  }

  void DOFMatrix::scal(double b) 
  {
    int i;
    DOFMatrix::Iterator rowIterator(this, USED_DOFS);
    for(rowIterator.reset(); !rowIterator.end(); ++rowIterator) {
      for(i=0; i < static_cast<int>((*rowIterator).size()); i++) {
	if((*rowIterator)[i].col >= 0) {
	  (*rowIterator)[i].entry *= b;
	}
      }
    }
  }

  void DOFMatrix::copy(const DOFMatrix& rhs) 
  {
    clear();
    DOFMatrix::Iterator rhsIterator(const_cast<DOFMatrix*>(&rhs), USED_DOFS);
    DOFMatrix::Iterator thisIterator(this, USED_DOFS);
638
639
640
    std::vector<MatEntry>::const_iterator colIt;
    std::vector<MatEntry>::const_iterator colBegin;
    std::vector<MatEntry>::const_iterator colEnd;
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
    for(rhsIterator.reset(), thisIterator.reset();
	!rhsIterator.end(); 
	++rhsIterator, ++thisIterator) 
      {
	colBegin = rhsIterator->begin();
	colEnd = rhsIterator->end();
	for(colIt = colBegin; colIt != colEnd; ++colIt) {
	  MatEntry matEntry;
	  matEntry.col = colIt->col;
	  matEntry.entry = colIt->entry;
	  thisIterator->push_back(matEntry);
	}
      }
  }

  void DOFMatrix::createPictureFile(const char* filename, int dim)
  {
    png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);

    if (!png_ptr)
       return;

    png_bytep rowPointers[dim];
    for (int i = 0; i < dim; i++) {
      rowPointers[i] = (png_byte*)png_malloc(png_ptr, dim);

      for (int j = 0; j < dim; j++) {
	rowPointers[i][j] = 255;
      }
    }

    double scalFactor = static_cast<double>(dim) / static_cast<double>(matrix.size());

    for (int i = 0; i < static_cast<int>(matrix.size()); i++) {    
      int pi = static_cast<int>(static_cast<double>(i) * scalFactor);

677
      TEST_EXIT_DBG((pi >= 0) && (pi < dim))("PI");
678
679
680
681
682

      for (int j = 0; j < static_cast<int>(matrix[i].size()); j++) {
	
	int pj = static_cast<int>(static_cast<double>(matrix[i][j].col) * scalFactor);

683
	TEST_EXIT_DBG((pj >= 0) && (pj < dim))("PJ");
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719

	rowPointers[pi][pj] = 0;
      }
    }
   
    FILE *fp = fopen(filename, "wb");
    TEST_EXIT(fp)("Cannot open file for writing matrix picture file!\n");

    png_infop info_ptr = png_create_info_struct(png_ptr);

    if (!info_ptr) {
       png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
       return;
    }

    png_init_io(png_ptr, fp);

    png_set_IHDR(png_ptr, info_ptr, dim, dim, 8,
		 PNG_COLOR_TYPE_GRAY, 
		 PNG_INTERLACE_NONE,
		 PNG_COMPRESSION_TYPE_DEFAULT,
		 PNG_FILTER_TYPE_DEFAULT);

    png_set_rows(png_ptr, info_ptr, rowPointers);

    png_write_png(png_ptr, info_ptr, PNG_TRANSFORM_IDENTITY, NULL);

    png_destroy_write_struct(&png_ptr, &info_ptr);

    fclose(fp);
  }


  int DOFMatrix::memsize() 
  {   
    int sizeDOFMatrix = sizeof(DOFMatrix);   
720
    int sizeMatrix = sizeof(std::vector<MatrixRow>);
721
722
723
724
725
726
727
728
    for (int i = 0; i < static_cast<int>(matrix.size()); i++) {
      sizeMatrix += sizeof(MatrixRow) + matrix[i].size() * sizeof(MatEntry);
    }
    
    return sizeDOFMatrix + sizeMatrix;
  }


729
  double norm(std::vector<MatEntry> *row)
730
731
  {
    double result = 0.0;
732
    std::vector<MatEntry>::iterator it;
733
734
735
736
737
738
739
    for (it = row->begin(); it < row->end(); ++it) {
      result += (*it).entry * (*it).entry;
    }

    return(sqrt(result));
  }

740
  double min(std::vector<MatEntry> *row)
741
742
743
744
745
746
  {
    double result = 0.0;
    if (row->size() > 0) {
      result = (*row)[0].entry;
    }

747
    std::vector<MatEntry>::iterator it;
748
749
750
751
752
753
754
755
756
    for (it = row->begin(); it < row->end(); ++it) {
      if ((*it).entry < result) {
	result = (*it).entry;
      }
    }

    return(result);
  }

757
  double max(std::vector<MatEntry> *row)
758
759
760
761
762
763
  {
    double result = 0.0;
    if (row->size() > 0) {
      result = (*row)[0].entry;
    }

764
    std::vector<MatEntry>::iterator it;
765
766
767
768
769
770
771
772
773
    for (it = row->begin(); it < row->end(); ++it) {
      if ((*it).entry > result) {
	result = (*it).entry;
      }
    }

    return(result);
  }

Thomas Witkowski's avatar
Thomas Witkowski committed
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
  void addDOFMatrix(DOFMatrix *result, const DOFMatrix *a, const DOFMatrix *b)
  {
    result->clear();

    DOFMatrix::Iterator resultIterator(result, USED_DOFS);
    DOFMatrix::Iterator aIterator(const_cast<DOFMatrix*>(a), USED_DOFS);
    DOFMatrix::Iterator bIterator(const_cast<DOFMatrix*>(b), USED_DOFS);

    for (resultIterator.reset(), aIterator.reset(); 
	 !aIterator.end(); 
	 ++resultIterator, ++aIterator) {

      *resultIterator = *aIterator;
    }      

    for (resultIterator.reset(), bIterator.reset(); 
	 !bIterator.end(); 
	 ++resultIterator, ++bIterator) {
      std::vector<MatEntry>::iterator resultRowIt;
      std::vector<MatEntry>::const_iterator bRowIt;

      for (bRowIt = bIterator->begin();
	   bRowIt != bIterator->end();
	   ++bRowIt) {
	bool added = false;
	for (resultRowIt = resultIterator->begin();
	     resultRowIt != resultIterator->end();
	     ++resultRowIt) {
	  if (bRowIt->col == resultRowIt->col) {
	    resultRowIt->entry += bRowIt->entry;
	    
	    added = true;
	    break;
	  }
	}

	if (!added) {
	  resultIterator->push_back(*bRowIt);
	}
      }	         
    }      
  }

817
}