add treiber stack

[model-checker-benchmarks.git] / treiber-stack / my_stack.h

diff --git a/treiber-stack/my_stack.h b/treiber-stack/my_stack.h
new file mode 100644 (file)
index 0000000..8d4d789
--- /dev/null
+++ b/treiber-stack/my_stack.h
@@ -0,0 +1,35 @@
+#include <stdatomic.h>
+
+#define release memory_order_release 
+#define acquire memory_order_acquire 
+#define relaxed memory_order_relaxed
+
+#define MAX_NODES                      0xf
+
+typedef unsigned long long pointer;
+typedef atomic_ullong pointer_t;
+
+#define MAKE_POINTER(ptr, count)       ((((pointer)count) << 32) | ptr)
+#define PTR_MASK 0xffffffffLL
+#define COUNT_MASK (0xffffffffLL << 32)
+
+static inline void set_count(pointer *p, unsigned int val) { *p = (*p & ~COUNT_MASK) | ((pointer)val << 32); }
+static inline void set_ptr(pointer *p, unsigned int val) { *p = (*p & ~PTR_MASK) | val; }
+static inline unsigned int get_count(pointer p) { return (p & COUNT_MASK) >> 32; }
+static inline unsigned int get_ptr(pointer p) { return p & PTR_MASK; }
+
+typedef struct node {
+       unsigned int value;
+       pointer_t next;
+
+} node_t;
+
+typedef struct {
+       pointer_t top;
+       node_t nodes[MAX_NODES + 1];
+} stack_t;
+
+void init_stack(stack_t *s, int num_threads);
+void push(stack_t *s, unsigned int val);
+unsigned int pop(stack_t *s);
+int get_thread_num();