#include "niml.h"

/********************************************************************
  Timing test for NIML element creation:
    - load 1 row at a time:  NI_add_row()
    - load all rows at once: NI_add_many_rows()
    - load by columns:       NI_add_column()
*********************************************************************/

typedef struct {      /* a location in space */
  int id ;
  float x,y,z ;
} AGNI_nod ;

/* prototypes */

NI_element * AGNI_nod_to_NIML_col ( int num_nod, AGNI_nod *nod ) ;
NI_element * AGNI_nod_to_NIML_row ( int num_nod, AGNI_nod *nod ) ;
NI_element * AGNI_nod_to_NIML_rows( int num_nod, AGNI_nod *nod ) ;

/*------------------------------------------------------------------*/

int main( int argc, char *argv[] )
{
   int ii, nn, ct, nb ;
   AGNI_nod *nod ;
   NI_element *nel1, *nel2, *nel3 ;
   NI_stream ns ;

   if( argc < 2 ){ printf("Usage: nisurf n\n"); exit(0); }

   nn = strtol(argv[1],NULL,10) ;
   if( nn < 1 ) exit(1) ;

   /* make up some fake nodes */

   nod = NI_malloc(AGNI_nod, sizeof(AGNI_nod)*nn ) ;
   for( ii=0 ; ii < nn ; ii++ ){
      nod[ii].id = ii     ;
      nod[ii].x  = 1.1*ii ;
      nod[ii].y  = 2.2*ii ;
      nod[ii].z  = 3.3*ii ;
   }

   /* put nodes into a NIML data element: one row at a time */

   ct = NI_clock_time() ;                   /* start timer */
   nel1 = AGNI_nod_to_NIML_row( nn , nod ) ; /* create element by row */
   fprintf(stderr,"Clock time for %d nodes by row = %d ms\n",
           nn,NI_clock_time()-ct) ;
   if( nn < 10 ){                           /* print element */
     ns = NI_stream_open( "fd:1" , "w" ) ;
     if( ns == NULL ){ fprintf(stderr,"Can't open fd:1\n"); exit(1); }
     nb = NI_write_element( ns , nel1 , NI_TEXT_MODE ) ;
     fprintf(stderr,"num bytes=%d\n\n",nb) ;
   }
   /* NI_free_element(nel1) ; */                  /* free element */

   /* put nodes into a NIML data element: all rows at once */

   ct = NI_clock_time() ;
   nel2 = AGNI_nod_to_NIML_rows( nn , nod ) ;
   fprintf(stderr,"Clock time for %d nodes by rows = %d ms\n",
           nn,NI_clock_time()-ct) ;
   if( nn < 10 ){
     nb = NI_write_element( ns , nel2 , NI_TEXT_MODE ) ;
     fprintf(stderr,"num bytes=%d\n\n",nb) ;
   }
   /* NI_free_element(nel2) ; */

   /* put nodes into a NIML data element: by columns */

   ct = NI_clock_time() ;
   nel3 = AGNI_nod_to_NIML_col( nn , nod ) ;
   fprintf(stderr,"Clock time for %d nodes by col = %d ms\n",
           nn,NI_clock_time()-ct) ;
   if( nn < 10 ){
     nb = NI_write_element( ns , nel3 , NI_TEXT_MODE ) ;
     fprintf(stderr,"num bytes=%d\n\n",nb) ;
   }
   /* NI_free_element(nel3) ; */

   /* play with ni_group */

   if( nn < 5 ){
      NI_group *ngr = NI_new_group_element() ;
      NI_add_to_group( ngr , nel1 ) ;
      NI_add_to_group( ngr , nel2 ) ;
      NI_add_to_group( ngr , nel3 ) ;
      nb = NI_write_element( ns , ngr , NI_BASE64_MODE ) ;
      fprintf(stderr,"num bytes=%d\n\n",nb) ;
      NI_free_element(ngr) ;
   }

   fprintf(stderr,"\n*** That's all folks ***\n") ;
   exit(0) ;
}

/*------------------------------------------------------------------*/
/*! Make a NIML data element from an array of AGNI_nod structs.
    Return value is NULL if you input stupid values.
--------------------------------------------------------------------*/

NI_element * AGNI_nod_to_NIML_row( int num_nod , AGNI_nod *nod )
{
   NI_element *nel ;
   int ii ;

   /* check inputs for sanity */

   if( num_nod < 1 || nod == NULL ) return NULL ;

   /* make a new data element, to be filled by rows */

   nel = NI_new_data_element( "surfixyz_row" , -1 ) ;

   /* define the mapping from struct components to element columns */

   NI_define_rowmap_VA( nel ,
                          NI_INT   , offsetof(AGNI_nod,id) ,
                          NI_FLOAT , offsetof(AGNI_nod,x)  ,
                          NI_FLOAT , offsetof(AGNI_nod,y)  ,
                          NI_FLOAT , offsetof(AGNI_nod,z)  ,
                        -1 ) ;

   /* load the structs, one row at a time */

   for( ii=0 ; ii < num_nod ; ii++ )
      NI_add_row( nel , nod+ii ) ;

   return nel ;
}

