path: root/tests/fstar/hashmap/Hashmap.Funs.fst

(** THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS *)
(** [hashmap]: function definitions *)
module Hashmap.Funs
open Primitives
include Hashmap.Types
include Hashmap.Clauses

#set-options "--z3rlimit 50 --fuel 1 --ifuel 1"

(** [hashmap::hash_key] *)
let hash_key_fwd (k : usize) : result usize = Return k

(** [hashmap::HashMap::{0}::allocate_slots] *)
let rec hash_map_allocate_slots_loop_fwd
  (t : Type0) (v : vec (list_t t)) (n : usize) :
  Tot (result (vec (list_t t)))
  (decreases (hash_map_allocate_slots_decreases t v n))
  =
  if n > 0
  then
    begin match vec_push_back (list_t t) v ListNil with
    | Fail e -> Fail e
    | Return slots ->
      begin match usize_sub n 1 with
      | Fail e -> Fail e
      | Return n0 -> hash_map_allocate_slots_loop_fwd t slots n0
      end
    end
  else Return v

(** [hashmap::HashMap::{0}::allocate_slots] *)
let hash_map_allocate_slots_fwd
  (t : Type0) (slots : vec (list_t t)) (n : usize) : result (vec (list_t t)) =
  hash_map_allocate_slots_loop_fwd t slots n

(** [hashmap::HashMap::{0}::new_with_capacity] *)
let hash_map_new_with_capacity_fwd
  (t : Type0) (capacity : usize) (max_load_dividend : usize)
  (max_load_divisor : usize) :
  result (hash_map_t t)
  =
  let v = vec_new (list_t t) in
  begin match hash_map_allocate_slots_fwd t v capacity with
  | Fail e -> Fail e
  | Return slots ->
    begin match usize_mul capacity max_load_dividend with
    | Fail e -> Fail e
    | Return i ->
      begin match usize_div i max_load_divisor with
      | Fail e -> Fail e
      | Return i0 ->
        Return (Mkhash_map_t 0 (max_load_dividend, max_load_divisor) i0 slots)
      end
    end
  end

(** [hashmap::HashMap::{0}::new] *)
let hash_map_new_fwd (t : Type0) : result (hash_map_t t) =
  hash_map_new_with_capacity_fwd t 32 4 5

(** [hashmap::HashMap::{0}::clear_slots] *)
let rec hash_map_clear_slots_loop_fwd_back
  (t : Type0) (v : vec (list_t t)) (i : usize) :
  Tot (result (vec (list_t t)))
  (decreases (hash_map_clear_slots_decreases t v i))
  =
  let i0 = vec_len (list_t t) v in
  if i < i0
  then
    begin match usize_add i 1 with
    | Fail e -> Fail e
    | Return i1 ->
      begin match vec_index_mut_back (list_t t) v i ListNil with
      | Fail e -> Fail e
      | Return slots -> hash_map_clear_slots_loop_fwd_back t slots i1
      end
    end
  else Return v

(** [hashmap::HashMap::{0}::clear_slots] *)
let hash_map_clear_slots_fwd_back
  (t : Type0) (slots : vec (list_t t)) : result (vec (list_t t)) =
  hash_map_clear_slots_loop_fwd_back t slots 0

(** [hashmap::HashMap::{0}::clear] *)
let hash_map_clear_fwd_back
  (t : Type0) (self : hash_map_t t) : result (hash_map_t t) =
  begin match hash_map_clear_slots_fwd_back t self.hash_map_slots with
  | Fail e -> Fail e
  | Return v ->
    Return (Mkhash_map_t 0 self.hash_map_max_load_factor self.hash_map_max_load
      v)
  end

(** [hashmap::HashMap::{0}::len] *)
let hash_map_len_fwd (t : Type0) (self : hash_map_t t) : result usize =
  Return self.hash_map_num_entries

