summaryrefslogtreecommitdiff
path: root/src/Obs/Syntax.idr
blob: 868e7617cc56a7a7f50ec82def978b1508ac3d65 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
||| Raw syntax for terms. This is before eliminating names.
module Obs.Syntax

import Obs.Pretty
import Obs.Universe

import Text.Bounded

%default total

-- Definition ------------------------------------------------------------------

data Syntax : Type

public export
record DeclForm where
  constructor MkDecl
  var : WithBounds String
  type : WithBounds Syntax

public export
record LambdaForm where
  constructor MkLambda
  var : WithBounds String
  body : WithBounds Syntax

public export
data Syntax : Type where
  Var : (var : String) -> Syntax
  -- Universes
  Universe : (s : Universe) -> Syntax
  -- Dependent Products
  Pi : (domain : DeclForm) -> (codomain : WithBounds Syntax) -> Syntax
  Lambda : (body : LambdaForm) -> Syntax
  App : (fun, arg : WithBounds Syntax) -> Syntax
  -- Dependent Sums
  Sigma : (index : DeclForm) -> (element : WithBounds Syntax) -> Syntax
  Pair : (first, second : WithBounds Syntax) -> Syntax
  First : (arg : WithBounds Syntax) -> Syntax
  Second : (arg : WithBounds Syntax) -> Syntax
  -- Containers
  Container : (input, output : WithBounds Syntax) ->
    (shapeSort, positionSort : WithBounds Universe) ->
    Syntax
  MkContainer : (shape, position, next : WithBounds Syntax) -> Syntax
  Shape : (arg : WithBounds Syntax) -> Syntax
  Position : (arg : WithBounds Syntax) -> Syntax
  Next : (arg : WithBounds Syntax) -> Syntax
  Sem : (pred : LambdaForm) -> (arg : WithBounds Syntax) -> Syntax
  -- Booleans
  Bool : Syntax
  True : Syntax
  False : Syntax
  If : (returnType : LambdaForm) -> (discriminant, true, false : WithBounds Syntax) -> Syntax
  -- True
  Top : Syntax
  Point : Syntax
  -- False
  Bottom : Syntax
  Absurd : (contradiction : WithBounds Syntax) -> Syntax
  -- Equality
  Equal : (left, right : WithBounds Syntax) -> Syntax
  Refl : (value : WithBounds Syntax) -> Syntax
  Transp : (indexedType, oldIndex, value, newIndex, equality : WithBounds Syntax) -> Syntax
  Cast : (oldType, newType, equality, value : WithBounds Syntax) -> Syntax
  CastId : (value : WithBounds Syntax) -> Syntax

public export
record Definition where
  constructor MkDefinition
  name   : WithBounds String
  ty     : WithBounds Syntax
  tm     : WithBounds Syntax

-- Pretty Print ----------------------------------------------------------------

prettyPrecBounds : Prec -> WithBounds Syntax -> Doc ann

prettyPrecDecl : DeclForm -> Doc ann
prettyPrecDecl (MkDecl var type) = prettyDecl (Just var.val) (prettyPrecBounds Open type)

prettyPrecLambda : Prec -> LambdaForm -> Doc ann
prettyPrecLambda d (MkLambda var body) = prettyLambda d var.val (prettyPrecBounds Open body)

prettyPrec : Prec -> Syntax -> Doc ann
prettyPrec d (Var {var}) = pretty var
prettyPrec d (Universe {s}) = prettyPrec d s
prettyPrec d (Pi {domain, codomain}) =
  prettyDeclForm d [prettyPrecDecl domain] "->" (prettyPrecBounds Open codomain)
prettyPrec d (Lambda {body}) = prettyPrecLambda d body
prettyPrec d (App {fun, arg}) = prettyApp d (prettyPrecBounds Open fun) [prettyPrecBounds App arg]
prettyPrec d (Sigma {index, element}) =
  prettyDeclForm d [prettyPrecDecl index] "**" (prettyPrecBounds Open element)
prettyPrec d (Pair {first, second}) =
  prettyPair (prettyPrecBounds Open first) (prettyPrecBounds Open second)
prettyPrec d (First {arg}) = prettyApp d (pretty "fst") [prettyPrecBounds App arg]
prettyPrec d (Second {arg}) = prettyApp d (pretty "snd") [prettyPrecBounds App arg]
prettyPrec d (Container {input, output, shapeSort, positionSort}) =
  prettyApp d (pretty "Container") $
  [ prettyPrecBounds App input
  , prettyPrecBounds App output
  , prettyPrec App shapeSort.val
  , prettyPrec App positionSort.val
  ]
prettyPrec d (MkContainer {shape, position, next}) =
  prettyApp d (pretty "MkContainer") $
  [ prettyPrecBounds App shape
  , prettyPrecBounds App position
  , prettyPrecBounds App next
  ]
prettyPrec d (Shape {arg}) = prettyApp d (pretty "Shape") [prettyPrecBounds App arg]
prettyPrec d (Position {arg}) = prettyApp d (pretty "Position") [prettyPrecBounds App arg]
prettyPrec d (Next {arg}) = prettyApp d (pretty "nextIndex") [prettyPrecBounds App arg]
prettyPrec d (Sem {pred, arg}) =
  prettyApp d (pretty "extension") [prettyPrecLambda App pred, prettyPrecBounds App arg]
prettyPrec d Bool = pretty "Bool"
prettyPrec d True = pretty "True"
prettyPrec d False = pretty "False"
prettyPrec d (If {returnType, discriminant, true, false}) =
  prettyApp d (pretty "if") $
  [ prettyPrecLambda App returnType
  , prettyPrecBounds App discriminant
  , prettyPrecBounds App true
  , prettyPrecBounds App false
  ]
prettyPrec d Top = pretty "()"
prettyPrec d Point = pretty "tt"
prettyPrec d Bottom = pretty "Void"
prettyPrec d (Absurd {contradiction}) =
  prettyApp d (pretty "absurd") [prettyPrecBounds App contradiction]
prettyPrec d (Equal {left, right}) =
  prettyEqual d (prettyPrecBounds Equal left) (prettyPrecBounds Equal right)
prettyPrec d (Refl {value}) = prettyApp d (pretty "refl") [prettyPrecBounds App value]
prettyPrec d (Transp {indexedType, oldIndex, value, newIndex, equality}) =
  prettyApp d (pretty "transp") $
  [ prettyPrecBounds App indexedType
  , prettyPrecBounds App oldIndex
  , prettyPrecBounds App value
  , prettyPrecBounds App newIndex
  , prettyPrecBounds App equality
  ]
prettyPrec d (Cast {oldType, newType, equality, value}) =
  prettyApp d (pretty "cast") $
  [ prettyPrecBounds App oldType
  , prettyPrecBounds App newType
  , prettyPrecBounds App equality
  , prettyPrecBounds App value
  ]
prettyPrec d (CastId {value}) = prettyApp d (pretty "castRefl") [prettyPrecBounds App value]

prettyPrecBounds d (MkBounded val _ _) = prettyPrec d val

export
Pretty Syntax where
  prettyPrec = Syntax.prettyPrec

export
Pretty Definition where
  pretty def = group $
    pretty def.name.val <++> colon <+> softline <+> pretty def.ty.val <+> hardline <+>
    pretty def.name.val <++> equals <+> softline <+> pretty def.tm.val