1#ifndef TREE_SITTER_ATOMIC_H_
 2#define TREE_SITTER_ATOMIC_H_
 3
 4#include <stddef.h>
 5#include <stdint.h>
 6
 7#ifdef __TINYC__
 8
 9static inline size_t atomic_load(const volatile size_t *p) {
10  return *p;
11}
12
13static inline uint32_t atomic_inc(volatile uint32_t *p) {
14  *p += 1;
15  return *p;
16}
17
18static inline uint32_t atomic_dec(volatile uint32_t *p) {
19  *p-= 1;
20  return *p;
21}
22
23#elif defined(_WIN32)
24
25#include <windows.h>
26
27static inline size_t atomic_load(const volatile size_t *p) {
28  return *p;
29}
30
31static inline uint32_t atomic_inc(volatile uint32_t *p) {
32  return InterlockedIncrement((long volatile *)p);
33}
34
35static inline uint32_t atomic_dec(volatile uint32_t *p) {
36  return InterlockedDecrement((long volatile *)p);
37}
38
39#else
40
41static inline size_t atomic_load(const volatile size_t *p) {
42#ifdef __ATOMIC_RELAXED
43  return __atomic_load_n(p, __ATOMIC_RELAXED);
44#else
45  return __sync_fetch_and_add((volatile size_t *)p, 0);
46#endif
47}
48
49static inline uint32_t atomic_inc(volatile uint32_t *p) {
50  #ifdef __ATOMIC_RELAXED
51    return __atomic_add_fetch(p, 1U, __ATOMIC_SEQ_CST);
52  #else
53    return __sync_add_and_fetch(p, 1U);
54  #endif
55}
56
57static inline uint32_t atomic_dec(volatile uint32_t *p) {
58  #ifdef __ATOMIC_RELAXED
59    return __atomic_sub_fetch(p, 1U, __ATOMIC_SEQ_CST);
60  #else
61    return __sync_sub_and_fetch(p, 1U);
62  #endif
63}
64
65#endif
66
67#endif  // TREE_SITTER_ATOMIC_H_