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
#[derive(Debug, Clone)]
pub enum Probabilistic<T> {
Pure(T),
Water {
failure: T,
success: T,
},
Sticks {
s0: T,
s1: T,
s2: T,
s3: T,
s4: T,
s5: T,
},
WhoGoesFirst {
ia_first: T,
a_first: T,
},
}
impl<T: Clone> Probabilistic<T> {
#[must_use]
pub fn choose(self) -> (T, Option<usize>) {
let prob: Prob<_> = self.into();
prob.choose()
}
#[must_use]
pub fn choose_when_no_ciurl(self) -> T {
let prob: Prob<_> = self.into();
match prob.choose() {
(t, None) => t,
_ => panic!("ciurl exists; call `choose` instead."),
}
}
}
impl<T: Clone> Into<Prob<(T, Option<usize>)>> for Probabilistic<T> {
fn into(self) -> Prob<(T, Option<usize>)> {
match self {
Probabilistic::WhoGoesFirst { ia_first, a_first } => Prob(vec![
((ia_first, None), 1.0 / 2.0),
((a_first, None), 1.0 / 2.0),
]),
Probabilistic::Pure(t) => Prob(vec![((t, None), 1.0)]),
Probabilistic::Water { failure, success } => Prob(vec![
((failure.clone(), Some(0)), 1.0 / 32.0),
((failure.clone(), Some(1)), 5.0 / 32.0),
((failure, Some(2)), 10.0 / 32.0),
((success.clone(), Some(3)), 10.0 / 32.0),
((success.clone(), Some(4)), 5.0 / 32.0),
((success, Some(5)), 1.0 / 32.0),
]),
Probabilistic::Sticks {
s0,
s1,
s2,
s3,
s4,
s5,
} => Prob(vec![
((s0, Some(0)), 1.0 / 32.0),
((s1, Some(1)), 5.0 / 32.0),
((s2, Some(2)), 10.0 / 32.0),
((s3, Some(3)), 10.0 / 32.0),
((s4, Some(4)), 5.0 / 32.0),
((s5, Some(5)), 1.0 / 32.0),
]),
}
}
}
#[readonly::make]
pub struct Prob<T>(pub Vec<(T, f64)>);
impl<T> Prob<T> {
#[must_use]
pub fn choose_by_uniform_random_variable(self, rand: f64) -> T {
if !(0.0..1.0).contains(&rand) {
panic!("Expects a float within the range of 0 up to but not including 1")
}
let mut threshold = 0.0;
for (t, prob) in self.0 {
if (threshold..(threshold + prob)).contains(&rand) {
return t;
}
threshold += prob;
}
panic!("Something went wrong in `choose_by_uniform_random_variable`")
}
#[must_use]
pub fn choose(self) -> T {
use rand::Rng;
let mut rng = rand::thread_rng();
self.choose_by_uniform_random_variable(rng.gen())
}
}