org.walkmod.javalang.compiler.reflection.CompatibleLambdaResultPredicate 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.HashMap;
import java.util.Map;

import org.walkmod.javalang.ast.SymbolData;
import org.walkmod.javalang.ast.expr.LambdaExpr;
import org.walkmod.javalang.ast.stmt.ExpressionStmt;
import org.walkmod.javalang.compiler.Predicate;
import org.walkmod.javalang.compiler.symbols.SymbolTable;
import org.walkmod.javalang.compiler.symbols.SymbolType;
import org.walkmod.javalang.visitors.VoidVisitor;

public class CompatibleLambdaResultPredicate implements Predicate {

	private LambdaExpr expr;

	private VoidVisitor typeResolver;

	private A ctx;

	private Map typeMapping;
	
	private SymbolTable symTable;

	public CompatibleLambdaResultPredicate(LambdaExpr expr,
			VoidVisitor typeResolver, A ctx,
			Map typeMapping, 
			SymbolTable symTable) {
		this.expr = expr;
		this.typeResolver = typeResolver;
		this.ctx = ctx;
		this.typeMapping = typeMapping;
		this.symTable = symTable;
	}

	@Override
	public boolean filter(Method method) throws Exception {
		
		SymbolData returnType = expr.getBody().getSymbolData();
		if (returnType == null) {
			int scopeLevel = symTable.getScopeLevel();
			try {
				expr.accept(typeResolver, ctx);
			} catch (Exception e) {
				int currentScopeLevel = symTable.getScopeLevel();
				while(currentScopeLevel > scopeLevel){
					symTable.popScope(true);
					currentScopeLevel --;
				}
				
				return false;
			}
			returnType = expr.getBody().getSymbolData();
			if (returnType == null) {
				returnType = new SymbolType(void.class);
			}
		}
		Map updateMapping = new HashMap();

		SymbolType st = null;

		if (void.class.equals(method.getReturnType())) {
			st = new SymbolType(void.class);
		} else {
			st = SymbolType.valueOf(method.getGenericReturnType(),
					(SymbolType) returnType, updateMapping, null);
		}
		Map aux = new HashMap(
				typeMapping);
		aux.putAll(updateMapping);

		boolean isCompatible = ("void".equals(st.getName()) && (expr.getBody() instanceof ExpressionStmt))
				|| (st.isCompatible((SymbolType) returnType));
		if (isCompatible) {
			expr.setSymbolData(SymbolType.valueOf(method.getDeclaringClass(),
					aux));
		}
		return isCompatible;
	}

}

Related Artifacts