Edits
[c11tester.git] / funcnode.cc
index 2b2525466d24652e4f7d703f8caf1ceed63077cb..2a8d7c1b21b6a096c091be86425473e24c65eac0 100644 (file)
+#include "action.h"
+#include "history.h"
 #include "funcnode.h"
+#include "funcinst.h"
 #include "predicate.h"
-
-FuncNode::FuncNode() :
+#include "concretepredicate.h"
+
+#include "model.h"
+#include "execution.h"
+#include "newfuzzer.h"
+#include <cmath>
+
+FuncNode::FuncNode(ModelHistory * history) :
+       history(history),
+       inst_counter(1),
+       marker(1),
+       thrd_markers(),
+       thrd_recursion_depth(),
        func_inst_map(),
        inst_list(),
        entry_insts(),
-       thrd_read_map()
-{}
+       thrd_inst_pred_maps(),
+       thrd_inst_id_maps(),
+       thrd_loc_inst_maps(),
+       thrd_predicate_tree_position(),
+       thrd_predicate_trace(),
+       edge_table(32),
+       out_edges()
+{
+       predicate_tree_entry = new Predicate(NULL, true);
+       predicate_tree_entry->add_predicate_expr(NOPREDICATE, NULL, true);
+
+       predicate_tree_exit = new Predicate(NULL, false, true);
+       predicate_tree_exit->set_depth(MAX_DEPTH);
+
+       /* Snapshot data structures below */
+       read_locations = new loc_set_t();
+       write_locations = new loc_set_t();
+       val_loc_map = new HashTable<uint64_t, loc_set_t *, uint64_t, 0, snapshot_malloc, snapshot_calloc, snapshot_free, int64_hash>();
+       loc_may_equal_map = new HashTable<void *, loc_set_t *, uintptr_t, 0>();
+
+       //values_may_read_from = new value_set_t();
+}
+
+/* Reallocate snapshotted memories when new executions start */
+void FuncNode::set_new_exec_flag()
+{
+       read_locations = new loc_set_t();
+       write_locations = new loc_set_t();
+       val_loc_map = new HashTable<uint64_t, loc_set_t *, uint64_t, 0, snapshot_malloc, snapshot_calloc, snapshot_free, int64_hash>();
+       loc_may_equal_map = new HashTable<void *, loc_set_t *, uintptr_t, 0>();
+
+       //values_may_read_from = new value_set_t();
+}
 
 /* Check whether FuncInst with the same type, position, and location
- * as act has been added to func_inst_map or not. If so, return it;
- * if not, add it and return it.
- *
- * @return FuncInst with the same type, position, and location as act */
-FuncInst * FuncNode::get_or_add_action(ModelAction *act)
+ * as act has been added to func_inst_map or not. If not, add it.
+ */
+void FuncNode::add_inst(ModelAction *act)
 {
        ASSERT(act);
        const char * position = act->get_position();
 
-       /* Actions THREAD_CREATE, THREAD_START, THREAD_YIELD, THREAD_JOIN,
-        * THREAD_FINISH, PTHREAD_CREATE, PTHREAD_JOIN,
-        * ATOMIC_LOCK, ATOMIC_TRYLOCK, and ATOMIC_UNLOCK are not tagged with their
-        * source line numbers
+       /* THREAD* actions, ATOMIC_LOCK, ATOMIC_TRYLOCK, and ATOMIC_UNLOCK
+        * actions are not tagged with their source line numbers
         */
        if (position == NULL)
-               return NULL;
-
-       if ( func_inst_map.contains(position) ) {
-               FuncInst * inst = func_inst_map.get(position);
+               return;
 
-               if (inst->get_type() != act->get_type() ) {
-                       // model_print("action with a different type occurs at line number %s\n", position);
-                       FuncInst * func_inst = inst->search_in_collision(act);
+       FuncInst * func_inst = func_inst_map.get(position);
 
-                       if (func_inst != NULL) {
-                               // return the FuncInst found in the collision list
-                               return func_inst;
-                       }
+       /* This position has not been inserted into hashtable before */
+       if (func_inst == NULL) {
+               func_inst = create_new_inst(act);
+               func_inst_map.put(position, func_inst);
+               return;
+       }
 
-                       func_inst = new FuncInst(act, this);
-                       inst->get_collisions()->push_back(func_inst);
-                       inst_list.push_back(func_inst); // delete?
+       /* Volatile variables that use ++ or -- syntax may result in read and write actions with the same position */
+       if (func_inst->get_type() != act->get_type()) {
+               FuncInst * collision_inst = func_inst->search_in_collision(act);
 
-                       return func_inst;
+               if (collision_inst == NULL) {
+                       collision_inst = create_new_inst(act);
+                       func_inst->add_to_collision(collision_inst);
+                       return;
+               } else {
+                       func_inst = collision_inst;
                }
+       }
 
-               return inst;
+       ASSERT(func_inst->get_type() == act->get_type());
+       int curr_execution_number = model->get_execution_number();
+
+       /* Reset locations when new executions start */
+       if (func_inst->get_execution_number() != curr_execution_number) {
+               func_inst->set_location(act->get_location());
+               func_inst->set_execution_number(curr_execution_number);
        }
 
+       /* Mark the memory location of such inst as not unique */
+       if (func_inst->get_location() != act->get_location())
+               func_inst->not_single_location();
+}
+
+FuncInst * FuncNode::create_new_inst(ModelAction * act)
+{
        FuncInst * func_inst = new FuncInst(act, this);
+       int exec_num = model->get_execution_number();
+       func_inst->set_execution_number(exec_num);
 
-       func_inst_map.put(position, func_inst);
        inst_list.push_back(func_inst);
 
        return func_inst;
 }
 
+
+/* Get the FuncInst with the same type, position, and location
+ * as act
+ *
+ * @return FuncInst with the same type, position, and location as act */
+FuncInst * FuncNode::get_inst(ModelAction *act)
+{
+       ASSERT(act);
+       const char * position = act->get_position();
+
+       /* THREAD* actions, ATOMIC_LOCK, ATOMIC_TRYLOCK, and ATOMIC_UNLOCK
+        * actions are not tagged with their source line numbers
+        */
+       if (position == NULL)
+               return NULL;
+
+       FuncInst * inst = func_inst_map.get(position);
+       if (inst == NULL)
+               return NULL;
+
+       action_type inst_type = inst->get_type();
+       action_type act_type = act->get_type();
+
+       if (inst_type == act_type) {
+               return inst;
+       }
+       /* RMWRCAS actions are converted to RMW or READ actions */
+       else if (inst_type == ATOMIC_RMWRCAS &&
+                                        (act_type == ATOMIC_RMW || act_type == ATOMIC_READ)) {
+               return inst;
+       }
+       /* Return the FuncInst in the collision list */
+       else {
+               return inst->search_in_collision(act);
+       }
+}
+
 void FuncNode::add_entry_inst(FuncInst * inst)
 {
        if (inst == NULL)
                return;
 
-       mllnode<FuncInst*>* it;
-       for (it = entry_insts.begin();it != NULL;it=it->getNext()) {
+       mllnode<FuncInst *> * it;
+       for (it = entry_insts.begin();it != NULL;it = it->getNext()) {
                if (inst == it->getVal())
                        return;
        }
@@ -70,27 +165,107 @@ void FuncNode::add_entry_inst(FuncInst * inst)
        entry_insts.push_back(inst);
 }
 
-/* @param inst_list a list of FuncInsts; this argument comes from ModelExecution
- * Link FuncInsts in a list - add one FuncInst to another's predecessors and successors
+void FuncNode::function_entry_handler(thread_id_t tid)
+{
+       init_marker(tid);
+       init_local_maps(tid);
+       init_predicate_tree_data_structure(tid);
+}
+
+void FuncNode::function_exit_handler(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+
+       reset_local_maps(tid);
+
+       thrd_recursion_depth[thread_id]--;
+       thrd_markers[thread_id]->pop_back();
+
+       Predicate * exit_pred = get_predicate_tree_position(tid);
+       if (exit_pred->get_exit() == NULL) {
+               // Exit predicate is unset yet
+               exit_pred->set_exit(predicate_tree_exit);
+       }
+
+       update_predicate_tree_weight(tid);
+       reset_predicate_tree_data_structure(tid);
+}
+
+/**
+ * @brief Convert ModelAdtion list to FuncInst list
+ * @param act_list A list of ModelActions
  */
-void FuncNode::link_insts(func_inst_list_t * inst_list)
+void FuncNode::update_tree(ModelAction * act)
+{
+       bool should_process = act->is_read() || act->is_write();
+       if (!should_process)
+               return;
+
+       HashTable<void *, value_set_t *, uintptr_t, 0> * write_history = history->getWriteHistory();
+
+       /* build inst_list from act_list for later processing */
+//     func_inst_list_t inst_list;
+
+       FuncInst * func_inst = get_inst(act);
+       void * loc = act->get_location();
+
+       if (func_inst == NULL)
+               return;
+
+//     inst_list.push_back(func_inst);
+
+       if (act->is_write()) {
+               if (!write_locations->contains(loc)) {
+                       write_locations->add(loc);
+                       history->update_loc_wr_func_nodes_map(loc, this);
+               }
+       }
+
+       if (act->is_read()) {
+
+               /* If func_inst may only read_from a single location, then:
+                *
+                * The first time an action reads from some location,
+                * import all the values that have been written to this
+                * location from ModelHistory and notify ModelHistory
+                * that this FuncNode may read from this location.
+                */
+               if (!read_locations->contains(loc) && func_inst->is_single_location()) {
+                       read_locations->add(loc);
+                       value_set_t * write_values = write_history->get(loc);
+                       add_to_val_loc_map(write_values, loc);
+                       history->update_loc_rd_func_nodes_map(loc, this);
+               }
+       }
+
+//     update_inst_tree(&inst_list); TODO
+
+       update_predicate_tree(act);
+
+//     print_predicate_tree();
+}
+
+/**
+ * @brief Link FuncInsts in inst_list  - add one FuncInst to another's predecessors and successors
+ * @param inst_list A list of FuncInsts
+ */
+void FuncNode::update_inst_tree(func_inst_list_t * inst_list)
 {
        if (inst_list == NULL)
                return;
+       else if (inst_list->size() == 0)
+               return;
 
+       /* start linking */
        sllnode<FuncInst *>* it = inst_list->begin();
        sllnode<FuncInst *>* prev;
 
-       if (inst_list->size() == 0)
-               return;
-
        /* add the first instruction to the list of entry insts */
        FuncInst * entry_inst = it->getVal();
        add_entry_inst(entry_inst);
 
-       it=it->getNext();
+       it = it->getNext();
        while (it != NULL) {
-               prev = it;
                prev = it->getPrev();
 
                FuncInst * prev_inst = prev->getVal();
@@ -99,81 +274,602 @@ void FuncNode::link_insts(func_inst_list_t * inst_list)
                prev_inst->add_succ(curr_inst);
                curr_inst->add_pred(prev_inst);
 
-               it=it->getNext();
+               it = it->getNext();
        }
 }
 
-/* @param tid thread id
- * Store the values read by atomic read actions into thrd_read_map */
-void FuncNode::store_read(ModelAction * act, uint32_t tid)
+void FuncNode::update_predicate_tree(ModelAction * next_act)
 {
-       ASSERT(act);
+       thread_id_t tid = next_act->get_tid();
+       int thread_id = id_to_int(tid);
+       uint32_t this_marker = thrd_markers[thread_id]->back();
+       int recursion_depth = thrd_recursion_depth[thread_id];
+
+       loc_inst_map_t * loc_inst_map = thrd_loc_inst_maps[thread_id]->back();
+       inst_pred_map_t * inst_pred_map = thrd_inst_pred_maps[thread_id]->back();
+       inst_id_map_t * inst_id_map = thrd_inst_id_maps[thread_id]->back();
+
+       Predicate * curr_pred = get_predicate_tree_position(tid);
+       NewFuzzer * fuzzer = (NewFuzzer *)model->get_execution()->getFuzzer();
+       Predicate * selected_branch = fuzzer->get_selected_child_branch(tid);
+
+       bool amended;
+       while (true) {
+               FuncInst * next_inst = get_inst(next_act);
+
+               Predicate * unset_predicate = NULL;
+               bool branch_found = follow_branch(&curr_pred, next_inst, next_act, &unset_predicate);
+
+               // A branch with unset predicate expression is detected
+               if (!branch_found && unset_predicate != NULL) {
+                       amended = amend_predicate_expr(curr_pred, next_inst, next_act);
+                       if (amended)
+                               continue;
+                       else {
+                               curr_pred = unset_predicate;
+                               branch_found = true;
+                       }
+               }
+
+               // Detect loops
+               if (!branch_found && inst_id_map->contains(next_inst)) {
+                       FuncInst * curr_inst = curr_pred->get_func_inst();
+                       uint32_t curr_id = inst_id_map->get(curr_inst);
+                       uint32_t next_id = inst_id_map->get(next_inst);
+
+                       if (curr_id >= next_id) {
+                               Predicate * old_pred = inst_pred_map->get(next_inst);
+                               Predicate * back_pred = old_pred->get_parent();
+
+                               // Add to the set of backedges
+                               curr_pred->add_backedge(back_pred);
+                               curr_pred = back_pred;
+
+                               continue;
+                       }
+               }
+
+               // Generate new branches
+               if (!branch_found) {
+                       SnapVector<struct half_pred_expr *> half_pred_expressions;
+                       infer_predicates(next_inst, next_act, &half_pred_expressions);
+                       generate_predicates(curr_pred, next_inst, &half_pred_expressions);
+                       continue;
+               }
+
+               if (next_act->is_write()) {
+                       curr_pred->set_write(true);
+               }
+
+               if (next_act->is_read()) {
+                       /* Only need to store the locations of read actions */
+                       loc_inst_map->put(next_act->get_location(), next_inst);
+               }
+
+               inst_pred_map->put(next_inst, curr_pred);
+               set_predicate_tree_position(tid, curr_pred);
+
+               if (!inst_id_map->contains(next_inst))
+                       inst_id_map->put(next_inst, inst_counter++);
+
+               curr_pred->incr_expl_count();
+               add_predicate_to_trace(tid, curr_pred);
+               if (next_act->is_read())
+                       next_inst->set_associated_read(tid, recursion_depth, this_marker, next_act->get_reads_from_value());
+
+               break;
+       }
+
+       // A check
+       if (next_act->is_read()) {
+               if (selected_branch != NULL && !amended)
+                       ASSERT(selected_branch == curr_pred);
+       }
+}
+
+/* Given curr_pred and next_inst, find the branch following curr_pred that
+ * contains next_inst and the correct predicate.
+ * @return true if branch found, false otherwise.
+ */
+bool FuncNode::follow_branch(Predicate ** curr_pred, FuncInst * next_inst,
+                                                                                                                ModelAction * next_act, Predicate ** unset_predicate)
+{
+       /* Check if a branch with func_inst and corresponding predicate exists */
+       bool branch_found = false;
+       thread_id_t tid = next_act->get_tid();
+
+       ModelVector<Predicate *> * branches = (*curr_pred)->get_children();
+       for (uint i = 0;i < branches->size();i++) {
+               Predicate * branch = (*branches)[i];
+               if (branch->get_func_inst() != next_inst)
+                       continue;
+
+               /* Check against predicate expressions */
+               bool predicate_correct = true;
+               PredExprSet * pred_expressions = branch->get_pred_expressions();
+
+               /* Only read and rmw actions my have unset predicate expressions */
+               if (pred_expressions->getSize() == 0) {
+                       predicate_correct = false;
+
+                       if (*unset_predicate == NULL)
+                               *unset_predicate = branch;
+                       else
+                               ASSERT(false);
+
+                       continue;
+               }
+
+               PredExprSetIter * pred_expr_it = pred_expressions->iterator();
+               while (pred_expr_it->hasNext()) {
+                       pred_expr * pred_expression = pred_expr_it->next();
+                       uint64_t last_read, next_read;
+                       bool equality;
+
+                       switch(pred_expression->token) {
+                       case NOPREDICATE:
+                               predicate_correct = true;
+                               break;
+                       case EQUALITY:
+                               FuncInst * to_be_compared;
+                               to_be_compared = pred_expression->func_inst;
+
+                               last_read = get_associated_read(tid, to_be_compared);
+                               ASSERT(last_read != VALUE_NONE);
+
+                               next_read = next_act->get_reads_from_value();
+                               equality = (last_read == next_read);
+                               if (equality != pred_expression->value)
+                                       predicate_correct = false;
+
+                               break;
+                       case NULLITY:
+                               next_read = next_act->get_reads_from_value();
+                               // TODO: implement likely to be null
+                               equality = ( (void*) (next_read & 0xffffffff) == NULL);
+                               if (equality != pred_expression->value)
+                                       predicate_correct = false;
+                               break;
+                       default:
+                               predicate_correct = false;
+                               model_print("unkown predicate token\n");
+                               break;
+                       }
+               }
+
+               delete pred_expr_it;
+
+               if (predicate_correct) {
+                       *curr_pred = branch;
+                       branch_found = true;
+                       break;
+               }
+       }
+
+       return branch_found;
+}
+
+/* Infer predicate expressions, which are generated in FuncNode::generate_predicates */
+void FuncNode::infer_predicates(FuncInst * next_inst, ModelAction * next_act,
+                                                                                                                               SnapVector<struct half_pred_expr *> * half_pred_expressions)
+{
+       void * loc = next_act->get_location();
+       int thread_id = id_to_int(next_act->get_tid());
+       loc_inst_map_t * loc_inst_map = thrd_loc_inst_maps[thread_id]->back();
+
+       if (next_inst->is_read()) {
+               /* read + rmw */
+               if ( loc_inst_map->contains(loc) ) {
+                       FuncInst * last_inst = loc_inst_map->get(loc);
+                       struct half_pred_expr * expression = new half_pred_expr(EQUALITY, last_inst);
+                       half_pred_expressions->push_back(expression);
+               } else if ( next_inst->is_single_location() ) {
+                       loc_set_t * loc_may_equal = loc_may_equal_map->get(loc);
+
+                       if (loc_may_equal != NULL) {
+                               loc_set_iter * loc_it = loc_may_equal->iterator();
+                               while (loc_it->hasNext()) {
+                                       void * neighbor = loc_it->next();
+                                       if (loc_inst_map->contains(neighbor)) {
+                                               FuncInst * last_inst = loc_inst_map->get(neighbor);
+
+                                               struct half_pred_expr * expression = new half_pred_expr(EQUALITY, last_inst);
+                                               half_pred_expressions->push_back(expression);
+                                       }
+                               }
+
+                               delete loc_it;
+                       }
+               } else {
+                       // next_inst is not single location
+                       uint64_t read_val = next_act->get_reads_from_value();
+
+                       // only infer NULLITY predicate when it is actually NULL.
+                       if ( (void*)read_val == NULL) {
+                               struct half_pred_expr * expression = new half_pred_expr(NULLITY, NULL);
+                               half_pred_expressions->push_back(expression);
+                       }
+               }
+       } else {
+               /* Pure writes */
+               // TODO: do anything here?
+       }
+}
 
-       void * location = act->get_location();
-       uint64_t read_from_val = act->get_reads_from_value();
+/* Able to generate complex predicates when there are multiple predciate expressions */
+void FuncNode::generate_predicates(Predicate * curr_pred, FuncInst * next_inst,
+                                                                                                                                        SnapVector<struct half_pred_expr *> * half_pred_expressions)
+{
+       if (half_pred_expressions->size() == 0) {
+               Predicate * new_pred = new Predicate(next_inst);
+               curr_pred->add_child(new_pred);
+               new_pred->set_parent(curr_pred);
+
+               /* entry predicates and predicates containing pure write actions
+                * have no predicate expressions */
+               if ( curr_pred->is_entry_predicate() )
+                       new_pred->add_predicate_expr(NOPREDICATE, NULL, true);
+               else if (next_inst->is_write()) {
+                       /* next_inst->is_write() <==> pure writes */
+                       new_pred->add_predicate_expr(NOPREDICATE, NULL, true);
+               }
 
-       /* resize and initialize */
-       uint32_t old_size = thrd_read_map.size();
-       if (old_size <= tid) {
-               thrd_read_map.resize(tid + 1);
-               for (uint32_t i = old_size;i < tid + 1;i++)
-                       thrd_read_map[i] = new read_map_t();
+               return;
        }
 
-       read_map_t * read_map = thrd_read_map[tid];
-       read_map->put(location, read_from_val);
+       SnapVector<Predicate *> predicates;
+
+       struct half_pred_expr * half_expr = (*half_pred_expressions)[0];
+       predicates.push_back(new Predicate(next_inst));
+       predicates.push_back(new Predicate(next_inst));
+
+       predicates[0]->add_predicate_expr(half_expr->token, half_expr->func_inst, true);
+       predicates[1]->add_predicate_expr(half_expr->token, half_expr->func_inst, false);
+
+       for (uint i = 1;i < half_pred_expressions->size();i++) {
+               half_expr = (*half_pred_expressions)[i];
+
+               uint old_size = predicates.size();
+               for (uint j = 0;j < old_size;j++) {
+                       Predicate * pred = predicates[j];
+                       Predicate * new_pred = new Predicate(next_inst);
+                       new_pred->copy_predicate_expr(pred);
+
+                       pred->add_predicate_expr(half_expr->token, half_expr->func_inst, true);
+                       new_pred->add_predicate_expr(half_expr->token, half_expr->func_inst, false);
+
+                       predicates.push_back(new_pred);
+               }
+       }
 
-       /* Store the memory locations where atomic reads happen */
-       // read_locations.add(location);
+       for (uint i = 0;i < predicates.size();i++) {
+               Predicate * pred= predicates[i];
+               curr_pred->add_child(pred);
+               pred->set_parent(curr_pred);
+       }
+
+       /* Free memories allocated by infer_predicate */
+       for (uint i = 0;i < half_pred_expressions->size();i++) {
+               struct half_pred_expr * tmp = (*half_pred_expressions)[i];
+               snapshot_free(tmp);
+       }
 }
 
-uint64_t FuncNode::query_last_read(void * location, uint32_t tid)
+/* Amend predicates that contain no predicate expressions. Currenlty only amend with NULLITY predicates */
+bool FuncNode::amend_predicate_expr(Predicate * curr_pred, FuncInst * next_inst, ModelAction * next_act)
 {
-       if (thrd_read_map.size() <= tid)
-               return 0xdeadbeef;
+       ModelVector<Predicate *> * children = curr_pred->get_children();
+
+       Predicate * unset_pred = NULL;
+       for (uint i = 0;i < children->size();i++) {
+               Predicate * child = (*children)[i];
+               if (child->get_func_inst() == next_inst) {
+                       unset_pred = child;
+                       break;
+               }
+       }
+
+       uint64_t read_val = next_act->get_reads_from_value();
 
-       read_map_t * read_map = thrd_read_map[tid];
+       // only generate NULLITY predicate when it is actually NULL.
+       if ( !next_inst->is_single_location() && (void*)read_val == NULL ) {
+               Predicate * new_pred = new Predicate(next_inst);
 
-       /* last read value not found */
-       if ( !read_map->contains(location) )
-               return 0xdeadbeef;
+               curr_pred->add_child(new_pred);
+               new_pred->set_parent(curr_pred);
 
-       uint64_t read_val = read_map->get(location);
-       return read_val;
+               unset_pred->add_predicate_expr(NULLITY, NULL, false);
+               new_pred->add_predicate_expr(NULLITY, NULL, true);
+
+               return true;
+       }
+
+       return false;
 }
 
-/* @param tid thread id
- * Reset read map for a thread. This function shall only be called
- * when a thread exits a function
- */
-void FuncNode::clear_read_map(uint32_t tid)
+void FuncNode::add_to_val_loc_map(uint64_t val, void * loc)
 {
-       if (thrd_read_map.size() <= tid)
+       loc_set_t * locations = val_loc_map->get(val);
+
+       if (locations == NULL) {
+               locations = new loc_set_t();
+               val_loc_map->put(val, locations);
+       }
+
+       update_loc_may_equal_map(loc, locations);
+       locations->add(loc);
+       // values_may_read_from->add(val);
+}
+
+void FuncNode::add_to_val_loc_map(value_set_t * values, void * loc)
+{
+       if (values == NULL)
                return;
 
-       thrd_read_map[tid]->reset();
+       value_set_iter * it = values->iterator();
+       while (it->hasNext()) {
+               uint64_t val = it->next();
+               add_to_val_loc_map(val, loc);
+       }
+
+       delete it;
+}
+
+void FuncNode::update_loc_may_equal_map(void * new_loc, loc_set_t * old_locations)
+{
+       if ( old_locations->contains(new_loc) )
+               return;
+
+       loc_set_t * neighbors = loc_may_equal_map->get(new_loc);
+
+       if (neighbors == NULL) {
+               neighbors = new loc_set_t();
+               loc_may_equal_map->put(new_loc, neighbors);
+       }
+
+       loc_set_iter * loc_it = old_locations->iterator();
+       while (loc_it->hasNext()) {
+               // new_loc: { old_locations, ... }
+               void * member = loc_it->next();
+               neighbors->add(member);
+
+               // for each i in old_locations, i : { new_loc, ... }
+               loc_set_t * _neighbors = loc_may_equal_map->get(member);
+               if (_neighbors == NULL) {
+                       _neighbors = new loc_set_t();
+                       loc_may_equal_map->put(member, _neighbors);
+               }
+               _neighbors->add(new_loc);
+       }
+
+       delete loc_it;
+}
+
+void FuncNode::set_predicate_tree_position(thread_id_t tid, Predicate * pred)
+{
+       int thread_id = id_to_int(tid);
+       ModelVector<Predicate *> * stack = thrd_predicate_tree_position[thread_id];
+       (*stack)[stack->size() - 1] = pred;
+}
+
+/* @return The position of a thread in a predicate tree */
+Predicate * FuncNode::get_predicate_tree_position(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+       return thrd_predicate_tree_position[thread_id]->back();
 }
 
-void FuncNode::generate_predicate(FuncInst *func_inst)
+void FuncNode::add_predicate_to_trace(thread_id_t tid, Predicate * pred)
 {
+       int thread_id = id_to_int(tid);
+       thrd_predicate_trace[thread_id]->back()->push_back(pred);
+}
+
+void FuncNode::init_marker(thread_id_t tid)
+{
+       marker++;
+
+       int thread_id = id_to_int(tid);
+       int old_size = thrd_markers.size();
+
+       if (old_size < thread_id + 1) {
+               thrd_markers.resize(thread_id + 1);
+
+               for (int i = old_size; i < thread_id + 1; i++) {
+                       thrd_markers[i] = new ModelVector<uint32_t>();
+                       thrd_recursion_depth.push_back(-1);
+               }
+       }
 
+       thrd_markers[thread_id]->push_back(marker);
+       thrd_recursion_depth[thread_id]++;
 }
 
-/* @param tid thread id
- * Print the values read by the last read actions for each memory location
+uint64_t FuncNode::get_associated_read(thread_id_t tid, FuncInst * inst)
+{
+       int thread_id = id_to_int(tid);
+       int recursion_depth = thrd_recursion_depth[thread_id];
+       uint marker = thrd_markers[thread_id]->back();
+
+       return inst->get_associated_read(tid, recursion_depth, marker);
+}
+
+/* Make sure elements of maps are initialized properly when threads enter functions */
+void FuncNode::init_local_maps(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+       int old_size = thrd_loc_inst_maps.size();
+
+       if (old_size < thread_id + 1) {
+               int new_size = thread_id + 1;
+
+               thrd_loc_inst_maps.resize(new_size);
+               thrd_inst_id_maps.resize(new_size);
+               thrd_inst_pred_maps.resize(new_size);
+
+               for (int i = old_size; i < new_size; i++) {
+                       thrd_loc_inst_maps[i] = new ModelVector<loc_inst_map_t *>;
+                       thrd_inst_id_maps[i] = new ModelVector<inst_id_map_t *>;
+                       thrd_inst_pred_maps[i] = new ModelVector<inst_pred_map_t *>;
+               }
+       }
+
+       ModelVector<loc_inst_map_t *> * map = thrd_loc_inst_maps[thread_id];
+       int index = thrd_recursion_depth[thread_id];
+
+       // If there are recursive calls, push more hashtables into the vector.
+       if (map->size() < (uint) index + 1) {
+               thrd_loc_inst_maps[thread_id]->push_back(new loc_inst_map_t(64));
+               thrd_inst_id_maps[thread_id]->push_back(new inst_id_map_t(64));
+               thrd_inst_pred_maps[thread_id]->push_back(new inst_pred_map_t(64));
+       }
+
+       ASSERT(map->size() == (uint) index + 1);
+}
+
+/* Reset elements of maps when threads exit functions */
+void FuncNode::reset_local_maps(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+       int index = thrd_recursion_depth[thread_id];
+
+       // When recursive call ends, keep only one hashtable in the vector
+       if (index > 0) {
+               delete thrd_loc_inst_maps[thread_id]->back();
+               delete thrd_inst_id_maps[thread_id]->back();
+               delete thrd_inst_pred_maps[thread_id]->back();
+
+               thrd_loc_inst_maps[thread_id]->pop_back();
+               thrd_inst_id_maps[thread_id]->pop_back();
+               thrd_inst_pred_maps[thread_id]->pop_back();
+       } else {
+               thrd_loc_inst_maps[thread_id]->back()->reset();
+               thrd_inst_id_maps[thread_id]->back()->reset();
+               thrd_inst_pred_maps[thread_id]->back()->reset();
+       }
+}
+
+void FuncNode::init_predicate_tree_data_structure(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+       int old_size = thrd_predicate_tree_position.size();
+
+       if (old_size < thread_id + 1) {
+               thrd_predicate_tree_position.resize(thread_id + 1);
+               thrd_predicate_trace.resize(thread_id + 1);
+
+               for (int i = old_size; i < thread_id + 1; i++) {
+                       thrd_predicate_tree_position[i] = new ModelVector<Predicate *>();
+                       thrd_predicate_trace[i] = new ModelVector<predicate_trace_t *>();
+               }
+       }
+
+       thrd_predicate_tree_position[thread_id]->push_back(predicate_tree_entry);
+       thrd_predicate_trace[thread_id]->push_back(new predicate_trace_t());
+}
+
+void FuncNode::reset_predicate_tree_data_structure(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+       thrd_predicate_tree_position[thread_id]->pop_back();
+
+       // Free memories allocated in init_predicate_tree_data_structure
+       delete thrd_predicate_trace[thread_id]->back();
+       thrd_predicate_trace[thread_id]->pop_back();
+}
+
+/* Add FuncNodes that this node may follow */
+void FuncNode::add_out_edge(FuncNode * other)
+{
+       if ( !edge_table.contains(other) ) {
+               edge_table.put(other, OUT_EDGE);
+               out_edges.push_back(other);
+               return;
+       }
+
+       edge_type_t edge = edge_table.get(other);
+       if (edge == IN_EDGE) {
+               edge_table.put(other, BI_EDGE);
+               out_edges.push_back(other);
+       }
+}
+
+/* Compute the distance between this FuncNode and the target node.
+ * Return -1 if the target node is unreachable or the actual distance
+ * is greater than max_step.
  */
-void FuncNode::print_last_read(uint32_t tid)
-{
-       ASSERT(thrd_read_map.size() > tid);
-       read_map_t * read_map = thrd_read_map[tid];
-/*
-       mllnode<void *> * it;
-       for (it = read_locations.begin();it != NULL;it=it->getNext()) {
-               if ( !read_map->contains(it->getVal()) )
-                       break;
+int FuncNode::compute_distance(FuncNode * target, int max_step)
+{
+       if (target == NULL)
+               return -1;
+       else if (target == this)
+               return 0;
+
+       SnapList<FuncNode *> queue;
+       HashTable<FuncNode *, int, uintptr_t, 0> distances(128);
+
+       queue.push_back(this);
+       distances.put(this, 0);
+
+       while (!queue.empty()) {
+               FuncNode * curr = queue.front();
+               queue.pop_front();
+               int dist = distances.get(curr);
+
+               if (max_step <= dist)
+                       return -1;
 
-               uint64_t read_val = read_map->get(it->getVal());
-               model_print("last read of thread %d at %p: 0x%x\n", tid, it->getVal(), read_val);
+               ModelList<FuncNode *> * outEdges = curr->get_out_edges();
+               mllnode<FuncNode *> * it;
+               for (it = outEdges->begin();it != NULL;it = it->getNext()) {
+                       FuncNode * out_node = it->getVal();
+
+                       /* This node has not been visited before */
+                       if ( !distances.contains(out_node) ) {
+                               if (out_node == target)
+                                       return dist + 1;
+
+                               queue.push_back(out_node);
+                               distances.put(out_node, dist + 1);
+                       }
+               }
        }
-*/
+
+       /* Target node is unreachable */
+       return -1;
+}
+
+void FuncNode::update_predicate_tree_weight(thread_id_t tid)
+{
+       predicate_trace_t * trace = thrd_predicate_trace[id_to_int(tid)]->back();
+
+       // Update predicate weights based on prediate trace
+       for (mllnode<Predicate *> * rit = trace->end(); rit != NULL; rit = rit->getPrev()) {
+               Predicate * node = rit->getVal();
+               ModelVector<Predicate *> * children = node->get_children();
+
+               if (children->size() == 0) {
+                       double weight = 100.0 / sqrt(node->get_expl_count() + node->get_fail_count() + 1);
+                       node->set_weight(weight);
+               } else {
+                       double weight_sum = 0.0;
+                       for (uint i = 0;i < children->size();i++) {
+                               Predicate * child = (*children)[i];
+                               double weight = child->get_weight();
+                               weight_sum += weight;
+                       }
+
+                       double average_weight = (double) weight_sum / (double) children->size();
+                       double weight = average_weight * pow(0.9, node->get_depth());
+                       node->set_weight(weight);
+               }
+       }
+}
+
+void FuncNode::print_predicate_tree()
+{
+       model_print("digraph function_%s {\n", func_name);
+       predicate_tree_entry->print_pred_subtree();
+       predicate_tree_exit->print_predicate();
+       model_print("}\n");     // end of graph
 }