ElInfo.cc 3.87 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#include "ElInfo.h"
#include "BasisFunction.h"
#include "Element.h"
#include "Line.h"
#include "Triangle.h"
#include "Tetrahedron.h"
#include "FiniteElemSpace.h"
#include "Flag.h"
#include "MacroElement.h"
#include "Mesh.h"
#include "Global.h"
#include "FixVec.h"
#include "DOFVector.h"

namespace AMDiS {

Thomas Witkowski's avatar
Thomas Witkowski committed
17
  std::vector<std::map<std::pair<int, unsigned long>, mtl::dense2D<double> > > ElInfo::subElemMatrices(5);
18

Thomas Witkowski's avatar
Thomas Witkowski committed
19
  std::vector<std::map<std::pair<int, unsigned long>, mtl::dense2D<double> > > ElInfo::subElemGradMatrices(5);
20

21
  ElInfo::ElInfo(Mesh *aMesh) 
Thomas Witkowski's avatar
Thomas Witkowski committed
22
23
24
25
    : mesh(aMesh),
      element(NULL),
      parent(NULL),
      macroElement(NULL),
26
      level(0),
27
      elType(0),
28
      iChild(0),
Thomas Witkowski's avatar
Thomas Witkowski committed
29
30
31
32
33
34
35
36
      coord(mesh->getDim(), NO_INIT),
      boundary(mesh->getDim(), DEFAULT_VALUE, INTERIOR),
      projection(mesh->getDim(), NO_INIT),
      oppCoord(mesh->getDim(), NO_INIT),
      neighbour(mesh->getDim(), NO_INIT),
      neighbourCoord(mesh->getDim(), NO_INIT),
      oppVertex(mesh->getDim(), NO_INIT),
      grdLambda(mesh->getDim(), NO_INIT),
37
38
      refinementPath(0),
      refinementPathLength(0)
39
  {
Thomas Witkowski's avatar
Thomas Witkowski committed
40
    projection.set(NULL);
41

42
    for (int i = 0; i < neighbourCoord.getSize(); i++)
Thomas Witkowski's avatar
Thomas Witkowski committed
43
      neighbourCoord[i].init(mesh->getDim());
Thomas Witkowski's avatar
Thomas Witkowski committed
44
45

    dimOfWorld = Global::getGeo(WORLD);
46
47
48
  }

  ElInfo::~ElInfo()
49
  {}
50

51
52
  void ElInfo::coordToWorld(const DimVec<double>& l, 
			    WorldVector<double>& w) const
53
54

  {
55
56
    testFlag(Mesh::FILL_COORDS);

57
    double c = l[0];
58

59
    for (int j = 0; j < dimOfWorld; j++)
Thomas Witkowski's avatar
Thomas Witkowski committed
60
      w[j] = c * coord[0][j];
61
  
62
    int vertices = Global::getGeo(VERTEX, l.getSize() - 1);
63

64
    for (int i = 1; i < vertices; i++) {
65
      c = l[i];
66
      for (int j = 0; j < dimOfWorld; j++)
Thomas Witkowski's avatar
Thomas Witkowski committed
67
	w[j] += c * coord[i][j];
68
69
70
71
72
73
    }
  }

