#include <qse/si/rwl.h>
#include <qse/si/thr.h>
#include <qse/si/sio.h>
#include <qse/si/intr.h>
#include <qse/cmn/mem.h>

#include <locale.h>

#if defined(_WIN32)
#	include <windows.h>
#endif

struct thr_xtn_t
{
	int id;
};
typedef struct thr_xtn_t thr_xtn_t;

static qse_rwl_t* rwl;
static qse_mtx_t* mtx;
static int stop_req = 0;

#define OUTMSG(msg,id)  do { \
	qse_ntime_t now; \
	qse_gettime(&now); \
	qse_mtx_lock (mtx, QSE_NULL); \
	qse_printf (QSE_T("%10ld.%7ld thread %d: %s\n"), (long int)now.sec, (long int)now.nsec, id, msg); \
	qse_mtx_unlock (mtx); \
} while(0)

static int thr_exec (qse_thr_t* thr, void* ctx)
{
	thr_xtn_t* xtn = qse_thr_getxtn(thr);

	OUTMSG (QSE_T("starting"), xtn->id);
	while (!stop_req)
	{
		if (xtn->id % 2)
		/*if (xtn->id > 0)*/
		{
			OUTMSG (QSE_T("read waiting"), xtn->id);

			if (qse_rwl_lockr (rwl, QSE_NULL) >= 0)
			{
				OUTMSG (QSE_T("read start"), xtn->id);
				/*sleep (1);*/
				OUTMSG (QSE_T("read done"), xtn->id);
				qse_rwl_unlockr (rwl);
			}
			else
			{
				OUTMSG (QSE_T("read fail"), xtn->id);
			}
		}
		else
		{
			qse_ntime_t x;
			OUTMSG (QSE_T("write waiting"), xtn->id);

			qse_init_ntime (&x, 3, 0);
			/*if (qse_rwl_lockw (rwl, QSE_NULL) >= 0)*/
			if (qse_rwl_lockw (rwl, &x) >= 0)
			{
				OUTMSG (QSE_T("write start"), xtn->id);
				/*sleep (1);*/
				OUTMSG (QSE_T("write done"), xtn->id);
				qse_rwl_unlockw (rwl);
				/*sleep (2);*/
			}
			else
			{
				OUTMSG (QSE_T("write fail"), xtn->id);
			}

		}
	}

	OUTMSG (QSE_T("exiting"), xtn->id);
	return 0;
}


static void test_001 (void)
{
	qse_mmgr_t* mmgr;
	qse_thr_t* t[6];
	qse_size_t i;
	thr_xtn_t* xtn;

	mmgr = QSE_MMGR_GETDFL();

	mtx = qse_mtx_open (mmgr, 0);
	rwl = qse_rwl_open (mmgr, 0, 0/*QSE_RWL_PREFER_WRITER*/);

	for (i = 0; i < QSE_COUNTOF(t); i++)
	{
		t[i] = qse_thr_open (mmgr, QSE_SIZEOF(thr_xtn_t));
		xtn = qse_thr_getxtn(t[i]);
		xtn->id = i;
	}

	for (i = 0; i < QSE_COUNTOF(t); i++) qse_thr_start (t[i], thr_exec, QSE_NULL, 0);
	for (i = 0; i < QSE_COUNTOF(t); i++) qse_thr_join (t[i]);
	for (i = 0; i < QSE_COUNTOF(t); i++) qse_thr_close (t[i]);

	qse_rwl_close (rwl);
	qse_mtx_close (mtx);
}

static void stop_run (void* arg)
{
	stop_req = 1;
}

int main ()
{
#if defined(_WIN32)
	char codepage[100];
	UINT old_cp = GetConsoleOutputCP();
	SetConsoleOutputCP (CP_UTF8);

#else
	setlocale (LC_ALL, "");
#endif

	qse_open_stdsios ();

	qse_set_intr_handler (stop_run, QSE_NULL);
	test_001 ();
	qse_clear_intr_handler ();

	qse_close_stdsios ();

#if defined(_WIN32)
	SetConsoleOutputCP (old_cp);
#endif
	return 0;
}