Homec4science
Diffusion akantu (features/68-automatic-differentiation) Browse

rAKA/test/test_model/test_model_solver/test_dof_manager_default.cc
ca61641b304bfeatures/68-automatic…

Last Change
Raw File

test_dof_manager_default.cc
View Options

/**
* @file test_dof_manager_default.cc
*
* @author Nicolas Richart <nicolas.richart@epfl.ch>
*
* @date creation: Fri Feb 26 2016
* @date last modification: Thu Feb 01 2018
*
* @brief Test default dof manager
*
* @section LICENSE
*
* Copyright (©) 2016-2018 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* Akantu is free software: you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* Akantu is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#include "dof_manager_default.hh"
#include "solver_callback.hh"
#include "sparse_matrix_aij.hh"
#include "time_step_solver.hh"
using namespace akantu;
/**
* =\o-----o-----o-> F
* | |
* |---- L ----|
*/
class MySolverCallback : public SolverCallback {
public:
MySolverCallback(Real F, DOFManagerDefault & dof_manager, UInt nb_dofs = 3)
: dof_manager(dof_manager), dispacement(nb_dofs, 1, "disp"),
blocked(nb_dofs, 1), forces(nb_dofs, 1), nb_dofs(nb_dofs) {
dof_manager.registerDOFs("disp", dispacement, _dst_generic);
dof_manager.registerBlockedDOFs("disp", blocked);
dispacement.set(0.);
forces.set(0.);
blocked.set(false);
forces(nb_dofs - 1, _x) = F;
blocked(0, _x) = true;
}
void assembleMatrix(const ID & matrix_id) {
if (matrix_id != "K")
return;
auto & K = dynamic_cast<SparseMatrixAIJ &>(dof_manager.getMatrix("K"));
K.clear();
for (UInt i = 1; i < nb_dofs - 1; ++i)
K.add(i, i, 2.);
for (UInt i = 0; i < nb_dofs - 1; ++i)
K.add(i, i + 1, -1.);
K.add(0, 0, 1);
K.add(nb_dofs - 1, nb_dofs - 1, 1);
// K *= 1 / L_{el}
K *= nb_dofs - 1;
}
MatrixType getMatrixType(const ID & matrix_id) {
if (matrix_id == "K")
return _symmetric;
return _mt_not_defined;
}
void assembleLumpedMatrix(const ID &) {}
void assembleResidual() { dof_manager.assembleToResidual("disp", forces); }
void predictor() {}
void corrector() {}
DOFManagerDefault & dof_manager;
Array<Real> dispacement;
Array<bool> blocked;
Array<Real> forces;
UInt nb_dofs;
};
int main(int argc, char * argv[]) {
initialize(argc, argv);
DOFManagerDefault dof_manager("test_dof_manager");
MySolverCallback callback(10., dof_manager, 11);
NonLinearSolver & nls =
dof_manager.getNewNonLinearSolver("my_nls", _nls_linear);
TimeStepSolver & tss =
dof_manager.getNewTimeStepSolver("my_tss", _tsst_static, nls);
tss.setIntegrationScheme("disp", _ist_pseudo_time);
tss.solveStep(callback);
dof_manager.getMatrix("K").saveMatrix("K_dof_manager_default.mtx");
Array<Real>::const_scalar_iterator disp_it = callback.dispacement.begin();
Array<Real>::const_scalar_iterator force_it = callback.forces.begin();
Array<bool>::const_scalar_iterator blocked_it = callback.blocked.begin();
std::cout << std::setw(8) << "disp"
<< " " << std::setw(8) << "force"
<< " " << std::setw(8) << "blocked" << std::endl;
for (; disp_it != callback.dispacement.end();
++disp_it, ++force_it, ++blocked_it) {
std::cout << std::setw(8) << *disp_it << " " << std::setw(8) << *force_it
<< " " << std::setw(8) << std::boolalpha << *blocked_it
<< std::endl;
}
finalize();
return EXIT_SUCCESS;
}
c4science · Help