  double ElInfo::calcDet() const
  {
    testFlag(Mesh::FILL_COORDS);
Thomas Witkowski's avatar
Thomas Witkowski committed
74
    return calcDet(coord);
75
76
  }

Thomas Witkowski's avatar
Thomas Witkowski committed
77
  double ElInfo::calcDet(const FixVec<WorldVector<double>, VERTEX> &coords) const
78
  {
79
    FUNCNAME("ElInfo::calcDet()");
80
81
82
83

    int dim = coords.getSize() - 1;
    int dow = Global::getGeo(WORLD);

84
    TEST_EXIT_DBG(dim <= dow)("dim > dow\n");
85
86
87
88
89
90

    double det = 0.0;

    if (dim == 0)
      return 1.0;

Thomas Witkowski's avatar
Thomas Witkowski committed
91
    switch (dow) {
92
93
94
95
    case 1:
      det = coords[1][0] - coords[0][0];
      break;
    case 2:
96
      if (dim == 1) {
Thomas Witkowski's avatar
Thomas Witkowski committed
97
98
99
100
101

	WorldVector<double> e1;

	e1[0] = coords[1][0] - coords[0][0];
	e1[1] = coords[1][1] - coords[0][1];
102
	det = norm(&e1);
Thomas Witkowski's avatar
Thomas Witkowski committed
103

104
      } else {
Thomas Witkowski's avatar
Thomas Witkowski committed
105
106
107
108

	det = (coords[1][0] - coords[0][0]) * (coords[2][1] - coords[0][1]) - 
	  (coords[1][1] - coords[0][1]) * (coords[2][0] - coords[0][0]);

109
110
111
112
      }
      break;
    case 3: 
      {
Thomas Witkowski's avatar
Thomas Witkowski committed
113
	WorldVector<double> e1, e2, e3, n;
114
	
Thomas Witkowski's avatar
Thomas Witkowski committed
115
116
117
118
119
120
121
122
	for (int i = 0; i < dow; i++) {
	  e1[i] = coords[1][i] - coords[0][i];
	  if (dim > 1)
	    e2[i] = coords[2][i] - coords[0][i];
	  if (dim > 2)
	    e3[i] = coords[3][i] - coords[0][i];
	}
  
123
	if (dim > 1) {
Thomas Witkowski's avatar
Thomas Witkowski committed
124
125
126
	  n[0] = e1[1] * e2[2] - e1[2] * e2[1];
	  n[1] = e1[2] * e2[0] - e1[0] * e2[2];
	  n[2] = e1[0] * e2[1] - e1[1] * e2[0];
127
	}
128
	
129
	if (dim == 1) {
130
131
132
133
	  det = norm(&e1);
	} else if (dim == 2) {
	  det = norm(&n);
	} else if (dim == 3) {
134
	  det = n[0] * e3[0] + n[1] * e3[1] + n[2] * e3[2];
135
	} else
Thomas Witkowski's avatar
Thomas Witkowski committed
136
	  ERROR_EXIT("not yet for problem dimension = %d", dim);
137
138
	break;
      } 
139
    default:
Thomas Witkowski's avatar
Thomas Witkowski committed
140
      ERROR_EXIT("not yet for Global::getGeo(WORLD) = %d", dow);
141
    }
142
    
143
144
145
    return abs(det);
  }

Thomas Witkowski's avatar
Thomas Witkowski committed
146

147
  void ElInfo::testFlag(const Flag& flags) const
Thomas Witkowski's avatar
Thomas Witkowski committed
148
  {    
149
    TEST_EXIT_DBG(fillFlag.isSet(flags))("flag not set\n");
150
151
152
153
154
  }


  void ElInfo::fillDetGrdLambda() 
  { 
Thomas Witkowski's avatar
Thomas Witkowski committed
155
156
    if (fillFlag.isSet(Mesh::FILL_GRD_LAMBDA)) {
      det = calcGrdLambda(grdLambda);
157
    } else {
Thomas Witkowski's avatar
Thomas Witkowski committed
158
159
      if (fillFlag.isSet(Mesh::FILL_DET))
	det = calcDet();
160
161
162
163
164
165
166
167
168
169
170
    }
  }

  BoundaryType ElInfo::getBoundary(GeoIndex pos, int i)
  {
    static int indexOffset[3][3] = {
      { 0, 0, 0},
      { 3, 0, 0},
      {10, 4, 0}
    };

Thomas Witkowski's avatar
Thomas Witkowski committed
171
    int dim = mesh->getDim();
172
    int posIndex = DIM_OF_INDEX(pos, dim);
173
    int offset = indexOffset[dim - 1][posIndex];
174
    
Thomas Witkowski's avatar
Thomas Witkowski committed
175
    return boundary[offset + i];
176
177
178
  }

}