X-Git-Url: https://hackdaworld.org/gitweb/?a=blobdiff_plain;f=nlsop.c;h=5b160bc0bad5da00dad9e4b25817688c9f5a2e3b;hb=27d07bd7fc4b8166dfcfc6177101b778e6b1048c;hp=3dafd1bbc746a348f47235eb0bcbd13ae00fa120;hpb=f5c23d407ff5052877f38fa4f6ff4edf047b2a76;p=physik%2Fnlsop.git

diff --git a/nlsop.c b/nlsop.c
index 3dafd1b..5b160bc 100644
--- a/nlsop.c
+++ b/nlsop.c
@@ -55,6 +55,7 @@ int usage(void)
  printf("-D <value> \t diffusion rate from cryst to amorph cells (default %f)\n",DR_AC);
  printf("-c <value> \t diffusion rate in cryst cells (default %f)\n",DR_CC);
  printf("-e <value> \t do diffusion every <value> steps (default %d)\n",DIFF_RATE);
+ puts("-g <file> <step> \t continue simulation from file and step (step > 0)!");
  printf("-W <value> \t write every <value> steps to save file (default %d)\n",RESAVE);
  puts("-C <file> \t convert file to gnuplot format");
  puts("-L <file> \t load from file");
@@ -194,7 +195,7 @@ int distrib_c(d3_lattice *d3_l,info *my_info,int step,double c_ratio)
        carry=0;
        if(!(*(d3_l->status+off)&AMORPH))
        {
-        carry=(*(d3_l->extra+off)-*(d3_l->extra+offset))/2;
+        carry=(int)(my_info->dr_cc*(*(d3_l->extra+off)-*(d3_l->extra+offset))/2);
         if(carry!=0)
         {
          *(d3_l->extra+offset)+=carry;
@@ -211,7 +212,7 @@ int distrib_c(d3_lattice *d3_l,info *my_info,int step,double c_ratio)
        carry=0;
        if(!(*(d3_l->status+off)&AMORPH))
        {
-	carry=(*(d3_l->extra+off)-*(d3_l->extra+offset))/2;
+	carry=(int)(my_info->dr_cc*(*(d3_l->extra+off)-*(d3_l->extra+offset))/2);
 	if(carry!=0)
         {
          *(d3_l->extra+offset)+=carry;
@@ -233,7 +234,7 @@ int distrib_c(d3_lattice *d3_l,info *my_info,int step,double c_ratio)
 int calc_pressure(d3_lattice *d3_l,int range)
 {
  int i,j,off;
- double count;
+ double count,max=0;
  int x,y,z;
 
  for(x=0;x<d3_l->max_x;x++)
@@ -254,11 +255,209 @@ int calc_pressure(d3_lattice *d3_l,int range)
       }
      }
     }
-    *(unsigned char *)(d3_l->v_ptr+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y)=(unsigned char)(count*255/MAX_VPTR);
+    if(count>max) max=count;
    }
   }
  }
 
