feat: Handle narrowed and hole shapes for not

This commit is contained in:
Jeremy Wall 2023-09-04 11:06:54 -04:00 committed by Jeremy Wall
parent b1d6571194
commit 902199fd98

View File

@ -62,18 +62,24 @@ fn derive_include_shape(
fn derive_not_shape(def: &NotDef, symbol_table: &mut BTreeMap<Rc<str>, Shape>) -> Shape { fn derive_not_shape(def: &NotDef, symbol_table: &mut BTreeMap<Rc<str>, Shape>) -> Shape {
let shape = def.expr.as_ref().derive_shape(symbol_table); let shape = def.expr.as_ref().derive_shape(symbol_table);
if let Shape::Boolean(b) = shape { if let Shape::Boolean(b) = &shape {
Shape::Boolean(PositionedItem::new(!b.val, def.pos.clone())) return Shape::Boolean(PositionedItem::new(!b.val, def.pos.clone()));
} else { } else if let Shape::Hole(_) = &shape {
// TODO(jwall): Display implementations for shapes. return Shape::Boolean(PositionedItem::new(true, def.pos.clone()));
Shape::TypeErr( } else if let Shape::Narrowed(shape_list) = &shape {
def.pos.clone(), for s in shape_list.types.iter() {
format!( if let Shape::Boolean(b) = s {
"Expected Boolean value in Not expression but got: {:?}", return Shape::Boolean(PositionedItem::new(!b.val, def.pos.clone()));
shape }
), }
) };
} Shape::TypeErr(
def.pos.clone(),
format!(
"Expected Boolean value in Not expression but got: {:?}",
shape
),
)
} }
fn derive_copy_shape(def: &CopyDef, symbol_table: &mut BTreeMap<Rc<str>, Shape>) -> Shape { fn derive_copy_shape(def: &CopyDef, symbol_table: &mut BTreeMap<Rc<str>, Shape>) -> Shape {