Keyword usage can now be used for statistics

git-svn-id: svn://svn.code.sf.net/p/magicseteditor/code/trunk@517 0fc631ac-6414-0410-93d0-97cfa31319b6

src/gui/control/graph.cpp

@@ -37,6 +37,24 @@ GraphElement::GraphElement(const String& v1, const String& v2) {
 	values.push_back(v2);
 }
 
+void GraphDataPre::splitList(size_t axis) {
+	size_t count = elements.size(); // only the elements that were already there
+	for (size_t i = 0 ; i < count ; ++i) {
+		GraphElement& e = *elements[i];
+		String& v = e.values[axis];
+		size_t comma = v.find_first_of(_(','));
+		while (comma != String::npos) {
+			// split
+			GraphElementP e2(new GraphElement(e));
+			e2->values[axis] = v.substr(0,comma);
+			elements.push_back(e2);
+			if (is_substr(v, comma, _(", "))) ++comma; // skip space after it
+			v = v.substr(comma + 1);
+			comma = v.find_first_of(_(','));
+		}
+	}
+}
+
 
 GraphData::GraphData(const GraphDataPre& d)
 	: axes(d.axes)
@@ -50,13 +68,12 @@ GraphData::GraphData(const GraphDataPre& d)
 		FOR_EACH_CONST(e, d.elements) {
 			counts[e->values[i]] += 1;
 		}
-		// TODO: allow some ordering in the groups
 		if (a->numeric) {
 			// TODO: start at something other than 0?
 			// TODO: support fractions?
 			size_t left = counts.size();
 			int i = 0;
-			while (left) {
+			while (!counts.empty() && i < 100) {
 				String is = String() << i++;
 				map<String,UInt>::const_iterator it = counts.find(is);
 				if (it == counts.end()) {
@@ -66,16 +83,19 @@ GraphData::GraphData(const GraphDataPre& d)
 					a->groups.push_back(GraphGroup(is, it->second));
 					a->max = max(a->max, it->second);
 					a->total += it->second;
+					counts.erase(is);
 					left--;
 				}
-				if (i > 100) {
-					// prevent infinite loops if there are non-numeric entries
-					// drop empty tail
-					while (a->groups.size() > 1 && a->groups.back().size == 0) {
-						a->groups.pop_back();
-					}
-					break;
-				}
+			}
+			// Also keep non-numeric entries
+			FOR_EACH(c, counts) {
+				a->groups.push_back(GraphGroup(c.first, c.second));
+				a->max = max(a->max, c.second);
+				a->total += c.second;
+			}
+			// drop empty tail
+			while (a->groups.size() > 1 && a->groups.back().size == 0) {
+				a->groups.pop_back();
 			}
 		} else if (a->order) {
 			// specific group order

src/gui/control/graph.hpp

@@ -87,6 +87,8 @@ class GraphDataPre {
   public:
 	vector<GraphAxisP>    axes;
 	vector<GraphElementP> elements;
+	/// Split compound elements, "a,b,c" -> "a" and "b" and "c"
+	void splitList(size_t axis);
 };
 
 /// Data to be displayed in a graph

src/gui/set/stats_panel.cpp

@@ -147,6 +147,8 @@ void StatsPanel::onCommand(int id) {
 
 // ----------------------------------------------------------------------------- : Filtering card list
 
+bool chosen(const String& choice, const String& input);
+
 class StatsFilter : public CardListFilter {
   public:
 	StatsFilter(Set& set)
@@ -155,7 +157,13 @@ class StatsFilter : public CardListFilter {
 	virtual bool keep(const CardP& card) {
 		Context& ctx = set.getContext(card);
 		FOR_EACH(v, values) {
-			if (v.first->script.invoke(ctx)->toString() != v.second) return false;
+			StatsDimension& dim = *v.first;
+			String value = dim.script.invoke(ctx)->toString();
+			if (dim.split_list) {
+				if (!chosen(v.second, value)) return false;
+			} else {
+				if (value != v.second) return false;
+			}
 		}
 		return true;
 	}
@@ -208,6 +216,12 @@ void StatsPanel::showCategory() {
 				d.elements.push_back(e);
 			}
 		}
+		// split lists
+		size_t dim_id = 0;
+		FOR_EACH(dim, cat.dimensions) {
+			if (dim->split_list) d.splitList(dim_id);
+			++dim_id;
+		}
 		graph->setLayout(cat.type);
 		graph->setData(d);
 		filterCards();