diff --git a/ui/loadout_manager.py b/ui/loadout_manager.py index 83f1ef4..8625fbe 100644 --- a/ui/loadout_manager.py +++ b/ui/loadout_manager.py @@ -140,6 +140,14 @@ class LoadoutConfig: clothing_items: List[str] = field(default_factory=list) pet: Optional[str] = None + # Mindforce + mindforce_implant: Optional[str] = None + mindforce_decay_pec: Decimal = Decimal("0") + + # Enhancers - tier-based (1 per tier, max 10 tiers) + # Format: {tier_number: enhancer} + enhancers: Dict[int, 'NexusEnhancer'] = field(default_factory=dict) + # Settings shots_per_hour: int = 3600 hits_per_hour: int = 720 @@ -161,9 +169,11 @@ class LoadoutConfig: total += self.weapon_scope.decay_pec if self.weapon_absorber: total += self.weapon_absorber.decay_pec - # Add enhancer decay (assuming 1 use per shot) - for enhancer in self.weapon_enhancers: - total += enhancer.decay_pec + # Add enhancer decay from all tiers + for tier, enhancer in self.enhancers.items(): + total += enhancer.decay + # Add mindforce decay if used + total += self.mindforce_decay_pec return total def get_total_ammo_per_shot(self) -> Decimal: