X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2FProgress.java;h=748d4a666c377123ff2eadd29c840ec4baf402e5;hb=34718cf9dc943306279713aec140a197fb27629a;hp=6f05110e8715305b438258b45cf1674b992a4451;hpb=f4053377fa15da2f11e82955bfab86e673fa371c;p=fanfix.git

diff --git a/src/be/nikiroo/utils/Progress.java b/src/be/nikiroo/utils/Progress.java
index 6f05110..748d4a6 100644
--- a/src/be/nikiroo/utils/Progress.java
+++ b/src/be/nikiroo/utils/Progress.java
@@ -9,6 +9,16 @@ import java.util.Map.Entry;
 
 /**
  * Progress reporting system, possibly nested.
+ * <p>
+ * A {@link Progress} can have a name, and that name will be reported through
+ * the event system (it will report the first non-null name in the stack from
+ * the {@link Progress} from which the event originated to the parent the event
+ * is listened on).
+ * <p>
+ * The {@link Progress} also has a table of keys/values shared amongst all the
+ * hierarchy (note that when adding a {@link Progress} to others, its values
+ * will be prioritized if some with the same keys were already present in the
+ * hierarchy).
  * 
  * @author niki
  */
@@ -35,6 +45,7 @@ public class Progress {
 		public void progress(Progress progress, String name);
 	}
 
+	private Map<Object, Object> map = new HashMap<Object, Object>();
 	private Progress parent = null;
 	private Object lock = new Object();
 	private String name;
@@ -138,7 +149,7 @@ public class Progress {
 			throw new RuntimeException("negative values not supported");
 		}
 
