/*******************************************************************************
* Copyright 2005-2006, CHISEL Group, University of Victoria, Victoria, BC,
* Canada. All rights reserved. This program and the accompanying materials are
* made available under the terms of the Eclipse Public License v1.0 which
* accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors: The Chisel Group, University of Victoria
******************************************************************************/
package org.eclipse.zest.core.viewers.internal;
import org.eclipse.draw2d.IFigure;
import org.eclipse.jface.viewers.IBaseLabelProvider;
import org.eclipse.jface.viewers.IColorProvider;
import org.eclipse.jface.viewers.IFontProvider;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.swt.SWT;
import org.eclipse.swt.SWTError;
import org.eclipse.swt.graphics.Color;
import org.eclipse.zest.core.viewers.IConnectionStyleProvider;
import org.eclipse.zest.core.viewers.IEntityConnectionStyleProvider;
import org.eclipse.zest.core.viewers.IEntityStyleProvider;
import org.eclipse.zest.core.viewers.ISelfStyleProvider;
import org.eclipse.zest.core.widgets.GraphConnection;
import org.eclipse.zest.core.widgets.GraphItem;
import org.eclipse.zest.core.widgets.GraphNode;
import org.eclipse.zest.core.widgets.ZestStyles;
/*
* Helper class used to style graph elements based on graph element stylers.
*
* @author Del Myers
*/
// @tag bug(151327-Styles) : created to help resolve this bug
public class GraphItemStyler {
public static void styleItem(GraphItem item, final IBaseLabelProvider labelProvider) {
if (item instanceof GraphNode) {
GraphNode node = (GraphNode) item;
// set defaults.
if (node.getGraphModel().getNodeStyle() != ZestStyles.NONE) {
node.setNodeStyle(node.getGraphModel().getNodeStyle());
} else {
node.setNodeStyle(SWT.NONE);
}
Object entity = node.getData();
if (labelProvider instanceof IEntityStyleProvider) {
styleNode(node, (IEntityStyleProvider) labelProvider);
}
if (labelProvider instanceof IColorProvider) {
IColorProvider colorProvider = (IColorProvider) labelProvider;
node.setForegroundColor(colorProvider.getForeground(entity));
node.setBackgroundColor(colorProvider.getBackground(entity));
}
if (labelProvider instanceof IFontProvider) {
IFontProvider fontProvider = (IFontProvider) labelProvider;
node.setFont(fontProvider.getFont(entity));
}
if (labelProvider instanceof ILabelProvider) {
String text = ((ILabelProvider) labelProvider).getText(node.getData());
node.setText((text != null) ? text : "");
node.setImage(((ILabelProvider) labelProvider).getImage(node.getData()));
}
if (labelProvider instanceof ISelfStyleProvider) {
((ISelfStyleProvider) labelProvider).selfStyleNode(entity, node);
}
} else if (item instanceof GraphConnection) {
GraphConnection conn = (GraphConnection) item;
// set defaults
if (conn.getGraphModel().getConnectionStyle() != ZestStyles.NONE) {
int s = conn.getGraphModel().getConnectionStyle();
conn.setConnectionStyle(s);
} else {
conn.setConnectionStyle(SWT.NONE);
}
if (labelProvider instanceof ILabelProvider) {
String text = ((ILabelProvider) labelProvider).getText(conn.getExternalConnection());
conn.setText((text != null) ? text : "");
conn.setImage(((ILabelProvider) labelProvider).getImage(conn.getExternalConnection()));
}
if (labelProvider instanceof IEntityConnectionStyleProvider) {
styleEntityConnection(conn, (IEntityConnectionStyleProvider) labelProvider);
} else if (labelProvider instanceof IConnectionStyleProvider) {
styleConnection(conn, (IConnectionStyleProvider) labelProvider);
}
int swt = getLineStyleForZestStyle(conn.getConnectionStyle());
conn.setLineStyle(swt);
if (labelProvider instanceof ISelfStyleProvider) {
((ISelfStyleProvider) labelProvider).selfStyleConnection(conn.getData(), conn);
}
}
}
/**
* @param conn
* @param provider
*/
private static void styleConnection(GraphConnection conn, IConnectionStyleProvider provider) {
Object rel = conn.getExternalConnection();
Color c;
int style = provider.getConnectionStyle(rel);
if (!ZestStyles.validateConnectionStyle(style)) {
throw new SWTError(SWT.ERROR_INVALID_ARGUMENT);
}
if (style != ZestStyles.NONE) {
conn.setConnectionStyle(style);
}
// @tag bug(152530-Bezier(fix))
// @tat TODO curves bezier: Add back the bezier connection stuff
// if (ZestStyles.checkStyle(conn.getConnectionStyle(),
// ZestStyles.CONNECTIONS_BEZIER)
// && provider instanceof IConnectionStyleBezierExtension) {
// IConnectionStyleBezierExtension bezier =
// (IConnectionStyleBezierExtension) provider;
// double d;
// if (!Double.isNaN((d = bezier.getStartAngle(rel)))) {
// conn.setStartAngle(d);
// }
// if (!Double.isNaN((d = bezier.getEndAngle(rel)))) {
// conn.setEndAngle(d);
// }
// if (!Double.isNaN((d = bezier.getStartDistance(rel)))) {
// conn.setStartLength(d);
// }
// if (!Double.isNaN((d = bezier.getEndDistance(rel)))) {
// conn.setEndLength(d);
// }
// }
if ((c = provider.getHighlightColor(rel)) != null) {
conn.setHighlightColor(c);
}
if ((c = provider.getColor(rel)) != null) {
conn.setLineColor(c);
}
IFigure tooltip;
if ((tooltip = provider.getTooltip(rel)) != null) {
conn.setTooltip(tooltip);
}
int w = -1;
if ((w = provider.getLineWidth(rel)) >= 0) {
conn.setLineWidth(w);
}
}
/**
* @param conn
* @param provider
*/
private static void styleEntityConnection(GraphConnection conn, IEntityConnectionStyleProvider provider) {
Object src = conn.getSource().getData();
Object dest = conn.getDestination().getData();
Color c;
int style = provider.getConnectionStyle(src, dest);
if (!ZestStyles.validateConnectionStyle(style)) {
throw new SWTError(SWT.ERROR_INVALID_ARGUMENT);
}
if (style != ZestStyles.NONE) {
conn.setConnectionStyle(style);
}
// @tag bug(152530-Bezier(fisx))
// @tag TODO curved connections bezier : add back the bezier connection stuff
// if (ZestStyles.checkStyle(conn.getConnectionStyle(),
// ZestStyles.CONNECTIONS_BEZIER)
// && provider instanceof IEntityConnectionStyleBezierExtension) {
// IEntityConnectionStyleBezierExtension bezier =
// (IEntityConnectionStyleBezierExtension) provider;
// double d;
// if (!Double.isNaN((d = bezier.getStartAngle(src, dest)))) {
// conn.setStartAngle(d);
// }
// if (!Double.isNaN((d = bezier.getEndAngle(src, dest)))) {
// conn.setEndAngle(d);
// }
// if (!Double.isNaN((d = bezier.getStartDistance(src, dest)))) {
// conn.setStartLength(d);
// }
// if (!Double.isNaN((d = bezier.getEndDistance(src, dest)))) {
// conn.setEndLength(d);
// }
// }
if ((c = provider.getColor(src, dest)) != null) {
conn.setLineColor(c);
}
if ((c = provider.getHighlightColor(src, dest)) != null) {
conn.setHighlightColor(c);
}
int w = -1;
if ((w = provider.getLineWidth(src, dest)) >= 0) {
conn.setLineWidth(w);
}
}
/**
* Styles the given node according to the properties in the style provider.
*
* @param node
* the graph element to style.
* @param data
* the element that is being styled.
* @param provider
* the style provier.
*/
// @tag bug(151327-Styles) : resolution
private static void styleNode(GraphNode node, IEntityStyleProvider provider) {
Object entity = node.getData();
// @tag ADJACENT : Removed highlight adjacent
//node.setHighlightAdjacentNodes(provider.highlightAdjacentEntities(entity));
// @tag ADJACENT : Removed highlight adjacent
/*
if (provider.highlightAdjacentEntities(entity)) {
Color c = provider.getAdjacentEntityHighlightColor(entity);
if (c != null) {
node.setHighlightAdjacentColor(c);
}
}
*/
Color c;
IFigure figure;
int width = -1;
if (provider.fisheyeNode(entity) == true) {
node.setNodeStyle(node.getNodeStyle() | ZestStyles.NODES_FISHEYE);
}
if ((c = provider.getBorderColor(entity)) != null) {
node.setBorderColor(c);
}
if ((c = provider.getBorderHighlightColor(entity)) != null) {
node.setBorderHighlightColor(c);
}
if ((c = provider.getNodeHighlightColor(entity)) != null) {
node.setHighlightColor(c);
}
if ((c = provider.getBackgroundColour(entity)) != null) {
node.setBackgroundColor(c);
}
if ((c = provider.getForegroundColour(entity)) != null) {
node.setForegroundColor(c);
}
if ((width = provider.getBorderWidth(entity)) >= 0) {
node.setBorderWidth(width);
}
if ((figure = provider.getTooltip(entity)) != null) {
node.setTooltip(figure);
}
}
/**
* Returns the SWT line style for the given zest connection style.
*
*/
public static int getLineStyleForZestStyle(int style) {
int lineStyles = ZestStyles.CONNECTIONS_DASH_DOT | ZestStyles.CONNECTIONS_DASH | ZestStyles.CONNECTIONS_DOT | ZestStyles.CONNECTIONS_SOLID;
style = style & lineStyles;
if (style == 0) {
style = ZestStyles.CONNECTIONS_SOLID;
}
switch (style) {
case ZestStyles.CONNECTIONS_DASH_DOT:
return SWT.LINE_DASHDOT;
case ZestStyles.CONNECTIONS_DASH:
return SWT.LINE_DASH;
case ZestStyles.CONNECTIONS_DOT:
return SWT.LINE_DOT;
}
return SWT.LINE_SOLID;
}
}