如何在二进制表达式中将参数表达式更改为常量表达式

时间:2018-03-28 08:56:20

标签: c# lambda expression

我的自定义访问者向右和向左看,并将参数更改为常量。
我知道只是改变节点是不可能的。
我应该返回包含常量而不是参数的新lambda表达式。 但我不能自己创建一个表达式:(
我有这段代码:

    public class ParametersTransformToConstantVisitor : ExpressionVisitor
    {

    private Dictionary<string, ConstantExpression> parameters = new Dictionary<string, ConstantExpression>();

    public ParametersTransformToConstantVisitor(Dictionary<string, ConstantExpression> parameters)
    {
        this.parameters = parameters;
    }

    protected override Expression VisitBinary(BinaryExpression node)
    {
        var constExprLeftName = new Lazy<string>(() => ((ParameterExpression) node.Left)?.Name);
        var constExprRightName = new Lazy<string>(() => ((ParameterExpression) node.Right)?.Name);
        var constExprName = new Lazy<string>(() => ((ParameterExpression) node.Reduce())?.Name);

        ParameterExpression leftParam = null;
        ParameterExpression rightParam = null;

        if (node.NodeType == ExpressionType.Parameter && parameters.ContainsKey(constExprName.Value))
        {
            return parameters[constExprName.Value];
        }

        if (node.Left.NodeType == ExpressionType.Parameter && parameters.ContainsKey(constExprLeftName.Value))
        {
            leftParam = (ParameterExpression) node.Left;
        }       

        if (node.Right.NodeType == ExpressionType.Parameter && parameters.ContainsKey(constExprLeftName.Value))
        {
            rightParam = (ParameterExpression) node.Right;
        }

        if (leftParam != null || rightParam != null)
        {
            //return Expression.Lambda();
        }       

        return base.VisitBinary(node);
    }
}

请帮我构建lambda表达式,

1 个答案:

答案 0 :(得分:3)

感觉就像你真正需要的一样:

Z = S n

即。每当访问者看到protected override Expression VisitParameter(ParameterExpression node) => parameters.TryGetValue(node.Name, out var ce) ? (Expression)ce : node; protected override Expression VisitLambda<T>(Expression<T> node) => Expression.Lambda(Visit(node.Body), node.Parameters); // don't visit the parameters if ParameterExpression地图中有相应的项目时,请使用该值。

parameters上的覆盖是因为VisitLambda仍然需要返回相同形状的 ,而默认实现也会访问(并因此交换)声明中的参数

访问者的工作是担心围绕更改重新组装树。

但请注意,如果您尝试创建无参数lambda ,则可能还需要重写root。或者您可以使用VisitLambda并忘记参数。

示例:

.Body

给出:

Expression<Func<int, int, string>> add = (x, y) => ((2 * x) + y).ToString();
Console.WriteLine(add);

var args = new Dictionary<string, ConstantExpression>
{
    ["x"] = Expression.Constant(4),
    ["y"] = Expression.Constant(1),
};

var visitor = new ParametersTransformToConstantVisitor(args);
var result = (LambdaExpression)visitor.Visit(add);
Console.WriteLine(result);

您可以通过以下方式将其变为无参数的lambda:

(x, y) => ((2 * x) + y).ToString()
(x, y) => ((2 * 4) + 1).ToString()

给出:

var withoutArgs = Expression.Lambda<Func<string>>(result.Body);
Console.WriteLine(withoutArgs);

次要补充:您可能还希望简化访问者:

() => ((2 * 4) + 1).ToString()

这会将输出更改为:

protected override Expression VisitBinary(BinaryExpression node)
{
    var visited = base.VisitBinary(node);

    if(visited is BinaryExpression be
        && be.Method == null && be.Conversion == null
        && !be.IsLifted
        && be.Left is ConstantExpression left
        && be.Right is ConstantExpression right)
    {

        object val;
        switch(be.NodeType)
        {
            case ExpressionType.Add:
                val = (dynamic)left.Value + (dynamic)right.Value;
                break;
            case ExpressionType.Multiply:
                val = (dynamic)left.Value * (dynamic)right.Value;
                break;
            case ExpressionType.Subtract:
                val = (dynamic)left.Value - (dynamic)right.Value;
                break;
            case ExpressionType.Divide:
                val = (dynamic)left.Value / (dynamic)right.Value;
                break;
            default:
                return visited; // unknown
        }

        return Expression.Constant(
            Convert.ChangeType(val, visited.Type), visited.Type);
    }
    return visited;
}

我们可能甚至还可以提升(x, y) => ((2 * x) + y).ToString() (x, y) => 9.ToString() () => 9.ToString()

ToString()

现在给了我们:

protected override Expression VisitMethodCall(MethodCallExpression node)
{
    var visited = base.VisitMethodCall(node);
    if (visited is MethodCallExpression mce)
    {
        if ((mce.Object == null || mce.Object is ConstantExpression)
            && mce.Arguments.All(x => x is ConstantExpression))
        {
            var obj = (mce.Object as ConstantExpression)?.Value;
            var args = mce.Arguments.Select(x => ((ConstantExpression)x).Value).ToArray();
            var result = mce.Method.Invoke(obj, args);
            return Expression.Constant(result, mce.Type);
        }
    }
    return visited;
}