ProductSystemFigure.java

package org.openlca.app.results.analysis.sankey.model;

import org.eclipse.draw2d.ColorConstants;
import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.Graphics;
import org.eclipse.draw2d.LineBorder;
import org.eclipse.draw2d.MouseEvent;
import org.eclipse.draw2d.MouseListener;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Font;
import org.eclipse.swt.graphics.FontData;
import org.eclipse.swt.graphics.RGB;
import org.eclipse.swt.widgets.Display;
import org.openlca.app.FaviColor;
import org.openlca.app.M;
import org.openlca.app.results.analysis.sankey.actions.SankeySelectionAction;
import org.openlca.app.results.analysis.sankey.layout.GraphLayoutManager;
import org.openlca.app.util.Colors;
import org.openlca.app.util.CostResultDescriptor;
import org.openlca.app.util.Numbers;
import org.openlca.core.model.descriptors.FlowDescriptor;
import org.openlca.core.model.descriptors.ImpactCategoryDescriptor;

public class ProductSystemFigure extends Figure {

	private boolean firstTime = true;
	private ProductSystemNode productSystemNode;
	private Font infoFont;

	public ProductSystemFigure(ProductSystemNode node) {
		setForegroundColor(ColorConstants.black);
		setBorder(new LineBorder(1));
		productSystemNode = node;
		addMouseListener(new SelectionChange());
	}

	Font getInfoFont() {
		return infoFont;
	}

	@Override
	public void paint(Graphics graphics) {
		if (firstTime)
			firePropertyChange("firstTimeInitialized", null, "not null");
		super.paint(graphics);
		if (firstTime) {
			((GraphLayoutManager) getLayoutManager()).layoutTree();
			firstTime = false;
		}
		paintInfoBox(graphics);
	}

	private void paintInfoBox(Graphics graphics) {
		graphics.pushState();
		Font normalFont = graphics.getFont();
		Font infoFont = getInfoFont(normalFont);
		graphics.setFont(infoFont);

		Object selection = productSystemNode.getSelection();
		double cutoffValue = productSystemNode.getCutoff() * 100;
		String cutoffText = M.Cutoff + ": "
				+ Numbers.format(cutoffValue, 3) + "%";
		if (selection != null) {
			graphics.drawText(M.ProductSystem + ": "
					+ productSystemNode.getProductSystem().getName(),
					new Point(5, 5));
			String label = selectionLabel(selection);
			graphics.drawText(label, new Point(5, 30));
			graphics.drawText(cutoffText, new Point(5, 60));

		} else {
			graphics.drawText(M.NoAnalysisOptionsSet, new Point(5, 5));
			graphics.drawText(M.ClickHereToChangeDisplay, new Point(5,
					30));
		}

		graphics.setFont(normalFont);

		drawColorScale(graphics);
		graphics.popState();
	}

	private Font getInfoFont(Font normalFont) {
		if (infoFont != null)
			return infoFont;
		infoFont = new Font(Display.getCurrent(),
				new FontData[] { new FontData(
						normalFont.getFontData()[0].getName(), 16, SWT.NONE) });
		return infoFont;
	}

	private String selectionLabel(Object selection) {
		if (selection instanceof FlowDescriptor) {
			FlowDescriptor flow = (FlowDescriptor) selection;
			return M.Flow + ": " + flow.getName();
		}
		if (selection instanceof ImpactCategoryDescriptor) {
			ImpactCategoryDescriptor impact = (ImpactCategoryDescriptor) selection;
			return M.ImpactCategory + ": " + impact.getName();
		}
		if (selection instanceof CostResultDescriptor) {
			CostResultDescriptor cost = (CostResultDescriptor) selection;
			String m = M.CostResult;
			return m + ": " + cost.getName();
		}
		return M.NoAnalysisOptionsSet;
	}

	private void drawColorScale(Graphics graphics) {
		int x = 25;
		int y = 140;
		for (int i = -100; i < 100; i += 2) {
			RGB rgb = FaviColor.getForContribution(i / 100d);
			graphics.setBackgroundColor(Colors.get(rgb));
			int posX = x + 3 * ((100 + i) / 2);
			graphics.fillRectangle(new Rectangle(posX, y, 4, 20));
		}

		// draw percentage texts
		graphics.drawText(M.Sankey_ScaleDescription, x + 35, y + 22);
		graphics.drawLine(x, y, x + 300, y);
		for (int i = 0; i <= 20; i++) {
			int height = 0;
			if (i == 0 || i == 20) {
				height = 19;
			} else if (i % 2 == 0) {
				height = 12;
			} else {
				height = 6;
			}
			graphics.drawLine(x + 15 * i, y, x + 15 * i, y + height);
			String percentage = "" + (i * 10 - 100);
			if (i % 2 == 0) {
				int left = 5 * (percentage.length() - 1);
				if (i == 10) {
					left += 2;
				} else if (i == 0 || i == 20) {
					left -= 1;
				}
				graphics.drawText(percentage, x + 15 * i - left, y - 16);
			}
		}
	}

	private class SelectionChange implements MouseListener {

		@Override
		public void mouseReleased(MouseEvent evt) {
		}

		@Override
		public void mouseDoubleClicked(MouseEvent evt) {
		}

		@Override
		public void mousePressed(MouseEvent arg0) {
			int x = arg0.getLocation().x;
			int y = arg0.getLocation().y;
			if (in(x, 350) && in(y, 120)) {
				SankeySelectionAction psa = new SankeySelectionAction();
				psa.setSankeyDiagram(productSystemNode.getEditor());
				psa.run();
			}
		}

		private boolean in(int val, int max) {
			return val >= 0 && val <= max;
		}
	}

}