rtlola-parser 0.4.0

//! This module contains `Display` implementations for the AST.

use std::fmt::{Display, Formatter, Result};

use itertools::Itertools;

use crate::ast::*;

/// Writes out the joined vector `v`, enclosed by the given strings `pref` and `suff`.
fn write_delim_list<T: Display>(
    f: &mut Formatter<'_>,
    v: &[T],
    pref: &str,
    suff: &str,
    join: &str,
) -> Result {
    write!(f, "{pref}")?;
    if let Some(e) = v.first() {
        write!(f, "{e}")?;
        for b in &v[1..] {
            write!(f, "{join}{b}")?;
        }
    }
    write!(f, "{suff}")?;
    Ok(())
}

/// Helper to format an Optional
fn format_opt<T: Display>(opt: &Option<T>, pref: &str, suff: &str) -> String {
    if let Some(ref e) = opt {
        format!("{pref}{e}{suff}")
    } else {
        String::new()
    }
}

/// Formats an optional type
fn format_type(ty: &Option<Type>) -> String {
    format_opt(ty, ": ", "")
}

fn format_tags(f: &mut Formatter<'_>, tags: &[Tag], global: bool) -> Result {
    if tags.is_empty() {
        return Ok(());
    }
    let tag_list = tags
        .iter()
        .map(|tag| {
            if let Some(value) = &tag.value {
                format!("{}=\"{}\"", &tag.key, value)
            } else {
                tag.key.to_string()
            }
        })
        .join(",");

    if global {
        writeln!(f, "#![{tag_list}]")
    } else {
        writeln!(f, "#[{tag_list}]")
    }
}

impl Display for Constant {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(
            f,
            "constant {}{} := {}",
            self.name,
            format_type(&self.ty),
            self.literal
        )
    }
}

impl Display for Input {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        format_tags(f, &self.tags, false)?;
        write!(f, "input {}", self.name)?;
        if !self.params.is_empty() {
            write_delim_list(f, &self.params, " (", ")", ", ")?;
        }
        write!(f, ": {}", self.ty)
    }
}

impl Display for Mirror {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(
            f,
            "output {} mirror {} when {}",
            self.name, self.target, self.filter
        )
    }
}

impl Display for Output {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        format_tags(f, &self.tags, false)?;
        match &self.kind {
            OutputKind::NamedOutput(name) => write!(f, "output {name}")?,
            OutputKind::Trigger => write!(f, "trigger")?,
        };
        if !self.params.is_empty() {
            write_delim_list(f, &self.params, " (", ")", ", ")?;
        }
        if let Some(ty) = &self.annotated_type {
            write!(f, ": {ty}")?;
        }
        if let Some(spawn) = &self.spawn {
            write!(f, " {spawn}")?;
        }
        for eval_spec in self.eval.iter() {
            write!(f, " {eval_spec}")?;
        }
        if let Some(close) = &self.close {
            write!(f, " {close}")?;
        }
        Ok(())
    }
}

impl Display for Parameter {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match &self.ty {
            None => write!(f, "{}", self.name),
            Some(ty) => write!(f, "{}: {}", self.name, ty),
        }
    }
}

impl Display for AnnotatedPacingType {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match self {
            AnnotatedPacingType::NotAnnotated(_) => Ok(()),
            AnnotatedPacingType::Global(freq) => write!(f, " @Global({freq})"),
            AnnotatedPacingType::Local(freq) => write!(f, " @Local({freq})"),
            AnnotatedPacingType::Unspecified(expr) => write!(f, " @{expr}"),
        }
    }
}

impl Display for SpawnSpec {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        if self.expression.is_some() || self.condition.is_some() {
            write!(f, "spawn")?;
        }
        write!(f, "{}", self.annotated_pacing)?;
        if let Some(condition) = &self.condition {
            write!(f, " when {condition}")?;
        }
        if let Some(expr) = &self.expression {
            write!(f, " with {expr}")?;
        }
        Ok(())
    }
}

impl Display for EvalSpec {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        if self.condition.is_some()
            || self.eval_expression.is_some()
            || !matches!(self.annotated_pacing, AnnotatedPacingType::NotAnnotated(_))
        {
            write!(f, "eval")?;
        }
        write!(f, "{}", self.annotated_pacing)?;
        if let Some(when) = &self.condition {
            write!(f, " when {when}")?;
        }
        if let Some(exp) = &self.eval_expression {
            write!(f, " with {exp}")?;
        }
        Ok(())
    }
}

impl Display for CloseSpec {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(f, "close")?;
        write!(f, "{}", self.annotated_pacing)?;
        write!(f, " when {}", self.condition)
    }
}

impl Display for Type {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(f, "{}", self.kind)
    }
}

