void force_calc()
{
#ifdef LB_GPU
if (lattice_switch & LATTICE_LB_GPU) lb_calc_particle_lattice_ia_gpu();
#endif
init_forces();
switch (cell_structure.type) {
case CELL_STRUCTURE_LAYERED:
layered_calculate_ia();
break;
case CELL_STRUCTURE_DOMDEC:
if(dd.use_vList) {
if (rebuild_verletlist)
build_verlet_lists_and_calc_verlet_ia();
else
calculate_verlet_ia();
}
else
calc_link_cell();
break;
case CELL_STRUCTURE_NSQUARE:
nsq_calculate_ia();
}
calc_long_range_forces();
#ifdef LB
if (lattice_switch & LATTICE_LB) calc_particle_lattice_ia() ;
#endif
#ifdef COMFORCE
calc_comforce();
#endif
#ifdef METADYNAMICS
/* Metadynamics main function */
meta_perform();
#endif
#ifdef LB_GPU
if (lattice_switch & LATTICE_LB_GPU) lb_send_forces_gpu();
#endif
/* this must be the last force to be calculated (Mehmet)*/
#ifdef COMFIXED
calc_comfixed();
#endif
}
void force_calc()
{
// Communication step: distribute ghost positions
cells_update_ghosts();
// VIRTUAL_SITES pos (and vel for DPD) update for security reason !!!
#ifdef VIRTUAL_SITES
update_mol_vel_pos();
ghost_communicator(&cell_structure.update_ghost_pos_comm);
#endif
#if defined(VIRTUAL_SITES_RELATIVE) && defined(LB)
// This is on a workaround stage:
// When using virtual sites relative and LB at the same time, it is necessary
// to reassemble the cell lists after all position updates, also of virtual
// particles.
if ((lattice_switch & LATTICE_LB) && cell_structure.type == CELL_STRUCTURE_DOMDEC && (!dd.use_vList) )
cells_update_ghosts();
#endif
espressoSystemInterface.update();
#ifdef COLLISION_DETECTION
prepare_collision_queue();
#endif
#ifdef LB_GPU
#ifdef SHANCHEN
if (lattice_switch & LATTICE_LB_GPU && this_node == 0) lattice_boltzmann_calc_shanchen_gpu();
#endif // SHANCHEN
// transfer_momentum_gpu check makes sure the LB fluid doesn't get updated on integrate 0
// this_node==0 makes sure it is the master node where the gpu exists
if (lattice_switch & LATTICE_LB_GPU && transfer_momentum_gpu && (this_node == 0) ) lb_calc_particle_lattice_ia_gpu();
#endif // LB_GPU
#ifdef ELECTROSTATICS
if (iccp3m_initialized && iccp3m_cfg.set_flag)
iccp3m_iteration();
#endif
init_forces();
for (ActorList::iterator actor = forceActors.begin();
actor != forceActors.end(); ++actor)
{
(*actor)->computeForces(espressoSystemInterface);
#ifdef ROTATION
(*actor)->computeTorques(espressoSystemInterface);
#endif
}
calc_long_range_forces();
switch (cell_structure.type) {
case CELL_STRUCTURE_LAYERED:
layered_calculate_ia();
break;
case CELL_STRUCTURE_DOMDEC:
if(dd.use_vList) {
if (rebuild_verletlist)
build_verlet_lists_and_calc_verlet_ia();
else
calculate_verlet_ia();
}
else
calc_link_cell();
break;
case CELL_STRUCTURE_NSQUARE:
nsq_calculate_ia();
}
#ifdef OIF_GLOBAL_FORCES
double area_volume[2]; //There are two global quantities that need to be evaluated: object's surface and object's volume. One can add another quantity.
area_volume[0] = 0.0;
area_volume[1] = 0.0;
for (int i=0;i< MAX_OBJECTS_IN_FLUID;i++){
calc_oif_global(area_volume,i);
if (fabs(area_volume[0])<1e-100 && fabs(area_volume[1])<1e-100) break;
add_oif_global_forces(area_volume,i);
}
#endif
#ifdef IMMERSED_BOUNDARY
// Must be done here. Forces need to be ghost-communicated
IBM_VolumeConservation();
#endif
#ifdef LB
if (lattice_switch & LATTICE_LB) calc_particle_lattice_ia() ;
#endif
#ifdef COMFORCE
calc_comforce();
#endif
#ifdef METADYNAMICS
/* Metadynamics main function */
meta_perform();
#endif
#ifdef CUDA
copy_forces_from_GPU();
#endif
// VIRTUAL_SITES distribute forces
#ifdef VIRTUAL_SITES
ghost_communicator(&cell_structure.collect_ghost_force_comm);
init_forces_ghosts();
distribute_mol_force();
#endif
// Communication Step: ghost forces
ghost_communicator(&cell_structure.collect_ghost_force_comm);
// apply trap forces to trapped molecules
#ifdef MOLFORCES
calc_and_apply_mol_constraints();
#endif
// should be pretty late, since it needs to zero out the total force
#ifdef COMFIXED
calc_comfixed();
#endif
// mark that forces are now up-to-date
recalc_forces = 0;
}
void integrate_sd(int n_steps)
{
/* Prepare the Integrator */
on_integration_start();
/* if any method vetoes (P3M not initialized), immediately bail out */
if (check_runtime_errors())
return;
INTEG_TRACE(fprintf(stderr,"%d: integrate_vv: integrating %d steps (recalc_forces=%d)\n",
this_node, n_steps, recalc_forces));
/* Integration Step:
Calculate forces f(t) as function of positions p(t) ( and velocities v(t) ) */
//if (recalc_forces) {
//thermo_heat_up();
ghost_communicator(&cell_structure.collect_ghost_force_comm);
#ifdef ROTATION
convert_initial_torques();
#endif
//thermo_cool_down();
/* Communication Step: ghost forces */
/*apply trap forces to trapped molecules*/
#ifdef MOLFORCES
// prob. works only with harmonic bounds
calc_and_apply_mol_constraints();
#endif
/* should be pretty late, since it needs to zero out the total force */
#ifdef COMFIXED
calc_comfixed();
#endif
//rescale_forces();
#ifdef COLLISION_DETECTION
//should not be neccessery, as integrator avoids collision
handle_collisions();
#endif
// end of force calculation
#ifdef GHMC
if(thermo_switch & THERMO_GHMC)
ghmc_init();
#endif
if (check_runtime_errors())
return;
n_verlet_updates = 0;
/* Integration loop */
for(int step=0;step<n_steps;step++) {
INTEG_TRACE(fprintf(stderr,"%d: STEP %d\n",this_node,step));
//sd_set_particles_apart();
#ifdef BOND_CONSTRAINT
save_old_pos();
#endif
#ifdef GHMC
if(thermo_switch & THERMO_GHMC) {
if (step % ghmc_nmd == 0)
ghmc_momentum_update();
}
#endif
if(thermo_switch & ~(THERMO_SD|THERMO_BD) ){
static bool warned_thermo_sd_other=false;
if (!warned_thermo_sd_other){
fprintf (stderr, "Warning, using another thermo than the one provided by StokesDynamics breaks (most likely) StokesDynamics.\n");
warned_thermo_sd_other=true;
}
}
if (thermo_switch & THERMO_SD && thermo_switch &THERMO_BD) {
fprintf (stderr, "Warning: cannot use BD and SD. Disabeling BD!\n");
thermo_switch &= ~THERMO_BD;
}
/* Integration Step: Step 3 of Velocity Verlet scheme:
Calculate f(t) as function of positions p(t) ( and ``velocities'' v(t) ) */
#ifdef LB
transfer_momentum = 1;
#endif
#ifdef LB_GPU
transfer_momentum_gpu = 1;
#endif
force_calc();
#ifdef CATALYTIC_REACTIONS
integrate_reaction();
#endif
if (check_runtime_errors())
break;
#ifdef LB
if (lattice_switch & LATTICE_LB)
lattice_boltzmann_update();
if (check_runtime_errors())
break;
#endif
#ifdef LB_GPU
if(this_node == 0){
#ifdef ELECTROKINETICS
if (ek_initialized) {
ek_integrate();
}
else {
#endif
if (lattice_switch & LATTICE_LB_GPU)
lattice_boltzmann_update_gpu();
#ifdef ELECTROKINETICS
}
#endif
}
#endif //LB_GPU
#ifdef ELECTROSTATICS
if(coulomb.method == COULOMB_MAGGS) {
maggs_propagate_B_field(0.5*time_step);
}
#endif
#ifdef NPT
if((this_node==0) && (integ_switch == INTEG_METHOD_NPT_ISO))
nptiso.p_inst_av += nptiso.p_inst;
#endif
#ifdef GHMC
if(thermo_switch & THERMO_GHMC) {
if (step % ghmc_nmd == ghmc_nmd-1)
ghmc_mc();
}
#endif
/** Integration Steps: Update the Positions
\[ p_i(t + dt) = p_i(t) + dt * \mu_{ij} * f_j(t) + dt * \mu_{ij} * f^B_j \]
*/
propagate_pos_sd(); // we dont have velocities
#ifdef BOND_CONSTRAINT
static bool bond_constraint_with_sd_warned=false;
if (!bond_constraint_with_sd_warned){ // warn only once
fprintf (stderr, "Warning, using BOND_CONSTRAINT with StokesDynamics might not work as expected!.\n");
bond_constraint_with_sd_warned=true;
}
/**Correct those particle positions that participate in a rigid/constrained bond */
cells_update_ghosts();
correct_pos_shake();
#endif
#ifdef ELECTROSTATICS
if(coulomb.method == COULOMB_MAGGS) {
maggs_propagate_B_field(0.5*time_step);
}
#endif
#ifdef NPT
if (check_runtime_errors())
break;
#endif
/* Propagate time: t = t+dt */
sim_time += time_step;
}
/* verlet list statistics */
if(n_verlet_updates>0) verlet_reuse = n_steps/(double) n_verlet_updates;
else verlet_reuse = 0;
#ifdef NPT
if(integ_switch == INTEG_METHOD_NPT_ISO) {
nptiso.invalidate_p_vel = 0;
MPI_Bcast(&nptiso.p_inst, 1, MPI_DOUBLE, 0, comm_cart);
MPI_Bcast(&nptiso.p_diff, 1, MPI_DOUBLE, 0, comm_cart);
MPI_Bcast(&nptiso.volume, 1, MPI_DOUBLE, 0, comm_cart);
if(this_node==0) nptiso.p_inst_av /= 1.0*n_steps;
MPI_Bcast(&nptiso.p_inst_av, 1, MPI_DOUBLE, 0, comm_cart);
}
#endif
#ifdef GHMC
if(thermo_switch & THERMO_GHMC)
ghmc_close();
#endif
}
