add yield support

[model-checker.git] / nodestack.h

diff --git a/nodestack.h b/nodestack.h
index 3cffac9c9b2219905020262302f4c580f9f03784..0a9422572eb0c4bd3ae6ce18b96757c57281a29f 100644 (file)
--- a/nodestack.h
+++ b/nodestack.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
  *
@@ -58,6 +65,8 @@ public:
 
        ModelAction * get_action() const { return 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 +114,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;
@@ -144,6 +153,7 @@ private:
 
        int misc_index;
        int misc_max;
+       int * yield_data;
 };
 
 typedef std::vector< Node *, ModelAlloc< Node * > > node_list_t;