#include <rtthread.h>
#include "tc_comm.h"

static rt_sem_t sem;
static rt_uint8_t t1_count, t2_count;
static rt_thread_t t1, t2, worker;
static void thread1_entry(void* parameter)
{
	rt_err_t result;

	while (1)
	{
		result = rt_sem_take(sem, RT_WAITING_FOREVER);
		if (result != RT_EOK)
		{
			tc_done(TC_STAT_FAILED);
			return;
		}

		t1_count ++;
		rt_kprintf("thread1: got semaphore, count: %d\n", t1_count);
	}
}

static void thread2_entry(void* parameter)
{
	rt_err_t result;

	while (1)
	{
		result = rt_sem_take(sem, RT_WAITING_FOREVER);
		if (result != RT_EOK)
		{
			tc_done(TC_STAT_FAILED);
			return;
		}

		t2_count ++;
		rt_kprintf("thread2: got semaphore, count: %d\n", t2_count);
	}
}

static void worker_thread_entry(void* parameter)
{
	rt_thread_delay(10);

	while (1)
	{
		rt_sem_release(sem);
		rt_thread_delay(5);
	}
}

int semaphore_priority_init()
{
	sem = rt_sem_create("sem", 0, RT_IPC_FLAG_PRIO);
	if (sem == RT_NULL)
	{
		tc_stat(TC_STAT_END | TC_STAT_FAILED);
		return 0;
	}

	t1_count = t2_count = 0;

	t1 = rt_thread_create("t1",
		thread1_entry, RT_NULL,
		THREAD_STACK_SIZE, THREAD_PRIORITY + 1, THREAD_TIMESLICE);
	if (t1 != RT_NULL)
		rt_thread_startup(t1);
	else
		tc_stat(TC_STAT_END | TC_STAT_FAILED);

	t2 = rt_thread_create("t2",
		thread2_entry, RT_NULL,
		THREAD_STACK_SIZE, THREAD_PRIORITY - 1, THREAD_TIMESLICE);
	if (t2 != RT_NULL)
		rt_thread_startup(t2);
	else
		tc_stat(TC_STAT_END | TC_STAT_FAILED);

	worker = rt_thread_create("worker",
		worker_thread_entry, RT_NULL,
		THREAD_STACK_SIZE, THREAD_PRIORITY, THREAD_TIMESLICE);
	if (worker != RT_NULL)
		rt_thread_startup(worker);
	else
		tc_stat(TC_STAT_END | TC_STAT_FAILED);

	return 0;
}

#ifdef RT_USING_TC
static void _tc_cleanup()
{
	/* lock scheduler */
	rt_enter_critical();

	/* delete t1, t2 and worker thread */
	rt_thread_delete(t1);
	rt_thread_delete(t2);
	rt_thread_delete(worker);

	if (t1_count > t2_count)
		tc_done(TC_STAT_FAILED);
	else
		tc_done(TC_STAT_PASSED);

	/* unlock scheduler */
	rt_exit_critical();
}

int _tc_semaphore_priority()
{
	/* set tc cleanup */
	tc_cleanup(_tc_cleanup);
	semaphore_priority_init();

	return 50;
}
FINSH_FUNCTION_EXPORT(_tc_semaphore_priority, a priority semaphore test);
#else
int rt_application_init()
{
	semaphore_priority_init();

	return 0;
}
#endif