salsa/calc-example/calc/src/ir.rs

use derive_new::new;
use ordered_float::OrderedFloat;
use salsa::debug::DebugWithDb;

// ANCHOR: input
#[salsa::input]
pub struct SourceProgram {
    #[return_ref]
    text: String,
}
// ANCHOR_END: input

// ANCHOR: interned_ids
#[salsa::interned]
pub struct VariableId {
    #[return_ref]
    pub text: String,
}

#[salsa::interned]
pub struct FunctionId {
    #[return_ref]
    pub text: String,
}
// ANCHOR_END: interned_ids

// ANCHOR: program
#[salsa::tracked]
pub struct Program {
    #[return_ref]
    statements: Vec<Statement>,
}
// ANCHOR_END: program

// ANCHOR: statements_and_expressions
#[derive(Eq, PartialEq, Debug, Hash, new)]
pub struct Statement {
    pub span: Span,

    pub data: StatementData,
}

#[derive(Eq, PartialEq, Debug, Hash)]
pub enum StatementData {
    /// Defines `fn <name>(<args>) = <body>`
    Function(Function),
    /// Defines `print <expr>`
    Print(Expression),
}

#[derive(Eq, PartialEq, Debug, Hash, new)]
pub struct Expression {
    pub span: Span,

    pub data: ExpressionData,
}

#[derive(Eq, PartialEq, Debug, Hash)]
pub enum ExpressionData {
    Op(Box<Expression>, Op, Box<Expression>),
    Number(OrderedFloat<f64>),
    Variable(VariableId),
    Call(FunctionId, Vec<Expression>),
}

#[derive(Eq, PartialEq, Copy, Clone, Hash, Debug)]
pub enum Op {
    Add,
    Subtract,
    Multiply,
    Divide,
}
// ANCHOR_END: statements_and_expressions

// ANCHOR: functions
#[salsa::tracked]
pub struct Function {
    #[id]
    name: FunctionId,

    name_span: Span,

    #[return_ref]
    args: Vec<VariableId>,

    #[return_ref]
    body: Expression,
}
// ANCHOR_END: functions

#[salsa::tracked]
pub struct Span {
    pub start: usize,
    pub end: usize,
}

// ANCHOR: diagnostic
#[salsa::accumulator]
pub struct Diagnostics(Diagnostic);

#[derive(new, Clone, Debug)]
pub struct Diagnostic {
    pub start: usize,
    pub end: usize,
    pub message: String,
}
// ANCHOR_END: diagnostic
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`use derive_new::new;`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`use ordered_float::OrderedFloat;`
			`use salsa::debug::DebugWithDb;`

rewrite calc example to use a `salsa::input` 2022-08-03 04:41:36 +00:00			`// ANCHOR: input`
			`#[salsa::input]`
			`pub struct SourceProgram {`
			`#[return_ref]`
			`text: String,`
			`}`
			`// ANCHOR_END: input`

entity-based salsa preview 2022-08-01 05:32:47 +00:00			`// ANCHOR: interned_ids`
			`#[salsa::interned]`
			`pub struct VariableId {`
			`#[return_ref]`
			`pub text: String,`
			`}`

			`#[salsa::interned]`
			`pub struct FunctionId {`
			`#[return_ref]`
			`pub text: String,`
			`}`
			`// ANCHOR_END: interned_ids`

regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`// ANCHOR: program`
package up the output into a `Program` 2022-08-19 09:53:33 +00:00			`#[salsa::tracked]`
			`pub struct Program {`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`#[return_ref]`
package up the output into a `Program` 2022-08-19 09:53:33 +00:00			`statements: Vec<Statement>,`
			`}`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`// ANCHOR_END: program`
package up the output into a `Program` 2022-08-19 09:53:33 +00:00
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`// ANCHOR: statements_and_expressions`
			`#[derive(Eq, PartialEq, Debug, Hash, new)]`
remove support for interned enums It's nicer to just define a struct with a data field. 2022-08-03 07:43:39 +00:00			`pub struct Statement {`
wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`pub span: Span,`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00
wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`pub data: StatementData,`
remove support for interned enums It's nicer to just define a struct with a data field. 2022-08-03 07:43:39 +00:00			`}`

regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`#[derive(Eq, PartialEq, Debug, Hash)]`
remove support for interned enums It's nicer to just define a struct with a data field. 2022-08-03 07:43:39 +00:00			`pub enum StatementData {`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			/// Defines `fn <name>(<args>) = <body>`
			`Function(Function),`
			/// Defines `print <expr>`
			`Print(Expression),`
			`}`

regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`#[derive(Eq, PartialEq, Debug, Hash, new)]`
remove support for interned enums It's nicer to just define a struct with a data field. 2022-08-03 07:43:39 +00:00			`pub struct Expression {`
wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`pub span: Span,`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00
wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`pub data: ExpressionData,`
remove support for interned enums It's nicer to just define a struct with a data field. 2022-08-03 07:43:39 +00:00			`}`

regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`#[derive(Eq, PartialEq, Debug, Hash)]`
remove support for interned enums It's nicer to just define a struct with a data field. 2022-08-03 07:43:39 +00:00			`pub enum ExpressionData {`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`Op(Box<Expression>, Op, Box<Expression>),`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`Number(OrderedFloat<f64>),`
			`Variable(VariableId),`
			`Call(FunctionId, Vec<Expression>),`
			`}`

			`#[derive(Eq, PartialEq, Copy, Clone, Hash, Debug)]`
			`pub enum Op {`
			`Add,`
			`Subtract,`
			`Multiply,`
			`Divide,`
			`}`
			`// ANCHOR_END: statements_and_expressions`

			`// ANCHOR: functions`
remove `#[salsa::entity]`, and `#[salsa::tracked]` 2022-08-03 13:09:22 +00:00			`#[salsa::tracked]`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`pub struct Function {`
			`#[id]`
			`name: FunctionId,`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00
			`name_span: Span,`

wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`#[return_ref]`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`args: Vec<VariableId>,`
regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00
			`#[return_ref]`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`body: Expression,`
			`}`
			`// ANCHOR_END: functions`

regular structs for stmt,expr / track span Docs are somewhat out of date. Tracking span should enable re-use of type-check results. 2022-08-19 10:46:54 +00:00			`#[salsa::tracked]`
			`pub struct Span {`
			`pub start: usize,`
			`pub end: usize,`
			`}`

entity-based salsa preview 2022-08-01 05:32:47 +00:00			`// ANCHOR: diagnostic`
			`#[salsa::accumulator]`
			`pub struct Diagnostics(Diagnostic);`

wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`#[derive(new, Clone, Debug)]`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`pub struct Diagnostic {`
wip type-checking, testing need to cleanup this commit 2022-08-19 12:41:58 +00:00			`pub start: usize,`
			`pub end: usize,`
entity-based salsa preview 2022-08-01 05:32:47 +00:00			`pub message: String,`
			`}`
			`// ANCHOR_END: diagnostic`