123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- extern crate specs;
- use specs::prelude::*;
- use super::{WantsToPickupItem, Name, InBackpack, Position, gamelog::GameLog, WantsToDrinkPotion,
- Potion, CombatStats, WantsToDropItem};
- pub struct ItemCollectionSystem {}
- impl<'a> System<'a> for ItemCollectionSystem {
- #[allow(clippy::type_complexity)]
- type SystemData = ( ReadExpect<'a, Entity>,
- WriteExpect<'a, GameLog>,
- WriteStorage<'a, WantsToPickupItem>,
- WriteStorage<'a, Position>,
- ReadStorage<'a, Name>,
- WriteStorage<'a, InBackpack>
- );
- fn run(&mut self, data : Self::SystemData) {
- let (player_entity, mut gamelog, mut wants_pickup, mut positions, names, mut backpack) = data;
- for pickup in wants_pickup.join() {
- positions.remove(pickup.item);
- backpack.insert(pickup.item, InBackpack{ owner: pickup.collected_by }).expect("Unable to insert backpack entry");
- if pickup.collected_by == *player_entity {
- gamelog.entries.insert(0, format!("You pick up the {}.", names.get(pickup.item).unwrap().name));
- }
- }
- wants_pickup.clear();
- }
- }
- pub struct PotionUseSystem {}
- impl<'a> System<'a> for PotionUseSystem {
- #[allow(clippy::type_complexity)]
- type SystemData = ( ReadExpect<'a, Entity>,
- WriteExpect<'a, GameLog>,
- Entities<'a>,
- WriteStorage<'a, WantsToDrinkPotion>,
- ReadStorage<'a, Name>,
- ReadStorage<'a, Potion>,
- WriteStorage<'a, CombatStats>
- );
- fn run(&mut self, data : Self::SystemData) {
- let (player_entity, mut gamelog, entities, mut wants_drink, names, potions, mut combat_stats) = data;
- for (entity, drink, stats) in (&entities, &wants_drink, &mut combat_stats).join() {
- let potion = potions.get(drink.potion);
- match potion {
- None => {}
- Some(potion) => {
- stats.hp = i32::max(stats.max_hp, stats.hp + potion.heal_amount);
- if entity == *player_entity {
- gamelog.entries.insert(0, format!("You drink the {}, healing {} hp.", names.get(drink.potion).unwrap().name, potion.heal_amount));
- }
- entities.delete(drink.potion).expect("Delete failed");
- }
- }
- }
- wants_drink.clear();
- }
- }
- pub struct ItemDropSystem {}
- impl<'a> System<'a> for ItemDropSystem {
- #[allow(clippy::type_complexity)]
- type SystemData = ( ReadExpect<'a, Entity>,
- WriteExpect<'a, GameLog>,
- Entities<'a>,
- WriteStorage<'a, WantsToDropItem>,
- ReadStorage<'a, Name>,
- WriteStorage<'a, Position>,
- WriteStorage<'a, InBackpack>
- );
- fn run(&mut self, data : Self::SystemData) {
- let (player_entity, mut gamelog, entities, mut wants_drop, names, mut positions, mut backpack) = data;
- for (entity, to_drop) in (&entities, &wants_drop).join() {
- let mut dropper_pos : Position = Position{x:0, y:0};
- {
- let dropped_pos = positions.get(entity).unwrap();
- dropper_pos.x = dropped_pos.x;
- dropper_pos.y = dropped_pos.y;
- }
- positions.insert(to_drop.item, Position{ x : dropper_pos.x, y : dropper_pos.y }).expect("Unable to insert position");
- backpack.remove(to_drop.item);
- if entity == *player_entity {
- gamelog.entries.insert(0, format!("You drop up the {}.", names.get(to_drop.item).unwrap().name));
- }
- }
- wants_drop.clear();
- }
- }
|