use drop, tail and take from Data.Vec in examples
[~helmut/bidiragda.git] / BFFPlug.agda
index 1d5570c..9f45db1 100644 (file)
@@ -3,8 +3,7 @@ open import Relation.Binary using (DecSetoid)
 
 module BFFPlug (A : DecSetoid ℓ₀ ℓ₀) where
 
-open import Data.Nat using (ℕ ; _≟_ ; _+_ ; _∸_ ; zero ; suc ; ⌈_/2⌉)
-open import Data.Nat.Properties using (m+n∸n≡m)
+open import Data.Nat using (ℕ ; _≟_ ; _+_ ; zero ; suc ; ⌈_/2⌉)
 open import Data.Maybe using (Maybe ; just ; nothing)
 open import Data.Vec using (Vec)
 open import Data.Product using (∃ ; _,_)
@@ -12,46 +11,44 @@ open import Relation.Binary using (module DecSetoid)
 open import Relation.Binary.PropositionalEquality using (refl ; cong ; subst ; sym ; module ≡-Reasoning) renaming (setoid to PropEq)
 open import Relation.Nullary using (yes ; no)
 open import Function using (flip ; id ; _∘_)
-open import Function.Equality using (_⟶_)
+open import Function.Equality using (_⟶_ ; _⟨$⟩_)
 open import Function.LeftInverse using (_RightInverseOf_)
 
+open import Generic using (≡-to-Π)
 import BFF
 import GetTypes
 import Examples
 
 open DecSetoid A using (Carrier)
-open GetTypes.VecVec public using (Get)
-open BFF.VecBFF A public
+open GetTypes.PartialVecVec public using (Get)
+open BFF.PartialVecBFF A public
 
-bffsameshape : (G : Get) → {n : ℕ} → Vec Carrier n → Vec Carrier (Get.getlen G n) → Maybe (Vec Carrier n)
-bffsameshape G {n} = bff G n
+bffsameshape : (G : Get) → {i : Get.|I| G} → Vec Carrier (Get.|gl₁| G i) → Vec Carrier (Get.|gl₂| G i) → Maybe (Vec Carrier (Get.|gl₁| G i))
+bffsameshape G {i} = bff G i
 
-bffplug : (G : Get) → (ℕ → ℕ → Maybe ℕ) → {n m : ℕ} → Vec Carrier n → Vec Carrier m → Maybe (∃ λ l → Vec Carrier l)
-bffplug G sput {n} {m} s v with sput n m
+bffplug : (G : Get) → (Get.|I| G → ℕ → Maybe (Get.|I| G)) → {i : Get.|I| G} → {m : ℕ} → Vec Carrier (Get.|gl₁| G i) → Vec Carrier m → Maybe (∃ λ j → Vec Carrier (Get.|gl₁| G j))
+bffplug G sput {i} {m} s v with sput i m
 ...                        | nothing = nothing
-...                        | just l with Get.getlen G l ≟ m
-...                                 | no getlenl≢m  = nothing
-bffplug G sput {n}     s v | just l | yes refl with bff G l s v
+...                        | just j with Get.|gl₂| G j ≟ m
+...                                 | no gl₂j≢m  = nothing
+bffplug G sput {i}     s v | just j | yes refl with bff G j s v
 ...                                            | nothing = nothing
-...                                            | just s′ = just (l , s′)
-
-as-Π : {A B : Set} → (f : A → B) → PropEq A ⟶ PropEq B
-as-Π f = record { _⟨$⟩_ = f; cong = cong f }
+...                                            | just s′ = just (j , s′)
 
 _SimpleRightInvOf_ : (ℕ → ℕ) → (ℕ → ℕ) → Set
-f SimpleRightInvOf g = as-Π f RightInverseOf as-Π g
+f SimpleRightInvOf g = ≡-to-Π f RightInverseOf ≡-to-Π g
 
-bffinv : (G : Get) → (nelteg : ℕ → ℕ) → nelteg SimpleRightInvOf Get.getlen G → {n m : ℕ} → Vec Carrier n → Vec Carrier m → Maybe (Vec Carrier (nelteg m))
-bffinv G nelteg inv {n} {m} s v = bff G (nelteg m) s (subst (Vec Carrier) (sym (inv m)) v)
+bffinv : (G : Get) → (nelteg : PropEq ℕ ⟶ Get.I G) → nelteg RightInverseOf Get.gl₂ G → {i : Get.|I| G} → {m : ℕ} → Vec Carrier (Get.|gl₁| G i) → Vec Carrier m → Maybe (Vec Carrier (Get.|gl₁| G (nelteg ⟨$⟩ m)))
+bffinv G nelteg inv {m = m} s v = bff G (nelteg ⟨$⟩ m) s (subst (Vec Carrier) (sym (inv m)) v)
 
 module InvExamples where
-  open Examples using (reverse' ; drop' ; sieve')
+  open Examples using (reverse' ; drop' ; sieve' ; tail' ; take')
   
   reverse-put : {n m : ℕ} → Vec Carrier n → Vec Carrier m → Maybe (Vec Carrier m)
-  reverse-put = bffinv reverse' id (λ _ → refl)
+  reverse-put = bffinv reverse' (≡-to-Π id) (λ _ → refl)
 
-  drop-put : (k : ℕ) → {n m : ℕ} → Vec Carrier n → Vec Carrier m → Maybe (Vec Carrier (m + k))
-  drop-put k = bffinv (drop' k) (flip _+_ k) (flip m+n∸n≡m k)
+  drop-put : (k : ℕ) → {n m : ℕ} → Vec Carrier (k + n) → Vec Carrier m → Maybe (Vec Carrier (k + m))
+  drop-put k = bffinv (drop' k) (≡-to-Π id) (λ _ → refl)
 
   double : ℕ → ℕ
   double zero    = zero
@@ -63,4 +60,10 @@ module InvExamples where
   sieve-inv-len (suc (suc x)) = cong (suc ∘ suc) (sieve-inv-len x)
 
   sieve-put : {n m : ℕ} → Vec Carrier n → Vec Carrier m → Maybe (Vec Carrier (double m))
-  sieve-put = bffinv sieve' double sieve-inv-len
+  sieve-put = bffinv sieve' (≡-to-Π double) sieve-inv-len
+
+  tail-put : {n m : ℕ} → Vec Carrier (suc n) → Vec Carrier m → Maybe (Vec Carrier (suc m))
+  tail-put = bffinv tail' (≡-to-Π id) (λ _ → refl)
+
+  take-put : (k : ℕ) → {n : ℕ}  → Vec Carrier (k + n) → Vec Carrier k → Maybe (Vec Carrier (k + n))
+  take-put k = bffsameshape (take' k)