Examples with BLangForeach org.wso2.ballerinalang.compiler.tree.statements.BLangForeach used on opensource projects

Example 6 with BLangForeach

use of org.wso2.ballerinalang.compiler.tree.statements.BLangForeach in project ballerina by ballerina-lang.

the class CodeGenerator method visit.

public void visit(BLangForeach foreach) {
    // Calculate temporary scope variables for iteration.
    Operand iteratorVar = getLVIndex(TypeTags.ITERATOR);
    Operand conditionVar = getLVIndex(TypeTags.BOOLEAN);
    // Create new Iterator for given collection.
    this.genNode(foreach.collection, env);
    this.emit(InstructionCodes.ITR_NEW, foreach.collection.regIndex, iteratorVar);
    Operand foreachStartAddress = new Operand(nextIP());
    Operand foreachEndAddress = new Operand(-1);
    Instruction gotoStartInstruction = InstructionFactory.get(InstructionCodes.GOTO, foreachStartAddress);
    Instruction gotoEndInstruction = InstructionFactory.get(InstructionCodes.GOTO, foreachEndAddress);
    // Checks given iterator has a next value.
    this.emit(InstructionCodes.ITR_HAS_NEXT, iteratorVar, conditionVar);
    this.emit(InstructionCodes.BR_FALSE, conditionVar, foreachEndAddress);
    // assign variables.
    generateForeachVarAssignment(foreach, iteratorVar);
    this.loopResetInstructionStack.push(gotoStartInstruction);
    this.loopExitInstructionStack.push(gotoEndInstruction);
    // generate foreach body.
    this.genNode(foreach.body, env);
    this.loopResetInstructionStack.pop();
    this.loopExitInstructionStack.pop();
    // move to next iteration.
    this.emit(gotoStartInstruction);
    foreachEndAddress.value = this.nextIP();
}

Example 7 with BLangForeach

use of org.wso2.ballerinalang.compiler.tree.statements.BLangForeach in project ballerina by ballerina-lang.

the class CodeGenerator method generateForeachVarAssignment.

// private helper methods of visitors.
private void generateForeachVarAssignment(BLangForeach foreach, Operand iteratorIndex) {
    List<BLangVariableReference> variables = foreach.varRefs.stream().map(expr -> (BLangVariableReference) expr).collect(Collectors.toList());
    // create Local variable Info entries.
    variables.stream().filter(// Ignoring ignored ("_") variables.
    v -> v.type.tag != TypeTags.NONE).forEach(varRef -> visitVarSymbol(varRef.symbol, lvIndexes, localVarAttrInfo));
    List<Operand> nextOperands = new ArrayList<>();
    nextOperands.add(iteratorIndex);
    nextOperands.add(new Operand(variables.size()));
    foreach.varTypes.forEach(v -> nextOperands.add(new Operand(v.tag)));
    nextOperands.add(new Operand(variables.size()));
    for (int i = 0; i < variables.size(); i++) {
        BLangVariableReference varRef = variables.get(i);
        nextOperands.add(Optional.ofNullable(varRef.symbol.varIndex).orElse(getRegIndex(foreach.varTypes.get(i).tag)));
    }
    this.emit(InstructionCodes.ITR_NEXT, nextOperands.toArray(new Operand[0]));
}

Example 8 with BLangForeach

use of org.wso2.ballerinalang.compiler.tree.statements.BLangForeach in project ballerina by ballerina-lang.

the class ASTBuilderUtil method createForeach.

static BLangForeach createForeach(DiagnosticPos pos, BLangBlockStmt target, BLangSimpleVarRef collectionVarRef, List<BLangSimpleVarRef> varRefs, List<BType> inputTypes) {
    final BLangForeach foreach = (BLangForeach) TreeBuilder.createForeachNode();
    foreach.pos = pos;
    target.addStatement(foreach);
    foreach.body = ASTBuilderUtil.createBlockStmt(pos);
    foreach.collection = collectionVarRef;
    foreach.varRefs.addAll(varRefs);
    foreach.varTypes = inputTypes;
    return foreach;
}

Example 9 with BLangForeach

use of org.wso2.ballerinalang.compiler.tree.statements.BLangForeach in project ballerina by ballerina-lang.

the class IterableCodeDesugar method generateStreamingIteratorBlock.

