implemented bmp writing (doesntwork by now)

[physik/nlsop.git] / nlsop.c

diff --git a/nlsop.c b/nlsop.c
index 26a6e61..ff89e7c 100644 (file)
--- a/nlsop.c
+++ b/nlsop.c
@@ -305,6 +305,131 @@ int calc_max_extra(d3_lattice *d3_l)
  return max;
 }
 
+int write_bmp(d3_lattice *d3_l,int window,u32 x,u32 y,u32 z)
+{
+ int fd,i,j,size=0;
+ int width=0,height=0;
+ char bmpfile[MAX_CHARS];
+ char buf[128];
+
+ if(window==1)
+ {
+  sprintf(bmpfile,"x-z_%d.bmp",y);
+  size=x*z;
+  width=x;
+  height=z;
+ }
+ if(window==2)
+ {
+  sprintf(bmpfile,"y-z_%d.bmp",x);
+  size=y*z;
+  width=y;
+  height=z;
+ }
+ if(window==3)
+ {
+  sprintf(bmpfile,"x-y_%d.bmp",z);
+  size=x*y;
+  width=x;
+  height=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(i=0;i<d3_l->max_x;i++)
+  {
+   for(j=0;j<d3_l->max_z;j++)
+   {
+    if(*(d3_l->status+i+y*d3_l->max_x+j*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(window==2)
+ {
+  for(i=0;i<d3_l->max_y;i++)
+  {
+   for(j=0;j<d3_l->max_z;j++)
+   {
+    if(*(d3_l->status+x+i*d3_l->max_x+j*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(window==3)
+ {
+  for(i=0;i<d3_l->max_x;i++)
+  {
+   for(j=0;j<d3_l->max_y;j++)
+   {
+    if(*(d3_l->status+i+j*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;
+    }
+   }
+  }
+ }
+ close(fd);
+
+ return 1;
+}
+
 int save_to_file(char *sf,d3_lattice *d3_l,info *my_inf)
 {
  int sf_fd,c;
@@ -455,7 +580,7 @@ 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;
  char s_file[MAX_CHARS];
  char s_file_tmp[MAX_CHARS];
@@ -782,7 +907,10 @@ int main(int argc,char **argv)
   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,max_extra);
-  scan_event(&d3_l,&x,&y,&z,&quit,&escape,&switchmode);
+  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);