Homec4science
Diffusion µSpectre (master) Browse

muSpectre/bin/test_plain_mat4.cc
ed63229013eamaster

Last Change
Raw File

test_plain_mat4.cc
View Options

//cpp script of test_plain_mat4.py necessary to better profile the program
#include "cell/cell_factory.hh"
#include "materials/material_linear_elastic4.hh"
#include "solver/solvers.hh"
#include "solver/solver_cg.hh"
#include <iostream>
//#include <iomanip>
#include <random>
using namespace muSpectre;
int main()
{
std::cout << "startig test_plain_mat4\n";
const Ccoord_t<threeD> resolution{51, 51, 51};
const Rcoord_t<threeD> lengths{5.0, 5.0, 5.0};
constexpr Formulation formulation{Formulation::finite_strain};
auto cell{make_cell<threeD, threeD>(resolution, lengths, formulation)};
//initialize my material
using Material_t = MaterialLinearElastic4<threeD, threeD>;
auto mat{std::make_unique<Material_t>("material")};
//introduce random number generators for Young and Poisson
//make random numbers as in std::uniform_real_distribution cppreference.com
std::random_device rd;
std::mt19937 gen(rd());
//set a seed for the random number generation to make reproducable results
gen.seed(15);
// for Young [5,10)
std::uniform_real_distribution<> dis_Y(5, 10);
//for Poisson [0.1, 0.4)
std::uniform_real_distribution<> dis_P(0.1, 0.4);
//fill the pixels
for (const auto && pixel:cell) {
mat->add_pixel(pixel, dis_Y(gen), dis_P(gen));
}
cell.add_material(std::move(mat)); //add the material to the cell
cell.initialise(FFT_PlanFlags::measure); //fft initialization, make faster fft
//set the deformation
//DelF = np.array([[0, 0.01, 0],
// [0, 0 , 0],
// [0, 0 , 0]])
Grad_t<threeD> DelF{Grad_t<threeD>::Zero()};
DelF(0, 1) = 0.01;
//set solver constants
const Real newton_tol {1e-6}; //tolerance for newton algo
const Real cg_tol {1e-6}; //tolerance for cg algo
const Real equil_tol {1e-6}; //tolerance for equilibrium / tol for div(P) ?
const Uint maxiter {100}; //maximum cg iterations
const Uint verbose {0}; //verbosity of solver
GradIncrements<threeD> grads{DelF};
SolverCG<threeD> cg{cell, cg_tol, maxiter, bool(verbose)};
auto newton_result {newton_cg(cell, grads, cg,
newton_tol, equil_tol, verbose)[0]};
//print some messages of the solver
/*
std::cout << "gradient F:\n" << newton_result.grad << "\n";
std::cout << "\n\nstress:\n" << newton_result.stress << "\n";
std::cout << "convergence_test: " << newton_result.success << "\n";
std::cout << "status: " << newton_result.status << "\n";
std::cout << "message: " << newton_result.message << "\n";
std::cout << "# iterations: " << newton_result.nb_it << "\n";
std::cout << "# cell evaluations: " << newton_result.nb_fev << "\n";
*/
std::cout << "test_plain_mat4 finished correct\n";
}
c4science · Help