cleaning + set hook function
[physik/posic.git] / moldyn.c
index c21f59f..9db5cd9 100644 (file)
--- a/moldyn.c
+++ b/moldyn.c
@@ -19,6 +19,8 @@
 
 int moldyn_init(t_moldyn *moldyn,int argc,char **argv) {
 
+       printf("[moldyn] init\n");
+
        memset(moldyn,0,sizeof(t_moldyn));
 
        rand_init(&(moldyn->random),NULL,1);
@@ -30,6 +32,7 @@ int moldyn_init(t_moldyn *moldyn,int argc,char **argv) {
 int moldyn_shutdown(t_moldyn *moldyn) {
 
        printf("[moldyn] shutdown\n");
+
        moldyn_log_shutdown(moldyn);
        link_cell_shutdown(moldyn);
        rand_close(&(moldyn->random));
@@ -40,12 +43,16 @@ int moldyn_shutdown(t_moldyn *moldyn) {
 
 int set_int_alg(t_moldyn *moldyn,u8 algo) {
 
+       printf("[moldyn] integration algorithm: ");
+
        switch(algo) {
                case MOLDYN_INTEGRATE_VERLET:
                        moldyn->integrate=velocity_verlet;
+                       printf("velocity verlet\n");
                        break;
                default:
                        printf("unknown integration algorithm: %02x\n",algo);
+                       printf("unknown\n");
                        return -1;
        }
 
@@ -56,6 +63,8 @@ int set_cutoff(t_moldyn *moldyn,double cutoff) {
 
        moldyn->cutoff=cutoff;
 
+       printf("[moldyn] cutoff [A]: %f\n",moldyn->cutoff);
+
        return 0;
 }
 
@@ -63,6 +72,8 @@ int set_temperature(t_moldyn *moldyn,double t_ref) {
 
        moldyn->t_ref=t_ref;
 
+       printf("[moldyn] temperature: %f\n",moldyn->t_ref);
+
        return 0;
 }
 
@@ -70,6 +81,8 @@ int set_pressure(t_moldyn *moldyn,double p_ref) {
 
        moldyn->p_ref=p_ref;
 
+       printf("[moldyn] pressure: %f\n",moldyn->p_ref);
+
        return 0;
 }
 
@@ -79,6 +92,18 @@ int set_pt_scale(t_moldyn *moldyn,u8 ptype,double ptc,u8 ttype,double ttc) {
        moldyn->t_tc=ttc;
        moldyn->p_tc=ptc;
 
+       printf("[moldyn] p/t scaling:\n");
+
+       printf("  p: %s",ptype?"yes":"no ");
+       if(ptype)
+               printf(" | type: %02x | factor: %f",ptype,ptc);
+       printf("\n");
+
+       printf("  t: %s",ttype?"yes":"no ");
+       if(ttype)
+               printf(" | type: %02x | factor: %f",ttype,ttc);
+       printf("\n");
+
        return 0;
 }
 
@@ -101,7 +126,7 @@ int set_dim(t_moldyn *moldyn,double x,double y,double z,u8 visualize) {
        printf("  y: %f\n",moldyn->dim.y);
        printf("  z: %f\n",moldyn->dim.z);
        printf("  volume: %f\n",moldyn->volume);
-       printf("  visualize simulation box: %s\n",visualize?"on":"off");
+       printf("  visualize simulation box: %s\n",visualize?"yes":"no");
 
        return 0;
 }
@@ -115,6 +140,8 @@ int set_nn_dist(t_moldyn *moldyn,double dist) {
 
 int set_pbc(t_moldyn *moldyn,u8 x,u8 y,u8 z) {
 
+       printf("[moldyn] periodic boundary conditions:\n");
+
        if(x)
                moldyn->status|=MOLDYN_STAT_PBX;
 
@@ -124,6 +151,10 @@ int set_pbc(t_moldyn *moldyn,u8 x,u8 y,u8 z) {
        if(z)
                moldyn->status|=MOLDYN_STAT_PBZ;
 
+       printf("  x: %s\n",x?"yes":"no");
+       printf("  y: %s\n",y?"yes":"no");
+       printf("  z: %s\n",z?"yes":"no");
+
        return 0;
 }
 
@@ -171,6 +202,8 @@ int moldyn_set_log(t_moldyn *moldyn,u8 type,int timer) {
        char filename[128];
        int ret;
 
+       printf("[moldyn] set log: ");
+
        switch(type) {
                case LOG_TOTAL_ENERGY:
                        moldyn->ewrite=timer;
@@ -183,6 +216,7 @@ int moldyn_set_log(t_moldyn *moldyn,u8 type,int timer) {
                                return moldyn->efd;
                        }
                        dprintf(moldyn->efd,"# total energy log file\n");
+                       printf("total energy (%d)\n",timer);
                        break;
                case LOG_TOTAL_MOMENTUM:
                        moldyn->mwrite=timer;
@@ -195,9 +229,11 @@ int moldyn_set_log(t_moldyn *moldyn,u8 type,int timer) {
                                return moldyn->mfd;
                        }
                        dprintf(moldyn->efd,"# total momentum log file\n");
+                       printf("total momentum (%d)\n",timer);
                        break;
                case SAVE_STEP:
                        moldyn->swrite=timer;
+                       printf("save file (%d)\n",timer);
                        break;
                case VISUAL_STEP:
                        moldyn->vwrite=timer;
@@ -206,9 +242,10 @@ int moldyn_set_log(t_moldyn *moldyn,u8 type,int timer) {
                                printf("[moldyn] visual init failure\n");
                                return ret;
                        }
+                       printf("visual file (%d)\n",timer);
                        break;
                default:
-                       printf("[moldyn] unknown log mechanism: %02x\n",type);
+                       printf("unknown log type: %02x\n",type);
                        return -1;
        }
 
@@ -429,6 +466,8 @@ int thermal_init(t_moldyn *moldyn,u8 equi_init) {
        atom=moldyn->atom;
        random=&(moldyn->random);
 
+       printf("[moldyn] thermal init (equi init: %s)\n",equi_init?"yes":"no");
+
        /* gaussian distribution of velocities */
        v3_zero(&p_total);
        for(i=0;i<moldyn->count;i++) {
@@ -460,6 +499,28 @@ int thermal_init(t_moldyn *moldyn,u8 equi_init) {
        return 0;
 }
 
+double temperature_calc(t_moldyn *moldyn) {
+
+       double double_ekin;
+       int i;
+       t_atom *atom;
+
+       atom=moldyn->atom;
+
+       for(i=0;i<moldyn->count;i++)
+               double_ekin+=atom[i].mass*v3_absolute_square(&(atom[i].v));
+
+       /* kinetic energy = 3/2 N k_B T */
+       moldyn->t=double_ekin/(3.0*K_BOLTZMANN*moldyn->count);
+
+       return moldyn->t;
+}
+
+double get_temperature(t_moldyn *moldyn) {
+
+       return moldyn->t;
+}
+
 int scale_velocity(t_moldyn *moldyn,u8 equi_init) {
 
        int i;
@@ -478,10 +539,11 @@ int scale_velocity(t_moldyn *moldyn,u8 equi_init) {
        count=0;
        for(i=0;i<moldyn->count;i++) {
                if((equi_init&TRUE)||(atom[i].attr&ATOM_ATTR_HB)) {
-                       e+=0.5*atom[i].mass*v3_absolute_square(&(atom[i].v));
+                       e+=atom[i].mass*v3_absolute_square(&(atom[i].v));
                        count+=1;
                }
        }
+       e*=0.5;
        if(count!=0) moldyn->t=e/(1.5*count*K_BOLTZMANN);
        else return 0;  /* no atoms involved in scaling! */
        
@@ -515,6 +577,34 @@ int scale_velocity(t_moldyn *moldyn,u8 equi_init) {
        return 0;
 }
 
+double pressure_calc(t_moldyn *moldyn) {
+
+       int i;
+       t_atom *atom;
+       double p1,p2,p=0;
+       
+       for(i=0;i<moldyn->count;i++) {
+               
+
+       }
+
+       p1=(moldyn->count*K_BOLTZMANN*moldyn->t-ONE_THIRD*moldyn->vt1);
+       p1/=moldyn->volume;
+
+       p2=(moldyn->count*K_BOLTZMANN*moldyn->t-ONE_THIRD*moldyn->vt2);
+       p2/=moldyn->volume;
+
+       printf("compare pressures: %f %f\n",p1/ATM,p2/ATM);
+
+       return moldyn->p;
+}      
+
+double get_pressure(t_moldyn *moldyn) {
+
+       return moldyn->p;
+
+}
+
 int scale_volume(t_moldyn *moldyn) {
 
        t_atom *atom;
@@ -592,11 +682,6 @@ double get_e_kin(t_moldyn *moldyn) {
        return moldyn->ekin;
 }
 
-double get_e_pot(t_moldyn *moldyn) {
-
-       return moldyn->energy;
-}
-
 double update_e_kin(t_moldyn *moldyn) {
 
        return(get_e_kin(moldyn));
@@ -659,6 +744,9 @@ int link_cell_init(t_moldyn *moldyn) {
        lc->cells=lc->nx*lc->ny*lc->nz;
        lc->subcell=malloc(lc->cells*sizeof(t_list));
 
+       if(lc->cells<27)
+               printf("[moldyn] FATAL: less then 27 subcells!\n");
+
        printf("[moldyn] initializing linked cells (%d)\n",lc->cells);
 
        for(i=0;i<lc->cells;i++)
@@ -797,10 +885,14 @@ int moldyn_add_schedule(t_moldyn *moldyn,int runs,double tau) {
        schedule->tau=ptr;
        schedule->tau[count-1]=tau;
 
+       printf("[moldyn] schedule added:\n");
+       printf("  number: %d | runs: %d | tau: %f\n",count-1,runs,tau);
+                                      
+
        return 0;
 }
 
-int moldyn_set_schedule_hook(t_moldyn *moldyn,void *hook,void *hook_params) {
+int moldyn_set_schedule_hook(t_moldyn *moldyn,set_hook hook,void *hook_params) {
 
        moldyn->schedule.hook=hook;
        moldyn->schedule.hook_params=hook_params;
@@ -863,6 +955,9 @@ int moldyn_integrate(t_moldyn *moldyn) {
        /* debugging, ignore */
        moldyn->debug=0;
 
+       /* tell the world */
+       printf("[moldyn] integration start, go get a coffee ...\n");
+
        /* executing the schedule */
        for(sched->count=0;sched->count<sched->total_sched;sched->count++) {
 
@@ -885,13 +980,21 @@ int moldyn_integrate(t_moldyn *moldyn) {
                        scale_volume(moldyn);
 
                /* check for log & visualization */
+//double ax;
+//double ao;
+//double av;
                if(e) {
                        if(!(i%e))
+//ao=sqrt(0.1/M_SI);
+//ax=((0.28-0.25)*sqrt(3)*LC_SI/2)*cos(ao*i);
+//av=ao*(0.28-0.25)*sqrt(3)*LC_SI/2*sin(ao*i);
+                               update_e_kin(moldyn);
                                dprintf(moldyn->efd,
                                        "%f %f %f %f\n",
-                                       moldyn->time,update_e_kin(moldyn),
+                                       moldyn->time,moldyn->ekin,
                                        moldyn->energy,
                                        get_total_energy(moldyn));
+//moldyn->atom[0].r.x,ax,av*av*M_SI,0.1*ax*ax,av*av*M_SI+0.1*ax*ax);
                }
                if(m) {
                        if(!(i%m)) {
@@ -946,7 +1049,7 @@ int moldyn_integrate(t_moldyn *moldyn) {
 int velocity_verlet(t_moldyn *moldyn) {
 
        int i,count;
-       double tau,tau_square;
+       double tau,tau_square,h;
        t_3dvec delta;
        t_atom *atom;
 
@@ -957,14 +1060,15 @@ int velocity_verlet(t_moldyn *moldyn) {
 
        for(i=0;i<count;i++) {
                /* new positions */
+               h=0.5/atom[i].mass;
                v3_scale(&delta,&(atom[i].v),tau);
                v3_add(&(atom[i].r),&(atom[i].r),&delta);
-               v3_scale(&delta,&(atom[i].f),0.5*tau_square/atom[i].mass);
+               v3_scale(&delta,&(atom[i].f),h*tau_square);
                v3_add(&(atom[i].r),&(atom[i].r),&delta);
                check_per_bound(moldyn,&(atom[i].r));
 
-               /* velocities */
-               v3_scale(&delta,&(atom[i].f),0.5*tau/atom[i].mass);
+               /* velocities [actually v(t+tau/2)] */
+               v3_scale(&delta,&(atom[i].f),h*tau);
                v3_add(&(atom[i].v),&(atom[i].v),&delta);
        }
 
@@ -975,7 +1079,7 @@ int velocity_verlet(t_moldyn *moldyn) {
        potential_force_calc(moldyn);
 
        for(i=0;i<count;i++) {
-               /* again velocities */
+               /* again velocities [actually v(t+tau)] */
                v3_scale(&delta,&(atom[i].f),0.5*tau/atom[i].mass);
                v3_add(&(atom[i].v),&(atom[i].v),&delta);
        }
@@ -986,7 +1090,7 @@ int velocity_verlet(t_moldyn *moldyn) {
 
 /*
  *
- * potentials & corresponding forces
+ * potentials & corresponding forces & virial routine
  * 
  */
 
@@ -1010,6 +1114,8 @@ int potential_force_calc(t_moldyn *moldyn) {
 
        /* reset energy */
        moldyn->energy=0.0;
+
+       moldyn->vt2=0.0;
        
        /* get energy and force of every atom */
        for(i=0;i<count;i++) {
@@ -1025,6 +1131,7 @@ int potential_force_calc(t_moldyn *moldyn) {
                virial->xy=0.0;
                virial->xz=0.0;
                virial->yz=0.0;
+               moldyn->vt1=0.0;
 
                /* reset site energy */
                itom[i].e=0.0;
@@ -1134,6 +1241,30 @@ printf("\n\n");
 printf("\n\n");
 #endif
 
+       moldyn->vt2=0.0;
+       for(i=0;i<count;i++)
+               moldyn->vt2-=v3_scalar_product(&(itom[i].r),&(itom[i].f));
+
+//printf("compare: vt1: %f vt2: %f\n",moldyn->vt1,moldyn->vt2);
+
+//pressure_calc(moldyn);
+
+       return 0;
+}
+
+/*
+ * virial calculation
+ */
+
+inline int virial_calc(t_atom *a,t_3dvec *f,t_3dvec *d) {
+
+       a->virial.xx-=f->x*d->x;
+       a->virial.yy-=f->y*d->y;
+       a->virial.zz-=f->z*d->z;
+       a->virial.xy-=f->x*d->y;
+       a->virial.xz-=f->x*d->z;
+       a->virial.yz-=f->y*d->z;
+
        return 0;
 }
 
@@ -1179,23 +1310,29 @@ int harmonic_oscillator(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc) {
 
        t_ho_params *params;
        t_3dvec force,distance;
-       double d;
+       double d,f;
        double sc,equi_dist;
 
        params=moldyn->pot2b_params;
        sc=params->spring_constant;
        equi_dist=params->equilibrium_distance;
 
+       if(ai<aj) return 0;
+
        v3_sub(&distance,&(aj->r),&(ai->r));
        
        if(bc) check_per_bound(moldyn,&distance);
        d=v3_norm(&distance);
        if(d<=moldyn->cutoff) {
-               /* energy is 1/2 (d-d0)^2, but we will add this twice ... */
-               moldyn->energy+=(0.25*sc*(d-equi_dist)*(d-equi_dist));
+               moldyn->energy+=(0.5*sc*(d-equi_dist)*(d-equi_dist));
                /* f = -grad E; grad r_ij = -1 1/r_ij distance */
-               v3_scale(&force,&distance,sc*(1.0-(equi_dist/d)));
+               f=sc*(1.0-equi_dist/d);
+               v3_scale(&force,&distance,f);
                v3_add(&(ai->f),&(ai->f),&force);
+               virial_calc(ai,&force,&distance);
+               virial_calc(aj,&force,&distance); /* f and d signe switched */
+               v3_scale(&force,&distance,-f);
+               v3_add(&(aj->f),&(aj->f),&force);
        }
 
        return 0;
@@ -1215,6 +1352,8 @@ int lennard_jones(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc) {
        sig6=params->sigma6;
        sig12=params->sigma12;
 
+       if(ai<aj) return 0;
+
        v3_sub(&distance,&(aj->r),&(ai->r));
        if(bc) check_per_bound(moldyn,&distance);
        d=v3_absolute_square(&distance);        /* 1/r^2 */
@@ -1223,16 +1362,20 @@ int lennard_jones(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc) {
                h2=d*d;                         /* 1/r^4 */
                h2*=d;                          /* 1/r^6 */
                h1=h2*h2;                       /* 1/r^12 */
-               /* energy is eps*..., but we will add this twice ... */
-               moldyn->energy+=0.5*eps*(sig12*h1-sig6*h2);
+               moldyn->energy+=(eps*(sig12*h1-sig6*h2)-params->uc);
                h2*=d;                          /* 1/r^8 */
                h1*=d;                          /* 1/r^14 */
                h2*=6*sig6;
                h1*=12*sig12;
                d=+h1-h2;
                d*=eps;
+               v3_scale(&force,&distance,d);
+               v3_add(&(aj->f),&(aj->f),&force);
                v3_scale(&force,&distance,-1.0*d); /* f = - grad E */
                v3_add(&(ai->f),&(ai->f),&force);
+               virial_calc(ai,&force,&distance);
+               virial_calc(aj,&force,&distance); /* f and d signe switched */
+               moldyn->vt1-=v3_scalar_product(&force,&distance);
        }
 
        return 0;
@@ -1526,7 +1669,6 @@ int tersoff_mult_post_2bp(t_moldyn *moldyn,t_atom *ai,t_atom *aj,u8 bc) {
        zeta=exchange->zeta_ij;
        if(zeta==0.0) {
                moldyn->debug++;                /* just for debugging ... */
-               db=0.0;
                b=chi;
                v3_scale(&force,dist_ij,df_a*b*f_c);
        }
@@ -1602,12 +1744,13 @@ if(ai==&(moldyn->atom[0])) {
        v3_add(&(ai->f),&(ai->f),&force);
 
        /* virial - plus sign, as dist_ij = - dist_ji - (really??) */
-       ai->virial.xx+=force.x*dist_ij->x;
-       ai->virial.yy+=force.y*dist_ij->y;
-       ai->virial.zz+=force.z*dist_ij->z;
-       ai->virial.xy+=force.x*dist_ij->y;
-       ai->virial.xz+=force.x*dist_ij->z;
-       ai->virial.yz+=force.y*dist_ij->z;
+// TEST ... with a minus instead
+       ai->virial.xx-=force.x*dist_ij->x;
+       ai->virial.yy-=force.y*dist_ij->y;
+       ai->virial.zz-=force.z*dist_ij->z;
+       ai->virial.xy-=force.x*dist_ij->y;
+       ai->virial.xz-=force.x*dist_ij->z;
+       ai->virial.yz-=force.y*dist_ij->z;
 
 #ifdef DEBUG
 if(ai==&(moldyn->atom[0])) {