do optimization for FuncNode::update_tree
[c11tester.git] / funcnode.cc
index 6b1f199792f299cb64ffdbda5143ddd2ca1acf23..188ed173c3c0bfda3bc60fad06c7d781fc4f5fbc 100644 (file)
@@ -1,30 +1,30 @@
 #include "funcnode.h"
 
 FuncNode::FuncNode() :
+       predicate_tree_initialized(false),
+       predicate_tree_entry(new Predicate(NULL, true)),
        func_inst_map(),
        inst_list(),
        entry_insts(),
-       thrd_read_map(),
-       read_locations()
+       thrd_read_map()
 {}
 
-/* 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.
+/* Check whether FuncInst with the same type, position, and location
+ * as act has been added to func_inst_map or not. If not, add it.
  *
- * @return FuncInst with the same type, position, and location as act */
-FuncInst * FuncNode::get_or_add_action(ModelAction *act)
+ * Note: currently, actions with the same position are filtered out by process_action,
+ * so the collision list of FuncInst is not used. May remove it later. 
+ */
+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;
+               return;
 
        if ( func_inst_map.contains(position) ) {
                FuncInst * inst = func_inst_map.get(position);
@@ -33,73 +33,137 @@ FuncInst * FuncNode::get_or_add_action(ModelAction *act)
                        // model_print("action with a different type occurs at line number %s\n", position);
                        FuncInst * func_inst = inst->search_in_collision(act);
 
-                       if (func_inst != NULL) {
-                               // return the FuncInst found in the collision list
-                               return func_inst;
-                       }
+                       if (func_inst != NULL)
+                               return;
 
                        func_inst = new FuncInst(act, this);
                        inst->get_collisions()->push_back(func_inst);
-                       inst_list.push_back(func_inst);         // delete?
-
-                       return func_inst;
+                       inst_list.push_back(func_inst); // delete?
                }
 
-               return inst;
+               return;
        }
 
        FuncInst * func_inst = new FuncInst(act, this);
 
        func_inst_map.put(position, func_inst);
        inst_list.push_back(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;
+
+//     ASSERT(inst->get_location() == act->get_location());
+
+       action_type inst_type = inst->get_type();
+       action_type act_type = act->get_type();
 
-       return func_inst;
+       // else if branch: an RMWRCAS action is converted to a RMW or READ action
+       if (inst_type == act_type)
+               return inst;
+       else if (inst_type == ATOMIC_RMWRCAS &&
+                       (act_type == ATOMIC_RMW || act_type == ATOMIC_READ))
+               return inst;
+
+       return NULL;
 }
 
