ElInfo.cc 4.42 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
#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 {

  int ElInfo::traverseId = -1;
  int ElInfo::traverseIdCounter = 0;

  ElInfo::ElInfo(Mesh *aMesh) 
    : mesh_(aMesh),
      element_(NULL),
      parent_(NULL),
      macroElement_(NULL),
25
26
      level_(0),
      iChild(0),
27
28
29
30
31
      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),
32
      neighbourCoord_(mesh_->getDim(), NO_INIT),
33
      oppVertex_(mesh_->getDim(), NO_INIT),
34
      grdLambda_(mesh_->getDim(), NO_INIT),
35
      subElemCoords(NULL)
36
37
  {
    projection_.set(NULL);
38
39
40
41

    for (int i = 0; i < neighbourCoord_.getSize(); i++) {
      neighbourCoord_[i].init(mesh_->getDim());
    }
Thomas Witkowski's avatar
Thomas Witkowski committed
42
43

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

  ElInfo::~ElInfo()
  {
48
49
50
    if (subElemCoords) {
      DELETE subElemCoords;
    }
51
52
53
54
55
56
57
58
59
60
61
62
63
64
  }


  /****************************************************************************/
  /*  transform local coordintes l to world coordinates; if w is non NULL      */
  /*  store them at w otherwise return a pointer to some local static         */
  /*  area containing world coordintes                                        */
  /****************************************************************************/
  const WorldVector<double> *ElInfo::coordToWorld(const DimVec<double>& l, 
						  WorldVector<double>* w) const

  {
    int dim = l.getSize() - 1;

65
    static WorldVector<double> world[4];
66
    WorldVector<double> *ret = w ? w : &world[omp_get_thread_num()];
67
    double c = l[0];
68

69
    for (int j = 0; j < dimOfWorld; j++)
Thomas Witkowski's avatar
Thomas Witkowski committed
70
      (*ret)[j] = c * coord_[0][j];
71
  
Thomas Witkowski's avatar
Thomas Witkowski committed
72
    int vertices = Global::getGeo(VERTEX, dim);
73

74
    for (int i = 1; i < vertices; i++) {
75
      c = l[i];
76
      for (int j = 0; j < dimOfWorld; j++)
Thomas Witkowski's avatar
Thomas Witkowski committed
77
	(*ret)[j] += c * coord_[i][j];
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
    }
    return ret;
  }

  /****************************************************************************/
  /*  compute volume of an element                                            */
  /****************************************************************************/

  double ElInfo::calcDet() const
  {
    testFlag(Mesh::FILL_COORDS);
    return calcDet(coord_);
  }

  double ElInfo::calcDet(const FixVec<WorldVector<double>, VERTEX> &coords)
  {
94
    FUNCNAME("ElInfo::calcDet()");
95
96
97
98

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

99
    TEST_EXIT_DBG(dim <= dow)("dim > dow\n");
100
101
102
103
104
105
106
107
108
109
110

    double det = 0.0;

    if (dim == 0)
      return 1.0;

    // dim = dim of world
    WorldVector<double> e1, e2, e3, v0;

    v0 = coords[0];

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

  void ElInfo::testFlag(const Flag& flags) const
  {
159
    TEST_EXIT_DBG(fillFlag_.isSet(flags))("flag not set\n");
160
161
162
163
164
  }


  void ElInfo::fillDetGrdLambda() 
  { 
165
    if (fillFlag_.isSet(Mesh::FILL_GRD_LAMBDA)) {
166
167
      det_ = calcGrdLambda(grdLambda_);
    } else {
168
      if (fillFlag_.isSet(Mesh::FILL_DET)) {
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
	det_ = calcDet();
      }
    }
  }

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

    int dim = mesh_->getDim();
    int posIndex = DIM_OF_INDEX(pos, dim);
184
    int offset = indexOffset[dim - 1][posIndex];
185
    
186
187
188
189
    return boundary_[offset + i];
  }

}