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;
	}

}