switch to snapshot/modelalloc versions of stl classes

[model-checker.git] / nodestack.h

diff --git a/nodestack.h b/nodestack.h
index 3cffac9c9b2219905020262302f4c580f9f03784..47c0272f95fa917e8710235fa0d56b02194a4d08 100644 (file)
--- a/nodestack.h
+++ b/nodestack.h
@@ -5,7 +5,7 @@
 #ifndef __NODESTACK_H__
 #define __NODESTACK_H__
 
-#include <vector>
+#include "stl_wrappers.h"
 #include <cstddef>
 #include <inttypes.h>
 
@@ -29,6 +29,13 @@ typedef enum {
        READ_FROM_NONE,
 } read_from_type_t;
 
+#define YIELD_E 1
+#define YIELD_D 2
+#define YIELD_S 4
+#define YIELD_P 8
+#define YIELD_INDEX(tid1, tid2, num_threads) (tid1*num_threads+tid2)
+
+
 /**
  * @brief A single node in a NodeStack
  *
@@ -57,7 +64,12 @@ public:
        enabled_type_t enabled_status(thread_id_t tid) const;
 
        ModelAction * get_action() const { return action; }
+       void set_uninit_action(ModelAction *act) { uninit_action = act; }
+       ModelAction * get_uninit_action() const { return uninit_action; }
+
        bool has_priority(thread_id_t tid) const;
+       void update_yield(Scheduler *);
+       bool has_priority_over(thread_id_t tid, thread_id_t tid2) const;
        int get_num_threads() const { return num_threads; }
        /** @return the parent Node to this Node; that is, the action that
         * occurred previously in the stack. */
@@ -105,7 +117,7 @@ public:
        MEMALLOC
 private:
        void explore(thread_id_t tid);
-
+       int get_yield_data(int tid1, int tid2) const;
        bool read_from_past_empty() const;
        bool increment_read_from_past();
        bool read_from_promise_empty() const;
@@ -115,6 +127,10 @@ private:
        read_from_type_t read_from_status;
 
        ModelAction * const action;
+
+       /** @brief ATOMIC_UNINIT action which was created at this Node */
+       ModelAction *uninit_action;
+
        Node * const parent;
        const int num_threads;
        std::vector< bool, ModelAlloc<bool> > explored_children;
@@ -144,9 +160,10 @@ private:
 
        int misc_index;
        int misc_max;
+       int * yield_data;
 };
 
-typedef std::vector< Node *, ModelAlloc< Node * > > node_list_t;
+typedef model_vector< Node * > node_list_t;
 
 /**
  * @brief A stack of nodes