diff --git a/src/entity/item/unit.rs b/src/entity/item/unit.rs index cc973ed..371f450 100644 --- a/src/entity/item/unit.rs +++ b/src/entity/item/unit.rs @@ -446,4 +446,14 @@ impl Unit { | UnitType::PbIncrease ) } + + pub fn modifier_stars(&self) -> i8 { + match self.modifier { + Some(UnitModifier::PlusPlus) => 1, + Some(UnitModifier::Plus) => 1, + Some(UnitModifier::Minus) => -1, + Some(UnitModifier::MinusMinus) => -1, + _ => 0, + } + } }