/*
 * This file tests uncore events on AMD fam15h Northbridge machines
 *  The Linux perf_event developers introduced fam15h Northbridge
 *   support in Linux 3.9 with an interfae similar to fam10h
 *   where the events were part of the core CPU
 *  They broke the ABI with Linux 3.10 and made fam15h NB a separate
 *   PMU, like the Intel uncore support.
 */

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

#include <sys/utsname.h>

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

#include "do_loops.h"

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

	int retval;
	int EventSet = PAPI_NULL;
	long long values[1];
	char event_name[BUFSIZ];
	int uncore_cidx=-1;
	const PAPI_hw_info_t *hwinfo;
	int quiet;
	struct utsname uname_info;

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

	uname(&uname_info);
	if (!quiet) printf("Found Linux %s\n",uname_info.release);

	/* Init the PAPI library */
	retval = PAPI_library_init( PAPI_VER_CURRENT );
	if ( retval != PAPI_VER_CURRENT ) {
		test_fail( __FILE__, __LINE__, "PAPI_library_init", retval );
	}

	/* Check for AMD machine */
	hwinfo = PAPI_get_hardware_info();
	if ( hwinfo == NULL ) {
		test_fail( __FILE__, __LINE__, "PAPI_get_hardware_info", 0 );
	}

	if (hwinfo->vendor != PAPI_VENDOR_AMD) {
		if (!quiet) printf("Test only for AMD machines\n");
		test_skip(__FILE__,__LINE__,"Test only for AMD processor",0);
	}

	if ( hwinfo->cpuid_family != 21) {
		if (!quiet) printf("Test only for fam15h AMD machines\n");
		test_skip(__FILE__,__LINE__,"Test only for fam15h AMD processor",0);
	}

	if (!strcmp(uname_info.release,"3.9")) {

		if (!quiet) printf("Detected 3.9 kernel, using perf_event\n");

		/* For kernel 3.9 use regular CPU component */

		/* Find the uncore PMU */
		uncore_cidx=PAPI_get_component_index("perf_event");
		if (uncore_cidx<0) {
			test_skip(__FILE__,__LINE__,"perf_event component not found",0);
		}

		/* Get a relevant event name */
		strncpy(event_name,"DRAM_ACCESSES:ALL", BUFSIZ);
	}
	else {

		/* 3.10 and later */

		if (!quiet) {
			printf("Detected > 3.9 kernel, using perf_event_uncore\n");
		}

		/* Find the uncore PMU */
		uncore_cidx=PAPI_get_component_index("perf_event_uncore");
		if (uncore_cidx<0) {
			test_skip(__FILE__,__LINE__,"perf_event_uncore component not found",0);
		}

		/* Get a relevant event name */
		/* This might change once libpfm4 gets new fam15h NB support */
		strncpy(event_name,"DRAM_ACCESSES:ALL", BUFSIZ);
	}

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

	/* Set a component for the EventSet */
	retval = PAPI_assign_eventset_component(EventSet, uncore_cidx);

	/* we need to set to a certain cpu for uncore to work */

	PAPI_cpu_option_t cpu_opt;

	cpu_opt.eventset=EventSet;
	cpu_opt.cpu_num=0;

	retval = PAPI_set_opt(PAPI_CPU_ATTACH,(PAPI_option_t*)&cpu_opt);
	if (retval != PAPI_OK) {
		test_skip( __FILE__, __LINE__,
			"this test; trying to PAPI_CPU_ATTACH; need to run as root",
			retval);
	}

	/* we need to set the granularity to system-wide for uncore to work */

	PAPI_granularity_option_t gran_opt;

	gran_opt.def_cidx=0;
	gran_opt.eventset=EventSet;
	gran_opt.granularity=PAPI_GRN_SYS;

	retval = PAPI_set_opt(PAPI_GRANUL,(PAPI_option_t*)&gran_opt);
	if (retval != PAPI_OK) {
		test_skip( __FILE__, __LINE__,
			"this test; trying to set PAPI_GRN_SYS",
			retval);
	}

	/* we need to set domain to be as inclusive as possible */

	PAPI_domain_option_t domain_opt;

	domain_opt.def_cidx=0;
	domain_opt.eventset=EventSet;
	domain_opt.domain=PAPI_DOM_ALL;

	retval = PAPI_set_opt(PAPI_DOMAIN,(PAPI_option_t*)&domain_opt);
	if (retval != PAPI_OK) {
		test_skip( __FILE__, __LINE__,
			"this test; trying to set PAPI_DOM_ALL; need to run as root",
			retval);
	}

	/* Add our uncore event */
	retval = PAPI_add_named_event(EventSet, event_name);
	if (retval != PAPI_OK) {
		if ( !quiet ) {
			fprintf(stderr,"Error trying to use event %s\n", event_name);
		}
		test_fail(__FILE__, __LINE__, "adding uncore event",retval);
	}


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

	/* our work code */
	do_flops( NUM_FLOPS );

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

	if ( !quiet ) {
		printf("AMD fam15h Northbridge test:\n");
		printf("Using event %s\n",event_name);
		printf("\t%s: %lld\n",event_name,values[0]);
	}

	test_pass( __FILE__ );

	return 0;
}