-		synchronized (getLock()) {
+		synchronized (lock) {
 			if (min > max) {
 				throw new RuntimeException(
 						"The minimum progress value must be <= the maximum progress value");
@@ -168,7 +179,7 @@ public class Progress {
 	 *             if max &lt; min
 	 */
 	public void setMax(int max) {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			if (max < min) {
 				throw new Error(
 						"The maximum progress value must be >= the minimum progress value");
@@ -199,7 +210,7 @@ public class Progress {
 					"The minimum progress value must be <= the maximum progress value");
 		}
 
-		synchronized (getLock()) {
+		synchronized (lock) {
 			this.min = min;
 			this.max = max;
 		}
@@ -225,7 +236,7 @@ public class Progress {
 	 *            the progress to set
 	 */
 	public void setProgress(int progress) {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			double childrenProgress = relativeProgress - relativeLocalProgress;
 
 			relativeLocalProgress = ((double) progress) / (max - min);
@@ -260,7 +271,7 @@ public class Progress {
 	 */
 	private void setRelativeProgress(Progress pg, String name,
 			double relativeProgress) {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			relativeProgress = Math.max(0, relativeProgress);
 			relativeProgress = Math.min(1, relativeProgress);
 			this.relativeProgress = relativeProgress;
@@ -286,7 +297,7 @@ public class Progress {
 	 *            the amount to add
 	 */
 	public void add(int step) {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			setProgress(getLocalProgress() + step);
 		}
 	}
@@ -305,7 +316,7 @@ public class Progress {
 	 * Mark the {@link Progress} as done by setting its value to max.
 	 */
 	public void done() {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			double childrenProgress = relativeProgress - relativeLocalProgress;
 			relativeLocalProgress = 1 - childrenProgress;
 			setRelativeProgress(this, name, 1d);
@@ -318,7 +329,7 @@ public class Progress {
 	 * @return the children (Who will think of the children??)
 	 */
 	public List<Progress> getChildren() {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			return new ArrayList<Progress>(children.keySet());
 		}
 	}
@@ -327,7 +338,8 @@ public class Progress {
 	 * Notify the listeners that this {@link Progress} changed value.
 	 * 
 	 * @param pg
-	 *            the emmiter
+	 *            the emmiter, that is, the (sub-){link Progress} that just
+	 *            reported some change, not always the same as <tt>this</tt>
 	 * @param name
 	 *            the current name (if it is NULL, the first non-null name in
 	 *            the hierarchy will overwrite it) of the {@link Progress} who
@@ -342,7 +354,7 @@ public class Progress {
 			name = this.name;
 		}
 
-		synchronized (getLock()) {
+		synchronized (lock) {
 			for (ProgressListener l : listeners) {
 				l.progress(pg, name);
 			}
@@ -360,7 +372,7 @@ public class Progress {
 	 *            the listener
 	 */
 	public void addProgressListener(ProgressListener l) {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			this.listeners.add(l);
 		}
 	}
@@ -374,7 +386,7 @@ public class Progress {
 	 * @return TRUE if it was found (and removed)
 	 */
 	public boolean removeProgressListener(ProgressListener l) {
-		synchronized (getLock()) {
+		synchronized (lock) {
 			return this.listeners.remove(l);
 		}
 	}
@@ -397,7 +409,7 @@ public class Progress {
 		if (weight < min || weight > max) {
 			throw new RuntimeException(String.format(
 					"Progress object %s cannot have a weight of %f, "
-							+ "it is outside of its parent (%s) range (%f)",
+							+ "it is outside of its parent (%s) range (%d)",
 					progress.name, weight, name, max));
 		}
 
@@ -408,41 +420,76 @@ public class Progress {
 					name, progress.parent.name));
 		}
 
-		progress.addProgressListener(new ProgressListener() {
+		ProgressListener progressListener = new ProgressListener() {
 			@Override
 			public void progress(Progress pg, String name) {
-				synchronized (getLock()) {
+				synchronized (lock) {
 					double total = relativeLocalProgress;
-					synchronized (getLock()) {
-						for (Entry<Progress, Double> entry : children
-								.entrySet()) {
-							total += (entry.getValue() / (max - min))
-									* entry.getKey().getRelativeProgress();
-						}
+					for (Entry<Progress, Double> entry : children.entrySet()) {
+						total += (entry.getValue() / (max - min))
+								* entry.getKey().getRelativeProgress();
 					}
 
 					setRelativeProgress(pg, name, total);
 				}
 			}
-		});
+		};
 
-		synchronized (getLock()) {
+		synchronized (lock) {
+			// Should not happen but just in case
+			if (this.map != progress.map) {
+				this.map.putAll(progress.map);
+			}
+			progress.map = this.map;
+			progress.parent = this;
 			this.children.put(progress, weight);
+			progress.addProgressListener(progressListener);
 		}
 	}
 
 	/**
-	 * The lock object to use (this one or the recursively-parent one).
+	 * Set the given value for the given key on this {@link Progress} and it's
+	 * children.
 	 * 
-	 * @return the lock object to use
+	 * @param key
+	 *            the key
+	 * @param value
+	 *            the value
 	 */
-	private Object getLock() {
-		synchronized (lock) {
-			if (parent != null) {
-				return parent.getLock();
-			}
+	public void put(Object key, Object value) {
+		map.put(key, value);
+	}
 
-			return lock;
+	/**
+	 * Return the value associated with this key as a {@link String} if any,
+	 * NULL if not.
+	 * <p>
+	 * If the value is not NULL but not a {@link String}, it will be converted
+	 * via {@link Object#toString()}.
+	 * 
+	 * @param key
+	 *            the key to check
+	 * 
+	 * @return the value or NULL
+	 */
+	public String getString(Object key) {
+		Object value = map.get(key);
+		if (value == null) {
+			return null;
 		}
+
+		return value.toString();
+	}
+
+	/**
+	 * Return the value associated with this key if any, NULL if not.
+	 * 
+	 * @param key
+	 *            the key to check
+	 * 
+	 * @return the value or NULL
+	 */
+	public Object get(Object key) {
+		return map.get(key);
 	}
 }