hierarchyviewer/src/com/android/hierarchyviewer/scene/ViewHierarchyScene.java - platform/sdk - Git at Google

 /*
  * Copyright (C) 2008 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 package com.android.hierarchyviewer.scene;

 import java.awt.Color;
 import java.awt.Font;
 import java.awt.GradientPaint;
 import java.awt.Graphics2D;
 import java.awt.Rectangle;
 import java.awt.geom.Point2D;

 import org.netbeans.api.visual.action.ActionFactory;
 import org.netbeans.api.visual.action.WidgetAction;
 import org.netbeans.api.visual.anchor.AnchorFactory;
 import org.netbeans.api.visual.border.BorderFactory;
 import org.netbeans.api.visual.graph.GraphScene;
 import org.netbeans.api.visual.layout.LayoutFactory;
 import org.netbeans.api.visual.model.ObjectState;
 import org.netbeans.api.visual.widget.ConnectionWidget;
 import org.netbeans.api.visual.widget.LabelWidget;
 import org.netbeans.api.visual.widget.LayerWidget;
 import org.netbeans.api.visual.widget.Widget;

 public class ViewHierarchyScene extends GraphScene<ViewNode, String> {
     private ViewNode root;
     private LayerWidget widgetLayer;
     private LayerWidget connectionLayer;

     private WidgetAction moveAction = ActionFactory.createMoveAction();

     public ViewHierarchyScene() {
         widgetLayer = new LayerWidget(this);
         connectionLayer = new LayerWidget(this);

         addChild(widgetLayer);
         addChild(connectionLayer);
     }

     public ViewNode getRoot() {
         return root;
     }

     void setRoot(ViewNode root) {
         this.root = root;
     }

     @Override
     protected Widget attachNodeWidget(ViewNode node) {
         Widget widget = createBox(node, node.name, node.id);
         widget.getActions().addAction(createSelectAction());
         widget.getActions().addAction(moveAction);
         widgetLayer.addChild(widget);
         return widget;
     }

     private Widget createBox(ViewNode node, String nodeName, String id) {
         final String shortName = getShortName(nodeName);
         node.setShortName(shortName);

         GradientWidget box = new GradientWidget(this, node);
         box.setLayout(LayoutFactory.createVerticalFlowLayout());
         box.setBorder(BorderFactory.createLineBorder(2, Color.BLACK));
         box.setOpaque(true);

         LabelWidget label = new LabelWidget(this);
         label.setFont(getDefaultFont().deriveFont(Font.PLAIN, 12.0f));
         label.setLabel(shortName);
         label.setBorder(BorderFactory.createEmptyBorder(6, 6, 0, 6));
         label.setAlignment(LabelWidget.Alignment.CENTER);

         box.addChild(label);

         label = new LabelWidget(this);
         label.setFont(getDefaultFont().deriveFont(Font.PLAIN, 10.0f));
         label.setLabel(getAddress(nodeName));
         label.setBorder(BorderFactory.createEmptyBorder(3, 6, 0, 6));
         label.setAlignment(LabelWidget.Alignment.CENTER);

         box.addressWidget = label;

         box.addChild(label);

         label = new LabelWidget(this);
         label.setFont(getDefaultFont().deriveFont(Font.PLAIN, 10.0f));
         label.setLabel(id);
         label.setBorder(BorderFactory.createEmptyBorder(3, 6, 6, 6));
         label.setAlignment(LabelWidget.Alignment.CENTER);

         box.addChild(label);

         return box;
     }

     private static String getAddress(String name) {
         String[] nameAndHashcode = name.split("@");
         return "@" + nameAndHashcode[1];
     }

     private static String getShortName(String name) {
         String[] nameAndHashcode = name.split("@");
         String[] packages = nameAndHashcode[0].split("\\.");
         return packages[packages.length - 1];
     }

     @Override
     protected Widget attachEdgeWidget(String edge) {
         ConnectionWidget connectionWidget = new ConnectionWidget(this);
         connectionLayer.addChild(connectionWidget);
         return connectionWidget;
     }

     @Override
     protected void attachEdgeSourceAnchor(String edge, ViewNode oldSourceNode, ViewNode sourceNode) {
         final ConnectionWidget connection = (ConnectionWidget) findWidget(edge);
         final Widget source = findWidget(sourceNode);
         connection.bringToBack();
         source.bringToFront();
         connection.setSourceAnchor(AnchorFactory.createRectangularAnchor(source));
     }

     @Override
     protected void attachEdgeTargetAnchor(String edge, ViewNode oldTargetNode, ViewNode targetNode) {
         final ConnectionWidget connection = (ConnectionWidget) findWidget(edge);
         final Widget target = findWidget(targetNode);
         connection.bringToBack();
         target.bringToFront();
         connection.setTargetAnchor(AnchorFactory.createRectangularAnchor(target));
     }

     private static class GradientWidget extends Widget implements ViewNode.StateListener {
         public static final GradientPaint BLUE_EXPERIENCE = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(168, 204, 241),
                 new Point2D.Double(0, 1),
                 new Color(44, 61, 146));
         public static final GradientPaint MAC_OSX_SELECTED = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(81, 141, 236),
                 new Point2D.Double(0, 1),
                 new Color(36, 96, 192));
         public static final GradientPaint MAC_OSX = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(167, 210, 250),
                 new Point2D.Double(0, 1),
                 new Color(99, 147, 206));
         public static final GradientPaint AERITH = new GradientPaint(
                 new Point2D.Double(0, 0),
                 Color.WHITE,
                 new Point2D.Double(0, 1),
                 new Color(64, 110, 161));
         public static final GradientPaint GRAY = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(226, 226, 226),
                 new Point2D.Double(0, 1),
                 new Color(250, 248, 248));
         public static final GradientPaint RED_XP = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(236, 81, 81),
                 new Point2D.Double(0, 1),
                 new Color(192, 36, 36));
         public static final GradientPaint NIGHT_GRAY = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(102, 111, 127),
                 new Point2D.Double(0, 1),
                 new Color(38, 45, 61));
         public static final GradientPaint NIGHT_GRAY_LIGHT = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(129, 138, 155),
                 new Point2D.Double(0, 1),
                 new Color(58, 66, 82));
         public static final GradientPaint NIGHT_GRAY_VERY_LIGHT = new GradientPaint(
                 new Point2D.Double(0, 0),
                 new Color(129, 138, 155, 60),
                 new Point2D.Double(0, 1),
                 new Color(58, 66, 82, 60));

         private static Color UNSELECTED = Color.BLACK;
         private static Color SELECTED = Color.WHITE;

         private final ViewNode node;

         private LabelWidget addressWidget;

         private boolean isSelected = false;
         private final GradientPaint selectedGradient = MAC_OSX_SELECTED;
         private final GradientPaint filteredGradient = RED_XP;
         private final GradientPaint focusGradient = NIGHT_GRAY_VERY_LIGHT;

         public GradientWidget(ViewHierarchyScene scene, ViewNode node) {
             super(scene);
             this.node = node;
             node.setStateListener(this);
         }

         @Override
         protected void notifyStateChanged(ObjectState previous, ObjectState state) {
             super.notifyStateChanged(previous, state);
             isSelected = state.isSelected() || state.isFocused() || state.isWidgetFocused();

             pickChildrenColor();
         }

         private void pickChildrenColor() {
             for (Widget child : getChildren()) {
                 child.setForeground(isSelected || node.filtered ? SELECTED : UNSELECTED);
             }

             repaint();
         }

         @Override
         protected void paintBackground() {
             super.paintBackground();

             Graphics2D g2 = getGraphics();
             Rectangle bounds = getBounds();

             if (!isSelected) {
                 if (!node.filtered) {
                     if (!node.hasFocus) {
                         g2.setColor(Color.WHITE);
                     } else {
                         g2.setPaint(new GradientPaint(bounds.x, bounds.y,
                                 focusGradient.getColor1(), bounds.x, bounds.x + bounds.height,
                                 focusGradient.getColor2()));
                     }
                 } else {
                     g2.setPaint(new GradientPaint(bounds.x, bounds.y, filteredGradient.getColor1(),
                         bounds.x, bounds.x + bounds.height, filteredGradient.getColor2()));
                 }
             } else {
                 g2.setPaint(new GradientPaint(bounds.x, bounds.y, selectedGradient.getColor1(),
                         bounds.x, bounds.x + bounds.height, selectedGradient.getColor2()));
             }
             g2.fillRect(bounds.x, bounds.y, bounds.width, bounds.height);
         }

         public void nodeStateChanged(ViewNode node) {
             pickChildrenColor();
         }

         public void nodeIndexChanged(ViewNode node) {
             if (addressWidget != null) {
                 addressWidget.setLabel("#" + node.index + addressWidget.getLabel());
             }
         }
     }
 }
	/*
	* Copyright (C) 2008 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	package com.android.hierarchyviewer.scene;

	import java.awt.Color;
	import java.awt.Font;
	import java.awt.GradientPaint;
	import java.awt.Graphics2D;
	import java.awt.Rectangle;
	import java.awt.geom.Point2D;

	import org.netbeans.api.visual.action.ActionFactory;
	import org.netbeans.api.visual.action.WidgetAction;
	import org.netbeans.api.visual.anchor.AnchorFactory;
	import org.netbeans.api.visual.border.BorderFactory;
	import org.netbeans.api.visual.graph.GraphScene;
	import org.netbeans.api.visual.layout.LayoutFactory;
	import org.netbeans.api.visual.model.ObjectState;
	import org.netbeans.api.visual.widget.ConnectionWidget;
	import org.netbeans.api.visual.widget.LabelWidget;
	import org.netbeans.api.visual.widget.LayerWidget;
	import org.netbeans.api.visual.widget.Widget;

	public class ViewHierarchyScene extends GraphScene<ViewNode, String> {
	private ViewNode root;
	private LayerWidget widgetLayer;
	private LayerWidget connectionLayer;

	private WidgetAction moveAction = ActionFactory.createMoveAction();

	public ViewHierarchyScene() {
	widgetLayer = new LayerWidget(this);
	connectionLayer = new LayerWidget(this);

	addChild(widgetLayer);
	addChild(connectionLayer);
	}

	public ViewNode getRoot() {
	return root;
	}

	void setRoot(ViewNode root) {
	this.root = root;
	}

	@Override
	protected Widget attachNodeWidget(ViewNode node) {
	Widget widget = createBox(node, node.name, node.id);
	widget.getActions().addAction(createSelectAction());
	widget.getActions().addAction(moveAction);
	widgetLayer.addChild(widget);
	return widget;
	}

	private Widget createBox(ViewNode node, String nodeName, String id) {
	final String shortName = getShortName(nodeName);
	node.setShortName(shortName);

	GradientWidget box = new GradientWidget(this, node);
	box.setLayout(LayoutFactory.createVerticalFlowLayout());
	box.setBorder(BorderFactory.createLineBorder(2, Color.BLACK));
	box.setOpaque(true);

	LabelWidget label = new LabelWidget(this);
	label.setFont(getDefaultFont().deriveFont(Font.PLAIN, 12.0f));
	label.setLabel(shortName);
	label.setBorder(BorderFactory.createEmptyBorder(6, 6, 0, 6));
	label.setAlignment(LabelWidget.Alignment.CENTER);

	box.addChild(label);

	label = new LabelWidget(this);
	label.setFont(getDefaultFont().deriveFont(Font.PLAIN, 10.0f));
	label.setLabel(getAddress(nodeName));
	label.setBorder(BorderFactory.createEmptyBorder(3, 6, 0, 6));
	label.setAlignment(LabelWidget.Alignment.CENTER);

	box.addressWidget = label;

	box.addChild(label);

	label = new LabelWidget(this);
	label.setFont(getDefaultFont().deriveFont(Font.PLAIN, 10.0f));
	label.setLabel(id);
	label.setBorder(BorderFactory.createEmptyBorder(3, 6, 6, 6));
	label.setAlignment(LabelWidget.Alignment.CENTER);

	box.addChild(label);

	return box;
	}

	private static String getAddress(String name) {
	String[] nameAndHashcode = name.split("@");
	return "@" + nameAndHashcode[1];
	}

	private static String getShortName(String name) {
	String[] nameAndHashcode = name.split("@");
	String[] packages = nameAndHashcode[0].split("\\.");
	return packages[packages.length - 1];
	}

	@Override
	protected Widget attachEdgeWidget(String edge) {
	ConnectionWidget connectionWidget = new ConnectionWidget(this);
	connectionLayer.addChild(connectionWidget);
	return connectionWidget;
	}

	@Override
	protected void attachEdgeSourceAnchor(String edge, ViewNode oldSourceNode, ViewNode sourceNode) {
	final ConnectionWidget connection = (ConnectionWidget) findWidget(edge);
	final Widget source = findWidget(sourceNode);
	connection.bringToBack();
	source.bringToFront();
	connection.setSourceAnchor(AnchorFactory.createRectangularAnchor(source));
	}

	@Override
	protected void attachEdgeTargetAnchor(String edge, ViewNode oldTargetNode, ViewNode targetNode) {
	final ConnectionWidget connection = (ConnectionWidget) findWidget(edge);
	final Widget target = findWidget(targetNode);
	connection.bringToBack();
	target.bringToFront();
	connection.setTargetAnchor(AnchorFactory.createRectangularAnchor(target));
	}

	private static class GradientWidget extends Widget implements ViewNode.StateListener {
	public static final GradientPaint BLUE_EXPERIENCE = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(168, 204, 241),
	new Point2D.Double(0, 1),
	new Color(44, 61, 146));
	public static final GradientPaint MAC_OSX_SELECTED = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(81, 141, 236),
	new Point2D.Double(0, 1),
	new Color(36, 96, 192));
	public static final GradientPaint MAC_OSX = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(167, 210, 250),
	new Point2D.Double(0, 1),
	new Color(99, 147, 206));
	public static final GradientPaint AERITH = new GradientPaint(
	new Point2D.Double(0, 0),
	Color.WHITE,
	new Point2D.Double(0, 1),
	new Color(64, 110, 161));
	public static final GradientPaint GRAY = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(226, 226, 226),
	new Point2D.Double(0, 1),
	new Color(250, 248, 248));
	public static final GradientPaint RED_XP = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(236, 81, 81),
	new Point2D.Double(0, 1),
	new Color(192, 36, 36));
	public static final GradientPaint NIGHT_GRAY = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(102, 111, 127),
	new Point2D.Double(0, 1),
	new Color(38, 45, 61));
	public static final GradientPaint NIGHT_GRAY_LIGHT = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(129, 138, 155),
	new Point2D.Double(0, 1),
	new Color(58, 66, 82));
	public static final GradientPaint NIGHT_GRAY_VERY_LIGHT = new GradientPaint(
	new Point2D.Double(0, 0),
	new Color(129, 138, 155, 60),
	new Point2D.Double(0, 1),
	new Color(58, 66, 82, 60));

	private static Color UNSELECTED = Color.BLACK;
	private static Color SELECTED = Color.WHITE;

	private final ViewNode node;

	private LabelWidget addressWidget;

	private boolean isSelected = false;
	private final GradientPaint selectedGradient = MAC_OSX_SELECTED;
	private final GradientPaint filteredGradient = RED_XP;
	private final GradientPaint focusGradient = NIGHT_GRAY_VERY_LIGHT;

	public GradientWidget(ViewHierarchyScene scene, ViewNode node) {
	super(scene);
	this.node = node;
	node.setStateListener(this);
	}

	@Override
	protected void notifyStateChanged(ObjectState previous, ObjectState state) {
	super.notifyStateChanged(previous, state);
	isSelected = state.isSelected() \|\| state.isFocused() \|\| state.isWidgetFocused();

	pickChildrenColor();
	}

	private void pickChildrenColor() {
	for (Widget child : getChildren()) {
	child.setForeground(isSelected \|\| node.filtered ? SELECTED : UNSELECTED);
	}

	repaint();
	}

	@Override
	protected void paintBackground() {
	super.paintBackground();

	Graphics2D g2 = getGraphics();
	Rectangle bounds = getBounds();

	if (!isSelected) {
	if (!node.filtered) {
	if (!node.hasFocus) {
	g2.setColor(Color.WHITE);
	} else {
	g2.setPaint(new GradientPaint(bounds.x, bounds.y,
	focusGradient.getColor1(), bounds.x, bounds.x + bounds.height,
	focusGradient.getColor2()));
	}
	} else {
	g2.setPaint(new GradientPaint(bounds.x, bounds.y, filteredGradient.getColor1(),
	bounds.x, bounds.x + bounds.height, filteredGradient.getColor2()));
	}
	} else {
	g2.setPaint(new GradientPaint(bounds.x, bounds.y, selectedGradient.getColor1(),
	bounds.x, bounds.x + bounds.height, selectedGradient.getColor2()));
	}
	g2.fillRect(bounds.x, bounds.y, bounds.width, bounds.height);
	}

	public void nodeStateChanged(ViewNode node) {
	pickChildrenColor();
	}

	public void nodeIndexChanged(ViewNode node) {
	if (addressWidget != null) {
	addressWidget.setLabel("#" + node.index + addressWidget.getLabel());
	}
	}
	}
	}