finished random reject method

[physik/nlsop.git] / nlsop.c

diff --git a/nlsop.c b/nlsop.c
index 5d44239..11f81f5 100644 (file)
--- a/nlsop.c
+++ b/nlsop.c
@@ -118,7 +118,7 @@ int process_cell(d3_lattice *d3_l,u32 x,u32 y,u32 z,info *my_info)
  return 1;
 }
 
-int distrib_c(d3_lattice *d3_l,info *my_info,int step,double c_ratio)
+int distrib_c(d3_lattice *d3_l,info *my_info,int step,double c_ratio,u32 rj_m,u32 *rj_g)
 {
  u32 x,y,z;
  int i,j,k,c;
@@ -130,7 +130,10 @@ int distrib_c(d3_lattice *d3_l,info *my_info,int step,double c_ratio)
  {
   x=get_rand(d3_l->max_x);
   y=get_rand(d3_l->max_y);
-  z=get_rand_lgp(d3_l->max_z,my_info->a_cd,my_info->b_cd);
+  // printf("nd: %d %d\n",x,y);
+  // z=get_rand_lgp(d3_l->max_z,my_info->a_cd,my_info->b_cd);
+  z=get_rand_reject(d3_l->max_z,rj_m,rj_g);
+  // printf("%d\n",z);
   *(d3_l->extra+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y)+=1;
   (my_info->cc)++;
  }
@@ -692,6 +695,71 @@ int get_c_ratio(double *c_ratio,char *pfile,info *my_info,d3_lattice *d3_l)
  return 1;
 }
 
+u32 get_reject_graph(info *my_info,d3_lattice *d3_l,char *file,u32 *graph) {
+ double a,b;
+ int i,j,k;
+ int fd;
+ char buf[32],*p;
+ unsigned char *flag;
+ u32 max;
+
+ max=0;
+ if((fd=open(file,O_RDONLY))<0)
+ {
+  puts("cannot open file to calculate rejection graph");
+  return -1;
+ }
+ if((flag=(unsigned char *)malloc(d3_l->max_z))==NULL)
+ {
+  puts("cannot malloc flag memory for rejection graph");
+  return -1;
+ }
+ memset(flag,0,d3_l->max_z);
+ memset(graph,0,d3_l->max_z*sizeof(u32));
+ /* get fixpoints */
+ k=1;
+ while(k)
+ {
+  for(i=0;i<32;i++)
+  {
+   k=read(fd,&buf[i],1);
+   if((buf[i]=='\n')||(k==0)) break;
+  }
+  if(k)
+  {
+   p=strtok(buf," ");
+   a=atof(p)/10; /* nm */
+   p=strtok(NULL," ");
+   b=atof(p);
+   if(a>d3_l->max_z*CELL_LENGTH) k=0;
+   else 
+   {
+    graph[(int)(a/CELL_LENGTH)]=(int)(URAND_MAX*b);
+    flag[(int)(a/CELL_LENGTH)]=1;
+   }
+  }
+ }
+ /* do (linear) interpolation here! */
+ i=0;
+ a=0;
+ while(i<d3_l->max_z)
+ {
+  /* graph[0] is 0! */
+  j=i;
+  i++;
+  while(flag[i]==0&&i<d3_l->max_z) i++;
+  for(k=j+1;k<i;k++) graph[k]=(int)((k-j)*((int)graph[i]-(int)graph[j])/(i-j))+graph[j];
+  if(graph[i]>max) max=graph[i];
+ }
+
+ free(flag);
+
+ // printf("debug: (interpolated c profile)\n");
+ // for(i=0;i<d3_l->max_z;i++) printf("%d %d\n",i,graph[i]);
+
+ return max;
+}
+
 int main(int argc,char **argv)
 {
  u32 x,y,z,x_c,y_c,z_c;
@@ -733,6 +801,9 @@ int main(int argc,char **argv)
 #ifdef USE_DFB_API
  int max_extra;
 #endif
+ u32 *c_profile;
+ u32 *n_e_loss;
+ u32 ne_max,ip_max;
 
  d3_l.max_x=X;
  d3_l.max_y=Y;
@@ -767,6 +838,8 @@ int main(int argc,char **argv)
  convert=0;
  strcpy(r_file,"");
  mode=0;
+ ne_max=0;
+ ip_max=0;
 
  for(i=1;i<argc;i++)
  {
@@ -927,10 +1000,10 @@ int main(int argc,char **argv)
  sprintf(s_txt,"steps: %d",my_info.steps);
  sprintf(dose_txt,"dose: %.2fe+17 C/cm²",my_info.steps*1.0/(d3_l.max_x*d3_l.max_y*CELL_LENGTH*CELL_LENGTH*1000));
  sprintf(r_txt,"pressure range: %d",my_info.range);
- sprintf(ap_txt,"a_ap: %.3f  b_ap: %.3f",my_info.a_ap,my_info.b_ap);
+ sprintf(ap_txt,"a_ap: %.4f  b_ap: %.3f",my_info.a_ap,my_info.b_ap);
  sprintf(el_txt,"a_el: %.3f  b_el: %.3f",my_info.a_el,my_info.b_el);
  sprintf(cd_txt,"a_cd: %.3f  b_cd: %.3f",my_info.a_cd,my_info.b_cd);
- sprintf(cp_txt,"a_cp: %.4f",my_info.a_cp);
+ sprintf(cp_txt,"a_cp: %.5f",my_info.a_cp);
  sprintf(dr_ac_txt,"a/c diffusion rate: %.4f",my_info.dr_ac);
  if(my_info.c_diff!=0) sprintf(dr_cc_txt,"c/c diffusion rate: %.4f",my_info.dr_cc);
  else sprintf(dr_cc_txt,"c/c diffusion rate: none");
@@ -966,11 +1039,32 @@ int main(int argc,char **argv)
 
  if((!strcmp(l_file,""))||(c_step))
  {
+  /* calculate ratio of c_simwindow / c_total */
   if(get_c_ratio(&c_ratio,p_file,&my_info,&d3_l)!=1)
   {
    puts("failed calculating ratio");
    return -1;
   }
+  /* compute graphs for random number rejection method */
+  if((c_profile=(u32 *)malloc(d3_l.max_z*sizeof(unsigned int)))==NULL)
+  {
+   puts("failed allocating memory for carbon profile graph");
+   return -1;
+  }
+  if((n_e_loss=(u32 *)malloc(d3_l.max_z*sizeof(unsigned int)))==NULL)
+  {
+   puts("failed allocating memory for nuclear energy loss graph");
+   return -1;
+  }
+  ip_max=get_reject_graph(&my_info,&d3_l,p_file,c_profile);
+
+#ifdef DEBUG_RAND
+ while(1)
+ {
+  printf("%d\n",get_rand_reject(d3_l.max_z,ip_max,c_profile));
+ }
+#endif
+
   i=(c_step?c_step:0);
   while((i<my_info.steps) && (quit==0) && (escape==0))
   {
@@ -979,9 +1073,10 @@ int main(int argc,char **argv)
     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,my_info.a_el,my_info.b_el);
+    printf("%d\n",z_c);
     process_cell(&d3_l,x_c,y_c,z_c,&my_info);
    }
-   distrib_c(&d3_l,&my_info,i,c_ratio);
+   distrib_c(&d3_l,&my_info,i,c_ratio,ip_max,c_profile);
 #ifdef USE_DFB_API
    if(i%refresh==0)
    {