// (c) 2010 Thomas Schoebel-Theuer / 1&1 Internet AG
#ifndef BRICK_ATOMIC_H
#define BRICK_ATOMIC_H

#include <linux/spinlock.h>

#include <asm/atomic.h>

#include "brick_checking.h"

#define ATOMIC_DEBUGGING

#ifndef CONFIG_MARS_DEBUG
#undef  ATOMIC_DEBUGGING
#endif

#define ATOMIC_MAXTRACE 64

/* Trivial wrapper to force type checking.
 */
typedef struct tatomic {
	atomic_t ta_atomic;
} tatomic_t;

typedef struct atomic_trace {
#ifdef ATOMIC_DEBUGGING
	atomic_t at_count;
	short at_lines[ATOMIC_MAXTRACE];
#endif
} atomic_trace_t;

#ifdef ATOMIC_DEBUGGING

#define tatomic_trace(_at, _cmd)					\
	({								\
		int _index = atomic_add_return(1, &(_at)->at_count) - 1; \
		if (_index >= 0 && _index < ATOMIC_MAXTRACE) {		\
			(_at)->at_lines[_index] = __LINE__;		\
		}							\
		_cmd;							\
	})

#define tatomic_out(_at,_MSG)						\
	({								\
		int __i;						\
		int __max = atomic_read(&(_at)->at_count);		\
		_MSG("at_count = %d\n", __max);				\
		if (__max > ATOMIC_MAXTRACE)				\
			__max = ATOMIC_MAXTRACE;			\
		for (__i = 0; __i < __max; __i++) {			\
			_MSG("%2d %4d\n", __i, (_at)->at_lines[__i]);	\
		}							\
	})

#define _CHECK_TATOMIC(_at,_atom,OP,_minval)				\
do {									\
	if (BRICK_CHECKING) {						\
		int __test = atomic_read(&(_atom)->ta_atomic);		\
		if (__test OP (_minval)) {				\
			atomic_set(&(_atom)->ta_atomic, _minval);	\
			BRICK_ERR("%d: tatomic " #_atom " " #OP " " #_minval " (%d)\n", __LINE__, __test); \
			tatomic_out(_at, BRICK_DMP);			\
		}							\
	}								\
} while (0)

#else

#define tatomic_trace(_at,_cmd)  _cmd

#define _CHECK_TATOMIC(_at,_atom,OP,_minval)				\
	_CHECK_ATOMIC(&(_atom)->ta_atomic, OP, _minval)

#endif

#define CHECK_TATOMIC(_at,_atom,_minval)		\
	_CHECK_TATOMIC(_at, _atom, <, _minval)

#define tatomic_inc(at,a)           tatomic_trace(at, atomic_inc(&(a)->ta_atomic))
#define tatomic_dec(at,a)           tatomic_trace(at, atomic_dec(&(a)->ta_atomic))
#define tatomic_dec_and_test(at,a)  tatomic_trace(at, atomic_dec_and_test(&(a)->ta_atomic))

#endif