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

feat(library/init/lean/environment): lower level interface for C++

Browse source
This commit is contained in:
Leonardo de Moura 2019-05-11 15:50:02 -07:00
parent 77c6aa6693
commit 5f563973f3
2 changed files with 172 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
library/init/lean/environment.lean
View file

@ -229,16 +229,16 @@ constant registerPersistentEnvExtension {α σ : Type} (name : Name) (initState
def CPPExtensionState := NonScalar
@[export lean.register_extension_core]
def registerCPPExtension (initial : CPPExtensionState) : IO (EnvExtension CPPExtensionState) :=
registerEnvExtension initial
unsafe def registerCPPExtension (initial : CPPExtensionState) : Option Nat :=
unsafeIO (do ext ← registerEnvExtension initial, pure ext.idx)
@[export lean.set_extension_core]
def setCPPExtensionState (ext : EnvExtension CPPExtensionState) (env : Environment) (s : CPPExtensionState) : Environment :=
ext.setState env s
unsafe def setCPPExtensionState (env : Environment) (idx : Nat) (s : CPPExtensionState) : Option Environment :=
unsafeIO (do exts ← envExtensionsRef.get, pure $ (exts.get idx).setState env s)
@[export lean.get_extension_core]
def getCPPExtensionState (ext : EnvExtension CPPExtensionState) (env : Environment) : CPPExtensionState :=
ext.getState env
unsafe def getCPPExtensionState (env : Environment) (idx : Nat) : Option CPPExtensionState :=
unsafeIO (do exts ← envExtensionsRef.get, pure $ (exts.get idx).getState env)
/- Legacy support for Modification objects -/

180
src/stage0/init/lean/environment.cpp generated
View file

@ -46,6 +46,7 @@ obj* l_Lean_EnvExtension_modifyStateUnsafe(obj*);
obj* l___private_init_lean_environment_4__getTrustLevel___boxed(obj*);
obj* l_Lean_SMap_find___main___at_Lean_Environment_find___spec__1___boxed(obj*, obj*);
uint8 l_AssocList_contains___main___at_Lean_Environment_add___spec__5(obj*, obj*);
extern "C" obj* lean_io_unsafe(obj*, obj*);
obj* l_Lean_SMap_insert___main___at_Lean_Environment_add___spec__1___closed__1;
obj* l_Lean_SMap_insert___main___at_Lean_Environment_add___spec__1(obj*, obj*, obj*);
obj* l_HashMapImp_moveEntries___main___at_Lean_Environment_add___spec__7(obj*, obj*, obj*);
@ -63,6 +64,7 @@ obj* l_Lean_Name_quickLt___boxed(obj*, obj*);
obj* l_Lean_ModuleId_Inhabited;
obj* l_Lean_EnvExtension_setState___boxed(obj*, obj*, obj*, obj*);
extern "C" obj* lean_io_initializing(obj*);
obj* l_IO_Prim_Ref_get___boxed(obj*, obj*, obj*);
obj* l_Lean_PersistentEnvExtension_inhabited___lambda__2(obj*);
obj* l_RBNode_ins___main___at_Lean_Environment_add___spec__3(obj*, obj*, obj*);
obj* l_Lean_EnvExtension_setState___closed__1;
@ -71,6 +73,7 @@ obj* mk_empty_environment_core(uint32, obj*);
}
extern obj* l_Lean_Name_DecidableEq;
obj* l_mkHashMap___at_Lean_Environment_Inhabited___spec__2(obj*);
obj* l_Lean_setCPPExtensionState___lambda__1(obj*, obj*, obj*, obj*, obj*);
obj* l_Array_miterateAux___main___at_Lean_mkEmptyEnvironment___spec__1___boxed(obj*, obj*, obj*, obj*);
obj* l_Lean_EnvExtensionEntry_Inhabited;
obj* l_RBNode_find___main___at_Lean_Environment_find___spec__2___boxed(obj*, obj*);
@ -84,14 +87,17 @@ obj* l_Lean_registerEnvExtensionUnsafe(obj*);
obj* l_beqOfEq___rarg(obj*, obj*, obj*);
obj* l_Lean_registerPersistentEnvExtension___rarg(obj*);
obj* l_Lean_Environment_Inhabited;
obj* l_Lean_setCPPExtensionState___lambda__1___boxed(obj*, obj*, obj*, obj*, obj*);
namespace lean {
obj* string_append(obj*, obj*);
}
obj* l_Lean_PersistentEnvExtension_getModuleEntries(obj*, obj*);
obj* l_Lean_PersistentEnvExtension_getEntries(obj*, obj*);
obj* l_Lean_getCPPExtensionState___lambda__1___boxed(obj*, obj*, obj*, obj*);
obj* l_Lean_PersistentEnvExtensionState_inhabited;
obj* l___private_init_lean_environment_6__envExtensionsRef;
obj* l_Lean_EnvExtension_getState___boxed(obj*);
obj* l_Lean_registerCPPExtension___lambda__1(obj*, obj*);
obj* l_Lean_EnvExtension_getState___rarg(obj*, obj*);
obj* l_Lean_PersistentEnvExtension_forceStateAux(obj*, obj*);
namespace lean {
@ -111,6 +117,7 @@ uint8 nat_dec_eq(obj*, obj*);
}
obj* l_Lean_EnvExtension_setStateUnsafe___rarg(obj*, obj*, obj*);
uint8 l_RBNode_isRed___main___rarg(obj*);
obj* l_Lean_setCPPExtensionState___closed__1;
namespace lean {
obj* set_extension_core(obj*, obj*, obj*);
}
@ -127,6 +134,7 @@ obj* l_Lean_mkModuleData(obj*, obj*);
obj* l_Lean_PersistentEnvExtension_getState(obj*, obj*);
obj* l_Array_anyMAux___main___at_Lean_registerPersistentEnvExtensionUnsafe___spec__1___boxed(obj*, obj*, obj*);
obj* l_Lean_EnvExtension_modifyState(obj*, obj*, obj*, obj*);
obj* l_Lean_registerCPPExtension___closed__1;
obj* l_Nat_foldAux___main___at_Lean_mkModuleData___spec__1___boxed(obj*, obj*, obj*, obj*, obj*);
obj* l_Lean_mkEmptyEnvironment___closed__2;
obj* l_Lean_Modification_Inhabited;
@ -138,9 +146,11 @@ extern obj* l_NonScalar_Inhabited;
obj* l_List_foldl___main___rarg(obj*, obj*, obj*);
obj* l___private_init_lean_environment_3__isQuotInit___boxed(obj*);
obj* l_Lean_PersistentEnvExtension_addEntry___boxed(obj*, obj*);
obj* l_EState_bind___rarg(obj*, obj*, obj*);
obj* l_Lean_PersistentEnvExtension_getState___boxed(obj*, obj*);
obj* l_Lean_SMap_insert___main___at_Lean_Environment_add___spec__1___closed__2;
obj* l_RBNode_fold___main___at_Lean_mkModuleData___spec__2(obj*, obj*);
obj* l_Lean_getCPPExtensionState___lambda__1(obj*, obj*, obj*, obj*);
obj* l_Lean_EnvExtension_setStateUnsafe___boxed(obj*);
obj* l_RBNode_insert___at_Lean_Environment_add___spec__2(obj*, obj*, obj*);
obj* l_Lean_PersistentEnvExtension_addEntry___rarg(obj*, obj*, obj*);
@ -149,7 +159,7 @@ obj* environment_switch_core(obj*);
}
uint8 l_Lean_Name_quickLt(obj*, obj*);
namespace lean {
obj* register_extension_core(obj*, obj*);
obj* register_extension_core(obj*);
}
namespace lean {
usize usize_modn(usize, obj*);
@ -4029,38 +4039,176 @@ x_0 = l_NonScalar_Inhabited;
return x_0;
}
}
namespace lean {
obj* register_extension_core(obj* x_0, obj* x_1) {
obj* l_Lean_registerCPPExtension___lambda__1(obj* x_0, obj* x_1) {
_start:
{
obj* x_2;
x_2 = l_Lean_registerEnvExtensionUnsafe___rarg(x_0, x_1);
return x_2;
obj* x_2; obj* x_5; obj* x_7; obj* x_8;
x_2 = lean::cnstr_get(x_0, 0);
lean::inc(x_2);
lean::dec(x_0);
x_5 = lean::cnstr_get(x_1, 1);
if (lean::is_exclusive(x_1)) {
lean::cnstr_release(x_1, 0);
x_7 = x_1;
} else {
lean::inc(x_5);
lean::dec(x_1);
x_7 = lean::box(0);
}
if (lean::is_scalar(x_7)) {
x_8 = lean::alloc_cnstr(0, 2, 0);
} else {
x_8 = x_7;
}
lean::cnstr_set(x_8, 0, x_2);
lean::cnstr_set(x_8, 1, x_5);
return x_8;
}
}
obj* _init_l_Lean_registerCPPExtension___closed__1() {
_start:
{
obj* x_0;
x_0 = lean::alloc_closure(reinterpret_cast<void*>(l_Lean_registerCPPExtension___lambda__1), 2, 0);
return x_0;
}
}
namespace lean {
obj* register_extension_core(obj* x_0) {
_start:
{
obj* x_1; obj* x_2; obj* x_3; obj* x_4;
x_1 = lean::alloc_closure(reinterpret_cast<void*>(l_Lean_registerEnvExtensionUnsafe___rarg), 2, 1);
lean::closure_set(x_1, 0, x_0);
x_2 = l_Lean_registerCPPExtension___closed__1;
x_3 = lean::alloc_closure(reinterpret_cast<void*>(l_EState_bind___rarg), 3, 2);
lean::closure_set(x_3, 0, x_1);
lean::closure_set(x_3, 1, x_2);
x_4 = lean_io_unsafe(lean::box(0), x_3);
return x_4;
}
}
}
obj* l_Lean_setCPPExtensionState___lambda__1(obj* x_0, obj* x_1, obj* x_2, obj* x_3, obj* x_4) {
_start:
{
obj* x_5; obj* x_7; obj* x_8; obj* x_9; obj* x_10; obj* x_12;
x_5 = lean::cnstr_get(x_4, 1);
if (lean::is_exclusive(x_4)) {
lean::cnstr_release(x_4, 0);
x_7 = x_4;
} else {
lean::inc(x_5);
lean::dec(x_4);
x_7 = lean::box(0);
}
x_8 = l_Lean_registerEnvExtensionUnsafe___rarg___closed__2;
x_9 = lean::array_get(x_8, x_3, x_0);
x_10 = l_Lean_EnvExtension_setStateUnsafe___rarg(x_9, x_1, x_2);
lean::dec(x_9);
if (lean::is_scalar(x_7)) {
x_12 = lean::alloc_cnstr(0, 2, 0);
} else {
x_12 = x_7;
}
lean::cnstr_set(x_12, 0, x_10);
lean::cnstr_set(x_12, 1, x_5);
return x_12;
}
}
obj* _init_l_Lean_setCPPExtensionState___closed__1() {
_start:
{
obj* x_0; obj* x_1;
x_0 = l___private_init_lean_environment_6__envExtensionsRef;
x_1 = lean::alloc_closure(reinterpret_cast<void*>(l_IO_Prim_Ref_get___boxed), 3, 2);
lean::closure_set(x_1, 0, lean::box(0));
lean::closure_set(x_1, 1, x_0);
return x_1;
}
}
namespace lean {
obj* set_extension_core(obj* x_0, obj* x_1, obj* x_2) {
_start:
{
obj* x_3;
x_3 = l_Lean_EnvExtension_setStateUnsafe___rarg(x_0, x_1, x_2);
lean::dec(x_0);
return x_3;
obj* x_3; obj* x_4; obj* x_5; obj* x_6;
x_3 = lean::alloc_closure(reinterpret_cast<void*>(l_Lean_setCPPExtensionState___lambda__1___boxed), 5, 3);
lean::closure_set(x_3, 0, x_1);
lean::closure_set(x_3, 1, x_0);
lean::closure_set(x_3, 2, x_2);
x_4 = l_Lean_setCPPExtensionState___closed__1;
x_5 = lean::alloc_closure(reinterpret_cast<void*>(l_EState_bind___rarg), 3, 2);
lean::closure_set(x_5, 0, x_4);
lean::closure_set(x_5, 1, x_3);
x_6 = lean_io_unsafe(lean::box(0), x_5);
return x_6;
}
}
}
obj* l_Lean_setCPPExtensionState___lambda__1___boxed(obj* x_0, obj* x_1, obj* x_2, obj* x_3, obj* x_4) {
_start:
{
obj* x_5;
x_5 = l_Lean_setCPPExtensionState___lambda__1(x_0, x_1, x_2, x_3, x_4);
lean::dec(x_0);
lean::dec(x_3);
return x_5;
}
}
obj* l_Lean_getCPPExtensionState___lambda__1(obj* x_0, obj* x_1, obj* x_2, obj* x_3) {
_start:
{
obj* x_4; obj* x_6; obj* x_7; obj* x_8; obj* x_9; obj* x_10;
x_4 = lean::cnstr_get(x_3, 1);
if (lean::is_exclusive(x_3)) {
lean::cnstr_release(x_3, 0);
x_6 = x_3;
} else {
lean::inc(x_4);
lean::dec(x_3);
x_6 = lean::box(0);
}
x_7 = l_Lean_registerEnvExtensionUnsafe___rarg___closed__2;
x_8 = lean::array_get(x_7, x_2, x_0);
x_9 = l_Lean_EnvExtension_getStateUnsafe___rarg(x_8, x_1);
if (lean::is_scalar(x_6)) {
x_10 = lean::alloc_cnstr(0, 2, 0);
} else {
x_10 = x_6;
}
lean::cnstr_set(x_10, 0, x_9);
lean::cnstr_set(x_10, 1, x_4);
return x_10;
}
}
namespace lean {
obj* get_extension_core(obj* x_0, obj* x_1) {
_start:
{
obj* x_2;
x_2 = l_Lean_EnvExtension_getStateUnsafe___rarg(x_0, x_1);
lean::dec(x_1);
return x_2;
obj* x_2; obj* x_3; obj* x_4; obj* x_5;
x_2 = lean::alloc_closure(reinterpret_cast<void*>(l_Lean_getCPPExtensionState___lambda__1___boxed), 4, 2);
lean::closure_set(x_2, 0, x_1);
lean::closure_set(x_2, 1, x_0);
x_3 = l_Lean_setCPPExtensionState___closed__1;
x_4 = lean::alloc_closure(reinterpret_cast<void*>(l_EState_bind___rarg), 3, 2);
lean::closure_set(x_4, 0, x_3);
lean::closure_set(x_4, 1, x_2);
x_5 = lean_io_unsafe(lean::box(0), x_4);
return x_5;
}
}
}
obj* l_Lean_getCPPExtensionState___lambda__1___boxed(obj* x_0, obj* x_1, obj* x_2, obj* x_3) {
_start:
{
obj* x_4;
x_4 = l_Lean_getCPPExtensionState___lambda__1(x_0, x_1, x_2, x_3);
lean::dec(x_0);
lean::dec(x_1);
lean::dec(x_2);
return x_4;
}
}
obj* _init_l_Lean_Modification_Inhabited() {
_start:
{
@ -4376,6 +4524,10 @@ lean::mark_persistent(l_Lean_registerPersistentEnvExtensionUnsafe___rarg___close
lean::mark_persistent(l_Lean_registerPersistentEnvExtensionUnsafe___rarg___closed__2);
l_Lean_CPPExtensionState_Inhabited = _init_l_Lean_CPPExtensionState_Inhabited();
lean::mark_persistent(l_Lean_CPPExtensionState_Inhabited);
l_Lean_registerCPPExtension___closed__1 = _init_l_Lean_registerCPPExtension___closed__1();
lean::mark_persistent(l_Lean_registerCPPExtension___closed__1);
l_Lean_setCPPExtensionState___closed__1 = _init_l_Lean_setCPPExtensionState___closed__1();
lean::mark_persistent(l_Lean_setCPPExtensionState___closed__1);
l_Lean_Modification_Inhabited = _init_l_Lean_Modification_Inhabited();
lean::mark_persistent(l_Lean_Modification_Inhabited);
w = l_Lean_regModListExtension(w);