Maps: Total and Partial Maps
The Coq Standard Library
From Coq Require Import Arith.Arith.
From Coq Require Import Bool.Bool.
Require Export Coq.Strings.String.
From Coq Require Import Logic.FunctionalExtensionality.
From Coq Require Import Lists.List.
Import ListNotations.
Documentation for the standard library can be found at
    http://coq.inria.fr/library/.
 
    The Search command is a good way to look for theorems involving
    objects of specific types.  Take a minute now to experiment with it. 
Identifiers
(The function string_dec comes from Coq's string library.
    If you check the result type of string_dec, you'll see that it
    does not actually return a bool, but rather a type that looks
    like {x = y} + {x <> y}, called a sumbool, which can be
    thought of as an "evidence-carrying boolean."  Formally, an
    element of sumbool is either a proof that two things are equal
    or a proof that they are unequal, together with a tag indicating
    which.  But for present purposes you can think of it as just a
    fancy bool.) 
 
 Now we need a few basic properties of string equality... 
Theorem eqb_string_refl : forall s : string, true = eqb_string s s.
Proof. intros s. unfold eqb_string. destruct (string_dec s s) as [|Hs].
- reflexivity.
- destruct Hs. reflexivity.
Qed.
Proof. intros s. unfold eqb_string. destruct (string_dec s s) as [|Hs].
- reflexivity.
- destruct Hs. reflexivity.
Qed.
The following useful property follows from an analogous
    lemma about strings: 
Theorem eqb_string_true_iff : forall x y : string,
eqb_string x y = true <-> x = y.
Proof.
intros x y.
unfold eqb_string.
destruct (string_dec x y) as [|Hs].
- subst. split. reflexivity. reflexivity.
- split.
+ intros contra. discriminate contra.
+ intros H. rewrite H in Hs. destruct Hs. reflexivity.
Qed.
Similarly: 
Theorem eqb_string_false_iff : forall x y : string,
eqb_string x y = false <-> x <> y.
Proof.
intros x y. rewrite <- eqb_string_true_iff.
rewrite not_true_iff_false. reflexivity. Qed.
This handy variant follows just by rewriting: 
Theorem false_eqb_string : forall x y : string,
x <> y -> eqb_string x y = false.
Proof.
intros x y. rewrite eqb_string_false_iff.
intros H. apply H. Qed.
Total Maps
Intuitively, a total map over an element type A is just a
    function that can be used to look up strings, yielding As. 
 
 The function t_empty yields an empty total map, given a default
    element; this map always returns the default element when applied
    to any string. 
More interesting is the update function, which (as before) takes
    a map m, a key x, and a value v and returns a new map that
    takes x to v and takes every other key to whatever m does. 
Definition t_update {A : Type} (m : total_map A)
(x : string) (v : A) :=
fun x' => if eqb_string x x' then v else m x'.
This definition is a nice example of higher-order programming:
    t_update takes a function m and yields a new function
    fun x' => ... that behaves like the desired map. 
 
 For example, we can build a map taking strings to bools, where
    "foo" and "bar" are mapped to true and every other key is
    mapped to false, like this: 
Next, let's introduce some new notations to facilitate working
    with maps. 
 
 First, we will use the following notation to create an empty
    total map with a default value. 
Notation "'_' '!->' v" := (t_empty v)
(at level 100, right associativity).
Example example_empty := (_ !-> false).
(at level 100, right associativity).
Example example_empty := (_ !-> false).
We then introduce a convenient notation for extending an existing
    map with some bindings. 
Notation "x '!->' v ';' m" := (t_update m x v)
(at level 100, v at next level, right associativity).
(at level 100, v at next level, right associativity).
The examplemap above can now be defined as follows: 
This completes the definition of total maps.  Note that we
    don't need to define a find operation because it is just
    function application! 
Example update_example1 : examplemap' "baz" = false.
Proof. reflexivity. Qed.
Example update_example2 : examplemap' "foo" = true.
Proof. reflexivity. Qed.
Example update_example3 : examplemap' "quux" = false.
Proof. reflexivity. Qed.
Example update_example4 : examplemap' "bar" = true.
Proof. reflexivity. Qed.
To use maps in later chapters, we'll need several fundamental
    facts about how they behave. 
 
 Even if you don't work the following exercises, make sure
    you thoroughly understand the statements of the lemmas! 
 
 (Some of the proofs require the functional extensionality axiom,
    which is discussed in the Logic chapter.) 
 
 
    First, the empty map returns its default element for all keys: 
Exercise: 1 star, standard, optional (t_apply_empty)
☐ 
 
 
    Next, if we update a map m at a key x with a new value v
    and then look up x in the map resulting from the update, we
    get back v: 
