Add eval representation of FormulaSpec

This commit is contained in:
Joscha 2021-11-25 02:32:08 +01:00
parent 21739c7314
commit e827556c84
5 changed files with 165 additions and 10 deletions

68
src/eval/formula_spec.rs Normal file
View file

@ -0,0 +1,68 @@
use crate::files::commands::{self, Expr, Time, Var};
use super::delta::{Delta, DeltaStep};
pub struct FormulaSpec {
// TODO Implement more efficient exprs and expr evaluation
pub start: Expr,
pub start_delta: Delta,
pub start_time: Option<Time>,
pub end_delta: Delta,
}
impl From<&commands::FormulaSpec> for FormulaSpec {
fn from(spec: &commands::FormulaSpec) -> Self {
let start: Expr = spec.start.as_ref().cloned().unwrap_or(Expr::Lit(1));
let start_delta: Delta = spec
.start_delta
.as_ref()
.map(|delta| delta.into())
.unwrap_or_default();
let mut end_delta: Delta = spec
.end_delta
.as_ref()
.map(|delta| delta.into())
.unwrap_or_default();
if let Some(time) = spec.end_time {
end_delta.steps.push(DeltaStep::Time(time));
}
Self {
start,
start_delta,
start_time: spec.start_time,
end_delta,
}
}
}
impl From<&commands::WeekdaySpec> for FormulaSpec {
fn from(spec: &commands::WeekdaySpec) -> Self {
let start = Expr::Eq(
Box::new(Expr::Var(Var::Weekday)),
Box::new(Expr::Var(spec.start.into())),
);
let mut end_delta = Delta::default();
if let Some(wd) = spec.end {
end_delta.steps.push(DeltaStep::Weekday(1, wd));
}
if let Some(delta) = &spec.end_delta {
for step in &delta.0 {
end_delta.steps.push((*step).into());
}
}
if let Some(time) = spec.end_time {
end_delta.steps.push(DeltaStep::Time(time));
}
Self {
start,
start_delta: Default::default(),
start_time: spec.start_time,
end_delta,
}
}
}