barrier/barrier.cpp at master · P2Tree/barrier · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include <iostream>

#include "barrier.h"

Barrier::Barrier(int Expect) : Expect(Expect), Count(0), Release(0) {
  for (unsigned I = 0; I < Expect; I++) {
    Actives[I] = 0;
  }
}

void Barrier::handleBarSync(unsigned PC, unsigned TID) {
  wait(Expect, PC, TID);
}

void Barrier::handleBarSync(unsigned Expect, unsigned PC, unsigned TID) {
  wait(Expect, PC, TID);
}

void Barrier::handleBarProduce(unsigned Expect, unsigned PC) {
  produce(Expect, PC);
}

void Barrier::handleBarConsume(unsigned PC, unsigned TID) {
  consume(PC, TID);
}

void Barrier::wait(unsigned Expect, unsigned PC, unsigned TID) {
  unique_lock<mutex> Lock(MutexLock);
  if (Actives[TID] != Release) {
    Actives[TID]++;
#ifdef DEBUG
    cout << "Dropped sync of " << TID << " in " << PC << ", Active = " << Actives[TID] << endl;
#endif
    return;
  }
  Count++;
  Actives[TID]++;
#ifdef DEBUG
  cout << "Enter sync of " << TID << " in " << PC << ", Active = " << Actives[TID] << ", Count = " << Count << endl;
#endif
  if (Count == Expect) {
    Count = 0;
    Release++;
    Cond.notify_all();
#ifdef DEBUG
    cout << "Notify sync of " << TID << " in " << PC << ", Release = " << Release << ", Count = " << Count << endl;
#endif
  } else {
#ifdef DEBUG
    cout << "Wait sync of " << TID << " in " << PC << endl;
#endif
    unsigned LocalRelease = Release + 1;
    Cond.wait(Lock, [&]{ return LocalRelease - Release <= 0; });
#ifdef DEBUG
    cout << "Awake sync of " << TID << " in " << PC << endl;
#endif
  }
}

void Barrier::produce(unsigned Expect, unsigned PC) {
  unique_lock<mutex> Lock(MutexLock);
  Count = Expect;
#ifdef DEBUG
  cout << "Enter produce in " << PC << ", Count = " << Count << endl;
#endif
  Cond.notify_all();
#ifdef DEBUG
  cout << "Notify produce in " << PC << ", Count = " << Count << endl;
#endif
}

void Barrier::consume(unsigned PC, unsigned TID) {
  unique_lock<mutex> Lock(MutexLock);
  Actives[TID]++;
#ifdef DEBUG
  cout << "Enter consume of " << TID << " in " << PC << ", Active = " << Actives[TID] << ", Count = " << Count << endl;
#endif
  if (Count != 0) {
    Count--;
#ifdef DEBUG
    cout << "Decrease consume of " << TID << " in " << PC << ", Count = " << Count << endl;
#endif
  } else {
#ifdef DEBUG
    cout << "Wait consume of " << TID << " in " << PC << endl;
#endif
    Cond.wait(Lock, [&]{ return Count != 0; });
    Count--;
#ifdef DEBUG
    cout << "Awake consume of " << TID << " in " << PC << ", Count = " << Count << endl;
#endif
  }
}