+ for(x=0;x<d3_l->max_x;x++)
+ {
+  for(y=0;y<d3_l->max_y;y++)
+  {
+   for(z=0;z<d3_l->max_z;z++)
+   {
+    count=0;
+    for(i=-range;i<=range;i++)
+    {
+     for(j=-range;j<=range;j++)
+     {
+      if(i!=0 && j!=0)
+      {
+       off=((x+d3_l->max_x+i)%d3_l->max_x)+((y+d3_l->max_y+j)%d3_l->max_x)*d3_l->max_x+z*d3_l->max_x*d3_l->max_y;
+       if(*(d3_l->status+off)&AMORPH) count+=((double)*(d3_l->extra+off))/(i*i+j*j);
+      }
+     }
+    }
+    *(unsigned char *)(d3_l->v_ptr+x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y)=(unsigned char)(count*255/max);
+   }
+  }
+ }
+
+ return 1;
+}
+
+int calc_max_extra(d3_lattice *d3_l)
+{
+ int x,y,z;
+ int off,max=0;
+
+ for(x=0;x<d3_l->max_x;x++)
+ {
+  for(y=0;y<d3_l->max_y;y++)
+  {
+   for(z=0;z<d3_l->max_z;z++)
+   {
+    off=x+y*d3_l->max_x+z*d3_l->max_x*d3_l->max_y;
+    if(*(d3_l->extra+off)>max) max=*(d3_l->extra+off);
+   }
+  }
+ }
+
+ return max;
+}
+
+int write_bmp(d3_lattice *d3_l,int window,u32 x,u32 y,u32 z)
+{
+ int fd,i,j,size=0,foo=0;
+ int width=0,height=0;
+ char bmpfile[MAX_CHARS];
+ char buf[128];
+
+ if(window==1)
+ {
+  sprintf(bmpfile,"x-z_%d.bmp",y);
+  foo=3*d3_l->max_x;
+  size=(foo+(4-foo%4))*d3_l->max_z;
+  width=d3_l->max_x;
+  height=d3_l->max_z;
+ }
+ if(window==2)
+ {
+  sprintf(bmpfile,"y-z_%d.bmp",x);
+  foo=3*d3_l->max_y;
+  size=(foo+(4-foo%4))*d3_l->max_z;
+  width=d3_l->max_y;
+  height=d3_l->max_z;
+ }
+ if(window==3)
+ {
+  sprintf(bmpfile,"x-y_%d.bmp",z);
+  foo=3*d3_l->max_x;
+  size=(foo+(4-foo%4))*d3_l->max_y;
+  width=d3_l->max_x;
+  height=d3_l->max_y;
+ }
+
+ if((fd=open(bmpfile,O_WRONLY|O_CREAT))<0)
+ {
+  puts("cannot open bmp file");
+  return -1;
+ }
+
+ /* bmpheader */
+ buf[0]='B'; /* std header start */
+ buf[1]='M';
+ buf[2]=(size+0x36)&0xff; /* file size */
+ buf[3]=(size+0x36)>>8;
+ memset(buf+4,0,6);
+ buf[10]=0x36; /* offset to data */
+ memset(buf+11,0,3);
+ buf[14]=0x28; /* length of bmp info header */
+ memset(buf+15,0,3);
+ buf[18]=width&0xff; /* width and height */
+ buf[19]=width>>8;
+ memset(buf+20,0,2);
+ buf[22]=height&0xff;
+ buf[23]=height>>8;
+ memset(buf+24,0,2);
+ buf[26]=1; /* # planes -> 1 */
+ buf[27]=0;
+ buf[28]=24; /* bits per pixel -> 2^24 (true color) */
+ buf[29]=0;
+ memset(buf+30,0,4); /* compression -> none */
+ buf[34]=size&0xff; /* data size */
+ buf[35]=size>>8;
+ memset(buf+36,0,2);
+ buf[38]=0x12; /* res: pixel/meter */
+ buf[39]=0x0b;
+ memset(buf+40,0,2);
+ buf[42]=0x12;
+ buf[43]=0x0b;
+ memset(buf+44,0,2); 
+ memset(buf+46,0,8); /* no colors, no important colors */
+
+ if(write(fd,buf,54)<54)
+ {
+  puts("failed writing bmp header");
+  return -1;
+ }
+ if(window==1)
+ {
+  for(j=0;j<d3_l->max_z;j++)
+  {
+   for(i=0;i<d3_l->max_x;i++)
+   {
+    if(*(d3_l->status+i+y*d3_l->max_x+(d3_l->max_z-j-1)*d3_l->max_x*d3_l->max_y)&RED) memset(buf,0xff,3);
+    else memset(buf,0,3);
+    if(write(fd,buf,3)<3)
+    {
+     puts("failed writing rgb values to bmp file");
+     return-1;
+    }
+   }
+   if(foo%4)
+   {
+    memset(buf,0,4-foo%4);
+    if(write(fd,buf,4-foo%4)<4-foo%4)
+    {
+     puts("failed writing 4 byte ending");
+     return -1;
+    }
+   } 
+  }
+ }
+ if(window==2)
+ {
+  for(j=0;j<d3_l->max_z;j++)
+  {
+   for(i=0;i<d3_l->max_y;i++)
+   {
+    if(*(d3_l->status+x+i*d3_l->max_x+(d3_l->max_z-j-1)*d3_l->max_x*d3_l->max_y)&RED) memset(buf,0xff,3);
+    else memset(buf,0,3);
+    if(write(fd,buf,3)<3)
+    {
+     puts("failed writing rgb values to bmp file");
+     return-1;
+    }
+   }
+   if(foo%4)
+   {
+    memset(buf,0,4-foo%4);
+    if(write(fd,buf,4-foo%4)<4-foo%4)
+    {
+     puts("failed writing 4 byte ending");
+     return -1;
+    }
+   }
+  }
+ }
+ if(window==3)
+ {
+  for(j=0;j<d3_l->max_y;j++)
+  {
+   for(i=0;i<d3_l->max_x;i++)
+   {
+    if(*(d3_l->status+i+(d3_l->max_y-j-1)*d3_l->max_x+z*d3_l->max_x*d3_l->max_y)&RED) memset(buf,0xff,3);
+    else memset(buf,0,3);
+    if(write(fd,buf,3)<3)
+    {
+     puts("failed writing rgb values to bmp file");
+     return -1;
+    }
+   }
+   if(foo%4)
+   {
+    memset(buf,0,4-foo%4);
+    if(write(fd,buf,4-foo%4)<4-foo%4)
+    {
+     puts("failed writing 4 byte ending");
+     return -1;
+    }
+   }
+  }
+ }
+ close(fd);
+
  return 1;
 }
 
