Fix snapshot code

[model-checker.git] / nodestack.h

diff --git a/nodestack.h b/nodestack.h
index 1a8bbbe9f371cc2bef422ec2218e96f989fb28a8..6ae96be8d508d9b598ae3bb3c396760fea6047da 100644 (file)
--- a/nodestack.h
+++ b/nodestack.h
@@ -54,7 +54,8 @@ typedef enum {
  */
 class Node {
 public:
-       Node(ModelAction *act, Node *par, int nthreads, Node *prevfairness);
+       Node(const struct model_params *params, ModelAction *act, Node *par,
+                       int nthreads, Node *prevfairness);
        ~Node();
        /* return true = thread choice has already been explored */
        bool has_been_explored(thread_id_t tid) const;
@@ -134,9 +135,12 @@ private:
        bool future_value_empty() const;
        bool increment_future_value();
        read_from_type_t read_from_status;
+       const struct model_params * get_params() const { return params; }
 
        ModelAction * const action;
 
+       const struct model_params * const params;
+
        /** @brief ATOMIC_UNINIT action which was created at this Node */
        ModelAction *uninit_action;
 
@@ -194,6 +198,7 @@ public:
        Node * get_next() const;
        void reset_execution();
        void pop_restofstack(int numAhead);
+       void full_reset();
        int get_total_nodes() { return total_nodes; }
 
        void print() const;
@@ -202,6 +207,8 @@ public:
 private:
        node_list_t node_list;
 
+       const struct model_params * get_params() const;
+
        /** @brief The model-checker execution object */
        const ModelExecution *execution;