pt scaling tests, though pressure estimation still a mess!
[physik/posic.git] / moldyn.h
1 /*
2  * moldyn.h - molecular dynamics library header file
3  *
4  * author: Frank Zirkelbach <frank.zirkelbach@physik.uni-augsburg.de>
5  *
6  */
7
8 #ifndef MOLDYN_H
9 #define MOLDYN_H
10
11 #include "math/math.h"
12 #include "random/random.h"
13 #include "list/list.h"
14
15 /*
16  * 
17  * datatypes
18  *
19  */
20
21 /* general */
22 typedef unsigned char u8;
23
24 /* virial */
25 typedef struct s_virial {
26         double xx;      /*                      | xx     xy     xz |    */
27         double yy;      /*      V       =       | yx     yy     yz |    */
28         double zz;      /*                      | zx     zy     zz |    */
29         double xy;      /*                                              */
30         double xz;      /*      with:   xy=yx, xz=zx, yz=zy             */
31         double yz;      /*                                              */
32 } t_virial;
33
34 /* the atom of the md simulation */
35 typedef struct s_atom {
36         t_3dvec r;              /* position */
37         t_3dvec v;              /* velocity */
38         t_3dvec f;              /* force */
39         t_virial virial;        /* virial */
40         double e;               /* site energy */
41         int element;            /* number of element in pse */
42         double mass;            /* atom mass */
43         u8 brand;               /* brand id */
44         int tag;                /* atom unique id (number of atom) */
45         u8 attr;                /* attributes */
46 } t_atom;
47
48 #define ATOM_ATTR_FP    0x01    /* fixed position (bulk material) */
49 #define ATOM_ATTR_HB    0x02    /* coupled to heat bath (velocity scaling) */
50
51 #define ATOM_ATTR_1BP           0x10    /* single paricle potential */
52 #define ATOM_ATTR_2BP           0x20    /* pair potential */
53 #define ATOM_ATTR_3BP           0x40    /* 3 body potential */ 
54
55 /* cell lists */
56 typedef struct s_linkcell {
57         int nx,ny,nz;           /* amount of cells in x, y and z direction */
58         int cells;              /* total amount of cells */
59         double len;             /* prefered cell edge length */
60         double x,y,z;           /* the actual cell lengthes */
61         t_list *subcell;        /* pointer to the cell lists */
62         int dnlc;               /* direct neighbour lists counter */
63 } t_linkcell;
64
65 #include "visual/visual.h"
66
67 /* moldyn schedule structure */
68 typedef struct s_moldyn_schedule {
69         int count;
70         int total_sched;
71         int *runs;
72         double *tau;
73         int (*hook)(void *moldyn,void *hook_params);
74         void *hook_params;
75 } t_moldyn_schedule;
76
77 /* moldyn main structure */
78 typedef struct s_moldyn {
79         int count;              /* total amount of atoms */
80         t_atom *atom;           /* pointer to the atoms */
81
82         t_3dvec dim;            /* dimensions of the simulation volume */
83         double volume;          /* volume of sim cell (dim.x*dim.y*dim.z) */
84
85         /* potential force function and parameter pointers */
86         int (*func1b)(struct s_moldyn *moldyn,t_atom *ai);
87         void *pot1b_params;
88         int (*func2b)(struct s_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc);
89         int (*func2b_post)(struct s_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc);
90         void *pot2b_params;
91         int (*func3b)(struct s_moldyn *moldyn,t_atom *ai,t_atom *aj,t_atom *ak,
92                       u8 bck);
93         void *pot3b_params;
94         //int (*potential_force_function)(struct s_moldyn *moldyn);
95
96         double cutoff;          /* cutoff radius */
97         double cutoff_square;   /* square of the cutoff radius */
98         double nnd;             /* nearest neighbour distance (optional) */
99
100         t_linkcell lc;          /* linked cell list interface */
101
102         double t_ref;           /* reference temperature */
103         double t;               /* actual temperature */
104
105         double p_ref;           /* reference pressure */
106         double p;               /* actual pressure (computed by virial) */
107         t_3dvec tp;             /* thermodynamic pressure dU/dV */
108         double dv;              /* dV for thermodynamic pressure calc */
109
110         /* pressure and temperature control (velocity/volume scaling) */
111         /* (t_tc in units of tau, p_tc in units of tau * isoth. compressib.) */
112         unsigned char pt_scale; /* type of p and t scaling */
113         double t_tc;            /* t berendsen control time constant */
114         double p_tc;            /* p berendsen control time constant */
115
116         /* simulation schedule */
117         t_moldyn_schedule schedule;
118         int current;            /* current position in schedule */
119
120         /* integration function pointer */
121         int (*integrate)(struct s_moldyn *moldyn);
122         int time_steps;         /* amount of iterations */
123         double tau;             /* delta t */
124         double time;            /* absolute time */
125         double tau_square;      /* delta t squared */
126         double elapsed;         /* total elapsed time */
127
128         double energy;          /* potential energy */
129         double ekin;            /* kinetic energy */
130
131         char vlsdir[128];       /* visualization/log/save directory */
132         t_visual vis;           /* visualization interface structure */
133         u8 vlsprop;             /* log/vis/save properties */
134         unsigned int ewrite;    /* how often to log energy */
135         int efd;                /* fd for energy log */
136         unsigned int mwrite;    /* how often to log momentum */
137         int mfd;                /* fd for momentum log */
138         unsigned int vwrite;    /* how often to visualize atom information */
139         unsigned int swrite;    /* how often to create a save file */
140         int rfd;                /* report file descriptor */
141         char rtitle[64];        /* report title */
142         char rauthor[64];       /* report author */
143         int pfd;                /* gnuplot script file descriptor */
144
145         u8 status;              /* general moldyn properties */
146
147         t_random random;        /* random interface */
148
149         double debug;           /* debugging stuff, ignore */
150 } t_moldyn;
151
152 #define MOLDYN_STAT_PBX                 0x01    /* periodic boudaries in x */
153 #define MOLDYN_STAT_PBY                 0x02    /* y */
154 #define MOLDYN_STAT_PBZ                 0x04    /* and z direction */
155
156 #define MOLDYN_PSCALE                   0x08    /* size controlled by piston */
157
158 #define MOLDYN_1BP                      0x10    /* care about single */
159 #define MOLDYN_2BP                      0x20    /* 2 body */
160 #define MOLDYN_3BP                      0x40    /* and 3 body particle pots */
161
162 #define T_SCALE_BERENDSEN               0x01    /* berendsen t control */
163 #define T_SCALE_DIRECT                  0x02    /* direct t control */
164 #define P_SCALE_BERENDSEN               0x04    /* berendsen p control */
165 #define P_SCALE_DIRECT                  0x08    /* direct p control */
166
167 /*
168  *
169  * potential parameter structures
170  *
171  */
172
173 /*
174  * harmonic oscillator potential parameter structure
175  */
176
177 typedef struct s_ho_params {
178         double spring_constant;
179         double equilibrium_distance;
180 } t_ho_params;
181
182 /*
183  * lennard jones potential parameter structure
184  */
185
186 typedef struct s_lj_params {
187         double sigma6;
188         double sigma12;
189         double epsilon4;
190         double uc;
191 } t_lj_params;
192
193 /*
194  * tersoff 
195  */
196
197 /* tersoff exchange structure to exchange 2bp and 3bp calculated values */
198 typedef struct s_tersoff_exchange {
199         double f_c,df_c;
200         double f_a,df_a;
201
202         t_3dvec dist_ij;
203         double d_ij2;
204         double d_ij;
205
206         double chi;
207
208         double *beta_i;
209         double *beta_j;
210         double *n_i;
211         double *n_j;
212         double *c_i;
213         double *c_j;
214         double *d_i;
215         double *d_j;
216         double *h_i;
217         double *h_j;
218
219         double ci2;
220         double cj2;
221         double di2;
222         double dj2;
223         double ci2di2;
224         double cj2dj2;
225         double betaini;
226         double betajnj;
227
228         u8 run3bp;
229         u8 run2bp_post;
230         u8 d_ij_between_rs;
231
232         double zeta_ij;
233         double zeta_ji;
234         t_3dvec dzeta_ij;
235         t_3dvec dzeta_ji;
236 } t_tersoff_exchange;
237
238 /* tersoff mult (2!) potential parameters */
239 typedef struct s_tersoff_mult_params {
240         double S[2];            /* tersoff cutoff radii */
241         double S2[2];           /* tersoff cutoff radii squared */
242         double R[2];            /* tersoff cutoff radii */
243         double Smixed;          /* mixed S radius */
244         double S2mixed;         /* mixed S radius squared */
245         double Rmixed;          /* mixed R radius */
246         double A[2];            /* factor of tersoff attractive part */
247         double B[2];            /* factor of tersoff repulsive part */
248         double Amixed;          /* mixed A factor */
249         double Bmixed;          /* mixed B factor */
250         double lambda[2];       /* tersoff lambda */
251         double lambda_m;        /* mixed lambda */
252         double mu[2];           /* tersoff mu */
253         double mu_m;            /* mixed mu */
254
255         double chi;
256
257         double beta[2];
258         double n[2];
259         double c[2];
260         double d[2];
261         double h[2];
262
263         t_tersoff_exchange exchange;    /* exchange between 2bp and 3bp calc */
264 } t_tersoff_mult_params;
265
266
267
268 /*
269  *
270  *  defines
271  *
272  */
273
274 #define ONE_THIRD       (1.0/3.0)
275
276 /*
277  * default values
278  *
279  * - length unit: 1 A (1 A = 1e-10 m)
280  * - time unit: 1 fs (1 fs = 1e-15 s)
281  * - mass unit: 1 amu (1 amu = 1.6605388628e-27 kg )
282  *
283  * fyi: in the following 1 N = (amu*A)/(fs*fs)
284  *
285  */
286
287 #define METER                           1e10                    /* A */
288 #define SECOND                          1e15                    /* fs */
289 #define AMU                             1.6605388628e-27        /* kg */
290 #define KILOGRAM                        (1.0/AMU)               /* amu */
291 #define NEWTON  (METER*KILOGRAM/(SECOND*SECOND))        /* A amu / fs^2 */
292 #define PASCAL  (NEWTON/(METER*METER))                  /* N / A^2 */
293 #define ATM     ((1.0133e5*PASCAL))                     /* N / A^2 */
294
295 #define MOLDYN_TEMP                     273.0
296 #define MOLDYN_TAU                      1.0
297 #define MOLDYN_CUTOFF                   10.0
298 #define MOLDYN_RUNS                     1000000
299
300 #define MOLDYN_INTEGRATE_VERLET         0x00
301 #define MOLDYN_INTEGRATE_DEFAULT        MOLDYN_INTEGRATE_VERLET
302
303 #define MOLDYN_POTENTIAL_HO             0x00
304 #define MOLDYN_POTENTIAL_LJ             0x01
305 #define MOLDYN_POTENTIAL_TM             0x02
306
307 #define LOG_TOTAL_ENERGY                0x01
308 #define LOG_TOTAL_MOMENTUM              0x02
309 #define SAVE_STEP                       0x04
310 #define VISUAL_STEP                     0x08
311 #define CREATE_REPORT                   0x10
312
313 #define TRUE                            1
314 #define FALSE                           0
315
316 #define VERBOSE                         1
317 #define QUIET                           0
318
319 /*
320  *
321  * phsical values / constants
322  *
323  *
324  */
325
326 #define K_BOLTZMANN             (1.380650524e-23*METER*NEWTON)  /* NA/K */
327 #define EV                      (1.6021765314e-19*METER*NEWTON) /* NA */
328
329 #define C                       0x06
330 #define M_C                     12.011                          /* amu */
331
332 #define SI                      0x0e
333 #define LC_SI                   (0.543105e-9*METER)             /* A */
334 #define M_SI                    28.08553                        /* amu */
335
336 //#define LJ_SIGMA_SI           ((0.25*sqrt(3.0)*LC_SI)/1.122462)       /* A */
337 #define LJ_SIGMA_SI             (LC_SI/1.122462)                        /* A */
338 #define LJ_EPSILON_SI           (2.1678*EV)                             /* NA */
339
340 #define TM_R_SI                 (2.7e-10*METER)                 /* A */
341 #define TM_S_SI                 (3.0e-10*METER)                 /* A */
342 #define TM_A_SI                 (1830.8*EV)                     /* NA */
343 #define TM_B_SI                 (471.18*EV)                     /* NA */
344 #define TM_LAMBDA_SI            (2.4799e10/METER)               /* 1/A */
345 #define TM_MU_SI                (1.7322e10/METER)               /* 1/A */
346 #define TM_BETA_SI              1.1000e-6
347 #define TM_N_SI                 0.78734
348 #define TM_C_SI                 1.0039e5
349 #define TM_D_SI                 16.217
350 #define TM_H_SI                 -0.59825
351
352 #define TM_R_C                  (1.8e-10*METER)                 /* A */
353 #define TM_S_C                  (2.1e-10*METER)                 /* A */
354 #define TM_A_C                  (1393.6*EV)                     /* NA */
355 #define TM_B_C                  (346.7*EV)                      /* NA */
356 #define TM_LAMBDA_C             (3.4879e10/METER)               /* 1/A */
357 #define TM_MU_C                 (2.2119e10/METER)               /* 1/A */
358 #define TM_BETA_C               1.5724e-7
359 #define TM_N_C                  0.72751
360 #define TM_C_C                  3.8049e4
361 #define TM_D_C                  4.384
362 #define TM_H_C                  -0.57058
363
364 #define TM_CHI_SIC              0.9776
365
366 /*
367  * lattice constants
368  */
369
370 #define CUBIC                   0x01
371 #define FCC                     0x02
372 #define DIAMOND                 0x04
373
374
375 /*
376  *
377  * function prototypes
378  *
379  */
380
381 typedef int (*pf_func1b)(t_moldyn *,t_atom *ai);
382 typedef int (*pf_func2b)(t_moldyn *,t_atom *,t_atom *,u8 bc);
383 typedef int (*pf_func2b_post)(t_moldyn *,t_atom *,t_atom *,u8 bc);
384 typedef int (*pf_func3b)(t_moldyn *,t_atom *,t_atom *,t_atom *,u8 bc);
385
386 int moldyn_init(t_moldyn *moldyn,int argc,char **argv);
387 int moldyn_shutdown(t_moldyn *moldyn);
388
389 int set_int_alg(t_moldyn *moldyn,u8 algo);
390 int set_cutoff(t_moldyn *moldyn,double cutoff);
391 int set_temperature(t_moldyn *moldyn,double t_ref);
392 int set_pressure(t_moldyn *moldyn,double p_ref);
393 int set_pt_scale(t_moldyn *moldyn,u8 ptype,double ptc,u8 ttype,double ttc);
394 int set_dim(t_moldyn *moldyn,double x,double y,double z,u8 visualize);
395 int set_nn_dist(t_moldyn *moldyn,double dist);
396 int set_pbc(t_moldyn *moldyn,u8 x,u8 y,u8 z);
397 int set_potential1b(t_moldyn *moldyn,pf_func1b func,void *params);
398 int set_potential2b(t_moldyn *moldyn,pf_func2b func,void *params);
399 int set_potential2b_post(t_moldyn *moldyn,pf_func2b_post func,void *params);
400 int set_potential3b(t_moldyn *moldyn,pf_func3b func,void *params);
401
402 int moldyn_set_log_dir(t_moldyn *moldyn,char *dir);
403 int moldyn_set_report(t_moldyn *moldyn,char *author,char *title);
404 int moldyn_set_log(t_moldyn *moldyn,u8 type,int timer);
405 int moldyn_log_shutdown(t_moldyn *moldyn);
406
407 int create_lattice(t_moldyn *moldyn,u8 type,double lc,int element,double mass,
408                    u8 attr,u8 brand,int a,int b,int c);
409 int cubic_init(int a,int b,int c,double lc,t_atom *atom,t_3dvec *origin);
410 int fcc_init(int a,int b,int c,double lc,t_atom *atom,t_3dvec *origin);
411 int diamond_init(int a,int b,int c,double lc,t_atom *atom,t_3dvec *origin);
412 int add_atom(t_moldyn *moldyn,int element,double mass,u8 brand,u8 attr,
413              t_3dvec *r,t_3dvec *v);
414 int destroy_atoms(t_moldyn *moldyn);
415
416 int thermal_init(t_moldyn *moldyn,u8 equi_init);
417 double temperature_calc(t_moldyn *moldyn);
418 double get_temperature(t_moldyn *moldyn);
419 int scale_velocity(t_moldyn *moldyn,u8 equi_init);
420 double pressure_calc(t_moldyn *moldyn);
421 double thermodynamic_pressure_calc(t_moldyn *moldyn);
422 double get_pressure(t_moldyn *moldyn);
423 int scale_volume(t_moldyn *moldyn);
424 int scale_dim(t_moldyn *moldyn,double scale,u8 x,u8 y,u8 z);
425 int scale_atoms(t_moldyn *moldyn,double scale,u8 x,u8 y,u8 z);
426
427 double get_e_kin(t_moldyn *moldyn);
428 double update_e_kin(t_moldyn *moldyn);
429 double get_total_energy(t_moldyn *moldyn);
430 t_3dvec get_total_p(t_moldyn *moldyn);
431
432 double estimate_time_step(t_moldyn *moldyn,double nn_dist);
433
434 int link_cell_init(t_moldyn *moldyn,u8 vol);
435 int link_cell_update(t_moldyn *moldyn);
436 int link_cell_neighbour_index(t_moldyn *moldyn,int i,int j,int k,t_list *cell);
437 int link_cell_shutdown(t_moldyn *moldyn);
438
439 typedef int (*set_hook)(void *,void *);
440
441 int moldyn_add_schedule(t_moldyn *moldyn,int runs,double tau);
442 int moldyn_set_schedule_hook(t_moldyn *moldyn,set_hook hook,void *hook_params);
443
444 int moldyn_integrate(t_moldyn *moldyn);
445 int velocity_verlet(t_moldyn *moldyn);
446
447 int potential_force_calc(t_moldyn *moldyn);
448 inline int virial_calc(t_atom *a,t_3dvec *f,t_3dvec *d)
449         __attribute__((always_inline));
450 inline int check_per_bound(t_moldyn *moldyn,t_3dvec *a)
451         __attribute__((always_inline));
452 int harmonic_oscillator(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc);
453 int lennard_jones(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc);
454 int tersoff_mult_complete_params(t_tersoff_mult_params *p);
455 int tersoff_mult_1bp(t_moldyn *moldyn,t_atom *ai);
456 int tersoff_mult_2bp(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc);
457 int tersoff_mult_post_2bp(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc);
458 int tersoff_mult_3bp(t_moldyn *moldyn,t_atom *ai,t_atom *aj,t_atom *ak,u8 bc);
459
460 int moldyn_bc_check(t_moldyn *moldyn);
461
462 #endif