@@ -404,6 +603,7 @@ int get_c_ratio(double *c_ratio,char *pfile,info *my_info,d3_lattice *d3_l)
   }
  }
  *c_ratio=d/all;
+ close(p_fd);
 
  return 1;
 }
@@ -411,8 +611,9 @@ int get_c_ratio(double *c_ratio,char *pfile,info *my_info,d3_lattice *d3_l)
 int main(int argc,char **argv)
 {
  u32 x,y,z,x_c,y_c,z_c;
- int i,quit,escape,switchmode,nowait;
+ int i,quit,escape,switchmode,nowait,bmp;
  int refresh,resave;
+ int c_step;
  char s_file[MAX_CHARS];
  char s_file_tmp[MAX_CHARS];
  char l_file[MAX_CHARS];
@@ -437,6 +638,7 @@ int main(int argc,char **argv)
  char diff_txt[MAX_TXT];
  char dr_ac_txt[MAX_TXT];
  char dr_cc_txt[MAX_TXT];
+ char dose_txt[MAX_TXT];
  char mode_txt[MAX_TXT];
  char *arg_v[MAX_ARGV];
 #endif
@@ -444,6 +646,9 @@ int main(int argc,char **argv)
  info my_info;
  unsigned char mode;
  double c_ratio;
+#ifdef USE_DFB_API
+ int max_extra;
+#endif
 
  d3_l.max_x=X;
  d3_l.max_y=Y;
@@ -468,6 +673,7 @@ int main(int argc,char **argv)
  quit=0;
  escape=0;
  switchmode=0;
+ c_step=0;
  strcpy(s_file,"");
  strcpy(l_file,"");
  strcpy(c_file,"");
@@ -562,6 +768,10 @@ int main(int argc,char **argv)
     case 'P':
      strcpy(p_file,argv[++i]);
      break;
+    case 'g':
+     strcpy(l_file,argv[++i]);
+     if(i<argc-1) if(argv[i+1][0]!='-') c_step=atoi(argv[++i]);
+     break;
     default:
      usage();
      return -1;
@@ -614,7 +824,7 @@ int main(int argc,char **argv)
    return 1;
   } 
 #ifdef USE_DFB_API
-    else d3_lattice_init(&argc,argv,&d3_l);
+  else d3_lattice_init(&argc,argv,&d3_l);
 #endif
  }
 
