org.walkmod.javalang.compiler.providers.RemoveUnusedSymbolsProvider 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.providers;

import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Stack;

import org.walkmod.javalang.ast.CompilationUnit;
import org.walkmod.javalang.ast.ImportDeclaration;
import org.walkmod.javalang.ast.Node;
import org.walkmod.javalang.ast.body.AnnotationDeclaration;
import org.walkmod.javalang.ast.body.AnnotationMemberDeclaration;
import org.walkmod.javalang.ast.body.ClassOrInterfaceDeclaration;
import org.walkmod.javalang.ast.body.ConstructorDeclaration;
import org.walkmod.javalang.ast.body.EmptyTypeDeclaration;
import org.walkmod.javalang.ast.body.EnumConstantDeclaration;
import org.walkmod.javalang.ast.body.EnumDeclaration;
import org.walkmod.javalang.ast.body.FieldDeclaration;
import org.walkmod.javalang.ast.body.MethodDeclaration;
import org.walkmod.javalang.ast.body.MultiTypeParameter;
import org.walkmod.javalang.ast.body.Parameter;
import org.walkmod.javalang.ast.body.TypeDeclaration;
import org.walkmod.javalang.ast.body.VariableDeclarator;
import org.walkmod.javalang.ast.expr.ObjectCreationExpr;
import org.walkmod.javalang.ast.expr.VariableDeclarationExpr;
import org.walkmod.javalang.ast.stmt.BlockStmt;
import org.walkmod.javalang.ast.type.ClassOrInterfaceType;
import org.walkmod.javalang.compiler.actions.ReferencesCounterAction;
import org.walkmod.javalang.compiler.actions.RemoveUnusedSymbolsAction;
import org.walkmod.javalang.compiler.symbols.SymbolAction;

public class RemoveUnusedSymbolsProvider implements SymbolActionProvider {

	private Stack> siblings = new Stack>();

	private List actions = Collections.emptyList();

	@Override
	public List getActions(ClassOrInterfaceDeclaration n) {
		buildActionList(n, n.getMembers());
		return actions;
	}

	@Override
	public List getActions(Parameter n) {

		return Collections.emptyList();
	}

	@Override
	public List getActions(ClassOrInterfaceType n) {

		return Collections.emptyList();
	}

	@Override
	public List getActions(MultiTypeParameter n) {

		return Collections.emptyList();
	}

	@Override
	public List getActions(VariableDeclarator n) {
		buildActionList();
		return actions;
	}

	@Override
	public List getActions(ImportDeclaration n) {
		buildActionList();
		return actions;
	}

	@Override
	public List getActions(MethodDeclaration n) {
		buildActionList(n, new LinkedList());
		return actions;
	}

	private void buildActionList(Node n, List children) {

		List lastsiblings = siblings.peek();
		while (!lastsiblings.contains(n) && !siblings.isEmpty()) {
			siblings.pop();
			lastsiblings = siblings.peek();
		}
		
		RemoveUnusedSymbolsAction unusedSymbols = new RemoveUnusedSymbolsAction(
				siblings.peek());
		
		actions = new LinkedList();
		actions.add(unusedSymbols);
		siblings.push(children);

	}

	private void buildActionList() {
		RemoveUnusedSymbolsAction unusedSymbols = new RemoveUnusedSymbolsAction(
				siblings.peek());
		actions = new LinkedList();
		actions.add(unusedSymbols);
	}

	@Override
	public List getActions(FieldDeclaration n) {
		buildActionList(n, n.getVariables());
		return actions;
	}

	@Override
	public List getActions(EnumDeclaration n) {

		buildActionList(n, n.getMembers());

		return actions;
	}

	@Override
	public List getActions(EnumConstantDeclaration n) {
		buildActionList();
		return actions;
	}

	@Override
	public List getActions(AnnotationDeclaration n) {

		buildActionList(n, n.getMembers());

		return actions;

	}

	@Override
	public List getActions(AnnotationMemberDeclaration n) {
		buildActionList();
		return actions;
	}

	@Override
	public List getActions(ConstructorDeclaration n) {
		buildActionList();
		return actions;
	}

	@Override
	public List getActions(CompilationUnit n) {
		siblings.clear();
		ReferencesCounterAction counter = new ReferencesCounterAction();
		List actions = new LinkedList();
		actions.add(counter);
		
		siblings.push(n.getTypes());
		if(n.getImports() != null){
			siblings.push(n.getImports());
		}
		return actions;
	}

	@Override
	public List getActions(EmptyTypeDeclaration n) {
		buildActionList(n, n.getMembers());

		return Collections.emptyList();
	}

	@Override
	public List getActions(TypeDeclaration n) {
		buildActionList(n, n.getMembers());
		return actions;
	}

	@Override
	public List getActions(BlockStmt n) {
		RemoveUnusedSymbolsAction unusedSymbols = new RemoveUnusedSymbolsAction(
				siblings.peek());
		actions = new LinkedList();
		actions.add(unusedSymbols);
		siblings.push(n.getStmts());
		return Collections.emptyList();
	}

	@Override
	public List getActions(VariableDeclarationExpr n) {
		RemoveUnusedSymbolsAction unusedSymbols = new RemoveUnusedSymbolsAction(
				siblings.peek());
		actions = new LinkedList();
		actions.add(unusedSymbols);
		siblings.push(n.getVars());
		return actions;
	}

	@Override
	public List getActions(ObjectCreationExpr n) {
		RemoveUnusedSymbolsAction unusedSymbols = new RemoveUnusedSymbolsAction(
				siblings.peek());
		actions = new LinkedList();
		actions.add(unusedSymbols);
		siblings.push(n.getAnonymousClassBody());
		return actions;
	}

}