-
-
Notifications
You must be signed in to change notification settings - Fork 4
/
inn.rs
114 lines (105 loc) · 3.35 KB
/
inn.rs
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
use crate::utils::pluralize;
use crate::world::{word, word::ListGenerator, Demographics, Place};
use rand::prelude::*;
pub fn generate(place: &mut Place, rng: &mut impl Rng, _demographics: &Demographics) {
place.name.replace_with(|_| name(rng));
}
fn name(rng: &mut impl Rng) -> String {
match rng.gen_range(0..6) {
0 => format!("The {}", thing(rng)),
1 => {
let (profession, s) = pluralize(word::profession(rng));
format!("{}{} Arms", profession, s)
}
2..=3 => {
let (thing1, thing2) = thing_thing(rng);
format!("{} and {}", thing1, thing2)
}
4 => format!("The {} {}", word::adjective(rng), thing(rng)),
5 => {
let (thing, s) = pluralize(thing(rng));
format!("{} {}{}", number(rng), thing, s)
}
_ => unreachable!(),
}
}
fn thing(rng: &mut impl Rng) -> &'static str {
match rng.gen_range(0..5) {
0 => word::animal(rng),
1 => word::enemy(rng),
2 => word::food(rng),
3 => word::profession(rng),
4 => word::symbol(rng),
_ => unreachable!(),
}
}
fn thing_thing(rng: &mut impl Rng) -> (&'static str, &'static str) {
// We're more likely to have two things in the same category.
let (thing1, thing2) = if rng.gen_bool(0.5) {
match rng.gen_range(0..5) {
0 => (word::animal(rng), word::animal(rng)),
1 => (word::enemy(rng), word::enemy(rng)),
2 => (word::food(rng), word::food(rng)),
3 => (word::profession(rng), word::profession(rng)),
4 => (word::symbol(rng), word::symbol(rng)),
_ => unreachable!(),
}
} else {
(thing(rng), thing(rng))
};
// 50% chance of rolling again if we don't get two words starting with the same letter.
// (This is distinct from 50% chance of repeated letters, since the next try probably also
// won't have repetition.)
if thing1 == thing2
|| rng.gen_bool(0.5)
&& thing1
.chars()
.next()
.map(|c| !thing2.starts_with(c))
.unwrap_or(false)
{
thing_thing(rng)
} else {
(thing1, thing2)
}
}
#[rustfmt::skip]
fn number(rng: &mut impl Rng) -> &'static str {
ListGenerator(&["Three", "Five", "Seven", "Ten"]).gen(rng)
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn name_test() {
let mut rng = SmallRng::seed_from_u64(0);
assert_eq!(
[
"Mutton and Malt",
"Wizards Arms",
"The Column",
"Coopers Arms",
"The Orange Unicorn",
"Imp and Satyr",
"Otter and Rye",
"Seven Printers",
"Shovel and Crown",
"Seven Porters",
"The Bread",
"Mace and Phalactary",
"Three Kegs",
"Blacksmiths Arms",
"Ten Beggars",
"Bandit and Hydra",
"The Burgundy Potatoes",
"The Green Cooper",
"The Giant",
"The Lucky Wheel"
]
.iter()
.map(|s| s.to_string())
.collect::<Vec<_>>(),
(0..20).map(|_| name(&mut rng)).collect::<Vec<String>>(),
);
}
}