Exercise: 2 stars, standard, optional (t_update_eq)
☐ 
 
 
    On the other hand, if we update a map m at a key x1 and then
    look up a different key x2 in the resulting map, we get the
    same result that m would have given: 
Exercise: 2 stars, standard, optional (t_update_neq)
Theorem t_update_neq : forall (A : Type) (m : total_map A) x1 x2 v,
x1 <> x2 ->
(x1 !-> v ; m) x2 = m x2.
Proof.
Admitted.
☐ 
 
 
    If we update a map m at a key x with a value v1 and then
    update again with the same key x and another value v2, the
    resulting map behaves the same (gives the same result when applied
    to any key) as the simpler map obtained by performing just
    the second update on m: 
Exercise: 2 stars, standard, optional (t_update_shadow)
Lemma t_update_shadow : forall (A : Type) (m : total_map A) x v1 v2,
(x !-> v2 ; x !-> v1 ; m) = (x !-> v2 ; m).
Proof.
Admitted.
☐ 
 
 For the final two lemmas about total maps, it's convenient to use
    the reflection idioms introduced in chapter IndProp.  We begin
    by proving a fundamental reflection lemma relating the equality
    proposition on ids with the boolean function eqb_id. 
 
 
    Use the proof of eqbP in chapter IndProp as a template to
    prove the following: 
Exercise: 2 stars, standard, optional (eqb_stringP)
☐ 
 
 Now, given strings x1 and x2, we can use the tactic
    destruct (eqb_stringP x1 x2) to simultaneously perform case
    analysis on the result of eqb_string x1 x2 and generate
    hypotheses about the equality (in the sense of =) of x1
    and x2. 
 
 
    With the example in chapter IndProp as a template, use
    eqb_stringP to prove the following theorem, which states that
    if we update a map to assign key x the same value as it already
    has in m, then the result is equal to m: 
Exercise: 2 stars, standard (t_update_same)
Theorem t_update_same : forall (A : Type) (m : total_map A) x,
(x !-> m x ; m) = m.
Proof.
Admitted.
☐ 
 
 
    Use eqb_stringP to prove one final property of the update
    function: If we update a map m at two distinct keys, it doesn't
    matter in which order we do the updates. 
Exercise: 3 stars, standard, recommended (t_update_permute)
Theorem t_update_permute : forall (A : Type) (m : total_map A)
v1 v2 x1 x2,
x2 <> x1 ->
(x1 !-> v1 ; x2 !-> v2 ; m)
=
(x2 !-> v2 ; x1 !-> v1 ; m).
Proof.
Admitted.
☐ 
Partial maps
Definition partial_map (A : Type) := total_map (option A).
Definition empty {A : Type} : partial_map A :=
t_empty None.
Definition update {A : Type} (m : partial_map A)
(x : string) (v : A) :=
(x !-> Some v ; m).
We introduce a similar notation for partial maps: 
We can also hide the last case when it is empty. 
Notation "x '|->' v" := (update empty x v)
(at level 100).
Example examplepmap :=
("Church" |-> true ; "Turing" |-> false).
(at level 100).
Example examplepmap :=
("Church" |-> true ; "Turing" |-> false).
We now straightforwardly lift all of the basic lemmas about total
    maps to partial maps.  
Lemma apply_empty : forall (A : Type) (x : string),
@empty A x = None.
Proof.
intros. unfold empty. rewrite t_apply_empty.
reflexivity.
Qed.
Lemma update_eq : forall (A : Type) (m : partial_map A) x v,
(x |-> v ; m) x = Some v.
Proof.
intros. unfold update. rewrite t_update_eq.
reflexivity.
Qed.
Theorem update_neq : forall (A : Type) (m : partial_map A) x1 x2 v,
x2 <> x1 ->
(x2 |-> v ; m) x1 = m x1.
Proof.
intros A m x1 x2 v H.
unfold update. rewrite t_update_neq. reflexivity.
apply H. Qed.
Lemma update_shadow : forall (A : Type) (m : partial_map A) x v1 v2,
(x |-> v2 ; x |-> v1 ; m) = (x |-> v2 ; m).
Proof.
intros A m x v1 v2. unfold update. rewrite t_update_shadow.
reflexivity.
Qed.
Theorem update_same : forall (A : Type) (m : partial_map A) x v,
m x = Some v ->
(x |-> v ; m) = m.
Proof.
intros A m x v H. unfold update. rewrite <- H.
apply t_update_same.
Qed.
Theorem update_permute : forall (A : Type) (m : partial_map A)
x1 x2 v1 v2,
x2 <> x1 ->
(x1 |-> v1 ; x2 |-> v2 ; m) = (x2 |-> v2 ; x1 |-> v1 ; m).
Proof.
intros A m x1 x2 v1 v2. unfold update.
apply t_update_permute.
Qed.
