(*
    Author:   Salomon Sickert
    Author:   Benedikt Seidl
    Author:   Alexander Schimpf (original entry: CAVA/LTL_Rewrite.thy)
    License:  BSD
*)

section ‹Rewrite Rules for LTL Simplification›

theory Rewriting
imports
  LTL "HOL-Library.Extended_Nat"
begin

text ‹This theory provides rewrite rules for the simplification of LTL formulas. It supports:
 \begin{itemize}
 \item Constants Removal
 \item @{const Next_ltln}-Normalisation
 \item Modal Simplification (based on pure eventual, pure universal, or suspendable formulas)
 \item Syntactic Implication Checking
 \end{itemize}
 It reuses parts of LTL\_Rewrite.thy (CAVA, LTL\_TO\_GBA). Furthermore, some rules were taken from
 cite‹"DBLP:conf/cav/SomenziB00"› and cite‹"DBLP:conf/tacas/BabiakKRS12"›. All functions are defined for @{type ltln}.›



subsection ‹Constant Eliminating Constructors›

definition mk_and
where
  "mk_and x y ≡ case x of false_n → false_n | true_n → y | _ → (case y of false_n → false_n | true_n → x | _ → x and_n y)"

definition mk_or
where
  "mk_or x y ≡ case x of false_n → y | true_n → true_n | _ → (case y of true_n → true_n | false_n → x | _ → x or_n y)"

fun remove_strong_ops
where
  "remove_strong_ops (x U_n y) = remove_strong_ops y"
| "remove_strong_ops (x M_n y) = x and_n y"
| "remove_strong_ops (x or_n y) = remove_strong_ops x or_n remove_strong_ops y"
| "remove_strong_ops x = x"

fun remove_weak_ops
where
  "remove_weak_ops (x R_n y) = remove_weak_ops y"
| "remove_weak_ops (x W_n y) = x or_n y"
| "remove_weak_ops (x and_n y) = remove_weak_ops x and_n remove_weak_ops y"
| "remove_weak_ops x = x"

definition mk_finally
where
  "mk_finally x ≡ case x of true_n → true_n | false_n → false_n | _ → F_n (remove_strong_ops x)"

definition mk_globally
where
  "mk_globally x ≡ case x of true_n → true_n | false_n → false_n | _ → G_n (remove_weak_ops x)"

definition mk_until
where
  "mk_until x y ≡ case x of false_n → y
    | true_n → mk_finally y
    | _ → (case y of true_n → true_n | false_n → false_n | _ → x U_n y)"

definition mk_release
where
  "mk_release x y ≡ case x of true_n → y
    | false_n → mk_globally y
    | _ → (case y of true_n → true_n | false_n → false_n | _ → x R_n y)"

definition mk_weak_until
where
  "mk_weak_until x y ≡ case y of true_n → true_n
    | false_n → mk_globally x
    | _ → (case x of true_n → true_n | false_n → y | _ → x W_n y)"

definition mk_strong_release
where
  "mk_strong_release x y ≡ case y of false_n → false_n
    | true_n → mk_finally x
    | _ → (case x of true_n → y | false_n → false_n | _ → x M_n y)"

definition mk_next
where
  "mk_next x ≡ case x of true_n → true_n | false_n → false_n | _ → X_n x"

definition mk_next_pow (‹X_n''›)
where
  "mk_next_pow n x ≡ case x of true_n → true_n | false_n → false_n | _ → (Next_ltln ^^ n) x"



lemma mk_and_semantics [simp]:
  "w ⊨_n mk_and x y ⟷ w ⊨_n x and_n y"
  unfolding mk_and_def by (cases x; cases y; simp)

lemma mk_or_semantics [simp]:
  "w ⊨_n mk_or x y ⟷ w ⊨_n x or_n y"
  unfolding mk_or_def by (cases x; cases y; simp)

lemma remove_strong_ops_sound [simp]:
  "w ⊨_n F_n (remove_strong_ops y) ⟷ w ⊨_n F_n y"
  by (induction y arbitrary: w) (auto; force)+

lemma remove_weak_ops_sound [simp]:
  "w ⊨_n G_n (remove_weak_ops y) ⟷ w ⊨_n G_n y"
  by (induction y arbitrary: w) (auto; force)+

lemma mk_finally_semantics [simp]:
  "w ⊨_n mk_finally x ⟷ w ⊨_n F_n x"
  by (simp add: mk_finally_def del: semantics_ltln.simps(8,9) remove_strong_ops.simps split: ltln.splits)

lemma mk_globally_semantics [simp]:
  "w ⊨_n mk_globally x ⟷ w ⊨_n G_n x"
  by (simp add: mk_globally_def del: semantics_ltln.simps(8,9) remove_weak_ops.simps split: ltln.splits)

lemma mk_until_semantics [simp]:
  "w ⊨_n mk_until x y ⟷ w ⊨_n x U_n y"
proof (cases x)
  case (True_ltln)
    show ?thesis
      unfolding True_ltln mk_until_def
      by (cases y) auto
next
  case (False_ltln)
    thus ?thesis
      by (force simp: mk_until_def)
qed (cases y; force simp: mk_until_def)+

lemma mk_release_semantics [simp]:
  "w ⊨_n mk_release x y ⟷ w ⊨_n x R_n y"
proof (cases x)
  case (False_ltln)
    thus ?thesis
      unfolding False_ltln mk_release_def
      by (cases y) auto
next
  case (True_ltln)
    thus ?thesis
      by (force simp: mk_release_def)
qed (cases y; force simp: mk_release_def)+

lemma mk_weak_until_semantics [simp]:
  "w ⊨_n mk_weak_until x y ⟷ w ⊨_n x W_n y"
proof (cases y)
  case (False_ltln)
    thus ?thesis
      unfolding False_ltln mk_weak_until_def
      by (cases x) auto
next
  case (True_ltln)
    thus ?thesis
      by (force simp: mk_weak_until_def)
qed (cases x; force simp: mk_weak_until_def)+

lemma mk_strong_release_semantics [simp]:
  "w ⊨_n mk_strong_release x y ⟷ w ⊨_n x M_n y"
proof (cases y)
  case (True_ltln)
    show ?thesis
      unfolding True_ltln mk_strong_release_def
      by (cases x) auto
next
  case (False_ltln)
    thus ?thesis
      by (force simp: mk_strong_release_def)
qed (cases x; force simp: mk_strong_release_def)+

lemma mk_next_semantics [simp]:
  "w ⊨_n mk_next x ⟷ w ⊨_n X_n x"
  unfolding mk_next_def by (cases x; auto)

lemma mk_next_pow_semantics [simp]:
  "w ⊨_n mk_next_pow i x ⟷ suffix i w ⊨_n x"
  by (induction i arbitrary: w; cases x)
     (auto simp: mk_next_pow_def)