impl Display for TypeKind {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match &self {
            TypeKind::Simple(name) => write!(f, "{name}"),
            TypeKind::Tuple(types) => write_delim_list(f, types, "(", ")", ", "),
            TypeKind::Optional(ty) => write!(f, "{ty}?"),
        }
    }
}

impl Display for TypeDeclField {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(f, "{}: {}", &self.name, &self.ty)
    }
}

impl Display for TypeDeclaration {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(f, "type {}", format_opt(&self.name, "", ""))?;
        write_delim_list(f, &self.fields, " { ", " }", ", ")
    }
}

impl Display for Expression {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match &self.kind {
            ExpressionKind::Lit(l) => write!(f, "{l}"),
            ExpressionKind::Ident(ident) => write!(f, "{ident}"),
            ExpressionKind::StreamAccess(expr, access) => match access {
                StreamAccessKind::Sync => write!(f, "{expr}"),
                StreamAccessKind::Hold => write!(f, "{expr}.hold()"),
                StreamAccessKind::Get => write!(f, "{expr}.get()"),
                StreamAccessKind::Fresh => write!(f, "{expr}.is_fresh()"),
            },
            ExpressionKind::Default(expr, val) => write!(f, "{expr}.defaults(to: {val})"),
            ExpressionKind::Offset(expr, val) => write!(f, "{expr}.offset(by: {val})"),
            ExpressionKind::DiscreteWindowAggregation {
                expr,
                duration,
                wait,
                aggregation,
            } => match wait {
                true => {
                    write!(
                        f,
                        "{expr}.aggregate(over_exactly_discrete: {duration}, using: {aggregation})"
                    )
                }
                false => {
                    write!(
                        f,
                        "{expr}.aggregate(over_discrete: {duration}, using: {aggregation})"
                    )
                }
            },
            ExpressionKind::SlidingWindowAggregation {
                expr,
                duration,
                wait,
                aggregation,
            } => match wait {
                true => {
                    write!(
                        f,
                        "{expr}.aggregate(over_exactly: {duration}, using: {aggregation})"
                    )
                }
                false => write!(
                    f,
                    "{expr}.aggregate(over: {duration}, using: {aggregation})"
                ),
            },
            ExpressionKind::InstanceAggregation {
                expr,
                selection,
                aggregation,
            } => write!(
                f,
                "{expr}.aggregate(over_instances: {selection}, using: {aggregation})"
            ),
            ExpressionKind::Binary(op, lhs, rhs) => write!(f, "{lhs} {op} {rhs}"),
            ExpressionKind::Unary(operator, operand) => write!(f, "{operator}{operand}"),
            ExpressionKind::Ite(cond, cons, alt) => {
                write!(f, "if {cond} then {cons} else {alt}")
            }
            ExpressionKind::ParenthesizedExpression(expr) => {
                write!(f, "({})", expr,)
            }
            ExpressionKind::MissingExpression => Ok(()),
            ExpressionKind::Tuple(exprs) => write_delim_list(f, exprs, "(", ")", ", "),
            ExpressionKind::Function(name, types, args) => {
                write!(f, "{}", name.name)?;
                if !types.is_empty() {
                    write_delim_list(f, types, "<", ">", ", ")?;
                }
                let args: Vec<String> = args
                    .iter()
                    .zip(&name.arg_names)
                    .map(|(arg, arg_name)| match arg_name {
                        None => format!("{arg}"),
                        Some(arg_name) => format!("{arg_name}: {arg}"),
                    })
                    .collect();
                write_delim_list(f, &args, "(", ")", ", ")
            }
            ExpressionKind::Field(expr, ident) => write!(f, "{expr}.{ident}"),
            ExpressionKind::Method(expr, name, types, args) => {
                write!(f, "{}.{}", expr, name.name)?;
                if !types.is_empty() {
                    write_delim_list(f, types, "<", ">", ", ")?;
                }
                let args: Vec<String> = args
                    .iter()
                    .zip(&name.arg_names)
                    .map(|(arg, arg_name)| match arg_name {
                        None => format!("{arg}"),
                        Some(arg_name) => format!("{arg_name}: {arg}"),
                    })
                    .collect();
                write_delim_list(f, &args, "(", ")", ", ")
            }
            ExpressionKind::Lambda(lambda) => {
                write!(f, "{lambda}")
            }
        }
    }
}

impl Display for LambdaExpr {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        let LambdaExpr { parameters, expr } = self;
        let parameters = parameters.iter().map(|p| p.to_string()).join(",");
        write!(f, "({parameters}) => {expr}")
    }
}

impl Display for FunctionName {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(f, "{}", self.name)?;
        let args: Vec<String> = self
            .arg_names
            .iter()
            .map(|arg_name| match arg_name {
                None => String::from("_:"),
                Some(arg_name) => format!("{arg_name}:"),
            })
            .collect();
        write_delim_list(f, &args, "(", ")", "")
    }
}

