edu.buffalo.cse.green.relationship.dependency.DependencyIGenerator.java Source code

Introduction

Here is the source code for edu.buffalo.cse.green.relationship.dependency.DependencyIGenerator.java

Source

/* 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 java.util.AbstractList;
import java.util.List;

import org.eclipse.jdt.core.IMethod;
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.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() {
        ChooseMethodsDialog dialog = new ChooseMethodsDialog(PlugIn.getDefaultShell(), getSourceType());
        dialog.addMethodDialogListener(new MethodDialogListener() {
            public void okPressed(List<IMethod> selectedMethods) {
                _selectedMethods = selectedMethods;
            }
        });

        dialog.open();
    }

    /**
     * @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;
    }
}