/*
    * This file is a part of CAST project.
    * (c) Copyright 2007, AGH University of Science & Technology
    * https://caribou.iisg.agh.edu.pl/trac/cast
    *
    * Licensed under the Eclipse Public License, Version 1.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.eclipse.org/legal/epl-v10.html
   */
  /*
   * File: LegendFigure.java
   * Created: 2007-00-00
   * Author: cast
   * $Id: LegendFigure.java 3054 2009-07-27 10:23:07Z kpietak $
   */
  
  package pl.edu.agh.cast.backward.figure;
  
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.eclipse.draw2d.ColorConstants;
  import org.eclipse.draw2d.Figure;
  import org.eclipse.draw2d.FrameBorder;
  import org.eclipse.draw2d.Label;
  import org.eclipse.draw2d.MarginBorder;
  import org.eclipse.draw2d.RectangleFigure;
  import org.eclipse.draw2d.ToolbarLayout;
  import org.eclipse.swt.graphics.Color;
  
  import pl.edu.agh.cast.common.HexColor;
  import pl.edu.agh.cast.editpart.LegendEditPart;
  import pl.edu.agh.cast.model.visual.backward.ConnectionGroup;
  import pl.edu.agh.cast.model.visual.backward.Diagram;
  import pl.edu.agh.cast.model.visual.backward.Legend;
  import pl.edu.agh.cast.model.visual.backward.Node;
  import pl.edu.agh.cast.util.Images;
  
  /**
   * A figure for displaying diagram {@link Legend}.
   *
   * @author AGH CAST Team
   */
  public class LegendFigure extends Figure {
  
  	/**
  	 * The figure {@link org.eclipse.gef.EditPart}.
  	 */
  	protected LegendEditPart controller;
  
  	/**
  	 * The legend to be shown on the figure.
  	 */
  	protected Legend legend;
  
  	private FrameBorder legendBorder;
  
  	/**
  	 * Maps {@link Node#getImageId()} to {@link Figure}.
  	 */
  	private Map<String, Figure> imageFigures;
  
  	/**
  	 * Maps {@link ConnectionGroup#LINE_COLOR} attribute value to {@link Figure}.
  	 */
  	private Map<HexColor, Figure> colorFigures;
  
  	/**
  	 * Creates new legend figure.
  	 *
  	 * @param legend
  	 *            legend to display
  	 * @param schemaLegendEditPart
  	 *            legend {@link org.eclipse.gef.EditPart}
  	 */
  	public LegendFigure(Legend legend, LegendEditPart schemaLegendEditPart) {
  		super();
  		this.legend = legend;
  		this.controller = schemaLegendEditPart;
  
  		init();
  	}
  
  	/**
  	 * Initializes the legend figure.
  	 */
  	public void init() {
  		imageFigures = new HashMap<String, Figure>();
  		colorFigures = new HashMap<HexColor, Figure>();
  		this.removeAll();
  		ToolbarLayout layout = new ToolbarLayout();
  		layout.setStretchMinorAxis(false);
  		layout.setMinorAlignment(ToolbarLayout.ALIGN_TOPLEFT);
  		setLayoutManager(layout);
  
 		// get all icons from actually presented nodes
 		Diagram model = (Diagram)(controller.getParent().getModel());
 		List<String> imageIds = new ArrayList<String>();
 		List<HexColor> colorIds = new ArrayList<HexColor>();
 		for (Node node : model.getNodes()) {
 			if (!imageIds.contains(node.getImageId())) {
 				imageIds.add(node.getImageId());
 			}
 			for (ConnectionGroup connectionGroup : node.getConnectionGroups()) {
 				HexColor lineColor = (HexColor)connectionGroup.getAttributeValue(ConnectionGroup.LINE_COLOR).getValue();
 				if (!colorIds.contains(lineColor)) {
 					colorIds.add(lineColor);
 				}
 			}
 		}
 
 		setIconsLegend(imageIds);
 		setColorsLegend(colorIds);
 
 		legendBorder = new FrameBorder(legend.getDiagramName());
 		this.setBackgroundColor(ColorConstants.listBackground);
 		this.setOpaque(true);
 		this.setBorder(legendBorder);
 	}
 
 	private void setColorsLegend(List<HexColor> colorIds) {
 		for (HexColor colorId : colorIds) {
 			addColoredLineFigure(colorId);
 		}
 
 	}
 
 	private void addColoredLineFigure(HexColor colorId) {
 		int lineWidth = 40;
 		int lineThickness = 3;
 		int marginSize = 5;
 
 		Figure labeledColoredLine = new Figure();
 		ToolbarLayout imageLayout = new ToolbarLayout(true);
 		imageLayout.setStretchMinorAxis(false);
 		imageLayout.setMinorAlignment(ToolbarLayout.ALIGN_CENTER);
 		labeledColoredLine.setLayoutManager(imageLayout);
 		RectangleFigure colorLineFigure = new RectangleFigure();
 		colorLineFigure.setSize(lineWidth, lineThickness);
 		// colorLineFigure.setFill(true);
 		colorLineFigure.setForegroundColor(new Color(null, colorId.getRed(), colorId.getGreen(), colorId.getBlue()));
 		colorLineFigure.setBackgroundColor(new Color(null, colorId.getRed(), colorId.getGreen(), colorId.getBlue()));
 
 		RectangleFigure marginFigure = new RectangleFigure();
 		marginFigure.setSize(marginSize, lineThickness);
 		marginFigure.setForegroundColor(ColorConstants.listBackground);
 		marginFigure.setBackgroundColor(ColorConstants.listBackground);
 
 		Label imageLabel = null;
 		imageLabel = new Label(legend.getLineColorLabel(colorId));
 		imageLabel.setOpaque(true);
 		// coloredLine.setOpaque(true);
 		labeledColoredLine.add(colorLineFigure);
 		labeledColoredLine.add(marginFigure);
 
 		labeledColoredLine.add(imageLabel);
 		labeledColoredLine.setOpaque(true);
 		MarginBorder colorLineBorder = new MarginBorder(marginSize);
 		labeledColoredLine.setBorder(colorLineBorder);
 
 		Figure oldColor = colorFigures.get(colorId);
 		if (oldColor != null) {
 			this.remove(oldColor);
 			colorFigures.remove(oldColor);
 		}
 		add(labeledColoredLine);
 		colorFigures.put(colorId, labeledColoredLine);
 	}
 
 	private void setIconsLegend(List<String> imageIds) {
 
 		if (imageIds.contains(null)) {
 			imageIds.remove(null);
 			if (!imageIds.contains(Images.ENTITY_BIG)) {
 				imageIds.add(Images.ENTITY_BIG);
 			}
 		}
 
 		Collections.sort(imageIds);
 
 		for (String imageId : imageIds) {
 			addImageFigure(imageId);
 		}
 
 	}
 
 	private void addImageFigure(String imageId) {
 		Figure labeledImageFigure = new Figure();
 		ToolbarLayout imageLayout = new ToolbarLayout(true);
 		imageLayout.setStretchMinorAxis(false);
 		imageLayout.setMinorAlignment(ToolbarLayout.ALIGN_CENTER);
 		labeledImageFigure.setLayoutManager(imageLayout);
 		ImageFigure icon = new ImageFigure(imageId);
 		Label imageLabel = null;
 		if (imageId != null) {
 			imageLabel = new Label(legend.getImageLabel(imageId));
 		}
 		if (imageLabel != null) {
 			imageLabel.setOpaque(true);
 			icon.setOpaque(true);
 			labeledImageFigure.add(icon);
 			labeledImageFigure.add(imageLabel);
 			labeledImageFigure.setOpaque(true);
 
 			Figure oldImage = imageFigures.get(imageId);
 			if (oldImage != null) {
 				this.remove(oldImage);
 				imageFigures.remove(oldImage);
 			}
 			add(labeledImageFigure);
 			imageFigures.put(imageId, labeledImageFigure);
 		}
 	}
 
 	public LegendEditPart getController() {
 		return controller;
 	}
 }