Don't have builtins return Defs

This commit is contained in:
Richard Feldman 2020-06-06 23:55:05 -04:00
parent 093579780f
commit e726d838ea
2 changed files with 34 additions and 29 deletions

View file

@ -1,6 +1,5 @@
use crate::def::Def;
use crate::expr::{Expr, Recursive};
use roc_collections::all::{MutMap, SendMap};
use roc_collections::all::MutMap;
use roc_module::ident::TagName;
use roc_module::operator::CalledVia;
use roc_module::symbol::Symbol;
@ -24,7 +23,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, Def> {
pub fn builtin_defs(var_store: &VarStore) -> MutMap<Symbol, Expr> {
mut_map! {
Symbol::LIST_GET => list_get(var_store),
Symbol::LIST_FIRST => list_first(var_store),
@ -43,7 +42,7 @@ pub fn builtin_defs(var_store: &VarStore) -> MutMap<Symbol, Def> {
}
/// Float.isZero : Float -> Bool
fn float_is_zero(var_store: &VarStore) -> Def {
fn float_is_zero(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -62,7 +61,7 @@ fn float_is_zero(var_store: &VarStore) -> Def {
}
/// Float.isNegative : Float -> Bool
fn float_is_negative(var_store: &VarStore) -> Def {
fn float_is_negative(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -81,7 +80,7 @@ fn float_is_negative(var_store: &VarStore) -> Def {
}
/// Float.isPositive : Float -> Bool
fn float_is_positive(var_store: &VarStore) -> Def {
fn float_is_positive(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -100,7 +99,7 @@ fn float_is_positive(var_store: &VarStore) -> Def {
}
/// Int.isNegative : Int -> Bool
fn int_is_negative(var_store: &VarStore) -> Def {
fn int_is_negative(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -116,7 +115,7 @@ fn int_is_negative(var_store: &VarStore) -> Def {
}
/// Int.isPositive : Int -> Bool
fn int_is_positive(var_store: &VarStore) -> Def {
fn int_is_positive(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -132,7 +131,7 @@ fn int_is_positive(var_store: &VarStore) -> Def {
}
/// Int.isZero : Int -> Bool
fn int_is_zero(var_store: &VarStore) -> Def {
fn int_is_zero(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -148,7 +147,7 @@ fn int_is_zero(var_store: &VarStore) -> Def {
}
/// Int.isOdd : Int -> Bool
fn int_is_odd(var_store: &VarStore) -> Def {
fn int_is_odd(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -171,7 +170,7 @@ fn int_is_odd(var_store: &VarStore) -> Def {
}
/// Int.isEven : Int -> Bool
fn int_is_even(var_store: &VarStore) -> Def {
fn int_is_even(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -194,7 +193,7 @@ fn int_is_even(var_store: &VarStore) -> Def {
}
/// List.get : List elem, Int -> Result elem [ OutOfBounds ]*
fn list_get(var_store: &VarStore) -> Def {
fn list_get(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -265,7 +264,7 @@ fn list_get(var_store: &VarStore) -> Def {
}
/// Int.rem : Int, Int -> Int
fn int_rem(var_store: &VarStore) -> Def {
fn int_rem(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -312,7 +311,7 @@ fn int_rem(var_store: &VarStore) -> Def {
}
/// Int.abs : Int -> Int
fn int_abs(var_store: &VarStore) -> Def {
fn int_abs(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -349,7 +348,7 @@ fn int_abs(var_store: &VarStore) -> Def {
}
/// Int.div : Int, Int -> Result Int [ DivByZero ]*
fn int_div(var_store: &VarStore) -> Def {
fn int_div(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -408,7 +407,7 @@ fn int_div(var_store: &VarStore) -> Def {
}
/// List.first : List elem -> Result elem [ ListWasEmpty ]*
fn list_first(var_store: &VarStore) -> Def {
fn list_first(var_store: &VarStore) -> Expr {
use crate::expr::Expr::*;
defn(
@ -500,7 +499,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) -> Def {
fn defn(fn_name: Symbol, args: Vec<Symbol>, var_store: &VarStore, body: Expr) -> Expr {
use crate::expr::Expr::*;
use crate::pattern::Pattern::*;
@ -509,19 +508,11 @@ fn defn(fn_name: Symbol, args: Vec<Symbol>, var_store: &VarStore, body: Expr) ->
.map(|symbol| (var_store.fresh(), no_region(Identifier(symbol))))
.collect();
let expr = Closure(
Closure(
var_store.fresh(),
fn_name,
Recursive::NotRecursive,
closure_args,
Box::new((no_region(body), var_store.fresh())),
);
Def {
loc_pattern: no_region(Identifier(fn_name)),
loc_expr: no_region(expr),
expr_var: var_store.fresh(),
pattern_vars: SendMap::default(),
annotation: None,
}
)
}