language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-28 06:14:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Int.isOdd and Int.isEven functions in code gen

Browse source
This commit is contained in:
Chad Stearns 2020-05-13 18:57:15 -04:00
parent 7df4771f7b
commit c8d62bd838
3 changed files with 64 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

48
compiler/can/src/builtins.rs
View file

@ -32,9 +32,57 @@ pub fn builtin_defs(var_store: &VarStore) -> Vec<Def> {
int_div(var_store),
int_abs(var_store),
int_rem(var_store),
int_is_odd(var_store),
int_is_even(var_store),
]
}
/// Int.isOdd : Int -> Bool
fn int_is_odd(var_store: &VarStore) -> Def {
use crate::expr::Expr::*;
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,
),
)
}
/// Int.isEven : Int -> Bool
fn int_is_even(var_store: &VarStore) -> Def {
use crate::expr::Expr::*;
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,
),
)
}
/// List.get : List elem, Int -> Result elem [ OutOfBounds ]*
fn list_get(var_store: &VarStore) -> Def {
use crate::expr::Expr::*;

12
compiler/gen/tests/gen_builtins.rs
View file

@ -257,6 +257,18 @@ mod gen_builtins {
);
}
#[test]
fn gen_is_odd() {
assert_evals_to!("Int.isOdd 4", false, bool);
assert_evals_to!("Int.isOdd 5", true, bool);
}
#[test]
fn gen_is_even() {
assert_evals_to!("Int.isEven 6", true, bool);
assert_evals_to!("Int.isEven 7", false, bool);
}
#[test]
fn lt_i64() {
assert_evals_to!("1 < 2", true, bool);

4
compiler/module/src/symbol.rs
View file

@ -615,6 +615,10 @@ define_builtins! {
25 INT_REM: "rem"
26 INT_REM_ARG_0: "rem#arg0"
27 INT_REM_ARG_1: "rem#arg1"
28 INT_IS_ODD: "isOdd"
29 INT_IS_ODD_ARG: "isOdd#arg"
30 INT_IS_EVEN: "isEven"
31 INT_IS_EVEN_ARG: "isEven#arg"
}
3 FLOAT: "Float" => {
0 FLOAT_FLOAT: "Float" imported // the Float.Float type alias