/****************************/
/* THIS IS OPEN SOURCE CODE */
/****************************/

/**
 * @author  Vince Weaver
 *
 * test case for RAPL component
 *
 * @brief
 *   Tests basic functionality of RAPL component
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#include "papi.h"
#include "papi_test.h"

#define MAX_RAPL_EVENTS 64


#ifdef BASIC_TEST

void run_test(int quiet) {

     if (!quiet) {
	printf("Sleeping 1 second...\n");
     }

     /* Sleep */
     sleep(1);
}

#else

#define MATRIX_SIZE 1024

     static double a[MATRIX_SIZE][MATRIX_SIZE];
     static double b[MATRIX_SIZE][MATRIX_SIZE];
     static double c[MATRIX_SIZE][MATRIX_SIZE];

/* Naive matrix multiply */
void run_test(int quiet) {

       double s;
       int i,j,k;

       if (!quiet) {
	 		printf("Doing a naive %dx%d MMM...\n",MATRIX_SIZE,MATRIX_SIZE);
       }

       for(i=0;i<MATRIX_SIZE;i++) {
	 for(j=0;j<MATRIX_SIZE;j++) {
	   a[i][j]=(double)i*(double)j;
	   b[i][j]=(double)i/(double)(j+5);
	 }
       }

       for(j=0;j<MATRIX_SIZE;j++) {
	 for(i=0;i<MATRIX_SIZE;i++) {
	   s=0;
	   for(k=0;k<MATRIX_SIZE;k++) {
	     s+=a[i][k]*b[k][j];
	   }
	   c[i][j] = s;
	 }
       }

       s=0.0;
       for(i=0;i<MATRIX_SIZE;i++) {
	 	for(j=0;j<MATRIX_SIZE;j++) {
	   		s+=c[i][j];
	 	}
       }

       if (!quiet) printf("Matrix multiply sum: s=%lf\n",s);
}

#endif

