/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package ui;

import java.awt.*;
import java.awt.geom.RoundRectangle2D;
import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.Map;
import javax.swing.JFrame;
import javax.swing.JPanel;

/**
 *
 * @author Administrator
 */
public class BarDrawer extends JPanel {

    private int colorIndex;
    private Color[] colors = {
	Color.decode("#DE683A"),
	Color.decode("#DEB53A"),
	Color.decode("#B2DE3A"),
	Color.decode("#3ADEAD"),
	Color.decode("#3A65DE"),
	Color.decode("#973ADE"),
	Color.decode("#DE3A3A")
    };
    public LinkedHashMap<FileCollectionPanel, Double> collectionMap;

    public BarDrawer(LinkedHashMap<FileCollectionPanel, Double> collectionMap) throws IOException {
	this.collectionMap = collectionMap;

	this.collectionMap = new LinkedHashMap<FileCollectionPanel, Double>();
//	collectionMap.put("Pictures", 13d);
//	collectionMap.put("Music", 12d);
//	collectionMap.put("Videos", 13d);
//	collectionMap.put("TXT Files", 12d);
//	collectionMap.put("EXE Files", 16d);
//	collectionMap.put("Compressed Files", 10d);
//	collectionMap.put("Downloads Folder", 1d);
//	collectionMap.put("Another collection", 22d);


//	collectionMap.put("Pictures", 66d);
//	collectionMap.put("Music", 1d);
//	collectionMap.put("Videos", 33d);


	FileCollectionPanel fcp1 = new FileCollectionPanel();
	fcp1.setTitle("Pictures");
	FileCollectionPanel fcp2 = new FileCollectionPanel();
	fcp2.setTitle("Music");
	FileCollectionPanel fcp3 = new FileCollectionPanel();
	fcp3.setTitle("Videos");
//
	this.collectionMap.put(fcp1, 66d);
	this.collectionMap.put(fcp2, 11d);
	this.collectionMap.put(fcp3, 23d);
    }

    @Override
    protected void paintComponent(Graphics gr) {
	super.paintComponent(gr);
	Graphics2D g = (Graphics2D) gr;
	setPreferredSize(new Dimension(getWidth(), 150));

	Map desktopHints = (Map) (Toolkit.getDefaultToolkit().getDesktopProperty("awt.font.desktophints"));
	if (desktopHints != null) {
	    g.addRenderingHints(desktopHints);
	}

	g.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
		RenderingHints.VALUE_ANTIALIAS_ON);

	colorIndex = 0;
	int realWidth = getWidth() - 1;
	int heightOfBar = (int) (getHeight() - 50);

//	g.setColor(Color.decode("#cccccc"));
	double radius = getHeight() / 3;
	RoundRectangle2D r = new RoundRectangle2D.Double(0, 0, realWidth, heightOfBar, radius, radius * 2);

	g.setClip(getBounds());
	g.setColor(Color.black);
	double barSegment = realWidth / collectionMap.size();
	if (barSegment > 120) {
	    barSegment = 120;
	}
	int collectionNameX = 0;

	int x = 0;
	for (FileCollectionPanel fcp : collectionMap.keySet()) {

	    g.setColor(getNextColor());

	    double percentage = collectionMap.get(fcp) + adjustmentNeeded(collectionMap, fcp);
	    double width = (((double) percentage / 100) * realWidth);

	    g.setClip(r);
	    g.fillRect(x, 0, (int) width, heightOfBar);

	    g.setFont(g.getFont().deriveFont(14f));
	    String titleShortened = fcp.getTitle();
	    int lettersToRemove = 1;
	    while (g.getFontMetrics().stringWidth(titleShortened) > barSegment - 15) {
		titleShortened = shortenString(fcp.getTitle(), lettersToRemove);
		lettersToRemove++;
		titleShortened += "...";
	    }

	    g.setClip(getBounds());
	    g.setClip(0, 0, getWidth(), getHeight());
	    g.fillRect(collectionNameX + 15, heightOfBar + 10, 10, 10);
	    g.setColor(Color.black);
	    g.drawString(titleShortened, collectionNameX + 30, heightOfBar + 20);
	    System.out.println("heightOfBar = " + heightOfBar);
	    System.out.println("collectionNameX = " + collectionNameX);
	    g.setFont(g.getFont().deriveFont(11f));
	    g.drawString(Double.toString(collectionMap.get(fcp)) + "%", collectionNameX + 30, heightOfBar + 20 + g.getFontMetrics().getHeight());

	    collectionNameX += barSegment;
	    x = x + (int) width;
	}

