Examples with RegIndex org.wso2.ballerinalang.programfile.Instruction.RegIndex used on opensource projects

Example 16 with RegIndex

use of org.wso2.ballerinalang.programfile.Instruction.RegIndex in project ballerina by ballerina-lang.

the class CodeGenerator method processJoinBlock.

/* generate code for Join block */
private void processJoinBlock(BLangForkJoin forkJoin, ForkjoinInfo forkjoinInfo, SymbolEnv forkJoinEnv, RegIndex joinVarRegIndex, Operand joinBlockAddr) {
    UTF8CPEntry joinType = new UTF8CPEntry(forkJoin.joinType.name());
    int joinTypeCPIndex = this.currentPkgInfo.addCPEntry(joinType);
    forkjoinInfo.setJoinType(forkJoin.joinType.name());
    forkjoinInfo.setJoinTypeCPIndex(joinTypeCPIndex);
    joinBlockAddr.value = nextIP();
    if (forkJoin.joinResultVar != null) {
        visitForkJoinParameterDefs(forkJoin.joinResultVar, forkJoinEnv);
        joinVarRegIndex.value = forkJoin.joinResultVar.symbol.varIndex.value;
    }
    if (forkJoin.joinedBody != null) {
        this.genNode(forkJoin.joinedBody, forkJoinEnv);
    }
}

Example 17 with RegIndex

use of org.wso2.ballerinalang.programfile.Instruction.RegIndex in project ballerina by ballerina-lang.

the class CodeGenerator method visit.

public void visit(BLangAssignment assignNode) {
    List<BLangExpression> lhrExprs = assignNode.varRefs;
    if (assignNode.declaredWithVar) {
        lhrExprs.stream().filter(lhsExr -> lhsExr.type.tag != TypeTags.NONE).map(lhsExr -> (BLangVariableReference) lhsExr).forEach(varRef -> {
            visitVarSymbol(varRef.symbol, lvIndexes, localVarAttrInfo);
        });
    }
    // Calculate the register indexes of lhs expressions.
    int nLHSExpr = lhrExprs.size();
    RegIndex[] regIndexes = new RegIndex[nLHSExpr];
    for (int i = 0; i < nLHSExpr; i++) {
        BLangExpression lExpr = lhrExprs.get(i);
        if (lExpr.type.tag == TypeTags.NONE) {
            continue;
        }
        if (lExpr.getKind() == NodeKind.SIMPLE_VARIABLE_REF && lExpr instanceof BLangLocalVarRef) {
            lExpr.regIndex = ((BLangVariableReference) lExpr).symbol.varIndex;
            regIndexes[i] = lExpr.regIndex;
        }
    }
    // Set calculated reg indexes and visit rhs expression
    BLangExpression rhsExpr = assignNode.expr;
    rhsExpr.regIndex = regIndexes[0];
    genNode(rhsExpr, this.env);
    // Set the reg indexes generated by visiting rhs expression to lhs expression
    regIndexes[0] = rhsExpr.regIndex;
    for (int i = 0; i < lhrExprs.size(); i++) {
        BLangExpression lExpr = lhrExprs.get(i);
        if (lExpr.type.tag == TypeTags.NONE) {
            continue;
        }
        if (lExpr.getKind() == NodeKind.SIMPLE_VARIABLE_REF && lExpr instanceof BLangLocalVarRef) {
            continue;
        }
        varAssignment = true;
        lExpr.regIndex = regIndexes[i];
        genNode(lExpr, this.env);
        varAssignment = false;
    }
}

Example 18 with RegIndex

use of org.wso2.ballerinalang.programfile.Instruction.RegIndex in project ballerina by ballerina-lang.

the class CodeGenerator method createEnumInfoEntry.

private void createEnumInfoEntry(BLangEnum enumNode) {
    BTypeSymbol enumSymbol = (BTypeSymbol) enumNode.symbol;
    // Add Enum name as an UTFCPEntry to the constant pool
    int enumNameCPIndex = addUTF8CPEntry(currentPkgInfo, enumSymbol.name.value);
    EnumInfo enumInfo = new EnumInfo(currentPackageRefCPIndex, enumNameCPIndex, enumSymbol.flags);
    currentPkgInfo.addEnumInfo(enumSymbol.name.value, enumInfo);
    enumInfo.enumType = (BEnumType) enumSymbol.type;
    for (int i = 0; i < enumNode.enumerators.size(); i++) {
        BLangEnumerator enumeratorNode = enumNode.enumerators.get(i);
        enumeratorNode.symbol.varIndex = new RegIndex(i, enumSymbol.type.tag);
        enumeratorNode.symbol.varIndex.isVarIndex = true;
        int enumeratorNameCPIndex = addUTF8CPEntry(currentPkgInfo, enumeratorNode.symbol.name.toString());
        EnumeratorInfo enumeratorInfo = new EnumeratorInfo(enumeratorNameCPIndex, i, enumInfo.enumType);
        enumInfo.enumeratorInfoList.add(enumeratorInfo);
    }
}

