Examples with ExplicitConstructorInvocationStmt com.github.javaparser.ast.stmt.ExplicitConstructorInvocationStmt used on opensource projects

Example 6 with ExplicitConstructorInvocationStmt

use of com.github.javaparser.ast.stmt.ExplicitConstructorInvocationStmt in project drools by kiegroup.

the class KiePMMLSegmentationFactory method setConstructor.

static void setConstructor(final String generatedClassName, final String segmentationName, final ConstructorDeclaration constructorDeclaration, final MULTIPLE_MODEL_METHOD multipleModelMethod, final List<String> segmentsClasses) {
    setConstructorSuperNameInvocation(generatedClassName, constructorDeclaration, segmentationName);
    final BlockStmt body = constructorDeclaration.getBody();
    final ExplicitConstructorInvocationStmt superStatement = CommonCodegenUtils.getExplicitConstructorInvocationStmt(body).orElseThrow(() -> new KiePMMLException(String.format(MISSING_CONSTRUCTOR_IN_BODY, body)));
    CommonCodegenUtils.setExplicitConstructorInvocationStmtArgument(superStatement, "multipleModelMethod", multipleModelMethod.getClass().getCanonicalName() + "." + multipleModelMethod.name());
    final List<AssignExpr> assignExprs = body.findAll(AssignExpr.class);
    assignExprs.forEach(assignExpr -> {
        if (assignExpr.getTarget().asNameExpr().getNameAsString().equals("segments")) {
            for (String segmentClass : segmentsClasses) {
                ClassOrInterfaceType kiePMMLSegmentClass = parseClassOrInterfaceType(segmentClass);
                ObjectCreationExpr objectCreationExpr = new ObjectCreationExpr();
                objectCreationExpr.setType(kiePMMLSegmentClass);
                NodeList<Expression> arguments = NodeList.nodeList(objectCreationExpr);
                MethodCallExpr methodCallExpr = new MethodCallExpr();
                methodCallExpr.setScope(assignExpr.getTarget().asNameExpr());
                methodCallExpr.setName("add");
                methodCallExpr.setArguments(arguments);
                ExpressionStmt expressionStmt = new ExpressionStmt();
                expressionStmt.setExpression(methodCallExpr);
                body.addStatement(expressionStmt);
            }
        }
    });
}

Example 7 with ExplicitConstructorInvocationStmt

use of com.github.javaparser.ast.stmt.ExplicitConstructorInvocationStmt in project drools by kiegroup.

the class KiePMMLScorecardModelFactoryTest method setConstructor.

@Test
public void setConstructor() {
    String fullCharacteristicsClassName = PACKAGE_NAME + ".fullCharacteristicsClassName";
    final CommonCompilationDTO<Scorecard> source = CommonCompilationDTO.fromGeneratedPackageNameAndFields(PACKAGE_NAME, basicComplexPartialScorePmml, basicComplexPartialScore, new HasClassLoaderMock());
    KiePMMLScorecardModelFactory.setConstructor(ScorecardCompilationDTO.fromCompilationDTO(source), scorecardTemplate, fullCharacteristicsClassName);
    final ConstructorDeclaration constructorDeclaration = scorecardTemplate.getDefaultConstructor().orElseThrow(() -> new KiePMMLInternalException(String.format(MISSING_DEFAULT_CONSTRUCTOR, scorecardTemplate.getName())));
    final BlockStmt body = constructorDeclaration.getBody();
    final ExplicitConstructorInvocationStmt retrieved = CommonCodegenUtils.getExplicitConstructorInvocationStmt(body).orElseThrow(() -> new KiePMMLException(String.format(MISSING_CONSTRUCTOR_IN_BODY, body)));
    Statement expected = JavaParserUtils.parseStatement(String.format("super(\"%1$s\", Collections.emptyList()" + ", new %2$s" + "(), %3$s, %4$s, %5$s, %6$s);\n", getSanitizedClassName(basicComplexPartialScore.getModelName()), fullCharacteristicsClassName, basicComplexPartialScore.getInitialScore(), basicComplexPartialScore.isUseReasonCodes(), REASONCODE_ALGORITHM.class.getName() + "." + REASONCODE_ALGORITHM.byName(basicComplexPartialScore.getReasonCodeAlgorithm().value()), basicComplexPartialScore.getBaselineScore()));
    assertTrue(JavaParserUtils.equalsNode(expected, retrieved));
}

