Newer
Older
/*----------------------------------------------------------------------
SerialReax - Reax Force Field Simulator
Copyright (2010) Purdue University
Hasan Metin Aktulga, haktulga@cs.purdue.edu
Joseph Fogarty, jcfogart@mail.usf.edu
Sagar Pandit, pandit@usf.edu
Ananth Y Grama, ayg@cs.purdue.edu
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 2 of
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
See the GNU General Public License for more details:
<http://www.gnu.org/licenses/>.
----------------------------------------------------------------------*/
Kurt A. O'Hearn
committed
#include "spuremd.h"
Kurt A. O'Hearn
committed
#include "allocate.h"
Kurt A. O'Hearn
committed
#include "box.h"
Kurt A. O'Hearn
committed
#include "control.h"
#include "ffield.h"
Kurt A. O'Hearn
committed
#include "io_tools.h"
Kurt A. O'Hearn
committed
#include "geo_tools.h"
Kurt A. O'Hearn
committed
#include "reset_tools.h"
#include "tool_box.h"
Kurt A. O'Hearn
committed
#include <ctype.h>
Kurt A. O'Hearn
committed
/* Handles additional entire geometry calculations after
* perturbing atom positions during a simulation step
*/
static void Post_Evolve( reax_system * const system, control_params * const control,
simulation_data * const data, static_storage * const workspace,
Kurt A. O'Hearn
committed
reax_list ** const lists, output_controls * const out_control )
int i;
rvec diff, cross;
/* remove rotational and translational velocity of the center of mass */
Kurt A. O'Hearn
committed
if ( control->ensemble != NVE && control->remove_CoM_vel > 0
Kurt A. O'Hearn
committed
&& data->step % control->remove_CoM_vel == 0 )
Kurt A. O'Hearn
committed
Compute_Center_of_Mass( system, data );
Kurt A. O'Hearn
committed
/* remove translational */
rvec_ScaledAdd( system->atoms[i].v, -1.0, data->vcm );
Kurt A. O'Hearn
committed
/* remove rotational */
rvec_ScaledSum( diff, 1.0, system->atoms[i].x, -1.0, data->xcm );
Kurt A. O'Hearn
committed
rvec_ScaledAdd( system->atoms[i].v, -1.0, cross );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( control->ensemble == NVE )
{
Compute_Kinetic_Energy( system, data );
}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
Compute_Total_Energy( data );
Kurt A. O'Hearn
committed
if ( control->compute_pressure == TRUE && control->ensemble != sNPT
&& control->ensemble != iNPT && control->ensemble != aNPT )
{
Compute_Pressure_Isotropic( system, control, data, out_control );
}
Kurt A. O'Hearn
committed
/* Parse input files
*
* geo_file: file containing geometry info of the structure to simulate
* ffield_file: file containing force field parameters
* control_file: file containing simulation parameters
*/
static void Read_Input_Files( const char * const geo_file,
const char * const ffield_file, const char * const control_file,
reax_system * const system, control_params * const control,
simulation_data * const data, static_storage * const workspace,
Kurt A. O'Hearn
committed
output_controls * const out_control, int reset )
Kurt A. O'Hearn
committed
if ( ffield_file != NULL )
Kurt A. O'Hearn
committed
Read_Force_Field( ffield_file, system, &system->reax_param );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( reset == FALSE || control_file != NULL )
{
Set_Control_Defaults( system, control, out_control );
}
Kurt A. O'Hearn
committed
if ( control_file != NULL )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
Read_Control_File( control_file, system, control, out_control );
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( reset == FALSE || control_file != NULL )
{
Set_Control_Derived_Values( system, control );
}
Kurt A. O'Hearn
committed
if ( geo_file != NULL )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
if ( control->geo_format == CUSTOM )
{
Read_Geo( geo_file, system, control, data, workspace );
}
else if ( control->geo_format == PDB )
{
Read_PDB( geo_file, system, control, data, workspace );
}
else if ( control->geo_format == BGF )
{
Read_BGF( geo_file, system, control, data, workspace );
}
else if ( control->geo_format == ASCII_RESTART )
{
Read_ASCII_Restart( geo_file, system, control, data, workspace );
control->restart = TRUE;
}
else if ( control->geo_format == BINARY_RESTART )
{
Read_Binary_Restart( geo_file, system, control, data, workspace );
control->restart = TRUE;
}
else
{
fprintf( stderr, "[ERROR] unknown geo file format. terminating!\n" );
exit( INVALID_GEO );
}
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
#if defined(DEBUG_FOCUS)
Print_Box( &system->box, stderr );
#endif
}
Kurt A. O'Hearn
committed
static void Allocate_Top_Level_Structs( spuremd_handle ** handle )
{
int i;
/* top-level allocation */
Kurt A. O'Hearn
committed
*handle = smalloc( sizeof(spuremd_handle), __FILE__, __LINE__ );
Kurt A. O'Hearn
committed
/* second-level allocations */
Kurt A. O'Hearn
committed
(*handle)->system = smalloc( sizeof(reax_system), __FILE__, __LINE__ );
(*handle)->control = smalloc( sizeof(control_params), __FILE__, __LINE__ );
(*handle)->data = smalloc( sizeof(simulation_data), __FILE__, __LINE__ );
(*handle)->workspace = smalloc( sizeof(static_storage), __FILE__, __LINE__ );
(*handle)->lists = smalloc( sizeof(reax_list *) * LIST_N, __FILE__, __LINE__ );
Kurt A. O'Hearn
committed
for ( i = 0; i < LIST_N; ++i )
{
Kurt A. O'Hearn
committed
(*handle)->lists[i] = smalloc( sizeof(reax_list), __FILE__, __LINE__ );
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
(*handle)->out_control = smalloc( sizeof(output_controls), __FILE__, __LINE__ );
Kurt A. O'Hearn
committed
}
static void Initialize_Top_Level_Structs( spuremd_handle * handle )
{
int i;
/* top-level initializations */
handle->output_enabled = TRUE;
handle->realloc = TRUE;
handle->callback = NULL;
handle->data->sim_id = 0;
/* second-level initializations */
handle->system->prealloc_allocated = FALSE;
Kurt A. O'Hearn
committed
handle->system->allocated = FALSE;
Kurt A. O'Hearn
committed
handle->system->ffield_params_allocated = FALSE;
handle->system->g.allocated = FALSE;
Kurt A. O'Hearn
committed
handle->workspace->allocated = FALSE;
Kurt A. O'Hearn
committed
handle->workspace->H.allocated = FALSE;
handle->workspace->H_full.allocated = FALSE;
handle->workspace->H_sp.allocated = FALSE;
handle->workspace->H_p.allocated = FALSE;
handle->workspace->H_spar_patt.allocated = FALSE;
handle->workspace->H_spar_patt_full.allocated = FALSE;
handle->workspace->H_app_inv.allocated = FALSE;
handle->workspace->L.allocated = FALSE;
handle->workspace->U.allocated = FALSE;
for ( i = 0; i < LIST_N; ++i )
{
handle->lists[i]->allocated = FALSE;
}
Kurt A. O'Hearn
committed
handle->out_control->allocated = FALSE;
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
/* Allocate top-level data structures and parse input files
* for the first simulation
*
* geo_file: file containing geometry info of the structure to simulate
* ffield_file: file containing force field parameters
* control_file: file containing simulation parameters
*/
Kurt A. O'Hearn
committed
void * setup( const char * const geo_file, const char * const ffield_file,
Kurt A. O'Hearn
committed
const char * const control_file )
Kurt A. O'Hearn
committed
spuremd_handle *spmd_handle;
Kurt A. O'Hearn
committed
Allocate_Top_Level_Structs( &spmd_handle );
Initialize_Top_Level_Structs( spmd_handle );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
spmd_handle->system->N_max = 0;
Kurt A. O'Hearn
committed
spmd_handle->system->max_num_molec_charge_constraints = 0;
Kurt A. O'Hearn
committed
spmd_handle->system->num_molec_charge_constraints = 0;
spmd_handle->system->max_num_custom_charge_constraints = 0;
spmd_handle->system->num_custom_charge_constraints = 0;
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
Read_Input_Files( geo_file, ffield_file, control_file,
Kurt A. O'Hearn
committed
spmd_handle->system, spmd_handle->control,
spmd_handle->data, spmd_handle->workspace,
Kurt A. O'Hearn
committed
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
spmd_handle->out_control, FALSE );
spmd_handle->system->N_max = (int) CEIL( SAFE_ZONE * spmd_handle->system->N );
return (void *) spmd_handle;
}
/* Allocate top-level data structures and parse input files
* for the first simulation
*
* num_atoms: num. atoms in this simulation
* types: integer representation of atom element (type)
* NOTE: must match the 0-based index from section 2 in the ReaxFF parameter file
* sim_box_info: simulation box information, where the entries are
* - box length per dimension (3 entries)
* - angles per dimension (3 entries)
* pos: coordinates of atom positions (consecutively arranged), in Angstroms
* ffield_file: file containing force field parameters
* control_file: file containing simulation parameters
*/
void * setup2( int num_atoms, const int * const atom_type,
const double * const pos, const double * const sim_box_info,
const char * const ffield_file, const char * const control_file )
{
int i;
// char atom_name[9];
rvec x;
spuremd_handle *spmd_handle;
Allocate_Top_Level_Structs( &spmd_handle );
Initialize_Top_Level_Structs( spmd_handle );
spmd_handle->system->max_num_molec_charge_constraints = 0;
spmd_handle->system->num_molec_charge_constraints = 0;
spmd_handle->system->max_num_custom_charge_constraints = 0;
spmd_handle->system->num_custom_charge_constraints = 0;
Kurt A. O'Hearn
committed
/* override default */
spmd_handle->output_enabled = FALSE;
Read_Input_Files( NULL, ffield_file, control_file,
spmd_handle->system, spmd_handle->control,
spmd_handle->data, spmd_handle->workspace,
spmd_handle->out_control, FALSE );
spmd_handle->system->N = num_atoms;
/* note: assign here to avoid compiler warning
* of uninitialized usage in PreAllocate_Space */
spmd_handle->system->N_max = 0;
Kurt A. O'Hearn
committed
spmd_handle->system->max_num_molec_charge_constraints = 0;
spmd_handle->system->num_molec_charge_constraints = 0;
spmd_handle->system->max_num_custom_charge_constraints = 0;
spmd_handle->system->num_custom_charge_constraints = 0;
Kurt A. O'Hearn
committed
PreAllocate_Space( spmd_handle->system, spmd_handle->control,
spmd_handle->workspace, (int) CEIL( SAFE_ZONE * spmd_handle->system->N ) );
Setup_Box( sim_box_info[0], sim_box_info[1], sim_box_info[2],
sim_box_info[3], sim_box_info[4], sim_box_info[5],
&spmd_handle->system->box );
for ( i = 0; i < spmd_handle->system->N; ++i )
{
assert( atom_type[i] >= 0
Kurt A. O'Hearn
committed
&& atom_type[i] < spmd_handle->system->reax_param.num_atom_types );
Kurt A. O'Hearn
committed
x[0] = pos[3 * i];
x[1] = pos[3 * i + 1];
x[2] = pos[3 * i + 2];
Fit_to_Periodic_Box( &spmd_handle->system->box, x );
spmd_handle->workspace->orig_id[i] = i + 1;
spmd_handle->system->atoms[i].type = atom_type[i];
strncpy( spmd_handle->system->atoms[i].name,
spmd_handle->system->reax_param.sbp[atom_type[i]].name,
sizeof(spmd_handle->system->atoms[i].name) - 1 );
spmd_handle->system->atoms[i].name[sizeof(spmd_handle->system->atoms[i].name) - 1] = '\0';
Kurt A. O'Hearn
committed
rvec_Copy( spmd_handle->system->atoms[i].x, x );
rvec_MakeZero( spmd_handle->system->atoms[i].v );
rvec_MakeZero( spmd_handle->system->atoms[i].f );
spmd_handle->system->atoms[i].q = 0.0;
/* check for dummy atom */
if ( strncmp( spmd_handle->system->atoms[i].name, "X\0", 2 ) == 0 )
{
spmd_handle->system->atoms[i].is_dummy = TRUE;
}
else
{
spmd_handle->system->atoms[i].is_dummy = FALSE;
}
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
spmd_handle->system->N_max = (int) CEIL( SAFE_ZONE * spmd_handle->system->N );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
return (void *) spmd_handle;
Kurt A. O'Hearn
committed
/* Setup callback function to be run after each simulation step
*
* handle: pointer to wrapper struct with top-level data structures
* callback: function pointer to attach for callback
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
Kurt A. O'Hearn
committed
*/
Kurt A. O'Hearn
committed
int setup_callback( const void * const handle, const callback_function callback )
{
int ret;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL && callback != NULL )
{
spmd_handle = (spuremd_handle*) handle;
spmd_handle->callback = callback;
ret = SPUREMD_SUCCESS;
}
return ret;
}
Kurt A. O'Hearn
committed
/* Run the simulation according to the prescribed parameters
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
Kurt A. O'Hearn
committed
*/
Kurt A. O'Hearn
committed
int simulate( const void * const handle )
Kurt A. O'Hearn
committed
int steps, ret, ret_forces, retries;
Kurt A. O'Hearn
committed
evolve_function Evolve;
Kurt A. O'Hearn
committed
spuremd_handle *spmd_handle;
Kurt A. O'Hearn
committed
ret = SPUREMD_FAILURE;
Kurt A. O'Hearn
committed
if ( handle != NULL )
Kurt A. O'Hearn
committed
spmd_handle = (spuremd_handle*) handle;
Initialize( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists,
spmd_handle->out_control, &Evolve,
Kurt A. O'Hearn
committed
spmd_handle->output_enabled,
Kurt A. O'Hearn
committed
spmd_handle->realloc );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
/* compute f_0 */
Kurt A. O'Hearn
committed
Reset( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
ret_forces = Compute_Forces( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control,
spmd_handle->realloc );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( ret_forces != SUCCESS )
{
Estimate_Storages( spmd_handle->system, spmd_handle->control,
spmd_handle->workspace, spmd_handle->lists,
spmd_handle->workspace->realloc.cm,
spmd_handle->workspace->realloc.bonds
|| spmd_handle->workspace->realloc.hbonds );
Kurt A. O'Hearn
committed
Reallocate_Part2( spmd_handle->system, spmd_handle->control,
spmd_handle->data, spmd_handle->workspace,
spmd_handle->lists );
Kurt A. O'Hearn
committed
ret_forces = Compute_Forces( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control,
spmd_handle->realloc );
if ( ret_forces != SUCCESS )
{
fprintf( stderr, "[ERROR] Unrecoverable memory allocation issue (Compute_Forces). Terminating...\n" );
exit( INVALID_GEO );
}
}
Kurt A. O'Hearn
committed
Compute_Kinetic_Energy( spmd_handle->system, spmd_handle->data );
Kurt A. O'Hearn
committed
if ( spmd_handle->control->compute_pressure == TRUE && spmd_handle->control->ensemble != sNPT
&& spmd_handle->control->ensemble != iNPT && spmd_handle->control->ensemble != aNPT )
{
Compute_Pressure_Isotropic( spmd_handle->system, spmd_handle->control,
spmd_handle->data, spmd_handle->out_control );
}
Kurt A. O'Hearn
committed
Compute_Total_Energy( spmd_handle->data );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( spmd_handle->output_enabled == TRUE )
{
Kurt A. O'Hearn
committed
Output_Results( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control );
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
Check_Energy( spmd_handle->data );
Kurt A. O'Hearn
committed
if ( spmd_handle->output_enabled == TRUE )
{
if ( spmd_handle->out_control->write_steps > 0
&& spmd_handle->data->step % spmd_handle->out_control->write_steps == 0 )
{
Write_PDB( spmd_handle->system, spmd_handle->lists[BONDS], spmd_handle->data,
spmd_handle->control, spmd_handle->workspace, spmd_handle->out_control );
}
}
Kurt A. O'Hearn
committed
if ( spmd_handle->callback != NULL )
{
spmd_handle->callback( spmd_handle->system->N, spmd_handle->system->atoms,
spmd_handle->data );
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
//}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
++spmd_handle->data->step;
retries = 0;
while ( spmd_handle->data->step <= spmd_handle->control->nsteps
&& retries < MAX_RETRIES )
{
Kurt A. O'Hearn
committed
ret = SUCCESS;
Kurt A. O'Hearn
committed
if ( spmd_handle->control->T_mode != 0 )
Kurt A. O'Hearn
committed
{
Temperature_Control( spmd_handle->control, spmd_handle->data,
spmd_handle->out_control );
}
Kurt A. O'Hearn
committed
ret = Evolve( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( ret == SUCCESS )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
Post_Evolve( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control );
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
if ( ret == SUCCESS )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
if ( spmd_handle->output_enabled == TRUE )
{
Output_Results( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control );
}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
Check_Energy( spmd_handle->data );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( spmd_handle->output_enabled == TRUE )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
steps = spmd_handle->data->step - spmd_handle->data->prev_steps;
Analysis( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control );
if ( spmd_handle->out_control->restart_freq > 0
&& steps % spmd_handle->out_control->restart_freq == 0 )
{
Write_Restart( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->out_control );
}
if ( spmd_handle->out_control->write_steps > 0
&& steps % spmd_handle->out_control->write_steps == 0 )
{
Write_PDB( spmd_handle->system, spmd_handle->lists[BONDS], spmd_handle->data,
spmd_handle->control, spmd_handle->workspace, spmd_handle->out_control );
}
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
if ( spmd_handle->callback != NULL )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
spmd_handle->callback( spmd_handle->system->N, spmd_handle->system->atoms,
spmd_handle->data );
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
++spmd_handle->data->step;
retries = 0;
}
else
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
++retries;
#if defined(DEBUG_FOCUS)
fprintf( stderr, "[INFO] p%d: retrying step %d...\n", system->my_rank, data->step );
#endif
Kurt A. O'Hearn
committed
}
}
Kurt A. O'Hearn
committed
spmd_handle->data->timing.end = Get_Time( );
spmd_handle->data->timing.elapsed = Get_Timing_Info( spmd_handle->data->timing.start );
Kurt A. O'Hearn
committed
if ( spmd_handle->output_enabled == TRUE
&& spmd_handle->out_control->log_update_freq > 0 )
Kurt A. O'Hearn
committed
{
fprintf( spmd_handle->out_control->log, "total: %.2f secs\n", spmd_handle->data->timing.elapsed );
}
Kurt A. O'Hearn
committed
spmd_handle->realloc = FALSE;
Kurt A. O'Hearn
committed
ret = SPUREMD_SUCCESS;
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
return ret;
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
/* Deallocate all data structures post-simulation
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
Kurt A. O'Hearn
committed
*/
Kurt A. O'Hearn
committed
int cleanup( const void * const handle )
Kurt A. O'Hearn
committed
{
int i, ret;
Kurt A. O'Hearn
committed
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
Finalize( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control,
Kurt A. O'Hearn
committed
spmd_handle->output_enabled, FALSE );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
sfree( spmd_handle->out_control, __FILE__, __LINE__ );
for ( i = 0; i < LIST_N; ++i )
{
Kurt A. O'Hearn
committed
sfree( spmd_handle->lists[i], __FILE__, __LINE__ );
}
Kurt A. O'Hearn
committed
sfree( spmd_handle->lists, __FILE__, __LINE__ );
sfree( spmd_handle->workspace, __FILE__, __LINE__ );
sfree( spmd_handle->data, __FILE__, __LINE__ );
sfree( spmd_handle->control, __FILE__, __LINE__ );
sfree( spmd_handle->system, __FILE__, __LINE__ );
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
sfree( spmd_handle, __FILE__, __LINE__ );
Kurt A. O'Hearn
committed
ret = SPUREMD_SUCCESS;
}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
return ret;
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
/* Reset for the next simulation by parsing input files and triggering
* reallocation if more space is needed
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
* geo_file: file containing geometry info of the structure to simulate
Kurt A. O'Hearn
committed
* ffield_file: file containing force field parameters
* control_file: file containing simulation parameters
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
Kurt A. O'Hearn
committed
int reset( const void * const handle, const char * const geo_file,
Kurt A. O'Hearn
committed
const char * const ffield_file, const char * const control_file )
{
Kurt A. O'Hearn
committed
int ret;
Kurt A. O'Hearn
committed
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
/* close files used in previous simulation */
if ( spmd_handle->output_enabled == TRUE )
{
Finalize_Out_Controls( spmd_handle->system, spmd_handle->control,
spmd_handle->workspace, spmd_handle->out_control );
}
spmd_handle->realloc = FALSE;
spmd_handle->data->sim_id++;
Kurt A. O'Hearn
committed
Read_Input_Files( geo_file, ffield_file, control_file,
Kurt A. O'Hearn
committed
spmd_handle->system, spmd_handle->control,
spmd_handle->data, spmd_handle->workspace,
Kurt A. O'Hearn
committed
spmd_handle->out_control, TRUE );
Kurt A. O'Hearn
committed
if ( spmd_handle->system->N > spmd_handle->system->N_max )
{
/* deallocate everything which needs more space
* (i.e., structures whose space is a function of the number of atoms),
* except for data structures allocated while parsing input files */
Finalize( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control,
spmd_handle->output_enabled, TRUE );
spmd_handle->system->N_max = (int) CEIL( SAFE_ZONE * spmd_handle->system->N );
spmd_handle->realloc = TRUE;
}
ret = SPUREMD_SUCCESS;
}
return ret;
}
Kurt A. O'Hearn
committed
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
/* Allocate top-level data structures and parse input files
* for the first simulation
*
* handle: pointer to wrapper struct with top-level data structures
* num_atoms: num. atoms in this simulation
* types: integer representation of atom element (type)
* NOTE: must match the 0-based index from section 2 in the ReaxFF parameter file
* sim_box_info: simulation box information, where the entries are
* - box length per dimension (3 entries)
* - angles per dimension (3 entries)
* pos: coordinates of atom positions (consecutively arranged), in Angstroms
* ffield_file: file containing force field parameters
* control_file: file containing simulation parameters
*/
int reset2( const void * const handle, int num_atoms,
const int * const atom_type, const double * const pos,
const double * const sim_box_info, const char * const ffield_file,
const char * const control_file )
{
int i, ret;
rvec x;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
/* close files used in previous simulation */
if ( spmd_handle->output_enabled == TRUE )
{
Finalize_Out_Controls( spmd_handle->system, spmd_handle->control,
spmd_handle->workspace, spmd_handle->out_control );
}
spmd_handle->realloc = FALSE;
spmd_handle->data->sim_id++;
Read_Input_Files( NULL, ffield_file, control_file,
spmd_handle->system, spmd_handle->control,
spmd_handle->data, spmd_handle->workspace,
spmd_handle->out_control, TRUE );
spmd_handle->system->N = num_atoms;
if ( spmd_handle->system->prealloc_allocated == FALSE
|| spmd_handle->system->N > spmd_handle->system->N_max )
{
PreAllocate_Space( spmd_handle->system, spmd_handle->control,
spmd_handle->workspace, (int) CEIL( SAFE_ZONE * spmd_handle->system->N ) );
}
Setup_Box( sim_box_info[0], sim_box_info[1], sim_box_info[2],
sim_box_info[3], sim_box_info[4], sim_box_info[5],
&spmd_handle->system->box );
for ( i = 0; i < spmd_handle->system->N; ++i )
{
assert( atom_type[i] >= 0
Kurt A. O'Hearn
committed
&& atom_type[i] < spmd_handle->system->reax_param.num_atom_types );
Kurt A. O'Hearn
committed
x[0] = pos[3 * i];
x[1] = pos[3 * i + 1];
x[2] = pos[3 * i + 2];
Fit_to_Periodic_Box( &spmd_handle->system->box, x );
spmd_handle->workspace->orig_id[i] = i + 1;
spmd_handle->system->atoms[i].type = atom_type[i];
strncpy( spmd_handle->system->atoms[i].name,
spmd_handle->system->reax_param.sbp[atom_type[i]].name,
sizeof(spmd_handle->system->atoms[i].name) - 1 );
spmd_handle->system->atoms[i].name[sizeof(spmd_handle->system->atoms[i].name) - 1] = '\0';
Kurt A. O'Hearn
committed
rvec_Copy( spmd_handle->system->atoms[i].x, x );
rvec_MakeZero( spmd_handle->system->atoms[i].v );
rvec_MakeZero( spmd_handle->system->atoms[i].f );
spmd_handle->system->atoms[i].q = 0.0;
/* check for dummy atom */
if ( strncmp( spmd_handle->system->atoms[i].name, "X\0", 2 ) == 0 )
{
spmd_handle->system->atoms[i].is_dummy = TRUE;
}
else
{
spmd_handle->system->atoms[i].is_dummy = FALSE;
}
Kurt A. O'Hearn
committed
}
if ( spmd_handle->system->N > spmd_handle->system->N_max )
{
/* deallocate everything which needs more space
* (i.e., structures whose space is a function of the number of atoms),
* except for data structures allocated while parsing input files */
Finalize( spmd_handle->system, spmd_handle->control, spmd_handle->data,
spmd_handle->workspace, spmd_handle->lists, spmd_handle->out_control,
spmd_handle->output_enabled, TRUE );
spmd_handle->system->N_max = (int) CEIL( SAFE_ZONE * spmd_handle->system->N );
spmd_handle->realloc = TRUE;
}
ret = SPUREMD_SUCCESS;
}
return ret;
}
Kurt A. O'Hearn
committed
/* Getter for atom positions
Kurt A. O'Hearn
committed
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
* pos: coordinates of atom positions, in Angstroms (allocated by caller)
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
Kurt A. O'Hearn
committed
*/
Kurt A. O'Hearn
committed
int get_atom_positions( const void * const handle, double * const pos )
Kurt A. O'Hearn
committed
{
Kurt A. O'Hearn
committed
int i, ret;
Kurt A. O'Hearn
committed
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
Kurt A. O'Hearn
committed
for ( i = 0; i < spmd_handle->system->N; ++i )
{
Kurt A. O'Hearn
committed
pos[3 * i] = spmd_handle->system->atoms[i].x[0];
pos[3 * i + 1] = spmd_handle->system->atoms[i].x[1];
pos[3 * i + 2] = spmd_handle->system->atoms[i].x[2];
}
ret = SPUREMD_SUCCESS;
Kurt A. O'Hearn
committed
}
return ret;
}
Kurt A. O'Hearn
committed
/* Getter for atom velocities
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
* vel: coordinates of atom velocities, in Angstroms / ps (allocated by caller)
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
Kurt A. O'Hearn
committed
int get_atom_velocities( const void * const handle, double * const vel )
Kurt A. O'Hearn
committed
{
int i, ret;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
for ( i = 0; i < spmd_handle->system->N; ++i )
{
Kurt A. O'Hearn
committed
vel[3 * i] = spmd_handle->system->atoms[i].v[0];
vel[3 * i + 1] = spmd_handle->system->atoms[i].v[1];
vel[3 * i + 2] = spmd_handle->system->atoms[i].v[2];
Kurt A. O'Hearn
committed
}
ret = SPUREMD_SUCCESS;
}
return ret;
}
/* Getter for atom forces
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
* f: coordinates of atom forces, in Angstroms * Daltons / ps^2 (allocated by caller)
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
Kurt A. O'Hearn
committed
int get_atom_forces( const void * const handle, double * const f )
Kurt A. O'Hearn
committed
{
int i, ret;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
for ( i = 0; i < spmd_handle->system->N; ++i )
{
Kurt A. O'Hearn
committed
f[3 * i] = spmd_handle->system->atoms[i].f[0];
f[3 * i + 1] = spmd_handle->system->atoms[i].f[1];
f[3 * i + 2] = spmd_handle->system->atoms[i].f[2];
Kurt A. O'Hearn
committed
}
ret = SPUREMD_SUCCESS;
}
return ret;
}
/* Getter for atom charges
*
* handle: pointer to wrapper struct with top-level data structures
Kurt A. O'Hearn
committed
* q: atom charges, in Coulombs (allocated by caller)
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
int get_atom_charges( const void * const handle, double * const q )
{
int i, ret;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
for ( i = 0; i < spmd_handle->system->N; ++i )
{
q[i] = spmd_handle->system->atoms[i].q;
}
ret = SPUREMD_SUCCESS;
}
return ret;
Kurt A. O'Hearn
committed
}
Kurt A. O'Hearn
committed
Kurt A. O'Hearn
committed
/* Getter for system energies
*
* handle: pointer to wrapper struct with top-level data structures
* e_pot: system potential energy, in kcal / mol (reference from caller)
* e_kin: system kinetic energy, in kcal / mol (reference from caller)
* e_tot: system total energy, in kcal / mol (reference from caller)
* t_scalar: temperature scalar, in K (reference from caller)
* vol: volume of the simulation box, in Angstroms^3 (reference from caller)
* pres: average pressure, in K (reference from caller)
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
int get_system_info( const void * const handle, double * const e_pot,
double * const e_kin, double * const e_tot, double * const temp,
double * const vol, double * const pres )
{
int ret;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
Kurt A. O'Hearn
committed
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
if ( e_pot != NULL )
{
*e_pot = spmd_handle->data->E_Pot;
}
if ( e_kin != NULL )
{
*e_kin = spmd_handle->data->E_Kin;
}
if ( e_tot != NULL )
{
*e_tot = spmd_handle->data->E_Tot;
}
if ( temp != NULL )
{
*temp = spmd_handle->data->therm.T;
}
if ( vol != NULL )
{
*vol = spmd_handle->system->box.volume;
}
if ( pres != NULL )
{
*pres = (spmd_handle->control->P[0] + spmd_handle->control->P[1]
+ spmd_handle->control->P[2]) / 3.0;
}
Kurt A. O'Hearn
committed
ret = SPUREMD_SUCCESS;
}
return ret;
}
Kurt A. O'Hearn
committed
/* Getter for total energy
*
* handle: pointer to wrapper struct with top-level data structures
* e_tot: system total energy, in kcal / mol (reference from caller)
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
int get_total_energy( const void * const handle, double * const e_tot )
{
int ret;
ret = get_system_info( handle, e_tot, NULL, NULL, NULL, NULL, NULL );
if ( ret == SUCCESS )
{
ret = SPUREMD_SUCCESS;
}
return ret;
}
Kurt A. O'Hearn
committed
/* Setter for writing output to files
*
* handle: pointer to wrapper struct with top-level data structures
* enabled: TRUE enables writing output to files, FALSE otherwise
Kurt A. O'Hearn
committed
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
Kurt A. O'Hearn
committed
*/
Kurt A. O'Hearn
committed
int set_output_enabled( const void * const handle, const int enabled )
{
int ret;
spuremd_handle *spmd_handle;
ret = SPUREMD_FAILURE;
if ( handle != NULL )
{
spmd_handle = (spuremd_handle*) handle;
spmd_handle->output_enabled = enabled;
ret = SPUREMD_SUCCESS;
}
return ret;
}
Kurt A. O'Hearn
committed
/* Setter for simulation parameter values as defined in the input control file
*
* handle: pointer to wrapper struct with top-level data structures
* control_keyword: keyword from the control file to set the value for
* control_value: value to set
*
* returns: SPUREMD_SUCCESS upon success, SPUREMD_FAILURE otherwise
*/
int set_control_parameter( const void * const handle, const char * const keyword,
const char ** const values )
{