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
open Declarations
open Libnames
open Constrexpr
open Constrintern
type module_internalization_error =
| NotAModuleNorModtype of qualid
| NotAModuleType of qualid
| NotAModule of qualid
| IncorrectWithInModule
| IncorrectModuleApplication
exception ModuleInternalizationError of module_internalization_error
type module_kind = Module | ModType | ModAny
let error_not_a_module_loc ~info kind loc qid =
let e = match kind with
| Module -> NotAModule qid
| ModType -> NotAModuleType qid
| ModAny -> NotAModuleNorModtype qid
in
let info = Option.cata (Loc.add_loc info) info loc in
Exninfo.iraise (ModuleInternalizationError e,info)
let error_incorrect_with_in_module loc =
Loc.raise ?loc (ModuleInternalizationError IncorrectWithInModule)
let error_application_to_module_type loc =
Loc.raise ?loc (ModuleInternalizationError IncorrectModuleApplication)
(** Searching for a module name in the Nametab.
According to the input module kind, modules or module types
or both are searched. The returned kind is never ModAny, and
it is equal to the input kind when this one isn't ModAny. *)
let lookup_module_or_modtype kind qid =
let loc = qid.CAst.loc in
try
if kind == ModType then raise Not_found;
let mp = Nametab.locate_module qid in
Dumpglob.dump_modref ?loc mp "modtype"; (mp,Module)
with Not_found ->
try
if kind == Module then raise Not_found;
let mp = Nametab.locate_modtype qid in
Dumpglob.dump_modref ?loc mp "mod"; (mp,ModType)
with Not_found as exn ->
let _, info = Exninfo.capture exn in
error_not_a_module_loc ~info kind loc qid
let lookup_module lqid = fst (lookup_module_or_modtype Module lqid)
let lookup_polymorphism env base kind fqid =
let m = match kind with
| Module -> (Environ.lookup_module base env).mod_type
| ModType -> (Environ.lookup_modtype base env).mod_type
| ModAny -> assert false
in
let rec defunctor = function
| NoFunctor m -> m
| MoreFunctor (_,_,m) -> defunctor m
in
let rec aux m fqid =
let open Names in
match fqid with
| [] -> assert false
| [id] ->
let test (lab,obj) =
match Id.equal (Label.to_id lab) id, obj with
| false, _ | _, (SFBmodule _ | SFBmodtype _) -> None
| true, SFBmind mind -> Some (Declareops.inductive_is_polymorphic mind)
| true, SFBconst const -> Some (Declareops.constant_is_polymorphic const)
in
(try CList.find_map test m with Not_found -> false )
| id::rem ->
let next = function
| MoreFunctor _ -> false
| NoFunctor body -> aux body rem
in
let test (lab,obj) =
match Id.equal (Label.to_id lab) id, obj with
| false, _ | _, (SFBconst _ | SFBmind _) -> None
| true, SFBmodule body -> Some (next body.mod_type)
| true, SFBmodtype body ->
Some (next body.mod_type)
in
(try CList.find_map test m with Not_found -> false )
in
aux (defunctor m) fqid
let intern_with_decl = function
| CWith_Module ({CAst.v=fqid},qid) ->
WithMod (fqid,lookup_module qid)
| CWith_Definition ({CAst.v=fqid},udecl,c) ->
WithDef (fqid,(udecl,c))
let loc_of_module l = l.CAst.loc
let rec intern_module_ast kind m = match m with
| {CAst.loc;v=CMident qid} ->
let (mp,kind) = lookup_module_or_modtype kind qid in
(MEident mp, mp, kind)
| {CAst.loc;v=CMapply (me1,me2)} ->
let me1', base, kind1 = intern_module_ast kind me1 in
let mp2, kind2 = lookup_module_or_modtype ModAny me2 in
if kind2 == ModType then
error_application_to_module_type (loc_of_module me2);
(MEapply (me1',mp2), base, kind1)
| {CAst.loc;v=CMwith (me,decl)} ->
let me,base,kind = intern_module_ast kind me in
if kind == Module then error_incorrect_with_in_module m.CAst.loc;
let decl = intern_with_decl decl in
(MEwith(me,decl), base, kind)
let interp_with_decl env base kind = function
| WithMod (fqid,mp) -> WithMod (fqid,mp), Univ.ContextSet.empty
| WithDef(fqid,(udecl,c)) ->
let sigma, udecl = interp_univ_decl_opt env udecl in
let c, ectx = interp_constr env sigma c in
let poly = lookup_polymorphism env base kind fqid in
begin match fst (UState.check_univ_decl ~poly ectx udecl) with
| UState.Polymorphic_entry ctx ->
let inst, ctx = Univ.abstract_universes ctx in
let c = EConstr.Vars.subst_univs_level_constr (Univ.make_instance_subst inst) c in
let c = EConstr.to_constr sigma c in
WithDef (fqid,(c, Some ctx)), Univ.ContextSet.empty
| UState.Monomorphic_entry ctx ->
let c = EConstr.to_constr sigma c in
WithDef (fqid,(c, None)), ctx
end
let rec interp_module_ast env kind base m cst = match m with
| MEident mp ->
MEident mp, cst
| MEapply (me,mp) ->
let me', cst = interp_module_ast env kind base me cst in
MEapply(me',mp), cst
| MEwith(me,decl) ->
let me, cst = interp_module_ast env kind base me cst in
let decl, cst' = interp_with_decl env base kind decl in
let cst = Univ.ContextSet.union cst cst' in
MEwith(me,decl), cst
let interp_module_ast env kind base m =
interp_module_ast env kind base m Univ.ContextSet.empty