X-Git-Url: https://hackdaworld.org/gitweb/?p=physik%2Fnlsop.git;a=blobdiff_plain;f=nlsop.c;h=041fac0151d440c8a0c48aaa7b12a59033796797;hp=761be356356bbedd2b5bf17516b5b092136399de;hb=c71815b4b5dc67546a9ca407c256d18a4abb3aff;hpb=6f26989bb9e70a48d947ada3af4b3defce65ed27

diff --git a/nlsop.c b/nlsop.c
index 761be35..041fac0 100644
--- a/nlsop.c
+++ b/nlsop.c
@@ -33,8 +33,8 @@ int usage(void)
  puts("usage:");
  puts("-h \t\t help");
  puts("-n \t\t no user interaction");
- printf("-a <value> \t slope of nuclear energy loss (default %d)\n",A_EL);
- printf("-b <value> \t nuclear energy loss offset (default %d)\n",B_EL);
+ printf("-a <value> \t slope of nuclear energy loss (default %f)\n",A_EL);
+ printf("-b <value> \t nuclear energy loss offset (default %f)\n",B_EL);
  printf("-x <value> \t # x cells (default %d)\n",X);
  printf("-y <value> \t # x cells (default %d)\n",Y);
  printf("-z <value> \t # x cells (default %d)\n",Z);
@@ -56,15 +56,15 @@ int usage(void)
  return 1;
 }
 
