mirror of
https://github.com/roc-lang/roc.git
synced 2025-10-03 00:24:34 +00:00
Merge remote-tracking branch 'origin/trunk' into specialize-separately
This commit is contained in:
commit
1b3b9ee1c4
22 changed files with 499 additions and 465 deletions
|
@ -65,7 +65,7 @@ pub fn canonicalize_annotation(
|
|||
scope: &mut Scope,
|
||||
annotation: &roc_parse::ast::TypeAnnotation,
|
||||
region: Region,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
) -> Annotation {
|
||||
let mut introduced_variables = IntroducedVariables::default();
|
||||
let mut aliases = SendMap::default();
|
||||
|
@ -95,7 +95,7 @@ fn can_annotation_help(
|
|||
annotation: &roc_parse::ast::TypeAnnotation,
|
||||
region: Region,
|
||||
scope: &mut Scope,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
introduced_variables: &mut IntroducedVariables,
|
||||
local_aliases: &mut SendMap<Symbol, Alias>,
|
||||
references: &mut MutSet<Symbol>,
|
||||
|
@ -407,7 +407,7 @@ fn can_assigned_field<'a>(
|
|||
field: &AssignedField<'a, TypeAnnotation<'a>>,
|
||||
region: Region,
|
||||
scope: &mut Scope,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
introduced_variables: &mut IntroducedVariables,
|
||||
local_aliases: &mut SendMap<Symbol, Alias>,
|
||||
field_types: &mut SendMap<Lowercase, Type>,
|
||||
|
@ -472,7 +472,7 @@ fn can_tag<'a>(
|
|||
tag: &Tag<'a>,
|
||||
region: Region,
|
||||
scope: &mut Scope,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
introduced_variables: &mut IntroducedVariables,
|
||||
local_aliases: &mut SendMap<Symbol, Alias>,
|
||||
tag_types: &mut Vec<(TagName, Vec<Type>)>,
|
||||
|
|
|
@ -24,7 +24,7 @@ use roc_types::subs::{VarStore, Variable};
|
|||
/// delegates to the compiler-internal List.getUnsafe function to do the actual
|
||||
/// lookup (if the bounds check passed). That internal function is hardcoded in code gen,
|
||||
/// which works fine because it doesn't involve any open tag unions.
|
||||
pub fn builtin_defs(var_store: &VarStore) -> MutMap<Symbol, Expr> {
|
||||
pub fn builtin_defs(var_store: &mut VarStore) -> MutMap<Symbol, Expr> {
|
||||
mut_map! {
|
||||
Symbol::LIST_LEN => list_len(var_store),
|
||||
Symbol::LIST_GET => list_get(var_store),
|
||||
|
@ -45,185 +45,203 @@ pub fn builtin_defs(var_store: &VarStore) -> MutMap<Symbol, Expr> {
|
|||
}
|
||||
|
||||
/// Float.tan : Float -> Float
|
||||
fn float_tan(var_store: &VarStore) -> Expr {
|
||||
fn float_tan(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::FLOAT_DIV,
|
||||
vec![
|
||||
call(
|
||||
Symbol::FLOAT_SIN,
|
||||
vec![Var(Symbol::FLOAT_TAN_ARG)],
|
||||
var_store,
|
||||
),
|
||||
call(
|
||||
Symbol::FLOAT_COS,
|
||||
vec![Var(Symbol::FLOAT_TAN_ARG)],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::FLOAT_TAN,
|
||||
vec![Symbol::FLOAT_TAN_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::FLOAT_DIV,
|
||||
vec![
|
||||
call(
|
||||
Symbol::FLOAT_SIN,
|
||||
vec![Var(Symbol::FLOAT_TAN_ARG)],
|
||||
var_store,
|
||||
),
|
||||
call(
|
||||
Symbol::FLOAT_COS,
|
||||
vec![Var(Symbol::FLOAT_TAN_ARG)],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Float.isZero : Float -> Bool
|
||||
fn float_is_zero(var_store: &VarStore) -> Expr {
|
||||
fn float_is_zero(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::FLOAT_EQ,
|
||||
vec![
|
||||
Float(var_store.fresh(), 0.0),
|
||||
Var(Symbol::FLOAT_IS_ZERO_ARG),
|
||||
],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::FLOAT_IS_ZERO,
|
||||
vec![Symbol::FLOAT_IS_ZERO_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::FLOAT_EQ,
|
||||
vec![
|
||||
Float(var_store.fresh(), 0.0),
|
||||
Var(Symbol::FLOAT_IS_ZERO_ARG),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Float.isNegative : Float -> Bool
|
||||
fn float_is_negative(var_store: &VarStore) -> Expr {
|
||||
fn float_is_negative(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::FLOAT_GT,
|
||||
vec![
|
||||
Float(var_store.fresh(), 0.0),
|
||||
Var(Symbol::FLOAT_IS_NEGATIVE_ARG),
|
||||
],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::FLOAT_IS_NEGATIVE,
|
||||
vec![Symbol::FLOAT_IS_NEGATIVE_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::FLOAT_GT,
|
||||
vec![
|
||||
Float(var_store.fresh(), 0.0),
|
||||
Var(Symbol::FLOAT_IS_NEGATIVE_ARG),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Float.isPositive : Float -> Bool
|
||||
fn float_is_positive(var_store: &VarStore) -> Expr {
|
||||
fn float_is_positive(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::FLOAT_GT,
|
||||
vec![
|
||||
Var(Symbol::FLOAT_IS_POSITIVE_ARG),
|
||||
Float(var_store.fresh(), 0.0),
|
||||
],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::FLOAT_IS_POSITIVE,
|
||||
vec![Symbol::FLOAT_IS_POSITIVE_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::FLOAT_GT,
|
||||
vec![
|
||||
Var(Symbol::FLOAT_IS_POSITIVE_ARG),
|
||||
Float(var_store.fresh(), 0.0),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.isNegative : Int -> Bool
|
||||
fn int_is_negative(var_store: &VarStore) -> Expr {
|
||||
fn int_is_negative(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::NUM_LT,
|
||||
vec![Var(Symbol::INT_IS_NEGATIVE_ARG), Int(var_store.fresh(), 0)],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::INT_IS_NEGATIVE,
|
||||
vec![Symbol::INT_IS_NEGATIVE_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::NUM_LT,
|
||||
vec![Var(Symbol::INT_IS_NEGATIVE_ARG), Int(var_store.fresh(), 0)],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.isPositive : Int -> Bool
|
||||
fn int_is_positive(var_store: &VarStore) -> Expr {
|
||||
fn int_is_positive(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::NUM_GT,
|
||||
vec![Var(Symbol::INT_IS_POSITIVE_ARG), Int(var_store.fresh(), 0)],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::INT_IS_POSITIVE,
|
||||
vec![Symbol::INT_IS_POSITIVE_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::NUM_GT,
|
||||
vec![Var(Symbol::INT_IS_POSITIVE_ARG), Int(var_store.fresh(), 0)],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.isZero : Int -> Bool
|
||||
fn int_is_zero(var_store: &VarStore) -> Expr {
|
||||
fn int_is_zero(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::INT_EQ_I64,
|
||||
vec![Var(Symbol::INT_IS_ZERO_ARG), Int(var_store.fresh(), 0)],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::INT_IS_ZERO,
|
||||
vec![Symbol::INT_IS_ZERO_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::INT_EQ_I64,
|
||||
vec![Var(Symbol::INT_IS_ZERO_ARG), Int(var_store.fresh(), 0)],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.isOdd : Int -> Bool
|
||||
fn int_is_odd(var_store: &VarStore) -> Expr {
|
||||
fn int_is_odd(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::INT_EQ_I64,
|
||||
vec![
|
||||
call(
|
||||
Symbol::INT_REM_UNSAFE,
|
||||
vec![Var(Symbol::INT_IS_ODD_ARG), Int(var_store.fresh(), 2)],
|
||||
var_store,
|
||||
),
|
||||
Int(var_store.fresh(), 1),
|
||||
],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::INT_IS_ODD,
|
||||
vec![Symbol::INT_IS_ODD_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::INT_EQ_I64,
|
||||
vec![
|
||||
call(
|
||||
Symbol::INT_REM_UNSAFE,
|
||||
vec![Var(Symbol::INT_IS_ODD_ARG), Int(var_store.fresh(), 2)],
|
||||
var_store,
|
||||
),
|
||||
Int(var_store.fresh(), 1),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.isEven : Int -> Bool
|
||||
fn int_is_even(var_store: &VarStore) -> Expr {
|
||||
fn int_is_even(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = call(
|
||||
Symbol::INT_EQ_I64,
|
||||
vec![
|
||||
call(
|
||||
Symbol::INT_REM_UNSAFE,
|
||||
vec![Var(Symbol::INT_IS_EVEN_ARG), Int(var_store.fresh(), 2)],
|
||||
var_store,
|
||||
),
|
||||
Int(var_store.fresh(), 0),
|
||||
],
|
||||
var_store,
|
||||
);
|
||||
|
||||
defn(
|
||||
Symbol::INT_IS_EVEN,
|
||||
vec![Symbol::INT_IS_EVEN_ARG],
|
||||
var_store,
|
||||
call(
|
||||
Symbol::INT_EQ_I64,
|
||||
vec![
|
||||
call(
|
||||
Symbol::INT_REM_UNSAFE,
|
||||
vec![Var(Symbol::INT_IS_EVEN_ARG), Int(var_store.fresh(), 2)],
|
||||
var_store,
|
||||
),
|
||||
Int(var_store.fresh(), 0),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// List.len : List * -> Int
|
||||
fn list_len(var_store: &VarStore) -> Expr {
|
||||
fn list_len(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
// Polymorphic wrapper around LowLevel::ListLen
|
||||
|
@ -238,162 +256,210 @@ fn list_len(var_store: &VarStore) -> Expr {
|
|||
}
|
||||
|
||||
/// List.get : List elem, Int -> Result elem [ OutOfBounds ]*
|
||||
fn list_get(var_store: &VarStore) -> Expr {
|
||||
fn list_get(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
// Perform a bounds check. If it passes, delegate to List.#getUnsafe
|
||||
let body = If {
|
||||
cond_var: var_store.fresh(),
|
||||
branch_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// index < List.len list
|
||||
call(
|
||||
Symbol::NUM_LT,
|
||||
vec![
|
||||
Var(Symbol::LIST_GET_ARG_INDEX),
|
||||
call(
|
||||
Symbol::LIST_LEN,
|
||||
vec![Var(Symbol::LIST_GET_ARG_LIST)],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// then-branch
|
||||
no_region(
|
||||
// Ok
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// List.getUnsafe list index
|
||||
Call(
|
||||
Box::new((
|
||||
var_store.fresh(),
|
||||
no_region(Var(Symbol::LIST_GET_UNSAFE)),
|
||||
var_store.fresh(),
|
||||
)),
|
||||
vec![
|
||||
(var_store.fresh(), no_region(Var(Symbol::LIST_GET_ARG_LIST))),
|
||||
(
|
||||
var_store.fresh(),
|
||||
no_region(Var(Symbol::LIST_GET_ARG_INDEX)),
|
||||
),
|
||||
],
|
||||
CalledVia::Space,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// else-branch
|
||||
no_region(
|
||||
// Err
|
||||
tag(
|
||||
"Err",
|
||||
vec![tag("OutOfBounds", Vec::new(), var_store)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
),
|
||||
};
|
||||
|
||||
defn(
|
||||
Symbol::LIST_GET,
|
||||
vec![Symbol::LIST_GET_ARG_LIST, Symbol::LIST_GET_ARG_INDEX],
|
||||
var_store,
|
||||
// Perform a bounds check. If it passes, delegate to List.#getUnsafe
|
||||
If {
|
||||
cond_var: var_store.fresh(),
|
||||
branch_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// index < List.len list
|
||||
call(
|
||||
Symbol::NUM_LT,
|
||||
vec![
|
||||
Var(Symbol::LIST_GET_ARG_INDEX),
|
||||
RunLowLevel(LowLevel::ListLen {
|
||||
arg: Symbol::LIST_GET_ARG_LIST,
|
||||
}),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// then-branch
|
||||
no_region(
|
||||
// Ok
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// List.getUnsafe list index
|
||||
Call(
|
||||
Box::new((
|
||||
var_store.fresh(),
|
||||
no_region(Var(Symbol::LIST_GET_UNSAFE)),
|
||||
var_store.fresh(),
|
||||
)),
|
||||
vec![
|
||||
(var_store.fresh(), no_region(Var(Symbol::LIST_GET_ARG_LIST))),
|
||||
(
|
||||
var_store.fresh(),
|
||||
no_region(Var(Symbol::LIST_GET_ARG_INDEX)),
|
||||
),
|
||||
],
|
||||
CalledVia::Space,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// else-branch
|
||||
no_region(
|
||||
// Err
|
||||
tag(
|
||||
"Err",
|
||||
vec![tag("OutOfBounds", Vec::new(), var_store)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
),
|
||||
},
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.rem : Int, Int -> Int
|
||||
fn int_rem(var_store: &VarStore) -> Expr {
|
||||
fn int_rem(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = If {
|
||||
branch_var: var_store.fresh(),
|
||||
cond_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if condition
|
||||
no_region(
|
||||
// Int.neq arg1 0
|
||||
call(
|
||||
Symbol::INT_NEQ_I64,
|
||||
vec![Var(Symbol::INT_REM_ARG_1), (Int(var_store.fresh(), 0))],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// arg1 was not zero
|
||||
no_region(
|
||||
// Ok (Int.#remUnsafe arg0 arg1)
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// Int.#remUnsafe arg0 arg1
|
||||
call(
|
||||
Symbol::INT_REM_UNSAFE,
|
||||
vec![Var(Symbol::INT_REM_ARG_0), Var(Symbol::INT_REM_ARG_1)],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(no_region(tag(
|
||||
"Err",
|
||||
vec![tag("DivByZero", Vec::new(), var_store)],
|
||||
var_store,
|
||||
))),
|
||||
};
|
||||
|
||||
defn(
|
||||
Symbol::INT_REM,
|
||||
vec![Symbol::INT_REM_ARG_0, Symbol::INT_REM_ARG_1],
|
||||
var_store,
|
||||
If {
|
||||
branch_var: var_store.fresh(),
|
||||
cond_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if condition
|
||||
no_region(
|
||||
// Int.neq arg1 0
|
||||
call(
|
||||
Symbol::INT_NEQ_I64,
|
||||
vec![Var(Symbol::INT_REM_ARG_1), (Int(var_store.fresh(), 0))],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// arg1 was not zero
|
||||
no_region(
|
||||
// Ok (Int.#remUnsafe arg0 arg1)
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// Int.#remUnsafe arg0 arg1
|
||||
call(
|
||||
Symbol::INT_REM_UNSAFE,
|
||||
vec![Var(Symbol::INT_REM_ARG_0), Var(Symbol::INT_REM_ARG_1)],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(no_region(tag(
|
||||
"Err",
|
||||
vec![tag("DivByZero", Vec::new(), var_store)],
|
||||
var_store,
|
||||
))),
|
||||
},
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// Int.abs : Int -> Int
|
||||
fn int_abs(var_store: &VarStore) -> Expr {
|
||||
fn int_abs(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
defn(
|
||||
Symbol::INT_ABS,
|
||||
vec![Symbol::INT_ABS_ARG],
|
||||
var_store,
|
||||
If {
|
||||
branch_var: var_store.fresh(),
|
||||
cond_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// Int.isLt 0 n
|
||||
// 0 < n
|
||||
call(
|
||||
Symbol::INT_LT,
|
||||
vec![Int(var_store.fresh(), 0), Var(Symbol::INT_ABS_ARG)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// int is at least 0, so just pass it along
|
||||
no_region(Var(Symbol::INT_ABS_ARG)),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// int is below 0, so negate it.
|
||||
no_region(call(
|
||||
Symbol::NUM_NEG,
|
||||
vec![Var(Symbol::INT_ABS_ARG)],
|
||||
let body = If {
|
||||
branch_var: var_store.fresh(),
|
||||
cond_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// Int.isLt 0 n
|
||||
// 0 < n
|
||||
call(
|
||||
Symbol::INT_LT,
|
||||
vec![Int(var_store.fresh(), 0), Var(Symbol::INT_ABS_ARG)],
|
||||
var_store,
|
||||
)),
|
||||
),
|
||||
),
|
||||
},
|
||||
)
|
||||
// int is at least 0, so just pass it along
|
||||
no_region(Var(Symbol::INT_ABS_ARG)),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// int is below 0, so negate it.
|
||||
no_region(call(
|
||||
Symbol::NUM_NEG,
|
||||
vec![Var(Symbol::INT_ABS_ARG)],
|
||||
var_store,
|
||||
)),
|
||||
),
|
||||
};
|
||||
|
||||
defn(Symbol::INT_ABS, vec![Symbol::INT_ABS_ARG], var_store, body)
|
||||
}
|
||||
|
||||
/// Int.div : Int, Int -> Result Int [ DivByZero ]*
|
||||
fn int_div(var_store: &VarStore) -> Expr {
|
||||
fn int_div(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
let body = If {
|
||||
branch_var: var_store.fresh(),
|
||||
cond_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// Int.neq denominator 0
|
||||
call(
|
||||
Symbol::INT_NEQ_I64,
|
||||
vec![
|
||||
Var(Symbol::INT_DIV_ARG_DENOMINATOR),
|
||||
(Int(var_store.fresh(), 0)),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// denominator was not zero
|
||||
no_region(
|
||||
// Ok (Int.#divUnsafe numerator denominator)
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// Int.#divUnsafe numerator denominator
|
||||
call(
|
||||
Symbol::INT_DIV_UNSAFE,
|
||||
vec![
|
||||
Var(Symbol::INT_DIV_ARG_NUMERATOR),
|
||||
Var(Symbol::INT_DIV_ARG_DENOMINATOR),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// denominator was zero
|
||||
no_region(tag(
|
||||
"Err",
|
||||
vec![tag("DivByZero", Vec::new(), var_store)],
|
||||
var_store,
|
||||
)),
|
||||
),
|
||||
};
|
||||
|
||||
defn(
|
||||
Symbol::INT_DIV,
|
||||
vec![
|
||||
|
@ -401,107 +467,65 @@ fn int_div(var_store: &VarStore) -> Expr {
|
|||
Symbol::INT_DIV_ARG_DENOMINATOR,
|
||||
],
|
||||
var_store,
|
||||
If {
|
||||
branch_var: var_store.fresh(),
|
||||
cond_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// Int.neq denominator 0
|
||||
call(
|
||||
Symbol::INT_NEQ_I64,
|
||||
vec![
|
||||
Var(Symbol::INT_DIV_ARG_DENOMINATOR),
|
||||
(Int(var_store.fresh(), 0)),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// denominator was not zero
|
||||
no_region(
|
||||
// Ok (Int.#divUnsafe numerator denominator)
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// Int.#divUnsafe numerator denominator
|
||||
call(
|
||||
Symbol::INT_DIV_UNSAFE,
|
||||
vec![
|
||||
Var(Symbol::INT_DIV_ARG_NUMERATOR),
|
||||
Var(Symbol::INT_DIV_ARG_DENOMINATOR),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// denominator was zero
|
||||
no_region(tag(
|
||||
"Err",
|
||||
vec![tag("DivByZero", Vec::new(), var_store)],
|
||||
var_store,
|
||||
)),
|
||||
),
|
||||
},
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
/// List.first : List elem -> Result elem [ ListWasEmpty ]*
|
||||
fn list_first(var_store: &VarStore) -> Expr {
|
||||
fn list_first(var_store: &mut VarStore) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
|
||||
// Perform a bounds check. If it passes, delegate to List.getUnsafe.
|
||||
let body = If {
|
||||
// TODO Use "when" instead of "if" so that we can have False be the first branch.
|
||||
// We want that for branch prediction; usually we expect the list to be nonempty.
|
||||
cond_var: var_store.fresh(),
|
||||
branch_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// List.isEmpty list
|
||||
call(
|
||||
Symbol::LIST_IS_EMPTY,
|
||||
vec![Var(Symbol::LIST_FIRST_ARG)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// list was empty
|
||||
no_region(
|
||||
// Err ListWasEmpty
|
||||
tag(
|
||||
"Err",
|
||||
vec![tag("ListWasEmpty", Vec::new(), var_store)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// list was not empty
|
||||
no_region(
|
||||
// Ok (List.#getUnsafe list 0)
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// List.#getUnsafe list 0
|
||||
call(
|
||||
Symbol::LIST_GET_UNSAFE,
|
||||
vec![(Var(Symbol::LIST_FIRST_ARG)), (Int(var_store.fresh(), 0))],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
),
|
||||
};
|
||||
|
||||
defn(
|
||||
Symbol::LIST_FIRST,
|
||||
vec![Symbol::LIST_FIRST_ARG],
|
||||
var_store,
|
||||
// Perform a bounds check. If it passes, delegate to List.getUnsafe.
|
||||
If {
|
||||
// TODO Use "when" instead of "if" so that we can have False be the first branch.
|
||||
// We want that for branch prediction; usually we expect the list to be nonempty.
|
||||
cond_var: var_store.fresh(),
|
||||
branch_var: var_store.fresh(),
|
||||
branches: vec![(
|
||||
// if-condition
|
||||
no_region(
|
||||
// List.isEmpty list
|
||||
call(
|
||||
Symbol::LIST_IS_EMPTY,
|
||||
vec![Var(Symbol::LIST_FIRST_ARG)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
// list was empty
|
||||
no_region(
|
||||
// Err ListWasEmpty
|
||||
tag(
|
||||
"Err",
|
||||
vec![tag("ListWasEmpty", Vec::new(), var_store)],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
)],
|
||||
final_else: Box::new(
|
||||
// list was not empty
|
||||
no_region(
|
||||
// Ok (List.#getUnsafe list 0)
|
||||
tag(
|
||||
"Ok",
|
||||
vec![
|
||||
// List.#getUnsafe list 0
|
||||
call(
|
||||
Symbol::LIST_GET_UNSAFE,
|
||||
vec![(Var(Symbol::LIST_FIRST_ARG)), (Int(var_store.fresh(), 0))],
|
||||
var_store,
|
||||
),
|
||||
],
|
||||
var_store,
|
||||
),
|
||||
),
|
||||
),
|
||||
},
|
||||
body,
|
||||
)
|
||||
}
|
||||
|
||||
|
@ -514,7 +538,7 @@ fn no_region<T>(value: T) -> Located<T> {
|
|||
}
|
||||
|
||||
#[inline(always)]
|
||||
fn tag(name: &'static str, args: Vec<Expr>, var_store: &VarStore) -> Expr {
|
||||
fn tag(name: &'static str, args: Vec<Expr>, var_store: &mut VarStore) -> Expr {
|
||||
Expr::Tag {
|
||||
variant_var: var_store.fresh(),
|
||||
ext_var: var_store.fresh(),
|
||||
|
@ -527,7 +551,7 @@ fn tag(name: &'static str, args: Vec<Expr>, var_store: &VarStore) -> Expr {
|
|||
}
|
||||
|
||||
#[inline(always)]
|
||||
fn call(symbol: Symbol, args: Vec<Expr>, var_store: &VarStore) -> Expr {
|
||||
fn call(symbol: Symbol, args: Vec<Expr>, var_store: &mut VarStore) -> Expr {
|
||||
Expr::Call(
|
||||
Box::new((
|
||||
var_store.fresh(),
|
||||
|
@ -542,7 +566,7 @@ fn call(symbol: Symbol, args: Vec<Expr>, var_store: &VarStore) -> Expr {
|
|||
}
|
||||
|
||||
#[inline(always)]
|
||||
fn defn(fn_name: Symbol, args: Vec<Symbol>, var_store: &VarStore, body: Expr) -> Expr {
|
||||
fn defn(fn_name: Symbol, args: Vec<Symbol>, var_store: &mut VarStore, body: Expr) -> Expr {
|
||||
use crate::expr::Expr::*;
|
||||
use crate::pattern::Pattern::*;
|
||||
|
||||
|
|
|
@ -17,14 +17,14 @@ pub enum Constraint {
|
|||
}
|
||||
|
||||
impl Constraint {
|
||||
pub fn instantiate_aliases(&mut self, var_store: &VarStore) {
|
||||
pub fn instantiate_aliases(&mut self, var_store: &mut VarStore) {
|
||||
Self::instantiate_aliases_help(self, &ImMap::default(), var_store, &mut ImSet::default())
|
||||
}
|
||||
|
||||
fn instantiate_aliases_help(
|
||||
&mut self,
|
||||
aliases: &ImMap<Symbol, Alias>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
introduced: &mut ImSet<Variable>,
|
||||
) {
|
||||
use Constraint::*;
|
||||
|
|
|
@ -110,7 +110,7 @@ impl Declaration {
|
|||
pub fn canonicalize_defs<'a>(
|
||||
env: &mut Env<'a>,
|
||||
mut output: Output,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
original_scope: &Scope,
|
||||
loc_defs: &'a bumpalo::collections::Vec<'a, &'a Located<ast::Def<'a>>>,
|
||||
pattern_type: PatternType,
|
||||
|
@ -253,7 +253,7 @@ pub fn canonicalize_defs<'a>(
|
|||
}
|
||||
}
|
||||
|
||||
correct_mutual_recursive_type_alias(env, &mut aliases, &var_store);
|
||||
correct_mutual_recursive_type_alias(env, &mut aliases, var_store);
|
||||
|
||||
// Now that we have the scope completely assembled, and shadowing resolved,
|
||||
// we're ready to canonicalize any body exprs.
|
||||
|
@ -725,7 +725,7 @@ fn canonicalize_pending_def<'a>(
|
|||
mut output: Output,
|
||||
scope: &mut Scope,
|
||||
can_defs_by_symbol: &mut MutMap<Symbol, Def>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
refs_by_symbol: &mut MutMap<Symbol, (Located<Ident>, References)>,
|
||||
aliases: &mut SendMap<Symbol, Alias>,
|
||||
) -> Output {
|
||||
|
@ -1184,7 +1184,7 @@ fn canonicalize_pending_def<'a>(
|
|||
#[inline(always)]
|
||||
pub fn can_defs_with_return<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: Scope,
|
||||
loc_defs: &'a bumpalo::collections::Vec<'a, &'a Located<ast::Def<'a>>>,
|
||||
loc_ret: &'a Located<ast::Expr<'a>>,
|
||||
|
@ -1236,7 +1236,7 @@ pub fn can_defs_with_return<'a>(
|
|||
}
|
||||
|
||||
fn decl_to_let(
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
decl: Declaration,
|
||||
loc_ret: Located<Expr>,
|
||||
aliases: SendMap<Symbol, Alias>,
|
||||
|
@ -1294,7 +1294,7 @@ fn closure_recursivity(symbol: Symbol, closures: &MutMap<Symbol, References>) ->
|
|||
|
||||
fn to_pending_def<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
def: &'a ast::Def<'a>,
|
||||
scope: &mut Scope,
|
||||
pattern_type: PatternType,
|
||||
|
@ -1399,7 +1399,7 @@ fn pending_typed_body<'a>(
|
|||
loc_pattern: &'a Located<ast::Pattern<'a>>,
|
||||
loc_ann: &'a Located<ast::TypeAnnotation<'a>>,
|
||||
loc_expr: &'a Located<ast::Expr<'a>>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: &mut Scope,
|
||||
pattern_type: PatternType,
|
||||
) -> PendingDef<'a> {
|
||||
|
@ -1420,7 +1420,7 @@ fn pending_typed_body<'a>(
|
|||
fn correct_mutual_recursive_type_alias<'a>(
|
||||
env: &mut Env<'a>,
|
||||
aliases: &mut SendMap<Symbol, Alias>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
) {
|
||||
let mut symbols_introduced = ImSet::default();
|
||||
|
||||
|
@ -1516,7 +1516,7 @@ fn make_tag_union_recursive<'a>(
|
|||
region: Region,
|
||||
others: Vec<Symbol>,
|
||||
typ: &mut Type,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
can_report_error: &mut bool,
|
||||
) {
|
||||
match typ {
|
||||
|
|
|
@ -170,7 +170,7 @@ pub struct WhenBranch {
|
|||
|
||||
pub fn canonicalize_expr<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: &mut Scope,
|
||||
region: Region,
|
||||
expr: &'a ast::Expr<'a>,
|
||||
|
@ -692,7 +692,7 @@ pub fn canonicalize_expr<'a>(
|
|||
#[inline(always)]
|
||||
fn canonicalize_when_branch<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: &mut Scope,
|
||||
_region: Region,
|
||||
branch: &'a ast::WhenBranch<'a>,
|
||||
|
@ -887,7 +887,7 @@ where
|
|||
|
||||
fn canonicalize_fields<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: &mut Scope,
|
||||
region: Region,
|
||||
fields: &'a [Located<ast::AssignedField<'a, ast::Expr<'a>>>],
|
||||
|
@ -924,7 +924,7 @@ fn canonicalize_fields<'a>(
|
|||
|
||||
fn canonicalize_field<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: &mut Scope,
|
||||
field: &'a ast::AssignedField<'a, ast::Expr<'a>>,
|
||||
region: Region,
|
||||
|
|
|
@ -49,7 +49,7 @@ pub fn canonicalize_module_defs<'a>(
|
|||
dep_idents: MutMap<ModuleId, IdentIds>,
|
||||
exposed_imports: MutMap<Ident, (Symbol, Region)>,
|
||||
mut exposed_symbols: MutSet<Symbol>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
) -> Result<ModuleOutput, RuntimeError> {
|
||||
let mut can_exposed_imports = MutMap::default();
|
||||
let mut scope = Scope::new(home);
|
||||
|
|
|
@ -8,7 +8,7 @@ use std::i64;
|
|||
|
||||
#[inline(always)]
|
||||
pub fn num_expr_from_result(
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
result: Result<i64, &str>,
|
||||
env: &mut Env,
|
||||
) -> Expr {
|
||||
|
@ -29,7 +29,7 @@ pub fn num_expr_from_result(
|
|||
|
||||
#[inline(always)]
|
||||
pub fn int_expr_from_result(
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
result: Result<i64, &str>,
|
||||
env: &mut Env,
|
||||
) -> Expr {
|
||||
|
@ -48,7 +48,7 @@ pub fn int_expr_from_result(
|
|||
|
||||
#[inline(always)]
|
||||
pub fn float_expr_from_result(
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
result: Result<f64, &str>,
|
||||
env: &mut Env,
|
||||
) -> Expr {
|
||||
|
|
|
@ -84,7 +84,7 @@ pub fn symbols_from_pattern_help(pattern: &Pattern, symbols: &mut Vec<Symbol>) {
|
|||
|
||||
pub fn canonicalize_pattern<'a>(
|
||||
env: &mut Env<'a>,
|
||||
var_store: &VarStore,
|
||||
var_store: &mut VarStore,
|
||||
scope: &mut Scope,
|
||||
pattern_type: PatternType,
|
||||
pattern: &ast::Pattern<'a>,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue