move definition of get-type to BFF and use it everywhere
[~helmut/bidiragda.git] / BFF.agda
1 module BFF where
2
3 open import Data.Nat using (ℕ)
4 open import Data.Fin using (Fin)
5 open import Data.Maybe using (Maybe ; just ; nothing ; maybe′)
6 open import Data.List using (List ; [] ; _∷_ ; map ; length)
7 open import Data.Vec using (Vec ; toList ; fromList ; tabulate ; allFin) renaming (lookup to lookupV ; map to mapV ; [] to []V ; _∷_ to _∷V_)
8 open import Function using (id ; _∘_ ; flip)
9
10 open import FinMap
11 open import CheckInsert
12
13 _>>=_ : {A B : Set} → Maybe A → (A → Maybe B) → Maybe B
14 _>>=_ = flip (flip maybe′ nothing)
15
16 fmap : {A B : Set} → (A → B) → Maybe A → Maybe B
17 fmap f = maybe′ (λ a → just (f a)) nothing
18
19 module ListBFF where
20   get-type : Set₁
21   get-type = {A : Set} → List A → List A
22
23   assoc : {A : Set} {n : ℕ} → EqInst A → List (Fin n) → List A → Maybe (FinMapMaybe n A)
24   assoc _  []       []       = just empty
25   assoc eq (i ∷ is) (b ∷ bs) = (assoc eq is bs) >>= (checkInsert eq i b)
26   assoc _  _        _        = nothing
27
28   enumerate : {A : Set} → (l : List A) → List (Fin (length l))
29   enumerate l = toList (tabulate id)
30
31   denumerate : {A : Set} (l : List A) → Fin (length l) → A
32   denumerate l = flip lookupV (fromList l)
33
34   bff : get-type → ({B : Set} → EqInst B → List B → List B → Maybe (List B))
35   bff get eq s v = let s′ = enumerate s
36                        g  = fromFunc (denumerate s)
37                        h  = assoc eq (get s′) v
38                        h′ = fmap (flip union g) h
39                    in fmap (flip map s′ ∘ flip lookup) h′
40
41 module VecBFF where
42   get-type : (ℕ → ℕ) → Set₁
43   get-type getlen = {A : Set} {n : ℕ} → Vec A n → Vec A (getlen n)
44
45   assoc : {A : Set} {n m : ℕ} → EqInst A → Vec (Fin n) m → Vec A m → Maybe (FinMapMaybe n A)
46   assoc _  []V       []V       = just empty
47   assoc eq (i ∷V is) (b ∷V bs) = (assoc eq is bs) >>= (checkInsert eq i b)
48
49   enumerate : {A : Set} {n : ℕ} → Vec A n → Vec (Fin n) n
50   enumerate _ = tabulate id
51
52   denumerate : {A : Set} {n : ℕ} → Vec A n → Fin n → A
53   denumerate = flip lookupV
54
55   bff : {getlen : ℕ → ℕ} → (get-type getlen) → ({B : Set} {n : ℕ} → EqInst B → Vec B n → Vec B (getlen n) → Maybe (Vec B n))
56   bff get eq s v = let s′ = enumerate s
57                        g  = fromFunc (denumerate s)
58                        h  = assoc eq (get s′) v
59                        h′ = fmap (flip union g) h
60                    in fmap (flip mapV s′ ∘ flip lookupV) h′
61
62 module VecRevBFF where
63   get-type : (ℕ → ℕ) → Set₁
64   get-type getlen = {A : Set} {n : ℕ} → Vec A (getlen n) → Vec A n
65
66   assoc : {A : Set} {m n : ℕ} → EqInst A → Vec (Fin m) n → Vec A n → Maybe (FinMapMaybe m A)
67   assoc _  []V       []V       = just empty
68   assoc eq (i ∷V is) (b ∷V bs) = (assoc eq is bs) >>= (checkInsert eq i b)
69
70   enumerate : {A : Set} {n : ℕ} → Vec A n → Vec (Fin n) n
71   enumerate _ = tabulate id
72
73   denumerate : {A : Set} {n : ℕ} → Vec A n → Fin n → A
74   denumerate = flip lookupV
75
76   bff : {getlen : ℕ → ℕ} → (get-type getlen) → ({m : ℕ} {B : Set} → EqInst B → Vec B (getlen m) → Vec B m → Maybe (Vec B (getlen m)))
77   bff get eq s v = let s′ = enumerate s
78                        g  = fromFunc (denumerate s)
79                        h  = assoc eq (get s′) v
80                        h′ = fmap (flip union g) h
81                    in fmap (flip mapV s′ ∘ (flip lookup)) h′