如何将Expr(表达式)数组传递给Haxe Macro?

时间:2016-04-01 17:42:44

标签: arrays syntax macros expression haxe

我基本上试图创建一个宏,通过为所有条件提供一个公共结果语句来自动生成[SetUp] public void ExampleSetup() { //whatever } [Test] public void ExampleTest() { //whatever } [TearDown] public void ExampleTearDown() { //do stuff someObject.ReportTestResult(); } 链。

这是我到目前为止所尝试的内容(修改代码仅作为示例):

if/else if

然后,理论上它可以这样使用:

import haxe.macro.Expr;

class LazyUtils {

    public macro static function tryUntilFalse( xBool:Expr, xConds:Array<Expr> ) {
        var con1, con2, con3, con4, con5;

        /*
         * Here's a switch to handle specific # of conditions, because for-loops
         * don't seem to be allowed here (at least in the ways I've tried so far).
         * 
         * If you know how to use for-loop for this, PLEASE do tell!
         */
        switch(xConds.length) {
            case 1: {
                con1 = conds[0];
                return macro {
                    if (!$con1) $xBool;
                }
            }
            case 2: {
                con1 = conds[0];
                con2 = conds[1];
                return macro {
                    if (!$con1) $xBool;
                    else if (!$con2) $xBool;
                }
            }
            case 3: {
                con1 = conds[0];
                con2 = conds[1];
                con3 = conds[2];
                return macro {
                    if (!$con1) $xBool;
                    else if (!$con2) $xBool;
                    else if (!$con3) $xBool;
                }
            }
            // ... so on and so forth
        }

        return macro { trace("Unhandled length of conditions :("); };
    }
}

哪个应该生成这个条件树:

class Main {
    static function main() {
        var isOK = true;
        LazyUtils.tryUntilFalse( isOK = false, [
            doSomething(),
            doSomethingElse(), //Returns false, so should stop here.
            doFinalThing()
        ]);
    }

    static function doSomething():Bool {
        // ???
        return true;
    }

    static function doSomethingElse():Bool {
        // ???
        return false;
    }

    static function doFinalThing():Bool {
        return true;
    }
}

或者,我想它可以输出这个:

if (!doSomething()) isOK = false;
else if (!doSomethingElse()) isOK = false;
else if (!doFinalThing()) isOK = false;

现在回过头来看,这是真的 - 编写一个完整的宏来生成可以更容易输入原始格式的代码可能没什么意义。

但是为了学习宏,有没有人知道是否可以像我在上面的代码示例中尝试的那样在if(!doSomething() || !doSomethingElse() || !doFinalThing()) isOK = false; 中传递多个表达式?

1 个答案:

答案 0 :(得分:4)

您可能无法使xConds参数表现得像您预期的那样,因为类型为Array<Expr>的表达式宏的最终参数隐式为rest argument。这意味着您最终得到了一个包含单个EArrayDecl表达式的数组。只需省略[]即可解决此问题。

关于生成if - else - 链 - 让我们来看看EIf

/**
    An `if(econd) eif` or `if(econd) eif else eelse` expression.
**/
EIf( econd : Expr, eif : Expr, eelse : Null<Expr> );

链可以被视为单链表 - eelse如果第一个EIf应该引用下一个EIf,依此类推,直到我们停止{{1} }为最后eelse = null。所以我们想为你的例子生成这个(伪代码):

EIf

递归效果很好。

通常使用reification比使用像我这样的原始表达式更方便,但是我不确定在动态生成这样的表达式时前者是否真的可行。

EIf(doSomething(), isOk = false, /* else */
    EIf(doSomethingElse, isOk = false, /* else */
        EIf(doFinalThing(), isOk = false, null)
    )
)

import haxe.macro.Context; import haxe.macro.Expr; class LazyUtils { public macro static function tryUntilFalse(setBool:Expr, conditions:Array<Expr>):Expr { return generateIfChain(setBool, conditions); } private static function generateIfChain(eif:Expr, conditions:Array<Expr>):Expr { // get the next condition var condition = conditions.shift(); if (condition == null) { return null; // no more conditions } // recurse deeper to generate the next if var nextIf = generateIfChain(eif, conditions); return { expr: EIf(condition, eif, nextIf), pos: Context.currentPos() }; } } (大部分未改变):

Main.hx

为了简单起见,我在调用站点用class Main { static function main() { var isOK = true; LazyUtils.tryUntilFalse(isOK = false, !doSomething(), !doSomethingElse(), //Returns false, so should stop here. !doFinalThing() ); } static function doSomething():Bool { trace("doSomething"); return true; } static function doSomethingElse():Bool { trace("doSomethingElse"); return false; } static function doFinalThing():Bool { trace("doFinalThing"); return true; } } 反转了函数调用参数,而不是在宏中处理它。

您可以使用!生成AST转储并检查正在生成的代码。结果如下:

-D dump=pretty