Version 1.3.5: Progress/ProgressBar fixes (woopsie)

[nikiroo-utils.git] / src / be / nikiroo / utils / ui / ProgressBar.java

package be.nikiroo.utils.ui;

import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.swing.JPanel;
import javax.swing.JProgressBar;
import javax.swing.SwingUtilities;

import be.nikiroo.utils.Progress;

/**
 * A graphical control to show the progress of a {@link Progress}.
 * <p>
 * This control is <b>NOT</b> thread-safe.
 * 
 * @author niki
 */
public class ProgressBar extends JPanel {
	private static final long serialVersionUID = 1L;

	private Map<Progress, JProgressBar> bars;
	private List<ActionListener> actionListeners;
	private List<ActionListener> updateListeners;
	private Progress pg;

	public ProgressBar() {
		bars = new HashMap<Progress, JProgressBar>();
		actionListeners = new ArrayList<ActionListener>();
		updateListeners = new ArrayList<ActionListener>();
	}

	public void setProgress(final Progress pg) {
		this.pg = pg;

		SwingUtilities.invokeLater(new Runnable() {
			public void run() {
				if (pg != null) {
					final JProgressBar bar = new JProgressBar();
					bar.setStringPainted(true);

					bars.clear();
					bars.put(pg, bar);

					bar.setMinimum(pg.getMin());
					bar.setMaximum(pg.getMax());
					bar.setValue(pg.getProgress());
					bar.setString(pg.getName());

					pg.addProgressListener(new Progress.ProgressListener() {
						public void progress(Progress progress, String name) {
							final Progress.ProgressListener l = this;
							SwingUtilities.invokeLater(new Runnable() {
								public void run() {
									Map<Progress, JProgressBar> newBars = new HashMap<Progress, JProgressBar>();
									newBars.put(pg, bar);

									bar.setMinimum(pg.getMin());
									bar.setMaximum(pg.getMax());
									bar.setValue(pg.getProgress());
									bar.setString(pg.getName());

									for (Progress pg : getChildrenAsOrderedList(pg)) {
										JProgressBar bar = bars.get(pg);
										if (bar == null) {
											bar = new JProgressBar();
											bar.setStringPainted(true);
										}

										newBars.put(pg, bar);

										bar.setMinimum(pg.getMin());
										bar.setMaximum(pg.getMax());
										bar.setValue(pg.getProgress());
										bar.setString(pg.getName());
									}

									if (ProgressBar.this.pg == null) {
										bars.clear();
									} else {
										bars = newBars;

										if (pg.isDone()) {
											pg.removeProgressListener(l);
											for (ActionListener listener : actionListeners) {
												listener.actionPerformed(new ActionEvent(
														ProgressBar.this, 0,
														"done"));
											}
										}

										update();
									}
								}
							});
						}
					});
				}

				update();
			}
		});
	}

	public void addActionListener(ActionListener l) {
		actionListeners.add(l);
	}

	public void clearActionListeners() {
		actionListeners.clear();
	}

	public void addUpdateListener(ActionListener l) {
		updateListeners.add(l);
	}

	public void clearUpdateListeners() {
		updateListeners.clear();
	}

	public int getProgress() {
		if (pg == null) {
			return 0;
		}

		return pg.getProgress();
	}

	// only named ones
	private List<Progress> getChildrenAsOrderedList(Progress pg) {
		List<Progress> children = new ArrayList<Progress>();
		for (Progress child : pg.getChildren()) {
			if (child.getName() != null && !child.getName().isEmpty()) {
				children.add(child);
			}
			children.addAll(getChildrenAsOrderedList(child));
		}

		return children;
	}

	private void update() {
		invalidate();
		removeAll();

		if (pg != null) {
			setLayout(new GridLayout(bars.size(), 1));
			add(bars.get(pg), 0);
			for (Progress child : getChildrenAsOrderedList(pg)) {
				JProgressBar jbar = bars.get(child);
				if (jbar != null) {
					add(jbar);
				}
			}
		}

		validate();
		repaint();

		for (ActionListener listener : updateListeners) {
			listener.actionPerformed(new ActionEvent(this, 0, "update"));
		}
	}
}