impl Display for Offset {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match self {
            Offset::Discrete(val) => write!(f, "{val}"),
            Offset::RealTime(val, unit) => write!(f, "{val}{unit}"),
        }
    }
}

impl Display for TimeUnit {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(
            f,
            "{}",
            match self {
                TimeUnit::Nanosecond => "ns",
                TimeUnit::Microsecond => "μs",
                TimeUnit::Millisecond => "ms",
                TimeUnit::Second => "s",
                TimeUnit::Minute => "min",
                TimeUnit::Hour => "h",
                TimeUnit::Day => "d",
                TimeUnit::Week => "w",
                TimeUnit::Year => "a",
            }
        )
    }
}

impl Display for WindowOperation {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        let op_str = match self {
            WindowOperation::Sum => "Σ",
            WindowOperation::Product => "Π",
            WindowOperation::Average => "avg",
            WindowOperation::Count => "#",
            WindowOperation::Integral => "∫",
            WindowOperation::Min => "min",
            WindowOperation::Max => "max",
            WindowOperation::Disjunction => "∃",
            WindowOperation::Conjunction => "∀",
            WindowOperation::Last => "last",
            WindowOperation::Variance => "σ²",
            WindowOperation::Covariance => "cov",
            WindowOperation::StandardDeviation => "σ",
            WindowOperation::NthPercentile(p) => return write!(f, "pctl{p}"),
            WindowOperation::TrueRatio => "true_ratio",
        };
        write!(f, "{op_str}")
    }
}

impl Display for UnOp {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(
            f,
            "{}",
            match self {
                UnOp::Not => "!",
                UnOp::Neg => "-",
                UnOp::BitNot => "~",
            }
        )
    }
}

impl Display for Literal {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match &self.kind {
            LitKind::Bool(val) => write!(f, "{val}"),
            LitKind::Numeric(val, unit) => write!(f, "{}{}", val, unit.clone().unwrap_or_default()),
            LitKind::Str(s) => write!(f, "\"{s}\""),
            LitKind::Tuple(elements) => write_delim_list(f, elements, "(", ")", ", "),
            LitKind::RawStr(s) => {
                // need to determine padding with `#`
                let mut padding = 0;
                while s.contains(&format!("{}\"", "#".repeat(padding))) {
                    padding += 1;
                }
                write!(f, "r{pad}\"{}\"{pad}", s, pad = "#".repeat(padding))
            }
        }
    }
}

impl Display for Ident {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        write!(f, "{}", self.name)
    }
}

impl Display for InstanceSelection {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        match self {
            InstanceSelection::Fresh => write!(f, "fresh"),
            InstanceSelection::All => write!(f, "all"),
            InstanceSelection::FilteredFresh(lambda) => write!(f, "fresh(where: {lambda})",),
            InstanceSelection::FilteredAll(lambda) => write!(f, "all(where: {lambda})"),
        }
    }
}

impl Display for InstanceOperation {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        let wo: WindowOperation = (*self).into();
        wo.fmt(f)
    }
}

impl Display for BinOp {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        use BinOp::*;
        match &self {
            // Arithmetic
            Add => write!(f, "+"),
            Sub => write!(f, "-"),
            Mul => write!(f, "*"),
            Div => write!(f, "/"),
            Rem => write!(f, "%"),
            Pow => write!(f, "**"),
            // Logical
            And => write!(f, "∧"),
            Or => write!(f, "∨"),
            Implies => write!(f, "->"),
            // Comparison
            Eq => write!(f, "="),
            Lt => write!(f, "<"),
            Le => write!(f, "≤"),
            Ne => write!(f, "≠"),
            Gt => write!(f, ">"),
            Ge => write!(f, "≥"),
            // Bitwise
            BitAnd => write!(f, "&"),
            BitOr => write!(f, "|"),
            BitXor => write!(f, "^"),
            Shl => write!(f, "<<"),
            Shr => write!(f, ">>"),
        }
    }
}

impl Display for Import {
    fn fmt(&self, f: &mut Formatter) -> Result {
        write!(f, "import {}", self.name)
    }
}

impl Display for RtLolaAst {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result {
        format_tags(f, &self.global_tags, true)?;
        for import in &self.imports {
            writeln!(f, "{import}")?;
        }
        for decl in &self.type_declarations {
            writeln!(f, "{decl}")?;
        }
        for constant in &self.constants {
            writeln!(f, "{constant}")?;
        }
        for input in &self.inputs {
            writeln!(f, "{input}")?;
        }
        for output in &self.outputs {
            writeln!(f, "{output}")?;
        }
        for mirror in &self.mirrors {
            writeln!(f, "{mirror}")?;
        }
        Ok(())
    }
}

impl Display for NodeId {
    fn fmt(&self, f: &mut Formatter) -> Result {
        write!(f, "{}", self.0)
    }
}