{-# OPTIONS --rewriting #-}

module A201801.AllVecLemmas where

open import A201801.Prelude
open import A201801.Category
open import A201801.Fin
open import A201801.FinLemmas
open import A201801.Vec
open import A201801.VecLemmas
open import A201801.AllVec


--------------------------------------------------------------------------------
{-
                      get (gets ξ η) i ≡ (get ξ ∘ ren∋ η) i                     comp-get-ren∋

                            gets ξ id⊇ ≡ ξ                                      id-gets
                      gets ξ (η₁ ∘ η₂) ≡ gets (gets ξ η₂) η₁                    comp-gets
                                                                                𝐠𝐞𝐭𝐬

                            id⊇◇ ∘⊇◇ 𝜂 ≡ 𝜂                                      lid∘⊇◇
                            𝜂 ∘⊇◇ id⊇◇ ≡ 𝜂                                      rid∘⊇◇
                    (𝜂₁ ∘⊇◇ 𝜂₂) ∘⊇◇ 𝜂₃ ≡ 𝜂₁ ∘⊇◇ (𝜂₂ ∘⊇◇ 𝜂₃)                     assoc∘⊇◇
                                                                                𝐎𝐏𝐄◇

                          ren∋◇ id⊇◇ 𝑖 ≡ 𝑖                                      id-ren∋◇
                   ren∋◇ (𝜂₁ ∘⊇◇ 𝜂₂) 𝑖 ≡ (ren∋◇ 𝜂₂ ∘ ren∋◇ 𝜂₁) 𝑖                comp-ren∋◇
                                                                                𝐫𝐞𝐧∋◇
-}
--------------------------------------------------------------------------------


comp-get-ren∋ : ∀ {X P A n n′} → {Ξ : Vec X n} {Ξ′ : Vec X n′} {e : n′ ≥ n} {I : Fin n}
                               → (ξ : All P Ξ′) (η : Ξ′ ⊇⟨ e ⟩ Ξ) (i : Ξ ∋⟨ I ⟩ A)
                               → get (gets ξ η) i ≡ (get ξ ∘ ren∋ η) i
comp-get-ren∋ ∙       done     ()
comp-get-ren∋ (ξ , b) (drop η) i       = comp-get-ren∋ ξ η i
comp-get-ren∋ (ξ , a) (keep η) zero    = refl
comp-get-ren∋ (ξ , b) (keep η) (suc i) = comp-get-ren∋ ξ η i


--------------------------------------------------------------------------------


id-gets : ∀ {X P n} → {Ξ : Vec X n}
                    → (ξ : All P Ξ)
                    → gets ξ id⊇ ≡ ξ
id-gets ∙       = refl
id-gets (ξ , a) = (_, a) & id-gets ξ


comp-gets : ∀ {X P n n′ n″ e₁ e₂} → {Ξ : Vec X n} {Ξ′ : Vec X n′} {Ξ″ : Vec X n″}
                                  → (ξ : All P Ξ″) (η₁ : Ξ′ ⊇⟨ e₁ ⟩ Ξ) (η₂ : Ξ″ ⊇⟨ e₂ ⟩ Ξ′)
                                  → gets ξ (η₁ ∘⊇ η₂) ≡ gets (gets ξ η₂) η₁
comp-gets ∙       η₁        done      = refl
comp-gets (ξ , b) η₁        (drop η₂) = comp-gets ξ η₁ η₂
comp-gets (ξ , b) (drop η₁) (keep η₂) = comp-gets ξ η₁ η₂
comp-gets (ξ , a) (keep η₁) (keep η₂) = (_, a) & comp-gets ξ η₁ η₂


𝐠𝐞𝐭𝐬 : ∀ {X P} → Presheaf (Opposite (𝐎𝐏𝐄 {X}))
                           (\ { (n , Ξ) → All P Ξ })
𝐠𝐞𝐭𝐬 = record
         { ℱ     = \ { (e , η) ξ → gets ξ η }
         ; idℱ   = funext! id-gets
         ; compℱ = \ { (e₁ , η₁) (e₂ , η₂) → funext! (\ ξ → comp-gets ξ η₂ η₁) }
         }


--------------------------------------------------------------------------------


{-# REWRITE lid∘⊇ #-}
lid∘⊇◇ : ∀ {X P n n′ e} → {Ξ : Vec X n} {Ξ′ : Vec X n′} {η : Ξ′ ⊇⟨ e ⟩ Ξ}
                           {ξ : All P Ξ} {ξ′ : All P Ξ′}
                        → (𝜂 : ξ′ ⊇◇⟨ η ⟩ ξ)
                        → id⊇◇ ∘⊇◇ 𝜂 ≡ 𝜂
lid∘⊇◇ done     = refl
lid∘⊇◇ (drop 𝜂) = drop & lid∘⊇◇ 𝜂
lid∘⊇◇ (keep 𝜂) = keep & lid∘⊇◇ 𝜂


{-# REWRITE rid∘⊇ #-}
rid∘⊇◇ : ∀ {X P n n′ e} → {Ξ : Vec X n} {Ξ′ : Vec X n′} {η : Ξ′ ⊇⟨ e ⟩ Ξ}
                           {ξ : All P Ξ} {ξ′ : All P Ξ′}
                        → (𝜂 : ξ′ ⊇◇⟨ η ⟩ ξ)
                        → 𝜂 ∘⊇◇ id⊇◇ ≡ 𝜂
rid∘⊇◇ done     = refl
rid∘⊇◇ (drop 𝜂) = drop & rid∘⊇◇ 𝜂
rid∘⊇◇ (keep 𝜂) = keep & rid∘⊇◇ 𝜂


{-# REWRITE assoc∘⊇ #-}
assoc∘⊇◇ : ∀ {X P n n′ n″ n‴ e₁ e₂ e₃} → {Ξ : Vec X n} {Ξ′ : Vec X n′} {Ξ″ : Vec X n″} {Ξ‴ : Vec X n‴}
                                          {η₁ : Ξ′ ⊇⟨ e₁ ⟩ Ξ} {η₂ : Ξ″ ⊇⟨ e₂ ⟩ Ξ′} {η₃ : Ξ‴ ⊇⟨ e₃ ⟩ Ξ″}
                                          {ξ : All P Ξ} {ξ′ : All P Ξ′} {ξ″ : All P Ξ″} {ξ‴ : All P Ξ‴}
                                       → (𝜂₁ : ξ′ ⊇◇⟨ η₁ ⟩ ξ) (𝜂₂ : ξ″ ⊇◇⟨ η₂ ⟩ ξ′) (𝜂₃ : ξ‴ ⊇◇⟨ η₃ ⟩ ξ″)
                                       → (𝜂₁ ∘⊇◇ 𝜂₂) ∘⊇◇ 𝜂₃ ≡ 𝜂₁ ∘⊇◇ (𝜂₂ ∘⊇◇ 𝜂₃)
assoc∘⊇◇ 𝜂₁        𝜂₂        done      = refl
assoc∘⊇◇ 𝜂₁        𝜂₂        (drop 𝜂₃) = drop & assoc∘⊇◇ 𝜂₁ 𝜂₂ 𝜂₃
assoc∘⊇◇ 𝜂₁        (drop 𝜂₂) (keep 𝜂₃) = drop & assoc∘⊇◇ 𝜂₁ 𝜂₂ 𝜂₃
assoc∘⊇◇ (drop 𝜂₁) (keep 𝜂₂) (keep 𝜂₃) = drop & assoc∘⊇◇ 𝜂₁ 𝜂₂ 𝜂₃
assoc∘⊇◇ (keep 𝜂₁) (keep 𝜂₂) (keep 𝜂₃) = keep & assoc∘⊇◇ 𝜂₁ 𝜂₂ 𝜂₃


instance
  𝐎𝐏𝐄◇ : ∀ {X P} → Category (Σ Nat (\ n → Σ (Vec X n) (All P)))
                             (\ { (n′ , (Ξ′ , ξ′)) (n , (Ξ , ξ)) →
                               Σ (n′ ≥ n) (\ e →
                                 Σ (Ξ′ ⊇⟨ e ⟩ Ξ) (\ η → ξ′ ⊇◇⟨ η ⟩ ξ)) })
  𝐎𝐏𝐄◇ = record
           { id     = id , (id⊇ , id⊇◇)
           ; _∘_    = \ { (e₁ , (η₁ , 𝜂₁)) (e₂ , (η₂ , 𝜂₂)) → e₁ ∘ e₂ , (η₁ ∘⊇ η₂ , 𝜂₁ ∘⊇◇ 𝜂₂) }
           ; lid∘   = \ { (e , (η , 𝜂)) → (\ 𝜂′ → e , (η , 𝜂′)) & lid∘⊇◇ 𝜂 }
           ; rid∘   = \ { (e , (η , 𝜂)) → (\ 𝜂′ → e , (η , 𝜂′)) & rid∘⊇◇ 𝜂 }
           ; assoc∘ = \ { (e₁ , (η₁ , 𝜂₁)) (e₂ , (η₂ , 𝜂₂)) (e₃ , (η₃ , 𝜂₃)) →
                        ((\ 𝜂′ → (e₁ ∘ e₂) ∘ e₃ , (((η₁ ∘⊇ η₂) ∘⊇ η₃) , 𝜂′))) & assoc∘⊇◇ 𝜂₁ 𝜂₂ 𝜂₃ }
           }


--------------------------------------------------------------------------------


{-# REWRITE id-ren∋ #-}
id-ren∋◇ : ∀ {X P A n I} → {Ξ : Vec X n} {i : Ξ ∋⟨ I ⟩ A} {a : P A} {ξ : All P Ξ}
                         → (𝑖 : ξ ∋◇⟨ i ⟩ a)
                         → ren∋◇ id⊇◇ 𝑖 ≡ 𝑖
id-ren∋◇ zero    = refl
id-ren∋◇ (suc 𝑖) = suc & id-ren∋◇ 𝑖


{-# REWRITE comp-ren∋ #-}
comp-ren∋◇ : ∀ {X P A n n′ n″ e₁ e₂ I} → {Ξ : Vec X n} {Ξ′ : Vec X n′} {Ξ″ : Vec X n″}
                                          {η₁ : Ξ′ ⊇⟨ e₁ ⟩ Ξ} {η₂ : Ξ″ ⊇⟨ e₂ ⟩ Ξ′} {i : Ξ ∋⟨ I ⟩ A}
                                          {a : P A} {ξ : All P Ξ} {ξ′ : All P Ξ′} {ξ″ : All P Ξ″}
                                       → (𝜂₁ : ξ′ ⊇◇⟨ η₁ ⟩ ξ) (𝜂₂ : ξ″ ⊇◇⟨ η₂ ⟩ ξ′) (𝑖 : ξ ∋◇⟨ i ⟩ a)
                                       → ren∋◇ (𝜂₁ ∘⊇◇ 𝜂₂) 𝑖 ≡ (ren∋◇ 𝜂₂ ∘ ren∋◇ 𝜂₁) 𝑖
comp-ren∋◇ 𝜂₁        done      𝑖       = refl
comp-ren∋◇ 𝜂₁        (drop 𝜂₂) 𝑖       = suc & comp-ren∋◇ 𝜂₁ 𝜂₂ 𝑖
comp-ren∋◇ (drop 𝜂₁) (keep 𝜂₂) 𝑖       = suc & comp-ren∋◇ 𝜂₁ 𝜂₂ 𝑖
comp-ren∋◇ (keep 𝜂₁) (keep 𝜂₂) zero    = refl
comp-ren∋◇ (keep 𝜂₁) (keep 𝜂₂) (suc 𝑖) = suc & comp-ren∋◇ 𝜂₁ 𝜂₂ 𝑖


𝐫𝐞𝐧∋◇ : ∀ {X P A} → {a : P A}
                  → Presheaf (𝐎𝐏𝐄◇ {X} {P})
                              (\ { (n , (Ξ , ξ)) →
                                Σ (Fin n) (\ I →
                                  Σ (Ξ ∋⟨ I ⟩ A) (\ i → ξ ∋◇⟨ i ⟩ a)) })
𝐫𝐞𝐧∋◇ = record
           { ℱ     = \ { (e , (η , 𝜂)) (I , (i , 𝑖)) → REN∋ e I , (ren∋ η i , ren∋◇ 𝜂 𝑖) }
           ; idℱ   = funext! (\ { (I , (i , 𝑖)) → ((_ ,_) ∘ (_ ,_)) & id-ren∋◇ 𝑖 })
           ; compℱ = \ { (e₁ , (η₁ , 𝜂₁)) (e₂ , (η₂ , 𝜂₂)) →
                       funext! (\ { (I , (i , 𝑖)) → ((_ ,_) ∘ (_ ,_)) & comp-ren∋◇ 𝜂₁ 𝜂₂ 𝑖 }) }
           }


--------------------------------------------------------------------------------