(** [hashmap::HashMap::{0}::insert_in_list] *)
let rec hash_map_insert_in_list_loop_fwd
  (t : Type0) (key : usize) (value : t) (ls : list_t t) :
  Tot (result bool)
  (decreases (hash_map_insert_in_list_decreases t key value ls))
  =
  begin match ls with
  | ListCons ckey cvalue tl ->
    if ckey = key
    then Return false
    else hash_map_insert_in_list_loop_fwd t key value tl
  | ListNil -> Return true
  end

(** [hashmap::HashMap::{0}::insert_in_list] *)
let hash_map_insert_in_list_fwd
  (t : Type0) (key : usize) (value : t) (ls : list_t t) : result bool =
  hash_map_insert_in_list_loop_fwd t key value ls

(** [hashmap::HashMap::{0}::insert_in_list] *)
let rec hash_map_insert_in_list_loop_back
  (t : Type0) (key : usize) (value : t) (ls : list_t t) :
  Tot (result (list_t t))
  (decreases (hash_map_insert_in_list_decreases t key value ls))
  =
  begin match ls with
  | ListCons ckey cvalue tl ->
    if ckey = key
    then Return (ListCons ckey value tl)
    else
      begin match hash_map_insert_in_list_loop_back t key value tl with
      | Fail e -> Fail e
      | Return l -> Return (ListCons ckey cvalue l)
      end
  | ListNil -> let l = ListNil in Return (ListCons key value l)
  end

(** [hashmap::HashMap::{0}::insert_in_list] *)
let hash_map_insert_in_list_back
  (t : Type0) (key : usize) (value : t) (ls : list_t t) : result (list_t t) =
  hash_map_insert_in_list_loop_back t key value ls

(** [hashmap::HashMap::{0}::insert_no_resize] *)
let hash_map_insert_no_resize_fwd_back
  (t : Type0) (self : hash_map_t t) (key : usize) (value : t) :
  result (hash_map_t t)
  =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_mut_fwd (list_t t) self.hash_map_slots hash_mod
        with
      | Fail e -> Fail e
      | Return l ->
        begin match hash_map_insert_in_list_fwd t key value l with
        | Fail e -> Fail e
        | Return inserted ->
          if inserted
          then
            begin match usize_add self.hash_map_num_entries 1 with
            | Fail e -> Fail e
            | Return i0 ->
              begin match hash_map_insert_in_list_back t key value l with
              | Fail e -> Fail e
              | Return l0 ->
                begin match
                  vec_index_mut_back (list_t t) self.hash_map_slots hash_mod l0
                  with
                | Fail e -> Fail e
                | Return v ->
                  Return (Mkhash_map_t i0 self.hash_map_max_load_factor
                    self.hash_map_max_load v)
                end
              end
            end
          else
            begin match hash_map_insert_in_list_back t key value l with
            | Fail e -> Fail e
            | Return l0 ->
              begin match
                vec_index_mut_back (list_t t) self.hash_map_slots hash_mod l0
                with
              | Fail e -> Fail e
              | Return v ->
                Return (Mkhash_map_t self.hash_map_num_entries
                  self.hash_map_max_load_factor self.hash_map_max_load v)
              end
            end
        end
      end
    end
  end

(** [core::num::u32::{9}::MAX] *)
let core_num_u32_max_body : result u32 = Return 4294967295
let core_num_u32_max_c : u32 = eval_global core_num_u32_max_body

(** [hashmap::HashMap::{0}::move_elements_from_list] *)
let rec hash_map_move_elements_from_list_loop_fwd_back
  (t : Type0) (hm : hash_map_t t) (ls : list_t t) :
  Tot (result (hash_map_t t))
  (decreases (hash_map_move_elements_from_list_decreases t hm ls))
  =
  begin match ls with
  | ListCons k v tl ->
    begin match hash_map_insert_no_resize_fwd_back t hm k v with
    | Fail e -> Fail e
    | Return ntable ->
      hash_map_move_elements_from_list_loop_fwd_back t ntable tl
    end
  | ListNil -> Return hm
  end

