/** Nonadaptive solver.*/
void solveNonadaptive(Mesh &mesh, NonlinSystem &nls,
Solution &Cp, Solution &Ci, Solution &phip, Solution &phii) {
begin_time();
//VectorView vview("electric field [V/m]", 0, 0, 600, 600);
ScalarView Cview("Concentration [mol/m3]", 0, 0, 800, 800);
ScalarView phiview("Voltage [V]", 650, 0, 600, 600);
#ifdef CONT_OUTPUT
phiview.show(&phii);
Cview.show(&Ci);
Cview.wait_for_keypress();
#endif
Solution Csln, phisln;
for (int n = 1; n <= NSTEP; n++) {
#ifdef VERBOSE
info("\n---- Time step %d ----", n);
#endif
int it = 1;
double res_l2_norm;
do {
#ifdef VERBOSE
info("\n -------- Time step %d, Newton iter %d --------\n", n, it);
#endif
it++;
nls.assemble();
nls.solve(2, &Csln, &phisln);
res_l2_norm = nls.get_residuum_l2_norm();
#ifdef VERBOSE
info("Residuum L2 norm: %g\n", res_l2_norm);
#endif
Ci.copy(&Csln);
phii.copy(&phisln);
} while (res_l2_norm > NEWTON_TOL);
#ifdef CONT_OUTPUT
phiview.show(&phii);
Cview.show(&Ci);
#endif
phip.copy(&phii);
Cp.copy(&Ci);
}
verbose("\nTotal run time: %g sec", end_time());
Cview.show(&Ci);
phiview.show(&phii);
//MeshView mview("small.mesh", 100, 30, 800, 800);
//mview.show(&mesh);
View::wait();
}
/** Adaptive solver.*/
void solveAdaptive(Mesh &Cmesh, Mesh &phimesh, Mesh &basemesh, NonlinSystem &nls, H1Space &C, H1Space &phi,
Solution &Cp, Solution &Ci, Solution &phip, Solution &phii) {
char title[100];
//VectorView vview("electric field [V/m]", 0, 0, 600, 600);
ScalarView Cview("Concentration [mol/m3]", 0, 0, 800, 800);
ScalarView phiview("Voltage [V]", 650, 0, 600, 600);
OrderView Cordview("C order", 0, 300, 600, 600);
OrderView phiordview("Phi order", 600, 300, 600, 600);
// Different Gnuplot graphs.
// convergence graph wrt. the number of degrees of freedom
GnuplotGraph graph_err;
graph_err.set_captions("", "Timestep", "Error (Energy Norm)");
graph_err.add_row("Reference solution", "k", "-", "O");
// convergence graph wrt. CPU time
GnuplotGraph graph_dof;
graph_dof.set_captions("", "Timestep", "DOF");
graph_dof.add_row(MULTIMESH ? "multi-mesh" : "single-mesh", "k", "-", "o");
sprintf(title, "Initial iteration");
phiview.set_title(title);
Cview.set_title(title);
phiview.show(&phii);
Cview.show(&Ci);
Cordview.show(&C);
phiordview.show(&phi);
Solution Csln_coarse, phisln_coarse, Csln_fine, phisln_fine;
int at_index = 1; //for saving screenshot
for (int n = 1; n <= NSTEP; n++) {
if (n % UNREF_FREQ == 0) {
Cmesh.copy(&basemesh);
if (MULTIMESH) {
phimesh.copy(&basemesh);
}
C.set_uniform_order(P_INIT);
phi.set_uniform_order(P_INIT);
int ndofs;
ndofs = C.assign_dofs();
phi.assign_dofs(ndofs);
}
#ifdef VERBOSE
info("\n---- Time step %d ----", n);
#endif
int at = 0;
bool done = false;
double err;
do {
at++;
at_index++;
int it = 1;
double res_l2_norm;
if (n > 1 || at > 1) {
nls.set_ic(&Csln_fine, &phisln_fine, &Ci, &phii);
} else {
/* No need to set anything, already set. */
nls.set_ic(&Ci, &phii, &Ci, &phii);
}
//Loop for coarse mesh solution
do {
it++;
#ifdef VERBOSE
info("\n -------- Time step %d, Newton iter %d --------\n", n, it);
#endif
nls.assemble();
nls.solve(2, &Csln_coarse, &phisln_coarse);
res_l2_norm = nls.get_residuum_l2_norm();
#ifdef VERBOSE
info("Residuum L2 norm: %g\n", res_l2_norm);
#endif
Ci.copy(&Csln_coarse);
phii.copy(&phisln_coarse);
} while (res_l2_norm > NEWTON_TOL_COARSE);
it = 1;
// Loop for fine mesh solution
RefNonlinSystem rs(&nls);
rs.prepare();
if (n > 1 || at > 1) {
rs.set_ic(&Csln_fine, &phisln_fine, &Ci, &phii);
} else {
rs.set_ic(&Ci, &phii, &Ci, &phii);
}
do {
it++;
#ifdef VERBOSE
info("\n -------- Time step %d, Adaptivity step %d, Newton iter %d (Fine mesh) --------\n", n, at, it);
#endif
rs.assemble();
rs.solve(2, &Csln_fine, &phisln_fine);
res_l2_norm = rs.get_residuum_l2_norm();
#ifdef VERBOSE
info("Residuum L2 norm: %g\n", res_l2_norm);
#endif
Ci.copy(&Csln_fine);
phii.copy(&phisln_fine);
} while (res_l2_norm > NEWTON_TOL_REF);
// Calculate element errors and total estimate
H1OrthoHP hp(2, &C, &phi);
info("\n Calculating element errors\n");
err = hp.calc_error_2(&Csln_coarse, &phisln_coarse, &Csln_fine, &phisln_fine) * 100;
info("Error: %g", err);
if (err < ERR_STOP) {
done = true;
} else {
hp.adapt(THRESHOLD, STRATEGY, ADAPT_TYPE);
}
int ndofs;
ndofs = C.assign_dofs();
ndofs += phi.assign_dofs(ndofs);
info("NDOFS after adapting: %d", ndofs);
if (ndofs >= NDOF_STOP) {
info("NDOFs reached to the max %d", NDOF_STOP);
done = true;
}
sprintf(title, "hp-mesh (C), time level %d, adaptivity %d", n, at);
Cordview.set_title(title);
sprintf(title, "hp-mesh (phi), time level %d, adaptivity %d", n, at);
phiordview.set_title(title);
Cordview.show(&C);
phiordview.show(&phi);
#ifdef SCREENSHOT
Cordview.save_numbered_screenshot("screenshots/Cord%03d.bmp", at_index, true);
phiordview.save_numbered_screenshot("screenshots/phiord%03d.bmp", at_index, true);
#endif
} while (!done);
phip.copy(&phii);
Cp.copy(&Ci);
graph_err.add_values(0, n, err);
graph_err.save("error.gp");
graph_dof.add_values(0, n, C.get_num_dofs() + phi.get_num_dofs());
graph_dof.save("dofs.gp");
if (n == 1) {
sprintf(title, "phi after time step %d, adjust the graph and PRESS ANY KEY", n);
} else {
sprintf(title, "phi after time step %d", n);
}
phiview.set_title(title);
phiview.show(&phii);
if (n == 1) {
sprintf(title, "C after time step %d, adjust the graph and PRESS ANY KEY", n);
} else {
sprintf(title, "C after time step %d", n);
}
Cview.set_title(title);
Cview.show(&Ci);
#ifdef SCREENSHOT
Cview.save_numbered_screenshot("screenshots/C%03d.bmp", n, true);
phiview.save_numbered_screenshot("screenshots/phi%03d.bmp", n, true);
#endif
if (n == 1) {
// Wait for key press, so one can go to 3D mode
// which is way more informative in case of Nernst Planck
Cview.wait_for_keypress();
}
}
View::wait();
}
