finished random reject method

[physik/nlsop.git] / nlsop.c

diff --git a/nlsop.c b/nlsop.c
index 0104695..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;
@@ -131,8 +131,9 @@ 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);
   // printf("nd: %d %d\n",x,y);
-  z=get_rand_lgp(d3_l->max_z,my_info->a_cd,my_info->b_cd);
-  // printf("ld: %d\n",z);
+  // 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)++;
  }
@@ -694,17 +695,26 @@ int get_c_ratio(double *c_ratio,char *pfile,info *my_info,d3_lattice *d3_l)
  return 1;
 }
 
-int get_reject_graph(info *my_info,d3_lattice *d3_l,char *file,u32 *graph) {
+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;
@@ -722,12 +732,32 @@ int get_reject_graph(info *my_info,d3_lattice *d3_l,char *file,u32 *graph) {
    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);
+   else 
+   {
+    graph[(int)(a/CELL_LENGTH)]=(int)(URAND_MAX*b);
+    flag[(int)(a/CELL_LENGTH)]=1;
+   }
   }
  }
- /* do interpolation here! */
+ /* 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];
+ }
 
- return 1;
+ 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)
@@ -771,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;
@@ -805,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++)
  {
@@ -1004,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))
   {
@@ -1017,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)
    {