Skip to content
Snippets Groups Projects
Commit 4612a93a authored by Elias Pipping's avatar Elias Pipping
Browse files

[Cleanup] Renaming vertex variables

parent b4237a76
No related branches found
No related tags found
No related merge requests found
...@@ -193,7 +193,7 @@ int main(int argc, char *argv[]) { ...@@ -193,7 +193,7 @@ int main(int argc, char *argv[]) {
auto const refinements = parset.get<size_t>("grid.refinements"); auto const refinements = parset.get<size_t>("grid.refinements");
grid->globalRefine(refinements); grid->globalRefine(refinements);
size_t const finestSize = grid->size(grid->maxLevel(), dims); size_t const fineVertexCount = grid->size(grid->maxLevel(), dims);
using GridView = GridType::LeafGridView; using GridView = GridType::LeafGridView;
GridView const leafView = grid->leafView(); GridView const leafView = grid->leafView();
...@@ -208,23 +208,22 @@ int main(int argc, char *argv[]) { ...@@ -208,23 +208,22 @@ int main(int argc, char *argv[]) {
Assembler<P1Basis, P1Basis> const p1Assembler(p1Basis, p1Basis); Assembler<P1Basis, P1Basis> const p1Assembler(p1Basis, p1Basis);
// Set up the boundary // Set up the boundary
Dune::BitSetVector<dims> velocityDirichletNodes(finestSize, false); Dune::BitSetVector<dims> velocityDirichletNodes(fineVertexCount, false);
Dune::BitSetVector<dims> const &displacementDirichletNodes = Dune::BitSetVector<dims> const &displacementDirichletNodes =
velocityDirichletNodes; velocityDirichletNodes;
Dune::BitSetVector<dims> accelerationDirichletNodes(finestSize, false); Dune::BitSetVector<dims> accelerationDirichletNodes(fineVertexCount, false);
Dune::BitSetVector<1> neumannNodes(finestSize, false); Dune::BitSetVector<1> neumannNodes(fineVertexCount, false);
Dune::BitSetVector<1> frictionalNodes(finestSize, false); Dune::BitSetVector<1> frictionalNodes(fineVertexCount, false);
VectorType coordinates(finestSize); VectorType vertexCoordinates(fineVertexCount);
{ {
Dune::MultipleCodimMultipleGeomTypeMapper< Dune::MultipleCodimMultipleGeomTypeMapper<
GridView, Dune::MCMGVertexLayout> const myVertexMapper(leafView); GridView, Dune::MCMGVertexLayout> const vertexMapper(leafView);
for (auto it = leafView.begin<dims>(); it != leafView.end<dims>(); ++it) { for (auto it = leafView.begin<dims>(); it != leafView.end<dims>(); ++it) {
assert(it->geometry().corners() == 1); assert(it->geometry().corners() == 1);
size_t const id = myVertexMapper.map(*it); size_t const id = vertexMapper.map(*it);
coordinates[id] = it->geometry().corner(0); vertexCoordinates[id] = it->geometry().corner(0);
auto const &localCoordinates = coordinates[id]; auto const &localCoordinates = vertexCoordinates[id];
// lower face // lower face
if (localCoordinates[1] == lowerLeft[1]) { if (localCoordinates[1] == lowerLeft[1]) {
...@@ -350,11 +349,11 @@ int main(int argc, char *argv[]) { ...@@ -350,11 +349,11 @@ int main(int argc, char *argv[]) {
neumannFunction, _relativeTime); neumannFunction, _relativeTime);
_ell += gravityFunctional; _ell += gravityFunctional;
}; };
VectorType ell(finestSize); VectorType ell(fineVertexCount);
createRHS(0.0, ell); createRHS(0.0, ell);
// {{{ Initial conditions // {{{ Initial conditions
SingletonVectorType alpha_initial(finestSize); SingletonVectorType alpha_initial(fineVertexCount);
alpha_initial = alpha_initial =
std::log(parset.get<double>("boundary.friction.initialState")); std::log(parset.get<double>("boundary.friction.initialState"));
...@@ -365,7 +364,7 @@ int main(int argc, char *argv[]) { ...@@ -365,7 +364,7 @@ int main(int argc, char *argv[]) {
ZeroNonlinearity<SmallVector, SmallMatrix> zeroNonlinearity; ZeroNonlinearity<SmallVector, SmallMatrix> zeroNonlinearity;
// Solve the stationary problem // Solve the stationary problem
VectorType u_initial(finestSize); VectorType u_initial(fineVertexCount);
u_initial = 0.0; u_initial = 0.0;
{ {
LinearFactoryType displacementFactory(parset.sub("solver.tnnmg"), // FIXME LinearFactoryType displacementFactory(parset.sub("solver.tnnmg"), // FIXME
...@@ -389,7 +388,7 @@ int main(int argc, char *argv[]) { ...@@ -389,7 +388,7 @@ int main(int argc, char *argv[]) {
initialDisplacementProblemSolver.preprocess(); initialDisplacementProblemSolver.preprocess();
initialDisplacementProblemSolver.solve(); initialDisplacementProblemSolver.solve();
} }
VectorType v_initial(finestSize); VectorType v_initial(fineVertexCount);
v_initial = 0.0; v_initial = 0.0;
{ {
// Prescribe a homogeneous velocity field in the x-direction // Prescribe a homogeneous velocity field in the x-direction
...@@ -400,13 +399,13 @@ int main(int argc, char *argv[]) { ...@@ -400,13 +399,13 @@ int main(int argc, char *argv[]) {
for (auto &x : v_initial) for (auto &x : v_initial)
x[0] = v_initial_const; x[0] = v_initial_const;
} }
VectorType a_initial(finestSize); VectorType a_initial(fineVertexCount);
a_initial = 0.0; a_initial = 0.0;
{ {
/* We solve Au + Cv + Ma + Psi(v) = ell, thus /* We solve Au + Cv + Ma + Psi(v) = ell, thus
Ma = - (Au + Cv + Psi(v) - ell) Ma = - (Au + Cv + Psi(v) - ell)
*/ */
VectorType accelerationRHS(finestSize); VectorType accelerationRHS(fineVertexCount);
{ {
accelerationRHS = 0.0; accelerationRHS = 0.0;
Arithmetic::addProduct(accelerationRHS, A, u_initial); Arithmetic::addProduct(accelerationRHS, A, u_initial);
...@@ -453,11 +452,11 @@ int main(int argc, char *argv[]) { ...@@ -453,11 +452,11 @@ int main(int argc, char *argv[]) {
auto multigridStep = factory.getSolver(); auto multigridStep = factory.getSolver();
{ {
std::fstream coordinateWriter("coordinates", std::fstream::out); std::fstream vertexCoordinateWriter("coordinates", std::fstream::out);
for (size_t i = 0; i < frictionalNodes.size(); ++i) for (size_t i = 0; i < frictionalNodes.size(); ++i)
if (frictionalNodes[i][0]) if (frictionalNodes[i][0])
coordinateWriter << coordinates[i] << std::endl; vertexCoordinateWriter << vertexCoordinates[i] << std::endl;
coordinateWriter.close(); vertexCoordinateWriter.close();
} }
std::fstream iterationWriter("iterations", std::fstream::out), std::fstream iterationWriter("iterations", std::fstream::out),
relaxationWriter("relaxation", std::fstream::out); relaxationWriter("relaxation", std::fstream::out);
...@@ -471,7 +470,7 @@ int main(int argc, char *argv[]) { ...@@ -471,7 +470,7 @@ int main(int argc, char *argv[]) {
alpha_initial, frictionalNodes, frictionData); alpha_initial, frictionalNodes, frictionData);
VectorType v = v_initial; VectorType v = v_initial;
SingletonVectorType alpha(finestSize); SingletonVectorType alpha(fineVertexCount);
auto const timesteps = parset.get<size_t>("timeSteps.number"), auto const timesteps = parset.get<size_t>("timeSteps.number"),
maximumStateFPI = parset.get<size_t>("v.fpi.maximumIterations"), maximumStateFPI = parset.get<size_t>("v.fpi.maximumIterations"),
...@@ -497,8 +496,8 @@ int main(int argc, char *argv[]) { ...@@ -497,8 +496,8 @@ int main(int argc, char *argv[]) {
createRHS(relativeTime, ell); createRHS(relativeTime, ell);
MatrixType velocityMatrix; MatrixType velocityMatrix;
VectorType velocityRHS(finestSize); VectorType velocityRHS(fineVertexCount);
VectorType velocityIterate(finestSize); VectorType velocityIterate(fineVertexCount);
stateUpdater->setup(tau); stateUpdater->setup(tau);
timeSteppingScheme->setup(ell, tau, relativeTime, velocityRHS, timeSteppingScheme->setup(ell, tau, relativeTime, velocityRHS,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment