#pragma once

#ifndef TATOMICVAR_H
#define TATOMICVAR_H

#undef DVAPI
#undef DVVAR
#ifdef TNZCORE_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

#ifdef _WIN32
#include <windows.h>
#elif defined(__sgi)
#include <sys/atomic_ops.h>
#elif defined(LINUX)
// #include <asm/atomic.h>
// it's broken, either include the kernel header
//   /usr/src/linux/include/asm/atomic.h
// or copy it here
#include "/usr/src/linux-2.4/include/asm/atomic.h"

#elif defined(powerpc)

// from linux-2.4.20-20.9/include/linux/asm-ppc

typedef struct {
	volatile int counter;
} atomic_t;

#define ATOMIC_INIT(i) \
	{                  \
		(i)            \
	}

#define atomic_read(v) ((v)->counter)
#define atomic_set(v, i) (((v)->counter) = (i))

extern void atomic_clear_mask(unsigned long mask, unsigned long *addr);
extern void atomic_set_mask(unsigned long mask, unsigned long *addr);

// we define it as we want to run on SMP dual processors...
#define CONFIG_SMP

#ifdef CONFIG_SMP
#define SMP_ISYNC "\n\tisync"
#else
#define SMP_ISYNC
#endif

static __inline__ void atomic_add(int a, atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%3		\n\
	add	%0,%2,%0\n\
	stwcx.	%0,0,%3\n\
	bne-	1b"
		: "=&r"(t), "=m"(v->counter)
		: "r"(a), "r"(&v->counter), "m"(v->counter)
		: "cc");
}

static __inline__ int atomic_add_return(int a, atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%2		\n\
	add	%0,%1,%0\n\
	stwcx.	%0,0,%2\n\
	bne-	1b" SMP_ISYNC
		: "=&r"(t)
		: "r"(a), "r"(&v->counter)
		: "cc", "memory");

	return t;
}

static __inline__ void atomic_sub(int a, atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%3		\n\
	subf	%0,%2,%0\n\
	stwcx.	%0,0,%3\n\
	bne-	1b"
		: "=&r"(t), "=m"(v->counter)
		: "r"(a), "r"(&v->counter), "m"(v->counter)
		: "cc");
}

static __inline__ int atomic_sub_return(int a, atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%2		\n\
	subf	%0,%1,%0\n\
	stwcx.	%0,0,%2\n\
	bne-	1b" SMP_ISYNC
		: "=&r"(t)
		: "r"(a), "r"(&v->counter)
		: "cc", "memory");

	return t;
}

static __inline__ void atomic_inc(atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%2		\n\
	addic	%0,%0,1\n\
	stwcx.	%0,0,%2\n\
	bne-	1b"
		: "=&r"(t), "=m"(v->counter)
		: "r"(&v->counter), "m"(v->counter)
		: "cc");
}

static __inline__ int atomic_inc_return(atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%1		\n\
	addic	%0,%0,1\n\
	stwcx.	%0,0,%1\n\
	bne-	1b" SMP_ISYNC
		: "=&r"(t)
		: "r"(&v->counter)
		: "cc", "memory");

	return t;
}

static __inline__ void atomic_dec(atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%2		\n\
	addic	%0,%0,-1\n\
	stwcx.	%0,0,%2\n\
	bne-	1b"
		: "=&r"(t), "=m"(v->counter)
		: "r"(&v->counter), "m"(v->counter)
		: "cc");
}

static __inline__ int atomic_dec_return(atomic_t *v)
{
	int t;

	__asm__ __volatile__(
		"1:	lwarx	%0,0,%1		\n\
	addic	%0,%0,-1\n\
	stwcx.	%0,0,%1\n\
	bne-	1b" SMP_ISYNC
		: "=&r"(t)
		: "r"(&v->counter)
		: "cc", "memory");

	return t;
}

#elif defined(i386)

#include <atomic>

typedef std::atomic<int> atomic_t;

static __inline__ void atomic_set(atomic_t *v, const int value)
{
	v->store(value);
}

static __inline__ int atomic_inc_return(atomic_t *v)
{
	return v->fetch_add(1) + 1; // post increment atomic
}

static __inline__ int atomic_dec_return(atomic_t *v)
{
	return v->fetch_sub(1) - 1; // post decriment atomic
}

static __inline__ int atomic_read(const atomic_t *v)
{
	return v->load();
}

static __inline__ int atomic_add(int num, const atomic_t *v)
{
	return const_cast<atomic_t *>(v)->fetch_add(num) + num; /* なんで const つけた? */
}

#else
@ @PLATFORM NOT SUPPORTED !@ @
#endif

/*! Platform specific 
    Provides class with increment & decrement absolutely done in interlocked way 
*/

class DVAPI TAtomicVar
{
public:
#if defined(LINUX) || defined(MACOSX)
	TAtomicVar()
	{
		atomic_set(&m_var, 0);
	}
#else
	TAtomicVar() : m_var(0)
	{
	}
#endif

	long operator++()
	{
#ifdef _WIN32
		return InterlockedIncrement(&m_var);
#elif defined(__sgi)
		return ++m_var;
#elif defined(LINUX)
		// atomic_inc(&m_var);
		// return atomic_read(&m_var);
		//  this is broken as it can return a value != from ++m_var
		atomic_inc(&m_var);
		return atomic_read(&m_var);
#elif defined(MACOSX)
		return atomic_inc_return(&m_var);
#endif
	}

	long operator+=(long value)
	{
#ifdef _WIN32
		InterlockedExchangeAdd(&m_var, value);
		return m_var;

#elif defined(__sgi)
		m_var += value;
		return m_var;
#elif defined(LINUX)
		// atomic_inc(&m_var);
		// return atomic_read(&m_var);
		//  this is broken as it can return a value != from ++m_var
		assert(false);

		return m_var;
#elif defined(MACOSX)
		atomic_add(value, &m_var);
		return atomic_read(&m_var);
#endif
	}

	long operator--()
	{
#ifdef _WIN32
		return InterlockedDecrement(&m_var);
#elif defined(__sgi)
		return --m_var;
#elif defined(LINUX)
		// atomic_dec(&m_var);
		// return atomic_read(&m_var);
		// broken as above...
		atomic_dec(&m_var);
		return atomic_read(&m_var);
#elif defined(MACOSX)
		// atomic_dec(&m_var);
		// return atomic_read(&m_var);
		// broken as above...
		return atomic_dec_return(&m_var);
#endif
	}
	bool operator<=(const long &rhs)
	{
#if defined(LINUX) || defined(MACOSX)
		return atomic_read(&m_var) <= rhs;
#else
		return m_var <= rhs;
#endif
	};
	operator long() const
	{
#if defined(LINUX) || defined(MACOSX)
		return atomic_read(&m_var);
#else
		return m_var;
#endif
	};

#ifdef _WIN32
	long m_var;
#elif defined(__sgi)
	long m_var;
#elif defined(LINUX) || defined(MACOSX)
	atomic_t m_var;
#endif

#if !defined(LINUX) || defined(LINUX) && (__GNUC__ == 3) && (__GNUC_MINOR__ > 1)
private: // to avoid well known bug in gcc3 ... fixed in later versions..
#endif
	TAtomicVar &operator=(const TAtomicVar &); //not implemented
	TAtomicVar(const TAtomicVar &v);		   //not implemented
};

#endif