Add some methods for WaitObj

[c11tester.git] / waitobj.cc

diff --git a/waitobj.cc b/waitobj.cc
index e9299f59c9a68e12d01d4eaf5d4620803c9e033c..3aaa632e8bc2e4ddd6017f3639221867536e9375 100644 (file)
--- a/waitobj.cc
+++ b/waitobj.cc
@@ -7,6 +7,28 @@ WaitObj::WaitObj(thread_id_t tid) :
        dist_table(32)
 {}
 
        dist_table(32)
 {}
 

+void WaitObj::add_waiting_for(thread_id_t other, int dist)
+{
+       waiting_for.add(other);
+       dist_table.put(other, dist);
+}
+
+void WaitObj::add_waited_by(thread_id_t other)
+{
+       waited_by.add(other);
+}
+
+void WaitObj::remove_waiting_for(thread_id_t other)
+{
+       waiting_for.remove(other);
+       dist_table.remove(other);
+}
+
+void WaitObj::remove_waited_by(thread_id_t other)
+{
+       waited_by.remove(other);
+}
+

 int WaitObj::lookup_dist(thread_id_t other_id)
 {
        if (dist_table.contains(other_id))
 int WaitObj::lookup_dist(thread_id_t other_id)
 {
        if (dist_table.contains(other_id))


@@ -15,6 +37,13 @@ int WaitObj::lookup_dist(thread_id_t other_id)
        return -1;
 }
 
        return -1;
 }
 

+void WaitObj::reset()
+{
+       waiting_for.reset();
+       waited_by.reset();
+       dist_table.reset();
+}
+

 void WaitObj::print_waiting_for()
 {
        if (waiting_for.getSize() == 0)
 void WaitObj::print_waiting_for()
 {
        if (waiting_for.getSize() == 0)