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

struct rt_thread thread1;
struct rt_thread thread2;
static char thread1_stack[THREAD_STACK_SIZE];
static char thread2_stack[THREAD_STACK_SIZE];
static rt_uint32_t count = 0;

/*
 * the priority of thread1 > the priority of thread2
 */
static void thread1_entry(void* parameter)
{
	while (1)
	{
		count ++;
		rt_kprintf("count = %d\n", count);

		rt_thread_delay(10);
	}
}

static void thread2_entry(void* parameter)
{
	rt_tick_t tick;

	tick = rt_tick_get();
	while (1)
	{
		if (rt_tick_get() - tick >= 100)
		{
			if (count == 0)
				tc_done(TC_STAT_FAILED);
			else
				tc_done(TC_STAT_PASSED);

			break;
		}
	}
}

int thread_priority_init()
{
	rt_err_t result;

	result = rt_thread_init(&thread1,
		"t1",
		thread1_entry, RT_NULL,
		&thread1_stack[0], sizeof(thread1_stack),
		THREAD_PRIORITY - 1, THREAD_TIMESLICE);
	if (result == RT_EOK)
		rt_thread_startup(&thread1);
	else
		tc_stat(TC_STAT_FAILED);

	rt_thread_init(&thread2,
		"t2",
		thread2_entry, RT_NULL,
		&thread2_stack[0], sizeof(thread2_stack),
		THREAD_PRIORITY + 1, THREAD_TIMESLICE);

	if (result == RT_EOK)
		rt_thread_startup(&thread2);
	else
		tc_stat(TC_STAT_FAILED);

	return 0;
}

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

	if (thread1.stat != RT_THREAD_CLOSE)
		rt_thread_detach(&thread1);
	if (thread2.stat != RT_THREAD_CLOSE)
		rt_thread_detach(&thread2);

	/* unlock scheduler */
	rt_exit_critical();
}
int _tc_thread_priority()
{
	count = 0;

	/* set tc cleanup */
	tc_cleanup(_tc_cleanup);
	thread_priority_init();

	return RT_TICK_PER_SECOND;
}
FINSH_FUNCTION_EXPORT(_tc_thread_priority, a priority thread test);
#else
int rt_application_init()
{
	thread_priority_init();

	return 0;
}
#endif