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

/* Non standard include */
#include <specific_comm.h>
#include <commtech.h>


struct thread_comm *tcomms;
volatile int cont = 1;
static int init = 0;
static int error = 0;
static pthread_mutex_t init_lock = PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t init_cond = PTHREAD_COND_INITIALIZER;

int init_library(void)
{
	tcomms = (struct thread_comm *) malloc(nb_prod * sizeof(struct thread_comm));
	if (tcomms == NULL)
	{
		fprintf(stderr, "Failed to allocate %lu bytes needed by the library to work\n", nb_prod * sizeof(struct thread_comm));
		return -1;
	}
	return 0;
}

int end_library(void)
{
	free(tcomms);
	return 0;
}

int get_thread_number(void)
{
	static int i = 0;
	static pthread_mutex_t i_lock = PTHREAD_MUTEX_INITIALIZER;
	int i_local;

	pthread_mutex_lock(&i_lock);
	i_local = i;
	i++;
	pthread_mutex_unlock(&i_lock);
	return i_local;
}

int init_producer_thread(void)
{
	int thread_num;

	thread_num = get_thread_number();
	if (init_thread_comm(&tcomms[thread_num]))
	{
		pthread_mutex_lock(&init_lock);
		error = 1;
		pthread_cond_signal(&init_cond);
		pthread_mutex_unlock(&init_lock);
		return -1;
	}
	if (thread_num == nb_prod - 1)
	{
		pthread_mutex_lock(&init_lock);
		init = 1;
		pthread_cond_signal(&init_cond);
		pthread_mutex_unlock(&init_lock);
	}
	return 0;
}

int end_producer_thread(void)
{
	return end_thread_comm();
}

int wait_initialization(void)
{
	pthread_mutex_lock(&init_lock);
	if (!init && !error)
		pthread_cond_wait(&init_cond, &init_lock);
	pthread_mutex_unlock(&init_lock);
	if (error)
		return -1;
	return 0;
}