Source file indTyping.ml
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
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
open Util
open Names
open Univ
open Term
open Constr
open Declarations
open Environ
open Entries
open Type_errors
open Context.Rel.Declaration
(** Check name unicity.
Redundant with safe_typing's add_field checks -> to remove?. *)
let check_constructors_names =
let rec check idset = function
| [] -> idset
| c::cl ->
if Id.Set.mem c idset then
raise (InductiveError (SameNamesConstructors c))
else
check (Id.Set.add c idset) cl
in
check
let mind_check_names mie =
let rec check indset cstset = function
| [] -> ()
| ind::inds ->
let id = ind.mind_entry_typename in
let cl = ind.mind_entry_consnames in
if Id.Set.mem id indset then
raise (InductiveError (SameNamesTypes id))
else
let cstset' = check_constructors_names cstset cl in
check (Id.Set.add id indset) cstset' inds
in
check Id.Set.empty Id.Set.empty mie.mind_entry_inds
type univ_info = { ind_squashed : bool; ind_has_relevant_arg : bool;
ind_min_univ : Universe.t option;
ind_univ : Universe.t;
missing : Universe.Set.t;
}
let check_univ_leq ?(is_real_arg=false) env u info =
let ind_univ = info.ind_univ in
let info = if not info.ind_has_relevant_arg && is_real_arg && not (Univ.Universe.is_sprop u)
then {info with ind_has_relevant_arg=true}
else info
in
if Univ.Universe.is_sprop u || UGraph.check_leq (universes env) u ind_univ
then { info with ind_min_univ = Option.map (Universe.sup u) info.ind_min_univ }
else if is_impredicative_univ env ind_univ
&& Option.is_empty info.ind_min_univ then { info with ind_squashed = true }
else {info with missing = Universe.Set.add u info.missing}
let check_context_univs ~ctor env info ctx =
let check_one d (info,env) =
let info = match d with
| LocalAssum (_,t) ->
let tj = Typeops.infer_type env t in
check_univ_leq ~is_real_arg:ctor env (Sorts.univ_of_sort tj.utj_type) info
| LocalDef _ -> info
in
info, push_rel d env
in
fst (Context.Rel.fold_outside ~init:(info,env) check_one ctx)
let check_indices_matter env_params info indices =
if not (indices_matter env_params) then info
else check_context_univs ~ctor:false env_params info indices
let check_arity ~template env_params env_ar ind =
let {utj_val=arity;utj_type=_} = Typeops.infer_type env_params ind.mind_entry_arity in
let indices, ind_sort = Reduction.dest_arity env_params arity in
let ind_min_univ = if template then Some Universe.type0m else None in
let univ_info = {
ind_squashed=false;
ind_has_relevant_arg=false;
ind_min_univ;
ind_univ=Sorts.univ_of_sort ind_sort;
missing=Universe.Set.empty;
}
in
let univ_info = check_indices_matter env_params univ_info indices in
let arity = it_mkProd_or_LetIn arity (Environ.rel_context env_params) in
let x = Context.make_annot (Name ind.mind_entry_typename) (Sorts.relevance_of_sort ind_sort) in
push_rel (LocalAssum (x, arity)) env_ar,
(arity, indices, univ_info)
let check_constructor_univs env_ar_par info (args,_) =
check_context_univs ~ctor:true env_ar_par info args
let check_constructors env_ar_par isrecord params lc (arity,indices,univ_info) =
let lc = Array.map_of_list (fun c -> (Typeops.infer_type env_ar_par c).utj_val) lc in
let splayed_lc = Array.map (Reduction.dest_prod_assum env_ar_par) lc in
let univ_info = match Array.length lc with
| 0 -> univ_info
| 1 ->
if isrecord then univ_info
else if (Environ.typing_flags env_ar_par).allow_uip
&& fst (splayed_lc.(0)) = []
&& List.for_all Context.Rel.Declaration.is_local_assum params
then univ_info
else check_univ_leq env_ar_par Univ.Universe.type0m univ_info
| _ -> check_univ_leq env_ar_par Univ.Universe.type0 univ_info
in
let univ_info = Array.fold_left (check_constructor_univs env_ar_par) univ_info splayed_lc in
let lc = Array.map (fun c -> Term.it_mkProd_or_LetIn c params) lc in
(arity, lc), (indices, splayed_lc), univ_info
let check_record data =
List.for_all (fun (_,(_,splayed_lc),info) ->
not info.ind_squashed
&& (Univ.Universe.is_sprop info.ind_univ
|| info.ind_has_relevant_arg)
&& (match splayed_lc with
| [|ctx,_|] ->
let module D = Context.Rel.Declaration in
List.exists D.is_local_assum ctx &&
List.for_all (fun d -> not (D.is_local_assum d)
|| not (Name.is_anonymous (D.get_name d)))
ctx
| _ -> false))
data
let allowed_sorts {ind_squashed;ind_univ;ind_min_univ=_;ind_has_relevant_arg=_;missing=_} =
if not ind_squashed then InType
else Sorts.family (Sorts.sort_of_univ ind_univ)
let unbounded_from_below u cstrs =
Univ.Constraints.for_all (fun (l, d, r) ->
match d with
| Eq -> not (Univ.Level.equal l u) && not (Univ.Level.equal r u)
| Lt | Le -> not (Univ.Level.equal r u))
cstrs
let template_polymorphic_univs ~ctor_levels uctx paramsctxt concl =
let check_level l =
Univ.Level.Set.mem l (Univ.ContextSet.levels uctx) &&
(let () = assert (not @@ Univ.Level.is_small l) in true) &&
unbounded_from_below l (Univ.ContextSet.constraints uctx) &&
not (Univ.Level.Set.mem l ctor_levels)
in
let univs = Univ.Universe.levels concl in
let univs = Univ.Level.Set.filter (fun l -> check_level l) univs in
let fold acc = function
| (LocalAssum (_, p)) ->
(let c = Term.strip_prod_assum p in
match kind c with
| Sort (Type u) ->
(match Univ.Universe.level u with
| Some l -> if Univ.Level.Set.mem l univs then Some l else None
| None -> None)
| _ -> None) :: acc
| LocalDef _ -> acc
in
let params = List.fold_left fold [] paramsctxt in
if Universe.is_type0m concl then Some (univs, params)
else if not @@ Univ.Level.Set.is_empty univs then Some (univs, params)
else None
let get_param_levels ctx params arity splayed_lc =
let min_univ = match arity with
| RegularArity _ ->
CErrors.anomaly
Pp.(strbrk "Ill-formed template mutual inductive declaration: all types must be template.")
| TemplateArity ar -> ar.template_level
in
let ctor_levels =
let add_levels c levels = Univ.Level.Set.union levels (Vars.universes_of_constr c) in
let param_levels =
List.fold_left (fun levels d -> match d with
| LocalAssum _ -> levels
| LocalDef (_,b,t) -> add_levels b (add_levels t levels))
Univ.Level.Set.empty params
in
Array.fold_left
(fun levels (d,c) ->
let levels =
List.fold_left (fun levels d ->
Context.Rel.Declaration.fold_constr add_levels d levels)
levels d
in
add_levels c levels)
param_levels
splayed_lc
in
match template_polymorphic_univs ~ctor_levels ctx params min_univ with
| None ->
CErrors.user_err
Pp.(strbrk "Ill-formed template inductive declaration: not polymorphic on any universe.")
| Some (_, param_levels) ->
param_levels
let get_template univs params data =
match univs with
| Polymorphic_ind_entry _ | Monomorphic_ind_entry -> None
| Template_ind_entry ctx ->
let params = List.map (fun ((arity, _), (_, splayed_lc), _) -> get_param_levels ctx params arity splayed_lc) data in
let fold min params =
let map u v = match u, v with
| (None, _) | (_, None) -> None
| Some u, Some v ->
let () = assert (Univ.Level.equal u v) in
Some u
in
List.map2 map min params
in
let params = match params with
| [] -> assert false
| hd :: rem -> List.fold_left fold hd rem
in
Some { template_param_levels = params; template_context = ctx }
let abstract_packets usubst ((arity,lc),(indices,splayed_lc),univ_info) =
if not (Universe.Set.is_empty univ_info.missing)
then raise (InductiveError (MissingConstraints (univ_info.missing,univ_info.ind_univ)));
let arity = Vars.subst_univs_level_constr usubst arity in
let lc = Array.map (Vars.subst_univs_level_constr usubst) lc in
let indices = Vars.subst_univs_level_context usubst indices in
let splayed_lc = Array.map (fun (args,out) ->
let args = Vars.subst_univs_level_context usubst args in
let out = Vars.subst_univs_level_constr usubst out in
args,out)
splayed_lc
in
let ind_univ = Univ.subst_univs_level_universe usubst univ_info.ind_univ in
let arity = match univ_info.ind_min_univ with
| None -> RegularArity {mind_user_arity = arity; mind_sort = Sorts.sort_of_univ ind_univ}
| Some min_univ -> TemplateArity { template_level = min_univ; }
in
let kelim = allowed_sorts univ_info in
(arity,lc), (indices,splayed_lc), kelim
let typecheck_inductive env ~sec_univs (mie:mutual_inductive_entry) =
let () = match mie.mind_entry_inds with
| [] -> CErrors.anomaly Pp.(str "empty inductive types declaration.")
| _ -> ()
in
mind_check_names mie;
assert (List.is_empty (Environ.rel_context env));
let env_univs =
match mie.mind_entry_universes with
| Template_ind_entry ctx ->
let env = set_universes_lbound env UGraph.Bound.Prop in
push_context_set ~strict:false ctx env
| Monomorphic_ind_entry -> env
| Polymorphic_ind_entry ctx -> push_context ctx env
in
let has_template_poly = match mie.mind_entry_universes with
| Template_ind_entry _ -> true
| Monomorphic_ind_entry | Polymorphic_ind_entry _ -> false
in
let env_params, params = Typeops.check_context env_univs mie.mind_entry_params in
let env_ar, data = List.fold_left_map (check_arity ~template:has_template_poly env_params) env_univs mie.mind_entry_inds in
let env_ar_par = push_rel_context params env_ar in
let isrecord = match mie.mind_entry_record with
| Some (Some _) -> true
| Some None | None -> false
in
let data = List.map2 (fun ind data ->
check_constructors env_ar_par isrecord params ind.mind_entry_lc data)
mie.mind_entry_inds data
in
let record = mie.mind_entry_record in
let data, record = match record with
| None | Some None -> data, record
| Some (Some _) ->
if check_record data then
data, record
else
let data = List.map (fun (a,b,univs) ->
a,b,check_univ_leq env_ar_par Univ.Universe.type0m univs)
data
in
data, Some None
in
let variance = match mie.mind_entry_variance with
| None -> None
| Some variances ->
match mie.mind_entry_universes with
| Monomorphic_ind_entry | Template_ind_entry _ ->
CErrors.user_err Pp.(str "Inductive cannot be both monomorphic and universe cumulative.")
| Polymorphic_ind_entry uctx ->
let univs = Instance.to_array @@ UContext.instance uctx in
let univs = Array.map2 (fun a b -> a,b) univs variances in
let univs = match sec_univs with
| None -> univs
| Some sec_univs ->
let sec_univs = Array.map (fun u -> u, None) sec_univs in
Array.append sec_univs univs
in
let variances = InferCumulativity.infer_inductive ~env_params univs mie.mind_entry_inds in
Some variances
in
let usubst, univs = match mie.mind_entry_universes with
| Monomorphic_ind_entry | Template_ind_entry _ ->
Univ.empty_level_subst, Monomorphic
| Polymorphic_ind_entry uctx ->
let (inst, auctx) = Univ.abstract_universes uctx in
let inst = Univ.make_instance_subst inst in
(inst, Polymorphic auctx)
in
let params = Vars.subst_univs_level_context usubst params in
let data = List.map (abstract_packets usubst) data in
let template = get_template mie.mind_entry_universes params data in
let env_ar_par =
let ctx = Environ.rel_context env_ar_par in
let ctx = Vars.subst_univs_level_context usubst ctx in
let env = Environ.pop_rel_context (Environ.nb_rel env_ar_par) env_ar_par in
Environ.push_rel_context ctx env
in
env_ar_par, univs, template, variance, record, params, Array.of_list data