304 lines
9.4 KiB
Rust
304 lines
9.4 KiB
Rust
use std::cmp::Ordering;
|
|
|
|
use crate::engine_constants::EngineConstants;
|
|
use crate::shared_game_state::SharedGameState;
|
|
use crate::weapon::{Weapon, WeaponLevel, WeaponType};
|
|
use crate::player::Player;
|
|
use crate::caret::CaretType;
|
|
use crate::common::Direction;
|
|
|
|
#[derive(Clone, Copy)]
|
|
/// (id, amount)
|
|
pub struct Item(pub u16, pub u16);
|
|
|
|
#[derive(Clone)]
|
|
pub struct Inventory {
|
|
pub current_item: u16,
|
|
pub current_weapon: u16,
|
|
items: Vec<Item>,
|
|
weapons: Vec<Weapon>,
|
|
}
|
|
|
|
#[derive(Clone, Copy)]
|
|
pub enum TakeExperienceResult {
|
|
None,
|
|
LevelDown,
|
|
}
|
|
|
|
impl Inventory {
|
|
#[allow(clippy::new_without_default)]
|
|
pub fn new() -> Inventory {
|
|
Inventory {
|
|
current_item: 0,
|
|
current_weapon: 0,
|
|
items: Vec::with_capacity(16),
|
|
weapons: Vec::with_capacity(16),
|
|
}
|
|
}
|
|
|
|
pub fn add_item(&mut self, item_id: u16) {
|
|
if !self.has_item(item_id) {
|
|
self.items.push(Item(item_id, 1));
|
|
} else {
|
|
if let Some(item) = self.get_item(item_id) {
|
|
item.1 += 1;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn consume_item(&mut self, item_id: u16) {
|
|
if let Some(item) = self.get_item(item_id) {
|
|
if item.1 > 1 {
|
|
item.1 -= 1;
|
|
return;
|
|
}
|
|
}
|
|
|
|
self.remove_item(item_id);
|
|
}
|
|
|
|
pub fn remove_item(&mut self, item_id: u16) {
|
|
self.items.retain(|item| item.0 != item_id);
|
|
}
|
|
|
|
pub fn get_item(&mut self, item_id: u16) -> Option<&mut Item> {
|
|
self.items.iter_mut().by_ref().find(|item| item.0 == item_id)
|
|
}
|
|
|
|
pub fn get_item_idx(&self, idx: usize) -> Option<&Item> {
|
|
self.items.get(idx)
|
|
}
|
|
|
|
pub fn has_item(&self, item_id: u16) -> bool {
|
|
self.items.iter().any(|item| item.0 == item_id)
|
|
}
|
|
|
|
pub fn has_item_amount(&self, item_id: u16, operator: Ordering, amount: u16) -> bool {
|
|
let result = self.items.iter().any(|item| item.0 == item_id && item.1.cmp(&amount) == operator);
|
|
|
|
if (operator == Ordering::Equal && amount == 0 && !result) || (operator == Ordering::Less && !self.has_item(item_id)) {
|
|
return true;
|
|
}
|
|
|
|
result
|
|
}
|
|
|
|
pub fn add_weapon(&mut self, weapon_id: WeaponType, max_ammo: u16) -> &mut Weapon {
|
|
if !self.has_weapon(weapon_id) {
|
|
self.weapons.push(Weapon::new(
|
|
weapon_id,
|
|
WeaponLevel::Level1,
|
|
0,
|
|
max_ammo,
|
|
max_ammo,
|
|
));
|
|
}
|
|
|
|
self.weapons.iter_mut().find(|w| w.wtype == weapon_id).unwrap()
|
|
}
|
|
|
|
pub fn trade_weapon(&mut self, old: Option<WeaponType>, new: WeaponType, max_ammo: u16) {
|
|
if let Some(wtype) = old {
|
|
if let Some(weapon) = self.get_weapon_by_type_mut(wtype) {
|
|
*weapon = Weapon::new(new, WeaponLevel::Level1, 0, max_ammo, max_ammo);
|
|
} else {
|
|
self.add_weapon(new, max_ammo);
|
|
}
|
|
} else {
|
|
self.add_weapon(new, max_ammo);
|
|
}
|
|
}
|
|
|
|
pub fn remove_weapon(&mut self, wtype: WeaponType) {
|
|
self.weapons.retain(|weapon| weapon.wtype != wtype);
|
|
}
|
|
|
|
pub fn get_weapon(&self, idx: usize) -> Option<&Weapon> {
|
|
self.weapons.get(idx)
|
|
}
|
|
|
|
pub fn get_weapon_by_type_mut(&mut self, wtype: WeaponType) -> Option<&mut Weapon> {
|
|
self.weapons.iter_mut().find(|weapon| weapon.wtype == wtype)
|
|
}
|
|
|
|
pub fn get_current_weapon(&self) -> Option<&Weapon> {
|
|
self.weapons.get(self.current_weapon as usize)
|
|
}
|
|
|
|
pub fn get_current_weapon_mut(&mut self) -> Option<&mut Weapon> {
|
|
self.weapons.get_mut(self.current_weapon as usize)
|
|
}
|
|
|
|
pub fn next_weapon(&mut self) {
|
|
if (1 + self.current_weapon as usize) < self.weapons.len() {
|
|
self.current_weapon += 1;
|
|
} else {
|
|
self.current_weapon = 0;
|
|
}
|
|
}
|
|
|
|
pub fn prev_weapon(&mut self) {
|
|
if self.current_weapon as usize > 0 {
|
|
self.current_weapon -= 1;
|
|
} else {
|
|
self.current_weapon = self.weapons.len().saturating_sub(1) as u16;
|
|
}
|
|
}
|
|
|
|
pub fn refill_all_ammo(&mut self) {
|
|
for weapon in self.weapons.iter_mut() {
|
|
weapon.ammo = weapon.max_ammo;
|
|
}
|
|
}
|
|
|
|
pub fn reset_all_weapon_xp(&mut self) {
|
|
for weapon in self.weapons.iter_mut() {
|
|
weapon.level = WeaponLevel::Level1;
|
|
weapon.experience = 0;
|
|
}
|
|
}
|
|
|
|
pub fn reset_current_weapon_xp(&mut self) {
|
|
if let Some(weapon) = self.get_current_weapon_mut() {
|
|
weapon.level = WeaponLevel::Level1;
|
|
weapon.experience = 0;
|
|
}
|
|
}
|
|
|
|
pub fn take_xp(&mut self, exp: u16, state: &mut SharedGameState) -> TakeExperienceResult {
|
|
let mut result = TakeExperienceResult::None;
|
|
|
|
if let Some(weapon) = self.get_current_weapon_mut() {
|
|
let lvl_table = state.constants.weapon.level_table[weapon.wtype as usize];
|
|
let mut tmp_exp = weapon.experience as isize - exp as isize;
|
|
|
|
if tmp_exp >= 0 {
|
|
weapon.experience = tmp_exp as u16;
|
|
} else {
|
|
while tmp_exp < 0 {
|
|
if weapon.level == WeaponLevel::Level1 {
|
|
weapon.experience = 0;
|
|
tmp_exp = 0;
|
|
} else if weapon.experience < exp {
|
|
weapon.level = weapon.level.prev();
|
|
let max_level = lvl_table[weapon.level as usize - 1] as isize;
|
|
weapon.experience = (max_level + tmp_exp.max(-max_level)) as u16;
|
|
tmp_exp += max_level;
|
|
|
|
if weapon.wtype != WeaponType::Spur {
|
|
result = TakeExperienceResult::LevelDown;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
result
|
|
}
|
|
|
|
pub fn add_xp(&mut self, exp: u16, player: &mut Player, state: &mut SharedGameState) {
|
|
if let Some(weapon) = self.get_current_weapon_mut() {
|
|
let curr_level_idx = weapon.level as usize - 1;
|
|
let lvl_table = state.constants.weapon.level_table[weapon.wtype as usize];
|
|
|
|
weapon.experience += exp;
|
|
|
|
if weapon.level == WeaponLevel::Level3 {
|
|
if weapon.experience > lvl_table[2] {
|
|
weapon.experience = lvl_table[2];
|
|
|
|
if player.equip.has_whimsical_star() && player.stars < 3 {
|
|
player.stars += 1;
|
|
}
|
|
}
|
|
} else if weapon.experience > lvl_table[curr_level_idx] {
|
|
weapon.level = weapon.level.next();
|
|
weapon.experience = 0;
|
|
|
|
if weapon.wtype != WeaponType::Spur {
|
|
state.sound_manager.play_sfx(27);
|
|
state.create_caret(player.x, player.y, CaretType::LevelUp, Direction::Left);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Get current experience state. Returns a (exp, max exp, max level/exp) tuple.
|
|
pub fn get_current_max_exp(&self, constants: &EngineConstants) -> (u16, u16, bool) {
|
|
if let Some(weapon) = self.weapons.get(self.current_weapon as usize) {
|
|
if weapon.level == WeaponLevel::None {
|
|
return (0, 0, false);
|
|
}
|
|
|
|
let level_idx = weapon.level as usize - 1;
|
|
let max_exp = constants.weapon.level_table[weapon.wtype as usize][level_idx];
|
|
let max = weapon.level == WeaponLevel::Level3 && weapon.experience == max_exp;
|
|
|
|
(weapon.experience, max_exp, max)
|
|
} else {
|
|
(0, 0, false)
|
|
}
|
|
}
|
|
|
|
/// Get current ammunition state. Returns a (ammo, max ammo) tuple.
|
|
pub fn get_current_ammo(&self) -> (u16, u16) {
|
|
if let Some(weapon) = self.weapons.get(self.current_weapon as usize) {
|
|
(weapon.ammo, weapon.max_ammo)
|
|
} else {
|
|
(0, 0)
|
|
}
|
|
}
|
|
|
|
pub fn get_current_level(&self) -> WeaponLevel {
|
|
if let Some(weapon) = self.weapons.get(self.current_weapon as usize) {
|
|
weapon.level
|
|
} else {
|
|
WeaponLevel::None
|
|
}
|
|
}
|
|
|
|
pub fn get_current_weapon_idx(&self) -> u16 {
|
|
self.current_weapon
|
|
}
|
|
|
|
pub fn get_weapon_count(&self) -> usize {
|
|
self.weapons.len()
|
|
}
|
|
|
|
pub fn has_weapon(&self, wtype: WeaponType) -> bool {
|
|
self.weapons.iter().any(|weapon| weapon.wtype == wtype)
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn inventory_test() {
|
|
let mut inventory = Inventory::new();
|
|
|
|
inventory.add_item(3);
|
|
assert_eq!(inventory.has_item(2), false);
|
|
assert_eq!(inventory.has_item(3), true);
|
|
|
|
assert_eq!(inventory.has_item_amount(3, Ordering::Equal, 1), true);
|
|
assert_eq!(inventory.has_item_amount(3, Ordering::Less, 2), true);
|
|
inventory.consume_item(3);
|
|
|
|
assert_eq!(inventory.has_item_amount(3, Ordering::Equal, 0), true);
|
|
assert_eq!(inventory.has_item_amount(3, Ordering::Less, 2), true);
|
|
|
|
inventory.add_item(2);
|
|
assert_eq!(inventory.has_item(2), true);
|
|
assert_eq!(inventory.has_item_amount(2, Ordering::Equal, 1), true);
|
|
assert_eq!(inventory.has_item_amount(2, Ordering::Less, 1), false);
|
|
|
|
inventory.add_item(4);
|
|
inventory.add_item(4);
|
|
inventory.add_item(4);
|
|
inventory.add_item(4);
|
|
|
|
assert_eq!(inventory.has_item(4), true);
|
|
assert_eq!(inventory.has_item_amount(4, Ordering::Greater, 3), true);
|
|
assert_eq!(inventory.has_item_amount(4, Ordering::Equal, 4), true);
|
|
assert_eq!(inventory.has_item_amount(4, Ordering::Less, 2), false);
|
|
}
|