CFGGraphEditPart.java :  » Code-Analyzer » soot » ca » mcgill » sable » soot » cfg » editParts » Java Open Source

/* Soot - a J*va Optimization Framework
 * Copyright (C) 2004 Jennifer Lhotak
 *
 * This library 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 2.1 of the License, or (at your option) any later version.
 *
 * This library 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 this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */


package ca.mcgill.sable.soot.cfg.editParts;

import org.eclipse.draw2d.*;
import org.eclipse.gef.editparts.*;
import org.eclipse.draw2d.graph.*;
import org.eclipse.gef.*;
import org.eclipse.draw2d.geometry.*;
import java.util.*;
import ca.mcgill.sable.soot.cfg.model.*;
import java.beans.*;


public class CFGGraphEditPart extends AbstractGraphicalEditPart 
  implements PropertyChangeListener {

  private int figureWidth = 20000;
  private int figureHeight = 20000;
  
  public CFGGraphEditPart() {
    super();
  }

  /* (non-Javadoc)
   * @see org.eclipse.gef.editparts.AbstractGraphicalEditPart#createFigure()
   */
  protected IFigure createFigure() {
    IFigure f = new Figure() {
      public void setBound(Rectangle rect){
        int x = bounds.x; 
        int y = bounds.y;

        boolean resize = (rect.width != bounds.width) || (rect.height != bounds.height),
            translate = (rect.x != x) || (rect.y != y);

        if (isVisible() && (resize || translate))
          erase();
        if (translate) {
          int dx = rect.x - x;
          int dy = rect.y - y;
          primTranslate(dx, dy);
        }
        bounds.width = rect.width;
        bounds.height = rect.height;
  
        if (resize || translate) {
          fireMoved();
          repaint();
        }
      }
    };
    f.setLayoutManager(new CFGGraphLayoutManager(this));
    return f;
  }

  protected void setFigure(IFigure figure){
    figure.getBounds().setSize(getFigureWidth(),getFigureHeight());
    super.setFigure(figure);
  }
  
  /* (non-Javadoc)
   * @see org.eclipse.gef.editparts.AbstractEditPart#createEditPolicies()
   */
  protected void createEditPolicies() {
  }
  
  public void contributeNodesToGraph(DirectedGraph graph, HashMap map){
    Iterator it = getChildren().iterator();
    while (it.hasNext()){
      Object next = it.next();
      if (next instanceof CFGNodeEditPart){
        ((CFGNodeEditPart)next).contributeNodesToGraph(graph, map);
      }
    }
    
  }
  public void contributeEdgesToGraph(DirectedGraph graph, HashMap map){
    Iterator it = getChildren().iterator();
    while (it.hasNext()){
      Object next = it.next();
      if (next instanceof CFGNodeEditPart){
        ((CFGNodeEditPart)next).contributeEdgesToGraph(graph, map);
      }
    }
    
  }

  public void applyGraphResults(DirectedGraph graph, HashMap map){
    Iterator it = getChildren().iterator();
    while (it.hasNext()){
      Object next = it.next();
      if (next instanceof CFGNodeEditPart){
        ((CFGNodeEditPart)next).applyGraphResults(graph, map);
      }
    }
    determineGraphBounds(graph);
  }
  
  public void resetChildColors(){
    Iterator it = getChildren().iterator();
    while (it.hasNext()){
      Object next = it.next();
      if (next instanceof CFGNodeEditPart){
        ((CFGNodeEditPart)next).resetColors();
      }
    }
  }
  
  private void determineGraphBounds(DirectedGraph graph){
    Iterator it = graph.nodes.iterator();
    int width = 0;
    int height = 0;
    while (it.hasNext()){
      Node n = (Node)it.next();
      if (width < n.x){
        width = n.x + 300;
      }
      height = max(height, n.height);
    }
    setFigureWidth(width);
    setFigureHeight(height);
  }
  
  private int max(int i, int j){
    return i < j ? j : i;
  }
  
  public CFGGraph getGraph(){
    return (CFGGraph)getModel();
  }
  
  public List getModelChildren(){
    return getGraph().getChildren();
  }

  public void activate(){
    super.activate();
    getGraph().addPropertyChangeListener(this);
  }
  
  public void deactivate(){
    super.deactivate();
    getGraph().removePropertyChangeListener(this);
  }
  
  public void propertyChange(PropertyChangeEvent event){
    if (event.getPropertyName().equals(CFGElement.CHILDREN)){
      refreshChildren();
    }
    else if (event.getPropertyName().equals(CFGElement.NEW_FLOW_DATA)){
      resetChildColors();
    }
    ((GraphicalEditPart)(getViewer().getContents())).getFigure().revalidate();
  }
  /**
   * @return
   */
  public int getFigureHeight() {
    return figureHeight;
  }

  /**
   * @return
   */
  public int getFigureWidth() {
    return figureWidth;
  }

  /**
   * @param i
   */
  public void setFigureHeight(int i) {
    figureHeight = i;
  }

  /**
   * @param i
   */
  public void setFigureWidth(int i) {
    figureWidth = i;
  }

  public void handleClickEvent(Object evt){
    ((CFGGraph)getModel()).handleClickEvent(evt);
  }
}