/** \file \internal
Basic diskless API tests.
   
Copyright 2011, UCAR/Unidata. See COPYRIGHT file for copying and
redistribution conditions.
*/

#include <config.h>
#include <nc_tests.h>
#include <stdio.h>
#include <stdlib.h>
#include <netcdf.h>

#define FILE_NAME "tst_diskless.nc"

int
main(int argc, char **argv) 
{
   printf("\n*** Testing the diskless API.\n");
   printf("*** testing diskless file with scalar vars...");
   {
#define RESISTOR "resistor_value"
#define CAPACITOR "capacitor_value"    
#define NUM555 "number_of_555_timer_chips"

      int ncid, varid0, varid1, varid2;
      int ndims_in, nvars_in, natts_in, unlimdimid_in;
      char name_in[NC_MAX_NAME + 1];
      nc_type type_in;
      size_t len_in;
      int i;
      float float_data = 3.14, float_data_in;
      int int_data = 42, int_data_in;
      short short_data = 2, short_data_in;

      /* Create a netCDF file (which exists only in memory). */
      if (nc_create(FILE_NAME, NC_DISKLESS|NC_NETCDF4|NC_CLASSIC_MODEL, 
		    &ncid)) ERR;

      /* Create some variables. */
      if (nc_def_var(ncid, RESISTOR, NC_INT, 0, NULL, &varid0)) ERR;
      if (nc_def_var(ncid, CAPACITOR, NC_FLOAT, 0, NULL, &varid1)) ERR;
      if (nc_def_var(ncid, NUM555, NC_SHORT, 0, NULL, &varid2)) ERR;

      /* Write some data to this file. */
      if (nc_put_vara_int(ncid, varid0, NULL, NULL, &int_data)) ERR;
      if (nc_put_vara_float(ncid, varid1, NULL, NULL, &float_data)) ERR;
      if (nc_put_vara_short(ncid, varid2, NULL, NULL, &short_data)) ERR;

      /* Now check the phony file. */
      if (nc_inq(ncid, &ndims_in, &nvars_in, &natts_in, &unlimdimid_in)) ERR;
      if (ndims_in != 0 || nvars_in != 3 || natts_in != 0 || unlimdimid_in != -1) ERR;

      /* Check variables. */
      if (nc_inq_var(ncid, varid0, name_in, &type_in, &ndims_in, NULL, &natts_in)) ERR;
      if (strcmp(name_in, RESISTOR) || type_in != NC_INT || ndims_in != 0 ||
	  natts_in != 0) ERR;
      if (nc_inq_var(ncid, varid1, name_in, &type_in, &ndims_in, NULL, &natts_in)) ERR;
      if (strcmp(name_in, CAPACITOR) || type_in != NC_FLOAT || ndims_in != 0 ||
	  natts_in != 0) ERR;
      if (nc_inq_var(ncid, varid2, name_in, &type_in, &ndims_in, NULL, &natts_in)) ERR;
      if (strcmp(name_in, NUM555) || type_in != NC_SHORT || natts_in != 0) ERR;

      /* Read my absolutely crucial data. */
      if (nc_get_vara_int(ncid, varid0, NULL, NULL, &int_data_in)) ERR;
      if (int_data_in != int_data) ERR;
      if (nc_get_vara_float(ncid, varid1, NULL, NULL, &float_data_in)) ERR;
      if (float_data_in != float_data) ERR;
      if (nc_get_vara_short(ncid, varid2, NULL, NULL, &short_data_in)) ERR;
      if (short_data_in != short_data) ERR;

      /* Close the file. */
      if (nc_close(ncid)) ERR;
   }
   SUMMARIZE_ERR;
   printf("*** testing creation of simple diskless file...");
   {
#define NDIMS 2
#define DIM0_NAME "Fun"
#define DIM1_NAME "Money"
#define DIM1_LEN 200
#define ATT0_NAME "home"
#define ATT0_TEXT "earthship"  
#define VAR0_NAME "nightlife"
#define VAR1_NAME "time"    
#define VAR2_NAME "taxi_distance"

      int ncid, dimid[NDIMS], dimid_in[NDIMS], varid0, varid1, varid2;
      int ndims_in, nvars_in, natts_in, unlimdimid_in;
      char name_in[NC_MAX_NAME + 1], att0_in[NC_MAX_NAME + 1];
      nc_type type_in;
      size_t len_in;
      short short_data[DIM1_LEN];
      size_t start[1] = {0};
      size_t count[1] = {DIM1_LEN};
      int i;
      float float_data = 42.22, float_data_in;

      /* This is some really important data that I want to save. */
      for (i = 0; i < DIM1_LEN; i++)
	 short_data[i] = i;
      
      /* Create a netCDF file (which exists only in memory). I am
       * confident that the world-famous netCDF format is the way to
       * store my data! */
      if (nc_create(FILE_NAME, NC_DISKLESS|NC_NETCDF4|NC_CLASSIC_MODEL, &ncid)) ERR;

      /* Create some atts. They will help document my data forever. */
      if (nc_put_att_text(ncid, NC_GLOBAL, ATT0_NAME, 
			  sizeof(ATT0_TEXT) + 1, ATT0_TEXT)) ERR;

      /* Create dimensions: money is limited, but fun is not! */
      if (nc_def_dim(ncid, DIM0_NAME, NC_UNLIMITED, &dimid[0])) ERR;
      if (nc_def_dim(ncid, DIM1_NAME, DIM1_LEN, &dimid[1])) ERR;

      /* Create some variables. The data they hold must persist
       * through the ages. */
      if (nc_def_var(ncid, VAR0_NAME, NC_INT, NDIMS, dimid, &varid0)) ERR;
      if (nc_def_var(ncid, VAR1_NAME, NC_FLOAT, 0, NULL, &varid1)) ERR;
      if (nc_def_var(ncid, VAR2_NAME, NC_SHORT, 1, &dimid[1], &varid2)) ERR;

      /* Write some data to this file. I'm glad I'm saving this
       * important data in such a safe format! */
      if (nc_put_vara_float(ncid, varid1, NULL, NULL, &float_data)) ERR;
      if (nc_put_vara_short(ncid, varid2, start, count, short_data)) ERR;

      /* Now check the phony file. Is my data safe? */
      if (nc_inq(ncid, &ndims_in, &nvars_in, &natts_in, &unlimdimid_in)) ERR;
      if (ndims_in != 2 || nvars_in != 3 || natts_in != 1 || unlimdimid_in != 0) ERR;

      /* Check attributes - they will be needed by future generations
       * of scientists to understand my data. */
      if (nc_get_att_text(ncid, NC_GLOBAL, ATT0_NAME, att0_in)) ERR;
      if (strcmp(att0_in, ATT0_TEXT)) ERR;
      
      /* Check dimensions. */
      if (nc_inq_dim(ncid, dimid[0], name_in, &len_in)) ERR;
      if (strcmp(name_in, DIM0_NAME) || len_in != 0) ERR;
      if (nc_inq_dim(ncid, dimid[1], name_in, &len_in)) ERR;
      if (strcmp(name_in, DIM1_NAME) || len_in != DIM1_LEN) ERR;

      /* Check variables. */
      if (nc_inq_var(ncid, varid0, name_in, &type_in, &ndims_in, dimid_in, &natts_in)) ERR;
      if (strcmp(name_in, VAR0_NAME) || type_in != NC_INT || ndims_in != NDIMS ||
	  dimid_in[0] != 0 || dimid_in[1] != 1 || natts_in != 0) ERR;
      if (nc_inq_var(ncid, varid1, name_in, &type_in, &ndims_in, dimid_in, &natts_in)) ERR;
      if (strcmp(name_in, VAR1_NAME) || type_in != NC_FLOAT || ndims_in != 0 ||
	  natts_in != 0) ERR;
      if (nc_inq_var(ncid, varid2, name_in, &type_in, &ndims_in, dimid_in, &natts_in)) ERR;
      if (strcmp(name_in, VAR2_NAME) || type_in != NC_SHORT || ndims_in != 1 || 
	  dimid_in[0] != 1 || natts_in != 0) ERR;

      /* Read my absolutely crucial data. */
      if (nc_get_vara_float(ncid, varid1, NULL, NULL, &float_data_in)) ERR;
      if (float_data_in != float_data) ERR;

      /* Close the file, losing all information. Hey! What kind of
       * storage format is this, anyway? */
      if (nc_close(ncid)) ERR;
   }
   SUMMARIZE_ERR;
   printf("*** testing diskless file with scalar vars and type conversion...");
   {
#define DUNE "dune"
#define STAR_TREK "capacitor_value"    
#define STAR_WARS "number_of_555_timer_chips"

      int ncid, varid0, varid1, varid2;
      int ndims_in, nvars_in, natts_in, unlimdimid_in;
      char name_in[NC_MAX_NAME + 1];
      nc_type type_in;
      size_t len_in;
      int i;
      float float_data = 3.14, float_data_in;
      int int_data = 42, int_data_in;
      short short_data = 2, short_data_in;

      /* Create a netCDF file (which exists only in memory). */
      if (nc_create(FILE_NAME, NC_DISKLESS|NC_NETCDF4|NC_CLASSIC_MODEL, 
		    &ncid)) ERR;

      /* Create some variables. */
      if (nc_def_var(ncid, DUNE, NC_INT, 0, NULL, &varid0)) ERR;
      if (nc_def_var(ncid, STAR_TREK, NC_FLOAT, 0, NULL, &varid1)) ERR;
      if (nc_def_var(ncid, STAR_WARS, NC_SHORT, 0, NULL, &varid2)) ERR;

      /* Write some data to this file. */
      if (nc_put_vara_int(ncid, varid0, NULL, NULL, &int_data)) ERR;
      if (nc_put_vara_float(ncid, varid1, NULL, NULL, &float_data)) ERR;
      if (nc_put_vara_short(ncid, varid2, NULL, NULL, &short_data)) ERR;

      /* Now check the phony file. */
      if (nc_inq(ncid, &ndims_in, &nvars_in, &natts_in, &unlimdimid_in)) ERR;
      if (ndims_in != 0 || nvars_in != 3 || natts_in != 0 || unlimdimid_in != -1) ERR;

      /* Check variables. */
      if (nc_inq_var(ncid, varid0, name_in, &type_in, &ndims_in, NULL, &natts_in)) ERR;
      if (strcmp(name_in, DUNE) || type_in != NC_INT || ndims_in != 0 ||
	  natts_in != 0) ERR;
      if (nc_inq_var(ncid, varid1, name_in, &type_in, &ndims_in, NULL, &natts_in)) ERR;
      if (strcmp(name_in, STAR_TREK) || type_in != NC_FLOAT || ndims_in != 0 ||
	  natts_in != 0) ERR;
      if (nc_inq_var(ncid, varid2, name_in, &type_in, &ndims_in, NULL, &natts_in)) ERR;
      if (strcmp(name_in, STAR_WARS) || type_in != NC_SHORT || natts_in != 0) ERR;

      /* Read my absolutely crucial data. */
      if (nc_get_vara_int(ncid, varid0, NULL, NULL, &int_data_in)) ERR;
      if (int_data_in != int_data) ERR;
      if (nc_get_vara_float(ncid, varid1, NULL, NULL, &float_data_in)) ERR;
      if (float_data_in != float_data) ERR;
      if (nc_get_vara_short(ncid, varid2, NULL, NULL, &short_data_in)) ERR;
      if (short_data_in != short_data) ERR;

      /* Close the file. */
      if (nc_close(ncid)) ERR;
   }
   SUMMARIZE_ERR;
   FINAL_RESULTS;
}