Examples with BLangMapLiteral org.wso2.ballerinalang.compiler.tree.expressions.BLangRecordLiteral.BLangMapLiteral used on opensource projects

Example 1 with BLangMapLiteral

use of org.wso2.ballerinalang.compiler.tree.expressions.BLangRecordLiteral.BLangMapLiteral in project ballerina by ballerina-lang.

the class Desugar method getInitExpr.

private BLangExpression getInitExpr(BType type) {
    // Don't need to create an empty init expressions if the type allows null.
    if (type.isNullable()) {
        return null;
    }
    switch(type.tag) {
        case TypeTags.INT:
        case TypeTags.FLOAT:
        case TypeTags.BOOLEAN:
        case TypeTags.STRING:
        case TypeTags.BLOB:
            // Int, float, boolean, string, blob types will get default values from VM side.
            break;
        case TypeTags.JSON:
            return new BLangJSONLiteral(new ArrayList<>(), type);
        case TypeTags.XML:
            return new BLangXMLSequenceLiteral(type);
        case TypeTags.TABLE:
            return new BLangTableLiteral(type);
        case TypeTags.STREAM:
            return new BLangStreamLiteral(type, null);
        case TypeTags.MAP:
            return new BLangMapLiteral(new ArrayList<>(), type);
        case TypeTags.STRUCT:
            if (((BStructSymbol) type.tsymbol).isObject) {
                return new BLangTypeInit();
            }
            return new BLangStructLiteral(new ArrayList<>(), type);
        case TypeTags.ARRAY:
            BLangArrayLiteral array = new BLangArrayLiteral();
            array.exprs = new ArrayList<>();
            array.type = type;
            return rewriteExpr(array);
        default:
            break;
    }
    return null;
}

Example 2 with BLangMapLiteral

use of org.wso2.ballerinalang.compiler.tree.expressions.BLangRecordLiteral.BLangMapLiteral 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);
        }
    }
}

Example 3 with BLangMapLiteral

use of org.wso2.ballerinalang.compiler.tree.expressions.BLangRecordLiteral.BLangMapLiteral in project ballerina by ballerina-lang.

the class Desugar method addArgInitExpr.

private void addArgInitExpr(BLangTransformer transformerNode, BLangVariable var) {
    BLangSimpleVarRef varRef = new BLangLocalVarRef(var.symbol);
    varRef.lhsVar = true;
    varRef.pos = var.pos;
    varRef.type = var.type;
    BLangExpression initExpr = null;
    switch(var.type.tag) {
        case TypeTags.MAP:
            initExpr = new BLangMapLiteral(new ArrayList<>(), var.type);
            break;
        case TypeTags.JSON:
            initExpr = new BLangJSONLiteral(new ArrayList<>(), var.type);
            break;
        case TypeTags.STRUCT:
            initExpr = new BLangStructLiteral(new ArrayList<>(), var.type);
            break;
        case TypeTags.INT:
        case TypeTags.FLOAT:
        case TypeTags.STRING:
        case TypeTags.BOOLEAN:
        case TypeTags.BLOB:
        case TypeTags.XML:
            return;
        case TypeTags.TABLE:
        case TypeTags.STREAM:
            // TODO: add this once the able initializing is supported.
            return;
        default:
            return;
    }
    initExpr.pos = var.pos;
    BLangAssignment assignStmt = (BLangAssignment) TreeBuilder.createAssignmentNode();
    assignStmt.pos = var.pos;
    assignStmt.addVariable(varRef);
    assignStmt.expr = initExpr;
    transformerNode.body.stmts.add(0, assignStmt);
}

Example 4 with BLangMapLiteral

use of org.wso2.ballerinalang.compiler.tree.expressions.BLangRecordLiteral.BLangMapLiteral in project ballerina by ballerina-lang.

the class Desugar method visit.

@Override
public void visit(BLangRecordLiteral recordLiteral) {
    recordLiteral.keyValuePairs.forEach(keyValue -> {
        BLangExpression keyExpr = keyValue.key.expr;
        if (keyExpr.getKind() == NodeKind.SIMPLE_VARIABLE_REF) {
            BLangSimpleVarRef varRef = (BLangSimpleVarRef) keyExpr;
            keyValue.key.expr = createStringLiteral(varRef.pos, varRef.variableName.value);
        } else {
            keyValue.key.expr = rewriteExpr(keyValue.key.expr);
        }
        keyValue.valueExpr = rewriteExpr(keyValue.valueExpr);
    });
    BLangExpression expr;
    if (recordLiteral.type.tag == TypeTags.STRUCT) {
        expr = new BLangStructLiteral(recordLiteral.keyValuePairs, recordLiteral.type);
    } else if (recordLiteral.type.tag == TypeTags.MAP) {
        expr = new BLangMapLiteral(recordLiteral.keyValuePairs, recordLiteral.type);
    } else if (recordLiteral.type.tag == TypeTags.TABLE) {
        expr = new BLangTableLiteral(recordLiteral.type);
    } else if (recordLiteral.type.tag == TypeTags.STREAM) {
        expr = new BLangStreamLiteral(recordLiteral.type, recordLiteral.name);
    } else {
        expr = new BLangJSONLiteral(recordLiteral.keyValuePairs, recordLiteral.type);
    }
    result = rewriteExpr(expr);
}