Examples with ForExpressionCode jetbrick.template.parser.code.ForExpressionCode used on opensource projects

Example 1 with ForExpressionCode

use of jetbrick.template.parser.code.ForExpressionCode in project jetbrick-template-1x by subchen.

the class JetTemplateCodeVisitor method visitFor_directive.

@Override
public Code visitFor_directive(For_directiveContext ctx) {
    BlockCode code = scopeCode.createBlockCode(16);
    String id_for = getUid("for");
    scopeCode = scopeCode.push();
    // 注意：for循环变量的作用域要放在 for 内部， 防止出现变量重定义错误
    ForExpressionCode for_expr_code = (ForExpressionCode) ctx.for_expression().accept(this);
    // for block
    forStack.push(id_for);
    Code for_block_code = ctx.block().accept(this);
    forStack.pop();
    scopeCode = scopeCode.pop();
    // for-else
    Else_directiveContext else_directive = ctx.else_directive();
    Code for_else_block = (else_directive == null) ? null : else_directive.accept(this);
    // 生成代码
    String id_foritem = getUid("foritem");
    String typeName = for_expr_code.getKlassName();
    String itemName = for_expr_code.getName();
    code.addLine("Object " + id_foritem + " = context.get(\"" + itemName + "\"); // save it");
    code.addLine("JetForIterator " + id_for + " = new JetForIterator(" + for_expr_code.toString() + ");");
    code.addLine("while (" + id_for + ".hasNext()) { // line: " + ctx.getStart().getLine());
    // class item = (class) it.next() ...
    code.addLine("  " + typeName + " " + itemName + " = (" + typeName + ") " + id_for + ".next();");
    code.addLine("  context.put(\"" + itemName + "\", " + itemName + ");");
    code.addChild(for_block_code);
    code.addLine("}");
    code.addLine("context.put(\"" + itemName + "\", " + id_foritem + "); // reset it");
    // for else ...
    if (for_else_block != null) {
        code.addLine("if (" + id_for + ".empty()) { // line: " + ctx.getStart().getLine());
        code.addChild(for_else_block);
        code.addLine("}");
    }
    return code;
}

Example 2 with ForExpressionCode

use of jetbrick.template.parser.code.ForExpressionCode in project jetbrick-template-1x by subchen.

the class JetTemplateCodeVisitor method visitFor_expression.

@Override
public Code visitFor_expression(For_expressionContext ctx) {
    String name = ctx.IDENTIFIER().getText();
    SegmentCode code = (SegmentCode) ctx.expression().accept(this);
    assert_not_void_expression(code);
    TypedKlass resultKlass = null;
    TypeContext type = ctx.type();
    if (type != null) {
        // 手动定义返回变量的类型
        SegmentCode c = (SegmentCode) type.accept(this);
        resultKlass = c.getTypedKlass();
    } else {
        // 根据 expression 来进行自动类型推导
        Class<?> rhsKlass = code.getKlass();
        if (rhsKlass.isArray()) {
            resultKlass = TypedKlass.create(rhsKlass.getComponentType(), code.getTypeArgs());
        } else if (Map.class.isAssignableFrom(rhsKlass)) {
            resultKlass = TypedKlass.create(Map.Entry.class, code.getTypeArgs());
        } else if (Collection.class.isAssignableFrom(rhsKlass)) {
            if (code.getTypeArgs() != null && code.getTypeArgs().length == 1) {
                resultKlass = code.getTypeArgs()[0];
            }
        }
    }
    if (resultKlass == null) {
        resultKlass = TypedKlass.Object;
    }
    // 必须是 Boxed 对象，因为需要强制类型转换 from iterator.next()
    resultKlass = resultKlass.asBoxedTypedKlass();
    if (!scopeCode.define(name, resultKlass)) {
        throw reportError("Duplicate local variable " + name, ctx.IDENTIFIER());
    }
    return new ForExpressionCode(resultKlass, name, code.toString(), ctx);
}