int newton ( int mode, int verbose, PolySys& p, PolySolSet& s )
{
double teval,tmgs;
bool success;
CT* sol = s.get_sol(0);
CT alpha;
CT *x_gpu;
CPUInstHom cpu_inst_hom;
Workspace workspace_cpu;
Parameter path_parameter(N_PREDICTOR, MAX_STEP, MAX_IT, MAX_DELTA_T, \
MAX_DELTA_T_END, MIN_DELTA_T, ERR_MAX_RES, ERR_MAX_DELTA_X, \
ERR_MAX_FIRST_DELTA_X, ERR_MIN_ROUND_OFF, MAX_IT_REFINE, \
ERR_MIN_ROUND_OFF_REFINE, STEP_INCREASE, STEP_DECREASE);
if(verbose > 0)
{
cout << "The first solution on input :" << endl;
for(int k=0; k<p.dim; k++)
{
cout << k << " :";
cout << setw(72) << scientific << setprecision(64) << sol[k];
}
}
alpha = CT(1,0);
cpu_inst_hom.init(p,p,p.dim,p.n_eq,1,alpha,verbose);
cpu_inst_hom.init_workspace(workspace_cpu);
if(mode == 0 || mode == 1)
{
workspace_cpu.init_x_t_idx();
workspace_cpu.update_x_t_value(sol,alpha);
success = CPU_Newton
(workspace_cpu,cpu_inst_hom,path_parameter,teval,tmgs);
if(verbose > 0)
{
cout.precision(64);
cout << "The first solution after CPU_Newton :" << endl;
for(int k=0; k<p.dim; k++)
cout << k << " :" << setw(72) << workspace_cpu.x[k];
}
}
if(mode == 0 || mode == 2)
{
alpha = CT(1,0);
success = GPU_Newton(cpu_inst_hom,path_parameter,sol,alpha,x_gpu,
1,verbose);
if(verbose > 0)
{
cout << "The first solution after GPU_Newton :" << endl;
for(int k=0; k<p.dim; k++)
cout << k << " :" << setw(72) << x_gpu[k];
}
}
if(mode == 1)
s.change_sol(0,workspace_cpu.x);
else
s.change_sol(0,x_gpu);
return 0;
}
void ada_write_sols ( PolySolSet& sols )
{
int fail = solcon_clear_dobldobl_solutions();
if(fail != 0)
std::cout << "failed to clear the solutions" << std::endl;
int dim = sols.dim;
int nbsols = sols.n_sol;
for(int sol_idx=0; sol_idx<nbsols; sol_idx++)
{
CT* sol = sols.get_sol(sol_idx);
double csol[4*dim+10];
csol[0] = 0.0; csol[1] = 0.0;
csol[2] = 0.0; csol[3] = 0.0;
int idx = 4;
for(int k=0; k<dim; k++)
{
csol[idx++] = sol[k].real.x[0];
csol[idx++] = sol[k].real.x[1];
csol[idx++] = sol[k].imag.x[0];
csol[idx++] = sol[k].imag.x[1];
}
csol[4*dim+4] = 0.0; csol[4*dim+5] = 0.0;
csol[4*dim+6] = 0.0; csol[4*dim+7] = 0.0;
csol[4*dim+8] = 0.0; csol[4*dim+9] = 0.0;
fail = solcon_append_dobldobl_solution(dim,1,csol);
if(fail != 0)
std::cout << "failed to append the solution" << std::endl;
}
}
int manytrack
( int mode, int verbose, double regamma, double imgamma,
PolySys& p, PolySys& q, PolySolSet& s )
{
double tpred,teval,tmgs;
bool success;
CT* sol = s.get_sol(0);
CT alpha,t;
CT *x_gpu;
CPUInstHom cpu_inst_hom;
Workspace workspace_cpu;
Parameter path_parameter(N_PREDICTOR, MAX_STEP, MAX_IT, MAX_DELTA_T, \
MAX_DELTA_T_END, MIN_DELTA_T, ERR_MAX_RES, ERR_MAX_DELTA_X, \
ERR_MAX_FIRST_DELTA_X, ERR_MIN_ROUND_OFF, MAX_IT_REFINE, \
ERR_MIN_ROUND_OFF_REFINE, STEP_INCREASE, STEP_DECREASE);
if(verbose > 0)
{
cout << "The first solution on input :" << endl;
for(int k=0; k<p.dim; k++)
{
cout << k << " :";
cout << setw(40) << scientific << setprecision(32) << sol[k];
}
}
int fail,n_path;
fail = solcon_number_of_dobldobl_solutions(&n_path);
alpha = CT(regamma,imgamma);
cpu_inst_hom.init(p,q,p.dim,p.n_eq,1,alpha,verbose);
cpu_inst_hom.init_workspace(workspace_cpu);
if(mode == 0 || mode == 1)
{
for(int path_idx=0; path_idx<n_path; path_idx++)
{
if(verbose > 0) cout << "tracking path " << path_idx << endl;
CT* sol0 = s.get_sol(path_idx);
CT* sol_new = NULL;
t = CT(0.0,0);
workspace_cpu.init_x_t_idx();
workspace_cpu.update_x_t_value(sol0,t);
workspace_cpu.update_x_t_idx();
workspace_cpu.path_idx = path_idx;
tpred = 0.0; teval = 0.0; tmgs = 0.0;
if(verbose > 0) cout << "calling path_tracker ..." << endl;
success = path_tracker(workspace_cpu,cpu_inst_hom,path_parameter,
tpred,teval,tmgs,0,verbose);
if(verbose > 0) cout << "done with call to path_tracker." << endl;
if(mode == 1) s.change_sol(path_idx,workspace_cpu.x_last);
if(verbose > 0)
{
cout.precision(32);
cout << "The solution " << path_idx << endl;
for(int k=0; k<p.dim; k++)
cout << k << " :" << setw(40) << workspace_cpu.x_last[k];
}
}
}
if(mode == 0 || mode == 2)
{
CT* sol0 = new CT[n_path*cpu_inst_hom.dim];
CT* sol_tmp_mult = sol0;
CT* tt = new CT[n_path];
CT** x_gpu = NULL;
for(int path_idx=0; path_idx<n_path; path_idx++)
{
CT* sol_tmp = s.get_sol(path_idx);
for(int sol_idx=0; sol_idx<cpu_inst_hom.dim; sol_idx++)
{
*sol_tmp_mult++ = sol_tmp[sol_idx];
}
tt[path_idx] = CT(0.0,0.0);
}
success = GPU_Path_mult
(cpu_inst_hom,path_parameter,sol0,tt,x_gpu,n_path,verbose);
for(int path_idx=0; path_idx<n_path; path_idx++)
{
s.change_sol(path_idx,x_gpu[path_idx]);
}
if(verbose > 0)
{
cout << "The first solution after GPU path tracker :" << endl;
cout.precision(32);
for(int k=0; k<p.dim; k++)
cout << k << " :" << setw(40) << x_gpu[0][k];
}
delete[] tt;
}
return 0;
}