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
use serde::{Deserialize, Serialize};
use std::collections::BTreeSet;
#[derive(Debug, Default, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub struct Capabilities(BTreeSet<Capability>);
impl Capabilities {
pub fn new<I>(capabilities: I) -> Self
where
I: Iterator<Item = Capability>,
{
Self(capabilities.collect())
}
pub fn all() -> Self {
let all = [
Capability::UniformContinuous,
Capability::UniformDiscrete,
Capability::LogUniformContinuous,
Capability::LogUniformDiscrete,
Capability::Categorical,
Capability::Conditional,
Capability::MultiObjective,
Capability::Concurrent,
]
.iter()
.copied()
.collect();
Self(all)
}
pub fn empty() -> Self {
Self(BTreeSet::new())
}
pub fn is_empty(&self) -> bool {
self.0.is_empty()
}
pub fn is_capable(&self, c: Capability) -> bool {
self.0.contains(&c)
}
pub fn incapables<'a>(&'a self, required: &'a Self) -> impl 'a + Iterator<Item = Capability> {
required.0.difference(&self.0).copied()
}
pub fn iter<'a>(&'a self) -> impl 'a + Iterator<Item = Capability> {
self.0.iter().copied()
}
pub fn add_capability(&mut self, c: Capability) -> &mut Self {
self.0.insert(c);
self
}
pub fn remove_capability(&mut self, c: Capability) -> &mut Self {
self.0.remove(&c);
self
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord, Serialize, Deserialize)]
#[allow(missing_docs)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum Capability {
UniformContinuous,
UniformDiscrete,
LogUniformContinuous,
LogUniformDiscrete,
Categorical,
Conditional,
MultiObjective,
Concurrent,
}