/* This file is part of Green.
*
* Copyright (C) 2005 The Research Foundation of State University of New York
* All Rights Under Copyright Reserved, The Research Foundation of S.U.N.Y.
*
* Green is free software, licensed under the terms of the Eclipse
* Public License, version 1.0. The license is available at
* http://www.eclipse.org/legal/epl-v10.html
*/
package edu.buffalo.cse.green.relationship.dependency;
import static edu.buffalo.cse.green.GreenException.GRERR_RELATIONSHIP_NO_METHODS;
import java.util.AbstractList;
import java.util.List;
import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.core.dom.Block;
import org.eclipse.jdt.core.dom.ClassInstanceCreation;
import org.eclipse.jdt.core.dom.ExpressionStatement;
import org.eclipse.jdt.core.dom.Statement;
import edu.buffalo.cse.green.GreenException;
import edu.buffalo.cse.green.PlugIn;
import edu.buffalo.cse.green.dialogs.ChooseMethodsDialog;
import edu.buffalo.cse.green.dialogs.MethodDialogListener;
import edu.buffalo.cse.green.relationships.CardinalRelationshipGenerator;
/**
* @see edu.buffalo.cse.green.relationship.CardinalRelationshipGenerator
*
* @author bcmartin
*/
public class DependencyIGenerator extends CardinalRelationshipGenerator {
private List<IMethod> _selectedMethods;
/**
* @see edu.buffalo.cse.green.relationships.RelationshipVisitor#preVisit()
*/
protected void preVisit() {
_selectedMethods = null; //Ensures that previously used methods do not carry over
try {
if (getSourceType().getMethods().length == 0)
GreenException.errorDialog(GRERR_RELATIONSHIP_NO_METHODS);
else {
ChooseMethodsDialog dialog = new ChooseMethodsDialog(
PlugIn.getDefaultShell(), getSourceType());
dialog.addMethodDialogListener(new MethodDialogListener() {
public void okPressed(List<IMethod> selectedMethods) {
_selectedMethods = selectedMethods;
}
});
dialog.open();
}
}
catch (JavaModelException jme) {
jme.printStackTrace();
}
}
/**
* @see edu.buffalo.cse.green.relationships.RelationshipGenerator#process(org.eclipse.jdt.core.dom.Block)
*/
protected boolean process(Block block) {
if (_selectedMethods == null) return true;
if (!correctTypeToGenerate()) return true;
IMethod method =
(IMethod) getMethodDeclaration().resolveBinding().getJavaElement();
if (!_selectedMethods.contains(method)) return true;
List<Statement> statements =
(AbstractList<Statement>) (List) block.statements();
for (int x = 0; x < _cardinality; x++) {
ClassInstanceCreation cic =
createInvocation(createTypeReference(getActualTargetType()));
ExpressionStatement stmt = getAST().newExpressionStatement(cic);
statements.add(stmt);
}
return true;
}
public boolean needConstructor() {
return false;
}
}