Lines
74.93 %
Functions
19.62 %
use crate::diagnostics::diagnostics_api::SymbolKind;
use crate::diagnostics::source_map::{HoverInfo, span_with_hover};
use crate::errors::FatalParseError;
use crate::parser::ParseContext;
use crate::parser::atom::parse_atom;
use crate::parser::comprehension::parse_quantifier_or_aggregate_expr;
use crate::util::TypecheckingContext;
use crate::{field, named_child};
use conjure_cp_core::ast::{Expression, Metadata, Moo};
use conjure_cp_core::{domain_int, matrix_expr, range};
use tree_sitter::Node;
pub fn parse_expression(
ctx: &mut ParseContext,
node: Node,
) -> Result<Option<Expression>, FatalParseError> {
match node.kind() {
"atom" => parse_atom(ctx, &node),
"bool_expr" => parse_boolean_expression(ctx, &node),
"arithmetic_expr" => parse_arithmetic_expression(ctx, &node),
"comparison_expr" => parse_comparison_expression(ctx, &node),
"dominance_relation" => parse_dominance_relation(ctx, &node),
"all_diff_comparison" => parse_all_diff_comparison(ctx, &node),
_ => Err(FatalParseError::internal_error(
format!("Unexpected expression type: '{}'", node.kind()),
Some(node.range()),
)),
}
fn parse_dominance_relation(
node: &Node,
if ctx.root.kind() == "dominance_relation" {
return Err(FatalParseError::internal_error(
"Nested dominance relations are not allowed".to_string(),
));
// NB: In all other cases, we keep the root the same;
// However, here we create a new context with the new root so downstream functions
// know we are inside a dominance relation
let mut inner_ctx = ParseContext {
source_code: ctx.source_code,
root: node,
symbols: ctx.symbols.clone(),
errors: ctx.errors,
source_map: &mut *ctx.source_map,
typechecking_context: ctx.typechecking_context,
};
let Some(inner) = parse_expression(&mut inner_ctx, field!(node, "expression"))? else {
return Ok(None);
Ok(Some(Expression::DominanceRelation(
Metadata::new(),
Moo::new(inner),
)))
fn parse_arithmetic_expression(
ctx.typechecking_context = TypecheckingContext::Arithmetic;
let inner = named_child!(node);
match inner.kind() {
"atom" => parse_atom(ctx, &inner),
"negative_expr" | "abs_value" | "sub_arith_expr" => parse_unary_expression(ctx, &inner),
"toInt_expr" => {
// add special handling for toInt, as it is arithmetic but takes a non-arithmetic operand
ctx.typechecking_context = TypecheckingContext::Unknown;
parse_unary_expression(ctx, &inner)
"exponent" | "product_expr" | "sum_expr" => parse_binary_expression(ctx, &inner),
"list_combining_expr_arith" => parse_list_combining_expression(ctx, &inner),
"aggregate_expr" => parse_quantifier_or_aggregate_expr(ctx, &inner),
format!("Expected arithmetic expression, found: {}", inner.kind()),
Some(inner.range()),
fn parse_comparison_expression(
"arithmetic_comparison" => {
// Arithmetic comparisons require arithmetic operands
parse_binary_expression(ctx, &inner)
"lex_comparison" => {
// TODO: check that both operands are comparable collections.
"equality_comparison" => {
// Equality works on any type
// TODO: add type checking to ensure both sides have the same type
"set_comparison" => {
// Set comparisons require set operands (no specific type checking for now)
// TODO: add typechecking for sets
"all_diff_comparison" => {
// TODO: check that operand is a collection with compatible element type.
parse_all_diff_comparison(ctx, &inner)
format!("Expected comparison expression, found '{}'", inner.kind()),
fn parse_boolean_expression(
ctx.typechecking_context = TypecheckingContext::Boolean;
"not_expr" | "sub_bool_expr" => parse_unary_expression(ctx, &inner),
"and_expr" | "or_expr" | "implication" | "iff_expr" => parse_binary_expression(ctx, &inner),
"list_combining_expr_bool" => parse_list_combining_expression(ctx, &inner),
"quantifier_expr" => parse_quantifier_or_aggregate_expr(ctx, &inner),
format!("Expected boolean expression, found '{}'", inner.kind()),
fn parse_list_combining_expression(
let operator_node = field!(node, "operator");
let operator_str = &ctx.source_code[operator_node.start_byte()..operator_node.end_byte()];
let Some(inner) = parse_atom(ctx, &field!(node, "arg"))? else {
match operator_str {
"and" => Ok(Some(Expression::And(Metadata::new(), Moo::new(inner)))),
"or" => Ok(Some(Expression::Or(Metadata::new(), Moo::new(inner)))),
"sum" => Ok(Some(Expression::Sum(Metadata::new(), Moo::new(inner)))),
"product" => Ok(Some(Expression::Product(Metadata::new(), Moo::new(inner)))),
"min" => Ok(Some(Expression::Min(Metadata::new(), Moo::new(inner)))),
"max" => Ok(Some(Expression::Max(Metadata::new(), Moo::new(inner)))),
format!("Invalid operator: '{operator_str}'"),
Some(operator_node.range()),
fn parse_all_diff_comparison(
let Some(inner) = parse_expression(ctx, field!(node, "arg"))? else {
Ok(Some(Expression::AllDiff(Metadata::new(), Moo::new(inner))))
fn parse_unary_expression(
let Some(inner) = parse_expression(ctx, field!(node, "expression"))? else {
"negative_expr" => Ok(Some(Expression::Neg(Metadata::new(), Moo::new(inner)))),
"abs_value" => Ok(Some(Expression::Abs(Metadata::new(), Moo::new(inner)))),
"not_expr" => Ok(Some(Expression::Not(Metadata::new(), Moo::new(inner)))),
"toInt_expr" => Ok(Some(Expression::ToInt(Metadata::new(), Moo::new(inner)))),
"sub_bool_expr" | "sub_arith_expr" => Ok(Some(inner)),
format!("Unrecognised unary operation: '{}'", node.kind()),
pub fn parse_binary_expression(
let mut parse_subexpr = |expr: Node| parse_expression(ctx, expr);
let Some(left) = parse_subexpr(field!(node, "left"))? else {
let Some(right) = parse_subexpr(field!(node, "right"))? else {
let op_node = field!(node, "operator");
let op_str = &ctx.source_code[op_node.start_byte()..op_node.end_byte()];
let mut description = format!("Operator '{op_str}'");
let expr = match op_str {
// NB: We are deliberately setting the index domain to 1.., not 1..2.
// Semantically, this means "a list that can grow/shrink arbitrarily".
// This is expected by rules which will modify the terms of the sum expression
// (e.g. by partially evaluating them).
"+" => Ok(Some(Expression::Sum(
Moo::new(matrix_expr![left, right; domain_int!(1..)]),
))),
"-" => Ok(Some(Expression::Minus(
Moo::new(left),
Moo::new(right),
"*" => Ok(Some(Expression::Product(
"/\\" => Ok(Some(Expression::And(
"\\/" => Ok(Some(Expression::Or(
"**" => Ok(Some(Expression::UnsafePow(
"/" => {
//TODO: add checks for if division is safe or not
Ok(Some(Expression::UnsafeDiv(
"%" => {
//TODO: add checks for if mod is safe or not
Ok(Some(Expression::UnsafeMod(
"=" => Ok(Some(Expression::Eq(
"!=" => Ok(Some(Expression::Neq(
"<=" => Ok(Some(Expression::Leq(
">=" => Ok(Some(Expression::Geq(
"<" => Ok(Some(Expression::Lt(
">" => Ok(Some(Expression::Gt(
"->" => Ok(Some(Expression::Imply(
"<->" => Ok(Some(Expression::Iff(
"<lex" => Ok(Some(Expression::LexLt(
">lex" => Ok(Some(Expression::LexGt(
"<=lex" => Ok(Some(Expression::LexLeq(
">=lex" => Ok(Some(Expression::LexGeq(
"in" => Ok(Some(Expression::In(
"subset" => Ok(Some(Expression::Subset(
"subsetEq" => Ok(Some(Expression::SubsetEq(
"supset" => Ok(Some(Expression::Supset(
"supsetEq" => Ok(Some(Expression::SupsetEq(
"union" => {
description = "set union: combines the elements from both operands".to_string();
Ok(Some(Expression::Union(
"intersect" => {
description =
"set intersection: keeps only elements common to both operands".to_string();
Ok(Some(Expression::Intersect(
format!("Invalid operator: '{op_str}'"),
Some(op_node.range()),
if expr.is_ok() {
let hover = HoverInfo {
description,
kind: Some(SymbolKind::Function),
ty: None,
decl_span: None,
span_with_hover(&op_node, ctx.source_code, ctx.source_map, hover);
expr