diff options
| -rw-r--r-- | src/rules/ability_scores.rs | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/src/rules/ability_scores.rs b/src/rules/ability_scores.rs index 205c1a3..2b9acf0 100644 --- a/src/rules/ability_scores.rs +++ b/src/rules/ability_scores.rs @@ -26,7 +26,7 @@ impl AbilityScore { #[derive(Debug)] pub struct AbilityScoreCollection { - scores: HashMap<AbilityScore, Vec<u32>>, + scores: HashMap<AbilityScore, u32>, } impl AbilityScoreCollection { @@ -36,14 +36,11 @@ impl AbilityScoreCollection { } } - pub fn add_score(&mut self, ability_score: AbilityScore, bonus: u32) { - self.scores - .entry(ability_score) - .or_insert_with(Vec::new) - .push(bonus); + pub fn add_score(&mut self, ability_score: AbilityScore, value: u32) { + self.scores.insert(ability_score, value); } - pub fn get_score(&self, ability_score: AbilityScore) -> Option<&Vec<u32>> { + pub fn get_score(&self, ability_score: AbilityScore) -> Option<&u32> { self.scores.get(&ability_score) } } |