(** [hashmap::HashMap::{0}::move_elements_from_list] *)
let hash_map_move_elements_from_list_fwd_back
  (t : Type0) (ntable : hash_map_t t) (ls : list_t t) : result (hash_map_t t) =
  hash_map_move_elements_from_list_loop_fwd_back t ntable ls

(** [hashmap::HashMap::{0}::move_elements] *)
let rec hash_map_move_elements_loop_fwd_back
  (t : Type0) (hm : hash_map_t t) (v : vec (list_t t)) (i : usize) :
  Tot (result ((hash_map_t t) & (vec (list_t t))))
  (decreases (hash_map_move_elements_decreases t hm v i))
  =
  let i0 = vec_len (list_t t) v in
  if i < i0
  then
    begin match vec_index_mut_fwd (list_t t) v i with
    | Fail e -> Fail e
    | Return l ->
      let ls = mem_replace_fwd (list_t t) l ListNil in
      begin match hash_map_move_elements_from_list_fwd_back t hm ls with
      | Fail e -> Fail e
      | Return ntable ->
        begin match usize_add i 1 with
        | Fail e -> Fail e
        | Return i1 ->
          let l0 = mem_replace_back (list_t t) l ListNil in
          begin match vec_index_mut_back (list_t t) v i l0 with
          | Fail e -> Fail e
          | Return slots ->
            hash_map_move_elements_loop_fwd_back t ntable slots i1
          end
        end
      end
    end
  else Return (hm, v)

(** [hashmap::HashMap::{0}::move_elements] *)
let hash_map_move_elements_fwd_back
  (t : Type0) (ntable : hash_map_t t) (slots : vec (list_t t)) (i : usize) :
  result ((hash_map_t t) & (vec (list_t t)))
  =
  hash_map_move_elements_loop_fwd_back t ntable slots i

(** [hashmap::HashMap::{0}::try_resize] *)
let hash_map_try_resize_fwd_back
  (t : Type0) (self : hash_map_t t) : result (hash_map_t t) =
  begin match scalar_cast U32 Usize core_num_u32_max_c with
  | Fail e -> Fail e
  | Return max_usize ->
    let capacity = vec_len (list_t t) self.hash_map_slots in
    begin match usize_div max_usize 2 with
    | Fail e -> Fail e
    | Return n1 ->
      let (i, i0) = self.hash_map_max_load_factor in
      begin match usize_div n1 i with
      | Fail e -> Fail e
      | Return i1 ->
        if capacity <= i1
        then
          begin match usize_mul capacity 2 with
          | Fail e -> Fail e
          | Return i2 ->
            begin match hash_map_new_with_capacity_fwd t i2 i i0 with
            | Fail e -> Fail e
            | Return ntable ->
              begin match
                hash_map_move_elements_fwd_back t ntable self.hash_map_slots 0
                with
              | Fail e -> Fail e
              | Return (ntable0, _) ->
                Return (Mkhash_map_t self.hash_map_num_entries (i, i0)
                  ntable0.hash_map_max_load ntable0.hash_map_slots)
              end
            end
          end
        else
          Return (Mkhash_map_t self.hash_map_num_entries (i, i0)
            self.hash_map_max_load self.hash_map_slots)
      end
    end
  end

(** [hashmap::HashMap::{0}::insert] *)
let hash_map_insert_fwd_back
  (t : Type0) (self : hash_map_t t) (key : usize) (value : t) :
  result (hash_map_t t)
  =
  begin match hash_map_insert_no_resize_fwd_back t self key value with
  | Fail e -> Fail e
  | Return self0 ->
    begin match hash_map_len_fwd t self0 with
    | Fail e -> Fail e
    | Return i ->
      if i > self0.hash_map_max_load
      then hash_map_try_resize_fwd_back t self0
      else Return self0
    end
  end

