From 1c1e17381bf011864d0f193bf8549a0869698dc6 Mon Sep 17 00:00:00 2001
From: Andrew Byrd <andrew@fastmail.net>
Date: Fri, 5 Jun 2015 21:41:03 +0200
Subject: [PATCH] move analyst cluster result envelope to OTP

---
 .../analyst/cluster/ResultEnvelope.java       | 138 ++++++++++++++++++
 1 file changed, 138 insertions(+)
 create mode 100644 src/main/java/org/opentripplanner/analyst/cluster/ResultEnvelope.java

diff --git a/src/main/java/org/opentripplanner/analyst/cluster/ResultEnvelope.java b/src/main/java/org/opentripplanner/analyst/cluster/ResultEnvelope.java
new file mode 100644
index 00000000000..1595e4fceae
--- /dev/null
+++ b/src/main/java/org/opentripplanner/analyst/cluster/ResultEnvelope.java
@@ -0,0 +1,138 @@
+package org.opentripplanner.analyst.cluster;
+
+import org.opentripplanner.analyst.ResultSet;
+
+import java.io.Serializable;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * This is a class that stores several result sets: an upper bound, a lower bound, and a central tendency.
+ * TODO consider renaming to something more specific. "Envelope" has a geometric meaning, but here we mean "the thing you put a letter in".
+ *
+ * @author mattwigway
+ */
+public class ResultEnvelope implements Serializable {
+
+	/**
+	 * The best case/upper bound (e.g. number of jobs reachable considering the best travel times seen during the
+	 * time window)
+	 */
+	public ResultSet bestCase;
+	
+	/**
+	 * The lower bound (e.g. number of jobs reachable considering the worst travel times seen during the time window)
+	 */
+	public ResultSet worstCase;
+
+	/**
+	 * The average case (e.g. the number of jobs reachable on average, departing randomly within the time window)
+	 */
+	public ResultSet avgCase;
+	
+	/**
+	 * The point estimate of the accessibility. If profile = false, this is the journey
+	 * time returned by OTP.
+	 */
+	public ResultSet pointEstimate;
+	
+	/**
+	 * The spread of results (future use). This could be the standard deviation or
+	 * interquartile range, for example. 
+	 */
+	public ResultSet spread;
+	
+	/**
+	 * Is this a profile request?
+	 * If so, upperBound and lowerBound will be defined, and pointEstimate not.
+	 * Otherwise, only pointEstimate will be defined.
+	 */
+	public boolean profile;
+
+	/** The ID of the job of which this resultenvelope is a part */
+	public String jobId;
+
+	/** The ID of the feature from whence this result envelope came */
+	public String id;
+	
+	/** The ID of the shapefile/pointset from whence this result envelope came */
+	public String destinationPointsetId;
+	
+	public ResultSet get (Which key) {
+		switch (key) {
+		case BEST_CASE:
+			return this.bestCase;
+		case WORST_CASE:
+			return this.worstCase;
+		case POINT_ESTIMATE:
+			return this.pointEstimate;
+		case SPREAD:
+			return this.spread;
+		case AVERAGE:
+			return this.avgCase;
+		default:
+			throw new IllegalStateException("Invalid result type!");
+		}
+	}
+	
+	public void put (Which key, ResultSet val) {
+		switch (key) {
+		case BEST_CASE:
+			this.bestCase = val;
+			break;
+		case WORST_CASE:
+			this.worstCase = val;
+			break;
+		case POINT_ESTIMATE:
+			this.pointEstimate = val;
+			break;
+		case SPREAD:
+			this.spread = val;
+			break;
+		case AVERAGE:
+			this.avgCase = val;
+			break;
+		}
+	}
+	
+	/**
+	 * Explode this result envelope into a result envelope for each contained attribute
+	 * We do this because we need to retrieve all of the values for a particular variable quickly, in order to display the map,
+	 * and looping over 10GB of data to do this is not tractable.
+	 */
+	public Map<String, ResultEnvelope> explode () {
+		Map<String, ResultEnvelope> exploded = new HashMap<String, ResultEnvelope>();
+		
+		// find a non-null resultset
+		for (String attr : (pointEstimate != null ? pointEstimate : avgCase).histograms.keySet()) {
+			ResultEnvelope env = new ResultEnvelope();
+			
+			for (Which which : Which.values()) {
+				ResultSet orig = this.get(which);
+				
+				if (orig != null) {
+					ResultSet rs = new ResultSet();
+					rs.id = orig.id;
+					rs.histograms.put(attr, orig.histograms.get(attr));
+					env.put(which, rs);
+					env.id = this.id;
+				}
+			}
+			
+			exploded.put(attr, env);
+		}
+		
+		return exploded;
+	}
+	
+	/**
+	 * Build an empty result envelope.
+	 */
+	public ResultEnvelope () {
+		// do nothing, restore default constructor
+	}
+	
+	public static enum Which {
+		BEST_CASE, WORST_CASE, POINT_ESTIMATE, SPREAD, AVERAGE
+	}
+}