1
0
Fork 0
roguelike-game/src/raws/rawmaster.rs

1035 lines
33 KiB
Rust

use std::collections::{HashMap, HashSet};
use ::bracket_lib::prelude::*;
use ::lazy_static::lazy_static;
use ::regex::Regex;
use ::specs::prelude::*;
use ::specs::saveload::{MarkedBuilder, SimpleMarker};
use super::{Raws, Reaction, SpawnTableEntry, RAWS};
use crate::components::*;
use crate::gamesystem::{mana_at_level, npc_hp};
use crate::map::MasterDungeonMap;
use crate::random_table::{MasterTable, RandomTable};
use crate::rng::roll_dice;
pub fn parse_dice_string(dice: &str) -> (i32, i32, i32) {
lazy_static! {
static ref DICE_RE: Regex = Regex::new(r"(\d+)d(\d+)([\+\-]\d+)?").unwrap();
}
let mut n_dice = 1;
let mut die_type = 4;
let mut die_bonus = 0;
for cap in DICE_RE.captures_iter(dice) {
if let Some(group) = cap.get(1) {
n_dice = group.as_str().parse::<i32>().expect("Not a digit");
}
if let Some(group) = cap.get(2) {
die_type = group.as_str().parse::<i32>().expect("Not a digit");
}
if let Some(group) = cap.get(3) {
die_bonus = group.as_str().parse::<i32>().expect("Not a digit");
}
}
(n_dice, die_type, die_bonus)
}
pub enum SpawnType {
AtPosition { x: i32, y: i32 },
Equipped { by: Entity },
Carried { by: Entity },
}
#[derive(Debug, Default)]
pub struct RawMaster {
raws: Raws,
item_index: HashMap<String, usize>,
mob_index: HashMap<String, usize>,
prop_index: HashMap<String, usize>,
loot_index: HashMap<String, usize>,
faction_index: HashMap<String, HashMap<String, Reaction>>,
spell_index: HashMap<String, usize>,
}
struct NewMagicItem {
name: String,
bonus: i32,
}
impl RawMaster {
pub fn empty() -> RawMaster {
RawMaster::default()
}
fn append_magic_template(items_to_build: &mut Vec<NewMagicItem>, item: &super::Item) {
if let Some(template) = &item.template_magic {
if item.weapon.is_some() || item.wearable.is_some() {
if template.include_cursed {
items_to_build.push(NewMagicItem {
name: item.name.clone(),
bonus: -1,
});
}
for bonus in template.bonus_min..=template.bonus_max {
items_to_build.push(NewMagicItem {
name: item.name.clone(),
bonus,
});
}
} else {
console::log(format!(
"{} is marked as templated, but isn't a weapon or armor.",
item.name
));
}
}
}
fn build_base_magic_item(&self, nmw: &NewMagicItem) -> super::Item {
let base_item_index = self.item_index[&nmw.name];
let mut base_item_copy = self.raws.items[base_item_index].clone();
base_item_copy.vendor_category = None; // Don't sell magic items!
if nmw.bonus == -1 {
base_item_copy.name = format!("{} -1", nmw.name);
} else {
base_item_copy.name = format!("{} +{}", nmw.name, nmw.bonus);
}
base_item_copy.magic = Some(super::MagicItem {
class: match nmw.bonus {
2 | 3 | 4 => "rare",
5 => "legendary",
_ => "common",
}
.to_string(),
naming: base_item_copy
.template_magic
.as_ref()
.unwrap()
.unidentified_name
.clone(),
cursed: if nmw.bonus == -1 { Some(true) } else { None },
});
if let Some(initiative_penalty) = base_item_copy.initiative_penalty.as_mut() {
*initiative_penalty -= nmw.bonus as f32;
}
if let Some(base_value) = base_item_copy.base_value.as_mut() {
*base_value += (nmw.bonus as f32 + 1.0) * 50.0;
}
if let Some(mut weapon) = base_item_copy.weapon.as_mut() {
weapon.hit_bonus += nmw.bonus;
let (n, die, plus) = parse_dice_string(&weapon.base_damage);
let final_bonus = plus + nmw.bonus;
if final_bonus > 0 {
weapon.base_damage = format!("{}d{}+{}", n, die, final_bonus);
} else {
weapon.base_damage = format!("{}d{}-{}", n, die, i32::abs(final_bonus));
}
}
if let Some(mut armor) = base_item_copy.wearable.as_mut() {
armor.armor_class += nmw.bonus as f32;
}
base_item_copy
}
fn build_magic_weapon_or_armor(&mut self, items_to_build: &[NewMagicItem]) {
for nmw in items_to_build.iter() {
let base_item_copy = self.build_base_magic_item(nmw);
let real_name = base_item_copy.name.clone();
self.raws.items.push(base_item_copy);
self.item_index
.insert(real_name.clone(), self.raws.items.len() - 1);
self.raws.spawn_table.push(SpawnTableEntry {
name: real_name.clone(),
weight: 10 - i32::abs(nmw.bonus),
min_depth: 1 + i32::abs((nmw.bonus - 1) * 3),
max_depth: 100,
add_map_depth_to_weight: None,
});
}
}
fn build_traited_weapons(&mut self, items_to_build: &[NewMagicItem]) {
items_to_build
.iter()
.filter(|i| i.bonus > 0)
.for_each(|nmw| {
for wt in self.raws.weapon_traits.iter() {
let mut base_item_copy = self.build_base_magic_item(nmw);
if let Some(mut weapon) = base_item_copy.weapon.as_mut() {
base_item_copy.name = format!("{} {}", wt.name, base_item_copy.name);
if let Some(base_value) = base_item_copy.base_value.as_mut() {
*base_value *= 2.0;
}
weapon.proc_chance = Some(0.25);
weapon.proc_effects = Some(wt.effects.clone());
let real_name = base_item_copy.name.clone();
self.raws.items.push(base_item_copy);
self.item_index
.insert(real_name.clone(), self.raws.items.len() - 1);
self.raws.spawn_table.push(SpawnTableEntry {
name: real_name.clone(),
weight: 9 - i32::abs(nmw.bonus),
min_depth: 2 + i32::abs((nmw.bonus - 1) * 3),
max_depth: 100,
add_map_depth_to_weight: None,
});
}
}
});
}
pub fn load(&mut self, raws: Raws) {
self.raws = raws;
self.item_index = HashMap::new();
let mut used_names: HashSet<String> = HashSet::new();
let mut items_to_build: Vec<NewMagicItem> = Vec::new();
for (i, item) in self.raws.items.iter().enumerate() {
if used_names.contains(&item.name) {
#[cfg(feature = "debug")]
console::log(format!(
"WARNING - duplicate item name in raws [{}]",
item.name
));
}
self.item_index.insert(item.name.clone(), i);
used_names.insert(item.name.clone());
RawMaster::append_magic_template(&mut items_to_build, item);
}
for (i, mob) in self.raws.mobs.iter().enumerate() {
if used_names.contains(&mob.name) {
#[cfg(feature = "debug")]
console::log(format!(
"WARNING - duplicate mob name in raws [{}]",
mob.name
));
}
self.mob_index.insert(mob.name.clone(), i);
used_names.insert(mob.name.clone());
}
for (i, prop) in self.raws.props.iter().enumerate() {
if used_names.contains(&prop.name) {
#[cfg(feature = "debug")]
console::log(format!(
"WARNING - duplicate prop name in raws [{}]",
prop.name
));
}
self.prop_index.insert(prop.name.clone(), i);
used_names.insert(prop.name.clone());
}
for spawn in self.raws.spawn_table.iter() {
if !used_names.contains(&spawn.name) {
#[cfg(feature = "debug")]
console::log(format!(
"WARNING - Spawn tables references unspecified entity [{}]",
spawn.name
));
}
}
for (i, loot) in self.raws.loot_tables.iter().enumerate() {
self.loot_index.insert(loot.name.clone(), i);
}
for faction in self.raws.faction_table.iter() {
let mut reactions: HashMap<String, Reaction> = HashMap::new();
for other in faction.responses.iter() {
reactions.insert(
other.0.clone(),
match other.1.as_str() {
"ignore" => Reaction::Ignore,
"flee" => Reaction::Flee,
_ => Reaction::Attack,
},
);
}
self.faction_index.insert(faction.name.clone(), reactions);
}
for (i, spell) in self.raws.spells.iter().enumerate() {
self.spell_index.insert(spell.name.clone(), i);
}
self.build_magic_weapon_or_armor(&items_to_build);
self.build_traited_weapons(&items_to_build);
}
}
#[inline(always)]
pub fn faction_reaction(my_faction: &str, their_faction: &str, raws: &RawMaster) -> Reaction {
if raws.faction_index.contains_key(my_faction) {
let mf = &raws.faction_index[my_faction];
return if mf.contains_key(their_faction) {
mf[their_faction]
} else if mf.contains_key("Default") {
mf["Default"]
} else {
Reaction::Ignore
};
}
Reaction::Ignore
}
fn find_slot_for_equippable_item(tag: &str, raws: &RawMaster) -> EquipmentSlot {
if !raws.item_index.contains_key(tag) {
panic!("Trying to equip an unknown item: {}", tag);
}
let item_index = raws.item_index[tag];
let item = &raws.raws.items[item_index];
if item.weapon.is_some() {
return EquipmentSlot::Melee;
} else if let Some(wearable) = &item.wearable {
return string_to_slot(&wearable.slot);
}
panic!("Trying to equip {}, but it has not slot tag", tag);
}
pub fn get_vendor_items(categories: &[String], raws: &RawMaster) -> Vec<(String, f32)> {
let mut result = Vec::new();
for item in raws.raws.items.iter() {
if let Some(cat) = &item.vendor_category {
if categories.contains(cat) && item.base_value.is_some() {
result.push((item.name.clone(), item.base_value.unwrap()))
}
}
}
result
}
pub fn get_scroll_tags() -> Vec<String> {
let raws = &RAWS.lock().unwrap();
let mut result = Vec::new();
for item in raws.raws.items.iter() {
if let Some(magic) = &item.magic {
if &magic.naming == "scroll" {
result.push(item.name.clone());
}
}
}
result
}
pub fn get_potion_tags() -> Vec<String> {
let raws = &RAWS.lock().unwrap();
let mut result = Vec::new();
for item in raws.raws.items.iter() {
if let Some(magic) = &item.magic {
if &magic.naming == "potion" {
result.push(item.name.clone());
}
}
}
result
}
pub fn is_tag_magic(tag: &str) -> bool {
let raws = &RAWS.lock().unwrap();
if raws.item_index.contains_key(tag) {
let item_template = &raws.raws.items[raws.item_index[tag]];
item_template.magic.is_some()
} else {
false
}
}
fn spawn_position<'a>(
pos: SpawnType,
new_entity: EntityBuilder<'a>,
tag: &str,
raws: &RawMaster,
) -> EntityBuilder<'a> {
let eb = new_entity;
// Spawn in the specified location
match pos {
SpawnType::AtPosition { x, y } => eb.with(Position { x, y }),
SpawnType::Carried { by } => eb.with(InBackpack { owner: by }),
SpawnType::Equipped { by } => {
let slot = find_slot_for_equippable_item(tag, raws);
eb.with(Equipped { owner: by, slot })
}
}
}
fn get_renderable_component(
renderable: &super::item_structs::Renderable,
) -> crate::components::Renderable {
crate::components::Renderable {
glyph: to_cp437(renderable.glyph.chars().next().unwrap()),
fg: RGB::from_hex(&renderable.fg).expect("Invalid color"),
bg: RGB::from_hex(&renderable.bg).expect("Invalid color"),
render_order: renderable.order,
}
}
pub fn string_to_slot(slot: &str) -> EquipmentSlot {
match slot {
"Shield" => EquipmentSlot::Shield,
"Head" => EquipmentSlot::Head,
"Torso" => EquipmentSlot::Torso,
"Legs" => EquipmentSlot::Legs,
"Feet" => EquipmentSlot::Feet,
"Hands" => EquipmentSlot::Hands,
"Melee" => EquipmentSlot::Melee,
_ => {
#[cfg(feature = "debug")]
console::log(format!("Warning: unknown equipment slot type [{}]", slot));
EquipmentSlot::Melee
}
}
}
fn parse_particle_line(n: &str) -> SpawnParticleLine {
let tokens: Vec<_> = n.split(';').collect();
SpawnParticleLine {
glyph: to_cp437(tokens[0].chars().next().unwrap()),
color: RGB::from_hex(tokens[1]).expect("Invalid hex rgb color"),
lifetime_ms: tokens[2].parse::<f32>().unwrap(),
}
}
fn parse_particle(n: &str) -> SpawnParticleBurst {
let tokens: Vec<_> = n.split(';').collect();
SpawnParticleBurst {
glyph: to_cp437(tokens[0].chars().next().unwrap()),
color: RGB::from_hex(tokens[1]).expect("Invalid hex rgb color"),
lifetime_ms: tokens[2].parse::<f32>().unwrap(),
}
}
macro_rules! apply_effects {
($effects:expr, $eb:expr) => {
// Effects are defined in the raws file as name => value pairs
for (effect_name, value) in $effects.iter() {
match effect_name.as_str() {
"provides_healing" => {
$eb = $eb.with(ProvidesHealing {
heal_amount: value.parse::<i32>().unwrap(),
});
}
"provides_mana" => {
$eb = $eb.with(ProvidesMana {
mana_amount: value.parse::<i32>().unwrap(),
})
}
"teach_spell" => {
$eb = $eb.with(TeachesSpell {
spell: value.to_string(),
})
}
"ranged" => {
$eb = $eb.with(Ranged {
range: value.parse::<i32>().unwrap(),
});
}
"damage" => {
$eb = $eb.with(InflictsDamage {
damage: value.parse::<i32>().unwrap(),
});
}
"area_of_effect" => {
$eb = $eb.with(AreaOfEffect {
radius: value.parse::<i32>().unwrap(),
});
}
"confusion" => {
$eb = $eb.with(Confusion {});
$eb = $eb.with(Duration {
turns: value.parse::<i32>().unwrap(),
});
}
"magic_mapping" => $eb = $eb.with(MagicMapper {}),
"town_portal" => $eb = $eb.with(TownPortal {}),
"food" => $eb = $eb.with(ProvidesFood {}),
"single_activation" => $eb = $eb.with(SingleActivation {}),
"particle_line" => $eb = $eb.with(parse_particle_line(&value)),
"particle" => $eb = $eb.with(parse_particle(&value)),
"remove_curse" => $eb = $eb.with(ProvidesRemoveCurse {}),
"identify" => $eb = $eb.with(ProvidesIdentification {}),
"slow" => {
$eb = $eb.with(Slow {
initiative_penalty: value.parse::<f32>().unwrap(),
})
}
"damage_over_time" => {
$eb = $eb.with(DamageOverTime {
damage: value.parse::<i32>().unwrap(),
})
}
"target_self" => $eb = $eb.with(AlwaysTargetsSelf {}),
_ => {
#[cfg(feature = "debug")]
console::log(format!(
"WARNING: consumable effect '{}' not implemented.",
effect_name
));
}
}
}
};
}
pub fn spawn_named_item(
raws: &RawMaster,
ecs: &mut World,
key: &str,
pos: SpawnType,
) -> Option<Entity> {
if raws.item_index.contains_key(key) {
let item_template = &raws.raws.items[raws.item_index[key]];
let dm = ecs.fetch::<MasterDungeonMap>();
let scroll_names = dm.scroll_mappings.clone();
let potion_names = dm.potion_mappings.clone();
let identified = dm.identified_items.clone();
std::mem::drop(dm);
let mut eb = ecs.create_entity().marked::<SimpleMarker<SerializeMe>>();
// Spawn in the specified location
eb = spawn_position(pos, eb, key, raws);
// Renderable
if let Some(renderable) = &item_template.renderable {
eb = eb.with(get_renderable_component(renderable));
}
eb = eb.with(Name::from(&item_template.name));
eb = eb.with(Item {
initiative_penalty: item_template.initiative_penalty.unwrap_or(0.),
weight_lbs: item_template.weight_lbs.unwrap_or(0.),
base_value: item_template.base_value.unwrap_or(0.),
});
if let Some(consumable) = &item_template.consumable {
let max_charges = consumable.charges.unwrap_or(1);
eb = eb.with(Consumable {
max_charges,
charges: max_charges,
});
apply_effects!(consumable.effects, eb);
}
if let Some(weapon) = &item_template.weapon {
eb = eb.with(Equippable {
slot: EquipmentSlot::Melee,
});
let (n_dice, die_type, bonus) = parse_dice_string(&weapon.base_damage);
let mut wpn = Weapon {
range: if weapon.range == "melee" {
None
} else {
Some(weapon.range.parse::<i32>().expect("Not a number"))
},
attribute: WeaponAttribute::Might,
damage_n_dice: n_dice,
damage_die_type: die_type,
damage_bonus: bonus,
hit_bonus: weapon.hit_bonus,
proc_chance: weapon.proc_chance,
proc_target: weapon.proc_target.clone(),
};
wpn.attribute = match weapon.attribute.as_str() {
"Quickness" => WeaponAttribute::Quickness,
_ => WeaponAttribute::Might,
};
eb = eb.with(wpn);
if let Some(proc_effects) = &weapon.proc_effects {
apply_effects!(proc_effects, eb);
}
}
if let Some(wearable) = &item_template.wearable {
let slot = string_to_slot(&wearable.slot);
eb = eb.with(Equippable { slot });
eb = eb.with(Wearable {
slot,
armor_class: wearable.armor_class,
});
}
if let Some(magic) = &item_template.magic {
let class = match magic.class.as_str() {
"rare" => MagicItemClass::Rare,
"legendary" => MagicItemClass::Legendary,
_ => MagicItemClass::Common,
};
eb = eb.with(MagicItem { class });
if !identified.contains(&item_template.name) {
#[allow(clippy::single_match)]
match magic.naming.as_str() {
"scroll" => {
eb = eb.with(ObfuscatedName {
name: scroll_names[&item_template.name].clone(),
})
}
"potion" => {
eb = eb.with(ObfuscatedName {
name: potion_names[&item_template.name].clone(),
})
}
_ => {
eb = eb.with(ObfuscatedName {
name: magic.naming.clone(),
})
}
}
}
if let Some(cursed) = magic.cursed {
if cursed {
eb = eb.with(CursedItem {});
}
}
}
if let Some(ab) = &item_template.attributes {
eb = eb.with(AttributeBonus {
might: ab.might,
fitness: ab.fitness,
quickness: ab.quickness,
intelligence: ab.intelligence,
})
}
return Some(eb.build());
}
None
}
pub fn spawn_named_mob(
raws: &RawMaster,
ecs: &mut World,
key: &str,
pos: SpawnType,
) -> Option<Entity> {
if raws.mob_index.contains_key(key) {
let mob_template = &raws.raws.mobs[raws.mob_index[key]];
let mut eb = ecs.create_entity().marked::<SimpleMarker<SerializeMe>>();
// Spawn in the specified location
eb = spawn_position(pos, eb, key, raws);
// Initiative of 2
eb = eb.with(Initiative { current: 2 });
// Renderable
if let Some(renderable) = &mob_template.renderable {
eb = eb.with(get_renderable_component(renderable));
if renderable.x_size.is_some() || renderable.y_size.is_some() {
eb = eb.with(TileSize {
x: renderable.x_size.unwrap_or(1),
y: renderable.y_size.unwrap_or(1),
});
}
}
eb = eb.with(Name::from(&mob_template.name));
eb = match mob_template.movement.as_ref() {
"random" => eb.with(MoveMode {
mode: Movement::Random,
}),
"random_waypoint" => eb.with(MoveMode {
mode: Movement::RandomWaypoint { path: None },
}),
_ => eb.with(MoveMode {
mode: Movement::Static,
}),
};
// Quips
if let Some(quips) = &mob_template.quips {
eb = eb.with(Quips {
available: quips.clone(),
});
}
if mob_template.blocks_tile {
eb = eb.with(BlocksTile {});
}
let mut mob_fitness = 11;
let mut mob_int = 11;
let mut attr = Attributes {
might: Attribute::new(11),
fitness: Attribute::new(11),
quickness: Attribute::new(11),
intelligence: Attribute::new(11),
};
if let Some(might) = mob_template.attributes.might {
attr.might = Attribute::new(might);
}
if let Some(fitness) = mob_template.attributes.fitness {
attr.fitness = Attribute::new(fitness);
mob_fitness = fitness;
}
if let Some(quickness) = mob_template.attributes.quickness {
attr.quickness = Attribute::new(quickness);
}
if let Some(intelligence) = mob_template.attributes.intelligence {
attr.intelligence = Attribute::new(intelligence);
mob_int = intelligence;
}
eb = eb.with(attr);
let mob_level = if mob_template.level.is_some() {
mob_template.level.unwrap()
} else {
1
};
let mob_hp = npc_hp(mob_fitness, mob_level);
let mob_mana = mana_at_level(mob_int, mob_level);
let pools = Pools {
level: mob_level,
xp: 0,
hit_points: Pool::new(mob_hp),
mana: Pool::new(mob_mana),
total_weight: 0.,
total_initiative_penalty: 0.,
gold: if let Some(gold) = &mob_template.gold {
let (n, d, b) = parse_dice_string(gold);
(roll_dice(n, d) + b) as f32
} else {
0.0
},
god_mode: false,
};
eb = eb.with(pools);
let mut skills = Skills::new(1);
if let Some(mobskills) = &mob_template.skills {
for sk in mobskills.iter() {
match sk.0.as_str() {
"Melee" => {
skills.skills.insert(Skill::Melee, *sk.1);
}
"Defense" => {
skills.skills.insert(Skill::Defense, *sk.1);
}
"Magic" => {
skills.skills.insert(Skill::Magic, *sk.1);
}
_ => {
#[cfg(feature = "debug")]
console::log(format!("Unknown skill referenced [{}]", sk.0));
}
}
}
}
eb = eb.with(skills);
eb = eb.with(Viewshed {
visible_tiles: Vec::new(),
range: mob_template.vision_range,
dirty: true,
});
if let Some(na) = &mob_template.natural {
let mut nature = NaturalAttackDefense {
armor_class: na.armor_class,
attacks: Vec::new(),
};
if let Some(attacks) = &na.attacks {
for nattack in attacks.iter() {
let (n, d, b) = parse_dice_string(&nattack.damage);
let attack = NaturalAttack {
name: nattack.name.clone(),
hit_bonus: nattack.hit_bonus,
damage_n_dice: n,
damage_die_type: d,
damage_bonus: b,
};
nature.attacks.push(attack);
}
}
eb = eb.with(nature);
}
if let Some(loot) = &mob_template.loot_table {
eb = eb.with(LootTable {
table: loot.clone(),
});
}
if let Some(light) = &mob_template.light {
eb = eb.with(LightSource {
range: light.range,
color: RGB::from_hex(&light.color).expect("Invalid color"),
});
}
if let Some(faction) = &mob_template.faction {
eb = eb.with(Faction::from(faction));
} else {
eb = eb.with(Faction::from("Mindless"));
}
if let Some(vendor) = &mob_template.vendor {
eb = eb.with(Vendor {
categories: vendor.clone(),
})
}
if let Some(ability_list) = &mob_template.abilities {
let mut a = SpecialAbilities::default();
for ability in ability_list.iter() {
a.abilities.push(SpecialAbility {
chance: ability.chance,
spell: ability.spell.clone(),
range: ability.range,
min_range: ability.min_range,
});
}
eb = eb.with(a);
}
if let Some(ability_list) = &mob_template.on_death {
let mut a = OnDeath::default();
for ability in ability_list.iter() {
a.abilities.push(SpecialAbility {
chance: ability.chance,
spell: ability.spell.clone(),
range: ability.range,
min_range: ability.min_range,
});
}
eb = eb.with(a);
}
let new_mob = eb.build();
// Are they wielding anything?
if let Some(wielding) = &mob_template.equipped {
for tag in wielding.iter() {
spawn_named_entity(raws, ecs, tag, SpawnType::Equipped { by: new_mob });
}
}
return Some(new_mob);
}
None
}
pub fn spawn_named_prop(
raws: &RawMaster,
ecs: &mut World,
key: &str,
pos: SpawnType,
) -> Option<Entity> {
if raws.prop_index.contains_key(key) {
let prop_template = &raws.raws.props[raws.prop_index[key]];
let mut eb = ecs.create_entity().marked::<SimpleMarker<SerializeMe>>();
// Spawn in the specified location
eb = spawn_position(pos, eb, key, raws);
// Renderable
if let Some(renderable) = &prop_template.renderable {
eb = eb.with(get_renderable_component(renderable));
}
eb = eb.with(Name::from(&prop_template.name));
if let Some(hidden) = prop_template.hidden {
if hidden {
eb = eb.with(Hidden {})
}
}
if let Some(blocks_tile) = prop_template.blocks_tile {
if blocks_tile {
eb = eb.with(BlocksTile {})
}
}
if let Some(blocks_visibility) = prop_template.blocks_visibility {
if blocks_visibility {
eb = eb.with(BlocksVisibility {})
}
}
if let Some(door_open) = prop_template.door_open {
eb = eb.with(Door { open: door_open });
}
if let Some(entry_trigger) = &prop_template.entry_trigger {
eb = eb.with(EntryTrigger {});
apply_effects!(entry_trigger.effects, eb);
}
if let Some(light) = &prop_template.light {
eb = eb
.with(LightSource {
range: light.range,
color: RGB::from_hex(&light.color).expect("Invalid color"),
})
.with(Viewshed {
range: light.range,
dirty: true,
visible_tiles: Vec::new(),
});
}
return Some(eb.build());
}
None
}
pub fn spawn_named_entity(
raws: &RawMaster,
ecs: &mut World,
key: &str,
pos: SpawnType,
) -> Option<Entity> {
if raws.item_index.contains_key(key) {
return spawn_named_item(raws, ecs, key, pos);
} else if raws.mob_index.contains_key(key) {
return spawn_named_mob(raws, ecs, key, pos);
} else if raws.prop_index.contains_key(key) {
return spawn_named_prop(raws, ecs, key, pos);
}
None
}
pub enum SpawnTableType {
Item,
Mob,
Prop,
}
pub fn spawn_type_by_name(raws: &RawMaster, key: &str) -> SpawnTableType {
if raws.item_index.contains_key(key) {
SpawnTableType::Item
} else if raws.mob_index.contains_key(key) {
SpawnTableType::Mob
} else {
SpawnTableType::Prop
}
}
pub fn spawn_named_spell(raws: &RawMaster, ecs: &mut World, key: &str) -> Option<Entity> {
if raws.spell_index.contains_key(key) {
let spell_template = &raws.raws.spells[raws.spell_index[key]];
let mut eb = ecs
.create_entity()
.marked::<SimpleMarker<SerializeMe>>()
.with(SpellTemplate {
mana_cost: spell_template.mana_cost,
})
.with(Name::from(spell_template.name.clone()));
apply_effects!(spell_template.effects, eb);
return Some(eb.build());
}
None
}
pub fn get_spawn_table_for_depth(raws: &RawMaster, depth: i32) -> MasterTable {
let available_options: Vec<&SpawnTableEntry> = raws
.raws
.spawn_table
.iter()
.filter(|a| depth >= a.min_depth && depth <= a.max_depth)
.collect();
let mut rt = MasterTable::new();
for e in available_options.iter() {
let mut weight = e.weight;
if e.add_map_depth_to_weight.is_some() {
weight += depth;
}
rt.add(e.name.clone(), weight, raws);
}
rt
}
pub fn get_item_drop(raws: &RawMaster, table: &str) -> Option<String> {
if raws.loot_index.contains_key(table) {
let mut rt = RandomTable::new();
let available_options = &raws.raws.loot_tables[raws.loot_index[table]];
for item in available_options.drops.iter() {
rt.add(item.name.clone(), item.weight);
}
return Some(rt.roll());
}
None
}
pub fn spawn_all_spells(ecs: &mut World) {
let raws = &RAWS.lock().unwrap();
for spell in raws.raws.spells.iter() {
spawn_named_spell(raws, ecs, &spell.name);
}
}
pub fn find_spell_entity(ecs: &World, name: &str) -> Option<Entity> {
let names = ecs.read_storage::<Name>();
let spell_templates = ecs.read_storage::<SpellTemplate>();
let entities = ecs.entities();
for (entity, sname, _template) in (&entities, &names, &spell_templates).join() {
if name == sname.name {
return Some(entity);
}
}
None
}
pub fn find_spell_entity_by_name(
name: &str,
names: &ReadStorage<Name>,
spell_templates: &ReadStorage<SpellTemplate>,
entities: &Entities,
) -> Option<Entity> {
for (entity, sname, _template) in (entities, names, spell_templates).join() {
if name == sname.name {
return Some(entity);
}
}
None
}