Examples with DecisionTableImpl org.kie.dmn.feel.runtime.decisiontables.DecisionTableImpl used on opensource projects

Example 1 with DecisionTableImpl

use of org.kie.dmn.feel.runtime.decisiontables.DecisionTableImpl in project drools by kiegroup.

the class DMNEvaluatorCompiler method compileDecisionTable.

protected DMNExpressionEvaluator compileDecisionTable(DMNCompilerContext ctx, DMNModelImpl model, DMNBaseNode node, String dtName, DecisionTable dt) {
    java.util.List<DTInputClause> inputs = new ArrayList<>();
    java.util.List<DMNType> inputTypes = new ArrayList<>();
    int index = 0;
    for (InputClause ic : dt.getInput()) {
        index++;
        String inputExpressionText = ic.getInputExpression().getText();
        String inputValuesText = Optional.ofNullable(ic.getInputValues()).map(UnaryTests::getText).orElse(null);
        java.util.List<UnaryTest> inputValues = null;
        DMNType inputType = model.getTypeRegistry().unknown();
        if (inputValuesText != null) {
            inputValues = textToUnaryTestList(ctx, inputValuesText, model, ic, Msg.ERR_COMPILING_FEEL_EXPR_ON_DT_INPUT_CLAUSE_IDX, inputValuesText, node.getIdentifierString(), index);
        } else if (ic.getInputExpression().getTypeRef() != null) {
            QName inputExpressionTypeRef = ic.getInputExpression().getTypeRef();
            QName resolvedInputExpressionTypeRef = DMNCompilerImpl.getNamespaceAndName(ic.getInputExpression(), model.getImportAliasesForNS(), inputExpressionTypeRef, model.getNamespace());
            BaseDMNTypeImpl typeRef = (BaseDMNTypeImpl) model.getTypeRegistry().resolveType(resolvedInputExpressionTypeRef.getNamespaceURI(), resolvedInputExpressionTypeRef.getLocalPart());
            inputType = typeRef;
            if (inputType == null) {
                MsgUtil.reportMessage(logger, DMNMessage.Severity.ERROR, dt, model, null, null, Msg.WRONG_TYPEREF_FOR_COLUMN, index, inputExpressionText, inputExpressionTypeRef);
                inputType = model.getTypeRegistry().unknown();
            } else {
                inputValues = typeRef.getAllowedValuesFEEL();
            }
        }
        CompiledExpression compiledInput = ctx.getFeelHelper().compileFeelExpression(ctx, inputExpressionText, model, dt, Msg.ERR_COMPILING_FEEL_EXPR_ON_DT_INPUT_CLAUSE_IDX, inputExpressionText, dtName, index);
        inputs.add(new DTInputClause(inputExpressionText, inputValuesText, inputValues, compiledInput, inputType.isCollection()));
        inputTypes.add(inputType);
    }
    java.util.List<DTOutputClause> outputs = new ArrayList<>();
    index = 0;
    boolean hasOutputValues = false;
    for (OutputClause oc : dt.getOutput()) {
        String outputName = oc.getName();
        if (outputName != null) {
            DMNCompilerHelper.checkVariableName(model, node.getSource(), outputName);
        }
        String id = oc.getId();
        String outputValuesText = Optional.ofNullable(oc.getOutputValues()).map(UnaryTests::getText).orElse(null);
        String defaultValue = Optional.ofNullable(oc.getDefaultOutputEntry()).map(LiteralExpression::getText).filter(t -> !t.isEmpty()).orElse(null);
        BaseDMNTypeImpl typeRef = inferTypeRef(model, dt, oc);
        java.util.List<UnaryTest> outputValues = null;
        if (outputValuesText != null) {
            outputValues = textToUnaryTestList(ctx, outputValuesText, model, oc, Msg.ERR_COMPILING_FEEL_EXPR_ON_DT_OUTPUT_CLAUSE_IDX, outputValuesText, node.getIdentifierString(), ++index);
        } else if (typeRef != model.getTypeRegistry().unknown()) {
            outputValues = typeRef.getAllowedValuesFEEL();
        }
        if (outputValues != null && !outputValues.isEmpty()) {
            hasOutputValues = true;
        }
        outputs.add(new DTOutputClause(outputName, id, outputValues, defaultValue, typeRef.getFeelType(), typeRef.isCollection()));
    }
    if (dt.getHitPolicy().equals(HitPolicy.PRIORITY) && !hasOutputValues) {
        MsgUtil.reportMessage(logger, DMNMessage.Severity.ERROR, dt.getParent(), model, null, null, Msg.MISSING_OUTPUT_VALUES, dtName);
    }
    java.util.List<DTDecisionRule> rules = new ArrayList<>();
    index = 0;
    for (DecisionRule dr : dt.getRule()) {
        DTDecisionRule rule = new DTDecisionRule(index);
        for (int i = 0; i < dr.getInputEntry().size(); i++) {
            UnaryTests ut = dr.getInputEntry().get(i);
            final java.util.List<UnaryTest> tests;
            if (ut == null || ut.getText() == null || ut.getText().isEmpty()) {
                tests = Collections.emptyList();
                MsgUtil.reportMessage(logger, DMNMessage.Severity.ERROR, ut, model, null, null, Msg.DTABLE_EMPTY_ENTRY, dt.getRule().indexOf(dr) + 1, dr.getInputEntry().indexOf(ut) + 1, dt.getParentDRDElement().getIdentifierString());
            } else {
                ctx.enterFrame();
                try {
                    ctx.setVariable("?", inputTypes.get(i));
                    tests = textToUnaryTestList(ctx, ut.getText(), model, dr, Msg.ERR_COMPILING_FEEL_EXPR_ON_DT_RULE_IDX, ut.getText(), node.getIdentifierString(), index + 1);
                } finally {
                    ctx.exitFrame();
                }
            }
            rule.getInputEntry().add((c, x) -> tests.stream().anyMatch(t -> {
                Boolean result = t.apply(c, x);
                return result != null && result;
            }));
        }
        for (LiteralExpression le : dr.getOutputEntry()) {
            String expressionText = le.getText();
            if (expressionText == null || expressionText.isEmpty()) {
                // addendum to DROOLS-2075 Allow empty output cell on DTs
                expressionText = "null";
            }
            CompiledExpression compiledExpression = ctx.getFeelHelper().compileFeelExpression(ctx, expressionText, model, dr, Msg.ERR_COMPILING_FEEL_EXPR_ON_DT_RULE_IDX, expressionText, dtName, index + 1);
            rule.getOutputEntry().add(compiledExpression);
        }
        rules.add(rule);
        index++;
    }
    String policy = dt.getHitPolicy().value() + (dt.getAggregation() != null ? " " + dt.getAggregation().value() : "");
    org.kie.dmn.feel.runtime.decisiontables.HitPolicy hp = org.kie.dmn.feel.runtime.decisiontables.HitPolicy.fromString(policy);
    java.util.List<String> parameterNames = getParameters(model, node, dt);
    // DROOLS-2799 DMN Optimize DT parameter binding for compilation:
    java.util.List<CompiledExpression> compiledParameterNames = new ArrayList<>();
    for (String pName : parameterNames) {
        CompiledExpression compiledExpression = ctx.getFeelHelper().compileFeelExpression(ctx, pName, model, dt, Msg.ERR_COMPILING_FEEL_EXPR_ON_DT_PARAM, pName, dtName);
        compiledParameterNames.add(compiledExpression);
    }
    // creates a FEEL instance which will be used by the invoker/impl (s)
    FEEL feelInstance = ctx.getFeelHelper().newFEELInstance();
    DecisionTableImpl dti = new DecisionTableImpl(dtName, parameterNames, inputs, outputs, rules, hp, feelInstance);
    dti.setCompiledParameterNames(compiledParameterNames);
    DTInvokerFunction dtf = new DTInvokerFunction(dti);
    DMNDTExpressionEvaluator dtee = new DMNDTExpressionEvaluator(node, feelInstance, dtf);
    return dtee;
}

