edoc: Use module erl_anno

7 files changed, 36 insertions, 38 deletions

diff --git a/lib/edoc/src/edoc.erl b/lib/edoc/src/edoc.erl
index 88e7ab5346..90f1fc3071 100644
--- a/lib/edoc/src/edoc.erl
+++ b/lib/edoc/src/edoc.erl
@@ -689,13 +689,12 @@ scan_and_parse(Epp) ->
 
 fix_last_line(Toks0) ->
     Toks1 = lists:reverse(Toks0),
-    {line, LastLine} = erl_scan:token_info(hd(Toks1), line),
+    LastLine = erl_scan:line(hd(Toks1)),
     fll(Toks1, LastLine, []).
 
-fll([{Category, Attributes0, Symbol} | L], LastLine, Ts) ->
-    F = fun(_OldLine) -> LastLine end,
-    Attributes = erl_scan:set_attribute(line, Attributes0, F),
-    lists:reverse(L, [{Category, Attributes, Symbol} | Ts]);
+fll([{Category, Anno0, Symbol} | L], LastLine, Ts) ->
+    Anno = erl_anno:set_line(LastLine, Anno0),
+    lists:reverse(L, [{Category, Anno, Symbol} | Ts]);
 fll([T | L], LastLine, Ts) ->
     fll(L, LastLine, [T | Ts]);
 fll(L, _LastLine, Ts) ->
diff --git a/lib/edoc/src/edoc_layout.erl b/lib/edoc/src/edoc_layout.erl
index 6309e88475..62d5eb9a18 100644
--- a/lib/edoc/src/edoc_layout.erl
+++ b/lib/edoc/src/edoc_layout.erl
@@ -535,7 +535,8 @@ t_clause(Name, Type) ->
 pp_clause(Pre, Type) ->
     Types = ot_utype([Type]),
     Atom = lists:duplicate(iolist_size(Pre), $a),
-    L1 = erl_pp:attribute({attribute,0,spec,{{list_to_atom(Atom),0},[Types]}}),
+    Attr = {attribute,0,spec,{{list_to_atom(Atom),0},[Types]}},
+    L1 = erl_pp:attribute(erl_parse:new_anno(Attr)),
     "-spec " ++ L2 = lists:flatten(L1),
     L3 = Pre ++ lists:nthtail(length(Atom), L2),
     re:replace(L3, "\n      ", "\n", [{return,list},global]).
@@ -555,7 +556,8 @@ format_type(Prefix, _Name, Type, Last, _Opts) ->
 
 pp_type(Prefix, Type) ->
     Atom = list_to_atom(lists:duplicate(iolist_size(Prefix), $a)),
-    L1 = erl_pp:attribute({attribute,0,type,{Atom,ot_utype(Type),[]}}),
+    Attr = {attribute,0,type,{Atom,ot_utype(Type),[]}},
+    L1 = erl_pp:attribute(erl_parse:new_anno(Attr)),
     {L2,N} = case lists:dropwhile(fun(C) -> C =/= $: end, lists:flatten(L1)) of
                  ":: " ++ L3 -> {L3,9}; % compensation for extra "()" and ":"
                  "::\n" ++ L3 -> {"\n"++L3,6}
@@ -1085,8 +1087,8 @@ ot_var(E) ->
     {var,0,list_to_atom(get_attrval(name, E))}.
 
 ot_atom(E) ->
-    {ok, [Atom], _} = erl_scan:string(get_attrval(value, E), 0),
-    Atom.
+    {ok, [{atom,A,Name}], _} = erl_scan:string(get_attrval(value, E), 0),
+    {atom,erl_anno:line(A),Name}.
 
 ot_integer(E) ->
     {integer,0,list_to_integer(get_attrval(value, E))}.
diff --git a/lib/edoc/src/edoc_lib.erl b/lib/edoc/src/edoc_lib.erl
index c248964dc4..dcc239f6b4 100644
--- a/lib/edoc/src/edoc_lib.erl
+++ b/lib/edoc/src/edoc_lib.erl
@@ -1012,7 +1012,7 @@ get_plugin(Key, Default, Opts) ->
 %% ---------------------------------------------------------------------
 %% Error handling
 
