use crate::{comp::Body, sync::Uid, DamageSource}; use serde::{Deserialize, Serialize}; use specs::{Component, FlaggedStorage}; use specs_idvs::IdvStorage; #[derive(Clone, Copy, Debug, Serialize, Deserialize)] pub struct Poise { base_max: u32, current: u32, maximum: u32, pub is_interrupted: bool, pub is_stunned: bool, pub is_dazed: bool, pub is_knockeddown: bool, } impl Default for Poise { fn default() -> Self { Self { current: 0, maximum: 0, base_max: 0, is_interrupted: false, is_stunned: false, is_dazed: false, is_knockeddown: false, } } } #[derive(Clone, Copy, Debug, Serialize, Deserialize)] pub enum PoiseState { Normal, Interrupted, Stunned, Dazed, KnockedDown, } impl Poise { pub fn new(body: Body) -> Self { let mut poise = Poise::default(); poise.update_max_poise(Some(body)); poise.set_to(poise.maximum()); poise } pub fn poise_state(&self) -> PoiseState { if self.current >= 5 * self.maximum / 10 { PoiseState::Normal } else if self.current >= 4 * self.maximum / 10 { PoiseState::Interrupted } else if self.current >= 3 * self.maximum / 10 { PoiseState::Stunned } else if self.current >= 2 * self.maximum / 10 { PoiseState::Dazed } else { PoiseState::KnockedDown } } pub fn current(&self) -> u32 { self.current } pub fn maximum(&self) -> u32 { self.maximum } pub fn set_to(&mut self, amount: u32) { let amount = amount.min(self.maximum); self.current = amount; } pub fn change_by(&mut self, change: i32) { self.current = ((self.current as i32 + change).max(0) as u32).min(self.maximum); } pub fn reset(&mut self) { self.current = self.maximum; } pub fn set_maximum(&mut self, amount: u32) { self.maximum = amount; self.current = self.current.min(self.maximum); } fn set_base_max(&mut self, amount: u32) { self.base_max = amount; self.current = self.current.min(self.maximum); } pub fn reset_max(&mut self) { self.maximum = self.base_max; } pub fn update_max_poise(&mut self, body: Option) { if let Some(body) = body { self.set_base_max(body.base_poise()); self.set_maximum(body.base_poise()); } } pub fn with_max_poise(mut self, amount: u32) -> Self { self.maximum = amount; self.current = amount; self } } impl Component for Poise { type Storage = FlaggedStorage>; }