RandomSec/src/main/java/com/metaweb/gridlock/browsing/facets/ListFacet.java

package com.metaweb.gridlock.browsing.facets;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import com.metaweb.gridlock.browsing.FilteredRows;
import com.metaweb.gridlock.browsing.accessors.DecoratedValue;
import com.metaweb.gridlock.browsing.filters.ExpressionEqualRowFilter;
import com.metaweb.gridlock.browsing.filters.RowFilter;
import com.metaweb.gridlock.expr.Evaluable;
import com.metaweb.gridlock.expr.Parser;
import com.metaweb.gridlock.model.Project;

public class ListFacet implements Facet {
	protected List<NominalFacetChoice> _selection = new LinkedList<NominalFacetChoice>();
	protected List<NominalFacetChoice> _choices = new LinkedList<NominalFacetChoice>();
	
	protected String 	_name;
	protected String 	_expression;
	protected int		_cellIndex;
	protected Evaluable _eval;
	
	public ListFacet() {
	}

	@Override
	public JSONObject getJSON(Properties options) throws JSONException {
		JSONObject o = new JSONObject();
		
		o.put("name", _name);
		o.put("expression", _expression);
		o.put("cellIndex", _cellIndex);
		
		List<JSONObject> a = new ArrayList<JSONObject>(_choices.size());
		for (NominalFacetChoice choice : _choices) {
			a.add(choice.getJSON(options));
		}
		o.put("choices", a);
		
		return o;
	}

	@Override
	public void initializeFromJSON(JSONObject o) throws Exception {
		_name = o.getString("name");
		_expression = o.getString("expression");
		_cellIndex = o.getInt("cellIndex");
		
		_eval = new Parser(_expression).getExpression();
		_selection.clear();
		
		JSONArray a = o.getJSONArray("selection");
		int length = a.length();
		
		for (int i = 0; i < length; i++) {
			JSONObject oc = a.getJSONObject(i);
			JSONObject ocv = oc.getJSONObject("v");
			
			DecoratedValue decoratedValue = new DecoratedValue(
				ocv.get("v"), ocv.getString("l"));
			
			NominalFacetChoice nominalFacetChoice = new NominalFacetChoice(decoratedValue);
			
			nominalFacetChoice.count = oc.getInt("c");
			nominalFacetChoice.selected = oc.getBoolean("s");
			
			_selection.add(nominalFacetChoice);
		}
	}

	@Override
	public RowFilter getRowFilter() {
		return _selection.size() == 0 ? null :
			new ExpressionEqualRowFilter(_eval, _cellIndex, createMatches());
	}

	@Override
	public void computeChoices(Project project, FilteredRows filteredRows) {
		ExpressionNominalRowGrouper grouper = 
			new ExpressionNominalRowGrouper(_eval, _cellIndex);
		
		filteredRows.accept(project, grouper);
		
		_choices.clear();
		_choices.addAll(grouper.choices.values());
		
		for (NominalFacetChoice choice : _selection) {
			if (grouper.choices.containsKey(choice.decoratedValue.value)) {
				grouper.choices.get(choice.decoratedValue.value).selected = true;
			} else {
				choice.count = 0;
				_choices.add(choice);
			}
		}
	}
	
	protected Object[] createMatches() {
		Object[] a = new Object[_selection.size()];
		for (int i = 0; i < a.length; i++) {
			a[i] = _selection.get(i).decoratedValue.value;
		}
		return a;
	}
}