#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
//#define _GNU_SOURCE          /* See feature_test_macros(7) */
#include <unistd.h>
#include <dlfcn.h>
#include <omp.h>
#include <omp-tools.h>

//#define THREAD_ADDR 1

typedef struct omp_t_data {
        // Thread data
        ompt_state_t ompt_state;
        ompt_wait_id_t ompt_wait_id;
        int omp_thread_num;
        ompt_data_t* ompt_thread_data;
        // Parallel data
        int omp_num_threads;
        int omp_level;
        int omp_active_level;
        ompt_data_t* ompt_parallel_data;
        // Task data
        int omp_max_threads;
        int omp_parallel;
        int omp_final;
        int omp_dynamic;
        int omp_nested;
        int omp_max_active_levels;
        // omp_get_schedule
        omp_sched_t omp_kind;
        int omp_modifier;
        // omp_get_proc_bind
        omp_proc_bind_t omp_proc_bind;
        ompt_frame_t* ompt_frame_list;
        ompt_data_t* ompt_task_data;
//        int max_task_depth;
} omp_t_data_t;

__thread omp_t_data_t thread_data;


static ompt_function_lookup_t ompt_lookup;

// TODO: remove macro
// #define declare_inquery_fn(F) static F##_t F;
// FOREACH_OMPT_INQUIRY_FN(declare_inquery_fn)
// #undef declare_inquery_fn

static ompt_set_callback_t ompt_set_callback;
static ompt_get_callback_t ompt_get_callback;
static ompt_get_state_t ompt_get_state;
static ompt_get_task_info_t ompt_get_task_info;
static ompt_get_thread_data_t ompt_get_thread_data;
static ompt_get_parallel_info_t ompt_get_parallel_info;
static ompt_get_unique_id_t ompt_get_unique_id;
static ompt_get_num_procs_t ompt_get_num_procs;
static ompt_get_num_places_t ompt_get_num_places;
static ompt_get_place_proc_ids_t ompt_get_place_proc_ids;
static ompt_get_place_num_t ompt_get_place_num;
static ompt_get_partition_place_nums_t ompt_get_partition_place_nums;
static ompt_get_proc_id_t ompt_get_proc_id;
static ompt_enumerate_states_t ompt_enumerate_states;
static ompt_enumerate_mutex_impls_t ompt_enumerate_mutex_impls;
int checks=0;

static void
on_ompt_callback_implicit_task(
        ompt_scope_endpoint_t endpoint,
        ompt_data_t *parallel_data,
        ompt_data_t *task_data,
        unsigned int team_size,
        unsigned int thread_num, int flags
        )
{
        if (endpoint == ompt_scope_begin)
                task_data->value = ompt_get_unique_id();
}

static void
on_ompt_callback_thread_begin(
        ompt_thread_t thread_type,
        ompt_data_t *t_data)
{
        t_data->value = ompt_get_unique_id();
}

static void on_ompt_callback_parallel_begin(
        ompt_data_t *encountering_task_data,
        const ompt_frame_t *encountering_task_frame, ompt_data_t *parallel_data,
        uint32_t requested_team_size, int flag,
        const void *codeptr_ra)
{
        parallel_data->value = ompt_get_unique_id();
}