int main (int argc, char **argv)
{

    int retval,cid,rapl_cid=-1,numcmp;
    int EventSet = PAPI_NULL;
    long long *values;
    int num_events=0;
    int code;
    char event_names[MAX_RAPL_EVENTS][PAPI_MAX_STR_LEN];
    char units[MAX_RAPL_EVENTS][PAPI_MIN_STR_LEN];
    int data_type[MAX_RAPL_EVENTS];
    int r,i;
    const PAPI_component_info_t *cmpinfo = NULL;
    PAPI_event_info_t evinfo;
    long long before_time,after_time;
    double elapsed_time;

#ifdef WRAP_TEST

	int do_wrap=0;

	if ( argc > 1 ) {
		if ( strstr( argv[1], "-w" ) ) {
			do_wrap = 1;
		}
	}

#endif

        /* Set TESTS_QUIET variable */
     tests_quiet( argc, argv );

	/* PAPI Initialization */
     retval = PAPI_library_init( PAPI_VER_CURRENT );
     if ( retval != PAPI_VER_CURRENT ) {
	test_fail(__FILE__, __LINE__,"PAPI_library_init failed\n",retval);
     }

     if (!TESTS_QUIET) {
		printf("Trying all RAPL events\n");
     }

     numcmp = PAPI_num_components();

     for(cid=0; cid<numcmp; cid++) {

	if ( (cmpinfo = PAPI_get_component_info(cid)) == NULL) {
	   test_fail(__FILE__, __LINE__,"PAPI_get_component_info failed\n", 0);
	}

	if (strstr(cmpinfo->name,"rapl")) {

	   rapl_cid=cid;

	   if (!TESTS_QUIET) {
	      printf("Found rapl component at cid %d\n",rapl_cid);
	   }

           if (cmpinfo->disabled) {
	      if (!TESTS_QUIET) {
		 printf("RAPL component disabled: %s\n",
                        cmpinfo->disabled_reason);
	      }
              test_skip(__FILE__,__LINE__,"RAPL component disabled",0);
           }
	   break;
	}
     }

     /* Component not found */
     if (cid==numcmp) {
       test_skip(__FILE__,__LINE__,"No rapl component found\n",0);
     }

     /* Create EventSet */
     retval = PAPI_create_eventset( &EventSet );
     if (retval != PAPI_OK) {
	test_fail(__FILE__, __LINE__,
                              "PAPI_create_eventset()",retval);
     }

     /* Add all events */

     code = PAPI_NATIVE_MASK;

     r = PAPI_enum_cmp_event( &code, PAPI_ENUM_FIRST, rapl_cid );

     while ( r == PAPI_OK ) {

        retval = PAPI_event_code_to_name( code, event_names[num_events] );
	if ( retval != PAPI_OK ) {
	   printf("Error translating %#x\n",code);
	   test_fail( __FILE__, __LINE__,
                            "PAPI_event_code_to_name", retval );
	}

	retval = PAPI_get_event_info(code,&evinfo);
	if (retval != PAPI_OK) {
	  test_fail( __FILE__, __LINE__,
             "Error getting event info\n",retval);
	}

	strncpy(units[num_events],evinfo.units,sizeof(units[0])-1);
	// buffer must be null terminated to safely use strstr operation on it below
	units[num_events][sizeof(units[0])-1] = '\0';

	data_type[num_events] = evinfo.data_type;

        retval = PAPI_add_event( EventSet, code );
        if (retval != PAPI_OK) {
	  break; /* We've hit an event limit */
	}
	num_events++;

        r = PAPI_enum_cmp_event( &code, PAPI_ENUM_EVENTS, rapl_cid );
     }

     values=calloc(num_events,sizeof(long long));
     if (values==NULL) {
	test_fail(__FILE__, __LINE__,
                              "No memory",retval);
     }

     if (!TESTS_QUIET) {
	printf("\nStarting measurements...\n\n");
     }

     /* Start Counting */
     before_time=PAPI_get_real_nsec();
     retval = PAPI_start( EventSet);
     if (retval != PAPI_OK) {
	test_fail(__FILE__, __LINE__, "PAPI_start()",retval);
     }

     /* Run test */
     run_test(TESTS_QUIET);

     /* Stop Counting */
     after_time=PAPI_get_real_nsec();
     retval = PAPI_stop( EventSet, values);
     if (retval != PAPI_OK) {
	test_fail(__FILE__, __LINE__, "PAPI_stop()",retval);
     }

     elapsed_time=((double)(after_time-before_time))/1.0e9;

     if (!TESTS_QUIET) {
        printf("\nStopping measurements, took %.3fs, gathering results...\n\n",
	       elapsed_time);

		printf("Scaled energy measurements:\n");

		for(i=0;i<num_events;i++) {
		   if (strstr(units[i],"nJ")) {

			  printf("%-40s%12.6f J\t(Average Power %.1fW)\n",
				event_names[i],
				(double)values[i]/1.0e9,
				((double)values[i]/1.0e9)/elapsed_time);
		   }
		}

		printf("\n");
		printf("Energy measurement counts:\n");

		for(i=0;i<num_events;i++) {
		   if (strstr(event_names[i],"ENERGY_CNT")) {
			  printf("%-40s%12lld\t%#08llx\n", event_names[i], values[i], values[i]);
		   }
		}

		printf("\n");
		printf("Scaled Fixed values:\n");

		for(i=0;i<num_events;i++) {
		   if (!strstr(event_names[i],"ENERGY")) {
			 if (data_type[i] == PAPI_DATATYPE_FP64) {

				 union {
				   long long ll;
				   double fp;
				 } result;

				result.ll=values[i];
				printf("%-40s%12.3f %s\n", event_names[i], result.fp, units[i]);
			  }
		   }
		}

		printf("\n");
		printf("Fixed value counts:\n");

		for(i=0;i<num_events;i++) {
		   if (!strstr(event_names[i],"ENERGY")) {
			  if (data_type[i] == PAPI_DATATYPE_UINT64) {
				printf("%-40s%12lld\t%#08llx\n", event_names[i], values[i], values[i]);
			  }
		   }
		}

     }

#ifdef WRAP_TEST
	double max_time;
	unsigned long long max_value = 0;
	int repeat;

	for(i=0;i<num_events;i++) {
		if (strstr(event_names[i],"ENERGY_CNT")) {
			if (max_value < (unsigned) values[i]) {
				max_value = values[i];
		  	}
		}
	}
	max_time = elapsed_time * ( (double)0xffffffff / (double)max_value );
	printf("\n");
	printf ("Approximate time to energy measurement wraparound: %.3f sec or %.3f min.\n", 
		max_time, max_time/60);

	if (do_wrap) {
		 printf ("Beginning wraparound execution.");
	     /* Start Counting */
		 before_time=PAPI_get_real_nsec();
		 retval = PAPI_start( EventSet);
		 if (retval != PAPI_OK) {
			test_fail(__FILE__, __LINE__, "PAPI_start()",retval);
		 }

		 /* Run test */
		repeat = (int)(max_time/elapsed_time);
		for (i=0;i< repeat;i++) {
			run_test(1);
			printf("."); fflush(stdout);
		}
		printf("\n");

		 /* Stop Counting */
		 after_time=PAPI_get_real_nsec();
		 retval = PAPI_stop( EventSet, values);
		 if (retval != PAPI_OK) {
			test_fail(__FILE__, __LINE__, "PAPI_stop()",retval);
		 }

		elapsed_time=((double)(after_time-before_time))/1.0e9;
		printf("\nStopping measurements, took %.3fs\n\n", elapsed_time);

		printf("Scaled energy measurements:\n");

		for(i=0;i<num_events;i++) {
		   if (strstr(units[i],"nJ")) {

			  printf("%-40s%12.6f J\t(Average Power %.1fW)\n",
				event_names[i],
				(double)values[i]/1.0e9,
				((double)values[i]/1.0e9)/elapsed_time);
		   }
		}
		printf("\n");
		printf("Energy measurement counts:\n");

		for(i=0;i<num_events;i++) {
		   if (strstr(event_names[i],"ENERGY_CNT")) {
			  printf("%-40s%12lld\t%#08llx\n", event_names[i], values[i], values[i]);
		   }
		}
	}

#endif

     /* Done, clean up */
     retval = PAPI_cleanup_eventset( EventSet );
     if (retval != PAPI_OK) {
	test_fail(__FILE__, __LINE__,
                              "PAPI_cleanup_eventset()",retval);
     }

     retval = PAPI_destroy_eventset( &EventSet );
     if (retval != PAPI_OK) {
	test_fail(__FILE__, __LINE__,
                              "PAPI_destroy_eventset()",retval);
     }

     test_pass( __FILE__ );

     return 0;
}