Example 19 with RegIndex

use of org.wso2.ballerinalang.programfile.Instruction.RegIndex in project ballerina by ballerina-lang.

the class CodeGenerator method getFuncOperands.

private Operand[] getFuncOperands(BLangInvocation iExpr, int funcRefCPIndex) {
    // call funcRefCPIndex, nArgRegs, argRegs[nArgRegs], nRetRegs, retRegs[nRetRegs]
    int i = 0;
    int nArgRegs = iExpr.requiredArgs.size() + iExpr.namedArgs.size() + iExpr.restArgs.size();
    int nRetRegs = iExpr.types.size();
    int flags = FunctionFlags.NOTHING;
    Operand[] operands = new Operand[nArgRegs + nRetRegs + 4];
    operands[i++] = getOperand(funcRefCPIndex);
    if (iExpr.async) {
        flags = FunctionFlags.markAsync(flags);
    }
    if (iExpr.actionInvocation) {
        flags = FunctionFlags.markObserved(flags);
    }
    operands[i++] = getOperand(flags);
    operands[i++] = getOperand(nArgRegs);
    // Write required arguments
    for (BLangExpression argExpr : iExpr.requiredArgs) {
        operands[i++] = genNode(argExpr, this.env).regIndex;
    }
    // Write named arguments
    i = generateNamedArgs(iExpr, operands, i);
    // Write rest arguments
    for (BLangExpression argExpr : iExpr.restArgs) {
        operands[i++] = genNode(argExpr, this.env).regIndex;
    }
    // Calculate registers to store return values
    operands[i++] = getOperand(nRetRegs);
    RegIndex[] iExprRegIndexes;
    if (iExpr.regIndex != null) {
        iExprRegIndexes = new RegIndex[nRetRegs];
        iExprRegIndexes[0] = iExpr.regIndex;
    } else {
        iExprRegIndexes = new RegIndex[nRetRegs];
    }
    for (int j = 0; j < nRetRegs; j++) {
        RegIndex regIndex = calcAndGetExprRegIndex(iExprRegIndexes[j], iExpr.getTypes().get(j).tag);
        iExprRegIndexes[j] = regIndex;
        operands[i++] = regIndex;
    }
    iExpr.setRegIndexes(iExprRegIndexes);
    return operands;
}

Example 20 with RegIndex

use of org.wso2.ballerinalang.programfile.Instruction.RegIndex in project ballerina by ballerina-lang.

the class CodeGenerator method visit.

@Override
public void visit(BLangMapLiteral mapLiteral) {
    Operand mapVarRegIndex = calcAndGetExprRegIndex(mapLiteral);
    Operand typeCPIndex = getTypeCPIndex(mapLiteral.type);
    emit(InstructionCodes.NEWMAP, mapVarRegIndex, typeCPIndex);
    // Handle Map init stuff
    for (BLangRecordKeyValue keyValue : mapLiteral.keyValuePairs) {
        BLangExpression keyExpr = keyValue.key.expr;
        genNode(keyExpr, this.env);
        BLangExpression valueExpr = keyValue.valueExpr;
        genNode(valueExpr, this.env);
        BMapType mapType = (BMapType) mapLiteral.type;
        int opcode = getValueToRefTypeCastOpcode(mapType.constraint.tag);
        if (opcode == InstructionCodes.NOP) {
            emit(InstructionCodes.MAPSTORE, mapVarRegIndex, keyExpr.regIndex, valueExpr.regIndex);
        } else {
            RegIndex refRegMapValue = getRegIndex(TypeTags.ANY);
            emit(opcode, valueExpr.regIndex, refRegMapValue);
            emit(InstructionCodes.MAPSTORE, mapVarRegIndex, keyExpr.regIndex, refRegMapValue);
        }
    }
}