/*------------------------------------------------------------------*/
/*! Make a NIML data element from an array of AGNI_nod structs.
    Return value is NULL if you input stupid values.
--------------------------------------------------------------------*/

NI_element * AGNI_nod_to_NIML_rows( int num_nod , AGNI_nod *nod )
{
   NI_element *nel ;
   int ii ;
   int *ic ; float *xc,*yc,*zc ;

   /* check inputs for sanity */

   if( num_nod < 1 || nod == NULL ) return NULL ;

   /* make a new data element, to be filled by rows */

   nel = NI_new_data_element( "surfixyz_rows" , -1 ) ;

   /* define the mapping from struct components to element columns */

   NI_define_rowmap_VA( nel ,
                          NI_INT   , offsetof(AGNI_nod,id) ,
                          NI_FLOAT , offsetof(AGNI_nod,x)  ,
                          NI_FLOAT , offsetof(AGNI_nod,y)  ,
                          NI_FLOAT , offsetof(AGNI_nod,z)  , -1 ) ;

   /* load all the rows at once */

   NI_add_many_rows( nel , num_nod , sizeof(AGNI_nod) , nod ) ;

   return nel ;
}

/*------------------------------------------------------------------*/
/*! Make a NIML data element from an array of AGNI_nod structs.
    Return value is NULL if you input stupid values.
--------------------------------------------------------------------*/

NI_element * AGNI_nod_to_NIML_col( int num_nod , AGNI_nod *nod )
{
   NI_element *nel ;
   int ii ;
   int *ic ; float *xc,*yc,*zc ;

   /* check inputs for sanity */

   if( num_nod < 1 || nod == NULL ) return NULL ;

   /* make a new data element, to be filled by columns */

   nel = NI_new_data_element( "surfixyz_col" , num_nod ) ;

   /* make the columns to be put in the element */

   ic = (int *)   malloc( sizeof(int)   * num_nod ) ;
   xc = (float *) malloc( sizeof(float) * num_nod ) ;
   yc = (float *) malloc( sizeof(float) * num_nod ) ;
   zc = (float *) malloc( sizeof(float) * num_nod ) ;

   /* load the columns from the struct array */

   for( ii=0 ; ii < num_nod ; ii++ ){
      ic[ii] = nod[ii].id ;
      xc[ii] = nod[ii].x ;
      yc[ii] = nod[ii].y ;
      zc[ii] = nod[ii].z ;
   }

   /* put columns into element */

   NI_add_column( nel , NI_INT   , ic ) ; free(ic) ;
   NI_add_column( nel , NI_FLOAT , xc ) ; free(xc) ;
   NI_add_column( nel , NI_FLOAT , yc ) ; free(yc) ;
   NI_add_column( nel , NI_FLOAT , zc ) ; free(zc) ;

   return nel ;
}

/*------------------------------------------------------------------*/
/*! Unload a <surfixyz> NI_element into an array of newly
    malloc()-ed AGNI_nod.
    Return value is number of nodes (will be 0 if an error occurs).
    *nod will point to the output array if the number of nodes is
    positive.
--------------------------------------------------------------------*/

int NIML_to_AGNI_nod( NI_element *nel , AGNI_nod **nod )
{
   int   num_nod , ii ;
   AGNI_nod *mynod ;

   /* check element for correctness */

   if( nel             == 0        ||   /* no data element?          */
       nel->vec_len    <  1        ||   /* empty element?             */
       nel->vec_filled <  1        ||   /* no data was filled in?      */
       nel->vec_num    <  4        ||   /* less than 4 columns?         */
       nel->vec_typ[0] != NI_INT   ||   /* must be int,float,float,float */
       nel->vec_typ[1] != NI_FLOAT ||
       nel->vec_typ[2] != NI_FLOAT ||
       nel->vec_typ[3] != NI_FLOAT   ) return 0 ;

   /* number of nodes is number of completely filled rows */

   num_nod = nel->vec_filled ;

   /* make space for the structs */

   mynod = (AGNI_nod *) malloc( sizeof(AGNI_nod) * num_nod ) ;

   /* define the mapping from struct components to element columns */

   NI_define_rowmap_VA( nel ,
                          NI_INT   , offsetof(AGNI_nod,id) ,
                          NI_FLOAT , offsetof(AGNI_nod,x)  ,
                          NI_FLOAT , offsetof(AGNI_nod,y)  ,
                          NI_FLOAT , offsetof(AGNI_nod,z)  ,
                        -1 ) ;

   /* load data from NIML element to the structs */

   for( ii=0 ; ii < num_nod ; ii++ )
      NI_get_row( nel , ii , mynod+ii ) ;

   /* we is done */

   *nod = mynod ; return num_nod ;
}