#include <stdio.h>
#include <stdlib.h>
#include "s2plot.h"
#include "us2spherical.h"

/* Demonstration of the spherical grid functions in us2spherical.c 
 * 
 * Usage: 
 * 
 * Make a local copy of cbuild.csh from the S2PLOT source distribution
 *
 * cp ${S2PATH}/scripts/cbuild.csh .
 *
 * Make the following edits to cbuild.csh to add us2spherical.c and us2spherical.o 
 *
 * $S2COMPILER ${S2EXTRAINC} -I${S2PATH}/src $source us2spherical.c
 * 
 * $S2CLINKER -o $target ${object} us2spherical.o -L${S2PATH}/${S2KERNEL} ${S2LINKS} ${MLLINKS} ${SWLINKS} ${GLLINKS} -L${S2X11PATH}/lib${S2LBITS} ${S2FORMSLINK}  -lX11 ${IMATH} -lm ${XLINKPATH} ${S2EXTRALIB}
 *
 * % ./cbuild.csh spheredemo
 *
 * % spheredemo [N]     
 * where N = 1..7
 *
 * Written by C.Fluke (Swinburne; cfluke@swin.edu.au) 
 * March 2014 */

void RGBtoCI(COLOUR col, int ci) 
{
   s2scr(ci, (float)col.r, (float)col.g, (float)col.b);
   s2sci(ci);
}

int main(int argc, char *argv[])
{
   int mode = 0;
   if (argc == 2) {
      mode = atoi(argv[1]);
   }
   s2opend("/?",argc,argv);
   s2swin(-1,1,-1,1,-1,1);

/* RA must be in decimal hours (0..24), DA must be in decimal degrees (-90..+90) */
   int N = 10;
   float *ra     = (float *)calloc(N, sizeof(float));
   float *de     = (float *)calloc(N, sizeof(float));
   float *radius = (float *)calloc(N, sizeof(float));
   COLOUR *col   = (COLOUR *)calloc(N,sizeof(COLOUR));
   int *sym      = (int *)calloc(N, sizeof(int));
   XYZ xyz;

   int i;
   for (i=0;i<N;i++) {
      ra[i] = drand48()*24.0;			/* [0..24] */
      de[i] = drand48()*180.0-90.0;		/* [-90..+90] */
      radius[i] = drand48()*1.5+0.5;		/* Eg. [0.5..2.0] */
      col[i].r  = drand48();			/* Random (R,G,B) colour for this point */
      col[i].g  = drand48();
      col[i].b  = drand48();
      sym[i]    = (int)floor(drand48()*6.0);	
      if (sym[i] == 3) sym[i] = 2; 
      if (sym[i] == 5) sym[i] = 4;
   }

   int ramin = 1.0;
   int ramax = 23.0;
   int demin = -80.0;
   int demax = +80.0;
   int ci    = 1000;				/* Colour index to change */
   int sz = 4;					/* Arrowhead size for mode 6 */


   switch(mode) {
      case 1 : 

		/* Convert RA (hours) and Declination (degrees) into XYZ coordinates */
		for (i=0;i<N;i++) {
                   xyz = us2rade2xyz(ra[i], de[i], radius[i]);
	           ns2vthpoint(xyz, col[i], 3);
                }
                break;

     case 2 :   /* Plot N pairs of RA (hours) and Declination (degrees) points as symbols */
	 	/* 0 = wireframe box 1 = point 2 = wireframe 3D cross 4 = shaded sphere 6 = shaded box	*/
                us2radept(N, ra, de, radius, sym);
                break;


     case 3 : 	/* Plot a single RA (hours) and Declination (degrees) point */
		/*  Sphere has radius r, plot using symbol sym */
	        RGBtoCI(col[0], ci);
                us2radept1(ra[0], de[0], radius[0], sym[0]);
	        s2sci(S2_PG_WHITE);
                break;
    
     case 4 :   /* Draw a line at constant RA (hours) between Declinations (degrees) */
		/*   demin and demax. Sphere has radius r */
	        s2slw(3.0);				/* Make these thick lines */
	        for (i=0;i<N;i++) {
	           RGBtoCI(col[i], ci);
                   us2raline(ra[i], demin, demax, radius[i]);
                }
	        s2sci(S2_PG_WHITE);
		s2slw(1.0);
                break;
 
     case 5 :   /* Draw a line at constant Declination (degrees) between RAs (hours) */
    		/*   ramin and ramax. Sphere has radius r */
	        s2slw(3.0);				/* Make these thick lines */
	        for (i=0;i<N;i++) {
	           RGBtoCI(col[i], ci);
                   us2deline(de[i], ramin, ramax, radius[i]);
                }
	        s2sci(S2_PG_WHITE);
		s2slw(1.0);
                break;

     case 6 :   /* Draw an arrow between radius r1 and radius r2 for constant RA (hour) */
		/*   and Declination (degrees). Arrow has colour ci and size sz  */
	        for (i=0;i<N;i++) {
    		   RGBtoCI(col[i], ci);
                   us2radearrow((double)ra[i], (double)de[i], 0.0, radius[i], ci, sz);
                }
                s2sci(S2_PG_WHITE);
                break;

     case 7 :   /* Draw a line from (ra1,de1) to (ra2,de2) - does not work in all cases */
		s2slw(3.0);
	        for (i=0;i<N;i++) {
    		   RGBtoCI(col[i], ci);
                   us2radeline(ra[i], de[i], ra[(i+1)%N], de[(i+1)%N], radius[0]);
                }
                s2sci(S2_PG_WHITE);
		s2slw(1.0);
                break;
   }

/* Draw a RA (hour) and Declination (degrees) grid with integer steps in */
/*  RA and Declination.  Sphere has radius r */
    int rastep = 3;
    int destep = 15;
    float sradius = 1.0;
    us2radegrid(rastep, destep, sradius);


   s2show(1);
 
   return 0;
}