-int process_cell(3d_lattice *3d_l,u32 x,u32 y,u32 z,int r,double a,double b,int *t_c)
+int process_cell(d3_lattice *d3_l,u32 x,u32 y,u32 z,int r,double a,double b,int *t_c)
 {
  unsigned char *thiz;
  int *conc;
  int i,j;
  double p;
 
- thiz=3d_l->status+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y;
- conc=3d_l->extra+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y;
+ thiz=d3_l->status+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y;
+ conc=d3_l->extra+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y;
  p=b*URAND_MAX;
  for(i=-r;i<=r;i++)
  {
@@ -79,7 +79,7 @@ int process_cell(3d_lattice *3d_l,u32 x,u32 y,u32 z,int r,double a,double b,int
  p*=*conc;
  if(!(*thiz&AMORPH))
  {
-  if(rand_get(URAND_MAX)<=p)
+  if(get_rand(URAND_MAX)<=p)
   {
    MAKE_AMORPH(thiz);
    *t_c=*t_c+1-*conc;
@@ -87,7 +87,7 @@ int process_cell(3d_lattice *3d_l,u32 x,u32 y,u32 z,int r,double a,double b,int
  } else
  {
   /* assume 1-p probability */
-  if(rand_get(URAND_MAX)>p)
+  if(get_rand(URAND_MAX)>p)
   {
    MAKE_CRYST(thiz);
    *t_c=*t_c+1+*conc;
@@ -97,13 +97,345 @@ int process_cell(3d_lattice *3d_l,u32 x,u32 y,u32 z,int r,double a,double b,int
  return 1;
 }
 
-int distrib_c(3d_lattice *3d_l,int t_c,double a,double b)
+int distrib_c(d3_lattice *d3_l,int t_c,double a,double b)
 {
- int i,j,total,area;
+ int i,j,k,total,area,sum;
+ int temp,left;
+ int *area_h;
+ u32 x,y,z;
 
+ area=d3_l->max_x*d3_l->max_y;
+ area_h=(int *)malloc(d3_l->max_z*sizeof(int));
+
+ total=0;
+ sum=0;
  for(i=0;i<d3_l->max_z;i++)
  {
- 
+  area_h[i]=0;
+  for(j=0;j<area;j++)
+  {
+   if(!(*(d3_l->status+(i*area)+j)&AMORPH))
+   {
+    sum+=(i+1)*a+b;
+    area_h[i]+=1;
+   }
+  }
+ }
+ for(i=0;i<d3_l->max_z;i++)
+ {
+  temp=((i+1)*a+b)*t_c/(sum+area_h[i]);
+  if(temp>1)
+  {
+   for(j=0;j<area;j++)
+   {
+    if(!(*(d3_l->status+(i*area)+j)&AMORPH))
+    {
+     *(d3_l->extra+(i*area)+j)=temp;
+     total+=temp;
+    }
+   }
+  }
+  left=(int)(((i+1)*a+b)*t_c/sum)%area_h[i];
+  while(left)
+  {
+   x=get_rand(d3_l->max_x);
+   y=get_rand(d3_l->max_y);
+   if(!(*(d3_l->status+(i*area)+x+y*d3_l->max_x)&AMORPH))
+   {
+    *(d3_l->extra+(i*area)+x+y*d3_l->max_x)+=1;
+    total+=1;
+    left-=1;
+   }
+  }
+ }
+ left=t_c-total;
+ while(left)
+ {
+  x=get_rand(d3_l->max_x);
+  y=get_rand(d3_l->max_y);
+  z=get_rand_lgp(d3_l->max_z,a,b);
+  if(!(*(d3_l->status+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y)&AMORPH))
+  {
+   *(d3_l->extra+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y)+=1;
+   left-=1;
+   total+=1;
+  }
+ }
+
+ return 1;
+}
+
+int save_to_file(char *sf,d3_lattice *d3_l)
+{
+ int sf_fd,c;
+
+ if((sf_fd=open(sf,O_WRONLY|O_CREAT))<0)
+ {
+  puts("cannot open save file");
+  return -1;
+ }
+ if(write(sf_fd,d3_l,sizeof(d3_lattice))<sizeof(d3_lattice))
+ {
+  puts("failed saving d3 lattice struct");
+  return -1;
+ }
+ c=d3_l->max_x*d3_l->max_y*d3_l->max_z;
+ if(write(sf_fd,d3_l->status,c*sizeof(unsigned char))<c*sizeof(unsigned char))
+ {
+  puts("failed saving status of d3 lattice sites");
+  return -1;
+ }
+ if(write(sf_fd,d3_l->extra,c*sizeof(int))<c*sizeof(int))
+ {
+  puts("failed saving sites concentration");
+  return -1;
+ }
+ close(sf_fd);
+
+ return 1;
+}
+
+int load_from_file(char *lf,d3_lattice *d3_l)
+{
+ int lf_fd,c;
+
+ if((lf_fd=open(lf,O_RDONLY))<0)
+ {
+  puts("cannot open load file");
+  return -1;
+ }
+ if(read(lf_fd,d3_l,sizeof(d3_lattice))<sizeof(d3_lattice))
+ {
+  puts("failed reading d3 lattice struct");
+  return -1;
+ }
+ c=d3_l->max_x*d3_l->max_y*d3_l->max_z;
+ if((d3_l->status=(unsigned char*)malloc(c*sizeof(unsigned char)))==NULL)
+ {
+  puts("cannot allocate status buffer");
+  return -1;
+ }
+ if((d3_l->extra=(int *)malloc(c*sizeof(int)))==NULL)
+ {
+  puts("cannot allocate concentration buffer");
+  return -1;
+ }
+ if(read(lf_fd,d3_l->status,c*sizeof(unsigned char))<c*sizeof(unsigned char))
+ {
+  puts("failed reading status of d3 lattice sites");
+  return -1;
+ }
+ if(read(lf_fd,d3_l->extra,c*sizeof(int))<c*sizeof(int))
+ {
+  puts("failed reading sites concentration");
+  return -1;
+ }
+ close(lf_fd);
+
+ return 1;
+}
+
+int main(int argc,char **argv)
+{
+ u32 max_x,max_y,max_z,x,y,z,x_c,y_c,z_c;
+ int i,quit,escape,nowait;
+ double a_el,b_el,a_cd,b_cd,a_ap,b_ap;
+ int cc,steps,range,refresh;
+ char s_file[MAX_CHARS];
+ char l_file[MAX_CHARS];
+ char r_file[MAX_CHARS];
+ char x_txt[MAX_TXT];
+ char y_txt[MAX_TXT];
+ char z_txt[MAX_TXT];
+ char status_txt[MAX_TXT];
+ char conc_txt[MAX_TXT];
+ char steps_txt[MAX_TXT];
+ char cc_txt[MAX_TXT];
+ char **arg_v;
+ d3_lattice d3_l;
+
+ max_x=X;
+ max_y=Y;
+ max_z=Z;
+ x=X/2-1;
+ y=Y/2-1;
+ z=Z/2-1;
+ steps=STEPS;
+ range=RANGE;
+ refresh=REFRESH;
+ a_el=A_EL;
+ b_el=B_EL;
+ a_cd=A_CD;
+ b_cd=B_CD;
+ a_ap=A_AP;
+ b_ap=B_AP;
+ cc=CC;
+ nowait=0;
+ strcpy(s_file,"");
+ strcpy(l_file,"");
+ strcpy(r_file,"");
+
+ for(i=1;i<argc;i++)
+ {
+  if(argv[i][0]=='-')
+  {
+   switch(argv[i][1])
+   {
+    case 'h':
+     usage();
+     return -1;
+    case 'n':
+     nowait=1;
+     break;
+    case 'a':
+     a_el=atof(argv[++i]);
+     break;
+    case 'b':
+     b_el=atof(argv[++i]);
+     break;
+    case 'x':
+     max_x=atoi(argv[++i]);
+     break;
+    case 'y':
+     max_y=atoi(argv[++i]);
+     break;
+    case 'z':
+     max_z=atoi(argv[++i]);
+     break;
+    case 'X':
+     x=atoi(argv[++i]);
+     break;
+    case 'Y':
+     y=atoi(argv[++i]);
+     break;
+    case 'Z':
+     z=atoi(argv[++i]);
+     break;
+    case 's':
+     steps=atoi(argv[++i]);
+     break;
+    case 'd':
+     refresh=atoi(argv[++i]);
+     break;
+    case 'r':
+     range=atoi(argv[++i]);
+     break;
+    case 'f':
+     a_ap=atof(argv[++i]);
+     break;
+    case 'p':
+     b_ap=atof(argv[++i]);
+     break;
+    case 'A':
+     a_cd=atof(argv[++i]);
+     break;
+    case 'B':
+     b_cd=atof(argv[++i]);
+     break;
+    case 'C':
+     cc=atoi(argv[++i]);
+     break;
+    case 'L':
+     strcpy(l_file,argv[++i]);
+     break;
+    case 'S':
+     strcpy(s_file,argv[++i]);
+     break;
+    case 'R':
+     strcpy(r_file,argv[++i]);
+     break;
+    default:
+     usage();
+     return -1;
+   }
+  } else usage();
+ }
+
+ if(!strcmp(r_file,"")) rand_init(NULL);
+ else rand_init(r_file);
+
+ if(!strcmp(l_file,""))
+ {
+  i=max_x*max_y*max_z;
+  d3_lattice_init(&argc,argv,&d3_l,max_x,max_y,max_z);
+  if((d3_l.status=(unsigned char *)malloc(i*sizeof(unsigned char)))==NULL)
+  {
+   puts("failed allocating status buffer");
+   return -1;
+  }
+  memset(d3_l.status,0,i*sizeof(unsigned char));
+  if((d3_l.extra=(int *)malloc(i*sizeof(int)))==NULL)
+  {
+   puts("failed allocating concentration buffer");
+   return -1;
+  }
+  memset(d3_l.extra,0,i*sizeof(int));
+ } else
+ {
+  load_from_file(l_file,&d3_l);
+  d3_lattice_init(&argc,argv,&d3_l,d3_l.max_x,d3_l.max_y,d3_l.max_z);
+ }
+
+ d3_event_init(&d3_l);
+
+ if(!strcmp(l_file,""))
+ {
+  while((i<steps) && (quit==0) && (escape==0))
+  {
+   x_c=get_rand(d3_l.max_x);
+   y_c=get_rand(d3_l.max_y);
+   z_c=get_rand_lgp(d3_l.max_z,a_el,b_el);
+   distrib_c(&d3_l,cc,a_cd,b_cd);
+   process_cell(&d3_l,x_c,y_c,z_c,range,a_ap,b_ap,&cc);
+   if(i%refresh==0)
+   {
+    sprintf(x_txt,"x: %d",x);
+    sprintf(y_txt,"y: %d",y);
+    sprintf(z_txt,"z: %d",z);
+    sprintf(status_txt,"status: %c",(*(d3_l.status+x+y*d3_l.max_x+z*d3_l.max_x*d3_l.max_y)&AMORPH)?'a':'c');
+    sprintf(conc_txt,"conc: %d",*(d3_l.extra+x+y*d3_l.max_x+z*d3_l.max_x*d3_l.max_y));
+    sprintf(steps_txt,"step: %d",i);
+    sprintf(cc_txt,"total c: %d",cc);
+    arg_v[1]=x_txt;
+    arg_v[2]=y_txt;
+    arg_v[3]=z_txt;
+    arg_v[4]=NULL;
+    arg_v[5]=status_txt;
+    arg_v[6]=conc_txt;
+    arg_v[7]=NULL;
+    arg_v[8]=NULL;
+    arg_v[9]=steps_txt;
+    arg_v[10]=cc_txt;
+    d3_lattice_draw(&d3_l,x,y,z,10,arg_v);
+    scan_event(&d3_l,&x,&y,&z,&quit,&escape);
+   }
+   i++;
+  }
+ }
+
+ if(strcmp(s_file,"")) save_to_file(s_file,&d3_l);
+
+ while((quit==0) && (escape==0) && (nowait==0))
+ {
+  sprintf(x_txt,"x: %d",x);
+  sprintf(y_txt,"y: %d",y);
+  sprintf(z_txt,"z: %d",z);
+  sprintf(status_txt,"status: %c",(*(d3_l.status+x+y*d3_l.max_x+z*d3_l.max_x*d3_l.max_y)&AMORPH)?'a':'c');
+  sprintf(conc_txt,"conc: %d",*(d3_l.extra+x+y*d3_l.max_x+z*d3_l.max_x*d3_l.max_y));
+  strcpy(steps_txt,"step: end!");
+  sprintf(cc_txt,"total c: %d",cc);
+  arg_v[1]=x_txt;
+  arg_v[2]=y_txt;
+  arg_v[3]=z_txt;
+  arg_v[4]=NULL;
+  arg_v[5]=status_txt;
+  arg_v[6]=conc_txt;
+  arg_v[7]=NULL;
+  arg_v[8]=NULL;
+  arg_v[9]=steps_txt;
+  arg_v[10]=cc_txt;
+  d3_lattice_draw(&d3_l,x,y,z,10,arg_v);
+  scan_event(&d3_l,&x,&y,&z,&quit,&escape);
  }
 
  return 1;