1 files changed, 22 insertions, 53 deletions

diff --git a/lib/compiler/src/v3_core.erl b/lib/compiler/src/v3_core.erl
index 1d5eb85a1c..a9b54005d1 100644
--- a/lib/compiler/src/v3_core.erl
+++ b/lib/compiler/src/v3_core.erl
@@ -536,7 +536,7 @@ expr({tuple,L,Es0}, St0) ->
     A = record_anno(L, St1),
     {annotate_tuple(A, Es1, St1),Eps,St1};
 expr({map,L,Es0}, St0) ->
-    map_build_pair_chain(#c_literal{val=#{}},Es0,lineno_anno(L,St0),St0);
+    map_build_pairs(#c_literal{val=#{}}, Es0, lineno_anno(L, St0), St0);
 expr({map,L,M0,Es0}, St0) ->
     try expr_map(M0,Es0,lineno_anno(L, St0),St0) of
 	{_,_,_}=Res -> Res
@@ -779,66 +779,35 @@ expr_map(M0,Es0,A,St0) ->
 		    Fc = fail_clause([Fpat], A, #c_literal{val=badarg}),
 		    {#icase{anno=#a{anno=A},args=[M1],clauses=Cs,fc=Fc},Mps,St3};
 		{_,_} ->
-		    {M2,Eps,St2} = map_build_pair_chain(M1,Es0,A,St1),
+		    {M2,Eps,St2} = map_build_pairs(M1, Es0, A, St1),
 		    {M2,Mps++Eps,St2}
 	    end;
 	false -> throw({bad_map,bad_map})
     end.
 
-%% Group continuous literal blocks and single variables, i.e.
-%% M0#{ a := 1, b := V1, K1 := V2, K2 := 42}
-%% becomes equivalent to
-%% M1 = M0#{ a := 1, b := V1 },
-%% M2 = M1#{ K1 := V1 },
-%% M3 = M2#{ K2 := 42 }
-
-map_build_pair_chain(M,Es,A,St) ->
-    %% hack, remove iset if only literal
-    case map_build_pair_chain(M,Es,A,St,[]) of
-	{_,[#iset{arg=#c_literal{}=Val}],St1} -> {Val,[],St1};
-	Normal -> Normal
+map_build_pairs(Map0, Es0, Ann, St0) ->
+    {Es,Pre,St1} = map_build_pairs_1(Es0, St0),
+    case ann_c_map(Ann, Map0, Es) of
+	#c_literal{}=Map ->
+	    {Map,[],St1};
+	#c_map{}=Map ->
+	    {Var,St2} = new_var(St1),
+	    {Var,Pre++[#iset{var=Var,arg=Map}],St2}
     end.
 
-map_build_pair_chain(M0,[],_,St,Mps) ->
-    {M0,Mps,St};
-map_build_pair_chain(M0,Es0,A,St0,Mps) ->
-    % group continuous literal blocks
-    % Anno = #a{anno=[compiler_generated]},
-    % order is important, we need to reverse the literals
-    case map_pair_block(Es0,[],[],St0) of
-	{{CesL,EspL},{[],[]},Es1,St1} ->
-	    {MVar,St2} = new_var(St1),
-	    Pre = [#iset{var=MVar, arg=ann_c_map(A,M0,reverse(CesL))}],
-	    map_build_pair_chain(MVar,Es1,A,St2,Mps++EspL++Pre);
-	{{[],[]},{CesV,EspV},Es1,St1} ->
-	    {MVar,St2} = new_var(St1),
-	    Pre = [#iset{var=MVar, arg=#c_map{arg=M0,es=CesV, anno=A}}],
-	    map_build_pair_chain(MVar,Es1,A,St2,Mps ++ EspV++Pre);
-	{{CesL,EspL},{CesV,EspV},Es1,St1} ->
-	    {MVarL,St2} = new_var(St1),
-	    {MVarV,St3} = new_var(St2),
-	    Pre = [#iset{var=MVarL, arg=ann_c_map(A,M0,reverse(CesL))},
-		   #iset{var=MVarV, arg=#c_map{arg=MVarL,es=CesV,anno=A}}],
-	    map_build_pair_chain(MVarV,Es1,A,St3,Mps++EspL++EspV++Pre)
-    end.
-
-map_pair_block([{Op,L,K0,V0}|Es],Ces,Esp,St0) ->
-    {K,Ep0,St1} = safe(K0, St0),
-    {V,Ep1,St2} = safe(V0, St1),
-    A = lineno_anno(L, St2),
-    Pair0 = map_op_to_c_map_pair(Op),
-    Pair1 = Pair0#c_map_pair{anno=A,key=K,val=V},
-    case cerl:is_literal(K) of
-	true ->
-	    map_pair_block(Es,[Pair1|Ces],Ep0 ++ Ep1 ++ Esp,St2);
-	false ->
-	    {{Ces,Esp},{[Pair1],Ep0++Ep1},Es,St2}
-    end;
-map_pair_block([],Ces,Esp,St) ->
-    {{Ces,Esp},{[],[]},[],St}.
+map_build_pairs_1([{Op0,L,K0,V0}|Es], St0) ->
+    {K,Pre0,St1} = safe(K0, St0),
+    {V,Pre1,St2} = safe(V0, St1),
+    {Pairs,Pre2,St3} = map_build_pairs_1(Es, St2),
+    As = lineno_anno(L, St3),
+    Op = map_op(Op0),
+    Pair = cerl:ann_c_map_pair(As, Op, K, V),
+    {[Pair|Pairs],Pre0++Pre1++Pre2,St3};
+map_build_pairs_1([], St) ->
+    {[],[],St}.
 
-map_op_to_c_map_pair(map_field_assoc) -> #c_map_pair{op=#c_literal{val=assoc}};
-map_op_to_c_map_pair(map_field_exact) -> #c_map_pair{op=#c_literal{val=exact}}.
+map_op(map_field_assoc) -> #c_literal{val=assoc};
+map_op(map_field_exact) -> #c_literal{val=exact}.
 
 is_valid_map_src(#c_literal{val = M}) when is_map(M) -> true;
 is_valid_map_src(#c_var{})  -> true;