Search in sources :

Example 6 with SeparatedItemStringBuilder

use of com.github.javaparser.utils.SeparatedItemStringBuilder in project javaparser by javaparser.

the class HashCodeVisitorGenerator method generateVisitMethodBody.

@Override
protected void generateVisitMethodBody(BaseNodeMetaModel node, MethodDeclaration visitMethod, CompilationUnit compilationUnit) {
    visitMethod.getParameters().forEach(p -> p.setFinal(true));
    final BlockStmt body = visitMethod.getBody().get();
    body.getStatements().clear();
    final SeparatedItemStringBuilder builder = new SeparatedItemStringBuilder("return ", "* 31 +", ";");
    final List<PropertyMetaModel> propertyMetaModels = node.getAllPropertyMetaModels();
    if (propertyMetaModels.isEmpty()) {
        builder.append("0");
    } else {
        for (PropertyMetaModel field : propertyMetaModels) {
            final String getter = field.getGetterMethodName() + "()";
            // Is this field another AST node? Visit it.
            if (field.getNodeReference().isPresent()) {
                if (field.isOptional()) {
                    builder.append("(n.%s.isPresent()? n.%s.get().accept(this, arg):0)", getter, getter);
                } else {
                    builder.append("(n.%s.accept(this, arg))", getter);
                }
            } else {
                Class<?> type = field.getType();
                if (type.equals(boolean.class)) {
                    builder.append("(n.%s?1:0)", getter);
                } else if (type.equals(int.class)) {
                    builder.append("n.%s", getter);
                } else {
                    builder.append("(n.%s.hashCode())", getter);
                }
            }
        }
    }
    body.addStatement(parseStatement(builder.toString()));
}
Also used : SeparatedItemStringBuilder(com.github.javaparser.utils.SeparatedItemStringBuilder) BlockStmt(com.github.javaparser.ast.stmt.BlockStmt) PropertyMetaModel(com.github.javaparser.metamodel.PropertyMetaModel)

Example 7 with SeparatedItemStringBuilder

use of com.github.javaparser.utils.SeparatedItemStringBuilder in project javaparser by javaparser.

the class ModifierVisitorGenerator method generateVisitMethodBody.

@Override
protected void generateVisitMethodBody(BaseNodeMetaModel node, MethodDeclaration visitMethod, CompilationUnit compilationUnit) {
    visitMethod.getParameters().forEach(p -> p.setFinal(true));
    BlockStmt body = visitMethod.getBody().get();
    body.getStatements().clear();
    for (PropertyMetaModel property : node.getAllPropertyMetaModels()) {
        if (property.isNode()) {
            if (property.isNodeList()) {
                body.addStatement(f("NodeList<%s> %s = modifyList(n.%s(), arg);", property.getTypeNameGenerified(), property.getName(), property.getGetterMethodName()));
            } else if (property.isOptional()) {
                body.addStatement(f("%s %s = n.%s().map(s -> (%s) s.accept(this, arg)).orElse(null);", property.getTypeNameGenerified(), property.getName(), property.getGetterMethodName(), property.getTypeNameGenerified()));
            } else {
                body.addStatement(f("%s %s = (%s) n.%s().accept(this, arg);", property.getTypeNameGenerified(), property.getName(), property.getTypeNameGenerified(), property.getGetterMethodName()));
            }
        }
    }
    if (node.is(BinaryExpr.class)) {
        body.addStatement("if (left == null) return right;");
        body.addStatement("if (right == null) return left;");
    } else {
        final SeparatedItemStringBuilder collapseCheck = new SeparatedItemStringBuilder("if(", "||", ") return null;");
        for (PropertyMetaModel property : node.getAllPropertyMetaModels()) {
            if (property.isRequired() && property.isNode()) {
                if (property.isNodeList()) {
                    if (property.isNonEmpty()) {
                        collapseCheck.append(f("%s.isEmpty()", property.getName()));
                    }
                } else {
                    collapseCheck.append(f("%s==null", property.getName()));
                }
            }
        }
        if (collapseCheck.hasItems()) {
            body.addStatement(collapseCheck.toString());
        }
    }
    for (PropertyMetaModel property : node.getAllPropertyMetaModels()) {
        if (property.isNode()) {
            body.addStatement(f("n.%s(%s);", property.getSetterMethodName(), property.getName()));
        }
    }
    body.addStatement("return n;");
}
Also used : SeparatedItemStringBuilder(com.github.javaparser.utils.SeparatedItemStringBuilder) BlockStmt(com.github.javaparser.ast.stmt.BlockStmt) PropertyMetaModel(com.github.javaparser.metamodel.PropertyMetaModel)

Aggregations

SeparatedItemStringBuilder (com.github.javaparser.utils.SeparatedItemStringBuilder)7 BlockStmt (com.github.javaparser.ast.stmt.BlockStmt)5 PropertyMetaModel (com.github.javaparser.metamodel.PropertyMetaModel)5 Modifier (com.github.javaparser.ast.Modifier)2 TokenRange (com.github.javaparser.TokenRange)1 ConstructorDeclaration (com.github.javaparser.ast.body.ConstructorDeclaration)1 ArrayList (java.util.ArrayList)1