#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
+#include <sys/time.h>
+#include <time.h>
#include <math.h>
#include "moldyn.h"
#include "report/report.h"
+/*
+ * global variables, pse and atom colors (only needed here)
+ */
+
+static char *pse_name[]={
+ "*",
+ "H",
+ "He",
+ "Li",
+ "Be",
+ "B",
+ "C",
+ "N",
+ "O",
+ "F",
+ "Ne",
+ "Na",
+ "Mg",
+ "Al",
+ "Si",
+ "P",
+ "S",
+ "Cl",
+ "Ar",
+};
+
+static char *pse_col[]={
+ "*",
+ "White",
+ "He",
+ "Li",
+ "Be",
+ "B",
+ "Gray",
+ "N",
+ "Blue",
+ "F",
+ "Ne",
+ "Na",
+ "Mg",
+ "Al",
+ "Yellow",
+ "P",
+ "S",
+ "Cl",
+ "Ar",
+};
+
+/*
+ * the moldyn functions
+ */
+
int moldyn_init(t_moldyn *moldyn,int argc,char **argv) {
printf("[moldyn] init\n");
memset(moldyn,0,sizeof(t_moldyn));
+ moldyn->argc=argc;
+ moldyn->args=argv;
+
rand_init(&(moldyn->random),NULL,1);
moldyn->random.status|=RAND_STAT_VERBOSE;
return 0;
}
+int set_bondlen(t_moldyn *moldyn,double b0,double b1,double bm) {
+
+ moldyn->bondlen[0]=b0*b0;
+ moldyn->bondlen[1]=b1*b1;
+ if(bm<0)
+ moldyn->bondlen[2]=b0*b1;
+ else
+ moldyn->bondlen[2]=bm*bm;
+
+ return 0;
+}
+
int set_temperature(t_moldyn *moldyn,double t_ref) {
moldyn->t_ref=t_ref;
break;
case VISUAL_STEP:
moldyn->vwrite=timer;
- ret=visual_init(&(moldyn->vis),moldyn->vlsdir);
+ ret=visual_init(moldyn,moldyn->vlsdir);
if(ret<0) {
printf("[moldyn] visual init failure\n");
return ret;
moldyn->vlsdir);
system(sc);
}
- if(&(moldyn->vis)) visual_tini(&(moldyn->vis));
return 0;
}
atom[ret].brand=brand;
atom[ret].tag=count+ret;
check_per_bound(moldyn,&(atom[ret].r));
+ atom[ret].r_0=atom[ret].r;
}
/* update total system mass */
return ret;
}
+int add_atom(t_moldyn *moldyn,int element,double mass,u8 brand,u8 attr,
+ t_3dvec *r,t_3dvec *v) {
+
+ t_atom *atom;
+ void *ptr;
+ int count;
+
+ atom=moldyn->atom;
+ count=(moldyn->count)++;
+
+ ptr=realloc(atom,(count+1)*sizeof(t_atom));
+ if(!ptr) {
+ perror("[moldyn] realloc (add atom)");
+ return -1;
+ }
+ moldyn->atom=ptr;
+
+ atom=moldyn->atom;
+ atom[count].r=*r;
+ atom[count].v=*v;
+ atom[count].element=element;
+ atom[count].mass=mass;
+ atom[count].brand=brand;
+ atom[count].tag=count;
+ atom[count].attr=attr;
+ check_per_bound(moldyn,&(atom[count].r));
+ atom[count].r_0=atom[count].r;
+
+ /* update total system mass */
+ total_mass_calc(moldyn);
+
+ return 0;
+}
+
+int del_atom(t_moldyn *moldyn,int tag) {
+
+ t_atom *new,*old;
+ int cnt;
+
+ old=moldyn->atom;
+
+ new=(t_atom *)malloc((moldyn->count-1)*sizeof(t_atom));
+ if(!new) {
+ perror("[moldyn]malloc (del atom)");
+ return -1;
+ }
+
+ for(cnt=0;cnt<tag;cnt++)
+ new[cnt]=old[cnt];
+
+ for(cnt=tag+1;cnt<moldyn->count;cnt++) {
+ new[cnt-1]=old[cnt];
+ new[cnt-1].tag=cnt-1;
+ }
+
+ moldyn->count-=1;
+ moldyn->atom=new;
+
+ free(old);
+
+ return 0;
+}
+
/* cubic init */
int cubic_init(int a,int b,int c,double lc,t_atom *atom,t_3dvec *origin) {
return count;
}
-int add_atom(t_moldyn *moldyn,int element,double mass,u8 brand,u8 attr,
- t_3dvec *r,t_3dvec *v) {
-
- t_atom *atom;
- void *ptr;
- int count;
-
- atom=moldyn->atom;
- count=(moldyn->count)++;
-
- ptr=realloc(atom,(count+1)*sizeof(t_atom));
- if(!ptr) {
- perror("[moldyn] realloc (add atom)");
- return -1;
- }
- moldyn->atom=ptr;
-
- atom=moldyn->atom;
- atom[count].r=*r;
- atom[count].v=*v;
- atom[count].element=element;
- atom[count].mass=mass;
- atom[count].brand=brand;
- atom[count].tag=count;
- atom[count].attr=attr;
-
- /* update total system mass */
- total_mass_calc(moldyn);
-
- return 0;
-}
-
int destroy_atoms(t_moldyn *moldyn) {
if(moldyn->atom) free(moldyn->atom);
atom=moldyn->atom;
moldyn->ekin=0.0;
- for(i=0;i<moldyn->count;i++)
- moldyn->ekin+=0.5*atom[i].mass*v3_absolute_square(&(atom[i].v));
+ for(i=0;i<moldyn->count;i++) {
+ atom[i].ekin=0.5*atom[i].mass*v3_absolute_square(&(atom[i].v));
+ moldyn->ekin+=atom[i].ekin;
+ }
return moldyn->ekin;
}
char dir[128];
double ds;
double energy_scale;
+ struct timeval t1,t2;
//double tp;
sched=&(moldyn->schedule);
moldyn->tau_square=moldyn->tau*moldyn->tau;
moldyn->cutoff_square=moldyn->cutoff*moldyn->cutoff;
- /* energy scaling factor */
- energy_scale=moldyn->count*EV;
+ /* get current time */
+ gettimeofday(&t1,NULL);
/* calculate initial forces */
potential_force_calc(moldyn);
#ifdef DEBUG
-return 0;
+//return 0;
#endif
/* some stupid checks before we actually start calculating bullshit */
moldyn->tau_square=moldyn->tau*moldyn->tau;
moldyn->time_steps=sched->runs[sched->count];
+ /* energy scaling factor (might change!) */
+ energy_scale=moldyn->count*EV;
+
/* integration according to schedule */
for(i=0;i<moldyn->time_steps;i++) {
/* check for log & visualization */
if(e) {
- if(!(i%e))
+ if(!(moldyn->total_steps%e))
dprintf(moldyn->efd,
"%f %f %f %f\n",
moldyn->time,moldyn->ekin/energy_scale,
get_total_energy(moldyn)/energy_scale);
}
if(m) {
- if(!(i%m)) {
+ if(!(moldyn->total_steps%m)) {
momentum=get_total_p(moldyn);
dprintf(moldyn->mfd,
"%f %f %f %f %f\n",moldyn->time,
}
}
if(p) {
- if(!(i%p)) {
+ if(!(moldyn->total_steps%p)) {
dprintf(moldyn->pfd,
"%f %f %f %f %f\n",moldyn->time,
moldyn->p/BAR,moldyn->p_avg/BAR,
}
}
if(t) {
- if(!(i%t)) {
+ if(!(moldyn->total_steps%t)) {
dprintf(moldyn->tfd,
"%f %f %f\n",
moldyn->time,moldyn->t,moldyn->t_avg);
}
}
if(s) {
- if(!(i%s)) {
+ if(!(moldyn->total_steps%s)) {
snprintf(dir,128,"%s/s-%07.f.save",
moldyn->vlsdir,moldyn->time);
- fd=open(dir,O_WRONLY|O_TRUNC|O_CREAT);
+ fd=open(dir,O_WRONLY|O_TRUNC|O_CREAT,
+ S_IRUSR|S_IWUSR);
if(fd<0) perror("[moldyn] save fd open");
else {
write(fd,moldyn,sizeof(t_moldyn));
}
}
if(v) {
- if(!(i%v)) {
- visual_atoms(&(moldyn->vis),moldyn->time,
- moldyn->atom,moldyn->count);
+ if(!(moldyn->total_steps%v)) {
+ visual_atoms(moldyn);
}
}
/* display progress */
- if(!(i%10)) {
- printf("\rsched:%d, steps:%d, T:%3.1f/%3.1f P:%4.1f/%4.1f V:%6.1f",
+ if(!(moldyn->total_steps%10)) {
+ /* get current time */
+ gettimeofday(&t2,NULL);
+
+ printf("\rsched:%d, steps:%d, T:%3.1f/%3.1f P:%4.1f/%4.1f V:%6.1f (%d)",
sched->count,i,
moldyn->t,moldyn->t_avg,
- moldyn->p_avg/BAR,moldyn->p/BAR,
- moldyn->volume);
+ moldyn->p_avg/BAR,moldyn->gp_avg/BAR,
+ moldyn->volume,
+ (int)(t2.tv_sec-t1.tv_sec));
fflush(stdout);
+
+ /* copy over time */
+ t1=t2;
}
/* increase absolute time */
}
/* check for hooks */
- if(sched->count+1<sched->total_sched) {
- if(sched->hook) {
- printf("\n ## schedule hook %d/%d start ##\n",
- sched->count+1,sched->total_sched-1);
- sched->hook(moldyn,sched->hook_params);
- printf(" ## schedule hook end ##\n");
- }
+ if(sched->hook) {
+ printf("\n ## schedule hook %d/%d start ##\n",
+ sched->count+1,sched->total_sched-1);
+ sched->hook(moldyn,sched->hook_params);
+ printf(" ## schedule hook end ##\n");
}
/* increase the schedule counter */
}
#ifdef DEBUG
- printf("\nATOM 0: %f %f %f\n\n",itom->f.x,itom->f.y,itom->f.z);
+ //printf("\nATOM 0: %f %f %f\n\n",itom->f.x,itom->f.y,itom->f.z);
+ if(moldyn->time>DSTART&&moldyn->time<DEND) {
+ printf("force:\n");
+ printf(" x: %0.40f\n",moldyn->atom[5832].f.x);
+ printf(" y: %0.40f\n",moldyn->atom[5832].f.y);
+ printf(" z: %0.40f\n",moldyn->atom[5832].f.z);
+ }
#endif
/* calculate global virial */
return 0;
}
+/*
+ * restore function
+ */
+
+int moldyn_load(t_moldyn *moldyn) {
+
+ // later ...
+
+ return 0;
+}
+
/*
* post processing functions
*/
}
}
+int analyze_bonds(t_moldyn *moldyn) {
+
+
+
+
+ return 0;
+}
+
+/*
+ * visualization code
+ */
+
+int visual_init(t_moldyn *moldyn,char *filebase) {
+
+ strncpy(moldyn->vis.fb,filebase,128);
+
+ return 0;
+}
+
+int visual_atoms(t_moldyn *moldyn) {
+
+ int i,j,fd;
+ char file[128+64];
+ t_3dvec dim;
+ double help;
+ t_visual *v;
+ t_atom *atom;
+ t_atom *btom;
+ t_linkcell *lc;
+ t_list neighbour[27];
+ u8 bc;
+ t_3dvec dist;
+ double d2;
+ u8 brand;
+
+ v=&(moldyn->vis);
+ dim.x=v->dim.x;
+ dim.y=v->dim.y;
+ dim.z=v->dim.z;
+ atom=moldyn->atom;
+ lc=&(moldyn->lc);
+
+ help=(dim.x+dim.y);
+
+ sprintf(file,"%s/atomic_conf_%07.f.xyz",v->fb,moldyn->time);
+ fd=open(file,O_WRONLY|O_CREAT|O_TRUNC,S_IRUSR|S_IWUSR);
+ if(fd<0) {
+ perror("open visual save file fd");
+ return -1;
+ }
+
+ /* write the actual data file */
+
+ // povray header
+ dprintf(fd,"# [P] %d %07.f <%f,%f,%f>\n",
+ moldyn->count,moldyn->time,help/40.0,help/40.0,-0.8*help);
+
+ // atomic configuration
+ for(i=0;i<moldyn->count;i++) {
+ // atom type, positions, color and kinetic energy
+ dprintf(fd,"%s %f %f %f %s %f\n",pse_name[atom[i].element],
+ atom[i].r.x,
+ atom[i].r.y,
+ atom[i].r.z,
+ pse_col[atom[i].element],
+ atom[i].ekin);
+
+ /*
+ * bond detection should usually be done by potential
+ * functions. brrrrr! EVIL!
+ *
+ * todo: potentials need to export a 'find_bonds' function!
+ */
+
+ // bonds between atoms
+ if(!(atom[i].attr&ATOM_ATTR_VB))
+ continue;
+ link_cell_neighbour_index(moldyn,
+ (atom[i].r.x+moldyn->dim.x/2)/lc->x,
+ (atom[i].r.y+moldyn->dim.y/2)/lc->y,
+ (atom[i].r.z+moldyn->dim.z/2)/lc->z,
+ neighbour);
+ for(j=0;j<27;j++) {
+ list_reset_f(&neighbour[j]);
+ if(neighbour[j].start==NULL)
+ continue;
+ bc=j<lc->dnlc?0:1;
+ do {
+ btom=neighbour[j].current->data;
+ if(btom==&atom[i]) // skip identical atoms
+ continue;
+ //if(btom<&atom[i]) // skip half of them
+ // continue;
+ v3_sub(&dist,&(atom[i].r),&(btom->r));
+ if(bc) check_per_bound(moldyn,&dist);
+ d2=v3_absolute_square(&dist);
+ brand=atom[i].brand;
+ if(brand==btom->brand) {
+ if(d2>moldyn->bondlen[brand])
+ continue;
+ }
+ else {
+ if(d2>moldyn->bondlen[2])
+ continue;
+ }
+ dprintf(fd,"# [B] %f %f %f %f %f %f\n",
+ atom[i].r.x,atom[i].r.y,atom[i].r.z,
+ btom->r.x,btom->r.y,btom->r.z);
+ } while(list_next_f(&neighbour[j])!=L_NO_NEXT_ELEMENT);
+ }
+ }
+
+ // boundaries
+ if(dim.x) {
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,-dim.y/2,-dim.z/2,
+ dim.x/2,-dim.y/2,-dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,-dim.y/2,-dim.z/2,
+ -dim.x/2,dim.y/2,-dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ dim.x/2,dim.y/2,-dim.z/2,
+ dim.x/2,-dim.y/2,-dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,dim.y/2,-dim.z/2,
+ dim.x/2,dim.y/2,-dim.z/2);
+
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,-dim.y/2,dim.z/2,
+ dim.x/2,-dim.y/2,dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,-dim.y/2,dim.z/2,
+ -dim.x/2,dim.y/2,dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ dim.x/2,dim.y/2,dim.z/2,
+ dim.x/2,-dim.y/2,dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,dim.y/2,dim.z/2,
+ dim.x/2,dim.y/2,dim.z/2);
+
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,-dim.y/2,dim.z/2,
+ -dim.x/2,-dim.y/2,-dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ -dim.x/2,dim.y/2,dim.z/2,
+ -dim.x/2,dim.y/2,-dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ dim.x/2,-dim.y/2,dim.z/2,
+ dim.x/2,-dim.y/2,-dim.z/2);
+ dprintf(fd,"# [D] %f %f %f %f %f %f\n",
+ dim.x/2,dim.y/2,dim.z/2,
+ dim.x/2,dim.y/2,-dim.z/2);
+ }
+
+ close(fd);
+
+ return 0;
+}
+
projects / physik / posic.git / blobdiff