lemma mk_next_pow_simp [simp, code_unfold]:
  "mk_next_pow 0 x = x"
  "mk_next_pow 1 x = mk_next x"
  by (cases x; simp add: mk_next_pow_def mk_next_def)+



subsection ‹Constant Propagation›

fun is_constant :: "'a ltln → bool"
where
  "is_constant true_n = True"
| "is_constant false_n = True"
| "is_constant _ = False"

lemma is_constant_constructorsI:
  "is_constant x ==> is_constant y ==> is_constant (mk_and x y)"
  "¬is_constant x ==> ¬is_constant y ==> ¬is_constant (mk_and x y)"
  "is_constant x ==> is_constant y ==> is_constant (mk_or x y)"
  "¬is_constant x ==> ¬is_constant y ==> ¬is_constant (mk_or x y)"
  "is_constant x ==> is_constant (mk_finally x)"
  "¬is_constant x ==> ¬is_constant (mk_finally x)"
  "is_constant x ==> is_constant (mk_globally x)"
  "¬is_constant x ==> ¬is_constant (mk_globally x)"
  "is_constant x ==> is_constant (mk_until y x)"
  "¬is_constant x ==> ¬is_constant (mk_until y x)"
  "is_constant x ==> is_constant (mk_release y x)"
  "¬is_constant x ==> ¬is_constant (mk_release y x)"
  "is_constant x ==> is_constant y ==> is_constant (mk_weak_until x y)"
  "¬is_constant x ==> ¬is_constant y ==> ¬is_constant (mk_weak_until x y)"
  "is_constant x ==> is_constant y ==> is_constant (mk_strong_release x y)"
  "¬is_constant x ==> ¬is_constant y ==> ¬is_constant (mk_strong_release x y)"
  "is_constant x ==> is_constant (mk_next x)"
  "¬is_constant x ==> ¬is_constant (mk_next x)"
  "is_constant x ==> is_constant (mk_next_pow n x)"
  by (cases x; cases y; simp add: mk_and_def mk_or_def mk_finally_def mk_globally_def mk_until_def mk_release_def mk_weak_until_def mk_strong_release_def mk_next_def mk_next_pow_def)+

lemma is_constant_constructors_simps:
  "mk_next_pow n x = false_n ⟷ x = false_n"
  "mk_next_pow n x = true_n ⟷ x = true_n"
  "is_constant (mk_next_pow n x) ⟷ is_constant x"
  by (induction n) (cases x; simp add: mk_next_pow_def)+

lemma is_constant_constructors_simps2:
  "is_constant (mk_and x y) ⟷ (x = true_n ∧ y = true_n ∨ x = false_n ∨ y = false_n)"
  "is_constant (mk_or x y) ⟷ (x = false_n ∧ y = false_n ∨ x = true_n ∨ y = true_n)"
  "is_constant (mk_finally x) ⟷ is_constant x"
  "is_constant (mk_globally x) ⟷ is_constant x"
  "is_constant (mk_until y x) ⟷ is_constant x"
  "is_constant (mk_release y x) ⟷ is_constant x"
  "is_constant (mk_next x) ⟷ is_constant x"
  by ((cases x; cases y; simp add: mk_and_def),
      (cases x; cases y; simp add: mk_or_def),
      (meson is_constant_constructorsI)+)

lemma is_constant_constructors_simps3:
  "is_constant (mk_weak_until x y) ⟷ (x = false_n ∧ y = false_n ∨ x = true_n ∨ y = true_n)"
  "is_constant (mk_strong_release x y) ⟷ (x = true_n ∧ y = true_n ∨ x = false_n ∨ y = false_n)"
  by (cases x; cases y; simp add: mk_weak_until_def mk_strong_release_def is_constant_constructors_simps2)+

lemma is_constant_semantics:
  "is_constant φ ==> ((∀w. w ⊨_n φ) ∨ ¬(∃w. w ⊨_n φ))"
  by (cases φ) auto

lemma until_constant_simp:
  "is_constant ψ ==> w ⊨_n φ U_n ψ ⟷ w ⊨_n ψ"
  by (cases ψ) auto

lemma release_constant_simp:
  "is_constant ψ ==> w ⊨_n φ R_n ψ ⟷ w ⊨_n ψ"
  by (cases ψ) auto

lemma mk_next_pow_dist:
  "mk_next_pow (i + j) φ = mk_next_pow i (mk_next_pow j φ)"
  by (cases j; simp) (cases φ; simp add: mk_next_pow_def funpow_add; simp add: funpow_swap1)

lemma mk_next_pow_until:
  "suffix (min i j) w ⊨_n (mk_next_pow (i - j) φ) U_n (mk_next_pow (j - i) ψ) ⟷ w ⊨_n (mk_next_pow i φ) U_n (mk_next_pow j ψ)"
  by (simp add: mk_next_pow_dist min_def add.commute)

lemma mk_next_pow_release:
  "suffix (min i j) w ⊨_n (mk_next_pow (i - j) φ) R_n (mk_next_pow (j - i) ψ) ⟷ w ⊨_n (mk_next_pow i φ) R_n (mk_next_pow j ψ)"
  by (simp add: mk_next_pow_dist min_def add.commute)

subsection ‹X-Normalisation›

text ‹The following rewrite functions pulls the X-operator up in the syntax tree. This preprocessing
 step enables the removal of X-operators in front of suspendable formulas. Furthermore constants are
 removed as far as possible.›

fun the_enat_0 :: "enat → nat"
where
  "the_enat_0 i = i"
| "the_enat_0 ∞ = 0"

lemma the_enat_0_simp [simp]:
  "the_enat_0 0 = 0"
  "the_enat_0 1 = 1"
  by (simp add: zero_enat_def one_enat_def)+

fun combine :: "('a ltln → 'a ltln → 'a ltln) → ('a ltln * enat) → ('a ltln * enat) → ('a ltln * enat)"
where
  "combine binop (φ, i) (ψ, j) = (
    let
      χ = binop (mk_next_pow (the_enat_0 (i - j)) φ) (mk_next_pow (the_enat_0 (j - i)) ψ)
    in
      (χ, if is_constant χ then ∞ else min i j))"

lemma fst_combine:
  "fst (combine binop (φ, i) (ψ, j)) = binop (mk_next_pow (the_enat_0 (i - j)) φ) (mk_next_pow (the_enat_0 (j - i)) ψ)"
  unfolding combine.simps by (meson fstI)

abbreviation to_ltln :: "('a ltln * enat) → 'a ltln"
where
  "to_ltln x ≡ mk_next_pow (the_enat_0 (snd x)) (fst x)"

