org.walkmod.javalang.compiler.reflection.LambdaParamsSymbolDataBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of javalang-compiler Show documentation

Library of compiler components to processs Java code.

There is a newer version: 2.3.10

/*
 * Copyright (C) 2015 Raquel Pau and Albert Coroleu.
 * 
 * Walkmod is free software: you can redistribute it and/or modify it under the terms of the GNU
 * Lesser General Public License as published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 * 
 * Walkmod is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
 * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License along with Walkmod. If
 * not, see .
 */
package org.walkmod.javalang.compiler.reflection;

import java.lang.reflect.Method;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.walkmod.javalang.ast.SymbolData;
import org.walkmod.javalang.ast.body.Parameter;
import org.walkmod.javalang.ast.expr.LambdaExpr;
import org.walkmod.javalang.ast.type.Type;
import org.walkmod.javalang.compiler.symbols.SymbolType;
import org.walkmod.javalang.visitors.VoidVisitor;

public class LambdaParamsSymbolDataBuilder extends FunctionalGenericsBuilder {

    public LambdaParamsSymbolDataBuilder(Method method, VoidVisitor typeResolver, Map mapping)
            throws Exception {
        super(method, typeResolver, mapping);
    }

    @Override
    public LambdaExpr build(LambdaExpr lambda) throws Exception {
        super.build(lambda);
        List lambdaParams = lambda.getParameters();
        if (lambdaParams != null) {
            Iterator lambdaIt = lambdaParams.iterator();
            int k = 0;
            SymbolType[] args = getArgs();
            VoidVisitor typeResolver = getTypeResolver();
            while (lambdaIt.hasNext()) {
                Parameter p = lambdaIt.next();
                Type type = p.getType();
                if (type == null) {
                    if (k < args.length) {
                        p.setSymbolData(args[k]);
                    }
                } else {
                    SymbolData sd = type.getSymbolData();
                    if (sd == null) {
                        type.accept(typeResolver, null);
                        sd = type.getSymbolData();
                        p.setSymbolData(sd);
                    }
                }
                k++;
            }
        }
        return lambda;
    }

}