#ifndef _ATOMIC_H
#define _ATOMIC_H

#include <stdint.h>

#include "atomic_arch.h"

#ifdef a_ll

#ifndef a_pre_llsc
#define a_pre_llsc()
#endif

#ifndef a_post_llsc
#define a_post_llsc()
#endif

#ifndef a_cas
#define a_cas a_cas
static inline int a_cas(volatile int* p, int t, int s) {
  int old;
  a_pre_llsc();
  do
    old = a_ll(p);
  while (old == t && !a_sc(p, s));
  a_post_llsc();
  return old;
}
#endif

#ifndef a_swap
#define a_swap a_swap
static inline int a_swap(volatile int* p, int v) {
  int old;
  a_pre_llsc();
  do
    old = a_ll(p);
  while (!a_sc(p, v));
  a_post_llsc();
  return old;
}
#endif

#ifndef a_fetch_add
#define a_fetch_add a_fetch_add
static inline int a_fetch_add(volatile int* p, int v) {
  int old;
  a_pre_llsc();
  do
    old = a_ll(p);
  while (!a_sc(p, (unsigned)old + v));
  a_post_llsc();
  return old;
}
#endif

#ifndef a_fetch_and
#define a_fetch_and a_fetch_and
static inline int a_fetch_and(volatile int* p, int v) {
  int old;
  a_pre_llsc();
  do
    old = a_ll(p);
  while (!a_sc(p, old & v));
  a_post_llsc();
  return old;
}
#endif

#ifndef a_fetch_or
#define a_fetch_or a_fetch_or
static inline int a_fetch_or(volatile int* p, int v) {
  int old;
  a_pre_llsc();
  do
    old = a_ll(p);
  while (!a_sc(p, old | v));
  a_post_llsc();
  return old;
}
#endif

#endif

#ifndef a_cas
#error missing definition of a_cas
#endif

#ifndef a_swap
#define a_swap a_swap
static inline int a_swap(volatile int* p, int v) {
  int old;
  do
    old = *p;
  while (a_cas(p, old, v) != old);
  return old;
}
#endif

#ifndef a_fetch_add
#define a_fetch_add a_fetch_add
static inline int a_fetch_add(volatile int* p, int v) {
  int old;
  do
    old = *p;
  while (a_cas(p, old, (unsigned)old + v) != old);
  return old;
}
#endif

#ifndef a_fetch_and
#define a_fetch_and a_fetch_and
static inline int a_fetch_and(volatile int* p, int v) {
  int old;
  do
    old = *p;
  while (a_cas(p, old, old & v) != old);
  return old;
}
#endif
#ifndef a_fetch_or
#define a_fetch_or a_fetch_or
static inline int a_fetch_or(volatile int* p, int v) {
  int old;
  do
    old = *p;
  while (a_cas(p, old, old | v) != old);
  return old;
}
#endif

#ifndef a_and
#define a_and a_and
static inline void a_and(volatile int* p, int v) {
  a_fetch_and(p, v);
}
#endif

#ifndef a_or
#define a_or a_or
static inline void a_or(volatile int* p, int v) {
  a_fetch_or(p, v);
}
#endif

#ifndef a_inc
#define a_inc a_inc
static inline void a_inc(volatile int* p) {
  a_fetch_add(p, 1);
}
#endif

#ifndef a_dec
#define a_dec a_dec
static inline void a_dec(volatile int* p) {
  a_fetch_add(p, -1);
}
#endif

#ifndef a_store
#define a_store a_store
static inline void a_store(volatile int* p, int v) {
#ifdef a_barrier
  a_barrier();
  *p = v;
  a_barrier();
#else
  a_swap(p, v);
#endif
}
#endif

#ifndef a_barrier
#define a_barrier a_barrier
static void a_barrier() {
  volatile int tmp = 0;
  a_cas(&tmp, 0, 0);
}
#endif

#ifndef a_spin
#define a_spin a_barrier
#endif

#ifndef a_and_64
#define a_and_64 a_and_64
static inline void a_and_64(volatile uint64_t* p, uint64_t v) {
  union {
    uint64_t v;
    uint32_t r[2];
  } u = {v};
  if (u.r[0] + 1)
    a_and((int*)p, u.r[0]);
  if (u.r[1] + 1)
    a_and((int*)p + 1, u.r[1]);
}
#endif

#ifndef a_or_64
#define a_or_64 a_or_64
static inline void a_or_64(volatile uint64_t* p, uint64_t v) {
  union {
    uint64_t v;
    uint32_t r[2];
  } u = {v};
  if (u.r[0])
    a_or((int*)p, u.r[0]);
  if (u.r[1])
    a_or((int*)p + 1, u.r[1]);
}
#endif

#ifndef a_cas_p
#define a_cas_p a_cas_p
static inline void* a_cas_p(volatile void* p, void* t, void* s) {
  return (void*)a_cas((volatile int*)p, (int)t, (int)s);
}
#endif

#ifndef a_or_l
#define a_or_l a_or_l
static inline void a_or_l(volatile void* p, long v) {
  if (sizeof(long) == sizeof(int))
    a_or(p, v);
  else
    a_or_64(p, v);
}
#endif

#ifndef a_crash
#define a_crash a_crash
static inline void a_crash() {
  *(volatile char*)0 = 0;
}
#endif

#ifndef a_ctz_64
#define a_ctz_64 a_ctz_64
static inline int a_ctz_64(uint64_t x) {
  static const char debruijn64[64] = {
      0,  1,  2,  53, 3,  7,  54, 27, 4,  38, 41, 8,  34, 55, 48, 28,
      62, 5,  39, 46, 44, 42, 22, 9,  24, 35, 59, 56, 49, 18, 29, 11,
      63, 52, 6,  26, 37, 40, 33, 47, 61, 45, 43, 21, 23, 58, 17, 10,
      51, 25, 36, 32, 60, 20, 57, 16, 50, 31, 19, 15, 30, 14, 13, 12};
  static const char debruijn32[32] = {
      0,  1,  23, 2,  29, 24, 19, 3,  30, 27, 25, 11, 20, 8, 4,  13,
      31, 22, 28, 18, 26, 10, 7,  12, 21, 17, 9,  6,  16, 5, 15, 14};
  if (sizeof(long) < 8) {
    uint32_t y = x;
    if (!y) {
      y = x >> 32;
      return 32 + debruijn32[(y & -y) * 0x076be629 >> 27];
    }
    return debruijn32[(y & -y) * 0x076be629 >> 27];
  }
  return debruijn64[(x & -x) * 0x022fdd63cc95386dull >> 58];
}
#endif

#ifndef a_ctz_l
#define a_ctz_l a_ctz_l
static inline int a_ctz_l(unsigned long x) {
  static const char debruijn32[32] = {
      0,  1,  23, 2,  29, 24, 19, 3,  30, 27, 25, 11, 20, 8, 4,  13,
      31, 22, 28, 18, 26, 10, 7,  12, 21, 17, 9,  6,  16, 5, 15, 14};
  if (sizeof(long) == 8)
    return a_ctz_64(x);
  return debruijn32[(x & -x) * 0x076be629 >> 27];
}
#endif

#endif