--type line() :: erl_scan:line().
+-type line() :: erl_anno:line().
 -type err()  :: 'eof'
 	      | {'missing', char()}
 	      | {line(), atom(), string()}
diff --git a/lib/edoc/src/edoc_macros.erl b/lib/edoc/src/edoc_macros.erl
index bdcb3fe81f..e1a54d5090 100644
--- a/lib/edoc/src/edoc_macros.erl
+++ b/lib/edoc/src/edoc_macros.erl
@@ -311,7 +311,7 @@ macro_content([C | Cs], As, L, N) ->
 macro_content([], _As, _L, _N) ->
     throw('end').
 
--type line() :: erl_scan:line().
+-type line() :: erl_anno:line().
 -type err()  :: 'unterminated_macro'
 	      | 'macro_name'
 	      | {'macro_name', string()}
diff --git a/lib/edoc/src/edoc_parser.yrl b/lib/edoc/src/edoc_parser.yrl
index 48c01c8dce..835e7ccaa6 100644
--- a/lib/edoc/src/edoc_parser.yrl
+++ b/lib/edoc/src/edoc_parser.yrl
@@ -338,7 +338,7 @@ build_def(S, P, As, T) ->
                                   args = lists:reverse(As)},
                    type = T};
         false ->
-            return_error(element(2, P), "variable expected after '('")
+            return_error(tok_line(P), "variable expected after '('")
     end.
 
 all_vars([#t_var{} | As]) ->
@@ -452,7 +452,7 @@ parse_throws(S, L) ->
 
 %% ---------------------------------------------------------------------
 
--spec throw_error(term(), erl_scan:line()) -> no_return().
+-spec throw_error(term(), erl_anno:line()) -> no_return().
 
 throw_error({parse_spec, E}, L) ->
     throw_error({"specification", E}, L);
diff --git a/lib/edoc/src/edoc_specs.erl b/lib/edoc/src/edoc_specs.erl
index 3bf81c6503..59f6cb8ddf 100644
--- a/lib/edoc/src/edoc_specs.erl
+++ b/lib/edoc/src/edoc_specs.erl
@@ -1,7 +1,7 @@
 %%
 %% %CopyrightBegin%
 %%
-%% Copyright Ericsson AB 1996-2014. All Rights Reserved.
+%% Copyright Ericsson AB 1996-2015. All Rights Reserved.
 %%
 %% The contents of this file are subject to the Erlang Public License,
 %% Version 1.1, (the "License"); you may not use this file except in
@@ -58,7 +58,7 @@ type(Form, TypeDocs) ->
                     end,
                 {#t_name{name = N}, T, As, Doc0}
         end,
-    #tag{name = type, line = element(2, Type),
+    #tag{name = type, line = get_line(element(2, Type)),
          origin = code,
          data = {#t_typedef{name = TypeName,
                             args = d2e(Args),
@@ -71,7 +71,7 @@ type(Form, TypeDocs) ->
 spec(Form, Clause) ->
     {Name, _Arity, TypeSpecs} = get_spec(Form),
     TypeSpec = lists:nth(Clause, TypeSpecs),
-    #tag{name = spec, line = element(2, TypeSpec),
+    #tag{name = spec, line = get_line(element(2, TypeSpec)),
          origin = code,
          data = aspec(d2e(TypeSpec), Name)}.
 
@@ -83,7 +83,7 @@ dummy_spec(Form) ->
     {#t_name{name = Name}, Arity, TypeSpecs} = get_spec(Form),
     As = string:join(lists:duplicate(Arity, "_X"), ","),
     S = lists:flatten(io_lib:format("~p(~s) -> true\n", [Name, As])),
-    #tag{name = spec, line = element(2, hd(TypeSpecs)),
+    #tag{name = spec, line = get_line(element(2, hd(TypeSpecs))),
          origin = code, data = S}.
 
 -spec docs(Forms::[syntaxTree()],
@@ -140,7 +140,7 @@ find_type_docs([F | Fs], Cs, Fun) ->
             %% Postcomments before the dot after the typespec are ignored.
             C2 = [C1 | [C ||
                            C <- erl_syntax:get_postcomments(F),
-                           get_line(erl_syntax:get_pos(C)) >= LastTypeLine]],
+                           erl_syntax:get_pos(C) >= LastTypeLine]],
             C3 = collect_comments(Fs, LastTypeLine),
             #tag{data = Doc0} = Fun(lists:reverse(C2 ++ C3), LastTypeLine),
             case strip(Doc0) of % Strip away "f(). \n"
@@ -157,7 +157,7 @@ find_type_docs([F | Fs], Cs, Fun) ->
 collect_comments([], _Line) ->
     [];
 collect_comments([F | Fs], Line) ->
-    L1 = get_line(erl_syntax:get_pos(F)),
+    L1 = erl_syntax:get_pos(F),
     if
         L1 =:= Line + 1;
         L1 =:= Line -> % a separate postcomment
@@ -190,29 +190,26 @@ get_name_and_last_line(F) ->
     {Name, Data} = erl_syntax_lib:analyze_wild_attribute(F),
     type = edoc_specs:tag(Name),
     Attr = {attribute, erl_syntax:get_pos(F), Name, Data},
-    Ref = make_ref(),
-    Fun = fun(L) -> {Ref, get_line(L)} end,
+    Fun = fun(A) ->
+                  Line = get_line(A),
+                  case get('$max_line') of
+                      Max when Max < Line ->
+                          _ = put('$max_line', Line);
+                      _ ->
+                          ok
+                  end
+          end,
+    undefined = put('$max_line', 0),
+    _ = erl_parse:map_anno(Fun, Attr),
+    Line = erase('$max_line'),
     TypeName = case Data of
                    {N, _T, As} when is_atom(N) -> % skip records
                        {N, length(As)}
                end,
-    Line = gll(erl_lint:modify_line(Attr, Fun), Ref),
     {TypeName, Line}.
 
-gll({Ref, Line}, Ref) ->
-    Line;
-gll([], _Ref) ->
-    0;
-gll(List, Ref) when is_list(List) ->
-    lists:max([gll(E, Ref) || E <- List]);
-gll(Tuple, Ref) when is_tuple(Tuple) ->
-    gll(tuple_to_list(Tuple), Ref);
-gll(_, _) ->
-    0.
-
-get_line(Pos) ->
-    {line, Line} = erl_scan:attributes_info(Pos, line),
-    Line.
+get_line(Anno) ->
+    erl_anno:line(Anno).
 
 %% Collect all Erlang types. Types in comments (@type) shadow Erlang
 %% types (-spec/-opaque).
@@ -348,7 +345,7 @@ d2e({type,_,constraint,[Sub,Ts0]}) ->
             Ts = [ST,T] = d2e([ST0,T0]),
             #t_def{name = ST, type = typevar_anno(T, Ts)};
         _ ->
-            throw_error(element(2, Sub), "cannot handle guard", [])
+            throw_error(get_line(element(2, Sub)), "cannot handle guard", [])
     end;
 d2e({type,_,union,Ts0}) ->
     Ts = d2e(Ts0),
diff --git a/lib/edoc/src/edoc_tags.erl b/lib/edoc/src/edoc_tags.erl
index c1c453511a..9e2e41e902 100644
--- a/lib/edoc/src/edoc_tags.erl
+++ b/lib/edoc/src/edoc_tags.erl
@@ -342,7 +342,7 @@ parse_typedef(Data, Line, _Env, Where) ->
 	    Def
     end.
 
--type line() :: erl_scan:line().
+-type line() :: erl_anno:line().
 
 -spec parse_file(_, line(), _, _) -> no_return().