@@ -625,10 +835,11 @@ int main(int argc,char **argv)
 #ifdef USE_DFB_API
  strcpy(a_txt,"args:");
  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: %.2f  b_ap: %.3f",my_info.a_ap,my_info.b_ap);
- sprintf(el_txt,"a_el: %.2f  b_el: %.3f",my_info.a_el,my_info.b_el);
- sprintf(cd_txt,"a_cd: %.2f  b_cd: %.3f",my_info.a_cd,my_info.b_cd);
+ sprintf(ap_txt,"a_ap: %.3f  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(dr_ac_txt,"a/c diffusion rate: %.4f",my_info.dr_ac);
  sprintf(dr_cc_txt,"c/c diffusion rate: %.4f",my_info.dr_cc);
@@ -649,10 +860,10 @@ int main(int argc,char **argv)
  arg_v[12]=zdiff_txt;
  arg_v[13]=NULL;
  arg_v[14]=a_txt;
- arg_v[15]=NULL;
- arg_v[16]=s_txt;
- arg_v[17]=r_txt;
- arg_v[18]=NULL;
+ arg_v[15]=s_txt;
+ arg_v[16]=dose_txt;
+ arg_v[17]=NULL;
+ arg_v[18]=r_txt;
  arg_v[19]=ap_txt;
  arg_v[20]=el_txt;
  arg_v[21]=cd_txt;
@@ -662,14 +873,14 @@ int main(int argc,char **argv)
  arg_v[25]=dr_cc_txt;
 #endif
 
- if(!strcmp(l_file,""))
+ if((!strcmp(l_file,""))||(c_step))
  {
   if(get_c_ratio(&c_ratio,p_file,&my_info,&d3_l)!=1)
   {
    puts("failed calculating ratio");
    return -1;
   }
-  i=0;
+  i=(c_step?c_step:0);
   while((i<my_info.steps) && (quit==0) && (escape==0))
   {
    x_c=get_rand(d3_l.max_x);
@@ -685,7 +896,7 @@ int main(int argc,char **argv)
     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  a_count: %d",i,amorph_count);
     sprintf(cc_txt,"total c: %d  c_count: %d",my_info.cc,cryst_count);
-    d3_lattice_draw(&d3_l,x,y,z,25,arg_v,mode);
+    d3_lattice_draw(&d3_l,x,y,z,25,arg_v,mode,0);
    }
 #endif
    if(i%resave==0 && strcmp(s_file,"") && resave!=0 && i!=0)
@@ -715,6 +926,7 @@ int main(int argc,char **argv)
  }
  /* calc values */
  calc_pressure(&d3_l,my_info.range);
+ max_extra=calc_max_extra(&d3_l);
 
  while((quit==0) && (escape==0) && (nowait==0))
  {
@@ -726,13 +938,16 @@ int main(int argc,char **argv)
   sprintf(xyz_txt,"x: %d  y: %d  z: %d",x+1,y+1,z+1);
   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: end!  a_count: %d",amorph_count);
-  sprintf(cc_txt,"total c: %d  c_count: %d",my_info.cc,cryst_count);
+  sprintf(steps_txt,"step: end");
+  sprintf(cc_txt,"total c: %d",my_info.cc);
   if(switchmode==0) strcpy(mode_txt,"view: a/c mode");
   if(switchmode==1) strcpy(mode_txt,"view: c conc mode");
   if(switchmode==2) strcpy(mode_txt,"view: a pressure mode");
-  d3_lattice_draw(&d3_l,x,y,z,25,arg_v,mode);
-  scan_event(&d3_l,&x,&y,&z,&quit,&escape,&switchmode);
+  d3_lattice_draw(&d3_l,x,y,z,25,arg_v,mode,max_extra);
+  bmp=0;
+  scan_event(&d3_l,&x,&y,&z,&quit,&escape,&switchmode,&bmp);
+  if(bmp) write_bmp(&d3_l,bmp,x,y,z);
+
  }
 
  d3_lattice_release(&d3_l);