Example 2 with DecisionTableImpl

use of org.kie.dmn.feel.runtime.decisiontables.DecisionTableImpl in project drools by kiegroup.

the class DecisionTableFunction method invoke.

/**
 *     @param inputExpressionList a list of the N>=0 input expressions in display order
 *     @param inputValuesList * a list of N input values, corresponding to the input expressions. Each
 *     list element is a unary tests literal (see below).
 *     @param outputs * a name (a string matching grammar rule 27) or a list of M>0 names
 *     @param outputValues * if outputs is a list, then output values is a list of lists of values, one list
 *     per output; else output values is a list of values for the one output.
 *     Each value is a string.
 *     @param ruleList a list of R>0 rules. A rule is a list of N input entries followed by M
 *     output entries. An input entry is a unary tests literal. An output entry is
 *     an expression represented as a string.
 *     @param hitPolicy * one of: "U", "A", “P”, “F”, "R", "O", "C", "C+", "C#", "C<", “C>”
 *     (default is "U")
 *     @param defaultOutputValue * if outputs is a list, then default output value is a context with entries
 *     composed of outputs and output values; else default output value is one
 *     of the output values.
 */
public Object invoke(@ParameterName("ctx") EvaluationContext ctx, @ParameterName("outputs") Object outputs, @ParameterName("input expression list") Object inputExpressionList, @ParameterName("input values list") List<?> inputValuesList, @ParameterName("output values") Object outputValues, @ParameterName("rule list") List<List> ruleList, @ParameterName("hit policy") String hitPolicy, @ParameterName("default output value") Object defaultOutputValue) {
    // input expression list can have a single element or be a list
    // TODO isn't ^ conflicting with the specs page 136 "input expression list: a LIST of the"
    List<String> inputExpressions = inputExpressionList instanceof List ? (List) inputExpressionList : Collections.singletonList((String) inputExpressionList);
    List<DTInputClause> inputs;
    if (inputValuesList != null) {
        List<UnaryTest> inputValues = inputValuesList.stream().map(o -> toUnaryTest(ctx, o)).collect(Collectors.toList());
        if (inputValues.size() != inputExpressions.size()) {
        // TODO handle compilation error
        }
        // zip inputExpression with its inputValue
        inputs = IntStream.range(0, inputExpressions.size()).mapToObj(i -> new DTInputClause(inputExpressions.get(i), inputValuesList.toString(), Collections.singletonList(inputValues.get(i)), null, false)).collect(Collectors.toList());
    } else {
        inputs = inputExpressions.stream().map(ie -> new DTInputClause(ie, null, null, null, false)).collect(Collectors.toList());
    }
    List<String> parseOutputs = outputs instanceof List ? (List) outputs : Collections.singletonList((String) outputs);
    List<DTOutputClause> outputClauses;
    if (outputValues != null) {
        if (parseOutputs.size() == 1) {
            outputClauses = new ArrayList<>();
            List<UnaryTest> outputValuesCompiled = objectToUnaryTestList(ctx, Collections.singletonList((List<Object>) outputValues)).get(0);
            outputClauses.add(new DTOutputClause(parseOutputs.get(0), outputValuesCompiled));
        } else {
            List<List<UnaryTest>> listOfList = objectToUnaryTestList(ctx, (List<List<Object>>) outputValues);
            // zip inputExpression with its inputValue
            outputClauses = IntStream.range(0, parseOutputs.size()).mapToObj(i -> new DTOutputClause(parseOutputs.get(i), listOfList.get(i))).collect(Collectors.toList());
        }
    } else {
        outputClauses = parseOutputs.stream().map(out -> new DTOutputClause(out, null)).collect(Collectors.toList());
    }
    // TODO parse default output value.
    FEEL feel = FEEL.newInstance();
    List<DTDecisionRule> decisionRules = IntStream.range(0, ruleList.size()).mapToObj(index -> toDecisionRule(ctx, feel, index, ruleList.get(index), inputExpressions.size())).collect(Collectors.toList());
    // TODO is there a way to avoid UUID and get from _evaluation_ ctx the name of the wrapping context?
    // TODO also in this case it is using an ad-hoc created FEEL instance instead of the "hosted" one.
    DecisionTableImpl dti = new DecisionTableImpl(UUID.randomUUID().toString(), inputExpressions, inputs, outputClauses, decisionRules, HitPolicy.fromString(hitPolicy), FEEL.newInstance());
    return new DTInvokerFunction(dti);
}