Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.fmap.dal.swing.impl.searchpanel;

import java.io.ByteArrayInputStream;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.json.Json;

import javax.json.JsonObject;

import javax.json.JsonReader;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.swing.searchpanel.SearchParameters;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

/**

 *

 * @author jjdelcerro

 */

public class DefaultSearchParameters implements SearchParameters, Cloneable {

    private List<String> resultColumnNames;

    private FeatureQuery query;

    public String name;

    private Map<String, JsonObject> values;

    public DefaultSearchParameters() {

        this.resultColumnNames = new ArrayList<>();

        this.query = null;

        this.values = new HashMap<>();

    }

    @Override

    public List<String> getResultColumnNames() {

        return this.resultColumnNames;

    }

    public List<String> getGroupByColumns() {

        return this.query.getGroupByColumns();

    }

    public Map<String, String> getAggregateFunctions() {

        return this.query.getAggregateFunctions();

    }

    public void setQuery(FeatureQuery query) {

        this.query = query;

    }

    public void setFilter(Expression exp) {

        this.query.setFilter(exp);

    }

    public FeatureQuery getQuery() {

        return this.query;

    }

    void setResultColumnNames(List<String> selectedNames) {

        throw new UnsupportedOperationException("Not supported yet."); 

    }

    @Override

    public Map<String, JsonObject> getValues() {

        return this.values;

    }

    public static void registerPersistence() {

        PersistenceManager manager = ToolsLocator.getPersistenceManager();

        if (manager.getDefinition("SearchParameters") == null) {

            DynStruct definition = manager.addDefinition(DefaultSearchParameters.class,

                    "DefaultSearchParameters", "DefaultSearchParameters persistence definition", null, null);

            definition.addDynFieldList("resultColumnNames").setClassOfItems(String.class);

            definition.addDynFieldMap("values").setClassOfItems(String.class);

            definition.addDynFieldObject("query").setClassOfValue(FeatureQuery.class).setMandatory(false);

            definition.addDynFieldString("name").setClassOfValue(String.class);

        }

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        state.set("resultColumnNames", this.resultColumnNames);

        HashMap<String, String> valuesMap = new HashMap<>();

        for (String key : this.values.keySet()) {

            JsonObject value = this.values.get(key);

            valuesMap.put(key, value.toString());

        }

        state.set("values", valuesMap);

        state.set("query", this.query);

        state.set("name", this.name);

    }

    @Override

    public void loadFromState(PersistentState state) throws PersistenceException {

        this.resultColumnNames = new ArrayList<>(state.getList("resultColumnNames"));

        Map<String, String> valuesState = state.getMap("values");

        HashMap<String, JsonObject> valuesMap = new HashMap<>();

        for (String key : valuesState.keySet()) {

            String value = valuesState.get(key);

            InputStream targetStream = new ByteArrayInputStream(value.getBytes());

            JsonReader reader = Json.createReader(targetStream);

            JsonObject jsonObject = reader.readObject();

            valuesMap.put(key, jsonObject);

        }

        this.values = valuesMap;

        this.query = (FeatureQuery) state.get("query");

        this.name = state.getString("name");

    }

    @Override

    public String toString() {

        if (this.query != null && this.query.getExpressionFilter() != null) {

            return this.query.getExpressionFilter().getPhrase();

        } else {

            return this.getName();

        }

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        DefaultSearchParameters clone = (DefaultSearchParameters) super.clone();

        clone.query = this.query.getCopy();

        clone.resultColumnNames = new ArrayList(this.resultColumnNames);

        HashMap<String, JsonObject> cloneValues = new HashMap<>(this.values);

        clone.values = cloneValues;

        return clone;

    }

    public DefaultSearchParameters getCopy() {

        try {

            return (DefaultSearchParameters) this.clone();

        } catch (CloneNotSupportedException ex) {

            return null;

        }

    }

    @Override

    public void copyFrom(SearchParameters params) {

        this.resultColumnNames = params.getResultColumnNames();

        this.query = params.getQuery().getCopy();

        this.name = params.getName();

        this.values = params.getValues();

    }

    @Override

    public String getName() {

        return this.name;

    }

    @Override

    public void setName(String name) {

        this.name= name;

    }

    @Override

    public FeatureType getFeatureType(FeatureStore store) {

        FeatureSet fset = null;

        try {

            fset = store.getFeatureSet(this.query);

            FeatureType ft = fset.getDefaultFeatureType();

            return ft;

        } catch (DataException ex) {

            return store.getDefaultFeatureTypeQuietly();

        } finally {

            DisposeUtils.disposeQuietly(fset);

        }

    }

}