#define register_callback_t(name, type)                       \
do{                                                           \
  type f_##name = &on_##name;                                 \
  if (ompt_set_callback(name, (ompt_callback_t)f_##name) ==   \
      ompt_set_never)                                         \
    printf("0: Could not register callback '" #name "'\n");   \
}while(0)

#define register_callback(name) register_callback_t(name, name##_t)


int ompt_initialize(
  ompt_function_lookup_t lookup,
  int initial_device_num,
  ompt_data_t *tool_data)
{
        ompt_lookup = lookup;
        // TODO: remove: printf("runtime_version: %s, omp_version: %i\n", runtime_version, omp_version);

        // TODO: remove macro
        // #define declare_inquery_fn(F) F = (F##_t)lookup(#F);
        // FOREACH_OMPT_INQUIRY_FN(declare_inquery_fn)
        // #undef declare_inquery_fn

        ompt_set_callback_t ompt_set_callback = (ompt_set_callback_t) lookup("ompt_set_callback");
        ompt_get_callback = (ompt_get_callback_t) lookup("ompt_get_callback");
        ompt_get_state = (ompt_get_state_t) lookup("ompt_get_state");
        ompt_get_task_info = (ompt_get_task_info_t) lookup("ompt_get_task_info");
        ompt_get_thread_data = (ompt_get_thread_data_t) lookup("ompt_get_thread_data");
        ompt_get_parallel_info = (ompt_get_parallel_info_t) lookup("ompt_get_parallel_info");
        ompt_get_unique_id = (ompt_get_unique_id_t) lookup("ompt_get_unique_id");

        ompt_get_num_procs = (ompt_get_num_procs_t) lookup("ompt_get_num_procs");
        ompt_get_num_places = (ompt_get_num_places_t) lookup("ompt_get_num_places");
        ompt_get_place_proc_ids = (ompt_get_place_proc_ids_t) lookup("ompt_get_place_proc_ids");
        ompt_get_place_num = (ompt_get_place_num_t) lookup("ompt_get_place_num");
        ompt_get_partition_place_nums = (ompt_get_partition_place_nums_t) lookup("ompt_get_partition_place_nums");
        ompt_get_proc_id = (ompt_get_proc_id_t) lookup("ompt_get_proc_id");
        ompt_enumerate_states = (ompt_enumerate_states_t) lookup("ompt_enumerate_states");
        ompt_enumerate_mutex_impls = (ompt_enumerate_mutex_impls_t) lookup("ompt_enumerate_mutex_impls");

        register_callback(ompt_callback_implicit_task);
        register_callback(ompt_callback_thread_begin);
        register_callback(ompt_callback_parallel_begin);

        return 1; // activate tool
}

void ompt_finalize(ompt_data_t *tool_data)
{
}

ompt_start_tool_result_t *ompt_start_tool(
        unsigned int omp_version,
        const char *runtime_version
)
{
        static ompt_start_tool_result_t ompt_start_tool_result = {&ompt_initialize,&ompt_finalize, 0};
        return &ompt_start_tool_result;
}

void collectParallelData(omp_t_data_t * data)
{
        data->omp_num_threads = omp_get_num_threads();
        data->omp_level = omp_get_level();
        data->omp_active_level = omp_get_active_level();
        ompt_get_parallel_info(0, &(data->ompt_parallel_data), NULL);
}

void collectTaskData(omp_t_data_t * data)
{
        data->omp_max_threads = omp_get_max_threads();
        data->omp_parallel = omp_in_parallel();
        data->omp_final = omp_in_final();
        data->omp_dynamic = omp_get_dynamic();
        data->omp_nested = omp_get_nested();
        data->omp_max_active_levels = omp_get_max_active_levels();
        omp_get_schedule(&(data->omp_kind), &(data->omp_modifier));
        data->omp_proc_bind = omp_get_proc_bind();
        ompt_get_task_info(0, NULL, &(data->ompt_task_data), &(data->ompt_frame_list), NULL, NULL);
}

void collectThreadData(omp_t_data_t * data)
{
        data->omp_thread_num = omp_get_thread_num();
        data->ompt_state = ompt_get_state(&(data->ompt_wait_id));
        data->ompt_thread_data = ompt_get_thread_data();
}

#ifdef  __cplusplus
extern "C"
#endif
__attribute__ ((noinline))
void* breakToolTest(omp_t_data_t * data){
        return data;
}

#ifdef  __cplusplus
extern "C"
#endif
void* ompd_tool_break(void* n){
        (void)n;
        asm("");
        return NULL;
}

void* ompd_tool_test(void* n)
{
        collectThreadData(&thread_data);
        collectParallelData(&thread_data);
        collectTaskData(&thread_data);
        breakToolTest(&thread_data);
        checks++;
        ompd_tool_break(NULL);
        return NULL;
}


__attribute__ ((__constructor__))
void init(void)
{
}

__attribute__ ((__destructor__))
void fini(void)
{
        printf("Finished %i testsuites.\n",checks);
}