+
 void FuncNode::add_entry_inst(FuncInst * inst)
 {
        if (inst == NULL)
                return;
 
-       func_inst_list_mt::iterator it;
-       for (it = entry_insts.begin(); it != entry_insts.end(); it++) {
-               if (inst == *it)
+       mllnode<FuncInst *> * it;
+       for (it = entry_insts.begin(); it != NULL; it = it->getNext()) {
+               if (inst == it->getVal())
                        return;
        }
 
        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
+/**
+ * @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(action_list_t * act_list)
 {
-       if (inst_list == NULL)
+       if (act_list == NULL)
+               return;
+       else if (act_list->size() == 0)
                return;
 
-       func_inst_list_t::iterator it = inst_list->begin();
-       func_inst_list_t::iterator prev;
+       /* build inst_list from act_list for later processing */
+       func_inst_list_t inst_list;
+       action_list_t read_act_list;
+
+       for (sllnode<ModelAction *> * it = act_list->begin(); it != NULL; it = it->getNext()) {
+               ModelAction * act = it->getVal();
+               FuncInst * func_inst = get_inst(act);
+
+               if (func_inst == NULL)
+                       continue;
+
+               inst_list.push_back(func_inst);
+
+//             model_print("position: %s ", act->get_position());
+//             act->print();
+
+               if (func_inst->is_read())
+                       read_act_list.push_back(act);
+       }
 
-       if (inst_list->size() == 0)
+       update_inst_tree(&inst_list);
+       update_predicate_tree(&read_act_list);
+}
+
+/** 
+ * @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;
+
        /* add the first instruction to the list of entry insts */
-       FuncInst * entry_inst = *it;
+       FuncInst * entry_inst = it->getVal();
        add_entry_inst(entry_inst);
 
-       it++;
-       while (it != inst_list->end()) {
-               prev = it;
-               prev--;
+       it = it->getNext();
+       while (it != NULL) {
+               prev = it->getPrev();
 
-               FuncInst * prev_inst = *prev;
-               FuncInst * curr_inst = *it;
+               FuncInst * prev_inst = prev->getVal();
+               FuncInst * curr_inst = it->getVal();
 
                prev_inst->add_succ(curr_inst);
                curr_inst->add_pred(prev_inst);
 
-               it++;
+               it = it->getNext();
        }
 }
 
@@ -112,42 +176,31 @@ void FuncNode::store_read(ModelAction * act, uint32_t tid)
        void * location = act->get_location();
        uint64_t read_from_val = act->get_reads_from_value();
 
-       if (thrd_read_map.size() <= tid)
+       /* resize and initialize */
+       uint32_t old_size = thrd_read_map.size();
+       if (old_size <= tid) {
                thrd_read_map.resize(tid + 1);
-
-       read_map_t * read_map = thrd_read_map[tid];
-       if (read_map == NULL) {
-               read_map = new read_map_t();
-               thrd_read_map[tid] = read_map;
+               for (uint32_t i = old_size; i < tid + 1;i++)
+                       thrd_read_map[i] = new read_map_t();
        }
 
+       read_map_t * read_map = thrd_read_map[tid];
        read_map->put(location, read_from_val);
 
        /* Store the memory locations where atomic reads happen */
-       bool push_loc = true;
-       ModelList<void *>::iterator it;
-       for (it = read_locations.begin(); it != read_locations.end(); it++) {
-               if (location == *it) {
-                       push_loc = false;
-                       break;
-               }
-       }
-
-       if (push_loc)
-               read_locations.push_back(location);
+       // read_locations.add(location);
 }
 
-uint64_t FuncNode::query_last_read(ModelAction * act, uint32_t tid)
+uint64_t FuncNode::query_last_read(void * location, uint32_t tid)
 {
        if (thrd_read_map.size() <= tid)
-               return 0xdeadbeef;
+               return VALUE_NONE;
 
        read_map_t * read_map = thrd_read_map[tid];
-       void * location = act->get_location();
 
        /* last read value not found */
        if ( !read_map->contains(location) )
-               return 0xdeadbeef;
+               return VALUE_NONE;
 
        uint64_t read_val = read_map->get(location);
        return read_val;
@@ -159,24 +212,174 @@ uint64_t FuncNode::query_last_read(ModelAction * act, uint32_t tid)
  */
 void FuncNode::clear_read_map(uint32_t tid)
 {
-       ASSERT(thrd_read_map.size() > tid);
+       if (thrd_read_map.size() <= tid)
+               return;
+
        thrd_read_map[tid]->reset();
 }
 
+void FuncNode::update_predicate_tree(action_list_t * act_list)
+{
+       if (act_list == NULL || act_list->size() == 0)
+               return;
+/*
+       if (predicate_tree_initialized) {
+               return;
+       }
+       predicate_tree_initialized = true;
+*/
+       /* map a FuncInst to the parent of its predicate */
+       HashTable<FuncInst *, Predicate *, uintptr_t, 0> inst_pred_map(128);
+       HashTable<void *, ModelAction *, uintptr_t, 0> loc_act_map(128);
+
+       sllnode<ModelAction *> *it = act_list->begin();
+       Predicate * curr_pred = predicate_tree_entry;
+
+       while (it != NULL) {
+               ModelAction * next_act = it->getVal();
+               FuncInst * next_inst = get_inst(next_act);
+               Predicate * old_pred = curr_pred;
+
+               bool branch_found = follow_branch(&curr_pred, next_inst, next_act, &loc_act_map);
+
+               // check back edges
+               if (!branch_found) {
+                       Predicate * back_pred = curr_pred->get_backedge();
+                       if (back_pred != NULL) {
+                               curr_pred = back_pred;
+                               continue;
+                       }
+
+                       if (inst_pred_map.contains(next_inst)) {
+                               back_pred = inst_pred_map.get(next_inst);
+                               curr_pred->set_backedge(back_pred);
+                               curr_pred = back_pred;
+                               continue;
+                       }
+               }
+
+               if (!inst_pred_map.contains(next_inst))
+                       inst_pred_map.put(next_inst, old_pred);
+
+               if (!branch_found) {
+                       if ( loc_act_map.contains(next_act->get_location()) ) {
+                               Predicate * new_pred1 = new Predicate(next_inst);
+                               new_pred1->add_predicate(EQUALITY, next_act->get_location(), true);
+
+                               Predicate * new_pred2 = new Predicate(next_inst);
+                               new_pred2->add_predicate(EQUALITY, next_act->get_location(), false);
+
+                               curr_pred->add_child(new_pred1);
+                               curr_pred->add_child(new_pred2);
+                               //new_pred1->add_parent(curr_pred);
+                               //new_pred2->add_parent(curr_pred);
+
+                               ModelAction * last_act = loc_act_map.get(next_act->get_location());
+                               uint64_t last_read = last_act->get_reads_from_value();
+                               uint64_t next_read = next_act->get_reads_from_value();
+
+                               if ( last_read == next_read )
+                                       curr_pred = new_pred1;
+                               else
+                                       curr_pred = new_pred2;
+                       } else {
+                               Predicate * new_pred = new Predicate(next_inst);
+                               curr_pred->add_child(new_pred);
+                               //new_pred->add_parent(curr_pred);
+
+                               curr_pred = new_pred;
+                       }
+               }
+
+               loc_act_map.put(next_act->get_location(), next_act);
+               it = it->getNext();
+       }
+
+//     model_print("function %s\n", func_name);
+//     print_predicate_tree();
+}
+
+/* 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,
+       HashTable<void *, ModelAction *, uintptr_t, 0> * loc_act_map)
+{
+       /* check if a branch with func_inst and corresponding predicate exists */
+       bool branch_found = false;
+       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;
+
+               PredExprSet * pred_expressions = branch->get_pred_expressions();
+
+               /* no predicate, follow the only branch */
+               if (pred_expressions->getSize() == 0) {
+//                     model_print("no predicate exists: "); next_inst->print();
+                       *curr_pred = branch;
+                       branch_found = true;
+                       break;
+               }
+
+               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;
+                       ModelAction * last_act;
+                       bool equality;
+
+                       switch(pred_expression->token) {
+                               case EQUALITY:
+                                       last_act = loc_act_map->get(next_act->get_location());
+                                       last_read = last_act->get_reads_from_value();
+                                       next_read = next_act->get_reads_from_value();
+
+                                       equality = (last_read == next_read);
+
+                                       if (equality == pred_expression->value) {
+                                               *curr_pred = branch;
+//                                             model_print("predicate: token: %d, location: %p, value: %d - ", pred_expression->token, pred_expression->location, pred_expression->value); next_inst->print();
+                                               branch_found = true;
+                                       }
+                                       break;
+                               case NULLITY:
+                                       break;
+                               default:
+                                       model_print("unkown predicate token\n");
+                                       break;
+                       }
+               }
+
+       }
+
+       return branch_found;
+}
+
+void FuncNode::print_predicate_tree()
+{
+       model_print("digraph function_%s {\n", func_name);
+       predicate_tree_entry->print_pred_subtree();
+       model_print("}\n");     // end of graph
+}
+
 /* @param tid thread id
- * Print the values read by the last read actions per memory location
+ * Print the values read by the last read actions for each memory location
  */
+/*
 void FuncNode::print_last_read(uint32_t tid)
 {
        ASSERT(thrd_read_map.size() > tid);
        read_map_t * read_map = thrd_read_map[tid];
 
-       ModelList<void *>::iterator it;
-       for (it = read_locations.begin(); it != read_locations.end(); it++) {
-               if ( !read_map->contains(*it) )
+       mllnode<void *> * it;
+       for (it = read_locations.begin();it != NULL;it=it->getNext()) {
+               if ( !read_map->contains(it->getVal()) )
                        break;
 
-               uint64_t read_val = read_map->get(*it);
-               model_print("last read of thread %d at %p: 0x%x\n", tid, *it, read_val);
+               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);
        }
 }
+*/