RandomSec/src/main/java/com/metaweb/gridlock/browsing/Engine.java

package com.metaweb.gridlock.browsing;

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 org.json.JSONWriter;

import com.metaweb.gridlock.Jsonizable;
import com.metaweb.gridlock.browsing.facets.Facet;
import com.metaweb.gridlock.browsing.facets.ListFacet;
import com.metaweb.gridlock.browsing.facets.RangeFacet;
import com.metaweb.gridlock.browsing.facets.TextSearchFacet;
import com.metaweb.gridlock.browsing.filters.RowFilter;
import com.metaweb.gridlock.model.Project;

public class Engine implements Jsonizable {
	protected Project 		_project;
	protected List<Facet> 	_facets = new LinkedList<Facet>();
	
	public Engine(Project project) {
		_project  = project;
	}
	
	public FilteredRows getAllFilteredRows() {
		return getFilteredRows(null);
	}

	public FilteredRows getFilteredRows(Facet except) {
		ConjunctiveFilteredRows cfr = new ConjunctiveFilteredRows();
		for (Facet facet : _facets) {
			if (facet != except) {
				RowFilter rowFilter = facet.getRowFilter();
				if (rowFilter != null) {
					cfr.add(rowFilter);
				}
			}
		}
		return cfr;
	}
	
	public void initializeFromJSON(JSONObject o) throws Exception {
		JSONArray a = o.getJSONArray("facets");
		int length = a.length();
		
		for (int i = 0; i < length; i++) {
			JSONObject fo = a.getJSONObject(i);
			String type = fo.has("type") ? fo.getString("type") : "list";
			
			Facet facet = null;
			if ("list".equals(type)) {
				facet = new ListFacet();
			} else if ("range".equals(type)) {
				facet = new RangeFacet();
			} else if ("text".equals(type)) {
				facet = new TextSearchFacet();
			}
			
			if (facet != null) {
				facet.initializeFromJSON(fo);
				_facets.add(facet);
			}
		}
	}
	
	public void computeFacets() throws JSONException {
		for (Facet facet : _facets) {
			FilteredRows filteredRows = getFilteredRows(facet);
			
			facet.computeChoices(_project, filteredRows);
		}
	}

	@Override
	public void write(JSONWriter writer, Properties options)
			throws JSONException {
		
		writer.object();
		writer.key("facets"); writer.array();
		for (Facet facet : _facets) {
			facet.write(writer, options);
		}
		writer.endArray();
		writer.endObject();
	}
}