	// draw lines
	g.setClip(r);
	for (int lineX = 15; lineX < realWidth - 1; lineX += 25) {
	    g.setColor(new Color(255, 255, 255, 75));
	    g.drawLine(lineX, 0, lineX - 15, heightOfBar - 1);

	    int gray = 128;
	    Color grayTransparent = new Color(gray, gray, gray, 75);
	    g.setColor(grayTransparent);
	    g.drawLine(lineX - 1, 0, lineX - 16, heightOfBar - 1);
	}

	// adding gradient
	Paint p = new GradientPaint(0.0f, 0.0f, new Color(255, 255, 255, 255), 0.0f, getHeight(), new Color(255, 255, 255, 0));

	Paint p2 = new LinearGradientPaint(0.0f, 0.0f, getWidth(), getHeight(), new float[]{0.5f, 1.0f}, new Color[]{new Color(255, 255, 255, 255), new Color(255, 255, 255, 0)});
	g.setPaint(p);
    }

    public static String shortenString(String s, int lettersToRemove) {
	if (lettersToRemove > s.length()) {
	    return Character.toString(s.charAt(0));
	}
	char[] cArr = new char[s.length() - lettersToRemove];
	for (int i = 0; i < cArr.length; i++) {
	    cArr[i] = s.charAt(i);
	}

	return new String(cArr);
    }

    /**
     * Checks for two things. First, it checks that the entry being dealt with
     * is the last entry in a {@link LinkedHashMap LinkedHashMap}. Second, if it
     * is, then the total of the all the values is counted. If the total is less
     * than 100, then the result of this method will be adjustment needed to get
     * to 100. E.g. if the total of all the doubles in the map is 98, and the
     * entry passed is the last one, then this method will return 2. Else, it
     * will return 0.
     *
     * @param map
     * @param value
     * @return
     */
    public double adjustmentNeeded(LinkedHashMap<FileCollectionPanel, Double> map, FileCollectionPanel fcp) {
	double minVal = Double.MAX_VALUE;
	FileCollectionPanel smallestFcp = null;
	for (FileCollectionPanel currentCollection : map.keySet()) {
	    double val = map.get(currentCollection);
	    if (minVal > val) {
		minVal = val;
		smallestFcp = currentCollection;
	    }
	}

	double total = 0;
	for (Double val : map.values()) {
	    total += val;
	}

	if ((total < 100d) && (smallestFcp.equals(fcp))) {
	    return 100d - total;
	}

	return 0;
    }

    public Color getSelectedColor() {
	return colors[colorIndex];
    }

    public Color getNextColor() {
	if (colorIndex < 0 || colorIndex >= colors.length - 1) {
	    colorIndex = 0;
	} else {
	    colorIndex++;
	}

	return colors[colorIndex];
    }

    public static void main(String[] args) throws IOException {
	JFrame frame = new JFrame();

	frame.add(new BarDrawer(null));
	frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
	frame.setSize(600, 150);
	frame.setVisible(true);
    }
}

class FileCollectionPanel extends JPanel {

    private String title;

    public FileCollectionPanel() {
    }

    /**
     * @return the title
     */
    public String getTitle() {
	return title;
    }

    /**
     * @param title the title to set
     */
    public void setTitle(String title) {
	this.title = title;
    }
}