use super::*;
use crate::parser::{Position, RlError};
use std::collections::HashMap;
#[derive(Clone, Copy, PartialEq, Eq, Debug, Hash, Default)]
pub struct InvariantId(pub SkillId, pub usize);
impl Id for InvariantId {
fn index(&self) -> usize {
self.1
}
}
impl InvariantId {
pub fn skill(&self) -> SkillId {
self.0
}
}
#[derive(Debug, Clone)]
pub struct Invariant {
id: InvariantId,
name: String,
guard: Expr,
effects: Vec<Effect>,
position: Option<Position>,
}
impl Invariant {
pub fn new<S: Into<String>>(
name: S,
guard: Expr,
effects: Vec<Effect>,
position: Option<Position>,
) -> Self {
let id = InvariantId::default();
let name = name.into();
Self {
id,
name,
guard,
effects,
position,
}
}
pub fn guard(&self) -> &Expr {
&self.guard
}
pub fn effects(&self) -> &Vec<Effect> {
&self.effects
}
pub fn resolve_resource(&mut self, map: &HashMap<String, ResourceId>) -> Result<(), RlError> {
self.guard.resolve_resource(map)?;
for x in self.effects.iter_mut() {
x.resolve_resource(map)?;
}
Ok(())
}
pub fn resolve_state(&mut self, map: &HashMap<String, StateId>) -> Result<(), RlError> {
self.guard.resolve_state(map)?;
for x in self.effects.iter_mut() {
x.resolve_state(map)?;
}
Ok(())
}
}
impl Named<InvariantId> for Invariant {
fn id(&self) -> InvariantId {
self.id
}
fn set_id(&mut self, id: InvariantId) {
self.id = id;
}
fn name(&self) -> &str {
&self.name
}
fn position(&self) -> Option<Position> {
self.position.clone()
}
}
impl ToLang for Invariant {
fn to_lang(&self, skillset: &Skillset) -> String {
let mut s = format!("{} {{\n", self.name);
s.push_str(&format!(
"\t\t\t\t\tguard {}\n",
self.guard.to_lang(skillset)
));
if !self.effects.is_empty() {
s.push_str("\t\t\t\t\teffect {\n");
for x in self.effects.iter() {
s.push_str(&format!("\t\t\t\t\t\t{}\n", x.to_lang(skillset)))
}
s.push_str("\t\t\t\t\t}\n");
}
s.push_str("\t\t\t\t}\n");
s
}
}
impl std::fmt::Display for Invariant {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.name)
}
}