Require Import Facts States ICSyntax.
Set Implicit Arguments.

Module ICSemantics (Sigma : State).
Module ICSyn := ICSyntax.ICSyntax Sigma.
Export ICSyn.

Implicit Types (P : Pred state) (Q QQ : NPred state) (x y z : state).
Implicit Types (a : action) (b : guard) (s t : term).

Definition conf := (term × state)%type.
Inductive step : conf → conf → Prop :=
| step_act a s x :
    step (Act a s, x) (s, a x)
| step_def s t x:
    step (Def s t, x) (t.[Def s s/], x)
| step_if_true b s t x :
    b x →
    step (If b s t, x) (s, x)
| step_if_false b s t x :
    ~~b x →
    step (If b s t, x) (t, x).
Notation mstep := (star step).

Inductive eval Q : term → Pred state :=
| eval_act a s x :
    eval Q s (a x) →
    eval Q (Act a s) x
| eval_if_true b s t x :
    b x →
    eval Q s x →
    eval Q (If b s t) x
| eval_if_false b s t x :
    ~~b x →
    eval Q t x →
    eval Q (If b s t) x
| eval_jump f x :
    Q f x →
    eval Q (Jump f) x
| eval_def P s t x :
    P <<= eval Q (Def s s) →
    eval (P .: Q) t x →
    eval Q (Def s t) x.

Fixpoint wp Q s :=
  match s with
    | Act a s ⇒ wp Q s \o a
    | If b s t ⇒ fun sigma ⇒ if b sigma then wp Q s sigma else wp Q t sigma
    | Jump f ⇒ Q f
    | Def s t ⇒
      let F P := wp (P .: Q) s in
      wp (Fix F .: Q) t
  end.

Lemma wp_mono Q QQ :
  (∀ n, Q n <<= QQ n) → (∀ s, wp Q s <<= wp QQ s).

Lemma wp_equiv Q QQ s x :
  (∀ n x, Q n x ↔ QQ n x) → (wp Q s x ↔ wp QQ s x).

Lemma wp_cons_mono Q s :
  monotone (fun P ⇒ wp (P .: Q) s).

Lemma wp_fix P Q s :
  wp (P .: Q) s <<= P → Fix (fun P ⇒ wp (P .: Q) s) <<= P.

Lemma wp_eval Q s :
  wp Q s <<= eval Q s.

Lemma eval_wp Q s :
  eval Q s <<= wp Q s.

Theorem coincidence Q s x : eval Q s x ↔ wp Q s x.

Lemma wp_ren Q (xi : var → var) s :
  wp Q s.[ren xi] = wp (xi >>> Q) s.

Lemma wp_weaken P Q s : wp (P .: Q) s.[ren (+1)] = wp Q s.

Lemma wp_subst Q (theta : var → term) s :
  wp Q s.[theta] = wp (wp Q \o theta) s.

Lemma step_admissible Q s t x y :
  step (s,x) (t,y) → (wp Q s x ↔ wp Q t y).

Lemma mstep_admissible Q f s x y :
  mstep (s,x) (Jump f,y) → (wp Q s x ↔ Q f y).

Definition terminates s x :=
  ∃ f y, mstep (s,x) (Jump f,y).

Lemma step_terminates s t x y :
  step (s,x) (t,y) → terminates t y → terminates s x.

Lemma eval_terminates Q s x :
  eval Q s x → ∀ theta,
    (∀ f, Q f <<= terminates (theta f)) →
    terminates s.[theta] x.

Corollary eval_terminates_id Q s :
  eval Q s <<= terminates s.

Theorem correspondence Q s x :
  wp Q s x ↔ ∃ f y, mstep (s,x) (Jump f,y) ∧ Q f y.

Definition point_pred f x : NPred state := fun g y ⇒ f = g ∧ x = y.

Corollary wp_step_equiv f s x y :
  mstep (s,x) (Jump f,y) ↔ wp (point_pred f y) s x.

Corollary wp_terminates s x :
  wp (@Top _) s x ↔ terminates s x.

Corollary wp_det Q s x :
  wp Q s x ↔ ∃ f y, wp (point_pred f y) s x ∧ Q f y.

End ICSemantics.