annotate lib/ur/top.ur @ 983:2cd8c1aa0d3a

Add tail call optimization to interpretation
author Adam Chlipala <adamc@hcoop.net>
date Thu, 24 Sep 2009 15:45:37 -0400
parents 37dd42935dad
children 10114d7b7477
rev   line source
adamc@623 1 (** Row folding *)
adamc@623 2
adamc@623 3 con folder = K ==> fn r :: {K} =>
adamc@623 4 tf :: ({K} -> Type)
adamc@653 5 -> (nm :: Name -> v :: K -> r :: {K} -> [[nm] ~ r] =>
adamc@653 6 tf r -> tf ([nm = v] ++ r))
adamc@623 7 -> tf [] -> tf r
adamc@623 8
adamc@822 9 fun fold [K] [tf :: {K} -> Type]
adamc@653 10 (f : (nm :: Name -> v :: K -> r :: {K} -> [[nm] ~ r] =>
adamc@653 11 tf r -> tf ([nm = v] ++ r)))
adamc@822 12 (i : tf []) [r :: {K}] (fl : folder r) = fl [tf] f i
adamc@653 13
adamc@627 14 structure Folder = struct
adamc@822 15 fun nil [K] [tf :: {K} -> Type]
adamc@653 16 (f : nm :: Name -> v :: K -> r :: {K} -> [[nm] ~ r] =>
adamc@653 17 tf r -> tf ([nm = v] ++ r))
adamc@627 18 (i : tf []) = i
adamc@627 19
adamc@822 20 fun cons [K] [r ::: {K}] [nm :: Name] [v :: K] [[nm] ~ r] (fold : folder r)
adamc@822 21 [tf :: {K} -> Type]
adamc@653 22 (f : nm :: Name -> v :: K -> r :: {K} -> [[nm] ~ r] =>
adamc@653 23 tf r -> tf ([nm = v] ++ r))
adamc@653 24 (i : tf []) = f [nm] [v] [r] ! (fold [tf] f i)
adamc@628 25
adamc@822 26 fun concat [K] [r1 ::: {K}] [r2 ::: {K}] [r1 ~ r2]
adamc@628 27 (f1 : folder r1) (f2 : folder r2)
adamc@822 28 [tf :: {K} -> Type]
adamc@653 29 (f : nm :: Name -> v :: K -> r :: {K} -> [[nm] ~ r] =>
adamc@653 30 tf r -> tf ([nm = v] ++ r))
adamc@628 31 (i : tf []) =
adamc@629 32 f1 [fn r1' => [r1' ~ r2] => tf (r1' ++ r2)]
adamc@822 33 (fn [nm :: Name] [v :: K] [r1' :: {K}] [[nm] ~ r1']
adamc@653 34 (acc : [r1' ~ r2] => tf (r1' ++ r2))
adamc@653 35 [[nm = v] ++ r1' ~ r2] =>
adamc@653 36 f [nm] [v] [r1' ++ r2] ! acc)
adamc@629 37 (fn [[] ~ r2] => f2 [tf] f i) !
adamc@630 38
adamc@822 39 fun mp [K1] [K2] [f ::: K1 -> K2] [r ::: {K1}]
adamc@630 40 (fold : folder r)
adamc@822 41 [tf :: {K2} -> Type]
adamc@653 42 (f : nm :: Name -> v :: K2 -> r :: {K2} -> [[nm] ~ r] =>
adamc@653 43 tf r -> tf ([nm = v] ++ r))
adamc@630 44 (i : tf []) =
adamc@630 45 fold [fn r => tf (map f r)]
adamc@822 46 (fn [nm :: Name] [v :: K1] [rest :: {K1}] [[nm] ~ rest] (acc : tf (map f rest)) =>
adamc@653 47 f [nm] [f v] [map f rest] ! acc)
adamc@630 48 i
adamc@627 49 end
adamc@627 50
adamc@623 51
adamc@422 52 fun not b = if b then False else True
adamc@422 53
adamc@898 54 con id = K ==> fn t :: K => t
adamc@356 55 con record (t :: {Type}) = $t
adamc@637 56 con fst = K1 ==> K2 ==> fn t :: (K1 * K2) => t.1
adamc@637 57 con snd = K1 ==> K2 ==> fn t :: (K1 * K2) => t.2
adamc@637 58 con fst3 = K1 ==> K2 ==> K3 ==> fn t :: (K1 * K2 * K3) => t.1
adamc@637 59 con snd3 = K1 ==> K2 ==> K3 ==> fn t :: (K1 * K2 * K3) => t.2
adamc@637 60 con thd3 = K1 ==> K2 ==> K3 ==> fn t :: (K1 * K2 * K3) => t.3
adamc@329 61
adamc@643 62 con mapU = K ==> fn f :: K => map (fn _ :: Unit => f)
adamc@445 63
adamc@339 64 con ex = fn tf :: (Type -> Type) =>
adamc@356 65 res ::: Type -> (choice :: Type -> tf choice -> res) -> res
adamc@339 66
adamc@822 67 fun ex [tf :: (Type -> Type)] [choice :: Type] (body : tf choice) : ex tf =
adamc@822 68 fn [res] (f : choice :: Type -> tf choice -> res) =>
adamc@356 69 f [choice] body
adamc@339 70
adamc@822 71 fun compose [t1 ::: Type] [t2 ::: Type] [t3 ::: Type]
adamc@356 72 (f1 : t2 -> t3) (f2 : t1 -> t2) (x : t1) = f1 (f2 x)
adamc@325 73
adamc@822 74 fun show_option [t ::: Type] (_ : show t) =
adamc@777 75 mkShow (fn opt : option t =>
adamc@777 76 case opt of
adamc@777 77 None => ""
adamc@777 78 | Some x => show x)
adamc@777 79
adamc@822 80 fun read_option [t ::: Type] (_ : read t) =
adamc@777 81 mkRead (fn s =>
adamc@777 82 case s of
adamc@777 83 "" => None
adamc@777 84 | _ => Some (readError s : t))
adamc@777 85 (fn s =>
adamc@777 86 case s of
adamc@777 87 "" => Some None
adamc@777 88 | _ => case read s of
adamc@777 89 None => None
adamc@777 90 | v => Some v)
adamc@777 91
adamc@822 92 fun txt [t] [ctx ::: {Unit}] [use ::: {Type}] (_ : show t) (v : t) =
adamc@564 93 cdata (show v)
adamc@328 94
adamc@898 95 fun mp [K] [tf1 :: K -> Type] [tf2 :: K -> Type] (f : t ::: K -> tf1 t -> tf2 t) [r :: {K}] (fl : folder r) =
adamc@898 96 fl [fn r :: {K} => $(map tf1 r) -> $(map tf2 r)]
adamc@898 97 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest] acc r =>
adamc@898 98 acc (r -- nm) ++ {nm = f r.nm})
adamc@898 99 (fn _ => {})
adamc@898 100
adamc@905 101 fun map2 [K] [tf1 :: K -> Type] [tf2 :: K -> Type] [tf3 :: K -> Type]
adamc@905 102 (f : t ::: K -> tf1 t -> tf2 t -> tf3 t) [r :: {K}] (fl : folder r) =
adamc@905 103 fl [fn r :: {K} => $(map tf1 r) -> $(map tf2 r) -> $(map tf3 r)]
adamc@905 104 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest] acc r1 r2 =>
adamc@905 105 acc (r1 -- nm) (r2 -- nm) ++ {nm = f r1.nm r2.nm})
adamc@905 106 (fn _ _ => {})
adamc@898 107
adamc@937 108 fun map3 [K] [tf1 :: K -> Type] [tf2 :: K -> Type] [tf3 :: K -> Type] [tf :: K -> Type]
adamc@937 109 (f : t ::: K -> tf1 t -> tf2 t -> tf3 t -> tf t) [r :: {K}] (fl : folder r) =
adamc@937 110 fl [fn r :: {K} => $(map tf1 r) -> $(map tf2 r) -> $(map tf3 r) -> $(map tf r)]
adamc@937 111 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest] acc r1 r2 r3 =>
adamc@937 112 acc (r1 -- nm) (r2 -- nm) (r3 -- nm) ++ {nm = f r1.nm r2.nm r3.nm})
adamc@937 113 (fn _ _ _ => {})
adamc@937 114
adamc@822 115 fun foldUR [tf :: Type] [tr :: {Unit} -> Type]
adamc@411 116 (f : nm :: Name -> rest :: {Unit}
adamc@629 117 -> [[nm] ~ rest] =>
adamc@411 118 tf -> tr rest -> tr ([nm] ++ rest))
adamc@898 119 (i : tr []) [r :: {Unit}] (fl : folder r) =
adamc@653 120 fl [fn r :: {Unit} => $(mapU tf r) -> tr r]
adamc@822 121 (fn [nm :: Name] [t :: Unit] [rest :: {Unit}] [[nm] ~ rest] acc r =>
adamc@653 122 f [nm] [rest] ! r.nm (acc (r -- nm)))
adamc@653 123 (fn _ => i)
adamc@411 124
adamc@822 125 fun foldUR2 [tf1 :: Type] [tf2 :: Type] [tr :: {Unit} -> Type]
adamc@418 126 (f : nm :: Name -> rest :: {Unit}
adamc@629 127 -> [[nm] ~ rest] =>
adamc@418 128 tf1 -> tf2 -> tr rest -> tr ([nm] ++ rest))
adamc@822 129 (i : tr []) [r :: {Unit}] (fl : folder r) =
adamc@653 130 fl [fn r :: {Unit} => $(mapU tf1 r) -> $(mapU tf2 r) -> tr r]
adamc@822 131 (fn [nm :: Name] [t :: Unit] [rest :: {Unit}] [[nm] ~ rest] acc r1 r2 =>
adamc@653 132 f [nm] [rest] ! r1.nm r2.nm (acc (r1 -- nm) (r2 -- nm)))
adamc@653 133 (fn _ _ => i)
adamc@418 134
adamc@822 135 fun foldURX2 [tf1 :: Type] [tf2 :: Type] [ctx :: {Unit}]
adamc@418 136 (f : nm :: Name -> rest :: {Unit}
adamc@629 137 -> [[nm] ~ rest] =>
adamc@720 138 tf1 -> tf2 -> xml ctx [] []) =
adamc@720 139 foldUR2 [tf1] [tf2] [fn _ => xml ctx [] []]
adamc@822 140 (fn [nm :: Name] [rest :: {Unit}] [[nm] ~ rest] v1 v2 acc =>
adamc@629 141 <xml>{f [nm] [rest] ! v1 v2}{acc}</xml>)
adamc@418 142 <xml/>
adamc@418 143
adamc@822 144 fun foldR [K] [tf :: K -> Type] [tr :: {K} -> Type]
adamc@623 145 (f : nm :: Name -> t :: K -> rest :: {K}
adamc@629 146 -> [[nm] ~ rest] =>
adamc@356 147 tf t -> tr rest -> tr ([nm = t] ++ rest))
adamc@822 148 (i : tr []) [r :: {K}] (fl : folder r) =
adamc@653 149 fl [fn r :: {K} => $(map tf r) -> tr r]
adamc@822 150 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest] (acc : _ -> tr rest) r =>
adamc@653 151 f [nm] [t] [rest] ! r.nm (acc (r -- nm)))
adamc@653 152 (fn _ => i)
adamc@336 153
adamc@822 154 fun foldR2 [K] [tf1 :: K -> Type] [tf2 :: K -> Type] [tr :: {K} -> Type]
adamc@623 155 (f : nm :: Name -> t :: K -> rest :: {K}
adamc@629 156 -> [[nm] ~ rest] =>
adamc@367 157 tf1 t -> tf2 t -> tr rest -> tr ([nm = t] ++ rest))
adamc@822 158 (i : tr []) [r :: {K}] (fl : folder r) =
adamc@653 159 fl [fn r :: {K} => $(map tf1 r) -> $(map tf2 r) -> tr r]
adamc@822 160 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest]
adamc@653 161 (acc : _ -> _ -> tr rest) r1 r2 =>
adamc@653 162 f [nm] [t] [rest] ! r1.nm r2.nm (acc (r1 -- nm) (r2 -- nm)))
adamc@653 163 (fn _ _ => i)
adamc@332 164
adamc@910 165 fun foldR3 [K] [tf1 :: K -> Type] [tf2 :: K -> Type] [tf3 :: K -> Type] [tr :: {K} -> Type]
adamc@910 166 (f : nm :: Name -> t :: K -> rest :: {K}
adamc@910 167 -> [[nm] ~ rest] =>
adamc@910 168 tf1 t -> tf2 t -> tf3 t -> tr rest -> tr ([nm = t] ++ rest))
adamc@910 169 (i : tr []) [r :: {K}] (fl : folder r) =
adamc@910 170 fl [fn r :: {K} => $(map tf1 r) -> $(map tf2 r) -> $(map tf3 r) -> tr r]
adamc@910 171 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest]
adamc@910 172 (acc : _ -> _ -> _ -> tr rest) r1 r2 r3 =>
adamc@910 173 f [nm] [t] [rest] ! r1.nm r2.nm r3.nm (acc (r1 -- nm) (r2 -- nm) (r3 -- nm)))
adamc@910 174 (fn _ _ _ => i)
adamc@910 175
adamc@822 176 fun foldRX [K] [tf :: K -> Type] [ctx :: {Unit}]
adamc@623 177 (f : nm :: Name -> t :: K -> rest :: {K}
adamc@629 178 -> [[nm] ~ rest] =>
adamc@720 179 tf t -> xml ctx [] []) =
adamc@720 180 foldR [tf] [fn _ => xml ctx [] []]
adamc@822 181 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest] r acc =>
adamc@629 182 <xml>{f [nm] [t] [rest] ! r}{acc}</xml>)
adamc@623 183 <xml/>
adamc@336 184
adamc@822 185 fun foldRX2 [K] [tf1 :: K -> Type] [tf2 :: K -> Type] [ctx :: {Unit}]
adamc@623 186 (f : nm :: Name -> t :: K -> rest :: {K}
adamc@629 187 -> [[nm] ~ rest] =>
adamc@720 188 tf1 t -> tf2 t -> xml ctx [] []) =
adamc@720 189 foldR2 [tf1] [tf2] [fn _ => xml ctx [] []]
adamc@822 190 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest]
adamc@623 191 r1 r2 acc =>
adamc@629 192 <xml>{f [nm] [t] [rest] ! r1 r2}{acc}</xml>)
adamc@623 193 <xml/>
adamc@445 194
adamc@910 195 fun foldRX3 [K] [tf1 :: K -> Type] [tf2 :: K -> Type] [tf3 :: K -> Type] [ctx :: {Unit}]
adamc@910 196 (f : nm :: Name -> t :: K -> rest :: {K}
adamc@910 197 -> [[nm] ~ rest] =>
adamc@910 198 tf1 t -> tf2 t -> tf3 t -> xml ctx [] []) =
adamc@910 199 foldR3 [tf1] [tf2] [tf3] [fn _ => xml ctx [] []]
adamc@910 200 (fn [nm :: Name] [t :: K] [rest :: {K}] [[nm] ~ rest]
adamc@910 201 r1 r2 r3 acc =>
adamc@910 202 <xml>{f [nm] [t] [rest] ! r1 r2 r3}{acc}</xml>)
adamc@910 203 <xml/>
adamc@910 204
adamc@822 205 fun queryI [tables ::: {{Type}}] [exps ::: {Type}]
adamc@682 206 [tables ~ exps] (q : sql_query tables exps)
adamc@682 207 (f : $(exps ++ map (fn fields :: {Type} => $fields) tables)
adamc@682 208 -> transaction unit) =
adamc@682 209 query q
adamc@682 210 (fn fs _ => f fs)
adamc@682 211 ()
adamc@682 212
adamc@822 213 fun queryX [tables ::: {{Type}}] [exps ::: {Type}] [ctx ::: {Unit}]
adamc@632 214 [tables ~ exps] (q : sql_query tables exps)
adamc@621 215 (f : $(exps ++ map (fn fields :: {Type} => $fields) tables)
adamc@720 216 -> xml ctx [] []) =
adamc@356 217 query q
adamc@356 218 (fn fs acc => return <xml>{acc}{f fs}</xml>)
adamc@360 219 <xml/>
adamc@341 220
adamc@822 221 fun queryX' [tables ::: {{Type}}] [exps ::: {Type}] [ctx ::: {Unit}]
adamc@632 222 [tables ~ exps] (q : sql_query tables exps)
adamc@621 223 (f : $(exps ++ map (fn fields :: {Type} => $fields) tables)
adamc@720 224 -> transaction (xml ctx [] [])) =
adamc@469 225 query q
adamc@469 226 (fn fs acc =>
adamc@469 227 r <- f fs;
adamc@469 228 return <xml>{acc}{r}</xml>)
adamc@469 229 <xml/>
adamc@469 230
adamc@822 231 fun oneOrNoRows [tables ::: {{Type}}] [exps ::: {Type}]
adamc@629 232 [tables ~ exps]
adamc@629 233 (q : sql_query tables exps) =
adamc@356 234 query q
adamc@356 235 (fn fs _ => return (Some fs))
adamc@356 236 None
adamc@440 237
adamc@822 238 fun oneRow [tables ::: {{Type}}] [exps ::: {Type}]
adamc@822 239 [tables ~ exps] (q : sql_query tables exps) =
adamc@440 240 o <- oneOrNoRows q;
adamc@440 241 return (case o of
adamc@440 242 None => error <xml>Query returned no rows</xml>
adamc@440 243 | Some r => r)
adamc@440 244
adamc@822 245 fun eqNullable [tables ::: {{Type}}] [agg ::: {{Type}}] [exps ::: {Type}]
adamc@822 246 [t ::: Type] (_ : sql_injectable (option t))
adamc@470 247 (e1 : sql_exp tables agg exps (option t))
adamc@470 248 (e2 : sql_exp tables agg exps (option t)) =
adamc@471 249 (SQL ({e1} IS NULL AND {e2} IS NULL) OR {e1} = {e2})
adamc@470 250
adamc@822 251 fun eqNullable' [tables ::: {{Type}}] [agg ::: {{Type}}] [exps ::: {Type}]
adamc@822 252 [t ::: Type] (_ : sql_injectable (option t))
adamc@470 253 (e1 : sql_exp tables agg exps (option t))
adamc@470 254 (e2 : option t) =
adamc@470 255 case e2 of
adamc@471 256 None => (SQL {e1} IS NULL)
adamc@564 257 | Some _ => sql_binary sql_eq e1 (sql_inject e2)