RESOLVED - bug 200128: [pde viz] Use the PDE Label provider to ensure proper decoration of the PDE Icon
https://bugs.eclipse.org/bugs/show_bug.cgi?id=200128
diff --git a/visualization/plugins/org.eclipse.pde.visualization.dependency/src/org/eclipse/pde/visualization/dependency/views/AbstractVisualizationLabelProvider.java b/visualization/plugins/org.eclipse.pde.visualization.dependency/src/org/eclipse/pde/visualization/dependency/views/AbstractVisualizationLabelProvider.java
index c7f77c4..a253894 100644
--- a/visualization/plugins/org.eclipse.pde.visualization.dependency/src/org/eclipse/pde/visualization/dependency/views/AbstractVisualizationLabelProvider.java
+++ b/visualization/plugins/org.eclipse.pde.visualization.dependency/src/org/eclipse/pde/visualization/dependency/views/AbstractVisualizationLabelProvider.java
@@ -21,6 +21,8 @@
 import org.eclipse.mylyn.zest.core.viewers.IEntityStyleProvider;
 import org.eclipse.mylyn.zest.core.widgets.ZestStyles;
 import org.eclipse.osgi.service.resolver.BundleDescription;
+import org.eclipse.pde.core.plugin.PluginRegistry;
+import org.eclipse.pde.internal.ui.PDELabelProvider;
 import org.eclipse.pde.visualization.dependency.Activator;
 import org.eclipse.swt.graphics.Color;
 import org.eclipse.swt.graphics.Image;
@@ -69,6 +71,7 @@
 	protected BundleDescription pinnedNode = null;
 	private GraphViewer viewer;
 	private boolean showVersionNumber = false;
+	private PDELabelProvider pdeLabelProvider = null;
 
 	/**
 	 * Create a new Abstract Visualization Label Provider
@@ -79,6 +82,7 @@
 	 *            is used to maintain state between the old one and the new one.
 	 */
 	public AbstractVisualizationLabelProvider(GraphViewer viewer, AbstractVisualizationLabelProvider currentLabelProvider) {
+		this.pdeLabelProvider = new PDELabelProvider();
 		this.viewer = viewer;
 		if (currentLabelProvider != null) {
 			this.showVersionNumber = currentLabelProvider.showVersionNumber;
@@ -94,6 +98,9 @@
 		if (element.getClass() == EntityConnectionData.class) {
 			return null;
 		}
+		if ( element instanceof BundleDescription ) {
+			return pdeLabelProvider.getImage(PluginRegistry.findModel((BundleDescription)element));
+		}
 		return Activator.getDefault().getImageRegistry().get(Activator.PLUGIN_OBJ);
 	}