/****************************/

/* 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

	 for(j=0;j

	   a[i][j]=(double)i*(double)j;

	   b[i][j]=(double)i/(double)(j+5);

	 }

       }

       for(j=0;j

	 for(i=0;i

	   s=0;

	   for(k=0;k

	     s+=a[i][k]*b[k][j];

	   }

	   c[i][j] = s;

	 }

       }

       s=0.0;

       for(i=0;i

	 	for(j=0;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

	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

		   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

		   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

		   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

		   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

		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

		   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

		   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;

}