fun rewrite_X_enat :: "'a ltln → ('a ltln * enat)"
where
  "rewrite_X_enat true_n = (true_n, ∞)"
| "rewrite_X_enat false_n = (false_n, ∞)"
| "rewrite_X_enat prop_n(a) = (prop_n(a), 0)"
| "rewrite_X_enat nprop_n(a) = (nprop_n(a), 0)"
| "rewrite_X_enat (φ and_n ψ) = combine mk_and (rewrite_X_enat φ) (rewrite_X_enat ψ)"
| "rewrite_X_enat (φ or_n ψ) = combine mk_or (rewrite_X_enat φ) (rewrite_X_enat ψ)"
| "rewrite_X_enat (φ U_n ψ) = combine mk_until (rewrite_X_enat φ) (rewrite_X_enat ψ)"
| "rewrite_X_enat (φ R_n ψ) = combine mk_release (rewrite_X_enat φ) (rewrite_X_enat ψ)"
| "rewrite_X_enat (φ W_n ψ) = combine mk_weak_until (rewrite_X_enat φ) (rewrite_X_enat ψ)"
| "rewrite_X_enat (φ M_n ψ) = combine mk_strong_release (rewrite_X_enat φ) (rewrite_X_enat ψ)"
| "rewrite_X_enat (X_n φ) = (λ(φ, n). (φ, eSuc n)) (rewrite_X_enat φ)"

definition
  "rewrite_X φ = to_ltln (rewrite_X_enat φ)"

lemma combine_infinity_invariant:
  assumes "i = ∞ ⟷ is_constant x"
  assumes "j = ∞ ⟷ is_constant y"
  shows "combine mk_and (x, i) (y, j) = (z, k) ==> (k = ∞ ⟷ is_constant z)"
    and "combine mk_or (x, i) (y, j) = (z, k) ==> (k = ∞ ⟷ is_constant z)"
    and "combine mk_until (x, i) (y, j) = (z, k) ==> (k = ∞ ⟷ is_constant z)"
    and "combine mk_release (x, i) (y, j) = (z, k) ==> (k = ∞ ⟷ is_constant z)"
    and "combine mk_weak_until (x, i) (y, j) = (z, k) ==> (k = ∞ ⟷ is_constant z)"
    and "combine mk_strong_release (x, i) (y, j) = (z, k) ==> (k = ∞ ⟷ is_constant z)"
  by (cases i; cases j; simp add: assms Let_def; force intro: is_constant_constructorsI)+

lemma combine_and_or_semantics:
  assumes "i = ∞ ⟷ is_constant φ"
  assumes "j = ∞ ⟷ is_constant ψ"
  shows "w ⊨_n to_ltln (combine mk_and (φ, i) (ψ, j)) ⟷ w ⊨_n to_ltln (φ, i) and_n to_ltln (ψ, j)"
    and "w ⊨_n to_ltln (combine mk_or (φ, i) (ψ, j)) ⟷ w ⊨_n to_ltln (φ, i) or_n to_ltln (ψ, j)"
  by ((cases i; cases j; simp add: min_def is_constant_constructors_simps is_constant_constructors_simps2 assms),
      (cases ψ; insert assms; auto),
      (cases φ; insert assms; auto),
      (blast elim!: is_constant.elims))+

lemma combine_until_release_semantics:
  assumes "i = ∞ ⟷ is_constant φ"
  assumes "j = ∞ ⟷ is_constant ψ"
  shows "w ⊨_n to_ltln (combine mk_until (φ, i) (ψ, j)) ⟷ w ⊨_n to_ltln (φ, i) U_n to_ltln (ψ, j)"
    and "w ⊨_n to_ltln (combine mk_release (φ, i) (ψ, j)) ⟷ w ⊨_n to_ltln (φ, i) R_n to_ltln (ψ, j)"
  by ((cases i; cases j; simp add: is_constant_constructors_simps is_constant_constructors_simps2
       until_constant_simp release_constant_simp mk_next_pow_until mk_next_pow_release del: semantics_ltln.simps),
      (blast dest: is_constant_semantics),
      (cases ψ; simp add: assms),
      (cases φ; insert assms; auto simp: add.commute))+

(* TODO this proof is a bit slow and could be optimized *)
lemma combine_weak_until_strong_release_semantics:
  assumes "i = ∞ ⟷ is_constant φ"
  assumes "j = ∞ ⟷ is_constant ψ"
  shows "w ⊨_n to_ltln (combine mk_weak_until (φ, i) (ψ, j)) ⟷ w ⊨_n to_ltln (φ, i) W_n to_ltln (ψ, j)"
    and "w ⊨_n to_ltln (combine mk_strong_release (φ, i) (ψ, j)) ⟷ w ⊨_n to_ltln (φ, i) M_n to_ltln (ψ, j)"
  by ((cases i; cases j; simp add: min_def is_constant_constructors_simps is_constant_constructors_simps3 del: semantics_ltln.simps),
      (cases φ; simp add: assms),
      (cases ψ; insert assms; auto simp: add.commute))+



lemma rewrite_X_enat_infinity_invariant:
  "snd (rewrite_X_enat φ) = ∞ ⟷ is_constant (fst (rewrite_X_enat φ))"
proof (induction φ)
  case (And_ltln φ ψ)
    thus ?case
      by (simp add: combine_infinity_invariant[OF And_ltln(1,2), unfolded prod.collapse])
next
  case (Or_ltln φ ψ)
    thus ?case
      by (simp add: combine_infinity_invariant[OF Or_ltln(1,2), unfolded prod.collapse])
next
  case (Until_ltln φ ψ)
    thus ?case
      by (simp add: combine_infinity_invariant[OF Until_ltln(1,2), unfolded prod.collapse])
next
  case (Release_ltln φ ψ)
    thus ?case
      by (simp add: combine_infinity_invariant[OF Release_ltln(1,2), unfolded prod.collapse])
next
  case (WeakUntil_ltln φ ψ)
    thus ?case
      by (simp add: combine_infinity_invariant[OF WeakUntil_ltln(1,2), unfolded prod.collapse])
next
  case (StrongRelease_ltln φ ψ)
    thus ?case
      by (simp add: combine_infinity_invariant[OF StrongRelease_ltln(1,2), unfolded prod.collapse])
next
  case (Next_ltln φ)
    thus ?case
      by (simp add: split_def) (metis eSuc_infinity eSuc_inject)
qed auto

lemma rewrite_X_enat_correct:
  "w ⊨_n φ ⟷ w ⊨_n to_ltln (rewrite_X_enat φ)"
proof (induction φ arbitrary: w)
  case (And_ltln φ ψ)
    thus ?case
      using combine_and_or_semantics[OF rewrite_X_enat_infinity_invariant rewrite_X_enat_infinity_invariant] by fastforce