Example 8 with ExplicitConstructorInvocationStmt

use of com.github.javaparser.ast.stmt.ExplicitConstructorInvocationStmt in project drools by kiegroup.

the class KiePMMLModelFactoryUtils method setConstructorSuperNameInvocation.

/**
 * Set the <b>name</b> parameter on <b>super</b> invocation
 * @param generatedClassName
 * @param constructorDeclaration
 * @param name
 */
public static void setConstructorSuperNameInvocation(final String generatedClassName, final ConstructorDeclaration constructorDeclaration, final String name) {
    constructorDeclaration.setName(generatedClassName);
    final BlockStmt body = constructorDeclaration.getBody();
    final ExplicitConstructorInvocationStmt superStatement = CommonCodegenUtils.getExplicitConstructorInvocationStmt(body).orElseThrow(() -> new KiePMMLException(String.format(MISSING_CONSTRUCTOR_IN_BODY, body)));
    CommonCodegenUtils.setExplicitConstructorInvocationStmtArgument(superStatement, "name", String.format("\"%s\"", name));
}

Example 9 with ExplicitConstructorInvocationStmt

use of com.github.javaparser.ast.stmt.ExplicitConstructorInvocationStmt in project drools by kiegroup.

the class KiePMMLModelFactoryUtilsTest method initTest.

@Before
public void initTest() {
    CompilationUnit clonedCompilationUnit = compilationUnit.clone();
    classOrInterfaceDeclaration = clonedCompilationUnit.getClassByName(TEMPLATE_CLASS_NAME).orElseThrow(() -> new RuntimeException("Failed to retrieve ClassOrInterfaceDeclaration " + TEMPLATE_CLASS_NAME + "  from " + TEMPLATE_SOURCE)).clone();
    constructorDeclaration = classOrInterfaceDeclaration.getDefaultConstructor().orElseThrow(() -> new RuntimeException("Failed to retrieve default constructor from " + TEMPLATE_SOURCE));
    assertNotNull(constructorDeclaration);
    assertNotNull(constructorDeclaration.getBody());
    staticGetterMethod = classOrInterfaceDeclaration.getMethodsByName(GET_MODEL).get(0);
    Optional<ExplicitConstructorInvocationStmt> optSuperInvocation = CommonCodegenUtils.getExplicitConstructorInvocationStmt(constructorDeclaration.getBody());
    assertTrue(optSuperInvocation.isPresent());
    superInvocation = optSuperInvocation.get();
    // as in the original template
    assertEquals("Template", constructorDeclaration.getName().asString());
    // as in
    assertEquals("super(name, Collections.emptyList(), operator, second);", superInvocation.toString());
    // the original template
    assertTrue(clonedCompilationUnit.getClassByName(TEMPLATE_CLASS_NAME).isPresent());
}

Example 10 with ExplicitConstructorInvocationStmt

use of com.github.javaparser.ast.stmt.ExplicitConstructorInvocationStmt in project drools by kiegroup.

the class KiePMMLTreeModelFactory method setConstructor.

static void setConstructor(final DroolsCompilationDTO<TreeModel> compilationDTO, final ClassOrInterfaceDeclaration modelTemplate) {
    KiePMMLModelFactoryUtils.init(compilationDTO, modelTemplate);
    final ConstructorDeclaration constructorDeclaration = modelTemplate.getDefaultConstructor().orElseThrow(() -> new KiePMMLInternalException(String.format(MISSING_DEFAULT_CONSTRUCTOR, modelTemplate.getName())));
    final BlockStmt body = constructorDeclaration.getBody();
    final ExplicitConstructorInvocationStmt superStatement = CommonCodegenUtils.getExplicitConstructorInvocationStmt(body).orElseThrow(() -> new KiePMMLException(String.format(MISSING_CONSTRUCTOR_IN_BODY, body)));
    CommonCodegenUtils.setExplicitConstructorInvocationStmtArgument(superStatement, "algorithmName", String.format("\"%s\"", compilationDTO.getModel().getAlgorithmName()));
}