(** [hashmap::HashMap::{0}::contains_key_in_list] *)
let rec hash_map_contains_key_in_list_loop_fwd
  (t : Type0) (i : usize) (ls : list_t t) :
  Tot (result bool)
  (decreases (hash_map_contains_key_in_list_decreases t i ls))
  =
  begin match ls with
  | ListCons ckey x tl ->
    if ckey = i
    then Return true
    else hash_map_contains_key_in_list_loop_fwd t i tl
  | ListNil -> Return false
  end

(** [hashmap::HashMap::{0}::contains_key_in_list] *)
let hash_map_contains_key_in_list_fwd
  (t : Type0) (key : usize) (ls : list_t t) : result bool =
  hash_map_contains_key_in_list_loop_fwd t key ls

(** [hashmap::HashMap::{0}::contains_key] *)
let hash_map_contains_key_fwd
  (t : Type0) (self : hash_map_t t) (key : usize) : result bool =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_fwd (list_t t) self.hash_map_slots hash_mod with
      | Fail e -> Fail e
      | Return l -> hash_map_contains_key_in_list_fwd t key l
      end
    end
  end

(** [hashmap::HashMap::{0}::get_in_list] *)
let rec hash_map_get_in_list_loop_fwd
  (t : Type0) (i : usize) (ls : list_t t) :
  Tot (result t) (decreases (hash_map_get_in_list_decreases t i ls))
  =
  begin match ls with
  | ListCons ckey cvalue tl ->
    if ckey = i then Return cvalue else hash_map_get_in_list_loop_fwd t i tl
  | ListNil -> Fail Failure
  end

(** [hashmap::HashMap::{0}::get_in_list] *)
let hash_map_get_in_list_fwd
  (t : Type0) (key : usize) (ls : list_t t) : result t =
  hash_map_get_in_list_loop_fwd t key ls

(** [hashmap::HashMap::{0}::get] *)
let hash_map_get_fwd
  (t : Type0) (self : hash_map_t t) (key : usize) : result t =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_fwd (list_t t) self.hash_map_slots hash_mod with
      | Fail e -> Fail e
      | Return l -> hash_map_get_in_list_fwd t key l
      end
    end
  end

(** [hashmap::HashMap::{0}::get_mut_in_list] *)
let rec hash_map_get_mut_in_list_loop_fwd
  (t : Type0) (i : usize) (ls : list_t t) :
  Tot (result t) (decreases (hash_map_get_mut_in_list_decreases t i ls))
  =
  begin match ls with
  | ListCons ckey cvalue tl ->
    if ckey = i
    then Return cvalue
    else hash_map_get_mut_in_list_loop_fwd t i tl
  | ListNil -> Fail Failure
  end

(** [hashmap::HashMap::{0}::get_mut_in_list] *)
let hash_map_get_mut_in_list_fwd
  (t : Type0) (ls : list_t t) (key : usize) : result t =
  hash_map_get_mut_in_list_loop_fwd t key ls

(** [hashmap::HashMap::{0}::get_mut_in_list] *)
let rec hash_map_get_mut_in_list_loop_back
  (t : Type0) (i : usize) (ls : list_t t) (ret : t) :
  Tot (result (list_t t))
  (decreases (hash_map_get_mut_in_list_decreases t i ls))
  =
  begin match ls with
  | ListCons ckey cvalue tl ->
    if ckey = i
    then Return (ListCons ckey ret tl)
    else
      begin match hash_map_get_mut_in_list_loop_back t i tl ret with
      | Fail e -> Fail e
      | Return l -> Return (ListCons ckey cvalue l)
      end
  | ListNil -> Fail Failure
  end

(** [hashmap::HashMap::{0}::get_mut_in_list] *)
let hash_map_get_mut_in_list_back
  (t : Type0) (ls : list_t t) (key : usize) (ret : t) : result (list_t t) =
  hash_map_get_mut_in_list_loop_back t key ls ret

(** [hashmap::HashMap::{0}::get_mut] *)
let hash_map_get_mut_fwd
  (t : Type0) (self : hash_map_t t) (key : usize) : result t =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_mut_fwd (list_t t) self.hash_map_slots hash_mod
        with
      | Fail e -> Fail e
      | Return l -> hash_map_get_mut_in_list_fwd t l key
      end
    end
  end

