ocaml/typing/ctype.ml

1612 lines
48 KiB
OCaml
Raw Normal View History

(***********************************************************************)
(* *)
(* Objective Caml *)
(* *)
(* Xavier Leroy and Jerome Vouillon, projet Cristal, INRIA Rocquencourt*)
(* *)
(* Copyright 1996 Institut National de Recherche en Informatique et *)
(* Automatique. Distributed only by permission. *)
(* *)
(***********************************************************************)
(* $Id$ *)
(* Operations on core types *)
open Misc
open Asttypes
open Types
(*
Type manipulation after type inference
======================================
If one wants to manipulate a type after type inference (for
instance, during code generation or in the debugger), one must
first make sure that the type levels are correct, using the
function [correct_levels]. Then, this type can be correctely
manipulated by [substitute], [expand_abbrev] and [moregeneral].
*)
(*
General notes
=============
- As much sharing as possible should be kept : it makes types
smaller and better abbreviated.
When necessary, some sharing can be lost. Types will still be
printed correctly (+++ TO DO...), and types defined by a class do
not depend on sharing thanks to constrained abbreviations. (Of
course, typing will still be correct.)
- All nodes of a type have a level : that way, one know whether a
node need to be duplicated or not when instantiating a type.
- Levels of a type are decreasing (generic level being considered
as greatest).
- The level of a type constructor is superior to the binding
time of its path.
- Recursive types without limitation should be handled (even if
there is still an occur check.
*)
(*
A faire
=======
- Renommer [substitute] en [apply]
- Revoir affichage des types.
- Types recursifs sans limitation.
- Etendre la portee d'un alias [... as 'a] a tout le type englobant.
- #-type implementes comme de vraies abreviations.
- Deplacer Ctype.repr dans Types ?
- Niveaux plus fins pour les identificateurs :
Champ [global] renomme en [level];
Niveau -1 : global
0 : module toplevel
1 : module contenu dans module toplevel
...
En fait, incrementer le niveau a chaque fois que l'on rentre dans
un module.
3 4 6
\ / /
1 2 5
\|/
0
[Subst] doit ecreter les niveaux (pour qu'un variable non
generalisable dans un module de niveau 2 ne se retrouve pas
generalisable lorsque l'on l'utilise au niveau 0).
- Traitement de la trace de l'unification separe de la fonction
[unify].
*)
(**** Errors ****)
exception Unify of (type_expr * type_expr) list
exception Subtype of
(type_expr * type_expr) list * (type_expr * type_expr) list
exception Cannot_expand
exception Recursive_abbrev
(**** Type level management ****)
let generic_level = (-1)
let current_level = ref 0
let global_level = ref 1
let init_def level = current_level := level
let begin_def () = incr current_level
let end_def () = decr current_level
let reset_global_level () =
global_level := !current_level + 1
(* Used to mark a type during a traversal. *)
let lowest_level = generic_level
let pivot_level = 2 * lowest_level - 1
(* pivot_level - lowest_level < lowest_level *)
(**** Some type creators ****)
let newty desc = { desc = desc; level = !current_level }
let newgenty desc = { desc = desc; level = generic_level }
let new_global_ty desc = { desc = desc; level = !global_level }
let newvar () = { desc = Tvar; level = !current_level }
let newgenvar () = newgenty Tvar
let new_global_var () = new_global_ty Tvar
let newmarkedvar () = { desc = Tvar; level = pivot_level - generic_level }
let newobj fields = newty (Tobject (fields, ref None))
let none = newty (Ttuple []) (* Clearly ill-formed type *)
(**** Representative of a type ****)
let rec repr =
function
{desc = Tlink t'} ->
(*
We do no path compression. Path compression does not seem to
improve notably efficiency, and it prevents from changing a
[Tlink] into another type (for instance, for undoing a
unification).
*)
repr t'
| t -> t
(**********************************)
(* Utilities for type traversal *)
(**********************************)
let saved_desc = ref []
(* Saved association of generic nodes with their description. *)
(* Restored type descriptions *)
let cleanup_types () =
List.iter (fun (ty, desc) -> ty.desc <- desc) !saved_desc;
saved_desc := []
(* Remove marks from a type. *)
let rec unmark_type ty =
let ty = repr ty in
if ty.level < lowest_level then begin
ty.level <- pivot_level - ty.level;
iter_type_expr unmark_type ty
end
(**********************************************)
(* Miscellaneous operations on object types *)
(**********************************************)
(**** Object field manipulation. ****)
let flatten_fields ty =
let rec flatten l ty =
let ty = repr ty in
match ty.desc with
Tfield(s, ty1, ty2) ->
flatten ((s, ty1)::l) ty2
| _ ->
(l, ty)
in
let (l, r) = flatten [] ty in
(List.rev l, r)
let build_fields =
List.fold_right (fun (s, ty1) ty2 -> newty (Tfield(s, ty1, ty2)))
let associate_fields fields1 fields2 =
let rec associate p s s' =
function
(l, []) ->
(List.rev p, (List.rev s) @ l, List.rev s')
| ([], l') ->
(List.rev p, List.rev s, (List.rev s') @ l')
| ((n, t)::r, (n', t')::r') when n = n' ->
associate ((t, t')::p) s s' (r, r')
| ((n, t)::r, ((n', t')::_ as l')) when n < n' ->
associate p ((n, t)::s) s' (r, l')
| (((n, t)::r as l), (n', t')::r') (* when n > n' *) ->
associate p s ((n', t')::s') (l, r')
in let sort = Sort.list (fun (n, _) (n', _) -> n < n') in
associate [] [] [] (sort fields1, sort fields2)
(**** Check whether an object is open ****)
(* +++ Il faudra penser a eventuellement expanser l'abreviation *)
let rec opened_object ty =
match (repr ty).desc with
Tobject (t, _) -> opened_object t
| Tfield(_, _, t) -> opened_object t
| Tvar -> true
| _ -> false
(**** Close an object ****)
let close_object ty =
let rec close ty =
let ty = repr ty in
match ty.desc with
Tvar -> ty.desc <- Tlink {desc = Tnil; level = ty.level}
| Tfield(_, _, ty') -> close ty'
| Tnil -> ()
| _ -> fatal_error "Ctype.close_object (1)"
in
match (repr ty).desc with
Tobject (ty, _) -> close ty
| Tconstr (_, _, _) -> () (* Already closed *)
| _ -> fatal_error "Ctype.close_object (2)"
(**** Object name manipulation ****)
(* +++ Bientot obsolete *)
let rec row_variable ty =
let ty = repr ty in
match ty.desc with
Tfield (_, _, ty) -> row_variable ty
| Tvar -> ty
| Tnil -> raise Not_found
| _ -> fatal_error "Ctype.row_variable"
let set_object_name ty params id =
match (repr ty).desc with
Tobject (fi, nm) ->
begin try
nm := Some (Path.Pident id, (row_variable fi)::params)
with Not_found ->
()
end
| Tconstr (_, _, _) ->
()
| _ ->
fatal_error "Ctype.set_object_name"
let remove_object_name ty =
match (repr ty).desc with
Tobject (_, nm) -> nm := None
| Tconstr (_, _, _) -> ()
| _ -> fatal_error "Ctype.remove_object_name"
(*****************************)
(* Type level manipulation *)
(*****************************)
(*
It would be a bit more efficient to remove abbreviation expansions
rather than generalizing them: these expansions will usually not be
used anymore. However, this is not possible in the general case, as
[expand_abbrev] (via [subst]) requires these expansions to be
preserved. Does it worth duplicating this code ?
*)
let rec generalize ty =
let ty = repr ty in
if ty.level > !current_level then begin
ty.level <- generic_level;
begin match ty.desc with
Tconstr (_, _, abbrev) ->
generalize_expans !abbrev
| _ -> ()
end;
iter_type_expr generalize ty
end
and generalize_expans =
function
Mnil -> ()
| Mcons(_, ty, rem) -> generalize ty; generalize_expans rem
| Mlink rem -> generalize_expans !rem
(*
Lower in-place the level of a generic type. That way, [subst] can
do "unification" on generic types.
*)
let rec ungeneralize ty =
let ty = repr ty in
if ty.level = generic_level then begin
ty.level <- !current_level;
begin match ty.desc with
Tconstr (_, _, abbrev) ->
ungeneralize_expans !abbrev
| _ -> ()
end;
iter_type_expr ungeneralize ty
end
and ungeneralize_expans =
function
Mnil -> ()
| Mcons(_, ty, rem) -> ungeneralize ty; ungeneralize_expans rem
| Mlink rem -> ungeneralize_expans !rem
let expand_abbrev' = (* Forward declaration *)
ref (fun env path args abbrev level -> raise Cannot_expand)
(*
Lower the levels of a type (assume [level] is not
[generic_level]).
*)
(*
The level of a type constructor must be greater than its binding
time. That way, a type constructor cannot escape the scope of its
definition, as would be the case in
let x = ref []
module M = struct type t let _ = (x : t list ref) end
(without this constraint, the type system would even be unsafe.)
*)
let rec update_level env level ty =
let ty = repr ty in
if ty.level > level then begin
let old_level = ty.level in
ty.level <- level;
begin match ty.desc with
Tconstr(p, tl, abbrev) when level < Path.binding_time p ->
(* Try first to replace an abbreviation by its expansion. *)
begin try
ty.desc <- Tlink (!expand_abbrev' env p tl abbrev old_level);
update_level env level ty
with Cannot_expand ->
(* +++ Levels should be restored... *)
raise (Unify [])
end
| _ ->
iter_type_expr (update_level env level) ty
end
end
(*
Function [update_level] will never try to expand an abbreviation in
this case ([current_level] is greater than the binding time of any
type constructor path). So, it can be called with the empty
environnement.
*)
let make_nongen ty = update_level Env.empty !current_level ty
(* Correct the levels of type [ty]. *)
let correct_levels env ty =
Subst.type_expr (Subst.limit_level (Env.level env) Subst.identity) ty
(*******************)
(* Instantiation *)
(*******************)
(*
Generic nodes are duplicated, while non-generic nodes are left
as-is. The instance cannot be generic.
During instantiation, the description of a generic node is first
replaced by a link to a stub ([Tlink (newvar ())]). Once the copy
is made, it replaces the stub.
After instantiation, the description of generic node, which was
stored in [saved_desc], must be put back, using [cleanup_types].
*)
let abbreviations = ref (ref Mnil)
(* Abbreviation memorized. *)
let rec copy ty =
let ty = repr ty in
if ty.level <> generic_level then
ty
else begin
let desc = ty.desc in
saved_desc := (ty, desc)::!saved_desc;
let t = newvar () in (* Stub *)
ty.desc <- Tlink t;
t.desc <-
begin match desc with
Tvar ->
Tvar
| Tarrow (t1, t2) ->
Tarrow (copy t1, copy t2)
| Ttuple tl ->
Ttuple (List.map copy tl)
| Tconstr (p, tl, _) ->
(*
One must allocate a new reference, so that abbrevia-
tions belonging to different branches of a type are
independent.
Moreover, a reference containing a [Mcons] must be
shared, so that the memorized expansion of an abbrevi-
ation can be released by changing the content of just
one reference.
*)
Tconstr (p, List.map copy tl,
ref (match ! !abbreviations with
Mcons _ -> Mlink !abbreviations
| abbrev -> abbrev))
| Tobject (t1, {contents = name}) ->
let name' =
match name with
None ->
None
| Some (p, tl) ->
Some (p, List.map copy tl)
in
Tobject (copy t1, ref name')
| Tfield (label, t1, t2) ->
Tfield (label, copy t1, copy t2)
| Tnil ->
Tnil
| Tlink t -> (* Actually unused *)
Tlink (copy t)
end;
t
end
(**** Variants of instantiations ****)
let instance sch =
let ty = copy sch in
cleanup_types ();
ty
let instance_list schl =
let tyl = List.map copy schl in
cleanup_types ();
tyl
let instance_constructor cstr =
let ty_res = copy cstr.cstr_res in
let ty_args = List.map copy cstr.cstr_args in
cleanup_types ();
(ty_args, ty_res)
let instance_label lbl =
let ty_res = copy lbl.lbl_res in
let ty_arg = copy lbl.lbl_arg in
cleanup_types ();
(ty_arg, ty_res)
let instance_parameterized_type sch_args sch =
let ty_args = List.map copy sch_args in
let ty = copy sch in
cleanup_types ();
(ty_args, ty)
let instance_parameterized_type_2 sch_args sch_lst sch =
let ty_args = List.map copy sch_args in
let ty_lst = List.map copy sch_lst in
let ty = copy sch in
cleanup_types ();
(ty_args, ty_lst, ty)
let instance_class cl =
let params = List.map copy cl.cty_params in
let args = List.map copy cl.cty_args in
let vars =
Vars.fold
(fun lab (mut, ty) ->
Vars.add lab (mut, copy ty))
cl.cty_vars
Vars.empty in
let self = copy cl.cty_self in
cleanup_types ();
(params, args, vars, self)
(**** Instantiation with parameter substitution ****)
let unify' = (* Forward declaration *)
ref (fun env ty1 ty2 -> raise (Unify []))
let rec subst env level abbrev path params args body =
if level <> generic_level then begin
let old_level = !current_level in
current_level := level;
let body0 = newvar () in (* Stub *)
begin match path with
None -> ()
| Some path -> memorize_abbrev abbrev path body0
end;
abbreviations := abbrev;
let (params', body') = instance_parameterized_type params body in
abbreviations := ref Mnil;
!unify' env body0 body';
List.iter2 (!unify' env) params' args;
current_level := old_level;
body'
end else begin
(* One cannot expand directly to a generic type. *)
begin_def ();
(* Make sure [!current_level] is high enough. *)
current_level := max !current_level (Env.level env);
(*
Arguments cannot be generic either, as they are unified to the
parameters.
*)
List.iter ungeneralize args;
let ty = subst env !current_level abbrev path params args body in
end_def ();
generalize ty;
List.iter generalize args;
ty
end
(*
Only the shape of the type matters, not whether is is generic or
not. [generic_level] might be somewhat slower, but it ensures
invariants on types are enforced (decreasing levels.), and we don't
care about efficiency here.
*)
let substitute env params args body =
subst env generic_level (ref Mnil) None params args body
(****************************)
(* Abbreviation expansion *)
(****************************)
(* Search whether the expansion has been memorized. *)
let rec find_expans p1 =
function
Mnil ->
None
| Mcons (p2, ty, _) when Path.same p1 p2 ->
Some ty
| Mcons (_, _, rem) ->
find_expans p1 rem
| Mlink {contents = rem} ->
find_expans p1 rem
let previous_env = ref Env.empty
(* Expand an abbreviation. The expansion is memorized. *)
(*
Assume the level is greater than the path binding time of the
expanded abbreviation.
*)
let expand_abbrev env path args abbrev level =
(*
If the environnement has changed, memorized expansions might not
be correct anymore, and so we flush the cache. This is safe but
quite pessimistic: it would be enough to flush the cache at the
ends of structures and signatures.
+++ Do it !
*)
if env != !previous_env then begin
cleanup_abbrev ();
previous_env := env
end;
match find_expans path !abbrev with
Some ty ->
if level <> generic_level then
update_level env level ty;
ty
| None ->
let decl =
(*
The type constructor is always in the environnement during
typing. This is not true anymore at a latter stage (code
generation and debugger), as a non-generic type variable
can be instanciated afterwards to a previously undefined
type constructor.
(+++ actually, it is still true for the moment, due to the
strong constraint on type levels and constructor binding
time.)
*)
try Env.find_type path env with Not_found -> raise Cannot_expand in
match decl.type_manifest with
Some body ->
subst env level abbrev (Some path) decl.type_params args body
| None ->
raise Cannot_expand
let _ = expand_abbrev' := expand_abbrev
(* Fully expand the head of a type. *)
let rec expand_head env ty =
let ty = repr ty in
match ty.desc with
Tconstr(p, tl, abbrev) ->
begin try
expand_head env (expand_abbrev env p tl abbrev ty.level)
with Cannot_expand ->
ty
end
| _ ->
ty
(* Recursively expand the head of a type.
Also expand #-types. *)
let rec full_expand env ty =
let ty = repr (expand_head env ty) in
match ty.desc with
Tobject (fi, {contents = Some (_, v::_)}) when (repr v).desc = Tvar ->
{ desc = Tobject (fi, ref None); level = ty.level }
| _ ->
ty
(*
Check whether the abbreviation expands to a well-defined type.
During the typing of a class, abbreviations for correspondings
types expand to non-generic types.
*)
let generic_abbrev env path =
try
let decl = Env.find_type path env in
match decl.type_manifest with
Some body -> (repr body).level = generic_level
| None -> false
with
Not_found ->
false
(*****************)
(* Unification *)
(*****************)
(**** Occur check ****)
exception Occur
(* +++ A supprimer (?) *)
let occur env ty0 ty =
let visited = ref ([] : type_expr list) in
let rec occur_rec ty =
if ty == ty0 then raise Occur;
match ty.desc with
Tconstr(p, tl, abbrev) ->
if not (List.memq ty !visited) then begin
visited := ty :: !visited;
try List.iter occur_rec tl with Occur ->
try
let ty' = expand_abbrev env p tl abbrev ty.level in
occur_rec ty'
with Cannot_expand -> ()
end
| Tobject (_, _) ->
()
| _ ->
iter_type_expr occur_rec ty
in
occur_rec ty
(**** Transform error trace ****)
(* +++ Move it to some other place ? *)
let expand_trace env trace =
List.fold_right
(fun (t1, t2) rem ->
(repr t1, full_expand env t1)::(repr t2, full_expand env t2)::rem)
trace []
let rec filter_trace =
function
(t1, t1')::(t2, t2')::rem ->
let rem' = filter_trace rem in
if (t1 == t1') & (t2 == t2')
then rem'
else (t1, t1')::(t2, t2')::rem'
| _ ->
[]
(**** Unification ****)
(* Return whether [t0] occurs in [ty]. Objects are also traversed. *)
let deep_occur t0 ty =
let rec occur_rec ty =
let ty = repr ty in
if ty.level >= lowest_level then begin
if ty == t0 then raise Occur;
ty.level <- pivot_level - ty.level;
iter_type_expr occur_rec ty
end
in
try
occur_rec ty; unmark_type ty; false
with Occur ->
unmark_type ty; true
(*
1. When unifying two non-abbreviated types, one type is made a link
to the other. When unifying an abbreviated type with a
non-abbreviated type, the non-abbreviated type is made a link to
the other one. When unifying to abbreviated types, these two
types are kept distincts, but they are made to (temporally)
expand to the same type.
2. Abbreviations with at least one parameter are systematically
expanded. The overhead does not seem to high, and that way
abbreviations where some parameters does not appear in the
expansion, such as ['a t = int], are correctly handled. In
particular, for this example, unifying ['a t] with ['b t] keeps
['a] and ['b] distincts. (Is it really important ?)
3. Unifying an abbreviation ['a t = 'a] with ['a] should not yield
['a t as 'a]. Indeed, the type variable would otherwise be lost.
This problem occurs for abbreviations expanding to a type
variable, but also to many other constrained abbreviations (for
instance, [(< x : 'a > -> unit) t = <x : 'a>]). The solution is
that, if an abbreviation is unified with some subpart of its
parameters, then the parameter actually does not get
abbreviated. It would be possible to check whether some
information is indeed lost, but it probably does not worth it.
*)
let rec unify env t1 t2 =
(* First step: special cases (optimizations) *)
if t1 == t2 then () else
let t1 = repr t1 in
let t2 = repr t2 in
if t1 == t2 then () else
try
match (t1.desc, t2.desc) with
(Tvar, Tconstr _) when deep_occur t1 t2 ->
unify2 env t1 t2
| (Tconstr _, Tvar) when deep_occur t2 t1 ->
unify2 env t1 t2
| (Tvar, _) ->
begin try occur env t1 t2 with Occur ->
raise (Unify [])
end;
update_level env t1.level t2;
t1.desc <- Tlink t2
| (_, Tvar) ->
begin try occur env t2 t1 with Occur ->
raise (Unify [])
end;
update_level env t2.level t1;
t2.desc <- Tlink t1
| (Tconstr (p1, [], _), Tconstr (p2, [], _)) when Path.same p1 p2 ->
update_level env t1.level t2;
t1.desc <- Tlink t2
| _ ->
unify2 env t1 t2
with Unify trace ->
raise (Unify ((t1, t2)::trace))
and unify2 env t1 t2 =
(* Second step: expansion of abbreviations *)
let t1' = expand_head env t1 in
let t2' = expand_head env t2 in
if t1' == t2' then () else
if (t1 == t1') || (t2 != t2') then
unify3 env t1 t1' t2 t2'
else
try unify3 env t2 t2' t1 t1' with Unify trace ->
raise (Unify (List.map (fun (x, y) -> (y, x)) trace))
and unify3 env t1 t1' t2 t2' =
(* Third step: truly unification *)
(* Assumes either [t1 == t1'] or [t2 != t2'] *)
let d1 = t1'.desc and d2 = t2'.desc in
if (t2 != t2') && (deep_occur t1' t2) then begin
(* See point 3. *)
update_level env t1'.level t2';
t1'.desc <- Tlink t2'
end else begin
update_level env t1'.level t2;
t1'.desc <- Tlink t2
end;
try
begin match (d1, d2) with
(Tvar, _) ->
begin try occur env t1' t2 with Occur ->
raise (Unify [])
end
| (_, Tvar) ->
begin try occur env t2' (newty d1) with Occur ->
raise (Unify [])
end;
if t1 == t1' then begin
(* The variable must be instantiated... *)
let ty = {desc = d1; level = t1'.level} in
update_level env t2'.level ty;
t2'.desc <- Tlink ty
end else begin
t1'.desc <- d1;
update_level env t2'.level t1;
t2'.desc <- Tlink t1
end
| (Tarrow (t1, u1), Tarrow (t2, u2)) ->
unify env t1 t2; unify env u1 u2
| (Ttuple tl1, Ttuple tl2) ->
unify_list env tl1 tl2
| (Tconstr (p1, tl1, _), Tconstr (p2, tl2, _)) when Path.same p1 p2 ->
unify_list env tl1 tl2
| (Tobject (fi1, nm1), Tobject (fi2, nm2)) ->
unify_fields env fi1 fi2;
begin match !nm2 with
Some (_, va::_) when (repr va).desc = Tvar -> ()
| _ -> nm2 := !nm1
end
| (Tfield _, Tfield _) -> (* Actually unused *)
unify_fields env t1 t2
| (Tnil, Tnil) ->
()
| (_, _) ->
raise (Unify [])
end
(*
(*
Can only be done afterwards, once the row variable has
(possibly) been instantiated.
*)
if t1 != t1' (* && t2 != t2' *) then begin
match (t1.desc, t2.desc) with
(Tconstr (p, ty::_, _), _)
when ((repr ty).desc != Tvar)
&& weak_abbrev p
&& not (deep_occur t1 t2) ->
update_level env t1.level t2;
t1.desc <- Tlink t2
| (_, Tconstr (p, ty::_, _))
when ((repr ty).desc != Tvar)
&& weak_abbrev p
&& not (deep_occur t2 t1) ->
update_level env t2.level t1;
t2.desc <- Tlink t1;
t1'.desc <- Tlink t2'
| _ ->
()
end
*)
with Unify trace ->
t1'.desc <- d1;
raise (Unify trace)
and unify_list env tl1 tl2 =
if List.length tl1 <> List.length tl2 then
raise (Unify []);
List.iter2 (unify env) tl1 tl2
and unify_fields env ty1 ty2 = (* Optimization *)
let (fields1, rest1) = flatten_fields ty1
and (fields2, rest2) = flatten_fields ty2 in
let (pairs, miss1, miss2) = associate_fields fields1 fields2 in
let va = newvar () in
unify env rest1 (build_fields miss2 va);
unify env (build_fields miss1 va) rest2;
List.iter (fun (t1, t2) -> unify env t1 t2) pairs
let unify env ty1 ty2 =
try
unify env ty1 ty2
with Unify trace ->
let trace = expand_trace env trace in
match trace with
t1::t2::rem ->
raise (Unify (t1::t2::filter_trace rem))
| _ ->
fatal_error "Ctype.unify"
let _ = unify' := unify
(**** Special cases of unification ****)
(* Unify [t] and ['a -> 'b]. Return ['a] and ['b]. *)
let rec filter_arrow env t =
let t = repr t in
match t.desc with
Tvar ->
let t1 = newvar () and t2 = newvar () in
let t' = newty (Tarrow (t1, t2)) in
update_level env t.level t';
t.desc <- Tlink t';
(t1, t2)
| Tarrow(t1, t2) ->
(t1, t2)
| Tconstr(p, tl, abbrev) ->
begin try
filter_arrow env (expand_abbrev env p tl abbrev t.level)
with Cannot_expand ->
raise (Unify [])
end
| _ ->
raise (Unify [])
(* Used by [filter_method]. *)
let rec filter_method_field env name ty =
let ty = repr ty in
match ty.desc with
Tvar ->
let ty1 = newvar () and ty2 = newvar () in
let ty' = newty (Tfield (name, ty1, ty2)) in
update_level env ty.level ty';
ty.desc <- Tlink ty';
ty1
| Tfield(n, ty1, ty2) ->
if n = name then
ty1
else
filter_method_field env name ty2
| _ ->
raise (Unify [])
(* Unify [ty] and [< name : 'a; .. >]. Return ['a]. *)
let rec filter_method env name ty =
let ty = repr ty in
match ty.desc with
Tvar ->
let ty1 = newvar () in
let ty' = newobj ty1 in
update_level env ty.level ty';
ty.desc <- Tlink ty';
filter_method_field env name ty1
| Tobject(f, _) ->
filter_method_field env name f
| Tconstr(p, tl, abbrev) ->
begin try
filter_method env name (expand_abbrev env p tl abbrev ty.level)
with Cannot_expand ->
raise (Unify [])
end
| _ ->
raise (Unify [])
(***********************************)
(* Matching between type schemes *)
(***********************************)
(*
Levels:
!current_level : generic variables from the pattern
!current_level - 1 : generic variables from the subject
A generic variable from the subject cannot be instantiated, and its
level must remain unchanged. A generic variable from the pattern
can be instantiated to anything.
Usually, the subject is given by the user, and the pattern is
unimportant. So, no need to propagate abbreviations.
*)
(*
Update the level of [ty]. Check that the levels of generic variables
from the subject are not lowered.
*)
let moregen_occur env level ty =
let rec occur_rec ty =
let ty = repr ty in
if ty.level >= !current_level - 1 then begin
if (ty.desc = Tvar) && (ty.level = !current_level - 1) then
raise Occur;
ty.level <- pivot_level - ty.level;
iter_type_expr occur_rec ty
end
in
if level < !current_level - 1 then begin
begin try
occur_rec ty; unmark_type ty
with Occur ->
unmark_type ty; raise (Unify [])
end;
update_level env level ty
end
let rec moregen env t1 t2 =
if t1 == t2 then () else
let t1 = repr t1 in
let t2 = repr t2 in
if t1 == t2 then () else
let d2 = ref t2.desc in
moregen_occur env t1.level t2;
(* Ensure termination *)
t2.desc <- Tlink t1;
try
begin match (t1.desc, !d2) with
(Tvar, _) when t1.level <> !current_level - 1 ->
t2.desc <- !d2;
moregen_occur env t1.level t2;
t1.desc <- Tlink t2
| (_, Tvar) when t2.level <> !current_level - 1 ->
moregen_occur env t2.level t1;
d2 := Tlink t1
| (Tarrow(t1, u1), Tarrow(t2, u2)) ->
moregen env t1 t2; moregen env u1 u2
| (Ttuple tl1, Ttuple tl2) ->
moregen_list env tl1 tl2
| (Tconstr(p1, tl1, abbrev1), Tconstr(p2, tl2, abbrev2))
when Path.same p1 p2 ->
begin try
moregen_list env tl1 tl2
with Unify _ ->
try
moregen env t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
with Cannot_expand ->
t2.desc <- !d2;
try
moregen env (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
with Cannot_expand ->
raise (Unify [])
end
| (Tconstr(p1, tl1, abbrev1), Tconstr(p2, tl2, abbrev2)) ->
begin try
moregen env t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
with Cannot_expand ->
t2.desc <- !d2;
try
moregen env (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
with Cannot_expand ->
raise (Unify [])
end
| (Tobject(f1, _), Tobject(f2, _)) ->
moregen_fields env f1 f2
| (Tconstr(p1, tl1, abbrev1), _) ->
t2.desc <- !d2;
begin try
moregen env (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
with Cannot_expand ->
raise (Unify [])
end
| (_, Tconstr(p2, tl2, abbrev2)) ->
begin try
moregen env t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
with Cannot_expand ->
raise (Unify [])
end
| (Tfield _, Tfield _) -> (* Actually unused *)
t2.desc <- !d2;
moregen_fields env t1 t2
| (Tnil, Tnil) ->
()
| (_, _) ->
raise (Unify [])
end;
t2.desc <- !d2
with exn ->
t2.desc <- !d2;
raise exn
and moregen_list env tl1 tl2 =
if List.length tl1 <> List.length tl2 then
raise (Unify []);
List.iter2 (moregen env) tl1 tl2
and moregen_fields env ty1 ty2 =
let (fields1, rest1) = flatten_fields ty1
and (fields2, rest2) = flatten_fields ty2 in
let (pairs, miss1, miss2) = associate_fields fields1 fields2 in
let va = newvar () in
moregen env (build_fields miss1 va) rest2;
moregen env rest1 (build_fields miss2 va);
List.iter (fun (t1, t2) -> moregen env t1 t2) pairs
let moregeneral env sch1 sch2 =
(* Make sure [!current_level] is high enough. *)
current_level := max !current_level (Env.level env);
begin_def ();
let ty2 = instance sch2 in
begin_def ();
let ty1 = instance sch1 in
try
moregen env ty1 ty2;
end_def ();
end_def ();
true
with Unify _ ->
end_def ();
end_def ();
false
(*********************************************)
(* Equivalence between parameterized types *)
(*********************************************)
(* +++ A voir... *)
(* Deux modes : avec ou sans subtitution *)
(* Equalite de deux listes de types : *)
(* [Ctype.equal env rename tyl1 tyl2] *)
let equal env rename tyl1 tyl2 =
let subst = ref [] in
let type_pairs = ref [] in
let rec eqtype t1 t2 =
let t1 = repr t1 in
let t2 = repr t2 in
List.exists (function (t1', t2') -> t1 == t1' & t2 == t2') !type_pairs
(* XXX Possibly slow... *)
||
begin
type_pairs := (t1, t2) :: !type_pairs;
match (t1.desc, t2.desc) with
(Tvar, Tvar) ->
if rename then begin
try
List.assq t1 !subst == t2
with Not_found ->
subst := (t1, t2) :: !subst;
true
end else
t1 == t2
| (Tarrow(t1, u1), Tarrow(t2, u2)) ->
eqtype t1 t2 & eqtype u1 u2
| (Ttuple tl1, Ttuple tl2) ->
eqtype_list tl1 tl2
| (Tconstr(p1, tl1, abbrev1), Tconstr(p2, tl2, abbrev2)) ->
(Path.same p1 p2 && eqtype_list tl1 tl2)
||
begin
try
eqtype (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
with Cannot_expand ->
try
eqtype t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
with Cannot_expand ->
false
end
| (Tobject (f1, _), Tobject (f2, _)) ->
eqtype_fields f1 f2
| (Tconstr(p1, tl1, abbrev1), _) ->
begin try
eqtype (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
with Cannot_expand ->
false
end
| (_, Tconstr(p2, tl2, abbrev2)) ->
begin try
eqtype t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
with Cannot_expand ->
false
end
| (Tfield _, Tfield _) ->
eqtype_fields t1 t2
| (Tnil, Tnil) ->
true
| (_, _) ->
false
end
and eqtype_list tl1 tl2 =
List.length tl1 = List.length tl2
&&
List.for_all2 eqtype tl1 tl2
and eqtype_fields ty1 ty2 = (* Optimization *)
let (fields1, rest1) = flatten_fields ty1
and (fields2, rest2) = flatten_fields ty2 in
let (pairs, miss1, miss2) = associate_fields fields1 fields2 in
eqtype rest1 rest2
&&
(miss1 = []) && (miss2 = [])
&&
List.for_all (function (t1, t2) -> eqtype t1 t2) pairs
in
eqtype_list tyl1 tyl2
(***************)
(* Subtyping *)
(***************)
(**** Build a subtype of a given type. ****)
let subtypes = ref []
(* XXX Types r<>cursifs ? *)
let rec build_subtype env t =
let t = repr t in
match t.desc with
Tlink t' -> (* Redundant ! *)
build_subtype env t'
| Tvar ->
(t, false)
| Tarrow(t1, t2) ->
let (t1', c1) = (t1, false) in
let (t2', c2) = build_subtype env t2 in
if c1 or c2 then (new_global_ty (Tarrow(t1', t2')), true)
else (t, false)
| Ttuple tlist ->
let (tlist', clist) =
List.split (List.map (build_subtype env) tlist)
in
if List.exists (function c -> c) clist then
(new_global_ty (Ttuple tlist'), true)
else (t, false)
| Tconstr(p, tl, abbrev) when generic_abbrev env p ->
let t' = expand_abbrev env p tl abbrev t.level in
let (t'', c) = build_subtype env t' in
if c then (t'', true)
else (t, false)
| Tconstr(p, tl, abbrev) ->
(t, false)
| Tobject (t1, _) when opened_object t1 ->
(t, false)
| Tobject (t1, _) ->
(begin try
List.assq t !subtypes
with Not_found ->
let t' = new_global_var () in
subtypes := (t, t')::!subtypes;
let (t1', _) = build_subtype env t1 in
t'.desc <- Tobject (t1', ref None);
t'
end,
true)
| Tfield(s, t1, t2) ->
let (t1', _) = build_subtype env t1 in
let (t2', _) = build_subtype env t2 in
(new_global_ty (Tfield(s, t1', t2')), true)
| Tnil ->
let v = new_global_var () in
(v, true)
let enlarge_type env ty =
subtypes := [];
let (ty', _) = build_subtype env ty in
subtypes := [];
ty'
(**** Check whether a type is a subtype of another type. ****)
(*
During the traversal, a trace of visited types is maintained. It
is printed in case of error.
Constraints (pairs of types that must be equals) are accumulated
rather than being enforced straight. Indeed, the result would
otherwise depend on the order in which these constraints are
enforced.
A function enforcing these constraints is returned. That way, type
variables can be bound to their actual values before this function
is called (see Typecore).
Only well-defined abbreviations are expanded (hence the tests
[generic_abbrev ...]).
*)
let subtypes = ref [];;
let subtype_error env trace =
raise (Subtype (expand_trace env (List.rev trace), []))
let rec subtype_rec env trace t1 t2 =
let t1 = repr t1 in
let t2 = repr t2 in
if t1 == t2 then [] else
if List.exists (fun (t1', t2') -> t1 == t1' & t2 == t2') !subtypes then
(* +++ Possibly slow *)
[]
else begin
subtypes := (t1, t2) :: !subtypes;
match (t1.desc, t2.desc) with
(Tvar, _) | (_, Tvar) ->
[(trace, t1, t2)]
| (Tarrow(t1, u1), Tarrow(t2, u2)) ->
(subtype_rec env ((t2, t1)::trace) t2 t1) @
(subtype_rec env ((u1, u2)::trace) u1 u2)
| (Ttuple tl1, Ttuple tl2) ->
subtype_list env trace tl1 tl2
| (Tconstr(p1, tl1, abbrev1), Tconstr _) when generic_abbrev env p1 ->
subtype_rec env trace (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
| (Tconstr _, Tconstr(p2, tl2, abbrev2)) when generic_abbrev env p2 ->
subtype_rec env trace t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
| (Tconstr _, Tconstr _) ->
[(trace, t1, t2)]
| (Tconstr(p1, tl1, abbrev1), _) when generic_abbrev env p1 ->
subtype_rec env trace (expand_abbrev env p1 tl1 abbrev1 t1.level) t2
| (_, Tconstr (p2, tl2, abbrev2)) when generic_abbrev env p2 ->
subtype_rec env trace t1 (expand_abbrev env p2 tl2 abbrev2 t2.level)
| (Tobject (f1, _), Tobject (f2, _))
when opened_object f1 & opened_object f2 ->
(* Same row variable implies same object. *)
[(trace, t1, t2)]
| (Tobject (f1, _), Tobject (f2, _)) ->
subtype_fields env trace f1 f2
| (_, _) ->
subtype_error env trace
end
and subtype_list env trace tl1 tl2 =
if List.length tl1 <> List.length tl2 then
subtype_error env trace;
List.fold_left2
(fun cstrs t1 t2 -> cstrs @ (subtype_rec env ((t1, t2)::trace) t1 t2))
[] tl1 tl2
and subtype_fields env trace ty1 ty2 =
let (fields1, rest1) = flatten_fields ty1 in
let (fields2, rest2) = flatten_fields ty2 in
let (pairs, miss1, miss2) = associate_fields fields1 fields2 in
[(trace, rest1, build_fields miss2 (newvar ()))]
@
begin match rest2.desc with
Tnil -> []
| _ -> [(trace, build_fields miss1 rest1, rest2)]
end
@
(List.fold_left
(fun cstrs (t1, t2) -> cstrs @ (subtype_rec env ((t1, t2)::trace) t1 t2))
[] pairs)
let subtype env ty1 ty2 =
subtypes := [];
(* Build constraint set. *)
let cstrs = subtype_rec env [(ty1, ty2)] ty1 ty2 in
(* Enforce constraints. *)
function () ->
List.iter
(function (trace0, t1, t2) ->
try unify env t1 t2 with Unify trace ->
raise (Subtype (expand_trace env (List.rev trace0),
List.tl (List.tl trace))))
cstrs;
subtypes := []
(*******************)
(* Miscellaneous *)
(*******************)
let unalias ty =
let ty = repr ty in
match ty.desc with
Tvar ->
ty
| _ ->
{desc = ty.desc; level = ty.level}
let unroll_abbrev id tl ty =
let ty = repr ty in
match ty.desc with
Tvar ->
ty
| _ ->
let ty' = {desc = ty.desc; level = ty.level} in
ty.desc <- Tlink {desc = Tconstr (Path.Pident id, tl, ref Mnil);
level = ty.level};
ty'
(* Return the arity (as for curried functions) of the given type. *)
let rec arity ty =
match (repr ty).desc with
Tarrow(t1, t2) -> 1 + arity t2
| _ -> 0
(*************************)
(* Remove dependencies *)
(*************************)
(*
Variables are left unchanged. Other type nodes are duplicated, with
levels set to generic level.
During copying, the description of a (non-variable) node is first
replaced by a link to a marked stub ([Tlink (newmarkedvar
())]). The mark allows to differentiate the original type from the
copy. Once the copy is made, it replaces the stub.
After copying, the description of node, which was stored in
[saved_desc], must be put back, using [cleanup_types], and the
marks on the copy must be removed.
*)
let rec nondep_type_rec env id ty =
let ty = repr ty in
if (ty.desc = Tvar) || (ty.level < lowest_level) then
ty
else begin
let desc = ty.desc in
saved_desc := (ty, desc)::!saved_desc;
let ty' = newmarkedvar () in (* Stub *)
ty.desc <- Tlink ty';
ty'.desc <-
begin match desc with
Tvar ->
fatal_error "Ctype.nondep_type_rec"
| Tarrow(t1, t2) ->
Tarrow(nondep_type_rec env id t1, nondep_type_rec env id t2)
| Ttuple tl ->
Ttuple(List.map (nondep_type_rec env id) tl)
| Tconstr(p, tl, abbrev) ->
if Path.isfree id p then
begin try
Tlink (nondep_type_rec env id
(expand_abbrev env p tl abbrev ty.level))
(*
The [Tlink] is important. The expanded type may be a
variable, or may not be completely copied yet
(recursive type), so one cannot just take its
description.
*)
with Cannot_expand ->
raise Not_found
end
else
Tconstr(p, List.map (nondep_type_rec env id) tl, ref Mnil)
| Tobject (t1, name) ->
Tobject (nondep_type_rec env id t1,
ref (match !name with
None -> None
| Some (p, tl) ->
if Path.isfree id p then None
else Some (p, List.map (nondep_type_rec env id) tl)))
| Tfield(label, t1, t2) ->
Tfield(label, nondep_type_rec env id t1, nondep_type_rec env id t2)
| Tnil ->
Tnil
| Tlink ty -> (* Actually unused *)
Tlink(nondep_type_rec env id ty)
end;
ty'
end
let nondep_type env id ty =
try
let ty' = nondep_type_rec env id ty in
cleanup_types ();
unmark_type ty';
ty'
with Not_found ->
cleanup_types ();
raise Not_found
(* Preserve sharing inside type declarations. *)
let nondep_type_decl env mid id is_covariant decl =
try
let params = List.map (nondep_type_rec env mid) decl.type_params in
let decl =
{ type_params = params;
type_arity = decl.type_arity;
type_kind =
begin try
match decl.type_kind with
Type_abstract ->
Type_abstract
| Type_variant cstrs ->
Type_variant(List.map
(fun (c, tl) -> (c, List.map (nondep_type_rec env mid) tl))
cstrs)
| Type_record lbls ->
Type_record(List.map
(fun (c, mut, t) -> (c, mut, nondep_type_rec env mid t))
lbls)
with Not_found when is_covariant ->
Type_abstract
end;
type_manifest =
begin try
match decl.type_manifest with
None -> None
| Some ty ->
Some (unroll_abbrev id params (nondep_type_rec env mid ty))
with Not_found when is_covariant ->
None
end }
in
cleanup_types ();
List.iter unmark_type decl.type_params;
begin match decl.type_kind with
Type_abstract -> ()
| Type_variant cstrs ->
List.iter (fun (c, tl) -> List.iter unmark_type tl) cstrs
| Type_record lbls ->
List.iter (fun (c, mut, t) -> unmark_type t) lbls
end;
begin match decl.type_manifest with
None -> ()
| Some ty -> unmark_type ty
end;
decl
with Not_found ->
cleanup_types ();
raise Not_found
(* Preserve sharing inside class types. *)
let nondep_class_type env id decl =
try
let decl =
{ cty_params = List.map (nondep_type_rec env id) decl.cty_params;
cty_args = List.map (nondep_type_rec env id) decl.cty_args;
cty_vars =
Vars.fold (fun l (m, t) -> Vars.add l (m, nondep_type_rec env id t))
decl.cty_vars Vars.empty;
cty_self = nondep_type_rec env id decl.cty_self;
cty_concr = decl.cty_concr;
cty_new =
begin match decl.cty_new with
None -> None
| Some ty -> Some (nondep_type_rec env id ty)
end }
in
cleanup_types ();
List.iter unmark_type decl.cty_params;
List.iter unmark_type decl.cty_args;
Vars.iter (fun l (m, t) -> unmark_type t) decl.cty_vars;
unmark_type decl.cty_self;
begin match decl.cty_new with
None -> ()
| Some ty -> unmark_type ty
end;
decl
with Not_found ->
cleanup_types ();
raise Not_found
(******************************)
(* Abbreviation correctness *)
(******************************)
(* +++ A supprimer... (occur-check) *)
let marked_types = ref []
let rec non_recursive_abbrev env path ty =
let ty = repr ty in
if ty.level >= lowest_level then begin
let level = ty.level in
ty.level <- pivot_level - level;
match ty.desc with
Tconstr(p, _, _) when Path.same path p ->
raise Recursive_abbrev
| Tconstr(p, args, abbrev) ->
begin try
let ty' = repr (expand_abbrev env p args abbrev level) in
if ty'.level >= lowest_level then begin
marked_types := ty' :: !marked_types;
non_recursive_abbrev env path ty'
end
with Cannot_expand -> () end
| Tobject (_, _) ->
()
| _ ->
iter_type_expr (non_recursive_abbrev env path) ty
end
let correct_abbrev env ident params ty =
marked_types := [ty];
non_recursive_abbrev env (Path.Pident ident) ty;
List.iter unmark_type !marked_types;
marked_types := []
(**************************************)
(* Check genericity of type schemes *)
(**************************************)
type closed_schema_result = Var of type_expr | Row_var of type_expr
exception Failed of closed_schema_result
let rec closed_schema_rec fullgen row ty =
let ty = repr ty in
if ty.level >= lowest_level then begin
let level = ty.level in
if fullgen then
ty.level <- pivot_level - generic_level (* Generalize type *)
else
ty.level <- pivot_level - level;
match ty.desc with
Tvar when level != generic_level ->
raise (Failed (if row then Row_var ty else Var ty))
| Tobject(f, {contents = Some (_, p)}) ->
closed_schema_rec fullgen true f;
List.iter (closed_schema_rec fullgen false) p
| Tobject(f, _) ->
closed_schema_rec fullgen true f
| Tfield(_, t1, t2) ->
closed_schema_rec fullgen false t1;
closed_schema_rec fullgen true t2
| _ ->
iter_type_expr (closed_schema_rec fullgen false) ty
end
(*
Return whether all variables of type [ty] are generic. The type is
also generalized if [fullgen] is true.
A file interface should be fully generic. On the other hand, class
abbreviation expansion cannot be made fully generic (this could
break type level invariant).
*)
let closed_schema fullgen ty =
try
closed_schema_rec fullgen false ty;
unmark_type ty;
true
with Failed _ ->
unmark_type ty;
false
(*
Check that all variables of type [ty] are generic. If this is not
the case, the non-generic variable is returned. The type is never
generalized.
*)
let closed_schema_verbose ty =
try
closed_schema_rec false false ty;
unmark_type ty;
None
with Failed status ->
unmark_type ty;
Some status