next
  case (Or_ltln φ ψ)
    thus ?case
      using combine_and_or_semantics[OF rewrite_X_enat_infinity_invariant rewrite_X_enat_infinity_invariant] by fastforce
next
  case (Until_ltln φ ψ)
    thus ?case
      unfolding rewrite_X_enat.simps combine_until_release_semantics[OF rewrite_X_enat_infinity_invariant rewrite_X_enat_infinity_invariant, unfolded prod.collapse] by fastforce
next
  case (Release_ltln φ ψ)
    thus ?case
      unfolding rewrite_X_enat.simps combine_until_release_semantics[OF rewrite_X_enat_infinity_invariant rewrite_X_enat_infinity_invariant, unfolded prod.collapse] by fastforce
next
  case (WeakUntil_ltln φ ψ)
    thus ?case
      unfolding rewrite_X_enat.simps combine_weak_until_strong_release_semantics[OF rewrite_X_enat_infinity_invariant rewrite_X_enat_infinity_invariant, unfolded prod.collapse] by fastforce
next
  case (StrongRelease_ltln φ ψ)
    thus ?case
      unfolding rewrite_X_enat.simps combine_weak_until_strong_release_semantics[OF rewrite_X_enat_infinity_invariant rewrite_X_enat_infinity_invariant, unfolded prod.collapse] by fastforce
next
  case (Next_ltln φ)
  moreover
    have " w ⊨_n to_ltln (rewrite_X_enat (X_n φ)) ⟷ suffix 1 w ⊨_n to_ltln (rewrite_X_enat φ)"
      by (simp add: split_def; cases "snd (rewrite_X_enat φ) ≠ ∞")
         (auto simp: eSuc_def, auto simp: rewrite_X_enat_infinity_invariant eSuc_def dest: is_constant_semantics)
    ultimately
    show ?case
       using semantics_ltln.simps(7) by blast
qed auto

lemma rewrite_X_sound [simp]:
  "w ⊨_n rewrite_X φ ⟷ w ⊨_n φ"
  using rewrite_X_enat_correct unfolding rewrite_X_def Let_def by auto

subsection ‹Pure Eventual, Pure Universal, and Suspendable Formulas›

fun pure_eventual :: "'a ltln → bool"
where
  "pure_eventual true_n = True"
| "pure_eventual false_n = True"
| "pure_eventual (μ and_n μ') = (pure_eventual μ ∧ pure_eventual μ')"
| "pure_eventual (μ or_n μ') = (pure_eventual μ ∧ pure_eventual μ')"
| "pure_eventual (μ U_n μ') = (μ = true_n ∨ pure_eventual μ')"
| "pure_eventual (μ R_n μ') = (pure_eventual μ ∧ pure_eventual μ')"
| "pure_eventual (μ W_n μ') = (pure_eventual μ ∧ pure_eventual μ')"
| "pure_eventual (μ M_n μ') = (pure_eventual μ ∧ pure_eventual μ' ∨ μ' = true_n)"
| "pure_eventual (X_n μ) = pure_eventual μ"
| "pure_eventual _ = False"

fun pure_universal :: "'a ltln → bool"
where
  "pure_universal true_n = True"
| "pure_universal false_n = True"
| "pure_universal (ν and_n ν') = (pure_universal ν ∧ pure_universal ν')"
| "pure_universal (ν or_n ν') = (pure_universal ν ∧ pure_universal ν')"
| "pure_universal (ν U_n ν') = (pure_universal ν ∧ pure_universal ν')"
| "pure_universal (ν R_n ν') = (ν = false_n ∨ pure_universal ν')"
| "pure_universal (ν W_n ν') = (pure_universal ν ∧ pure_universal ν' ∨ ν' = false_n)"
| "pure_universal (ν M_n ν') = (pure_universal ν ∧ pure_universal ν')"
| "pure_universal (X_n ν) = pure_universal ν"
| "pure_universal _ = False"

fun suspendable :: "'a ltln → bool"
where
  "suspendable true_n = True"
| "suspendable false_n = True"
| "suspendable (ξ and_n ξ') = (suspendable ξ ∧ suspendable ξ')"
| "suspendable (ξ or_n ξ') = (suspendable ξ ∧ suspendable ξ')"
| "suspendable (φ U_n ξ) = (φ = true_n ∧ pure_universal ξ ∨ suspendable ξ)"
| "suspendable (φ R_n ξ) = (φ = false_n ∧ pure_eventual ξ ∨ suspendable ξ)"
| "suspendable (φ W_n ξ) = (suspendable φ ∧ suspendable ξ ∨ pure_eventual φ ∧ ξ = false_n)"
| "suspendable (φ M_n ξ) = (suspendable φ ∧ suspendable ξ ∨ pure_universal φ ∧ ξ = true_n)"
| "suspendable (X_n ξ) = suspendable ξ"
| "suspendable _ = False"

lemma pure_eventual_left_append:
  "pure_eventual μ ==> w ⊨_n μ ==> (u ⌢ w) ⊨_n μ"