(** [hashmap::HashMap::{0}::get_mut] *)
let hash_map_get_mut_back
  (t : Type0) (self : hash_map_t t) (key : usize) (ret : t) :
  result (hash_map_t t)
  =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_mut_fwd (list_t t) self.hash_map_slots hash_mod
        with
      | Fail e -> Fail e
      | Return l ->
        begin match hash_map_get_mut_in_list_back t l key ret with
        | Fail e -> Fail e
        | Return l0 ->
          begin match
            vec_index_mut_back (list_t t) self.hash_map_slots hash_mod l0 with
          | Fail e -> Fail e
          | Return v ->
            Return (Mkhash_map_t self.hash_map_num_entries
              self.hash_map_max_load_factor self.hash_map_max_load v)
          end
        end
      end
    end
  end

(** [hashmap::HashMap::{0}::remove_from_list] *)
let rec hash_map_remove_from_list_loop_fwd
  (t : Type0) (i : usize) (ls : list_t t) :
  Tot (result (option t))
  (decreases (hash_map_remove_from_list_decreases t i ls))
  =
  begin match ls with
  | ListCons ckey x tl ->
    if ckey = i
    then
      let mv_ls = mem_replace_fwd (list_t t) (ListCons ckey x tl) ListNil in
      begin match mv_ls with
      | ListCons i0 cvalue tl0 -> Return (Some cvalue)
      | ListNil -> Fail Failure
      end
    else hash_map_remove_from_list_loop_fwd t i tl
  | ListNil -> Return None
  end

(** [hashmap::HashMap::{0}::remove_from_list] *)
let hash_map_remove_from_list_fwd
  (t : Type0) (key : usize) (ls : list_t t) : result (option t) =
  hash_map_remove_from_list_loop_fwd t key ls

(** [hashmap::HashMap::{0}::remove_from_list] *)
let rec hash_map_remove_from_list_loop_back
  (t : Type0) (i : usize) (ls : list_t t) :
  Tot (result (list_t t))
  (decreases (hash_map_remove_from_list_decreases t i ls))
  =
  begin match ls with
  | ListCons ckey x tl ->
    if ckey = i
    then
      let mv_ls = mem_replace_fwd (list_t t) (ListCons ckey x tl) ListNil in
      begin match mv_ls with
      | ListCons i0 cvalue tl0 -> Return tl0
      | ListNil -> Fail Failure
      end
    else
      begin match hash_map_remove_from_list_loop_back t i tl with
      | Fail e -> Fail e
      | Return l -> Return (ListCons ckey x l)
      end
  | ListNil -> Return ListNil
  end

(** [hashmap::HashMap::{0}::remove_from_list] *)
let hash_map_remove_from_list_back
  (t : Type0) (key : usize) (ls : list_t t) : result (list_t t) =
  hash_map_remove_from_list_loop_back t key ls

(** [hashmap::HashMap::{0}::remove] *)
let hash_map_remove_fwd
  (t : Type0) (self : hash_map_t t) (key : usize) : result (option t) =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_mut_fwd (list_t t) self.hash_map_slots hash_mod
        with
      | Fail e -> Fail e
      | Return l ->
        begin match hash_map_remove_from_list_fwd t key l with
        | Fail e -> Fail e
        | Return x ->
          begin match x with
          | None -> Return None
          | Some x0 ->
            begin match usize_sub self.hash_map_num_entries 1 with
            | Fail e -> Fail e
            | Return _ -> Return (Some x0)
            end
          end
        end
      end
    end
  end

(** [hashmap::HashMap::{0}::remove] *)
let hash_map_remove_back
  (t : Type0) (self : hash_map_t t) (key : usize) : result (hash_map_t t) =
  begin match hash_key_fwd key with
  | Fail e -> Fail e
  | Return hash ->
    let i = vec_len (list_t t) self.hash_map_slots in
    begin match usize_rem hash i with
    | Fail e -> Fail e
    | Return hash_mod ->
      begin match vec_index_mut_fwd (list_t t) self.hash_map_slots hash_mod
        with
      | Fail e -> Fail e
      | Return l ->
        begin match hash_map_remove_from_list_fwd t key l with
        | Fail e -> Fail e
        | Return x ->
          begin match x with
          | None ->
            begin match hash_map_remove_from_list_back t key l with
            | Fail e -> Fail e
            | Return l0 ->
              begin match
                vec_index_mut_back (list_t t) self.hash_map_slots hash_mod l0
                with
              | Fail e -> Fail e
              | Return v ->
                Return (Mkhash_map_t self.hash_map_num_entries
                  self.hash_map_max_load_factor self.hash_map_max_load v)
              end
            end
          | Some x0 ->
            begin match usize_sub self.hash_map_num_entries 1 with
            | Fail e -> Fail e
            | Return i0 ->
              begin match hash_map_remove_from_list_back t key l with
              | Fail e -> Fail e
              | Return l0 ->
                begin match
                  vec_index_mut_back (list_t t) self.hash_map_slots hash_mod l0
                  with
                | Fail e -> Fail e
                | Return v ->
                  Return (Mkhash_map_t i0 self.hash_map_max_load_factor
                    self.hash_map_max_load v)
                end
              end
            end
          end
        end
      end
    end
  end

(** [hashmap::test1] *)
let test1_fwd : result unit =
  begin match hash_map_new_fwd u64 with
  | Fail e -> Fail e
  | Return hm ->
    begin match hash_map_insert_fwd_back u64 hm 0 42 with
    | Fail e -> Fail e
    | Return hm0 ->
      begin match hash_map_insert_fwd_back u64 hm0 128 18 with
      | Fail e -> Fail e
      | Return hm1 ->
        begin match hash_map_insert_fwd_back u64 hm1 1024 138 with
        | Fail e -> Fail e
        | Return hm2 ->
          begin match hash_map_insert_fwd_back u64 hm2 1056 256 with
          | Fail e -> Fail e
          | Return hm3 ->
            begin match hash_map_get_fwd u64 hm3 128 with
            | Fail e -> Fail e
            | Return i ->
              if not (i = 18)
              then Fail Failure
              else
                begin match hash_map_get_mut_back u64 hm3 1024 56 with
                | Fail e -> Fail e
                | Return hm4 ->
                  begin match hash_map_get_fwd u64 hm4 1024 with
                  | Fail e -> Fail e
                  | Return i0 ->
                    if not (i0 = 56)
                    then Fail Failure
                    else
                      begin match hash_map_remove_fwd u64 hm4 1024 with
                      | Fail e -> Fail e
                      | Return x ->
                        begin match x with
                        | None -> Fail Failure
                        | Some x0 ->
                          if not (x0 = 56)
                          then Fail Failure
                          else
                            begin match hash_map_remove_back u64 hm4 1024 with
                            | Fail e -> Fail e
                            | Return hm5 ->
                              begin match hash_map_get_fwd u64 hm5 0 with
                              | Fail e -> Fail e
                              | Return i1 ->
                                if not (i1 = 42)
                                then Fail Failure
                                else
                                  begin match hash_map_get_fwd u64 hm5 128 with
                                  | Fail e -> Fail e
                                  | Return i2 ->
                                    if not (i2 = 18)
                                    then Fail Failure
                                    else
                                      begin match hash_map_get_fwd u64 hm5 1056
                                        with
                                      | Fail e -> Fail e
                                      | Return i3 ->
                                        if not (i3 = 256)
                                        then Fail Failure
                                        else Return ()
                                      end
                                  end
                              end
                            end
                        end
                      end
                  end
                end
            end
          end
        end
      end
    end
  end

(** Unit test for [hashmap::test1] *)
let _ = assert_norm (test1_fwd = Return ())