#ifndef ATOMIC_H
#define ATOMIC_H

#include "cpu.h"
#include "irq.h"

typedef struct {
	int counter;
} atomic_t;

static inline int atomic_add_return(int i, atomic_t *v)
{
	int val;
	CPU_SR_ALLOC();

	CPU_CRITICAL_ENTER();

	val = v->counter;
	v->counter = val += i;

	CPU_CRITICAL_EXIT();

	return val;
}


static inline int atomic_sub_return(int i, atomic_t *v)
{
	int val;
	CPU_SR_ALLOC();

	CPU_CRITICAL_ENTER();

	val = v->counter;
	v->counter = val -= i;

	CPU_CRITICAL_EXIT();

	return val;
}

static inline int atomic_set(atomic_t *v, int i)
{
	int val = 0;
	CPU_SR_ALLOC();

	CPU_CRITICAL_ENTER();

	v->counter = i;

	CPU_CRITICAL_EXIT();

	return val;
}


#define DEFINE_ATOMIC(x) \
    atomic_t x = {.counter = 0}

#define atomic_add(i, v)    atomic_add_return(i, v)
#define atomic_sub(i, v)    atomic_sub_return(i, v)

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

#define atomic_inc(v)       atomic_add(1, v)
#define atomic_dec(v)       atomic_sub(1, v)

#define atomic_inc_and_test(v)  (atomic_add_return(1, v) == 0)
#define atomic_dec_and_test(v)  (atomic_sub_return(1, v) == 0)
#define atomic_inc_return(v)    (atomic_add_return(1, v))
#define atomic_dec_return(v)    (atomic_sub_return(1, v))
#define atomic_sub_and_test(i, v) (atomic_sub_return(i, v) == 0)

#define atomic_add_negative(i,v) (atomic_add_return(i, v) < 0)




#endif