proof (induction μ arbitrary: u w)
  case (Until_ltln μ μ')
    moreover
    then obtain i where "suffix i w ⊨_n μ'"
      by auto
    hence "μ = true_n ==> ?case"
      by simp (metis suffix_conc_length suffix_suffix)
    moreover
    have "pure_eventual μ' ==> (u ⌢ w) ⊨_n μ'"
      by (metis ‹suffix i w ⊨_n μ'› Until_ltln(2) prefix_suffix)
    hence "pure_eventual μ' ==> ?case"
      by force
    ultimately
    show ?case
      by fastforce
next
  case (Release_ltln μ μ')
    thus ?case
      by (cases "∀i. suffix i w ⊨_n μ'"; simp_all)
         (metis linear suffix_conc_snd gr0I not_less0 prefix_suffix suffix_0)+
next
  case (WeakUntil_ltln μ μ')
    thus ?case
      by (cases "∀i. suffix i w ⊨_n μ'"; simp_all)
         (metis zero_le le0 nat_le_linear prefix_suffix suffix_0 suffix_conc_length suffix_conc_snd suffix_subseq_join)+
next
  case (StrongRelease_ltln μ μ')
    moreover
    then obtain i where "suffix i w ⊨_n μ and_n μ'"
      by auto
    hence "μ' = true_n ==> ?case"
      by simp (metis suffix_conc_length suffix_suffix)
    moreover
    have "pure_eventual μ ==> pure_eventual μ' ==> (u ⌢ w) ⊨_n μ and_n μ'"
      by (metis ‹suffix i w ⊨_n μ and_n μ'› calculation(1) calculation(2) prefix_suffix semantics_ltln.simps(5))
    hence "pure_eventual μ ==> pure_eventual μ' ==> ?case"
      by force
    ultimately
    show ?case
      by fastforce
qed (auto, metis diff_zero le_0_eq not_less_eq_eq suffix_conc_length suffix_conc_snd word_split)

lemma pure_universal_suffix_closed:
  "pure_universal ν ==> (u ⌢ w) ⊨_n ν ==> w ⊨_n ν"
proof (induction ν arbitrary: u w)
  case (Until_ltln ν ν')
    hence "∃i. suffix i (u ⌢ w) ⊨_n ν' ∧ (∀j<i. suffix j (u ⌢ w) ⊨_n ν)"
      using semantics_ltln.simps(8) by blast
    thus ?case
      by simp (metis Until_ltln(1-3) le_0_eq le_eq_less_or_eq le_less_linear prefix_suffix pure_universal.simps(5) suffix_conc_fst suffix_conc_snd)
next
  case (Release_ltln ν ν')
    moreover
    hence "∀i. suffix i (u ⌢ w) ⊨_n ν' ∨ (∃j<i. suffix j (u ⌢ w) ⊨_n ν)"
      by simp
    ultimately
    show ?case
      by simp (metis semantics_ltln.simps(2) not_less0 prefix_suffix suffix_0 suffix_conc_length suffix_suffix)
next
  case (WeakUntil_ltln ν ν')
    moreover
    hence "∀i. suffix i (u ⌢ w) ⊨_n ν ∨ (∃j≤i. suffix j (u ⌢ w) ⊨_n ν')"
      by simp
    ultimately
    show ?case
      by simp (metis (full_types) le_antisym prefix_suffix semantics_ltln.simps(2) suffix_0 suffix_conc_length suffix_suffix zero_le)
next
  case (StrongRelease_ltln ν ν')
    hence "∃i. suffix i (u ⌢ w) ⊨_n ν ∧ (∀j≤i. suffix j (u ⌢ w) ⊨_n ν')"
      using semantics_ltln.simps(11) by blast
    thus ?case
      by simp (metis StrongRelease_ltln(1-3) diff_is_0_eq nat_le_linear prefix_conc_length prefix_suffix pure_universal.simps(8) subsequence_length suffix_conc_snd suffix_subseq_join)
next
  case (Next_ltln μ)
    thus ?case
      by (metis prefix_suffix pure_universal.simps(9) semantics_ltln.simps(7) semiring_normalization_rules(24) suffix_conc_length suffix_suffix)
qed auto

lemma suspendable_prefix_invariant:
  "suspendable ξ ==> (u ⌢ w) ⊨_n ξ ⟷ w ⊨_n ξ"
proof (induction ξ arbitrary: u w)
  case (Until_ltln ξ ξ')
    show ?case
    proof (cases "suspendable ξ'")
      case False
        hence "ξ = true_n" and "pure_universal ξ'"
          using Until_ltln by simp+
        thus ?thesis
          by (simp; metis (no_types) linear pure_universal_suffix_closed suffix_conc_fst suffix_conc_length suffix_conc_snd suffix_suffix)
    qed (simp; metis Until_ltln(2) not_less0 prefix_suffix)
next
  case (Release_ltln ξ ξ')
    show ?case
    proof (cases "suspendable ξ'")
      case False
        hence "ξ = false_n" and "pure_eventual ξ'"
          using Release_ltln by simp+
        thus ?thesis
          by (simp; metis (no_types) le_iff_add add_diff_cancel_left' linear pure_eventual_left_append suffix_0 suffix_conc_fst suffix_conc_snd)
    qed (simp; metis Release_ltln(2) not_less0 prefix_suffix)
next
  case (WeakUntil_ltln ξ ξ')
    show ?case
    proof (cases "suspendable ξ ∧ suspendable ξ'")
      case False
        hence "ξ' = false_n" and "pure_eventual ξ"
          using WeakUntil_ltln by simp+
        thus ?thesis
          by (simp; metis (no_types) le_iff_add add_diff_cancel_left' linear pure_eventual_left_append suffix_0 suffix_conc_fst suffix_conc_snd)
      qed (simp; metis (full_types) WeakUntil_ltln.IH prefix_suffix)
next
  case (StrongRelease_ltln ξ ξ')
    show ?case
    proof (cases "suspendable ξ ∧ suspendable ξ'")
      case False
        hence "ξ' = true_n" and "pure_universal ξ"
          using StrongRelease_ltln by simp+
        thus ?thesis
          by (simp; metis (no_types) linear pure_universal_suffix_closed suffix_conc_fst suffix_conc_length suffix_conc_snd suffix_suffix)
    qed (simp; metis (full_types) StrongRelease_ltln.IH(1) StrongRelease_ltln.IH(2) prefix_suffix)
qed (simp_all, metis prefix_suffix)

theorem pure_eventual_until_simp:
  assumes "pure_eventual μ"
  shows "w ⊨_n φ U_n μ ⟷ w ⊨_n μ"
proof -
  have "∧i. suffix i w ⊨_n μ ==> w ⊨_n μ"
    using pure_eventual_left_append[OF assms] prefix_suffix by metis
  thus ?thesis
    by force
qed

theorem pure_universal_release_simp:
  assumes "pure_universal ν"
  shows "w ⊨_n φ R_n ν ⟷ w ⊨_n ν"
proof -
  have "∧i. w ⊨_n ν ==> suffix i w ⊨_n ν"
    using pure_universal_suffix_closed[OF assms] prefix_suffix by metis
  thus ?thesis
    by force
qed

theorem pure_universal_weak_until_simp:
  assumes "pure_universal φ" and "pure_universal ψ"
  shows "w ⊨_n φ W_n ψ ⟷ w ⊨_n φ or_n ψ"
proof -
  have "∧i. w ⊨_n φ ==> suffix i w ⊨_n φ" and "∧i. w ⊨_n ψ ==> suffix i w ⊨_n ψ"
    using assms pure_universal_suffix_closed prefix_suffix by metis+
  thus ?thesis
    by force
qed

theorem pure_eventual_strong_release_simp:
  assumes "pure_eventual φ" and "pure_eventual ψ"
  shows "w ⊨_n φ M_n ψ ⟷ w ⊨_n φ and_n ψ"
proof -
  have "∧i. suffix i w ⊨_n φ ==> w ⊨_n φ" and "∧i. suffix i w ⊨_n ψ ==> w ⊨_n ψ"
    using assms pure_eventual_left_append prefix_suffix by metis+
  thus ?thesis
    by force
qed


theorem suspendable_formula_simp:
  assumes "suspendable ξ"
  shows "w ⊨_n X_n ξ ⟷ w ⊨_n ξ" (is ?t1)
    and "w ⊨_n φ U_n ξ ⟷ w ⊨_n ξ" (is ?t2)
    and "w ⊨_n φ R_n ξ ⟷ w ⊨_n ξ" (is ?t3)
proof -
  have "∧i. suffix i w ⊨_n ξ ⟷ w ⊨_n ξ"
    using suspendable_prefix_invariant[OF assms] prefix_suffix by metis
  thus ?t1 ?t2 ?t3
    by force+
qed

theorem suspendable_formula_simp2:
  assumes "suspendable φ" and "suspendable ψ"
  shows "w ⊨_n φ W_n ψ ⟷ w ⊨_n φ or_n ψ" (is ?t1)
    and "w ⊨_n φ M_n ψ ⟷ w ⊨_n φ and_n ψ" (is ?t2)
proof -
  have "∧i. suffix i w ⊨_n φ ⟷ w ⊨_n φ" and "∧i. suffix i w ⊨_n ψ ⟷ w ⊨_n ψ"
    using assms suspendable_prefix_invariant prefix_suffix by metis+
  thus ?t1 ?t2
    by force+
qed

fun rewrite_modal :: "'a ltln → 'a ltln"
where
  "rewrite_modal true_n = true_n"
| "rewrite_modal false_n = false_n"
| "rewrite_modal (φ and_n ψ) = (rewrite_modal φ and_n rewrite_modal ψ)"
| "rewrite_modal (φ or_n ψ) = (rewrite_modal φ or_n rewrite_modal ψ)"
| "rewrite_modal (φ U_n ψ) = (if pure_eventual ψ ∨ suspendable ψ then rewrite_modal ψ else (rewrite_modal φ U_n rewrite_modal ψ))"
| "rewrite_modal (φ R_n ψ) = (if pure_universal ψ ∨ suspendable ψ then rewrite_modal ψ else (rewrite_modal φ R_n rewrite_modal ψ))"
| "rewrite_modal (φ W_n ψ) = (if pure_universal φ ∧ pure_universal ψ ∨ suspendable φ ∧ suspendable ψ then (rewrite_modal φ or_n rewrite_modal ψ) else (rewrite_modal φ W_n rewrite_modal ψ))"
| "rewrite_modal (φ M_n ψ) = (if pure_eventual φ ∧ pure_eventual ψ ∨ suspendable φ ∧ suspendable ψ then (rewrite_modal φ and_n rewrite_modal ψ) else (rewrite_modal φ M_n rewrite_modal ψ))"
| "rewrite_modal (X_n φ) = (if suspendable φ then rewrite_modal φ else X_n (rewrite_modal φ))"
| "rewrite_modal φ = φ"

lemma rewrite_modal_sound [simp]:
  "w ⊨_n rewrite_modal φ ⟷ w ⊨_n φ"
proof (induction φ arbitrary: w)
  case (Until_ltln φ ψ)
    thus ?case
      apply (cases "pure_eventual ψ ∨ suspendable ψ")
      apply (insert pure_eventual_until_simp[of ψ] suspendable_formula_simp[of ψ])
      apply fastforce+
      done
next
  case (Release_ltln φ ψ)
    thus ?case
      apply (cases "pure_universal ψ ∨ suspendable ψ")
      apply (insert pure_universal_release_simp[of ψ] suspendable_formula_simp[of ψ])
      apply fastforce+
      done
next
  case (WeakUntil_ltln φ ψ)
    thus ?case
      apply (cases "pure_universal φ ∧ pure_universal ψ ∨ suspendable φ ∧ suspendable ψ")
      apply (insert pure_universal_weak_until_simp[of φ ψ] suspendable_formula_simp2[of φ ψ])
      apply fastforce+
      done
next
  case (StrongRelease_ltln φ ψ)
    thus ?case
      apply (cases "pure_eventual φ ∧ pure_eventual ψ ∨ suspendable φ ∧ suspendable ψ")
      apply (insert pure_eventual_strong_release_simp[of φ ψ] suspendable_formula_simp2[of φ ψ])
      apply fastforce+
      done
next
  case (Next_ltln φ)
    thus ?case
      apply (cases "suspendable φ")
      apply (insert suspendable_formula_simp[of φ])
      apply fastforce+
      done
qed auto

lemma rewrite_modal_size:
  "size (rewrite_modal φ) ≤ size φ"
  by (induction φ) auto

subsection ‹Syntactical Implication Based Simplification›

inductive syntactical_implies :: "'a ltln → 'a ltln → bool" (‹_ ⊨_s _› [80, 80])
where
  "_ ⊨_s true_n"
| "false_n ⊨_s _"
| "φ = φ ==> φ ⊨_s φ"

| "φ ⊨_s χ ==> (φ and_n ψ) ⊨_s χ"
| "ψ ⊨_s χ ==> (φ and_n ψ) ⊨_s χ"
| "φ ⊨_s ψ ==> φ ⊨_s χ ==> φ ⊨_s (ψ and_n χ)"

| "φ ⊨_s ψ ==> φ ⊨_s (ψ or_n χ)"
| "φ ⊨_s χ ==> φ ⊨_s (ψ or_n χ)"
| "φ ⊨_s χ ==> ψ ⊨_s χ ==> (φ or_n ψ) ⊨_s χ"

| "φ ⊨_s χ ==> φ ⊨_s (ψ U_n χ)"
| "φ ⊨_s χ ==> ψ ⊨_s χ ==> (φ U_n ψ) ⊨_s χ"
| "φ ⊨_s χ ==> ψ ⊨_s ν ==> (φ U_n ψ) ⊨_s (χ U_n ν)"

| "χ ⊨_s φ ==> (ψ R_n χ) ⊨_s φ"
| "χ ⊨_s φ ==> χ ⊨_s ψ ==> χ ⊨_s (φ R_n ψ)"
| "φ ⊨_s χ ==> ψ ⊨_s ν ==> (φ R_n ψ) ⊨_s (χ R_n ν)"

| "(false_n R_n φ) ⊨_s ψ ==> (false_n R_n φ) ⊨_s X_n ψ"
| "φ ⊨_s (true_n U_n ψ) ==> (X_n φ) ⊨_s (true_n U_n ψ)"
| "φ ⊨_s ψ ==> (X_n φ) ⊨_s (X_n ψ)"

lemma syntactical_implies_correct:
  "φ ⊨_s ψ ==> w ⊨_n φ ==> w ⊨_n ψ"
  by (induction arbitrary: w rule: syntactical_implies.induct; auto; force)

fun rewrite_syn_imp
where
  "rewrite_syn_imp (φ and_n ψ) = (
    if φ ⊨_s ψ then
      rewrite_syn_imp φ
    else if ψ ⊨_s φ then
      rewrite_syn_imp ψ
    else if φ ⊨_s (not_n ψ) ∨ ψ ⊨_s (not_n φ) then
      false_n
    else
      mk_and (rewrite_syn_imp φ) (rewrite_syn_imp ψ))"
| "rewrite_syn_imp (φ or_n ψ) = (
    if φ ⊨_s ψ then
      rewrite_syn_imp ψ
    else if ψ ⊨_s φ then
      rewrite_syn_imp φ
    else if (not_n φ) ⊨_s ψ ∨ (not_n ψ) ⊨_s φ then
      true_n
    else
      mk_or (rewrite_syn_imp φ) (rewrite_syn_imp ψ))"
| "rewrite_syn_imp (φ U_n ψ) = (
    if φ ⊨_s ψ then
      rewrite_syn_imp ψ
    else if (not_n φ) ⊨_s ψ then
      mk_finally (rewrite_syn_imp ψ)
    else
      mk_until (rewrite_syn_imp φ) (rewrite_syn_imp ψ))"
| "rewrite_syn_imp (φ R_n ψ) = (
    if ψ ⊨_s φ then
      rewrite_syn_imp ψ
    else if ψ ⊨_s (not_n φ) then
      mk_globally (rewrite_syn_imp ψ)
    else
      mk_release (rewrite_syn_imp φ) (rewrite_syn_imp ψ))"
| "rewrite_syn_imp (X_n φ) = mk_next (rewrite_syn_imp φ)"
| "rewrite_syn_imp φ = φ"

lemma rewrite_syn_imp_sound:
  "w ⊨_n rewrite_syn_imp φ ⟷ w ⊨_n φ"
proof (induction φ arbitrary: w)
  case And_ltln
    thus ?case
      by (simp add: Let_def; metis syntactical_implies_correct not_{n_semantics})
next
  case (Or_ltln φ ψ)
    moreover
    have "(not_n φ) ⊨_s ψ ==> ∀w. w ⊨_n φ or_n ψ"
      by (auto intro: syntactical_implies_correct[of "not_n φ"])
    moreover
    have "(not_n ψ) ⊨_s φ ==> ∀w. w ⊨_n φ or_n ψ"
      by (auto intro: syntactical_implies_correct[of "not_n ψ"])
    ultimately
    show ?case
      by (auto intro: syntactical_implies_correct)
next
  case (Until_ltln φ ψ)
    moreover
    have "φ ⊨_s ψ ==> ?case"
      by (force simp add: Until_ltln dest: syntactical_implies_correct)
    moreover
    {
      assume A: "(not_n φ) ⊨_s ψ" and B: "¬ φ ⊨_s ψ"
      hence [simp]: "rewrite_syn_imp (φ U_n ψ) = mk_finally (rewrite_syn_imp ψ)"
        by simp
      {
        assume "∃i. suffix i w ⊨_n ψ"
        moreover
        define i where "i ≡ LEAST i. suffix i w ⊨_n ψ"
        ultimately
        have "∀j < i. ¬suffix j w ⊨_n ψ" and "suffix i w ⊨_n ψ"
          by (blast dest: not_less_Least , metis LeastI ‹∃i. suffix i w ⊨_n ψ› i_def)
        hence "∀j < i. suffix j w ⊨_n φ" and "suffix i w ⊨_n ψ"
          using syntactical_implies_correct[OF A] by auto
      }
      hence ?case
        by (simp del: rewrite_syn_imp.simps; unfold Until_ltln(2)) blast
    }
    ultimately
    show ?case
      by fastforce
next
  case (Release_ltln φ ψ)
    moreover
    have "ψ ⊨_s φ ==> ?case"
      by (force simp add: Release_ltln dest: syntactical_implies_correct)
    moreover
    {
      assume A: "ψ ⊨_s (not_n φ)" and B: "¬ ψ ⊨_s φ"
      hence [simp]: "rewrite_syn_imp (φ R_n ψ) = mk_globally (rewrite_syn_imp ψ)"
        by simp
      {
        assume "∃i. ¬suffix i w ⊨_n ψ"
        moreover
        define i where "i ≡ LEAST i. ¬suffix i w ⊨_n ψ"
        ultimately
        have "∀j < i. suffix j w ⊨_n ψ" and "¬ suffix i w ⊨_n ψ"
          by (blast dest: not_less_Least , metis LeastI ‹∃i. ¬suffix i w ⊨_n ψ› i_def)
        hence "∀j < i. ¬suffix j w ⊨_n φ" and "¬ suffix i w ⊨_n ψ"
          using syntactical_implies_correct[OF A] by auto
      }
      hence ?case
        by (simp del: rewrite_syn_imp.simps; unfold Release_ltln(2)) blast
    }
    ultimately
    show ?case
      by fastforce
qed auto

subsection ‹Iterated Rewriting›

fun iterate
where
  "iterate f x 0 = x"
| "iterate f x (Suc n) = (let x' = f x in if x = x' then x else iterate f x' n)"

definition
  "rewrite_iter_fast φ ≡ iterate (rewrite_modal o rewrite_X) φ (size φ)"

definition
  "rewrite_iter_slow φ ≡ iterate (rewrite_syn_imp o rewrite_modal o rewrite_X) φ (size φ)"

text ‹The rewriting functions defined in the previous subsections can be used as-is. However, in the
 most cases one wants to iterate these rules until the formula cannot be simplified further.
 @{const rewrite_iter_fast} pulls X operators up in the syntax tree and the uses the
 "modal" simplification rules. @{const rewrite_iter_slow} additionally tries to simplify the
 formula using syntactic implication checking.›

lemma iterate_sound:
  assumes "∧φ. w ⊨_n f φ ⟷ w ⊨_n φ"
  shows "w ⊨_n iterate f φ n ⟷ w ⊨_n φ"
  by (induction n arbitrary: φ; simp add: assms Let_def)

theorem rewrite_iter_fast_sound [simp]:
  "w ⊨_n rewrite_iter_fast φ ⟷ w ⊨_n φ"
  using iterate_sound[of _ "rewrite_modal o rewrite_X"]
  unfolding comp_def rewrite_modal_sound rewrite_X_sound rewrite_iter_fast_def
  by blast

theorem rewrite_iter_slow_sound [simp]:
  "w ⊨_n rewrite_iter_slow φ ⟷ w ⊨_n φ"
  using iterate_sound[of _ "rewrite_syn_imp o rewrite_modal o rewrite_X"]
  unfolding comp_def rewrite_modal_sound rewrite_X_sound rewrite_syn_imp_sound rewrite_iter_slow_def
  by blast

subsection ‹Preservation of atoms›

lemma iterate_atoms:
  assumes
    "∧φ. atoms_ltln (f φ) ⊆ atoms_ltln φ"
  shows
    "atoms_ltln (iterate f φ n) ⊆ atoms_ltln φ"
  by (induction n arbitrary: φ) (auto, metis (mono_tags, lifting) assms in_mono)

lemma rewrite_modal_atoms:
  "atoms_ltln (rewrite_modal φ) ⊆ atoms_ltln φ"
  by (induction φ) auto

lemma mk_and_atoms:
  "atoms_ltln (mk_and φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  by (auto simp: mk_and_def split: ltln.splits)

lemma mk_or_atoms:
  "atoms_ltln (mk_or φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  by (auto simp: mk_or_def split: ltln.splits)

lemma remove_strong_ops_atoms:
  "atoms_ltln (remove_strong_ops φ) ⊆ atoms_ltln φ"
  by (induction φ) auto

lemma remove_weak_ops_atoms:
  "atoms_ltln (remove_weak_ops φ) ⊆ atoms_ltln φ"
  by (induction φ) auto

lemma mk_finally_atoms:
  "atoms_ltln (mk_finally φ) ⊆ atoms_ltln φ"
  by (auto simp: mk_finally_def split: ltln.splits) (insert remove_strong_ops_atoms, fast+)

lemma mk_globally_atoms:
  "atoms_ltln (mk_globally φ) ⊆ atoms_ltln φ"
  by (auto simp: mk_globally_def split: ltln.splits) (insert remove_weak_ops_atoms, fast+)

lemma mk_until_atoms:
  "atoms_ltln (mk_until φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  by (auto simp: mk_until_def split: ltln.splits) (insert mk_finally_atoms, fastforce+)

lemma mk_release_atoms:
  "atoms_ltln (mk_release φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  by (auto simp: mk_release_def split: ltln.splits) (insert mk_globally_atoms, fastforce+)

lemma mk_weak_until_atoms:
  "atoms_ltln (mk_weak_until φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  by (auto simp: mk_weak_until_def split: ltln.splits) (insert mk_globally_atoms, fastforce+)

lemma mk_strong_release_atoms:
  "atoms_ltln (mk_strong_release φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  by (auto simp: mk_strong_release_def split: ltln.splits) (insert mk_finally_atoms, fastforce+)

lemma mk_next_atoms:
  "atoms_ltln (mk_next φ) = atoms_ltln φ"
  by (auto simp: mk_next_def split: ltln.splits)

lemma mk_next_pow_atoms:
  "atoms_ltln (mk_next_pow n φ) = atoms_ltln φ"
  by (induction n) (auto simp: mk_next_pow_def split: ltln.splits)

lemma combine_atoms:
  assumes
    "∧φ ψ. atoms_ltln (f φ ψ) ⊆ atoms_ltln φ ∪ atoms_ltln ψ"
  shows
    "atoms_ltln (fst (combine f x y)) ⊆ atoms_ltln (fst x) ∪ atoms_ltln (fst y)"
  by (metis assms fst_combine mk_next_pow_atoms prod.collapse)

lemmas combine_mk_atoms =
  combine_atoms[OF mk_and_atoms]
  combine_atoms[OF mk_or_atoms]
  combine_atoms[OF mk_until_atoms]
  combine_atoms[OF mk_release_atoms]
  combine_atoms[OF mk_weak_until_atoms]
  combine_atoms[OF mk_strong_release_atoms]

lemma rewrite_X_enat_atoms:
  "atoms_ltln (fst (rewrite_X_enat φ)) ⊆ atoms_ltln φ"
  by (induction φ) (simp_all add: case_prod_beta, insert combine_mk_atoms, fast+)

lemma rewrite_X_atoms:
  "atoms_ltln (rewrite_X φ) ⊆ atoms_ltln φ"
  by (induction φ) (simp_all add: rewrite_X_def mk_next_pow_atoms case_prod_beta, insert combine_mk_atoms, fast+)

lemma rewrite_syn_imp_atoms:
  "atoms_ltln (rewrite_syn_imp φ) ⊆ atoms_ltln φ"
proof (induction φ)
  case (And_ltln φ1 φ2)
  then show ?case
    using mk_and_atoms by simp fast
next
  case (Or_ltln φ1 φ2)
  then show ?case
    using mk_or_atoms by simp fast
next
  case (Next_ltln φ)
  then show ?case
    using mk_next_atoms by simp fast
next
  case (Until_ltln φ1 φ2)
  then show ?case
    using mk_finally_atoms mk_until_atoms by simp fast
next
  case (Release_ltln φ1 φ2)
  then show ?case
    using mk_globally_atoms mk_release_atoms by simp fast
qed simp_all

lemma rewrite_iter_fast_atoms:
  "atoms_ltln (rewrite_iter_fast φ) ⊆ atoms_ltln φ"
proof -
  have 1: "∧φ. atoms_ltln (rewrite_modal (rewrite_X φ)) ⊆ atoms_ltln φ"
    using rewrite_modal_atoms rewrite_X_atoms by force

  show ?thesis
    by (simp add: rewrite_iter_fast_def 1 iterate_atoms)
qed

lemma rewrite_iter_slow_atoms:
  "atoms_ltln (rewrite_iter_slow φ) ⊆ atoms_ltln φ"
proof -
  have 1: "∧φ. atoms_ltln (rewrite_syn_imp (rewrite_modal (rewrite_X φ))) ⊆ atoms_ltln φ"
    using rewrite_syn_imp_atoms rewrite_modal_atoms rewrite_X_atoms by force

  show ?thesis
    by (simp add: rewrite_iter_slow_def 1 iterate_atoms)
qed

subsection ‹Simplifier›

text ‹We now define a convenience wrapper for the rewriting engine›

datatype mode = Nop | Fast | Slow

fun simplify :: "mode → 'a ltln → 'a ltln"
where
  "simplify Nop = id"
| "simplify Fast = rewrite_iter_fast"
| "simplify Slow = rewrite_iter_slow"

theorem simplify_correct:
  "w ⊨_n simplify m φ ⟷ w ⊨_n φ"
  by (cases m) simp+

lemma simplify_atoms:
  "atoms_ltln (simplify m φ) ⊆ atoms_ltln φ"
  by (cases m) (insert rewrite_iter_fast_atoms rewrite_iter_slow_atoms, fastforce+)

subsection ‹Code Generation›

code_pred syntactical_implies .

export_code simplify checking

lemma "rewrite_iter_fast (F_n (G_n (X_n prop_n(''a'')))) = (F_n (G_n prop_n(''a'')))"
  by eval

lemma "rewrite_iter_fast (X_n prop_n(''a'') U_n (X_n nprop_n(''a''))) = X_n (prop_n(''a'') U_n nprop_n(''a''))"
  by eval

lemma "rewrite_iter_slow (X_n prop_n(''a'') U_n (X_n nprop_n(''a''))) = X_n (F_n nprop_n(''a''))"
  by eval

end