max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

feat(library/init/meta/fun_info): add fold_explicit_args

Browse source
This commit is contained in:
Leonardo de Moura 2017-02-17 10:21:06 -08:00
parent d402b2a467
commit e16c3a0bee
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
library/init/meta/fun_info.lean
View file

@ -109,4 +109,21 @@ meta constant get_subsingleton_info (f : expr) (nargs : option nat := none) (md
meta constant get_spec_subsingleton_info (t : expr) (md := semireducible) : tactic (list subsingleton_info)
meta constant get_spec_prefix_size (t : expr) (nargs : nat) (md := semireducible) : tactic nat
private meta def is_next_explicit : list param_info → bool
| [] := tt
| (p::ps) := bnot p^.is_implicit && bnot p^.is_inst_implicit
meta def fold_explicit_args_aux {α} (f : α → expr → tactic α) : list expr → list param_info → α → tactic α
| [] _ a := return a
| (e::es) ps a :=
if is_next_explicit ps
then f a e >>= fold_explicit_args_aux es ps^.tail
else fold_explicit_args_aux es ps^.tail a
meta def fold_explicit_args {α} (e : expr) (a : α) (f : α → expr → tactic α) : tactic α :=
if e^.is_app then do
info ← get_fun_info e^.get_app_fn (some e^.get_app_num_args),
fold_explicit_args_aux f e^.get_app_args info^.params a
else return a
end tactic