Examples with RecordUpdateExpr jkind.lustre.RecordUpdateExpr used on opensource projects

Example 1 with RecordUpdateExpr

use of jkind.lustre.RecordUpdateExpr in project AGREE by loonwerks.

the class LustreToMATLABExprVisitor method visit.

@Override
public MATLABExpr visit(RecordUpdateExpr e) {
    MATLABIdExpr recordIdExpr = (MATLABIdExpr) e.record.accept(this);
    // Assign the specific field of the variable created from the recordExpr associated with it
    // to the value specified in the RecordUpdateExpr
    SortedMap<String, MATLABExpr> fields = new TreeMap<>(new StringNaturalOrdering());
    // get the type for the field
    Expr curExpr = e;
    while (curExpr instanceof RecordUpdateExpr) {
        curExpr = ((RecordUpdateExpr) curExpr).record;
    }
    MATLABType type = null;
    String recordName = "";
    if (curExpr instanceof RecordExpr) {
        recordName = ((RecordExpr) curExpr).id;
        if (recordTypeMap.get(recordName) != null) {
            String fieldName = e.field;
            fieldName = updateName(fieldName, recordName);
            type = recordTypeMap.get(recordName).get(fieldName);
        }
    } else {
        recordName = recordIdExpr.id;
    }
    // conduct explicit type cast if a field value is a constant of double type or int type
    MATLABTypeCastExprVisitor typeCastVisitor = new MATLABTypeCastExprVisitor(type);
    MATLABExpr fieldExpr = typeCastVisitor.visit(e.value.accept(this));
    fields.put(updateName(e.field, recordName), fieldExpr);
    String originalVar = null;
    if (e.record instanceof IdExpr) {
        originalVar = updateName(e.record.toString(), "");
    } else {
        originalVar = updateName(recordIdExpr.id.split("_var")[0] + "_var" + localVarIndex, "");
    }
    localVarIndex++;
    String newVar = updateName(recordIdExpr.id.split("_var")[0] + "_var" + localVarIndex, "");
    localBusVarInits.add(new MATLABLocalBusVarInit(originalVar, newVar, fields));
    // In the expression that uses the RecordUpdateExpr, just reference the variable
    return new MATLABIdExpr(newVar);
}

Example 2 with RecordUpdateExpr

use of jkind.lustre.RecordUpdateExpr in project AGREE by loonwerks.

the class ExpressionFlattener method flattenRecordExpression.

private static void flattenRecordExpression(final Expr expr, final RecordType recordType, final Map<String, Expr> results) {
    if (expr instanceof RecordUpdateExpr) {
        final RecordUpdateExpr updateExpr = (RecordUpdateExpr) expr;
        // Handle base record
        flattenRecordExpression(updateExpr.record, recordType, results);
        // Handle the update
        results.put(updateExpr.field, updateExpr.value);
    } else if (expr instanceof RecordExpr) {
        final RecordExpr recordExpr = (RecordExpr) expr;
        results.putAll(recordExpr.fields);
    } else {
        // IdExpr
        for (final String field : recordType.fields.keySet()) {
            results.put(field, new RecordAccessExpr(expr, field));
        }
    }
}