diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/lean/betree/Base/Primitives.lean (renamed from tests/lean/misc/constants/Base/Primitives.lean) | 0 | ||||
-rw-r--r-- | tests/lean/betree/BetreeMain/Clauses/Template.lean | 185 | ||||
-rw-r--r-- | tests/lean/betree/BetreeMain/Funs.lean | 1222 | ||||
-rw-r--r-- | tests/lean/betree/BetreeMain/Opaque.lean | 33 | ||||
-rw-r--r-- | tests/lean/betree/BetreeMain/Types.lean | 61 | ||||
-rw-r--r-- | tests/lean/betree/lakefile.lean | 18 | ||||
-rw-r--r-- | tests/lean/misc-constants/Base/Primitives.lean (renamed from tests/lean/misc/external/Base/Primitives.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-constants/Constants.lean (renamed from tests/lean/misc/constants/Constants.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-constants/lakefile.lean (renamed from tests/lean/misc/constants/lakefile.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-external/Base/Primitives.lean (renamed from tests/lean/misc/loops/Base/Primitives.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-external/External/Funs.lean (renamed from tests/lean/misc/external/External/Funs.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-external/External/Opaque.lean (renamed from tests/lean/misc/external/External/Opaque.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-external/External/Types.lean (renamed from tests/lean/misc/external/External/Types.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-external/lakefile.lean (renamed from tests/lean/misc/external/lakefile.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-loops/Base/Primitives.lean (renamed from tests/lean/misc/no_nested_borrows/Base/Primitives.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-loops/Loops/Clauses/Clauses.lean (renamed from tests/lean/misc/loops/Loops/Clauses/Clauses.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-loops/Loops/Clauses/Template.lean (renamed from tests/lean/misc/loops/Loops/Clauses/Template.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-loops/Loops/Funs.lean (renamed from tests/lean/misc/loops/Loops/Funs.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-loops/Loops/Types.lean (renamed from tests/lean/misc/loops/Loops/Types.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-loops/lakefile.lean (renamed from tests/lean/misc/loops/lakefile.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-no_nested_borrows/Base/Primitives.lean (renamed from tests/lean/misc/paper/Base/Primitives.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-no_nested_borrows/NoNestedBorrows.lean (renamed from tests/lean/misc/no_nested_borrows/NoNestedBorrows.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-no_nested_borrows/lakefile.lean (renamed from tests/lean/misc/no_nested_borrows/lakefile.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-paper/Base/Primitives.lean | 392 | ||||
-rw-r--r-- | tests/lean/misc-paper/Paper.lean (renamed from tests/lean/misc/paper/Paper.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-paper/lakefile.lean (renamed from tests/lean/misc/paper/lakefile.lean) | 0 | ||||
-rw-r--r-- | tests/lean/misc-polonius_list/Base/Primitives.lean | 392 | ||||
-rw-r--r-- | tests/lean/misc-polonius_list/PoloniusList.lean | 36 | ||||
-rw-r--r-- | tests/lean/misc-polonius_list/lakefile.lean | 18 |
29 files changed, 2357 insertions, 0 deletions
diff --git a/tests/lean/misc/constants/Base/Primitives.lean b/tests/lean/betree/Base/Primitives.lean index 5b64e908..5b64e908 100644 --- a/tests/lean/misc/constants/Base/Primitives.lean +++ b/tests/lean/betree/Base/Primitives.lean diff --git a/tests/lean/betree/BetreeMain/Clauses/Template.lean b/tests/lean/betree/BetreeMain/Clauses/Template.lean new file mode 100644 index 00000000..1d18174e --- /dev/null +++ b/tests/lean/betree/BetreeMain/Clauses/Template.lean @@ -0,0 +1,185 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree_main]: templates for the decreases clauses +import Base.Primitives +import BetreeMain.Types + +/- [betree_main::betree::List::{1}::len]: termination measure -/ +@[simp] +def betree_list_len_terminates (T : Type) (self : betree_list_t T) := self + +/- [betree_main::betree::List::{1}::len]: decreases_by tactic -/ +syntax "betree_list_len_decreases" term+ : tactic +macro_rules +| `(tactic| betree_list_len_decreases $self) =>`(tactic| sorry) + +/- [betree_main::betree::List::{1}::split_at]: termination measure -/ +@[simp] +def betree_list_split_at_terminates (T : Type) (self : betree_list_t T) + (n : UInt64) := + (self, n) + +/- [betree_main::betree::List::{1}::split_at]: decreases_by tactic -/ +syntax "betree_list_split_at_decreases" term+ : tactic +macro_rules +| `(tactic| betree_list_split_at_decreases $self $n) =>`(tactic| sorry) + +/- [betree_main::betree::List::{2}::partition_at_pivot]: termination measure -/ +@[simp] +def betree_list_partition_at_pivot_terminates (T : Type) + (self : betree_list_t (UInt64 × T)) (pivot : UInt64) := + (self, pivot) + +/- [betree_main::betree::List::{2}::partition_at_pivot]: decreases_by tactic -/ +syntax "betree_list_partition_at_pivot_decreases" term+ : tactic +macro_rules +| `(tactic| betree_list_partition_at_pivot_decreases $self $pivot) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::lookup_in_bindings]: termination measure -/ +@[simp] +def betree_node_lookup_in_bindings_terminates (key : UInt64) + (bindings : betree_list_t (UInt64 × UInt64)) := + (key, bindings) + +/- [betree_main::betree::Node::{5}::lookup_in_bindings]: decreases_by tactic -/ +syntax "betree_node_lookup_in_bindings_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_lookup_in_bindings_decreases $key $bindings) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::lookup_first_message_for_key]: termination measure -/ +@[simp] +def betree_node_lookup_first_message_for_key_terminates (key : UInt64) + (msgs : betree_list_t (UInt64 × betree_message_t)) := + (key, msgs) + +/- [betree_main::betree::Node::{5}::lookup_first_message_for_key]: decreases_by tactic -/ +syntax "betree_node_lookup_first_message_for_key_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_lookup_first_message_for_key_decreases $key $msgs) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::apply_upserts]: termination measure -/ +@[simp] +def betree_node_apply_upserts_terminates + (msgs : betree_list_t (UInt64 × betree_message_t)) (prev : Option UInt64) + (key : UInt64) (st : State) := + (msgs, prev, key, st) + +/- [betree_main::betree::Node::{5}::apply_upserts]: decreases_by tactic -/ +syntax "betree_node_apply_upserts_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_apply_upserts_decreases $msgs $prev $key $st) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::lookup]: termination measure -/ +@[simp] +def betree_node_lookup_terminates (self : betree_node_t) (key : UInt64) + (st : State) := + (self, key, st) + +/- [betree_main::betree::Node::{5}::lookup]: decreases_by tactic -/ +syntax "betree_node_lookup_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_lookup_decreases $self $key $st) =>`(tactic| sorry) + +/- [betree_main::betree::Internal::{4}::lookup_in_children]: termination measure -/ +@[simp] +def betree_internal_lookup_in_children_terminates (self : betree_internal_t) + (key : UInt64) (st : State) := + (self, key, st) + +/- [betree_main::betree::Internal::{4}::lookup_in_children]: decreases_by tactic -/ +syntax "betree_internal_lookup_in_children_decreases" term+ : tactic +macro_rules +| `(tactic| betree_internal_lookup_in_children_decreases $self $key $st) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::lookup_mut_in_bindings]: termination measure -/ +@[simp] +def betree_node_lookup_mut_in_bindings_terminates (key : UInt64) + (bindings : betree_list_t (UInt64 × UInt64)) := + (key, bindings) + +/- [betree_main::betree::Node::{5}::lookup_mut_in_bindings]: decreases_by tactic -/ +syntax "betree_node_lookup_mut_in_bindings_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_lookup_mut_in_bindings_decreases $key $bindings) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::apply_messages_to_leaf]: termination measure -/ +@[simp] +def betree_node_apply_messages_to_leaf_terminates + (bindings : betree_list_t (UInt64 × UInt64)) + (new_msgs : betree_list_t (UInt64 × betree_message_t)) := + (bindings, new_msgs) + +/- [betree_main::betree::Node::{5}::apply_messages_to_leaf]: decreases_by tactic -/ +syntax "betree_node_apply_messages_to_leaf_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_apply_messages_to_leaf_decreases $bindings +$new_msgs) =>`(tactic| sorry) + +/- [betree_main::betree::Node::{5}::filter_messages_for_key]: termination measure -/ +@[simp] +def betree_node_filter_messages_for_key_terminates (key : UInt64) + (msgs : betree_list_t (UInt64 × betree_message_t)) := + (key, msgs) + +/- [betree_main::betree::Node::{5}::filter_messages_for_key]: decreases_by tactic -/ +syntax "betree_node_filter_messages_for_key_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_filter_messages_for_key_decreases $key $msgs) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::lookup_first_message_after_key]: termination measure -/ +@[simp] +def betree_node_lookup_first_message_after_key_terminates (key : UInt64) + (msgs : betree_list_t (UInt64 × betree_message_t)) := + (key, msgs) + +/- [betree_main::betree::Node::{5}::lookup_first_message_after_key]: decreases_by tactic -/ +syntax "betree_node_lookup_first_message_after_key_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_lookup_first_message_after_key_decreases $key $msgs) => + `(tactic| sorry) + +/- [betree_main::betree::Node::{5}::apply_messages_to_internal]: termination measure -/ +@[simp] +def betree_node_apply_messages_to_internal_terminates + (msgs : betree_list_t (UInt64 × betree_message_t)) + (new_msgs : betree_list_t (UInt64 × betree_message_t)) := + (msgs, new_msgs) + +/- [betree_main::betree::Node::{5}::apply_messages_to_internal]: decreases_by tactic -/ +syntax "betree_node_apply_messages_to_internal_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_apply_messages_to_internal_decreases $msgs +$new_msgs) =>`(tactic| sorry) + +/- [betree_main::betree::Node::{5}::apply_messages]: termination measure -/ +@[simp] +def betree_node_apply_messages_terminates (self : betree_node_t) + (params : betree_params_t) (node_id_cnt : betree_node_id_counter_t) + (msgs : betree_list_t (UInt64 × betree_message_t)) (st : State) := + (self, params, node_id_cnt, msgs, st) + +/- [betree_main::betree::Node::{5}::apply_messages]: decreases_by tactic -/ +syntax "betree_node_apply_messages_decreases" term+ : tactic +macro_rules +| `(tactic| betree_node_apply_messages_decreases $self $params $node_id_cnt +$msgs $st) =>`(tactic| sorry) + +/- [betree_main::betree::Internal::{4}::flush]: termination measure -/ +@[simp] +def betree_internal_flush_terminates (self : betree_internal_t) + (params : betree_params_t) (node_id_cnt : betree_node_id_counter_t) + (content : betree_list_t (UInt64 × betree_message_t)) (st : State) := + (self, params, node_id_cnt, content, st) + +/- [betree_main::betree::Internal::{4}::flush]: decreases_by tactic -/ +syntax "betree_internal_flush_decreases" term+ : tactic +macro_rules +| `(tactic| betree_internal_flush_decreases $self $params $node_id_cnt $content +$st) =>`(tactic| sorry) + diff --git a/tests/lean/betree/BetreeMain/Funs.lean b/tests/lean/betree/BetreeMain/Funs.lean new file mode 100644 index 00000000..e40ca4ca --- /dev/null +++ b/tests/lean/betree/BetreeMain/Funs.lean @@ -0,0 +1,1222 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree_main]: function definitions +import Base.Primitives +import BetreeMain.Types +import BetreeMain.Opaque +import BetreeMain.Clauses.Clauses + +section variable (opaque_defs: OpaqueDefs) + +/- [betree_main::betree::load_internal_node] -/ +def betree_load_internal_node_fwd + (id : UInt64) (st : State) : + Result (State × (betree_list_t (UInt64 × betree_message_t))) + := + opaque_defs.betree_utils_load_internal_node_fwd id st + +/- [betree_main::betree::store_internal_node] -/ +def betree_store_internal_node_fwd + (id : UInt64) (content : betree_list_t (UInt64 × betree_message_t)) + (st : State) : + Result (State × Unit) + := + do + let (st0, _) ← + opaque_defs.betree_utils_store_internal_node_fwd id content st + Result.ret (st0, ()) + +/- [betree_main::betree::load_leaf_node] -/ +def betree_load_leaf_node_fwd + (id : UInt64) (st : State) : + Result (State × (betree_list_t (UInt64 × UInt64))) + := + opaque_defs.betree_utils_load_leaf_node_fwd id st + +/- [betree_main::betree::store_leaf_node] -/ +def betree_store_leaf_node_fwd + (id : UInt64) (content : betree_list_t (UInt64 × UInt64)) (st : State) : + Result (State × Unit) + := + do + let (st0, _) ← opaque_defs.betree_utils_store_leaf_node_fwd id content st + Result.ret (st0, ()) + +/- [betree_main::betree::fresh_node_id] -/ +def betree_fresh_node_id_fwd (counter : UInt64) : Result UInt64 := + do + let _ ← UInt64.checked_add counter (UInt64.ofNatCore 1 (by intlit)) + Result.ret counter + +/- [betree_main::betree::fresh_node_id] -/ +def betree_fresh_node_id_back (counter : UInt64) : Result UInt64 := + UInt64.checked_add counter (UInt64.ofNatCore 1 (by intlit)) + +/- [betree_main::betree::NodeIdCounter::{0}::new] -/ +def betree_node_id_counter_new_fwd : Result betree_node_id_counter_t := + Result.ret + { betree_node_id_counter_next_node_id := (UInt64.ofNatCore 0 (by intlit)) } + +/- [betree_main::betree::NodeIdCounter::{0}::fresh_id] -/ +def betree_node_id_counter_fresh_id_fwd + (self : betree_node_id_counter_t) : Result UInt64 := + do + let _ ← UInt64.checked_add self.betree_node_id_counter_next_node_id + (UInt64.ofNatCore 1 (by intlit)) + Result.ret self.betree_node_id_counter_next_node_id + +/- [betree_main::betree::NodeIdCounter::{0}::fresh_id] -/ +def betree_node_id_counter_fresh_id_back + (self : betree_node_id_counter_t) : Result betree_node_id_counter_t := + do + let i ← UInt64.checked_add self.betree_node_id_counter_next_node_id + (UInt64.ofNatCore 1 (by intlit)) + Result.ret { betree_node_id_counter_next_node_id := i } + +/- [core::num::u64::{10}::MAX] -/ +def core_num_u64_max_body : Result UInt64 := + Result.ret (UInt64.ofNatCore 18446744073709551615 (by intlit)) +def core_num_u64_max_c : UInt64 := eval_global core_num_u64_max_body (by simp) + +/- [betree_main::betree::upsert_update] -/ +def betree_upsert_update_fwd + (prev : Option UInt64) (st : betree_upsert_fun_state_t) : Result UInt64 := + match h: prev with + | Option.none => + match h: st with + | betree_upsert_fun_state_t.BetreeUpsertFunStateAdd v => Result.ret v + | betree_upsert_fun_state_t.BetreeUpsertFunStateSub i => + Result.ret (UInt64.ofNatCore 0 (by intlit)) + | Option.some prev0 => + match h: st with + | betree_upsert_fun_state_t.BetreeUpsertFunStateAdd v => + do + let margin ← UInt64.checked_sub core_num_u64_max_c prev0 + if h: margin >= v + then UInt64.checked_add prev0 v + else Result.ret core_num_u64_max_c + | betree_upsert_fun_state_t.BetreeUpsertFunStateSub v => + if h: prev0 >= v + then UInt64.checked_sub prev0 v + else Result.ret (UInt64.ofNatCore 0 (by intlit)) + +/- [betree_main::betree::List::{1}::len] -/ +def betree_list_len_fwd + (T : Type) (self : betree_list_t T) : (Result UInt64) := + match h: self with + | betree_list_t.BetreeListCons t tl => + do + let i ← betree_list_len_fwd T tl + UInt64.checked_add (UInt64.ofNatCore 1 (by intlit)) i + | betree_list_t.BetreeListNil => Result.ret (UInt64.ofNatCore 0 (by intlit)) +termination_by betree_list_len_fwd self => betree_list_len_terminates T self +decreasing_by betree_list_len_decreases self + +/- [betree_main::betree::List::{1}::split_at] -/ +def betree_list_split_at_fwd + (T : Type) (self : betree_list_t T) (n : UInt64) : + (Result ((betree_list_t T) × (betree_list_t T))) + := + if h: n = (UInt64.ofNatCore 0 (by intlit)) + then Result.ret (betree_list_t.BetreeListNil, self) + else + match h: self with + | betree_list_t.BetreeListCons hd tl => + do + let i ← UInt64.checked_sub n (UInt64.ofNatCore 1 (by intlit)) + let p ← betree_list_split_at_fwd T tl i + let (ls0, ls1) := p + let l := ls0 + Result.ret (betree_list_t.BetreeListCons hd l, ls1) + | betree_list_t.BetreeListNil => Result.fail Error.panic +termination_by betree_list_split_at_fwd self n => + betree_list_split_at_terminates T self n +decreasing_by betree_list_split_at_decreases self n + +/- [betree_main::betree::List::{1}::push_front] -/ +def betree_list_push_front_fwd_back + (T : Type) (self : betree_list_t T) (x : T) : Result (betree_list_t T) := + let tl := mem_replace_fwd (betree_list_t T) self betree_list_t.BetreeListNil + let l := tl + Result.ret (betree_list_t.BetreeListCons x l) + +/- [betree_main::betree::List::{1}::pop_front] -/ +def betree_list_pop_front_fwd (T : Type) (self : betree_list_t T) : Result T := + let ls := mem_replace_fwd (betree_list_t T) self betree_list_t.BetreeListNil + match h: ls with + | betree_list_t.BetreeListCons x tl => Result.ret x + | betree_list_t.BetreeListNil => Result.fail Error.panic + +/- [betree_main::betree::List::{1}::pop_front] -/ +def betree_list_pop_front_back + (T : Type) (self : betree_list_t T) : Result (betree_list_t T) := + let ls := mem_replace_fwd (betree_list_t T) self betree_list_t.BetreeListNil + match h: ls with + | betree_list_t.BetreeListCons x tl => Result.ret tl + | betree_list_t.BetreeListNil => Result.fail Error.panic + +/- [betree_main::betree::List::{1}::hd] -/ +def betree_list_hd_fwd (T : Type) (self : betree_list_t T) : Result T := + match h: self with + | betree_list_t.BetreeListCons hd l => Result.ret hd + | betree_list_t.BetreeListNil => Result.fail Error.panic + +/- [betree_main::betree::List::{2}::head_has_key] -/ +def betree_list_head_has_key_fwd + (T : Type) (self : betree_list_t (UInt64 × T)) (key : UInt64) : + Result Bool + := + match h: self with + | betree_list_t.BetreeListCons hd l => let (i, _) := hd + Result.ret (i = key) + | betree_list_t.BetreeListNil => Result.ret false + +/- [betree_main::betree::List::{2}::partition_at_pivot] -/ +def betree_list_partition_at_pivot_fwd + (T : Type) (self : betree_list_t (UInt64 × T)) (pivot : UInt64) : + (Result ((betree_list_t (UInt64 × T)) × (betree_list_t (UInt64 × T)))) + := + match h: self with + | betree_list_t.BetreeListCons hd tl => + let (i, t) := hd + if h: i >= pivot + then + Result.ret (betree_list_t.BetreeListNil, betree_list_t.BetreeListCons (i, + t) tl) + else + do + let p ← betree_list_partition_at_pivot_fwd T tl pivot + let (ls0, ls1) := p + let l := ls0 + Result.ret (betree_list_t.BetreeListCons (i, t) l, ls1) + | betree_list_t.BetreeListNil => + Result.ret (betree_list_t.BetreeListNil, betree_list_t.BetreeListNil) +termination_by betree_list_partition_at_pivot_fwd self pivot => + betree_list_partition_at_pivot_terminates T self pivot +decreasing_by betree_list_partition_at_pivot_decreases self pivot + +/- [betree_main::betree::Leaf::{3}::split] -/ +def betree_leaf_split_fwd + (self : betree_leaf_t) (content : betree_list_t (UInt64 × UInt64)) + (params : betree_params_t) (node_id_cnt : betree_node_id_counter_t) + (st : State) : + Result (State × betree_internal_t) + := + do + let p ← + betree_list_split_at_fwd (UInt64 × UInt64) content + params.betree_params_split_size + let (content0, content1) := p + let p0 ← betree_list_hd_fwd (UInt64 × UInt64) content1 + let (pivot, _) := p0 + let id0 ← betree_node_id_counter_fresh_id_fwd node_id_cnt + let node_id_cnt0 ← betree_node_id_counter_fresh_id_back node_id_cnt + let id1 ← betree_node_id_counter_fresh_id_fwd node_id_cnt0 + let (st0, _) ← betree_store_leaf_node_fwd id0 content0 st + let (st1, _) ← betree_store_leaf_node_fwd id1 content1 st0 + let n := betree_node_t.BetreeNodeLeaf + { + betree_leaf_id := id0, + betree_leaf_size := params.betree_params_split_size + } + let n0 := betree_node_t.BetreeNodeLeaf + { + betree_leaf_id := id1, + betree_leaf_size := params.betree_params_split_size + } + Result.ret (st1, + { + betree_internal_id := self.betree_leaf_id, + betree_internal_pivot := pivot, + betree_internal_left := n, + betree_internal_right := n0 + }) + +/- [betree_main::betree::Leaf::{3}::split] -/ +def betree_leaf_split_back + (self : betree_leaf_t) (content : betree_list_t (UInt64 × UInt64)) + (params : betree_params_t) (node_id_cnt : betree_node_id_counter_t) + (st : State) : + Result betree_node_id_counter_t + := + do + let p ← + betree_list_split_at_fwd (UInt64 × UInt64) content + params.betree_params_split_size + let (content0, content1) := p + let _ ← betree_list_hd_fwd (UInt64 × UInt64) content1 + let id0 ← betree_node_id_counter_fresh_id_fwd node_id_cnt + let node_id_cnt0 ← betree_node_id_counter_fresh_id_back node_id_cnt + let id1 ← betree_node_id_counter_fresh_id_fwd node_id_cnt0 + let (st0, _) ← betree_store_leaf_node_fwd id0 content0 st + let _ ← betree_store_leaf_node_fwd id1 content1 st0 + betree_node_id_counter_fresh_id_back node_id_cnt0 + +/- [betree_main::betree::Node::{5}::lookup_in_bindings] -/ +def betree_node_lookup_in_bindings_fwd + (key : UInt64) (bindings : betree_list_t (UInt64 × UInt64)) : + (Result (Option UInt64)) + := + match h: bindings with + | betree_list_t.BetreeListCons hd tl => + let (i, i0) := hd + if h: i = key + then Result.ret (Option.some i0) + else + if h: i > key + then Result.ret Option.none + else betree_node_lookup_in_bindings_fwd key tl + | betree_list_t.BetreeListNil => Result.ret Option.none +termination_by betree_node_lookup_in_bindings_fwd key bindings => + betree_node_lookup_in_bindings_terminates key bindings +decreasing_by betree_node_lookup_in_bindings_decreases key bindings + +/- [betree_main::betree::Node::{5}::lookup_first_message_for_key] -/ +def betree_node_lookup_first_message_for_key_fwd + (key : UInt64) (msgs : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + match h: msgs with + | betree_list_t.BetreeListCons x next_msgs => + let (i, m) := x + if h: i >= key + then Result.ret (betree_list_t.BetreeListCons (i, m) next_msgs) + else betree_node_lookup_first_message_for_key_fwd key next_msgs + | betree_list_t.BetreeListNil => Result.ret betree_list_t.BetreeListNil +termination_by betree_node_lookup_first_message_for_key_fwd key msgs => + betree_node_lookup_first_message_for_key_terminates key msgs +decreasing_by betree_node_lookup_first_message_for_key_decreases key msgs + +/- [betree_main::betree::Node::{5}::lookup_first_message_for_key] -/ +def betree_node_lookup_first_message_for_key_back + (key : UInt64) (msgs : betree_list_t (UInt64 × betree_message_t)) + (ret0 : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + match h: msgs with + | betree_list_t.BetreeListCons x next_msgs => + let (i, m) := x + if h: i >= key + then Result.ret ret0 + else + do + let next_msgs0 ← + betree_node_lookup_first_message_for_key_back key next_msgs ret0 + Result.ret (betree_list_t.BetreeListCons (i, m) next_msgs0) + | betree_list_t.BetreeListNil => Result.ret ret0 +termination_by betree_node_lookup_first_message_for_key_back key msgs ret0 => + betree_node_lookup_first_message_for_key_terminates key msgs +decreasing_by betree_node_lookup_first_message_for_key_decreases key msgs + +/- [betree_main::betree::Node::{5}::apply_upserts] -/ +def betree_node_apply_upserts_fwd + (msgs : betree_list_t (UInt64 × betree_message_t)) (prev : Option UInt64) + (key : UInt64) (st : State) : + (Result (State × UInt64)) + := + do + let b ← betree_list_head_has_key_fwd betree_message_t msgs key + if h: b + then + do + let msg ← betree_list_pop_front_fwd (UInt64 × betree_message_t) msgs + let (_, m) := msg + match h: m with + | betree_message_t.BetreeMessageInsert i => Result.fail Error.panic + | betree_message_t.BetreeMessageDelete => Result.fail Error.panic + | betree_message_t.BetreeMessageUpsert s => + do + let v ← betree_upsert_update_fwd prev s + let msgs0 ← + betree_list_pop_front_back (UInt64 × betree_message_t) msgs + betree_node_apply_upserts_fwd msgs0 (Option.some v) key st + else + do + let (st0, v) ← + opaque_defs.core_option_option_unwrap_fwd UInt64 prev st + let _ ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) msgs + (key, betree_message_t.BetreeMessageInsert v) + Result.ret (st0, v) +termination_by betree_node_apply_upserts_fwd msgs prev key st => + betree_node_apply_upserts_terminates msgs prev key st +decreasing_by betree_node_apply_upserts_decreases msgs prev key st + +/- [betree_main::betree::Node::{5}::apply_upserts] -/ +def betree_node_apply_upserts_back + (msgs : betree_list_t (UInt64 × betree_message_t)) (prev : Option UInt64) + (key : UInt64) (st : State) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + do + let b ← betree_list_head_has_key_fwd betree_message_t msgs key + if h: b + then + do + let msg ← betree_list_pop_front_fwd (UInt64 × betree_message_t) msgs + let (_, m) := msg + match h: m with + | betree_message_t.BetreeMessageInsert i => Result.fail Error.panic + | betree_message_t.BetreeMessageDelete => Result.fail Error.panic + | betree_message_t.BetreeMessageUpsert s => + do + let v ← betree_upsert_update_fwd prev s + let msgs0 ← + betree_list_pop_front_back (UInt64 × betree_message_t) msgs + betree_node_apply_upserts_back msgs0 (Option.some v) key st + else + do + let (_, v) ← opaque_defs.core_option_option_unwrap_fwd UInt64 prev st + betree_list_push_front_fwd_back (UInt64 × betree_message_t) msgs (key, + betree_message_t.BetreeMessageInsert v) +termination_by betree_node_apply_upserts_back msgs prev key st => + betree_node_apply_upserts_terminates msgs prev key st +decreasing_by betree_node_apply_upserts_decreases msgs prev key st + +/- [betree_main::betree::Node::{5}::lookup] -/ +mutual def betree_node_lookup_fwd + (self : betree_node_t) (key : UInt64) (st : State) : + (Result (State × (Option UInt64))) + := + match h: self with + | betree_node_t.BetreeNodeInternal node => + do + let (st0, msgs) ← + betree_load_internal_node_fwd node.betree_internal_id st + let pending ← betree_node_lookup_first_message_for_key_fwd key msgs + match h: pending with + | betree_list_t.BetreeListCons p l => + let (k, msg) := p + if h: k != key + then + do + let (st1, opt) ← + betree_internal_lookup_in_children_fwd node key st0 + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + (betree_list_t.BetreeListCons (k, msg) l) + Result.ret (st1, opt) + else + match h: msg with + | betree_message_t.BetreeMessageInsert v => + do + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + (betree_list_t.BetreeListCons (k, + betree_message_t.BetreeMessageInsert v) l) + Result.ret (st0, Option.some v) + | betree_message_t.BetreeMessageDelete => + do + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + (betree_list_t.BetreeListCons (k, + betree_message_t.BetreeMessageDelete) l) + Result.ret (st0, Option.none) + | betree_message_t.BetreeMessageUpsert ufs => + do + let (st1, v) ← + betree_internal_lookup_in_children_fwd node key st0 + let (st2, v0) ← + betree_node_apply_upserts_fwd (betree_list_t.BetreeListCons (k, + betree_message_t.BetreeMessageUpsert ufs) l) v key st1 + let node0 ← + betree_internal_lookup_in_children_back node key st0 + let pending0 ← + betree_node_apply_upserts_back (betree_list_t.BetreeListCons + (k, betree_message_t.BetreeMessageUpsert ufs) l) v key st1 + let msgs0 ← + betree_node_lookup_first_message_for_key_back key msgs pending0 + let (st3, _) ← + betree_store_internal_node_fwd node0.betree_internal_id msgs0 + st2 + Result.ret (st3, Option.some v0) + | betree_list_t.BetreeListNil => + do + let (st1, opt) ← + betree_internal_lookup_in_children_fwd node key st0 + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + betree_list_t.BetreeListNil + Result.ret (st1, opt) + | betree_node_t.BetreeNodeLeaf node => + do + let (st0, bindings) ← betree_load_leaf_node_fwd node.betree_leaf_id st + let opt ← betree_node_lookup_in_bindings_fwd key bindings + Result.ret (st0, opt) +termination_by betree_node_lookup_fwd self key st => + betree_node_lookup_terminates self key st +decreasing_by betree_node_lookup_decreases self key st + +/- [betree_main::betree::Node::{5}::lookup] -/ +def betree_node_lookup_back + (self : betree_node_t) (key : UInt64) (st : State) : + (Result betree_node_t) + := + match h: self with + | betree_node_t.BetreeNodeInternal node => + do + let (st0, msgs) ← + betree_load_internal_node_fwd node.betree_internal_id st + let pending ← betree_node_lookup_first_message_for_key_fwd key msgs + match h: pending with + | betree_list_t.BetreeListCons p l => + let (k, msg) := p + if h: k != key + then + do + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + (betree_list_t.BetreeListCons (k, msg) l) + let node0 ← betree_internal_lookup_in_children_back node key st0 + Result.ret (betree_node_t.BetreeNodeInternal node0) + else + match h: msg with + | betree_message_t.BetreeMessageInsert v => + do + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + (betree_list_t.BetreeListCons (k, + betree_message_t.BetreeMessageInsert v) l) + Result.ret (betree_node_t.BetreeNodeInternal node) + | betree_message_t.BetreeMessageDelete => + do + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + (betree_list_t.BetreeListCons (k, + betree_message_t.BetreeMessageDelete) l) + Result.ret (betree_node_t.BetreeNodeInternal node) + | betree_message_t.BetreeMessageUpsert ufs => + do + let (st1, v) ← + betree_internal_lookup_in_children_fwd node key st0 + let (st2, _) ← + betree_node_apply_upserts_fwd (betree_list_t.BetreeListCons (k, + betree_message_t.BetreeMessageUpsert ufs) l) v key st1 + let node0 ← + betree_internal_lookup_in_children_back node key st0 + let pending0 ← + betree_node_apply_upserts_back (betree_list_t.BetreeListCons + (k, betree_message_t.BetreeMessageUpsert ufs) l) v key st1 + let msgs0 ← + betree_node_lookup_first_message_for_key_back key msgs pending0 + let _ ← + betree_store_internal_node_fwd node0.betree_internal_id msgs0 + st2 + Result.ret (betree_node_t.BetreeNodeInternal node0) + | betree_list_t.BetreeListNil => + do + let _ ← + betree_node_lookup_first_message_for_key_back key msgs + betree_list_t.BetreeListNil + let node0 ← betree_internal_lookup_in_children_back node key st0 + Result.ret (betree_node_t.BetreeNodeInternal node0) + | betree_node_t.BetreeNodeLeaf node => + do + let (_, bindings) ← betree_load_leaf_node_fwd node.betree_leaf_id st + let _ ← betree_node_lookup_in_bindings_fwd key bindings + Result.ret (betree_node_t.BetreeNodeLeaf node) +termination_by betree_node_lookup_back self key st => + betree_node_lookup_terminates self key st +decreasing_by betree_node_lookup_decreases self key st + +/- [betree_main::betree::Internal::{4}::lookup_in_children] -/ +def betree_internal_lookup_in_children_fwd + (self : betree_internal_t) (key : UInt64) (st : State) : + (Result (State × (Option UInt64))) + := + if h: key < self.betree_internal_pivot + then betree_node_lookup_fwd self.betree_internal_left key st + else betree_node_lookup_fwd self.betree_internal_right key st +termination_by betree_internal_lookup_in_children_fwd self key st => + betree_internal_lookup_in_children_terminates self key st +decreasing_by betree_internal_lookup_in_children_decreases self key st + +/- [betree_main::betree::Internal::{4}::lookup_in_children] -/ +def betree_internal_lookup_in_children_back + (self : betree_internal_t) (key : UInt64) (st : State) : + (Result betree_internal_t) + := + if h: key < self.betree_internal_pivot + then + do + let n ← betree_node_lookup_back self.betree_internal_left key st + Result.ret + { + betree_internal_id := self.betree_internal_id, + betree_internal_pivot := self.betree_internal_pivot, + betree_internal_left := n, + betree_internal_right := self.betree_internal_right + } + else + do + let n ← betree_node_lookup_back self.betree_internal_right key st + Result.ret + { + betree_internal_id := self.betree_internal_id, + betree_internal_pivot := self.betree_internal_pivot, + betree_internal_left := self.betree_internal_left, + betree_internal_right := n + } +termination_by betree_internal_lookup_in_children_back self key st => + betree_internal_lookup_in_children_terminates self key st +decreasing_by betree_internal_lookup_in_children_decreases self key st + +/- [betree_main::betree::Node::{5}::lookup_mut_in_bindings] -/ +def betree_node_lookup_mut_in_bindings_fwd + (key : UInt64) (bindings : betree_list_t (UInt64 × UInt64)) : + (Result (betree_list_t (UInt64 × UInt64))) + := + match h: bindings with + | betree_list_t.BetreeListCons hd tl => + let (i, i0) := hd + if h: i >= key + then Result.ret (betree_list_t.BetreeListCons (i, i0) tl) + else betree_node_lookup_mut_in_bindings_fwd key tl + | betree_list_t.BetreeListNil => Result.ret betree_list_t.BetreeListNil +termination_by betree_node_lookup_mut_in_bindings_fwd key bindings => + betree_node_lookup_mut_in_bindings_terminates key bindings +decreasing_by betree_node_lookup_mut_in_bindings_decreases key bindings + +/- [betree_main::betree::Node::{5}::lookup_mut_in_bindings] -/ +def betree_node_lookup_mut_in_bindings_back + (key : UInt64) (bindings : betree_list_t (UInt64 × UInt64)) + (ret0 : betree_list_t (UInt64 × UInt64)) : + (Result (betree_list_t (UInt64 × UInt64))) + := + match h: bindings with + | betree_list_t.BetreeListCons hd tl => + let (i, i0) := hd + if h: i >= key + then Result.ret ret0 + else + do + let tl0 ← betree_node_lookup_mut_in_bindings_back key tl ret0 + Result.ret (betree_list_t.BetreeListCons (i, i0) tl0) + | betree_list_t.BetreeListNil => Result.ret ret0 +termination_by betree_node_lookup_mut_in_bindings_back key bindings ret0 => + betree_node_lookup_mut_in_bindings_terminates key bindings +decreasing_by betree_node_lookup_mut_in_bindings_decreases key bindings + +/- [betree_main::betree::Node::{5}::apply_to_leaf] -/ +def betree_node_apply_to_leaf_fwd_back + (bindings : betree_list_t (UInt64 × UInt64)) (key : UInt64) + (new_msg : betree_message_t) : + Result (betree_list_t (UInt64 × UInt64)) + := + do + let bindings0 ← betree_node_lookup_mut_in_bindings_fwd key bindings + let b ← betree_list_head_has_key_fwd UInt64 bindings0 key + if h: b + then + do + let hd ← betree_list_pop_front_fwd (UInt64 × UInt64) bindings0 + match h: new_msg with + | betree_message_t.BetreeMessageInsert v => + do + let bindings1 ← + betree_list_pop_front_back (UInt64 × UInt64) bindings0 + let bindings2 ← + betree_list_push_front_fwd_back (UInt64 × UInt64) bindings1 + (key, v) + betree_node_lookup_mut_in_bindings_back key bindings bindings2 + | betree_message_t.BetreeMessageDelete => + do + let bindings1 ← + betree_list_pop_front_back (UInt64 × UInt64) bindings0 + betree_node_lookup_mut_in_bindings_back key bindings bindings1 + | betree_message_t.BetreeMessageUpsert s => + do + let (_, i) := hd + let v ← betree_upsert_update_fwd (Option.some i) s + let bindings1 ← + betree_list_pop_front_back (UInt64 × UInt64) bindings0 + let bindings2 ← + betree_list_push_front_fwd_back (UInt64 × UInt64) bindings1 + (key, v) + betree_node_lookup_mut_in_bindings_back key bindings bindings2 + else + match h: new_msg with + | betree_message_t.BetreeMessageInsert v => + do + let bindings1 ← + betree_list_push_front_fwd_back (UInt64 × UInt64) bindings0 (key, + v) + betree_node_lookup_mut_in_bindings_back key bindings bindings1 + | betree_message_t.BetreeMessageDelete => + betree_node_lookup_mut_in_bindings_back key bindings bindings0 + | betree_message_t.BetreeMessageUpsert s => + do + let v ← betree_upsert_update_fwd Option.none s + let bindings1 ← + betree_list_push_front_fwd_back (UInt64 × UInt64) bindings0 (key, + v) + betree_node_lookup_mut_in_bindings_back key bindings bindings1 + +/- [betree_main::betree::Node::{5}::apply_messages_to_leaf] -/ +def betree_node_apply_messages_to_leaf_fwd_back + (bindings : betree_list_t (UInt64 × UInt64)) + (new_msgs : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × UInt64))) + := + match h: new_msgs with + | betree_list_t.BetreeListCons new_msg new_msgs_tl => + do + let (i, m) := new_msg + let bindings0 ← betree_node_apply_to_leaf_fwd_back bindings i m + betree_node_apply_messages_to_leaf_fwd_back bindings0 new_msgs_tl + | betree_list_t.BetreeListNil => Result.ret bindings +termination_by betree_node_apply_messages_to_leaf_fwd_back bindings new_msgs => + betree_node_apply_messages_to_leaf_terminates bindings new_msgs +decreasing_by betree_node_apply_messages_to_leaf_decreases bindings new_msgs + +/- [betree_main::betree::Node::{5}::filter_messages_for_key] -/ +def betree_node_filter_messages_for_key_fwd_back + (key : UInt64) (msgs : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + match h: msgs with + | betree_list_t.BetreeListCons p l => + let (k, m) := p + if h: k = key + then + do + let msgs0 ← + betree_list_pop_front_back (UInt64 × betree_message_t) + (betree_list_t.BetreeListCons (k, m) l) + betree_node_filter_messages_for_key_fwd_back key msgs0 + else Result.ret (betree_list_t.BetreeListCons (k, m) l) + | betree_list_t.BetreeListNil => Result.ret betree_list_t.BetreeListNil +termination_by betree_node_filter_messages_for_key_fwd_back key msgs => + betree_node_filter_messages_for_key_terminates key msgs +decreasing_by betree_node_filter_messages_for_key_decreases key msgs + +/- [betree_main::betree::Node::{5}::lookup_first_message_after_key] -/ +def betree_node_lookup_first_message_after_key_fwd + (key : UInt64) (msgs : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + match h: msgs with + | betree_list_t.BetreeListCons p next_msgs => + let (k, m) := p + if h: k = key + then betree_node_lookup_first_message_after_key_fwd key next_msgs + else Result.ret (betree_list_t.BetreeListCons (k, m) next_msgs) + | betree_list_t.BetreeListNil => Result.ret betree_list_t.BetreeListNil +termination_by betree_node_lookup_first_message_after_key_fwd key msgs => + betree_node_lookup_first_message_after_key_terminates key msgs +decreasing_by betree_node_lookup_first_message_after_key_decreases key msgs + +/- [betree_main::betree::Node::{5}::lookup_first_message_after_key] -/ +def betree_node_lookup_first_message_after_key_back + (key : UInt64) (msgs : betree_list_t (UInt64 × betree_message_t)) + (ret0 : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + match h: msgs with + | betree_list_t.BetreeListCons p next_msgs => + let (k, m) := p + if h: k = key + then + do + let next_msgs0 ← + betree_node_lookup_first_message_after_key_back key next_msgs ret0 + Result.ret (betree_list_t.BetreeListCons (k, m) next_msgs0) + else Result.ret ret0 + | betree_list_t.BetreeListNil => Result.ret ret0 +termination_by betree_node_lookup_first_message_after_key_back key msgs ret0 => + betree_node_lookup_first_message_after_key_terminates key msgs +decreasing_by betree_node_lookup_first_message_after_key_decreases key msgs + +/- [betree_main::betree::Node::{5}::apply_to_internal] -/ +def betree_node_apply_to_internal_fwd_back + (msgs : betree_list_t (UInt64 × betree_message_t)) (key : UInt64) + (new_msg : betree_message_t) : + Result (betree_list_t (UInt64 × betree_message_t)) + := + do + let msgs0 ← betree_node_lookup_first_message_for_key_fwd key msgs + let b ← betree_list_head_has_key_fwd betree_message_t msgs0 key + if h: b + then + match h: new_msg with + | betree_message_t.BetreeMessageInsert i => + do + let msgs1 ← betree_node_filter_messages_for_key_fwd_back key msgs0 + let msgs2 ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) msgs1 + (key, betree_message_t.BetreeMessageInsert i) + betree_node_lookup_first_message_for_key_back key msgs msgs2 + | betree_message_t.BetreeMessageDelete => + do + let msgs1 ← betree_node_filter_messages_for_key_fwd_back key msgs0 + let msgs2 ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) msgs1 + (key, betree_message_t.BetreeMessageDelete) + betree_node_lookup_first_message_for_key_back key msgs msgs2 + | betree_message_t.BetreeMessageUpsert s => + do + let p ← betree_list_hd_fwd (UInt64 × betree_message_t) msgs0 + let (_, m) := p + match h: m with + | betree_message_t.BetreeMessageInsert prev => + do + let v ← betree_upsert_update_fwd (Option.some prev) s + let msgs1 ← + betree_list_pop_front_back (UInt64 × betree_message_t) msgs0 + let msgs2 ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) + msgs1 (key, betree_message_t.BetreeMessageInsert v) + betree_node_lookup_first_message_for_key_back key msgs msgs2 + | betree_message_t.BetreeMessageDelete => + do + let v ← betree_upsert_update_fwd Option.none s + let msgs1 ← + betree_list_pop_front_back (UInt64 × betree_message_t) msgs0 + let msgs2 ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) + msgs1 (key, betree_message_t.BetreeMessageInsert v) + betree_node_lookup_first_message_for_key_back key msgs msgs2 + | betree_message_t.BetreeMessageUpsert ufs => + do + let msgs1 ← + betree_node_lookup_first_message_after_key_fwd key msgs0 + let msgs2 ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) + msgs1 (key, betree_message_t.BetreeMessageUpsert s) + let msgs3 ← + betree_node_lookup_first_message_after_key_back key msgs0 msgs2 + betree_node_lookup_first_message_for_key_back key msgs msgs3 + else + do + let msgs1 ← + betree_list_push_front_fwd_back (UInt64 × betree_message_t) msgs0 + (key, new_msg) + betree_node_lookup_first_message_for_key_back key msgs msgs1 + +/- [betree_main::betree::Node::{5}::apply_messages_to_internal] -/ +def betree_node_apply_messages_to_internal_fwd_back + (msgs : betree_list_t (UInt64 × betree_message_t)) + (new_msgs : betree_list_t (UInt64 × betree_message_t)) : + (Result (betree_list_t (UInt64 × betree_message_t))) + := + match h: new_msgs with + | betree_list_t.BetreeListCons new_msg new_msgs_tl => + do + let (i, m) := new_msg + let msgs0 ← betree_node_apply_to_internal_fwd_back msgs i m + betree_node_apply_messages_to_internal_fwd_back msgs0 new_msgs_tl + | betree_list_t.BetreeListNil => Result.ret msgs +termination_by betree_node_apply_messages_to_internal_fwd_back msgs new_msgs => + betree_node_apply_messages_to_internal_terminates msgs new_msgs +decreasing_by betree_node_apply_messages_to_internal_decreases msgs new_msgs + +/- [betree_main::betree::Node::{5}::apply_messages] -/ +mutual def betree_node_apply_messages_fwd + (self : betree_node_t) (params : betree_params_t) + (node_id_cnt : betree_node_id_counter_t) + (msgs : betree_list_t (UInt64 × betree_message_t)) (st : State) : + (Result (State × Unit)) + := + match h: self with + | betree_node_t.BetreeNodeInternal node => + do + let (st0, content) ← + betree_load_internal_node_fwd node.betree_internal_id st + let content0 ← + betree_node_apply_messages_to_internal_fwd_back content msgs + let num_msgs ← + betree_list_len_fwd (UInt64 × betree_message_t) content0 + if h: num_msgs >= params.betree_params_min_flush_size + then + do + let (st1, content1) ← + betree_internal_flush_fwd node params node_id_cnt content0 st0 + let (node0, _) ← + betree_internal_flush_back node params node_id_cnt content0 st0 + let (st2, _) ← + betree_store_internal_node_fwd node0.betree_internal_id content1 + st1 + Result.ret (st2, ()) + else + do + let (st1, _) ← + betree_store_internal_node_fwd node.betree_internal_id content0 st0 + Result.ret (st1, ()) + | betree_node_t.BetreeNodeLeaf node => + do + let (st0, content) ← betree_load_leaf_node_fwd node.betree_leaf_id st + let content0 ← betree_node_apply_messages_to_leaf_fwd_back content msgs + let len ← betree_list_len_fwd (UInt64 × UInt64) content0 + let i ← UInt64.checked_mul (UInt64.ofNatCore 2 (by intlit)) + params.betree_params_split_size + if h: len >= i + then + do + let (st1, _) ← + betree_leaf_split_fwd node content0 params node_id_cnt st0 + let (st2, _) ← + betree_store_leaf_node_fwd node.betree_leaf_id + betree_list_t.BetreeListNil st1 + Result.ret (st2, ()) + else + do + let (st1, _) ← + betree_store_leaf_node_fwd node.betree_leaf_id content0 st0 + Result.ret (st1, ()) +termination_by betree_node_apply_messages_fwd self params node_id_cnt msgs st + => + betree_node_apply_messages_terminates self params node_id_cnt msgs st +decreasing_by + betree_node_apply_messages_decreases self params node_id_cnt msgs st + +/- [betree_main::betree::Node::{5}::apply_messages] -/ +def betree_node_apply_messages_back + (self : betree_node_t) (params : betree_params_t) + (node_id_cnt : betree_node_id_counter_t) + (msgs : betree_list_t (UInt64 × betree_message_t)) (st : State) : + (Result (betree_node_t × betree_node_id_counter_t)) + := + match h: self with + | betree_node_t.BetreeNodeInternal node => + do + let (st0, content) ← + betree_load_internal_node_fwd node.betree_internal_id st + let content0 ← + betree_node_apply_messages_to_internal_fwd_back content msgs + let num_msgs ← + betree_list_len_fwd (UInt64 × betree_message_t) content0 + if h: num_msgs >= params.betree_params_min_flush_size + then + do + let (st1, content1) ← + betree_internal_flush_fwd node params node_id_cnt content0 st0 + let (node0, node_id_cnt0) ← + betree_internal_flush_back node params node_id_cnt content0 st0 + let _ ← + betree_store_internal_node_fwd node0.betree_internal_id content1 + st1 + Result.ret (betree_node_t.BetreeNodeInternal node0, node_id_cnt0) + else + do + let _ ← + betree_store_internal_node_fwd node.betree_internal_id content0 st0 + Result.ret (betree_node_t.BetreeNodeInternal node, node_id_cnt) + | betree_node_t.BetreeNodeLeaf node => + do + let (st0, content) ← betree_load_leaf_node_fwd node.betree_leaf_id st + let content0 ← betree_node_apply_messages_to_leaf_fwd_back content msgs + let len ← betree_list_len_fwd (UInt64 × UInt64) content0 + let i ← UInt64.checked_mul (UInt64.ofNatCore 2 (by intlit)) + params.betree_params_split_size + if h: len >= i + then + do + let (st1, new_node) ← + betree_leaf_split_fwd node content0 params node_id_cnt st0 + let _ ← + betree_store_leaf_node_fwd node.betree_leaf_id + betree_list_t.BetreeListNil st1 + let node_id_cnt0 ← + betree_leaf_split_back node content0 params node_id_cnt st0 + Result.ret (betree_node_t.BetreeNodeInternal new_node, node_id_cnt0) + else + do + let _ ← betree_store_leaf_node_fwd node.betree_leaf_id content0 st0 + Result.ret (betree_node_t.BetreeNodeLeaf + { betree_leaf_id := node.betree_leaf_id, betree_leaf_size := len }, + node_id_cnt) +termination_by betree_node_apply_messages_back self params node_id_cnt msgs st + => + betree_node_apply_messages_terminates self params node_id_cnt msgs st +decreasing_by + betree_node_apply_messages_decreases self params node_id_cnt msgs st + +/- [betree_main::betree::Internal::{4}::flush] -/ +def betree_internal_flush_fwd + (self : betree_internal_t) (params : betree_params_t) + (node_id_cnt : betree_node_id_counter_t) + (content : betree_list_t (UInt64 × betree_message_t)) (st : State) : + (Result (State × (betree_list_t (UInt64 × betree_message_t)))) + := + do + let p ← + betree_list_partition_at_pivot_fwd betree_message_t content + self.betree_internal_pivot + let (msgs_left, msgs_right) := p + let len_left ← betree_list_len_fwd (UInt64 × betree_message_t) msgs_left + if h: len_left >= params.betree_params_min_flush_size + then + do + let (st0, _) ← + betree_node_apply_messages_fwd self.betree_internal_left params + node_id_cnt msgs_left st + let (_, node_id_cnt0) ← + betree_node_apply_messages_back self.betree_internal_left params + node_id_cnt msgs_left st + let len_right ← + betree_list_len_fwd (UInt64 × betree_message_t) msgs_right + if h: len_right >= params.betree_params_min_flush_size + then + do + let (st1, _) ← + betree_node_apply_messages_fwd self.betree_internal_right params + node_id_cnt0 msgs_right st0 + let _ ← + betree_node_apply_messages_back self.betree_internal_right params + node_id_cnt0 msgs_right st0 + Result.ret (st1, betree_list_t.BetreeListNil) + else Result.ret (st0, msgs_right) + else + do + let (st0, _) ← + betree_node_apply_messages_fwd self.betree_internal_right params + node_id_cnt msgs_right st + let _ ← + betree_node_apply_messages_back self.betree_internal_right params + node_id_cnt msgs_right st + Result.ret (st0, msgs_left) +termination_by betree_internal_flush_fwd self params node_id_cnt content st => + betree_internal_flush_terminates self params node_id_cnt content st +decreasing_by + betree_internal_flush_decreases self params node_id_cnt content st + +/- [betree_main::betree::Internal::{4}::flush] -/ +def betree_internal_flush_back + (self : betree_internal_t) (params : betree_params_t) + (node_id_cnt : betree_node_id_counter_t) + (content : betree_list_t (UInt64 × betree_message_t)) (st : State) : + (Result (betree_internal_t × betree_node_id_counter_t)) + := + do + let p ← + betree_list_partition_at_pivot_fwd betree_message_t content + self.betree_internal_pivot + let (msgs_left, msgs_right) := p + let len_left ← betree_list_len_fwd (UInt64 × betree_message_t) msgs_left + if h: len_left >= params.betree_params_min_flush_size + then + do + let (st0, _) ← + betree_node_apply_messages_fwd self.betree_internal_left params + node_id_cnt msgs_left st + let (n, node_id_cnt0) ← + betree_node_apply_messages_back self.betree_internal_left params + node_id_cnt msgs_left st + let len_right ← + betree_list_len_fwd (UInt64 × betree_message_t) msgs_right + if h: len_right >= params.betree_params_min_flush_size + then + do + let (n0, node_id_cnt1) ← + betree_node_apply_messages_back self.betree_internal_right params + node_id_cnt0 msgs_right st0 + Result.ret + ({ + betree_internal_id := self.betree_internal_id, + betree_internal_pivot := self.betree_internal_pivot, + betree_internal_left := n, + betree_internal_right := n0 + }, node_id_cnt1) + else + Result.ret + ({ + betree_internal_id := self.betree_internal_id, + betree_internal_pivot := self.betree_internal_pivot, + betree_internal_left := n, + betree_internal_right := self.betree_internal_right + }, node_id_cnt0) + else + do + let (n, node_id_cnt0) ← + betree_node_apply_messages_back self.betree_internal_right params + node_id_cnt msgs_right st + Result.ret + ({ + betree_internal_id := self.betree_internal_id, + betree_internal_pivot := self.betree_internal_pivot, + betree_internal_left := self.betree_internal_left, + betree_internal_right := n + }, node_id_cnt0) +termination_by betree_internal_flush_back self params node_id_cnt content st => + betree_internal_flush_terminates self params node_id_cnt content st +decreasing_by + betree_internal_flush_decreases self params node_id_cnt content st + +/- [betree_main::betree::Node::{5}::apply] -/ +def betree_node_apply_fwd + (self : betree_node_t) (params : betree_params_t) + (node_id_cnt : betree_node_id_counter_t) (key : UInt64) + (new_msg : betree_message_t) (st : State) : + Result (State × Unit) + := + do + let l := betree_list_t.BetreeListNil + let (st0, _) ← + betree_node_apply_messages_fwd self params node_id_cnt + (betree_list_t.BetreeListCons (key, new_msg) l) st + let _ ← + betree_node_apply_messages_back self params node_id_cnt + (betree_list_t.BetreeListCons (key, new_msg) l) st + Result.ret (st0, ()) + +/- [betree_main::betree::Node::{5}::apply] -/ +def betree_node_apply_back + (self : betree_node_t) (params : betree_params_t) + (node_id_cnt : betree_node_id_counter_t) (key : UInt64) + (new_msg : betree_message_t) (st : State) : + Result (betree_node_t × betree_node_id_counter_t) + := + let l := betree_list_t.BetreeListNil + betree_node_apply_messages_back self params node_id_cnt + (betree_list_t.BetreeListCons (key, new_msg) l) st + +/- [betree_main::betree::BeTree::{6}::new] -/ +def betree_be_tree_new_fwd + (min_flush_size : UInt64) (split_size : UInt64) (st : State) : + Result (State × betree_be_tree_t) + := + do + let node_id_cnt ← betree_node_id_counter_new_fwd + let id ← betree_node_id_counter_fresh_id_fwd node_id_cnt + let (st0, _) ← + betree_store_leaf_node_fwd id betree_list_t.BetreeListNil st + let node_id_cnt0 ← betree_node_id_counter_fresh_id_back node_id_cnt + Result.ret (st0, + { + betree_be_tree_params := { + betree_params_min_flush_size := min_flush_size, + betree_params_split_size := split_size + }, + betree_be_tree_node_id_cnt := node_id_cnt0, + betree_be_tree_root := (betree_node_t.BetreeNodeLeaf + { + betree_leaf_id := id, + betree_leaf_size := (UInt64.ofNatCore 0 (by intlit)) + }) + }) + +/- [betree_main::betree::BeTree::{6}::apply] -/ +def betree_be_tree_apply_fwd + (self : betree_be_tree_t) (key : UInt64) (msg : betree_message_t) + (st : State) : + Result (State × Unit) + := + do + let (st0, _) ← + betree_node_apply_fwd self.betree_be_tree_root self.betree_be_tree_params + self.betree_be_tree_node_id_cnt key msg st + let _ ← + betree_node_apply_back self.betree_be_tree_root + self.betree_be_tree_params self.betree_be_tree_node_id_cnt key msg st + Result.ret (st0, ()) + +/- [betree_main::betree::BeTree::{6}::apply] -/ +def betree_be_tree_apply_back + (self : betree_be_tree_t) (key : UInt64) (msg : betree_message_t) + (st : State) : + Result betree_be_tree_t + := + do + let (n, nic) ← + betree_node_apply_back self.betree_be_tree_root + self.betree_be_tree_params self.betree_be_tree_node_id_cnt key msg st + Result.ret + { + betree_be_tree_params := self.betree_be_tree_params, + betree_be_tree_node_id_cnt := nic, + betree_be_tree_root := n + } + +/- [betree_main::betree::BeTree::{6}::insert] -/ +def betree_be_tree_insert_fwd + (self : betree_be_tree_t) (key : UInt64) (value : UInt64) (st : State) : + Result (State × Unit) + := + do + let (st0, _) ← + betree_be_tree_apply_fwd self key (betree_message_t.BetreeMessageInsert + value) st + let _ ← + betree_be_tree_apply_back self key (betree_message_t.BetreeMessageInsert + value) st + Result.ret (st0, ()) + +/- [betree_main::betree::BeTree::{6}::insert] -/ +def betree_be_tree_insert_back + (self : betree_be_tree_t) (key : UInt64) (value : UInt64) (st : State) : + Result betree_be_tree_t + := + betree_be_tree_apply_back self key (betree_message_t.BetreeMessageInsert + value) st + +/- [betree_main::betree::BeTree::{6}::delete] -/ +def betree_be_tree_delete_fwd + (self : betree_be_tree_t) (key : UInt64) (st : State) : + Result (State × Unit) + := + do + let (st0, _) ← + betree_be_tree_apply_fwd self key betree_message_t.BetreeMessageDelete st + let _ ← + betree_be_tree_apply_back self key betree_message_t.BetreeMessageDelete + st + Result.ret (st0, ()) + +/- [betree_main::betree::BeTree::{6}::delete] -/ +def betree_be_tree_delete_back + (self : betree_be_tree_t) (key : UInt64) (st : State) : + Result betree_be_tree_t + := + betree_be_tree_apply_back self key betree_message_t.BetreeMessageDelete st + +/- [betree_main::betree::BeTree::{6}::upsert] -/ +def betree_be_tree_upsert_fwd + (self : betree_be_tree_t) (key : UInt64) (upd : betree_upsert_fun_state_t) + (st : State) : + Result (State × Unit) + := + do + let (st0, _) ← + betree_be_tree_apply_fwd self key (betree_message_t.BetreeMessageUpsert + upd) st + let _ ← + betree_be_tree_apply_back self key (betree_message_t.BetreeMessageUpsert + upd) st + Result.ret (st0, ()) + +/- [betree_main::betree::BeTree::{6}::upsert] -/ +def betree_be_tree_upsert_back + (self : betree_be_tree_t) (key : UInt64) (upd : betree_upsert_fun_state_t) + (st : State) : + Result betree_be_tree_t + := + betree_be_tree_apply_back self key (betree_message_t.BetreeMessageUpsert upd) + st + +/- [betree_main::betree::BeTree::{6}::lookup] -/ +def betree_be_tree_lookup_fwd + (self : betree_be_tree_t) (key : UInt64) (st : State) : + Result (State × (Option UInt64)) + := + betree_node_lookup_fwd self.betree_be_tree_root key st + +/- [betree_main::betree::BeTree::{6}::lookup] -/ +def betree_be_tree_lookup_back + (self : betree_be_tree_t) (key : UInt64) (st : State) : + Result betree_be_tree_t + := + do + let n ← betree_node_lookup_back self.betree_be_tree_root key st + Result.ret + { + betree_be_tree_params := self.betree_be_tree_params, + betree_be_tree_node_id_cnt := self.betree_be_tree_node_id_cnt, + betree_be_tree_root := n + } + +/- [betree_main::main] -/ +def main_fwd : Result Unit := + Result.ret () + +/- Unit test for [betree_main::main] -/ +#assert (main_fwd == .ret ()) + diff --git a/tests/lean/betree/BetreeMain/Opaque.lean b/tests/lean/betree/BetreeMain/Opaque.lean new file mode 100644 index 00000000..b3db37c2 --- /dev/null +++ b/tests/lean/betree/BetreeMain/Opaque.lean @@ -0,0 +1,33 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree_main]: opaque function definitions +import Base.Primitives +import BetreeMain.Types + +structure OpaqueDefs where + + /- [betree_main::betree_utils::load_internal_node] -/ + betree_utils_load_internal_node_fwd + : + UInt64 -> State -> Result (State × (betree_list_t (UInt64 × + betree_message_t))) + + /- [betree_main::betree_utils::store_internal_node] -/ + betree_utils_store_internal_node_fwd + : + UInt64 -> betree_list_t (UInt64 × betree_message_t) -> State -> Result + (State × Unit) + + /- [betree_main::betree_utils::load_leaf_node] -/ + betree_utils_load_leaf_node_fwd + : UInt64 -> State -> Result (State × (betree_list_t (UInt64 × UInt64))) + + /- [betree_main::betree_utils::store_leaf_node] -/ + betree_utils_store_leaf_node_fwd + : + UInt64 -> betree_list_t (UInt64 × UInt64) -> State -> Result (State × + Unit) + + /- [core::option::Option::{0}::unwrap] -/ + core_option_option_unwrap_fwd + (T : Type) : Option T -> State -> Result (State × T) + diff --git a/tests/lean/betree/BetreeMain/Types.lean b/tests/lean/betree/BetreeMain/Types.lean new file mode 100644 index 00000000..2726e1f0 --- /dev/null +++ b/tests/lean/betree/BetreeMain/Types.lean @@ -0,0 +1,61 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree_main]: type definitions +import Base.Primitives + +/- [betree_main::betree::List] -/ +inductive betree_list_t (T : Type) := +| BetreeListCons : T -> betree_list_t T -> betree_list_t T +| BetreeListNil : betree_list_t T + +/- [betree_main::betree::UpsertFunState] -/ +inductive betree_upsert_fun_state_t := +| BetreeUpsertFunStateAdd : UInt64 -> betree_upsert_fun_state_t +| BetreeUpsertFunStateSub : UInt64 -> betree_upsert_fun_state_t + +/- [betree_main::betree::Message] -/ +inductive betree_message_t := +| BetreeMessageInsert : UInt64 -> betree_message_t +| BetreeMessageDelete : betree_message_t +| BetreeMessageUpsert : betree_upsert_fun_state_t -> betree_message_t + +/- [betree_main::betree::Leaf] -/ +structure betree_leaf_t where + + betree_leaf_id : UInt64 betree_leaf_size : UInt64 + + +/- [betree_main::betree::Node] -/ +mutual inductive betree_node_t := +| BetreeNodeInternal : betree_internal_t -> betree_node_t +| BetreeNodeLeaf : betree_leaf_t -> betree_node_t + +/- [betree_main::betree::Internal] -/ +inductive betree_internal_t := + + betree_internal_id : UInt64 + betree_internal_pivot : UInt64 + betree_internal_left : betree_node_t + betree_internal_right : betree_node_t + + +/- [betree_main::betree::Params] -/ +structure betree_params_t where + + betree_params_min_flush_size : UInt64 betree_params_split_size : UInt64 + + +/- [betree_main::betree::NodeIdCounter] -/ +structure betree_node_id_counter_t where + + betree_node_id_counter_next_node_id : UInt64 + + +/- [betree_main::betree::BeTree] -/ +structure betree_be_tree_t where + + betree_be_tree_params : betree_params_t + betree_be_tree_node_id_cnt : betree_node_id_counter_t + betree_be_tree_root : betree_node_t + +/- The state type used in the state-error monad -/ axiom State : Type + diff --git a/tests/lean/betree/lakefile.lean b/tests/lean/betree/lakefile.lean new file mode 100644 index 00000000..aa702300 --- /dev/null +++ b/tests/lean/betree/lakefile.lean @@ -0,0 +1,18 @@ +import Lake +open Lake DSL + +require mathlib from git + "https://github.com/leanprover-community/mathlib4.git" + +package «betree_main» { + -- add package configuration options here +} + +lean_lib «Base» { + -- add library configuration options here +} + +lean_lib «BetreeMain» { + -- add library configuration options here +} + diff --git a/tests/lean/misc/external/Base/Primitives.lean b/tests/lean/misc-constants/Base/Primitives.lean index 5b64e908..5b64e908 100644 --- a/tests/lean/misc/external/Base/Primitives.lean +++ b/tests/lean/misc-constants/Base/Primitives.lean diff --git a/tests/lean/misc/constants/Constants.lean b/tests/lean/misc-constants/Constants.lean index 57f6e403..57f6e403 100644 --- a/tests/lean/misc/constants/Constants.lean +++ b/tests/lean/misc-constants/Constants.lean diff --git a/tests/lean/misc/constants/lakefile.lean b/tests/lean/misc-constants/lakefile.lean index ed8eebc2..ed8eebc2 100644 --- a/tests/lean/misc/constants/lakefile.lean +++ b/tests/lean/misc-constants/lakefile.lean diff --git a/tests/lean/misc/loops/Base/Primitives.lean b/tests/lean/misc-external/Base/Primitives.lean index 5b64e908..5b64e908 100644 --- a/tests/lean/misc/loops/Base/Primitives.lean +++ b/tests/lean/misc-external/Base/Primitives.lean diff --git a/tests/lean/misc/external/External/Funs.lean b/tests/lean/misc-external/External/Funs.lean index 4e1f36a1..4e1f36a1 100644 --- a/tests/lean/misc/external/External/Funs.lean +++ b/tests/lean/misc-external/External/Funs.lean diff --git a/tests/lean/misc/external/External/Opaque.lean b/tests/lean/misc-external/External/Opaque.lean index d3582de3..d3582de3 100644 --- a/tests/lean/misc/external/External/Opaque.lean +++ b/tests/lean/misc-external/External/Opaque.lean diff --git a/tests/lean/misc/external/External/Types.lean b/tests/lean/misc-external/External/Types.lean index 386832f4..386832f4 100644 --- a/tests/lean/misc/external/External/Types.lean +++ b/tests/lean/misc-external/External/Types.lean diff --git a/tests/lean/misc/external/lakefile.lean b/tests/lean/misc-external/lakefile.lean index b883f4b9..b883f4b9 100644 --- a/tests/lean/misc/external/lakefile.lean +++ b/tests/lean/misc-external/lakefile.lean diff --git a/tests/lean/misc/no_nested_borrows/Base/Primitives.lean b/tests/lean/misc-loops/Base/Primitives.lean index 5b64e908..5b64e908 100644 --- a/tests/lean/misc/no_nested_borrows/Base/Primitives.lean +++ b/tests/lean/misc-loops/Base/Primitives.lean diff --git a/tests/lean/misc/loops/Loops/Clauses/Clauses.lean b/tests/lean/misc-loops/Loops/Clauses/Clauses.lean index 5ddb65ca..5ddb65ca 100644 --- a/tests/lean/misc/loops/Loops/Clauses/Clauses.lean +++ b/tests/lean/misc-loops/Loops/Clauses/Clauses.lean diff --git a/tests/lean/misc/loops/Loops/Clauses/Template.lean b/tests/lean/misc-loops/Loops/Clauses/Template.lean index d1e72d65..d1e72d65 100644 --- a/tests/lean/misc/loops/Loops/Clauses/Template.lean +++ b/tests/lean/misc-loops/Loops/Clauses/Template.lean diff --git a/tests/lean/misc/loops/Loops/Funs.lean b/tests/lean/misc-loops/Loops/Funs.lean index 5a81ebff..5a81ebff 100644 --- a/tests/lean/misc/loops/Loops/Funs.lean +++ b/tests/lean/misc-loops/Loops/Funs.lean diff --git a/tests/lean/misc/loops/Loops/Types.lean b/tests/lean/misc-loops/Loops/Types.lean index f4b6809e..f4b6809e 100644 --- a/tests/lean/misc/loops/Loops/Types.lean +++ b/tests/lean/misc-loops/Loops/Types.lean diff --git a/tests/lean/misc/loops/lakefile.lean b/tests/lean/misc-loops/lakefile.lean index 0d20ba1f..0d20ba1f 100644 --- a/tests/lean/misc/loops/lakefile.lean +++ b/tests/lean/misc-loops/lakefile.lean diff --git a/tests/lean/misc/paper/Base/Primitives.lean b/tests/lean/misc-no_nested_borrows/Base/Primitives.lean index 5b64e908..5b64e908 100644 --- a/tests/lean/misc/paper/Base/Primitives.lean +++ b/tests/lean/misc-no_nested_borrows/Base/Primitives.lean diff --git a/tests/lean/misc/no_nested_borrows/NoNestedBorrows.lean b/tests/lean/misc-no_nested_borrows/NoNestedBorrows.lean index a20ee9fd..a20ee9fd 100644 --- a/tests/lean/misc/no_nested_borrows/NoNestedBorrows.lean +++ b/tests/lean/misc-no_nested_borrows/NoNestedBorrows.lean diff --git a/tests/lean/misc/no_nested_borrows/lakefile.lean b/tests/lean/misc-no_nested_borrows/lakefile.lean index e4460813..e4460813 100644 --- a/tests/lean/misc/no_nested_borrows/lakefile.lean +++ b/tests/lean/misc-no_nested_borrows/lakefile.lean diff --git a/tests/lean/misc-paper/Base/Primitives.lean b/tests/lean/misc-paper/Base/Primitives.lean new file mode 100644 index 00000000..5b64e908 --- /dev/null +++ b/tests/lean/misc-paper/Base/Primitives.lean @@ -0,0 +1,392 @@ +import Lean +import Lean.Meta.Tactic.Simp +import Init.Data.List.Basic +import Mathlib.Tactic.RunCmd + +------------- +-- PRELUDE -- +------------- + +-- Results & monadic combinators + +inductive Error where + | assertionFailure: Error + | integerOverflow: Error + | arrayOutOfBounds: Error + | maximumSizeExceeded: Error + | panic: Error +deriving Repr, BEq + +open Error + +inductive Result (α : Type u) where + | ret (v: α): Result α + | fail (e: Error): Result α +deriving Repr, BEq + +open Result + +/- HELPERS -/ + +def ret? {α: Type} (r: Result α): Bool := + match r with + | Result.ret _ => true + | Result.fail _ => false + +def massert (b:Bool) : Result Unit := + if b then .ret () else fail assertionFailure + +def eval_global {α: Type} (x: Result α) (_: ret? x): α := + match x with + | Result.fail _ => by contradiction + | Result.ret x => x + +/- DO-DSL SUPPORT -/ + +def bind (x: Result α) (f: α -> Result β) : Result β := + match x with + | ret v => f v + | fail v => fail v + +-- Allows using Result in do-blocks +instance : Bind Result where + bind := bind + +-- Allows using return x in do-blocks +instance : Pure Result where + pure := fun x => ret x + +/- CUSTOM-DSL SUPPORT -/ + +-- Let-binding the Result of a monadic operation is oftentimes not sufficient, +-- because we may need a hypothesis for equational reasoning in the scope. We +-- rely on subtype, and a custom let-binding operator, in effect recreating our +-- own variant of the do-dsl + +def Result.attach {α: Type} (o : Result α): Result { x : α // o = ret x } := + match o with + | .ret x => .ret ⟨x, rfl⟩ + | .fail e => .fail e + +macro "let" e:term " ⟵ " f:term : doElem => + `(doElem| let ⟨$e, h⟩ ← Result.attach $f) + +-- TODO: any way to factorize both definitions? +macro "let" e:term " <-- " f:term : doElem => + `(doElem| let ⟨$e, h⟩ ← Result.attach $f) + +-- We call the hypothesis `h`, in effect making it unavailable to the user +-- (because too much shadowing). But in practice, once can use the French single +-- quote notation (input with f< and f>), where `‹ h ›` finds a suitable +-- hypothesis in the context, this is equivalent to `have x: h := by assumption in x` +#eval do + let y <-- .ret (0: Nat) + let _: y = 0 := by cases ‹ ret 0 = ret y › ; decide + let r: { x: Nat // x = 0 } := ⟨ y, by assumption ⟩ + .ret r + +---------------------- +-- MACHINE INTEGERS -- +---------------------- + +-- NOTE: we reuse the fixed-width integer types from prelude.lean: UInt8, ..., +-- USize. They are generally defined in an idiomatic style, except that there is +-- not a single type class to rule them all (more on that below). The absence of +-- type class is intentional, and allows the Lean compiler to efficiently map +-- them to machine integers during compilation. + +-- USize is designed properly: you cannot reduce `getNumBits` using the +-- simplifier, meaning that proofs do not depend on the compile-time value of +-- USize.size. (Lean assumes 32 or 64-bit platforms, and Rust doesn't really +-- support, at least officially, 16-bit microcontrollers, so this seems like a +-- fine design decision for now.) + +-- Note from Chris Bailey: "If there's more than one salient property of your +-- definition then the subtyping strategy might get messy, and the property part +-- of a subtype is less discoverable by the simplifier or tactics like +-- library_search." So, we will not add refinements on the return values of the +-- operations defined on Primitives, but will rather rely on custom lemmas to +-- invert on possible return values of the primitive operations. + +-- Machine integer constants, done via `ofNatCore`, which requires a proof that +-- the `Nat` fits within the desired integer type. We provide a custom tactic. + +syntax "intlit" : tactic + +macro_rules + | `(tactic| intlit) => `(tactic| + match USize.size, usize_size_eq with + | _, Or.inl rfl => decide + | _, Or.inr rfl => decide) + +-- This is how the macro is expected to be used +#eval USize.ofNatCore 0 (by intlit) + +-- Also works for other integer types (at the expense of a needless disjunction) +#eval UInt32.ofNatCore 0 (by intlit) + +-- The machine integer operations (e.g. sub) are always total, which is not what +-- we want. We therefore define "checked" variants, below. Note that we add a +-- tiny bit of complexity for the USize variant: we first check whether the +-- result is < 2^32; if it is, we can compute the definition, rather than +-- returning a term that is computationally stuck (the comparison to USize.size +-- cannot reduce at compile-time, per the remark about regarding `getNumBits`). +-- This is useful for the various #asserts that we want to reduce at +-- type-checking time. + +-- Further thoughts: look at what has been done here: +-- https://github.com/leanprover-community/mathlib4/blob/master/Mathlib/Data/Fin/Basic.lean +-- and +-- https://github.com/leanprover-community/mathlib4/blob/master/Mathlib/Data/UInt.lean +-- which both contain a fair amount of reasoning already! +def USize.checked_sub (n: USize) (m: USize): Result USize := + -- NOTE: the test USize.toNat n - m >= 0 seems to always succeed? + if n >= m then + let n' := USize.toNat n + let m' := USize.toNat n + let r := USize.ofNatCore (n' - m') (by + have h: n' - m' <= n' := by + apply Nat.sub_le_of_le_add + case h => rewrite [ Nat.add_comm ]; apply Nat.le_add_left + apply Nat.lt_of_le_of_lt h + apply n.val.isLt + ) + return r + else + fail integerOverflow + +@[simp] +theorem usize_fits (n: Nat) (h: n <= 4294967295): n < USize.size := + match USize.size, usize_size_eq with + | _, Or.inl rfl => Nat.lt_of_le_of_lt h (by decide) + | _, Or.inr rfl => Nat.lt_of_le_of_lt h (by decide) + +def USize.checked_add (n: USize) (m: USize): Result USize := + if h: n.val + m.val < USize.size then + .ret ⟨ n.val + m.val, h ⟩ + else + .fail integerOverflow + +def USize.checked_rem (n: USize) (m: USize): Result USize := + if h: m > 0 then + .ret ⟨ n.val % m.val, by + have h1: ↑m.val < USize.size := m.val.isLt + have h2: n.val.val % m.val.val < m.val.val := @Nat.mod_lt n.val m.val h + apply Nat.lt_trans h2 h1 + ⟩ + else + .fail integerOverflow + +def USize.checked_mul (n: USize) (m: USize): Result USize := + if h: n.val * m.val < USize.size then + .ret ⟨ n.val * m.val, h ⟩ + else + .fail integerOverflow + +def USize.checked_div (n: USize) (m: USize): Result USize := + if m > 0 then + .ret ⟨ n.val / m.val, by + have h1: ↑n.val < USize.size := n.val.isLt + have h2: n.val.val / m.val.val <= n.val.val := @Nat.div_le_self n.val m.val + apply Nat.lt_of_le_of_lt h2 h1 + ⟩ + else + .fail integerOverflow + +-- Test behavior... +#eval assert! USize.checked_sub 10 20 == fail integerOverflow; 0 + +#eval USize.checked_sub 20 10 +-- NOTE: compare with concrete behavior here, which I do not think we want +#eval USize.sub 0 1 +#eval UInt8.add 255 255 + +-- We now define a type class that subsumes the various machine integer types, so +-- as to write a concise definition for scalar_cast, rather than exhaustively +-- enumerating all of the possible pairs. We remark that Rust has sane semantics +-- and fails if a cast operation would involve a truncation or modulo. + +class MachineInteger (t: Type) where + size: Nat + val: t -> Fin size + ofNatCore: (n:Nat) -> LT.lt n size -> t + +set_option hygiene false in +run_cmd + for typeName in [`UInt8, `UInt16, `UInt32, `UInt64, `USize].map Lean.mkIdent do + Lean.Elab.Command.elabCommand (← `( + namespace $typeName + instance: MachineInteger $typeName where + size := size + val := val + ofNatCore := ofNatCore + end $typeName + )) + +-- Aeneas only instantiates the destination type (`src` is implicit). We rely on +-- Lean to infer `src`. + +def scalar_cast { src: Type } (dst: Type) [ MachineInteger src ] [ MachineInteger dst ] (x: src): Result dst := + if h: MachineInteger.val x < MachineInteger.size dst then + .ret (MachineInteger.ofNatCore (MachineInteger.val x).val h) + else + .fail integerOverflow + +------------- +-- VECTORS -- +------------- + +-- Note: unlike F*, Lean seems to use strict upper bounds (e.g. USize.size) +-- rather than maximum values (usize_max). +def Vec (α : Type u) := { l : List α // List.length l < USize.size } + +def vec_new (α : Type u): Vec α := ⟨ [], by { + match USize.size, usize_size_eq with + | _, Or.inl rfl => simp + | _, Or.inr rfl => simp + } ⟩ + +#check vec_new + +def vec_len (α : Type u) (v : Vec α) : USize := + let ⟨ v, l ⟩ := v + USize.ofNatCore (List.length v) l + +#eval vec_len Nat (vec_new Nat) + +def vec_push_fwd (α : Type u) (_ : Vec α) (_ : α) : Unit := () + +-- NOTE: old version trying to use a subtype notation, but probably better to +-- leave Result elimination to auxiliary lemmas with suitable preconditions +-- TODO: I originally wrote `List.length v.val < USize.size - 1`; how can one +-- make the proof work in that case? Probably need to import tactics from +-- mathlib to deal with inequalities... would love to see an example. +def vec_push_back_old (α : Type u) (v : Vec α) (x : α) : { res: Result (Vec α) // + match res with | fail _ => True | ret v' => List.length v'.val = List.length v.val + 1} + := + if h : List.length v.val + 1 < USize.size then + ⟨ return ⟨List.concat v.val x, + by + rw [List.length_concat] + assumption + ⟩, by simp ⟩ + else + ⟨ fail maximumSizeExceeded, by simp ⟩ + +#eval do + -- NOTE: the // notation is syntactic sugar for Subtype, a refinement with + -- fields val and property. However, Lean's elaborator can automatically + -- select the `val` field if the context provides a type annotation. We + -- annotate `x`, which relieves us of having to write `.val` on the right-hand + -- side of the monadic let. + let v := vec_new Nat + let x: Vec Nat ← (vec_push_back_old Nat v 1: Result (Vec Nat)) -- WHY do we need the type annotation here? + -- TODO: strengthen post-condition above and do a demo to show that we can + -- safely eliminate the `fail` case + return (vec_len Nat x) + +def vec_push_back (α : Type u) (v : Vec α) (x : α) : Result (Vec α) + := + if h : List.length v.val + 1 <= 4294967295 then + return ⟨ List.concat v.val x, + by + rw [List.length_concat] + have h': 4294967295 < USize.size := by intlit + apply Nat.lt_of_le_of_lt h h' + ⟩ + else if h: List.length v.val + 1 < USize.size then + return ⟨List.concat v.val x, + by + rw [List.length_concat] + assumption + ⟩ + else + fail maximumSizeExceeded + +def vec_insert_fwd (α : Type u) (v: Vec α) (i: USize) (_: α): Result Unit := + if i.val < List.length v.val then + .ret () + else + .fail arrayOutOfBounds + +def vec_insert_back (α : Type u) (v: Vec α) (i: USize) (x: α): Result (Vec α) := + if i.val < List.length v.val then + .ret ⟨ List.set v.val i.val x, by + have h: List.length v.val < USize.size := v.property + rewrite [ List.length_set v.val i.val x ] + assumption + ⟩ + else + .fail arrayOutOfBounds + +def vec_index_fwd (α : Type u) (v: Vec α) (i: USize): Result α := + if h: i.val < List.length v.val then + .ret (List.get v.val ⟨i.val, h⟩) + else + .fail arrayOutOfBounds + +def vec_index_back (α : Type u) (v: Vec α) (i: USize) (_: α): Result Unit := + if i.val < List.length v.val then + .ret () + else + .fail arrayOutOfBounds + +def vec_index_mut_fwd (α : Type u) (v: Vec α) (i: USize): Result α := + if h: i.val < List.length v.val then + .ret (List.get v.val ⟨i.val, h⟩) + else + .fail arrayOutOfBounds + +def vec_index_mut_back (α : Type u) (v: Vec α) (i: USize) (x: α): Result (Vec α) := + if i.val < List.length v.val then + .ret ⟨ List.set v.val i.val x, by + have h: List.length v.val < USize.size := v.property + rewrite [ List.length_set v.val i.val x ] + assumption + ⟩ + else + .fail arrayOutOfBounds + +---------- +-- MISC -- +---------- + +def mem_replace_fwd (a : Type) (x : a) (_ : a) : a := + x + +def mem_replace_back (a : Type) (_ : a) (y : a) : a := + y + +/-- Aeneas-translated function -- useful to reduce non-recursive definitions. + Use with `simp [ aeneas ]` -/ +register_simp_attr aeneas + +-------------------- +-- ASSERT COMMAND -- +-------------------- + +open Lean Elab Command Term Meta + +syntax (name := assert) "#assert" term: command + +@[command_elab assert] +unsafe +def assertImpl : CommandElab := fun (_stx: Syntax) => do + runTermElabM (fun _ => do + let r ← evalTerm Bool (mkConst ``Bool) _stx[1] + if not r then + logInfo "Assertion failed for: " + logInfo _stx[1] + logError "Expression reduced to false" + pure ()) + +#eval 2 == 2 +#assert (2 == 2) + +------------------- +-- SANITY CHECKS -- +------------------- + +-- TODO: add more once we have signed integers + +#assert (USize.checked_rem 1 2 == .ret 1) diff --git a/tests/lean/misc/paper/Paper.lean b/tests/lean/misc-paper/Paper.lean index 4faf36ee..4faf36ee 100644 --- a/tests/lean/misc/paper/Paper.lean +++ b/tests/lean/misc-paper/Paper.lean diff --git a/tests/lean/misc/paper/lakefile.lean b/tests/lean/misc-paper/lakefile.lean index d8affff8..d8affff8 100644 --- a/tests/lean/misc/paper/lakefile.lean +++ b/tests/lean/misc-paper/lakefile.lean diff --git a/tests/lean/misc-polonius_list/Base/Primitives.lean b/tests/lean/misc-polonius_list/Base/Primitives.lean new file mode 100644 index 00000000..5b64e908 --- /dev/null +++ b/tests/lean/misc-polonius_list/Base/Primitives.lean @@ -0,0 +1,392 @@ +import Lean +import Lean.Meta.Tactic.Simp +import Init.Data.List.Basic +import Mathlib.Tactic.RunCmd + +------------- +-- PRELUDE -- +------------- + +-- Results & monadic combinators + +inductive Error where + | assertionFailure: Error + | integerOverflow: Error + | arrayOutOfBounds: Error + | maximumSizeExceeded: Error + | panic: Error +deriving Repr, BEq + +open Error + +inductive Result (α : Type u) where + | ret (v: α): Result α + | fail (e: Error): Result α +deriving Repr, BEq + +open Result + +/- HELPERS -/ + +def ret? {α: Type} (r: Result α): Bool := + match r with + | Result.ret _ => true + | Result.fail _ => false + +def massert (b:Bool) : Result Unit := + if b then .ret () else fail assertionFailure + +def eval_global {α: Type} (x: Result α) (_: ret? x): α := + match x with + | Result.fail _ => by contradiction + | Result.ret x => x + +/- DO-DSL SUPPORT -/ + +def bind (x: Result α) (f: α -> Result β) : Result β := + match x with + | ret v => f v + | fail v => fail v + +-- Allows using Result in do-blocks +instance : Bind Result where + bind := bind + +-- Allows using return x in do-blocks +instance : Pure Result where + pure := fun x => ret x + +/- CUSTOM-DSL SUPPORT -/ + +-- Let-binding the Result of a monadic operation is oftentimes not sufficient, +-- because we may need a hypothesis for equational reasoning in the scope. We +-- rely on subtype, and a custom let-binding operator, in effect recreating our +-- own variant of the do-dsl + +def Result.attach {α: Type} (o : Result α): Result { x : α // o = ret x } := + match o with + | .ret x => .ret ⟨x, rfl⟩ + | .fail e => .fail e + +macro "let" e:term " ⟵ " f:term : doElem => + `(doElem| let ⟨$e, h⟩ ← Result.attach $f) + +-- TODO: any way to factorize both definitions? +macro "let" e:term " <-- " f:term : doElem => + `(doElem| let ⟨$e, h⟩ ← Result.attach $f) + +-- We call the hypothesis `h`, in effect making it unavailable to the user +-- (because too much shadowing). But in practice, once can use the French single +-- quote notation (input with f< and f>), where `‹ h ›` finds a suitable +-- hypothesis in the context, this is equivalent to `have x: h := by assumption in x` +#eval do + let y <-- .ret (0: Nat) + let _: y = 0 := by cases ‹ ret 0 = ret y › ; decide + let r: { x: Nat // x = 0 } := ⟨ y, by assumption ⟩ + .ret r + +---------------------- +-- MACHINE INTEGERS -- +---------------------- + +-- NOTE: we reuse the fixed-width integer types from prelude.lean: UInt8, ..., +-- USize. They are generally defined in an idiomatic style, except that there is +-- not a single type class to rule them all (more on that below). The absence of +-- type class is intentional, and allows the Lean compiler to efficiently map +-- them to machine integers during compilation. + +-- USize is designed properly: you cannot reduce `getNumBits` using the +-- simplifier, meaning that proofs do not depend on the compile-time value of +-- USize.size. (Lean assumes 32 or 64-bit platforms, and Rust doesn't really +-- support, at least officially, 16-bit microcontrollers, so this seems like a +-- fine design decision for now.) + +-- Note from Chris Bailey: "If there's more than one salient property of your +-- definition then the subtyping strategy might get messy, and the property part +-- of a subtype is less discoverable by the simplifier or tactics like +-- library_search." So, we will not add refinements on the return values of the +-- operations defined on Primitives, but will rather rely on custom lemmas to +-- invert on possible return values of the primitive operations. + +-- Machine integer constants, done via `ofNatCore`, which requires a proof that +-- the `Nat` fits within the desired integer type. We provide a custom tactic. + +syntax "intlit" : tactic + +macro_rules + | `(tactic| intlit) => `(tactic| + match USize.size, usize_size_eq with + | _, Or.inl rfl => decide + | _, Or.inr rfl => decide) + +-- This is how the macro is expected to be used +#eval USize.ofNatCore 0 (by intlit) + +-- Also works for other integer types (at the expense of a needless disjunction) +#eval UInt32.ofNatCore 0 (by intlit) + +-- The machine integer operations (e.g. sub) are always total, which is not what +-- we want. We therefore define "checked" variants, below. Note that we add a +-- tiny bit of complexity for the USize variant: we first check whether the +-- result is < 2^32; if it is, we can compute the definition, rather than +-- returning a term that is computationally stuck (the comparison to USize.size +-- cannot reduce at compile-time, per the remark about regarding `getNumBits`). +-- This is useful for the various #asserts that we want to reduce at +-- type-checking time. + +-- Further thoughts: look at what has been done here: +-- https://github.com/leanprover-community/mathlib4/blob/master/Mathlib/Data/Fin/Basic.lean +-- and +-- https://github.com/leanprover-community/mathlib4/blob/master/Mathlib/Data/UInt.lean +-- which both contain a fair amount of reasoning already! +def USize.checked_sub (n: USize) (m: USize): Result USize := + -- NOTE: the test USize.toNat n - m >= 0 seems to always succeed? + if n >= m then + let n' := USize.toNat n + let m' := USize.toNat n + let r := USize.ofNatCore (n' - m') (by + have h: n' - m' <= n' := by + apply Nat.sub_le_of_le_add + case h => rewrite [ Nat.add_comm ]; apply Nat.le_add_left + apply Nat.lt_of_le_of_lt h + apply n.val.isLt + ) + return r + else + fail integerOverflow + +@[simp] +theorem usize_fits (n: Nat) (h: n <= 4294967295): n < USize.size := + match USize.size, usize_size_eq with + | _, Or.inl rfl => Nat.lt_of_le_of_lt h (by decide) + | _, Or.inr rfl => Nat.lt_of_le_of_lt h (by decide) + +def USize.checked_add (n: USize) (m: USize): Result USize := + if h: n.val + m.val < USize.size then + .ret ⟨ n.val + m.val, h ⟩ + else + .fail integerOverflow + +def USize.checked_rem (n: USize) (m: USize): Result USize := + if h: m > 0 then + .ret ⟨ n.val % m.val, by + have h1: ↑m.val < USize.size := m.val.isLt + have h2: n.val.val % m.val.val < m.val.val := @Nat.mod_lt n.val m.val h + apply Nat.lt_trans h2 h1 + ⟩ + else + .fail integerOverflow + +def USize.checked_mul (n: USize) (m: USize): Result USize := + if h: n.val * m.val < USize.size then + .ret ⟨ n.val * m.val, h ⟩ + else + .fail integerOverflow + +def USize.checked_div (n: USize) (m: USize): Result USize := + if m > 0 then + .ret ⟨ n.val / m.val, by + have h1: ↑n.val < USize.size := n.val.isLt + have h2: n.val.val / m.val.val <= n.val.val := @Nat.div_le_self n.val m.val + apply Nat.lt_of_le_of_lt h2 h1 + ⟩ + else + .fail integerOverflow + +-- Test behavior... +#eval assert! USize.checked_sub 10 20 == fail integerOverflow; 0 + +#eval USize.checked_sub 20 10 +-- NOTE: compare with concrete behavior here, which I do not think we want +#eval USize.sub 0 1 +#eval UInt8.add 255 255 + +-- We now define a type class that subsumes the various machine integer types, so +-- as to write a concise definition for scalar_cast, rather than exhaustively +-- enumerating all of the possible pairs. We remark that Rust has sane semantics +-- and fails if a cast operation would involve a truncation or modulo. + +class MachineInteger (t: Type) where + size: Nat + val: t -> Fin size + ofNatCore: (n:Nat) -> LT.lt n size -> t + +set_option hygiene false in +run_cmd + for typeName in [`UInt8, `UInt16, `UInt32, `UInt64, `USize].map Lean.mkIdent do + Lean.Elab.Command.elabCommand (← `( + namespace $typeName + instance: MachineInteger $typeName where + size := size + val := val + ofNatCore := ofNatCore + end $typeName + )) + +-- Aeneas only instantiates the destination type (`src` is implicit). We rely on +-- Lean to infer `src`. + +def scalar_cast { src: Type } (dst: Type) [ MachineInteger src ] [ MachineInteger dst ] (x: src): Result dst := + if h: MachineInteger.val x < MachineInteger.size dst then + .ret (MachineInteger.ofNatCore (MachineInteger.val x).val h) + else + .fail integerOverflow + +------------- +-- VECTORS -- +------------- + +-- Note: unlike F*, Lean seems to use strict upper bounds (e.g. USize.size) +-- rather than maximum values (usize_max). +def Vec (α : Type u) := { l : List α // List.length l < USize.size } + +def vec_new (α : Type u): Vec α := ⟨ [], by { + match USize.size, usize_size_eq with + | _, Or.inl rfl => simp + | _, Or.inr rfl => simp + } ⟩ + +#check vec_new + +def vec_len (α : Type u) (v : Vec α) : USize := + let ⟨ v, l ⟩ := v + USize.ofNatCore (List.length v) l + +#eval vec_len Nat (vec_new Nat) + +def vec_push_fwd (α : Type u) (_ : Vec α) (_ : α) : Unit := () + +-- NOTE: old version trying to use a subtype notation, but probably better to +-- leave Result elimination to auxiliary lemmas with suitable preconditions +-- TODO: I originally wrote `List.length v.val < USize.size - 1`; how can one +-- make the proof work in that case? Probably need to import tactics from +-- mathlib to deal with inequalities... would love to see an example. +def vec_push_back_old (α : Type u) (v : Vec α) (x : α) : { res: Result (Vec α) // + match res with | fail _ => True | ret v' => List.length v'.val = List.length v.val + 1} + := + if h : List.length v.val + 1 < USize.size then + ⟨ return ⟨List.concat v.val x, + by + rw [List.length_concat] + assumption + ⟩, by simp ⟩ + else + ⟨ fail maximumSizeExceeded, by simp ⟩ + +#eval do + -- NOTE: the // notation is syntactic sugar for Subtype, a refinement with + -- fields val and property. However, Lean's elaborator can automatically + -- select the `val` field if the context provides a type annotation. We + -- annotate `x`, which relieves us of having to write `.val` on the right-hand + -- side of the monadic let. + let v := vec_new Nat + let x: Vec Nat ← (vec_push_back_old Nat v 1: Result (Vec Nat)) -- WHY do we need the type annotation here? + -- TODO: strengthen post-condition above and do a demo to show that we can + -- safely eliminate the `fail` case + return (vec_len Nat x) + +def vec_push_back (α : Type u) (v : Vec α) (x : α) : Result (Vec α) + := + if h : List.length v.val + 1 <= 4294967295 then + return ⟨ List.concat v.val x, + by + rw [List.length_concat] + have h': 4294967295 < USize.size := by intlit + apply Nat.lt_of_le_of_lt h h' + ⟩ + else if h: List.length v.val + 1 < USize.size then + return ⟨List.concat v.val x, + by + rw [List.length_concat] + assumption + ⟩ + else + fail maximumSizeExceeded + +def vec_insert_fwd (α : Type u) (v: Vec α) (i: USize) (_: α): Result Unit := + if i.val < List.length v.val then + .ret () + else + .fail arrayOutOfBounds + +def vec_insert_back (α : Type u) (v: Vec α) (i: USize) (x: α): Result (Vec α) := + if i.val < List.length v.val then + .ret ⟨ List.set v.val i.val x, by + have h: List.length v.val < USize.size := v.property + rewrite [ List.length_set v.val i.val x ] + assumption + ⟩ + else + .fail arrayOutOfBounds + +def vec_index_fwd (α : Type u) (v: Vec α) (i: USize): Result α := + if h: i.val < List.length v.val then + .ret (List.get v.val ⟨i.val, h⟩) + else + .fail arrayOutOfBounds + +def vec_index_back (α : Type u) (v: Vec α) (i: USize) (_: α): Result Unit := + if i.val < List.length v.val then + .ret () + else + .fail arrayOutOfBounds + +def vec_index_mut_fwd (α : Type u) (v: Vec α) (i: USize): Result α := + if h: i.val < List.length v.val then + .ret (List.get v.val ⟨i.val, h⟩) + else + .fail arrayOutOfBounds + +def vec_index_mut_back (α : Type u) (v: Vec α) (i: USize) (x: α): Result (Vec α) := + if i.val < List.length v.val then + .ret ⟨ List.set v.val i.val x, by + have h: List.length v.val < USize.size := v.property + rewrite [ List.length_set v.val i.val x ] + assumption + ⟩ + else + .fail arrayOutOfBounds + +---------- +-- MISC -- +---------- + +def mem_replace_fwd (a : Type) (x : a) (_ : a) : a := + x + +def mem_replace_back (a : Type) (_ : a) (y : a) : a := + y + +/-- Aeneas-translated function -- useful to reduce non-recursive definitions. + Use with `simp [ aeneas ]` -/ +register_simp_attr aeneas + +-------------------- +-- ASSERT COMMAND -- +-------------------- + +open Lean Elab Command Term Meta + +syntax (name := assert) "#assert" term: command + +@[command_elab assert] +unsafe +def assertImpl : CommandElab := fun (_stx: Syntax) => do + runTermElabM (fun _ => do + let r ← evalTerm Bool (mkConst ``Bool) _stx[1] + if not r then + logInfo "Assertion failed for: " + logInfo _stx[1] + logError "Expression reduced to false" + pure ()) + +#eval 2 == 2 +#assert (2 == 2) + +------------------- +-- SANITY CHECKS -- +------------------- + +-- TODO: add more once we have signed integers + +#assert (USize.checked_rem 1 2 == .ret 1) diff --git a/tests/lean/misc-polonius_list/PoloniusList.lean b/tests/lean/misc-polonius_list/PoloniusList.lean new file mode 100644 index 00000000..d679230d --- /dev/null +++ b/tests/lean/misc-polonius_list/PoloniusList.lean @@ -0,0 +1,36 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [polonius_list] +import Base.Primitives + +structure OpaqueDefs where + + /- [polonius_list::List] -/ + inductive list_t (T : Type) := + | ListCons : T -> list_t T -> list_t T + | ListNil : list_t T + + /- [polonius_list::get_list_at_x] -/ + def get_list_at_x_fwd + (ls : list_t UInt32) (x : UInt32) : Result (list_t UInt32) := + match h: ls with + | list_t.ListCons hd tl => + if h: hd = x + then Result.ret (list_t.ListCons hd tl) + else get_list_at_x_fwd tl x + | list_t.ListNil => Result.ret list_t.ListNil + + /- [polonius_list::get_list_at_x] -/ + def get_list_at_x_back + (ls : list_t UInt32) (x : UInt32) (ret0 : list_t UInt32) : + Result (list_t UInt32) + := + match h: ls with + | list_t.ListCons hd tl => + if h: hd = x + then Result.ret ret0 + else + do + let tl0 ← get_list_at_x_back tl x ret0 + Result.ret (list_t.ListCons hd tl0) + | list_t.ListNil => Result.ret ret0 + diff --git a/tests/lean/misc-polonius_list/lakefile.lean b/tests/lean/misc-polonius_list/lakefile.lean new file mode 100644 index 00000000..f941effc --- /dev/null +++ b/tests/lean/misc-polonius_list/lakefile.lean @@ -0,0 +1,18 @@ +import Lake +open Lake DSL + +require mathlib from git + "https://github.com/leanprover-community/mathlib4.git" + +package «polonius_list» { + -- add package configuration options here +} + +lean_lib «Base» { + -- add library configuration options here +} + +lean_lib «PoloniusList» { + -- add library configuration options here +} + |