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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
use super::{Message, ResultMessage, RowResult};
use std::collections::{HashMap, HashSet};
use std::sync::mpsc::{channel, Receiver, Sender};
use std::thread::JoinHandle;
struct InBuffer {
receiver: Receiver<Message>,
row_requests: HashMap<u64, Vec<Vec<u32>>>,
results_requests: HashMap<u64, ResultMessage>,
}
impl InBuffer {
fn new(receiver: Receiver<Message>) -> Self {
Self {
receiver,
row_requests: HashMap::new(),
results_requests: HashMap::new(),
}
}
fn read(&mut self) -> Option<Vec<RowResult>> {
loop {
match self
.receiver
.recv()
.expect("Channel to Output Daemon broke")
{
Message::ResultMessage(results) => {
if let Some(result_walls) = self.row_requests.get(&results.id) {
return Some(Self::calc_results(
results.valid_walls().as_ref(),
result_walls,
));
} else {
self.results_requests.insert(results.id, results);
}
}
Message::OutputMessage((id, output)) => {
if let Some(results) = self.results_requests.get(&id) {
return Some(Self::calc_results(
results.valid_walls().as_ref(),
output.as_ref(),
));
} else {
self.row_requests.insert(id, output);
}
}
Message::CpuDone => {
return None;
}
Message::Terminate => {
return None;
}
_ => {
println!("Invalid MessageType");
}
}
}
}
fn calc_results(res_req: &[Vec<u32>], row_req: &[Vec<u32>]) -> Vec<RowResult> {
let mut out = Vec::new();
for (rows, perms) in row_req.iter().zip(res_req.iter()) {
for p in perms {
let mut new = rows.clone();
new.push(*p);
out.push(RowResult::new(new));
}
}
out
}
}
pub struct Output {
input: InBuffer,
permutations: Vec<Vec<u32>>,
permutations_mask: Vec<u64>,
results: HashSet<RowResult>,
result_sender: Sender<Message>,
}
impl Output {
pub fn launch_sevice(
permutations: &[Vec<u32>],
permutations_mask: &[u64],
result_sender: Sender<Message>,
) -> (Sender<Message>, JoinHandle<()>) {
let (sender, receiver) = channel();
let input = InBuffer::new(receiver);
let output = Self {
input,
permutations: permutations.into(),
permutations_mask: permutations_mask.into(),
results: HashSet::new(),
result_sender,
};
(
sender,
std::thread::Builder::new()
.name("GPU Output Deamon".into())
.spawn(move || {
output.run();
})
.unwrap(),
)
}
fn run(mut self) {
loop {
if let Some(walls) = self.input.read() {
for wall in walls {
/*if !self.results.contains(&wall) {
self.result_sender
.send(Message::RowResult(wall.clone()))
.or_else(|_| Err(println!("Failed to transmit result back")));
}*/
self.results.insert(wall);
}
} else {
for wall in self.results {
wall.output()
}
self.result_sender.send(Message::GpuDone).unwrap();
// wait for second exit signal
self.input.read();
return;
}
}
}
}
|