language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-03 00:24:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Annotate low-level usages

Browse source
This commit is contained in:
Richard Feldman 2020-06-21 21:05:57 -04:00
parent 52dfb75dc0
commit cad0a1421e
3 changed files with 75 additions and 89 deletions
Download patch file Download diff file Expand all files Collapse all files

133
compiler/uniq/src/sharing.rs
View file

@ -1,8 +1,8 @@
use roc_can::expr::Expr;
use roc_collections::all::{ImMap, ImSet};
use roc_module::ident::Lowercase;
use roc_module::low_level::LowLevel;
use roc_module::symbol::Symbol;
use roc_region::all::Located;
use roc_types::subs::Variable;
// fake field names for container elements
@ -618,25 +618,27 @@ pub fn annotate_usage(expr: &Expr, usage: &mut VarUsage) {
annotate_usage(&loc_expr.value, usage);
}
Call(fun, loc_args, _) => {
if let Var(symbol) = fun.1.value {
// call by name
special_case_builtins(usage, symbol, loc_args);
} else {
// unknown call
annotate_usage(&fun.1.value, usage);
match fun.1.value {
Var(symbol) => {
// call by name
usage.register_unique(symbol);
for (_, arg) in loc_args {
annotate_usage(&arg.value, usage);
for (_, arg) in loc_args {
annotate_usage(&arg.value, usage);
}
}
_ => {
// unknown call
annotate_usage(&fun.1.value, usage);
for (_, arg) in loc_args {
annotate_usage(&arg.value, usage);
}
}
}
}
RunLowLevel { op, args, ret_var } => {
todo!(
"TODO implement UNIQ RunLowLevel for {:?}({:?}) -> {:?}",
op,
args,
ret_var
);
annotate_low_level_usage(usage, *op, args, *ret_var);
}
Closure(_, _, _, _, body) => {
annotate_usage(&body.0.value, usage);
@ -701,75 +703,62 @@ fn get_access_chain<'a>(expr: &'a Expr, chain: &mut Vec<Lowercase>) -> Option<&'
}
}
fn special_case_builtins(
fn annotate_low_level_usage(
usage: &mut VarUsage,
symbol: Symbol,
loc_args: &[(Variable, Located<Expr>)],
op: LowLevel,
args: &[(Variable, Expr)],
_ret_var: Variable,
) {
use Expr::Var;
use LowLevel::*;
use Mark::*;
use Usage::*;
match symbol {
Symbol::LIST_GET => {
debug_assert!(loc_args.len() == 2);
let loc_list = &loc_args[0].1;
let loc_index = &loc_args[1].1;
if let Var(list_var) = loc_list.value {
usage.register_with(
list_var,
&Access(Container::List, Seen, FieldAccess::list_access()),
);
} else {
annotate_usage(&loc_list.value, usage);
match op {
ListLen | ListIsEmpty | ListGetUnsafe => {
match &args[0].1 {
Var(list_var) => {
usage.register_with(
*list_var,
&Access(Container::List, Seen, FieldAccess::list_seen()),
);
}
list => {
annotate_usage(list, usage);
}
}
annotate_usage(&loc_index.value, usage);
}
Symbol::LIST_SET => {
debug_assert!(loc_args.len() == 3);
let loc_list = &loc_args[0].1;
let loc_index = &loc_args[1].1;
let loc_value = &loc_args[2].1;
if let Var(list_var) = loc_list.value {
usage.register_with(
list_var,
&Update(
Container::List,
ImSet::default(),
FieldAccess::list_update(),
),
);
} else {
annotate_usage(&loc_list.value, usage);
}
annotate_usage(&loc_index.value, usage);
annotate_usage(&loc_value.value, usage);
}
Symbol::LIST_IS_EMPTY | Symbol::LIST_LEN => {
debug_assert!(loc_args.len() == 1);
let loc_list = &loc_args[0].1;
if let Var(list_var) = loc_list.value {
usage.register_with(
list_var,
&Access(Container::List, Seen, FieldAccess::list_seen()),
);
} else {
annotate_usage(&loc_list.value, usage);
for (_, arg) in &args[1..] {
annotate_usage(arg, usage);
}
}
_ => {
usage.register_unique(symbol);
ListSetUnsafe => {
match &args[0].1 {
Var(list_var) => {
usage.register_with(
*list_var,
&Update(
Container::List,
ImSet::default(),
FieldAccess::list_update(),
),
);
}
list => {
annotate_usage(list, usage);
}
}
for (_, arg) in loc_args {
annotate_usage(&arg.value, usage);
for (_, arg) in &args[1..] {
annotate_usage(arg, usage);
}
}
NumAdd | NumSub | NumMul | NumGt | NumGte | NumLt | NumLte | Eq | NotEq | And | Or
| Not => {
for (_, arg) in args {
annotate_usage(&arg, usage);
}
}
}