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

perf(library/vm/vm_array): minor optimization

Browse source
This commit is contained in:
Leonardo de Moura 2017-02-21 15:29:55 -08:00
parent 3f87fd15eb
commit d9dcb4461e
1 changed files with 13 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/library/vm/vm_array.cpp
View file

@ -74,37 +74,37 @@ vm_obj array_read(vm_obj const &, vm_obj const &, vm_obj const & a, vm_obj const
vm_obj array_write(vm_obj const &, vm_obj const &, vm_obj const & a, vm_obj const & i, vm_obj const & v) {
/* TODO(Leo): handle case where n is too big */
unsigned idx = force_to_unsigned(i);
lean_vm_check(idx < to_array(a).size());
parray<vm_obj> const & p = to_array(a);
lean_vm_check(idx < p.size());
if (a.raw()->get_rc() == 1) {
parray<vm_obj> & new_a = const_cast<parray<vm_obj> &>(to_array(a));
new_a.set(idx, v);
return to_obj(new_a);
const_cast<parray<vm_obj> &>(p).set(idx, v);
return a;
} else {
parray<vm_obj> new_a = to_array(a);
parray<vm_obj> new_a = p;
new_a.set(idx, v);
return to_obj(new_a);
}
}
vm_obj array_push_back(vm_obj const &, vm_obj const &, vm_obj const & a, vm_obj const & v) {
parray<vm_obj> const & p = to_array(a);
if (a.raw()->get_rc() == 1) {
parray<vm_obj> & new_a = const_cast<parray<vm_obj> &>(to_array(a));
new_a.push_back(v);
return to_obj(new_a);
const_cast<parray<vm_obj> &>(p).push_back(v);
return a;
} else {
parray<vm_obj> new_a = to_array(a);
parray<vm_obj> new_a = p;
new_a.push_back(v);
return to_obj(new_a);
}
}
vm_obj array_pop_back(vm_obj const &, vm_obj const &, vm_obj const & a) {
parray<vm_obj> const & p = to_array(a);
if (a.raw()->get_rc() == 1) {
parray<vm_obj> & new_a = const_cast<parray<vm_obj> &>(to_array(a));
new_a.pop_back();
return to_obj(new_a);
const_cast<parray<vm_obj> &>(p).pop_back();
return a;
} else {
parray<vm_obj> new_a = to_array(a);
parray<vm_obj> new_a = p;
new_a.pop_back();
return to_obj(new_a);
}