2008-07-22 Marek Safar <marek.safar@gmail.com>

[mono.git] / mcs / mcs / lambda.cs

diff --git a/mcs/mcs/lambda.cs b/mcs/mcs/lambda.cs
index 06eb6ea9b767d218cec4b4f2f8436dfcc00fb15b..b11dc6000e254d2c8c6beec1f9ce9445c9891e56 100644 (file)
--- a/mcs/mcs/lambda.cs
+++ b/mcs/mcs/lambda.cs
@@ -134,8 +134,7 @@ namespace Mono.CSharp {
                public LambdaMethod (TypeContainer host, Parameters parameters,
                                        ToplevelBlock block, Type return_type, Type delegate_type,
                                        Location loc)
-                       : base (host, parameters, block,
-                               return_type, delegate_type, loc)
+                       : base (host, parameters, block, return_type, delegate_type, loc)
                {
                }
 
@@ -147,7 +146,22 @@ namespace Mono.CSharp {
 
                public override Expression CreateExpressionTree (EmitContext ec)
                {
-                       return Block.CreateExpressionTree (ec);
+                       //
+                       // Remove IL method implementation when expression tree is requested
+                       //
+                       method.Parent.PartialContainer.RemoveMethod (method);
+
+                       Expression args = parameters.CreateExpressionTree (ec, loc);
+                       Expression expr = Block.CreateExpressionTree (ec);
+                       if (expr == null)
+                               return null;
+
+                       ArrayList arguments = new ArrayList (2);
+                       arguments.Add (new Argument (expr));
+                       arguments.Add (new Argument (args));
+                       return CreateExpressionFactoryCall ("Lambda",
+                               new TypeArguments (loc, new TypeExpression (type, loc)),
+                               arguments);
                }
        }
 
@@ -170,10 +184,13 @@ namespace Mono.CSharp {
 
                public override void Emit (EmitContext ec)
                {
-                       if (ec.ReturnType == TypeManager.void_type)
+                       if (ec.ReturnType == TypeManager.void_type) {
                                ((ExpressionStatement) Expr).EmitStatement (ec);
-                       else
-                               base.Emit (ec);
+                               ec.ig.Emit (OpCodes.Ret);
+                               return;
+                       }
+
+                       base.Emit (ec);
                }
 
                protected override bool DoResolve (EmitContext ec)
@@ -186,11 +203,8 @@ namespace Mono.CSharp {
                                if (Expr == null)
                                        return false;
                                
-                               if (Expr is ExpressionStatement) {
-                                       if (Expr is Invocation)
-                                               return TypeManager.IsEqual (Expr.Type, TypeManager.void_type);
+                               if (Expr is ExpressionStatement)
                                        return true;
-                               }
                                
                                Expr.Error_InvalidExpressionStatement ();
                                return false;