/*

* File:    zero_omp.c

* Author:  Philip Mucci

*          mucci@cs.utk.edu

* Mods:    Nils Smeds

*          smeds@pdc.kth.se

*          Anders Nilsson

*          anni@pdc.kth.se

*/

/* This file performs the following test: start, stop and timer

functionality for 2 slave OMP threads

   - It attempts to use the following two counters. It may use less

depending on hardware counter resource limitations. These are counted

in the default counting domain and default granularity, depending on

the platform. Usually this is the user domain (PAPI_DOM_USER) and

thread context (PAPI_GRN_THR).

     + PAPI_FP_INS

     + PAPI_TOT_CYC

Each thread inside the Thread routine:

   - Get cyc.

   - Get us.

   - Start counters

   - Do flops

   - Stop and read counters

   - Get us.

   - Get cyc.

Master serial thread:

   - Get us.

   - Get cyc.

   - Run parallel for loop

   - Get us.

   - Get cyc.

*/

#include <stdio.h>

#include <stdlib.h>

#include "papi.h"

#include "papi_test.h"

#include "do_loops.h"

#ifdef _OPENMP

#include <omp.h>

#else

#error "This compiler does not understand OPENMP"

#endif

const PAPI_hw_info_t *hw_info = NULL;

void

Thread( int n )

{

	int retval, num_tests = 1;

	int EventSet1 = PAPI_NULL;

	int PAPI_event, mask1;

	int num_events1;

	long long **values;

	long long elapsed_us, elapsed_cyc;

	char event_name[PAPI_MAX_STR_LEN];

	if (!TESTS_QUIET) {

		printf( "Thread %#x started\n", omp_get_thread_num(  ) );

	}

	/* add PAPI_TOT_CYC and one of the events in

	   PAPI_FP_INS, PAPI_FP_OPS or PAPI_TOT_INS,

	   depending on the availability of the event

	   on the platform */

	EventSet1 = add_two_events( &num_events1, &PAPI_event, &mask1 );

	if (num_events1==0) {

		if (!TESTS_QUIET) printf("No events added!\n");

		test_fail(__FILE__,__LINE__,"No events",0);

	}

	retval = PAPI_event_code_to_name( PAPI_event, event_name );

	if ( retval != PAPI_OK )

		test_fail( __FILE__, __LINE__, "PAPI_event_code_to_name", retval );

	values = allocate_test_space( num_tests, num_events1 );

	elapsed_us = PAPI_get_real_usec(  );

	elapsed_cyc = PAPI_get_real_cyc(  );

	retval = PAPI_start( EventSet1 );

	if ( retval != PAPI_OK )

		test_fail( __FILE__, __LINE__, "PAPI_start", retval );

	do_flops( n );

	retval = PAPI_stop( EventSet1, values[0] );

	if ( retval != PAPI_OK )

		test_fail( __FILE__, __LINE__, "PAPI_stop", retval );

	elapsed_us = PAPI_get_real_usec(  ) - elapsed_us;

	elapsed_cyc = PAPI_get_real_cyc(  ) - elapsed_cyc;

	remove_test_events( &EventSet1, mask1 );

	if ( !TESTS_QUIET ) {

		printf( "Thread %#x %-12s : \t%lld\n", omp_get_thread_num(  ),

				event_name, values[0][1] );

		printf( "Thread %#x PAPI_TOT_CYC: \t%lld\n", omp_get_thread_num(  ),

				values[0][0] );

		printf( "Thread %#x Real usec   : \t%lld\n", omp_get_thread_num(  ),

				elapsed_us );

		printf( "Thread %#x Real cycles : \t%lld\n", omp_get_thread_num(  ),

				elapsed_cyc );

	}

	/* It is illegal for the threads to exit in OpenMP */

	/* test_pass(__FILE__,0,0); */

	free_test_space( values, num_tests );

	PAPI_unregister_thread(  );

	if (!TESTS_QUIET) {

		printf( "Thread %#x finished\n", omp_get_thread_num(  ) );

	}

}

int

main( int argc, char **argv )

{

	int retval;

	long long elapsed_us, elapsed_cyc;

	int quiet;

	/* Set TESTS_QUIET variable */

	quiet = tests_quiet( argc, argv );

	retval = PAPI_library_init( PAPI_VER_CURRENT );

	if ( retval != PAPI_VER_CURRENT ) {

		test_fail( __FILE__, __LINE__, "PAPI_library_init", retval );

	}

	hw_info = PAPI_get_hardware_info(  );

	if ( hw_info == NULL ) {

		test_fail( __FILE__, __LINE__, "PAPI_get_hardware_info", 2 );

	}

	if (PAPI_query_event(PAPI_TOT_INS)!=PAPI_OK) {

		if (!quiet) printf("Can't find PAPI_TOT_INS\n");

		test_skip(__FILE__,__LINE__,"Event missing",1);

	}

	if (PAPI_query_event(PAPI_TOT_CYC)!=PAPI_OK) {

		if (!quiet) printf("Can't find PAPI_TOT_CYC\n");

		test_skip(__FILE__,__LINE__,"Event missing",1);

	}

	elapsed_us = PAPI_get_real_usec(  );

	elapsed_cyc = PAPI_get_real_cyc(  );

	retval = PAPI_thread_init( ( unsigned long ( * )( void ) )

						( omp_get_thread_num ) );

	if ( retval != PAPI_OK ) {

		if ( retval == PAPI_ECMP ) {

			if (!quiet) printf("Trouble init threads\n");

			test_skip( __FILE__, __LINE__, "PAPI_thread_init", retval );

		}

		else {

			test_fail( __FILE__, __LINE__, "PAPI_thread_init", retval );

		}

	}

#pragma omp parallel

	{

		Thread( 1000000 * ( omp_get_thread_num(  ) + 1 ) );

	}

	omp_set_num_threads( 1 );

	Thread( 1000000 * ( omp_get_thread_num(  ) + 1 ) );

	omp_set_num_threads( omp_get_max_threads(  ) );

#pragma omp parallel

	{

		Thread( 1000000 * ( omp_get_thread_num(  ) + 1 ) );

	}

	elapsed_cyc = PAPI_get_real_cyc(  ) - elapsed_cyc;

	elapsed_us = PAPI_get_real_usec(  ) - elapsed_us;

	if ( !TESTS_QUIET ) {

		printf( "Master real usec   : \t%lld\n", elapsed_us );

		printf( "Master real cycles : \t%lld\n", elapsed_cyc );

	}

	test_pass( __FILE__ );

	return 0;

}