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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
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
124
125
126
127
128
129
130
131
132
|
// Gmsh - Copyright (C) 1997-2020 C. Geuzaine, J.-F. Remacle
//
// See the LICENSE.txt file for license information. Please report all
// issues on https://gitlab.onelab.info/gmsh/gmsh/issues.
#ifndef ELASTICITY_SOLVER_H
#define ELASTICITY_SOLVER_H
#include <map>
#include <string>
#include "GmshConfig.h"
#include "SVector3.h"
#include "dofManager.h"
#include "functionSpace.h"
template <class scalar> class simpleFunction;
class GModel;
class PView;
class groupOfElements;
class gLevelset;
struct LagrangeMultiplierField {
int _tag;
groupOfElements *g;
double _tau;
SVector3 _d;
simpleFunction<double> *_f;
LagrangeMultiplierField() : _tag(0), g(0) {}
};
struct elasticField {
int _tag; // tag for the dofManager
groupOfElements *g; // support for this field
double _e, _nu; // specific elastic datas (should be somewhere else)
elasticField() : _tag(0), g(0) {}
};
struct BoundaryCondition {
int _tag; // tag for the dofManager
enum location { UNDEF, ON_VERTEX, ON_EDGE, ON_FACE, ON_VOLUME };
location onWhat; // on vertices or elements
groupOfElements *g; // support for this BC
BoundaryCondition() : _tag(0), onWhat(UNDEF), g(0) {}
};
struct dirichletBC : public BoundaryCondition {
int _comp; // component
simpleFunction<double> *_f;
dirichletBC() : BoundaryCondition(), _comp(0), _f(0) {}
};
struct neumannBC : public BoundaryCondition {
simpleFunction<SVector3> *_f;
neumannBC() : BoundaryCondition(), _f(NULL) {}
};
// an elastic solver ...
class elasticitySolver {
public:
GModel *pModel;
int _dim, _tag;
dofManager<double> *pAssembler;
FunctionSpace<SVector3> *LagSpace;
std::vector<FunctionSpace<double> *> LagrangeMultiplierSpaces;
// young modulus and poisson coefficient per physical
std::vector<elasticField> elasticFields;
std::vector<LagrangeMultiplierField> LagrangeMultiplierFields;
// neumann BC
std::vector<neumannBC> allNeumann;
// dirichlet BC
std::vector<dirichletBC> allDirichlet;
public:
elasticitySolver(int tag) : _tag(tag), pAssembler(0), LagSpace(0) {}
elasticitySolver(GModel *model, int tag);
void addDirichletBC(int dim, int entityId, int component, double value);
void addDirichletBC(int dim, std::string phys, int component, double value);
void addNeumannBC(int dim, int entityId, const std::vector<double> value);
void addNeumannBC(int dim, std::string phys, const std::vector<double> value);
void addElasticDomain(int tag, double e, double nu);
void addElasticDomain(std::string phys, double e, double nu);
virtual ~elasticitySolver()
{
if(LagSpace) delete LagSpace;
for(std::size_t i = 0; i < LagrangeMultiplierSpaces.size(); i++)
if(LagrangeMultiplierSpaces[i]) delete LagrangeMultiplierSpaces[i];
LagrangeMultiplierSpaces.clear();
if(pAssembler) delete pAssembler;
}
void assemble(linearSystem<double> *lsys);
void readInputFile(const std::string &meshFileName);
void read(const std::string s) { readInputFile(s.c_str()); }
virtual void setMesh(const std::string &meshFileName, int dim = 0);
void cutMesh(gLevelset *ls);
void setElasticDomain(int phys, double E, double nu);
void setLagrangeMultipliers(int phys, double tau, const SVector3 &d, int tag,
simpleFunction<double> *f);
void changeLMTau(int tag, double tau);
void setEdgeDisp(int edge, int comp, simpleFunction<double> *f);
void solve();
void postSolve();
void exportKb();
void computeEffectiveStiffness(std::vector<double> stiff);
void computeEffectiveStrain(std::vector<double> strain);
double computeDisplacementError(simpleFunction<double> *f0,
simpleFunction<double> *f1,
simpleFunction<double> *f2);
double computeL2Norm(simpleFunction<double> *f0, simpleFunction<double> *f1,
simpleFunction<double> *f2);
double computeLagNorm(int tag, simpleFunction<double> *f);
#if defined(HAVE_POST)
virtual PView *buildDisplacementView(const std::string postFileName);
virtual PView *buildStrainView(const std::string postFileName);
virtual PView *buildStressesView(const std::string postFileName);
virtual PView *buildLagrangeMultiplierView(const std::string &posFileName,
int tag = -1);
virtual PView *buildElasticEnergyView(const std::string postFileName);
virtual PView *buildVonMisesView(const std::string postFileName);
virtual PView *buildVolumeView(const std::string postFileName);
virtual PView *buildErrorView(const std::string postFileName,
simpleFunction<double> *f0,
simpleFunction<double> *f1,
simpleFunction<double> *f2);
#endif
};
#endif
|
