-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathEdge.cpp
58 lines (46 loc) · 891 Bytes
/
Edge.cpp
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
#include <iostream>
#include "Edge.h"
using namespace std;
/*
* Message methods.
*/
Message::Message(int code, string message) {
_code = code;
_msg = message;
}
Message::~Message() {
}
/*
* Pipe Methods.
*/
Pipe::Pipe(NodeID id) {
_id = id;
}
void Pipe::writeMsg(Message *msg) {
unique_lock<mutex> lock (_mutex);
_queue.push(*msg);
}
/*
* Edge methods.
*/
Edge::Edge(Node *a, Node*b, int w) {
_a = a;
_b = b;
_weight = w;
// Initialize Queue.
_pipe_a = new Pipe(_a -> _id);
_pipe_b = new Pipe(_b -> _id);
// Initialize mutex and cv.
}
void Edge::sendMessage(Pipe *pipe, Message *msg) {
pipe -> writeMsg(msg);
}
// TODO Do we need this to be thread safe.
int Edge::getState() {
unique_lock<mutex> lock(_mu);
return _state;
}
void Edge::setState(int state) {
unique_lock<mutex> lock(_mu);
_state = state;
}