Commit d508988c authored by incardon's avatar incardon

Adding missing files

parent 15bd7db0
This diff is collapsed.
/*
* petsc_solver_unit_tests.hpp
*
* Created on: Jun 19, 2017
* Author: i-bird
*/
#ifndef OPENFPM_NUMERICS_SRC_SOLVERS_PETSC_SOLVER_UNIT_TESTS_CPP_
#define OPENFPM_NUMERICS_SRC_SOLVERS_PETSC_SOLVER_UNIT_TESTS_CPP_
#define BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>
#include "petsc_solver_report_unit_tests.hpp"
#include "Grid/grid_dist_id.hpp"
#include "Matrix/SparseMatrix.hpp"
#include "Vector/Vector.hpp"
#include "FiniteDifference/Laplacian.hpp"
#include "FiniteDifference/FDScheme.hpp"
#include "Solvers/petsc_solver.hpp"
BOOST_AUTO_TEST_SUITE( mg_solvers_test )
BOOST_AUTO_TEST_CASE( laplacian_3D_int_zero_mg )
{
constexpr unsigned int phi = 0;
typedef Field<phi,poisson_nn_helm> phi_f;
Vcluster & v_cl = create_vcluster();
if (v_cl.getProcessingUnits() != 3)
return;
Ghost<3,long int> g(2);
Ghost<3,long int> stencil_max(2);
Box<3,float> domain({0.0,0.0,0.0},{1.0,1.0,1.0});
periodicity<3> p({PERIODIC,PERIODIC,PERIODIC});
typedef Lap<phi_f,poisson_nn_helm,CENTRAL_SYM> poisson;
grid_dist_id<3,float,aggregate<float>> psi({64,64,64},domain,g,p);
grid_dist_id<3,float,aggregate<float>> psi2(psi.getDecomposition(),{64,64,64},g);
// Fill B
size_t center_x = psi.size(0) / 2;
size_t center_y = psi.size(1) / 2;
size_t center_z = psi.size(2) / 2;
auto it = psi.getDomainIterator();
while (it.isNext())
{
auto key = it.get();
auto gkey = it.getGKey(key);
float sx = (float)(gkey.get(0))-center_x;
float sy = (float)(gkey.get(1))-center_y;
float sz = (float)(gkey.get(2))-center_z;
float gs = 100.0*exp(-((sx*sx)+(sy*sy)+(sz*sz))/100.0);
psi.get<0>(key) = sin(2*M_PI*sx/psi.size(0))*sin(2*M_PI*sy/psi.size(1))*sin(2*M_PI*sz/psi.size(2))*gs;
psi2.get<0>(key) = sin(2*M_PI*sx/psi.size(0))*sin(2*M_PI*sy/psi.size(1))*sin(2*M_PI*sz/psi.size(2))*gs;
++it;
}
FDScheme<poisson_nn_helm> fd(stencil_max, domain, psi);
fd.template impose_dit<0>(poisson(),psi,psi.getDomainIterator());
petsc_solver<double> solver;
solver.setSolver(KSPBCGS);
solver.setAbsTol(0.01);
solver.setMaxIter(500);
////////
solver.setPreconditioner(PCHYPRE_BOOMERAMG);
solver.setPreconditionerAMG_nl(6);
solver.setPreconditionerAMG_maxit(3);
solver.setPreconditionerAMG_relax("SOR/Jacobi");
solver.setPreconditionerAMG_cycleType("V",6,6);
solver.setPreconditionerAMG_coarsen("HMIS");
solver.setPreconditionerAMG_coarsenNodalType(0);
timer tm_solve;
tm_solve.start();
auto x_ = solver.solve(fd.getA(),fd.getB());
tm_solve.stop();
fd.template copy<phi>(x_,psi);
psi.write("AMG_psi");
bool check = compare("AMG_psi_" + std::to_string(v_cl.getProcessUnitID()) + ".vtk","test/AMG_psi_test_" + std::to_string(v_cl.getProcessUnitID()) + ".vtk");
BOOST_REQUIRE_EQUAL(check,true);
// Resolve
timer tm_solve2;
tm_solve2.start();
auto x2_ = solver.solve(fd.getB());
tm_solve2.stop();
fd.template copy<phi>(x_,psi2);
psi2.write("AMG_psi2");
check = compare("AMG_psi2_" + std::to_string(v_cl.getProcessUnitID()) + ".vtk","test/AMG_psi2_test_" + std::to_string(v_cl.getProcessUnitID()) + ".vtk");
BOOST_REQUIRE_EQUAL(check,true);
}
BOOST_AUTO_TEST_SUITE_END()
#endif /* OPENFPM_NUMERICS_SRC_SOLVERS_PETSC_SOLVER_UNIT_TESTS_CPP_ */
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment