darwinview/src/darwin.c

#include <GL/glut.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <math.h>
#define WINDOW "image"
#define UP 101
#define DOWN 103
#define RIGHT 102
#define LEFT 100
#define MAX 700 
#define SCALE .06
 
int NX, NY, NZ;

void do_byteswap_f32( float arr[], int nel ), global(), local( int, int, int );
void readnames( char[] ), readarray( float[], char[], int ), readjet(), readxz( float[], char[] );
void readyz( float[], char[] ), readdepths( char[] );
void TimerFunction( int ), stroke( char[], int, int );

float data[MAX][MAX*MAX], mxval, mnval, jet[64][3];
float globalmx=0, globalmn=100;
int win[MAX], ilev=1, howmany, sets, count=0, glo=0, usr=0, anim=0, endian=0;
int xmax, ymax, yoffset=0, xoffset=0, yz=0, logscale=0, xz=0, nonegs=1;
int depths[MAX], scaledepth=0, totaldepth=0, scalecount=0;
char initfns[MAX][MAX], fns[MAX][MAX][MAX];

void menu( int value ){           // called when menu is opened on right click

 switch( value ){
  case 1: usr=glo=0;            // unset glo & usr, sets local max/min
          break;
  case 2: glo=1;                // enables global max/min  
          usr=0;                // unsets usr 
          mxval=globalmx;       // sets max to globalmx
          mnval=globalmn;       // sets min to globalmn
          break;
  case 3: usr=1;                // switch to user-set max/min 
          glo=0;                // unset glo
          printf( "Max=" );  scanf( "%f", &mxval );  // prompt user for new max
          printf( "Min=" );  scanf( "%f", &mnval );  // prompt user for new min
          break;
  case 4: logscale=( logscale+1 )%2;  // switch log scale on/off
          break;
  case 5: nonegs=( nonegs+1 )%2;      // switch allowance of negatives on/off 
          break;
  case 6: endian=( endian+1 )%2;      // switch between big/little endian
          break;
 }
 glutPostRedisplay();
}

void display(){                 // called on glutPostRedisplay
 int i, j, h, ioff, q; 
 float r, g, b, k, y;
 double num, logmx, logmn;
 char str[MAX]; 


 for( q=0; q<sets; q++ ){           // runs display func for each subwindow
  glutSetWindow( win[q] );          // sets which subwindow to display to
  glClear( GL_COLOR_BUFFER_BIT );   // background to black

  if( xz ) {  xmax=NX;  ymax=NZ;  }
  else{
   if( yz ){  xmax=NY;  ymax=NZ;  } 
   else    {  xmax=NX;  ymax=NY;  } 
  }

  if( glo || usr ){                  // if global or user-set max/min 
   if( xz )
    readxz( data[q], fns[q][count] );
   else{
    if( yz )
     readyz( data[q], fns[q][count] );
    else
     readarray( data[q], fns[q][count], ilev );  // read new array w/o calculating local max/min
   }
  }
  else                              // if local max/min is set 
   local( q, count, ilev );         // read new array and calculate local max/min

  if( logscale ){
   if( usr ){
    logmx=(double)mxval;
    logmn=(double)mnval;
   }
   else{
    logmx=log10( mxval );
    logmn=log10( mnval );
   }
  }
 
  if( scaledepth && xz || scaledepth && yz )  ymax=totaldepth;
  ioff=0; h=0;                        // ioff will count both i&j b/c data is 1D
  for( j=0; j<ymax; j++ ){            // cycles through y values
   for( i=0; i<xmax; i++ ){           // cycles through x values
    r=g=b=0;                          // set color values to black
    if( data[q][ioff]==0 );           // if data=0, values stay black
    else{
     if( logscale ){
      num=(double)data[q][ioff];
      num=log10( num );
      if( num<logmn ) num=0;
      else{
       if( num>logmx ) num=63;
       else
        num=63*( num-logmn )/( logmx-logmn );
      }
     }
     else{
      if( data[q][ioff]<mnval )  num=0;   // if data is less than min, =min
      else{
       if( data[q][ioff]>mxval )  num=63; // if data is more than max, =max
       else
        num=63*( data[q][ioff]-mnval )/( mxval-mnval );  // scale num from 0-63 (not defined for 64)
      }
     }
     r=jet[(int)num][0];            // set red val
     g=jet[(int)num][1];            // set green val
     b=jet[(int)num][2];            // set blue val  
    }

    glColor3f( r, g, b );           // put r, g, b into effect
    if( scaledepth && xz || scaledepth && yz )
     glRectf( i, j, i+1, j+depths[h] );
    else
     glRectf( i, j, i+1, j+1 );      // draws a square for data value 
    ioff++;                    
   } 
   if( scaledepth && xz || scaledepth && yz ){
    j+=depths[h]-1;
    h++;
   }
  } 

  glColor3f( 1, 1, 1 );             // set color to white
  glRectf( xmax, 0, xmax+1, ymax+1 );     // draws a border
  glRectf( 0, ymax, xmax, ymax+1 );       // draws a border  
  for( i=0; i<64; i++ ){            // draws color bar    
   glColor3f( jet[i][0], jet[i][1], jet[i][2]);         // sets color
   k=(float)i;                      // turns i into a float
   y=(float)ymax/64;                  // sets height of rectangles
   glRectf( xmax+10, y*k, xmax+20, (k+1)*y );               // draws rectangle 
  } 
  glColor3f( 1, 1, 1 );             // color to white

  if( logscale )
   sprintf( str, "%.2f", logmx );
  else
   sprintf( str, "%.1e", mxval );    // labels color bar with max val
  stroke( str, xmax+2, ymax-1 );
  
  if( logscale )
   sprintf( str, "%.2f", logmn );
  else
   sprintf( str, "%.1e", mnval );    // labels color bar with min val
  stroke( str, xmax+2, 1 );

  if( xz )
   sprintf( str, "N-S slice %d", yoffset+1);
  else{
   if( yz )
    sprintf( str, "E-W slice %d", xoffset+1); 
   else
    sprintf( str, "Level %d", ilev ); // labels current level 
  }
  stroke( str, 1, ymax+5 );

  sprintf( str, "Time %d", count+1 ); // labels current time 
  stroke( str, xmax/2, ymax+5 );

  if( glo )                         // labels how max/min have been set
   sprintf( str, "Global" );        // if glo is set, display Global
  if( usr ) 
   sprintf( str, "User-set" );      // if usr is set, display User-set
  if( !usr && !glo ) 
   sprintf( str, "Local" );         // else display Local 
  stroke( str, xmax+8, ymax+8 );
  
  if( anim ){                       // tell user if autoplay is on
   sprintf( str, "Autoplay" );
   stroke( str, xmax-35, ymax+8 );
  }

  if( logscale ){
   sprintf( str, "Log Scale" );     // tell user if log scale is on
   stroke( str, xmax-25, ymax+15); 
  }

  stroke( fns[q][count], 1, ymax+15 ); // labels current file
 
  glutSwapBuffers();                // double buffering set to animate smoothly
  glFlush(); 
 }
}

void stroke( char str[], int x, int y ){  // called to display text onscreen
 int i;

 glBlendFunc( GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA );
 glEnable( GL_BLEND );
 glEnable( GL_LINE_SMOOTH );
 glMatrixMode( GL_MODELVIEW );

 glPushMatrix();

 glScalef( SCALE, SCALE, SCALE );
 glTranslatef( (1/SCALE)*x, (1/SCALE)*y, 0 );

 for( i=0; i<strlen( str ); i++)    // display each character of str
  glutStrokeCharacter( GLUT_STROKE_ROMAN, str[i] );

 glPopMatrix();
}

void key( unsigned char key, int x, int y ){   // called on key press
 int i, tmp;
 
 switch(key){
  case 'q':   exit(0);              // quits on 'q'
              break;
  case 'r':   count=0;              // resets back to first time    
              ilev=1;               // and first level
              break;
  case 'a':   anim=(anim+1)%2;      // turns anim on/off
              break;
  case 'x':   xz=(xz+1)%2;
              yz=0;
              break;
  case 'y':   yz=(yz+1)%2;
              xz=0;
              break;
  case 'z':   if( xz || yz ){
               scaledepth=( scaledepth+1 )%2;
               scalecount++;
               if( scalecount==1 ) 
                readdepths( ".darwinview/depths" );
              }
              break; 
 } 
 glutPostRedisplay();
}

void TimerFunction( int value ){    // called when anim is set and arrow key is pressed
 int i;

 switch(value){                     // increments in the correct direction
  case DOWN : if( xz && yoffset>0 )
               yoffset--;
              else{
               if( yz && xoffset<NX-1 )
                xoffset++;
               else 
                if( ilev<NZ && !xz && !yz ) 
                 ilev++;              // if down arrow pressed, move down a level      
              }
              break;
  case UP   : if( xz && yoffset<NY-1 )
               yoffset++;
              else{
               if( yz && xoffset>0 )
                xoffset--; 
               else
                if( ilev>1 && !xz && !yz )    
                 ilev--;              // if up arrow is pressed, move up a level
              }
              break;
  case LEFT : if( count>0 )      
               count--;             // if left arrow is pressed, move back one time step 
              break;
  case RIGHT: if( count<howmany-1 )
               count++;             // if right arrow is pressed, moves forward one time step 
              break;
 }
 
 glutPostRedisplay();            

 if ( anim )
  switch( value ){    
   case DOWN : if( xz && yoffset==0 )
                yoffset=NY-1;
               else{ 
                if( yz && xoffset==NX-1 )
                 xoffset=0;
                else
                 if( ilev==NZ && !xz && !yz )
                  ilev=1;                   // if end reached, restart 
               }
               glutTimerFunc( 100, TimerFunction, value ); // recalls itself
               break;
   case UP   : if( xz && yoffset==NY-1 )
                yoffset=0;
               else{
                if( yz && xoffset==0 )
                 xoffset=NX-1;
                else
                 if( ilev==1 )
                  ilev=NZ;                   // if end reached, restart 
               }
               glutTimerFunc( 100, TimerFunction, value ); // recalls itself 
               break;
   case LEFT : if( count==0 ) count=howmany-1;             // if end reached, restart 
               glutTimerFunc( 100, TimerFunction, value ); // recalls itself
               break;
   case RIGHT: if( count==howmany-1 ) count=0;             // if end reached, restart                 
               glutTimerFunc( 100, TimerFunction, value ); // recalls itself
               break;
  } 
}

void specialkey( int key, int x, int y ){
 int i;

 if( anim )                       // if animation is set, call the timer function
  glutTimerFunc( 100, TimerFunction, key);             // to scroll automatically

 switch(key){
  case DOWN   :  if( xz && yoffset>0 ) 
                  yoffset--;
                 else{ 
                  if( yz && xoffset<NY-1 )
                   xoffset++;
                  else
                   if( ilev<NZ && !xz && !yz )      // if you haven't reached the bottom
                    ilev++;              // keep going down
                 } 
                 break;
  case UP     :  if( xz && yoffset<NY-1 )
                  yoffset++;
                 else{
                  if( yz && xoffset>0 )
                   xoffset--;
                  else
                   if( ilev>1 )          // if you haven't reached the top
                    ilev--;              // keep going up
                 }
                 break;
  case RIGHT  :  if( count<howmany-1 ) // if you haven't reached the last file
                  count++;             // keep going right
                 break;
  case LEFT   :  if( count>0 )         // if you haven't reached the first file
                  count--;             // keep going left
                 break;
 }
 glutPostRedisplay();            
}

void global(){                         // calculates the max/min for the total data set
 FILE* fp;
 int h, i, j;
 for( h=0; h<sets; h++)                // cycles through each window 
  for( i=0; i<howmany; i++ )           // cycles through each time 
   for( j=0; j<NZ; j++ ){              // cycles through each level for each file
    local( h, i, j );                  // calculates local max/min for specific data set 
    if( mxval > globalmx ) globalmx = mxval;   // sets highest value to globalmx
    if( mnval < globalmn ) globalmn = mnval;   // sets lowest value to globalmn
   }
}

void local( int i, int time, int lev ){        // calculates local max/min
 int j;
 
 mxval=0;  
 mnval=100;
 if( xz )
  readxz( data[i], fns[i][time]);
 else{
  if( yz )
   readyz( data[i], fns[i][time]);
  else
   readarray( data[i], fns[i][time], lev );      // read new array of data
 }
 for( j=0; j<xmax*ymax; j++ ){
  if( data[i][j] > mxval ) 
   mxval=data[i][j];                           // set largest val to mxval 
  if( data[i][j] < mnval && data[i][j]!=0 ) 
   mnval=data[i][j];                           // set smallest positive val to mnval 
 }
}

void readnames( char filename[] ){             // reads in list of filenames 
 FILE* fp;
 int i=0, j=0;
 
 fp=fopen( filename, "r" );                    // opens list of filenames
 while( !feof(fp) ){                           // reads until the end of the file 
  fscanf( fp, "%s", initfns[i] );              // places filename into an array of strings 
  i++;                                         // counts how many filenames there are  
 } 
 fclose( fp );                                 // close file
 sets=i-1;                                  // saves number of initial filenames 
 
 for(i=0; i<sets; i++){                     // goes through each read-in filename 
  fp=fopen( initfns[i], "r" );                 // opens each filename  
  j=0;
  while( !feof(fp) ){                          // reads in filenames from each filename 
   fscanf( fp, "%s", fns[i][j]);               
   j++;
  }
  fclose(fp);                                  // close file
 }
 howmany=j-1;                                  // saves number of data filenames  
}

void readdepths( char filename[] ){
 int i;
 FILE* fp;

 fp=fopen( filename, "r" );

 for( i=NZ-1; i>=0; i-- ){
  fscanf( fp, "%d ", &depths[i] );
  totaldepth+=depths[i];
 } 

 fclose( fp );
}

void readjet(){               //reads in color scale values
 FILE* fp;
 int i, j; 
  
 fp=fopen( ".darwinview/jet.h", "r" );  // opens file containing values
 for( i=0; i<64; i++ )        // reads in 64 sets of r, g, b values
  for( j=0; j<3; j++ )
   fscanf( fp, "%f", &jet[i][j] );
 fclose( fp );                // closes file
}

void readxz( float arr[], char filename[] ){
 int i, j;
 float tmp[MAX][MAX], tmp2[MAX*MAX];
 FILE* fp;
 
 fp=fopen( filename, "rb" );   
 for( i=0; i<NZ; i++){
  fseek( fp, (NX*yoffset*4)+(i*NX*NY*4), SEEK_SET );
  fread( &arr[NX*i], sizeof( arr[0] ), NX, fp );  // reads in data to fill one level 
 }
 fclose( fp );

 for( i=0; i<NZ; i++ )
  for( j=0; j<NX; j++)
   tmp[i][j]=arr[i*NX+j];

 for( i=0; i<NZ; i++ )
  for( j=0; j<NX; j++)
   tmp2[i*NX+j]=tmp[NZ-i-1][j];

 for( i=0; i<NX*NZ; i++)
  arr[i]=tmp2[i];
}

void readyz( float arr[], char filename[] ){
 int i, j;
 float tmp[MAX][MAX], tmp2[MAX*MAX];
 FILE* fp;  
 
 fp=fopen( filename, "rb" );
 for( i=0; i<NY*NZ; i++ ){
  fseek( fp, (xoffset*4)+(i*NX*4), SEEK_SET ); 
  fread( &arr[i], sizeof( arr[0] ), 1, fp );
 }
 fclose(fp);

 for( i=0; i<NZ; i++ )
  for( j=0; j<NY; j++)
   tmp[i][j]=arr[i*NY+j];

 for( i=0; i<NZ; i++ )
  for( j=0; j<NY; j++)
   tmp2[i*NY+j]=tmp[NZ-i-1][j];

 for( i=0; i<NY*NZ; i++)
  arr[i]=tmp2[i];
}

void readarray( float arr[], char filename[], int il ){   // reads new data
 FILE* fp;
 int i;
 
 fp=fopen( filename, "rb" );                 // opens the file containing data
 fseek( fp, (il-1)*NX*NY*4, SEEK_SET );      // seeks to the correct place using ilev
 fread( arr, sizeof( arr[0] ), NX*NY, fp );  // reads in data to fill one level 
 fclose( fp );                               // close file

 if( nonegs ) 
  for( i=0; i<NX*NY; i++ )                    
   if( arr[i] < 0 ) 
    arr[i] = pow(10, -20);   // sets negative values to 10^-20 

 if( endian )
  do_byteswap_f32( arr, NX*NY );            // switches endian 
}

void do_byteswap_f32( float arr[], int nel ){ // switches endian

 int i;
 char src[4], trg[4];

 for( i=0; i<nel; i++ ){
  memcpy( src, &(arr[i]), 4 );
  trg[0]=src[3];
  trg[1]=src[2];
  trg[2]=src[1];
  trg[3]=src[0];
  memcpy( &(arr[i]), trg, 4 );
 }
}

void black( ){
 glClear( GL_COLOR_BUFFER_BIT );
 glutSwapBuffers();
 glFlush();
}


int main( int argc, char *argv[] ){
 int i, setsx, setsy, tmpx, tmpy, winx, winy, parent;
 char str[MAX], filename[MAX];
 FILE* fp;

 if( strcmp(argv[1], "binary") == 0 )
  fp=fopen( ".darwinview/binconfig", "r" );
 else
  if( strcmp(argv[1],"netcdf") == 0 )
   fp=fopen( ".darwinview/ncconfig", "r" );

 fscanf( fp, "%dx%d ", &winx, &winy ); 
 winy-=60;  winx-=20;                     // adjusts resolution so edges won't get cut off             
 fscanf( fp, "%d %d %d ", &NX, &NY, &NZ );
 fscanf( fp, "%dx%d ", &setsx, &setsy );                // prompts user for dimensions of subwindows
 fscanf( fp, "%s", filename );

 fclose( fp );

 xmax=NX; ymax=NY;
 
 readjet();                               // stores color values
 readnames( filename );                   // gets list of filenames to read from
 global();                                // calculates max and min for all data

 glutInit( &argc, argv );                 
 glutInitDisplayMode( GLUT_RGB | GLUT_DOUBLE );   // sets rgb mode and double buffering
 glutInitWindowSize( winx, winy );                // parent window will cover screen
 glutInitWindowPosition( 10, 10 );                // location of parent window
 parent=glutCreateWindow( WINDOW );               // creates parent window

 glClearColor( 0, 0, 0, 0 );                      // sets clear color to black
 gluOrtho2D( 0, winx , winy, 0 );                 // sets(0,0) as top left corner 
 glutDisplayFunc( display );
 glutKeyboardFunc( key );                         // called on key press
 glutSpecialFunc( specialkey );                   // called on special key press (arrow keys)
 

 for( i=0; i<setsx*setsy; i++ ){
  tmpx = (i%setsx)*(winx/setsx);                  // x coordinate of top left corner of subwindow
  tmpy = (i/setsx)*(winy/setsy);                  // y coordinate of top left corner of subwindow 

  win[i]=glutCreateSubWindow( parent, tmpx, tmpy, winx/setsx, winy/setsy ); // creates subwindow
  gluOrtho2D( 0, NX+35, 0, NY+25 );               // sets how data is mapped to subwindow
  glutKeyboardFunc( key );                        // called on key press
  glutSpecialFunc( specialkey );                  // called on special key press (arrow keys)
  if( i >= sets )                                 // 
   glutDisplayFunc( black );
  else
   glutDisplayFunc( display );                     // sets display func for subwindow
  

  glutCreateMenu( menu );                                // adds a menu
   glutAddMenuEntry( "Local Color Scale", 1 );           // adds a menu entry 
   glutAddMenuEntry( "Global Color Scale", 2 );          // adds a menu entry 
   glutAddMenuEntry( "User-Set Color Scale", 3 );        // adds a menu entry 
   glutAddMenuEntry( "Log Scale (on/off)", 4 );          // adds a menu entry 
   glutAddMenuEntry( "Allow Negatives (on/off)", 5 );    // adds a menu entry
   glutAddMenuEntry( "Switch Endian (big/little) ", 6);  // adds a menu entry
  glutAttachMenu( GLUT_RIGHT_BUTTON );                   // menu called on right click
 }

 glutMainLoop();                                  // begin processing events
 return 0;
}