private void generateStreamingIteratorBlock(IterableContext ctx, BLangFunction funcNode, LinkedList<Operation> streamOperations) {
    final Operation firstOperation = ctx.getFirstOperation();
    final DiagnosticPos pos = firstOperation.pos;
    // Generate streaming based function Body.
    if (isReturningIteratorFunction(ctx)) {
        if (ctx.resultType.tag != TypeTags.TABLE) {
            // This should be the select operation. No need of a count variable.
            createCounterVarDefStmt(funcNode, ctx);
        }
        createResultVarDefStmt(funcNode, ctx);
        ctx.iteratorResultVariables = createIteratorResultVariables(ctx, streamOperations.getLast().retVar, funcNode);
    } else {
        ctx.iteratorResultVariables = Collections.emptyList();
    }
    // Create variables required.
    final List<BLangVariable> foreachVariables = createForeachVariables(ctx, ctx.getFirstOperation().argVar, funcNode);
    // Define all undefined variables.
    defineRequiredVariables(ctx, streamOperations, foreachVariables, funcNode);
    // Generate foreach iteration.
    final BLangForeach foreachStmt = ASTBuilderUtil.createForeach(pos, funcNode.body, ASTBuilderUtil.createVariableRef(pos, ctx.collectionVar.symbol), ASTBuilderUtil.createVariableRefList(pos, foreachVariables), ctx.foreachTypes);
    if (foreachVariables.size() > 1) {
        // Create tuple, for lambda invocation.
        final BLangAssignment assignmentStmt = ASTBuilderUtil.createAssignmentStmt(pos, foreachStmt.body);
        assignmentStmt.declaredWithVar = true;
        assignmentStmt.varRefs.add(ASTBuilderUtil.createVariableRef(pos, ctx.getFirstOperation().argVar.symbol));
        final BLangBracedOrTupleExpr tupleExpr = (BLangBracedOrTupleExpr) TreeBuilder.createBracedOrTupleExpression();
        for (BLangVariable foreachVariable : foreachVariables) {
            tupleExpr.expressions.add(ASTBuilderUtil.createVariableRef(pos, foreachVariable.symbol));
        }
        tupleExpr.isBracedExpr = foreachVariables.size() == 1;
        tupleExpr.type = new BTupleType(getTupleTypeList(ctx.getFirstOperation().inputType));
        assignmentStmt.expr = tupleExpr;
    }
    // Generate Operations related
    ctx.operations.forEach(operation -> generateOperationCode(foreachStmt.body, operation));
    // Generate aggregator and result
    if (isReturningIteratorFunction(ctx)) {
        if (ctx.iteratorResultVariables.size() > 1) {
            // Destructure return Values.
            final BLangTupleDestructure tupleAssign = (BLangTupleDestructure) TreeBuilder.createTupleDestructureStatementNode();
            tupleAssign.pos = pos;
            tupleAssign.declaredWithVar = true;
            foreachStmt.body.addStatement(tupleAssign);
            tupleAssign.expr = ASTBuilderUtil.createVariableRef(pos, ctx.getLastOperation().retVar.symbol);
            tupleAssign.varRefs.addAll(ASTBuilderUtil.createVariableRefList(pos, ctx.iteratorResultVariables));
        }
        generateAggregator(foreachStmt.body, ctx);
        generateFinalResult(funcNode.body, ctx);
    }
    final BLangReturn returnStmt = ASTBuilderUtil.createReturnStmt(firstOperation.pos, funcNode.body);
    if (isReturningIteratorFunction(ctx)) {
        returnStmt.addExpression(ASTBuilderUtil.createVariableRef(pos, ctx.resultVar.symbol));
    }
}

Example 10 with BLangForeach

use of org.wso2.ballerinalang.compiler.tree.statements.BLangForeach in project ballerina by ballerina-lang.

the class BLangPackageBuilder method addForeachStatement.

public void addForeachStatement(DiagnosticPos pos, Set<Whitespace> ws) {
    BLangForeach foreach = (BLangForeach) TreeBuilder.createForeachNode();
    foreach.addWS(ws);
    foreach.pos = pos;
    foreach.setCollection(exprNodeStack.pop());
    foreach.addWS(commaWsStack.pop());
    List<ExpressionNode> lExprList = exprNodeListStack.pop();
    lExprList.forEach(expressionNode -> foreach.addVariable((BLangVariableReference) expressionNode));
    BLangBlockStmt foreachBlock = (BLangBlockStmt) this.blockNodeStack.pop();
    foreachBlock.pos = pos;
    foreach.setBody(foreachBlock);
    addStmtToCurrentBlock(foreach);
}