Skip to content

Commit

Permalink
feature: add CatchVariableReferenceFunction, CatchVariableScopeFunction
Browse files Browse the repository at this point in the history
  • Loading branch information
pvojtechovsky committed Jan 22, 2017
1 parent 7694957 commit e24c1e1
Show file tree
Hide file tree
Showing 2 changed files with 99 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
/**
* Copyright (C) 2006-2017 INRIA and contributors
* Spoon - http://spoon.gforge.inria.fr/
*
* This software is governed by the CeCILL-C License under French law and
* abiding by the rules of distribution of free software. You can use, modify
* and/or redistribute the software under the terms of the CeCILL-C license as
* circulated by CEA, CNRS and INRIA at http://www.cecill.info.
*
* This program 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 CeCILL-C License for more details.
*
* The fact that you are presently reading this means that you have had
* knowledge of the CeCILL-C license and that you accept its terms.
*/
package spoon.reflect.visitor.filter;

import spoon.reflect.code.CtCatchVariable;
import spoon.reflect.reference.CtCatchVariableReference;
import spoon.reflect.visitor.chain.CtConsumableFunction;
import spoon.reflect.visitor.chain.CtConsumer;

/**
* This Query expects a {@link CtCatchVariable} as input
* and returns all {@link CtCatchVariableReference}s, which refers this input.
* <br>
* Usage:<br>
* <pre> {@code
* CtCatchVariable var = ...;
* var
* .map(new CatchVariableReferenceFunction())
* .forEach((CtCatchVariableReference ref)->...process references...);
* }
* </pre>
*/
public class CatchVariableReferenceFunction implements CtConsumableFunction<CtCatchVariable<?>> {

public CatchVariableReferenceFunction() {
}

@Override
public void apply(CtCatchVariable<?> localVariable, CtConsumer<Object> outputConsumer) {
localVariable
.map(new CatchVariableScopeFunction())
.select(new DirectReferenceFilter<CtCatchVariableReference<?>>(localVariable.getReference()))
.forEach(outputConsumer);
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
/**
* Copyright (C) 2006-2017 INRIA and contributors
* Spoon - http://spoon.gforge.inria.fr/
*
* This software is governed by the CeCILL-C License under French law and
* abiding by the rules of distribution of free software. You can use, modify
* and/or redistribute the software under the terms of the CeCILL-C license as
* circulated by CEA, CNRS and INRIA at http://www.cecill.info.
*
* This program 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 CeCILL-C License for more details.
*
* The fact that you are presently reading this means that you have had
* knowledge of the CeCILL-C license and that you accept its terms.
*/
package spoon.reflect.visitor.filter;

import spoon.reflect.code.CtCatch;
import spoon.reflect.code.CtCatchVariable;
import spoon.reflect.visitor.chain.CtConsumableFunction;
import spoon.reflect.visitor.chain.CtConsumer;

/**
* This Query expects a {@link CtCatchVariable} as input
* and returns all CtElements,
* which are in visibility scope of that catch variable.
* In other words, it returns all elements,
* which might be reference to that catch variable.
* <br>
* It can be used to search for variable declarations or
* variable references which might be in name conflict with input catch variable.
* <br>
* Usage:<br>
* <pre> {@code
* CtCatchVariable var = ...;
* var.map(new CatchVariableScopeFunction()).forEach(...process result...);
* }
* </pre>
*/
public class CatchVariableScopeFunction implements CtConsumableFunction<CtCatchVariable<?>> {

public CatchVariableScopeFunction() {
}

@Override
public void apply(CtCatchVariable<?> catchVariable, CtConsumer<Object> outputConsumer) {
catchVariable.getParent(CtCatch.class).getBody().filterChildren(null).forEach(outputConsumer);
}
}

0 comments on commit e24c1e1

Please sign in to comment.