1	marissa	1.1	#include <GL/glut.h>
2			#include <stdlib.h>
3			#include <stdio.h>
4			#include <string.h>
5			#include <math.h>
6			#define WINDOW "image"
7			#define UP 101
8			#define DOWN 103
9			#define RIGHT 102
10			#define LEFT 100
11	marissa	1.4	#define MAX 700
12	marissa	1.7	#define SCALE .06
13	marissa	1.1
14	marissa	1.3	int NX, NY, NZ;
15	marissa	1.1
16			void do_byteswap_f32( float arr[], int nel ), global(), local( int, int, int );
17	marissa	1.4	void readnames( char[] ), readarray( float[], char[], int ), readjet(), readxz( float[], char[] );
18	marissa	1.7	void readyz( float[], char[] ), readdepths( char[] );
19			void TimerFunction( int ), stroke( char[], int, int );
20	marissa	1.1
21	marissa	1.4	float data[MAX][MAX*MAX], mxval, mnval, jet[64][3];
22	marissa	1.1	float globalmx=0, globalmn=100;
23	marissa	1.7	int win[MAX], ilev=1, howmany, sets, count=0, glo=0, usr=0, anim=0, endian=0;
24			int xmax, ymax, yoffset=0, xoffset=0, yz=0, logscale=0, xz=0, nonegs=1;
25			int depths[MAX], scaledepth=0, totaldepth=0, scalecount=0;
26	marissa	1.1	char initfns[MAX][MAX], fns[MAX][MAX][MAX];
27
28	marissa	1.6	void menu( int value ){ // called when menu is opened on right click
29	marissa	1.1
30			switch( value ){
31	marissa	1.2	case 1: usr=glo=0; // unset glo & usr, sets local max/min
32	marissa	1.1	break;
33	marissa	1.2	case 2: glo=1; // enables global max/min
34			usr=0; // unsets usr
35			mxval=globalmx; // sets max to globalmx
36			mnval=globalmn; // sets min to globalmn
37	marissa	1.1	break;
38	marissa	1.2	case 3: usr=1; // switch to user-set max/min
39			glo=0; // unset glo
40			printf( "Max=" ); scanf( "%f", &mxval ); // prompt user for new max
41			printf( "Min=" ); scanf( "%f", &mnval ); // prompt user for new min
42			break;
43	marissa	1.7	case 4: logscale=( logscale+1 )%2; // switch log scale on/off
44	marissa	1.1	break;
45	marissa	1.7	case 5: nonegs=( nonegs+1 )%2; // switch allowance of negatives on/off
46	marissa	1.3	break;
47	marissa	1.7	case 6: endian=( endian+1 )%2; // switch between big/little endian
48	marissa	1.3	break;
49	marissa	1.1	}
50	marissa	1.7	glutPostRedisplay();
51	marissa	1.1	}
52	marissa	1.7
53	marissa	1.4	void display(){ // called on glutPostRedisplay
54	marissa	1.7	int i, j, h, ioff, q;
55	marissa	1.2	float r, g, b, k, y;
56			double num, logmx, logmn;
57	marissa	1.1	char str[MAX];
58
59	marissa	1.7
60	marissa	1.4	for( q=0; q<sets; q++ ){ // runs display func for each subwindow
61	marissa	1.2	glutSetWindow( win[q] ); // sets which subwindow to display to
62			glClear( GL_COLOR_BUFFER_BIT ); // background to black
63	marissa	1.7
64	marissa	1.5	if( xz ) { xmax=NX; ymax=NZ; }
65			else{
66			if( yz ){ xmax=NY; ymax=NZ; }
67			else { xmax=NX; ymax=NY; }
68			}
69	marissa	1.4
70			if( glo \|\| usr ){ // if global or user-set max/min
71			if( xz )
72			readxz( data[q], fns[q][count] );
73	marissa	1.5	else{
74			if( yz )
75			readyz( data[q], fns[q][count] );
76			else
77			readarray( data[q], fns[q][count], ilev ); // read new array w/o calculating local max/min
78			}
79	marissa	1.4	}
80	marissa	1.2	else // if local max/min is set
81			local( q, count, ilev ); // read new array and calculate local max/min
82
83			if( logscale ){
84			if( usr ){
85			logmx=(double)mxval;
86			logmn=(double)mnval;
87			}
88			else{
89			logmx=log10( mxval );
90			logmn=log10( mnval );
91			}
92			}
93	marissa	1.4
94	marissa	1.7	if( scaledepth && xz \|\| scaledepth && yz ) ymax=totaldepth;
95			ioff=0; h=0; // ioff will count both i&j b/c data is 1D
96	marissa	1.4	for( j=0; j<ymax; j++ ){ // cycles through y values
97			for( i=0; i<xmax; i++ ){ // cycles through x values
98	marissa	1.7	r=g=b=0; // set color values to black
99	marissa	1.2	if( data[q][ioff]==0 ); // if data=0, values stay black
100	marissa	1.1	else{
101	marissa	1.2	if( logscale ){
102			num=(double)data[q][ioff];
103			num=log10( num );
104			if( num<logmn ) num=0;
105	marissa	1.3	else{
106			if( num>logmx ) num=63;
107			else
108			num=63*( num-logmn )/( logmx-logmn );
109			}
110	marissa	1.2	}
111	marissa	1.1	else{
112	marissa	1.2	if( data[q][ioff]<mnval ) num=0; // if data is less than min, =min
113	marissa	1.3	else{
114	marissa	1.2	if( data[q][ioff]>mxval ) num=63; // if data is more than max, =max
115			else
116			num=63*( data[q][ioff]-mnval )/( mxval-mnval ); // scale num from 0-63 (not defined for 64)
117	marissa	1.3	}
118	marissa	1.1	}
119	marissa	1.2	r=jet[(int)num][0]; // set red val
120			g=jet[(int)num][1]; // set green val
121			b=jet[(int)num][2]; // set blue val
122	marissa	1.1	}
123	marissa	1.2
124			glColor3f( r, g, b ); // put r, g, b into effect
125	marissa	1.7	if( scaledepth && xz \|\| scaledepth && yz )
126			glRectf( i, j, i+1, j+depths[h] );
127			else
128			glRectf( i, j, i+1, j+1 ); // draws a square for data value
129	marissa	1.1	ioff++;
130			}
131	marissa	1.7	if( scaledepth && xz \|\| scaledepth && yz ){
132			j+=depths[h]-1;
133			h++;
134			}
135	marissa	1.1	}
136	marissa	1.7
137	marissa	1.2	glColor3f( 1, 1, 1 ); // set color to white
138	marissa	1.4	glRectf( xmax, 0, xmax+1, ymax+1 ); // draws a border
139			glRectf( 0, ymax, xmax, ymax+1 ); // draws a border
140	marissa	1.2	for( i=0; i<64; i++ ){ // draws color bar
141			glColor3f( jet[i][0], jet[i][1], jet[i][2]); // sets color
142			k=(float)i; // turns i into a float
143	marissa	1.4	y=(float)ymax/64; // sets height of rectangles
144			glRectf( xmax+10, yk, xmax+20, (k+1)y ); // draws rectangle
145	marissa	1.1	}
146	marissa	1.2	glColor3f( 1, 1, 1 ); // color to white
147	marissa	1.1
148	marissa	1.2	if( logscale )
149			sprintf( str, "%.2f", logmx );
150			else
151	marissa	1.3	sprintf( str, "%.1e", mxval ); // labels color bar with max val
152	marissa	1.7	stroke( str, xmax+2, ymax-1 );
153	marissa	1.1
154	marissa	1.2	if( logscale )
155			sprintf( str, "%.2f", logmn );
156			else
157	marissa	1.3	sprintf( str, "%.1e", mnval ); // labels color bar with min val
158	marissa	1.7	stroke( str, xmax+2, 1 );
159	marissa	1.1
160	marissa	1.4	if( xz )
161	marissa	1.6	sprintf( str, "N-S slice %d", yoffset+1);
162	marissa	1.5	else{
163			if( yz )
164	marissa	1.6	sprintf( str, "E-W slice %d", xoffset+1);
165	marissa	1.5	else
166			sprintf( str, "Level %d", ilev ); // labels current level
167			}
168	marissa	1.7	stroke( str, 1, ymax+5 );
169	marissa	1.1
170	marissa	1.2	sprintf( str, "Time %d", count+1 ); // labels current time
171	marissa	1.7	stroke( str, xmax/2, ymax+5 );
172	marissa	1.2
173			if( glo ) // labels how max/min have been set
174			sprintf( str, "Global" ); // if glo is set, display Global
175	marissa	1.1	if( usr )
176	marissa	1.2	sprintf( str, "User-set" ); // if usr is set, display User-set
177	marissa	1.1	if( !usr && !glo )
178	marissa	1.2	sprintf( str, "Local" ); // else display Local
179	marissa	1.7	stroke( str, xmax+8, ymax+8 );
180	marissa	1.1
181	marissa	1.2	if( anim ){ // tell user if autoplay is on
182	marissa	1.1	sprintf( str, "Autoplay" );
183	marissa	1.7	stroke( str, xmax-35, ymax+8 );
184	marissa	1.2	}
185
186			if( logscale ){
187			sprintf( str, "Log Scale" ); // tell user if log scale is on
188	marissa	1.7	stroke( str, xmax-25, ymax+15);
189	marissa	1.1	}
190
191	marissa	1.7	stroke( fns[q][count], 1, ymax+15 ); // labels current file
192	marissa	1.1
193	marissa	1.2	glutSwapBuffers(); // double buffering set to animate smoothly
194	marissa	1.1	glFlush();
195			}
196			}
197
198	marissa	1.7	void stroke( char str[], int x, int y ){ // called to display text onscreen
199	marissa	1.1	int i;
200
201	marissa	1.7	glBlendFunc( GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA );
202			glEnable( GL_BLEND );
203			glEnable( GL_LINE_SMOOTH );
204			glMatrixMode( GL_MODELVIEW );
205
206			glPushMatrix();
207
208			glScalef( SCALE, SCALE, SCALE );
209			glTranslatef( (1/SCALE)x, (1/SCALE)y, 0 );
210
211	marissa	1.2	for( i=0; i<strlen( str ); i++) // display each character of str
212	marissa	1.7	glutStrokeCharacter( GLUT_STROKE_ROMAN, str[i] );
213
214			glPopMatrix();
215	marissa	1.1	}
216
217	marissa	1.2	void key( unsigned char key, int x, int y ){ // called on key press
218	marissa	1.3	int i, tmp;
219	marissa	1.1
220			switch(key){
221	marissa	1.2	case 'q': exit(0); // quits on 'q'
222	marissa	1.1	break;
223	marissa	1.2	case 'r': count=0; // resets back to first time
224			ilev=1; // and first level
225	marissa	1.1	break;
226	marissa	1.2	case 'a': anim=(anim+1)%2; // turns anim on/off
227	marissa	1.1	break;
228	marissa	1.4	case 'x': xz=(xz+1)%2;
229	marissa	1.5	yz=0;
230	marissa	1.4	break;
231	marissa	1.5	case 'y': yz=(yz+1)%2;
232			xz=0;
233	marissa	1.7	break;
234			case 'z': if( xz \|\| yz ){
235			scaledepth=( scaledepth+1 )%2;
236			scalecount++;
237			if( scalecount==1 )
238			readdepths( ".darwinview/depths" );
239			}
240			break;
241	marissa	1.1	}
242	marissa	1.7	glutPostRedisplay();
243	marissa	1.1	}
244
245	marissa	1.2	void TimerFunction( int value ){ // called when anim is set and arrow key is pressed
246	marissa	1.1	int i;
247
248	marissa	1.2	switch(value){ // increments in the correct direction
249	marissa	1.5	case DOWN : if( xz && yoffset>0 )
250			yoffset--;
251	marissa	1.4	else{
252	marissa	1.5	if( yz && xoffset<NX-1 )
253			xoffset++;
254			else
255			if( ilev<NZ && !xz && !yz )
256			ilev++; // if down arrow pressed, move down a level
257	marissa	1.4	}
258	marissa	1.1	break;
259	marissa	1.5	case UP : if( xz && yoffset<NY-1 )
260			yoffset++;
261	marissa	1.4	else{
262	marissa	1.5	if( yz && xoffset>0 )
263			xoffset--;
264			else
265			if( ilev>1 && !xz && !yz )
266			ilev--; // if up arrow is pressed, move up a level
267	marissa	1.4	}
268	marissa	1.1	break;
269			case LEFT : if( count>0 )
270	marissa	1.2	count--; // if left arrow is pressed, move back one time step
271	marissa	1.1	break;
272			case RIGHT: if( count<howmany-1 )
273	marissa	1.2	count++; // if right arrow is pressed, moves forward one time step
274	marissa	1.1	break;
275			}
276
277			glutPostRedisplay();
278
279	marissa	1.5	if ( anim )
280	marissa	1.1	switch( value ){
281	marissa	1.5	case DOWN : if( xz && yoffset==0 )
282			yoffset=NY-1;
283	marissa	1.4	else{
284	marissa	1.5	if( yz && xoffset==NX-1 )
285			xoffset=0;
286			else
287			if( ilev==NZ && !xz && !yz )
288			ilev=1; // if end reached, restart
289	marissa	1.4	}
290	marissa	1.1	glutTimerFunc( 100, TimerFunction, value ); // recalls itself
291			break;
292	marissa	1.5	case UP : if( xz && yoffset==NY-1 )
293			yoffset=0;
294	marissa	1.4	else{
295	marissa	1.5	if( yz && xoffset==0 )
296			xoffset=NX-1;
297			else
298			if( ilev==1 )
299			ilev=NZ; // if end reached, restart
300	marissa	1.4	}
301	marissa	1.1	glutTimerFunc( 100, TimerFunction, value ); // recalls itself
302			break;
303	marissa	1.2	case LEFT : if( count==0 ) count=howmany-1; // if end reached, restart
304	marissa	1.1	glutTimerFunc( 100, TimerFunction, value ); // recalls itself
305			break;
306	marissa	1.2	case RIGHT: if( count==howmany-1 ) count=0; // if end reached, restart
307	marissa	1.1	glutTimerFunc( 100, TimerFunction, value ); // recalls itself
308			break;
309			}
310			}
311
312			void specialkey( int key, int x, int y ){
313			int i;
314
315	marissa	1.2	if( anim ) // if animation is set, call the timer function
316			glutTimerFunc( 100, TimerFunction, key); // to scroll automatically
317	marissa	1.1
318			switch(key){
319	marissa	1.5	case DOWN : if( xz && yoffset>0 )
320			yoffset--;
321	marissa	1.4	else{
322	marissa	1.5	if( yz && xoffset<NY-1 )
323			xoffset++;
324			else
325			if( ilev<NZ && !xz && !yz ) // if you haven't reached the bottom
326			ilev++; // keep going down
327	marissa	1.4	}
328	marissa	1.1	break;
329	marissa	1.5	case UP : if( xz && yoffset<NY-1 )
330			yoffset++;
331	marissa	1.4	else{
332	marissa	1.5	if( yz && xoffset>0 )
333			xoffset--;
334			else
335			if( ilev>1 ) // if you haven't reached the top
336			ilev--; // keep going up
337	marissa	1.4	}
338	marissa	1.1	break;
339			case RIGHT : if( count<howmany-1 ) // if you haven't reached the last file
340			count++; // keep going right
341			break;
342			case LEFT : if( count>0 ) // if you haven't reached the first file
343			count--; // keep going left
344			break;
345			}
346			glutPostRedisplay();
347			}
348
349	marissa	1.2	void global(){ // calculates the max/min for the total data set
350	marissa	1.1	FILE* fp;
351			int h, i, j;
352	marissa	1.3	for( h=0; h<sets; h++) // cycles through each window
353	marissa	1.2	for( i=0; i<howmany; i++ ) // cycles through each time
354	marissa	1.3	for( j=0; j<NZ; j++ ){ // cycles through each level for each file
355	marissa	1.2	local( h, i, j ); // calculates local max/min for specific data set
356	marissa	1.1	if( mxval > globalmx ) globalmx = mxval; // sets highest value to globalmx
357			if( mnval < globalmn ) globalmn = mnval; // sets lowest value to globalmn
358			}
359			}
360
361	marissa	1.2	void local( int i, int time, int lev ){ // calculates local max/min
362	marissa	1.1	int j;
363
364			mxval=0;
365			mnval=100;
366	marissa	1.4	if( xz )
367			readxz( data[i], fns[i][time]);
368	marissa	1.5	else{
369			if( yz )
370			readyz( data[i], fns[i][time]);
371			else
372			readarray( data[i], fns[i][time], lev ); // read new array of data
373			}
374	marissa	1.4	for( j=0; j<xmax*ymax; j++ ){
375	marissa	1.2	if( data[i][j] > mxval )
376			mxval=data[i][j]; // set largest val to mxval
377	marissa	1.3	if( data[i][j] < mnval && data[i][j]!=0 )
378	marissa	1.2	mnval=data[i][j]; // set smallest positive val to mnval
379	marissa	1.1	}
380			}
381
382	marissa	1.2	void readnames( char filename[] ){ // reads in list of filenames
383	marissa	1.1	FILE* fp;
384			int i=0, j=0;
385
386	marissa	1.2	fp=fopen( filename, "r" ); // opens list of filenames
387			while( !feof(fp) ){ // reads until the end of the file
388			fscanf( fp, "%s", initfns[i] ); // places filename into an array of strings
389			i++; // counts how many filenames there are
390	marissa	1.1	}
391	marissa	1.2	fclose( fp ); // close file
392	marissa	1.3	sets=i-1; // saves number of initial filenames
393	marissa	1.1
394	marissa	1.3	for(i=0; i<sets; i++){ // goes through each read-in filename
395	marissa	1.2	fp=fopen( initfns[i], "r" ); // opens each filename
396	marissa	1.1	j=0;
397	marissa	1.2	while( !feof(fp) ){ // reads in filenames from each filename
398			fscanf( fp, "%s", fns[i][j]);
399	marissa	1.1	j++;
400			}
401	marissa	1.2	fclose(fp); // close file
402	marissa	1.1	}
403	marissa	1.2	howmany=j-1; // saves number of data filenames
404	marissa	1.1	}
405
406	marissa	1.7	void readdepths( char filename[] ){
407			int i;
408			FILE* fp;
409
410			fp=fopen( filename, "r" );
411
412			for( i=NZ-1; i>=0; i-- ){
413			fscanf( fp, "%d ", &depths[i] );
414			totaldepth+=depths[i];
415			}
416
417			fclose( fp );
418			}
419
420	marissa	1.2	void readjet(){ //reads in color scale values
421	marissa	1.1	FILE* fp;
422			int i, j;
423
424	marissa	1.6	fp=fopen( ".darwinview/jet.h", "r" ); // opens file containing values
425	marissa	1.2	for( i=0; i<64; i++ ) // reads in 64 sets of r, g, b values
426	marissa	1.1	for( j=0; j<3; j++ )
427			fscanf( fp, "%f", &jet[i][j] );
428	marissa	1.2	fclose( fp ); // closes file
429	marissa	1.1	}
430
431	marissa	1.4	void readxz( float arr[], char filename[] ){
432	marissa	1.5	int i, j;
433			float tmp[MAX][MAX], tmp2[MAX*MAX];
434	marissa	1.4	FILE* fp;
435
436			fp=fopen( filename, "rb" );
437			for( i=0; i<NZ; i++){
438	marissa	1.5	fseek( fp, (NXyoffset4)+(iNXNY*4), SEEK_SET );
439	marissa	1.4	fread( &arr[NX*i], sizeof( arr[0] ), NX, fp ); // reads in data to fill one level
440			}
441			fclose( fp );
442	marissa	1.5
443			for( i=0; i<NZ; i++ )
444			for( j=0; j<NX; j++)
445			tmp[i][j]=arr[i*NX+j];
446
447			for( i=0; i<NZ; i++ )
448			for( j=0; j<NX; j++)
449			tmp2[i*NX+j]=tmp[NZ-i-1][j];
450
451			for( i=0; i<NX*NZ; i++)
452			arr[i]=tmp2[i];
453			}
454
455			void readyz( float arr[], char filename[] ){
456			int i, j;
457			float tmp[MAX][MAX], tmp2[MAX*MAX];
458			FILE* fp;
459
460			fp=fopen( filename, "rb" );
461			for( i=0; i<NY*NZ; i++ ){
462			fseek( fp, (xoffset4)+(iNX*4), SEEK_SET );
463			fread( &arr[i], sizeof( arr[0] ), 1, fp );
464			}
465			fclose(fp);
466
467			for( i=0; i<NZ; i++ )
468			for( j=0; j<NY; j++)
469			tmp[i][j]=arr[i*NY+j];
470
471			for( i=0; i<NZ; i++ )
472			for( j=0; j<NY; j++)
473			tmp2[i*NY+j]=tmp[NZ-i-1][j];
474
475			for( i=0; i<NY*NZ; i++)
476			arr[i]=tmp2[i];
477	marissa	1.4	}
478
479	marissa	1.1	void readarray( float arr[], char filename[], int il ){ // reads new data
480	marissa	1.3	FILE* fp;
481	marissa	1.1	int i;
482
483	marissa	1.5	fp=fopen( filename, "rb" ); // opens the file containing data
484	marissa	1.2	fseek( fp, (il-1)NXNY*4, SEEK_SET ); // seeks to the correct place using ilev
485	marissa	1.1	fread( arr, sizeof( arr[0] ), NX*NY, fp ); // reads in data to fill one level
486	marissa	1.2	fclose( fp ); // close file
487
488	marissa	1.3	if( nonegs )
489			for( i=0; i<NX*NY; i++ )
490			if( arr[i] < 0 )
491			arr[i] = pow(10, -20); // sets negative values to 10^-20
492
493			if( endian )
494			do_byteswap_f32( arr, NX*NY ); // switches endian
495	marissa	1.1	}
496
497	marissa	1.3	void do_byteswap_f32( float arr[], int nel ){ // switches endian
498
499			int i;
500			char src[4], trg[4];
501
502			for( i=0; i<nel; i++ ){
503			memcpy( src, &(arr[i]), 4 );
504			trg[0]=src[3];
505			trg[1]=src[2];
506			trg[2]=src[1];
507			trg[3]=src[0];
508			memcpy( &(arr[i]), trg, 4 );
509			}
510			}
511
512	marissa	1.6	void black( ){
513	marissa	1.3	glClear( GL_COLOR_BUFFER_BIT );
514			glutSwapBuffers();
515	marissa	1.6	glFlush();
516	marissa	1.1	}
517
518	marissa	1.6
519	marissa	1.1	int main( int argc, char *argv[] ){
520	marissa	1.3	int i, setsx, setsy, tmpx, tmpy, winx, winy, parent;
521	marissa	1.6	char str[MAX], filename[MAX];
522			FILE* fp;
523
524			if( strcmp(argv[1], "binary") == 0 )
525			fp=fopen( ".darwinview/binconfig", "r" );
526			else
527			if( strcmp(argv[1],"netcdf") == 0 )
528			fp=fopen( ".darwinview/ncconfig", "r" );
529	marissa	1.1
530	marissa	1.6	fscanf( fp, "%dx%d ", &winx, &winy );
531	marissa	1.2	winy-=60; winx-=20; // adjusts resolution so edges won't get cut off
532	marissa	1.6	fscanf( fp, "%d %d %d ", &NX, &NY, &NZ );
533			fscanf( fp, "%dx%d ", &setsx, &setsy ); // prompts user for dimensions of subwindows
534			fscanf( fp, "%s", filename );
535
536			fclose( fp );
537	marissa	1.1
538	marissa	1.4	xmax=NX; ymax=NY;
539	marissa	1.7
540	marissa	1.2	readjet(); // stores color values
541			readnames( filename ); // gets list of filenames to read from
542			global(); // calculates max and min for all data
543
544			glutInit( &argc, argv );
545			glutInitDisplayMode( GLUT_RGB \| GLUT_DOUBLE ); // sets rgb mode and double buffering
546			glutInitWindowSize( winx, winy ); // parent window will cover screen
547			glutInitWindowPosition( 10, 10 ); // location of parent window
548			parent=glutCreateWindow( WINDOW ); // creates parent window
549	marissa	1.1
550	marissa	1.2	glClearColor( 0, 0, 0, 0 ); // sets clear color to black
551			gluOrtho2D( 0, winx , winy, 0 ); // sets(0,0) as top left corner
552	marissa	1.1	glutDisplayFunc( display );
553	marissa	1.2	glutKeyboardFunc( key ); // called on key press
554			glutSpecialFunc( specialkey ); // called on special key press (arrow keys)
555	marissa	1.3
556	marissa	1.1
557			for( i=0; i<setsx*setsy; i++ ){
558	marissa	1.2	tmpx = (i%setsx)*(winx/setsx); // x coordinate of top left corner of subwindow
559			tmpy = (i/setsx)*(winy/setsy); // y coordinate of top left corner of subwindow
560	marissa	1.1
561	marissa	1.2	win[i]=glutCreateSubWindow( parent, tmpx, tmpy, winx/setsx, winy/setsy ); // creates subwindow
562	marissa	1.7	gluOrtho2D( 0, NX+35, 0, NY+25 ); // sets how data is mapped to subwindow
563	marissa	1.2	glutKeyboardFunc( key ); // called on key press
564			glutSpecialFunc( specialkey ); // called on special key press (arrow keys)
565	marissa	1.3	if( i >= sets ) //
566			glutDisplayFunc( black );
567			else
568			glutDisplayFunc( display ); // sets display func for subwindow
569
570	marissa	1.2
571	marissa	1.3	glutCreateMenu( menu ); // adds a menu
572			glutAddMenuEntry( "Local Color Scale", 1 ); // adds a menu entry
573			glutAddMenuEntry( "Global Color Scale", 2 ); // adds a menu entry
574			glutAddMenuEntry( "User-Set Color Scale", 3 ); // adds a menu entry
575			glutAddMenuEntry( "Log Scale (on/off)", 4 ); // adds a menu entry
576			glutAddMenuEntry( "Allow Negatives (on/off)", 5 ); // adds a menu entry
577			glutAddMenuEntry( "Switch Endian (big/little) ", 6); // adds a menu entry
578			glutAttachMenu( GLUT_RIGHT_BUTTON ); // menu called on right click
579	marissa	1.1	}
580
581	marissa	1.2	glutMainLoop(); // begin processing events
582	marissa	1.1	return 0;
583			}
584