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.feature.impl;

import java.util.ArrayList;

import java.util.List;

import javax.json.JsonObject;

import javax.json.JsonValue;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistentState;

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

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

import org.gvsig.json.Json;

import org.gvsig.json.JsonManager;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.json.SupportToJson;

/**

 *

 * @author osc

 */

public class DefaultFeatureExtraColumns implements FeatureExtraColumns {

    private List<EditableFeatureAttributeDescriptor> extraColumns;

    public DefaultFeatureExtraColumns() {

        this.extraColumns = new ArrayList();

    }

    @Override

    public boolean isEmpty() {

      return this.extraColumns.isEmpty();

    }

    @Override

    public EditableFeatureAttributeDescriptor get(String columnName) {

        for (EditableFeatureAttributeDescriptor extraColumn : this.extraColumns) {

            if (StringUtils.equals(extraColumn.getName(), columnName)) {

                return extraColumn;

            }

        }

        return null;

    }

    @Override

    public EditableFeatureAttributeDescriptor add(String name) {

        //TODO que no hayan dos columnas iguales

//        exi.addExtraColumn()

        EditableFeatureAttributeDescriptor newAttr = new DefaultEditableFeatureAttributeDescriptor(null, false);

        newAttr.setName(name);

        extraColumns.add(newAttr);

        return newAttr;

    }

    @Override

    public List<EditableFeatureAttributeDescriptor> getColumns() {

        return this.extraColumns;

    }

    @Override

    public void merge(FeatureExtraColumns other) {

        this.extraColumns.addAll(other.getColumns());

    }

    @Override

    public int getIndexOf(String name) {

        for (int i = 0; i < extraColumns.size(); i++) {

            EditableFeatureAttributeDescriptor extraColumn = extraColumns.get(i);

            if (StringUtils.equalsIgnoreCase(extraColumn.getName(), name)) {

                return i;

            }

        }

        return -1;

    }

    @Override

    public EditableFeatureAttributeDescriptor get(int index) {

        return extraColumns.get(index);

    }

    @Override

    public int size() {

        return extraColumns.size();

    }

    @Override

    public void copyFrom(FeatureExtraColumns extraColumn) {

        this.extraColumns.clear();

        for (EditableFeatureAttributeDescriptor col : extraColumn.getColumns()) {

            //TODO revisar si getCopy o no

            this.extraColumns.add(col);

        }

    }

    @Override

    public void clear() {

        this.extraColumns.clear();

    }

    @Override

    public FeatureExtraColumns getCopy() {

        DefaultFeatureExtraColumns other = new DefaultFeatureExtraColumns();

        ArrayList otherCols = new ArrayList();

        for (EditableFeatureAttributeDescriptor extraColumn : this.extraColumns) {

            EditableFeatureAttributeDescriptor newAttr = DALLocator.getDataManager().createFeatureAttributeDescriptor();

            newAttr.copyFrom(extraColumn);

            otherCols.add(newAttr);

        }

        other.extraColumns = otherCols;

        return other;

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

//        ArrayList<DefaultFeatureAttributeDescriptor> extraListCols = new ArrayList<>();

//        for (EditableFeatureAttributeDescriptor extraColumn : extraColumns) {

//            DefaultEditableFeatureAttributeDescriptor mycol = (DefaultEditableFeatureAttributeDescriptor) extraColumn;

//            DefaultFeatureAttributeDescriptor fad = new DefaultFeatureAttributeDescriptor();

//            fad.copyFrom(mycol);

//            extraListCols.add(fad);

//        }

         state.set("extraColumns", extraColumns);

    }

    @Override

    public void loadFromState(PersistentState state) throws PersistenceException {

        List<DefaultFeatureAttributeDescriptor> extraColumnsDescriptors = (List) state.getList("extraColumns");

        ArrayList<EditableFeatureAttributeDescriptor> cols = new ArrayList<EditableFeatureAttributeDescriptor>();

        for (DefaultFeatureAttributeDescriptor extraColumnsDescriptor : extraColumnsDescriptors) {

            DefaultEditableFeatureAttributeDescriptor myEditableDescriptor = new DefaultEditableFeatureAttributeDescriptor(extraColumnsDescriptor);

            cols.add(myEditableDescriptor);

        }

        this.extraColumns = cols;

    }

    public static void registerPersistent() {

        DynStruct definition =

            ToolsLocator.getPersistenceManager()

            .addDefinition(DefaultFeatureExtraColumns.class,

                "DefaultFeatureExtraColumn",

                "DefaultFeatureExtraColumn Persistent definition",

                null,

                null);

        definition.addDynFieldArray("extraColumns")

                .setClassOfItems(FeatureAttributeDescriptor.class)

                .setMandatory(false);

    }

    @Override

    public void remove(String attribute) {

        for (EditableFeatureAttributeDescriptor extraColumn : extraColumns) {

            if (extraColumn.getName().equalsIgnoreCase(attribute)) {

                this.extraColumns.remove(extraColumn);

                break;

            }

        }

    }

    @Override

    public void fromJson(JsonObject json) {

        this.clear();

        if( json.containsKey("extraColumns") ) {

            for (JsonValue jsonValue : json.getJsonArray("extraColumns")) {

                JsonObject value = (JsonObject) jsonValue;

                EditableFeatureAttributeDescriptor newAttr = new DefaultEditableFeatureAttributeDescriptor(null, false);

                newAttr.fromJson(value);

                this.extraColumns.add(newAttr);

            }

        }

    }

    @Override

    public JsonObject toJson() {

        return this.toJsonBuilder().build();

    }

    @Override

    public JsonObjectBuilder toJsonBuilder() {

        JsonObjectBuilder builder = Json.createObjectBuilder();

        builder.add_class(this);

        builder.add("extraColumns", Json.createArrayBuilder().addAll(this.extraColumns));

        return builder;

    }

    private static class TheJsonSerializer implements JsonManager.JsonSerializer {

        public TheJsonSerializer() {            

        }

        @Override

        public Class getObjectClass() {

            return DefaultFeatureExtraColumns.class;

        }

        @Override

        public Object toObject(JsonObject json) {

            DefaultFeatureExtraColumns o = new DefaultFeatureExtraColumns();

            o.fromJson(json);

            return o;

        }

        @Override

        public JsonObjectBuilder toJsonBuilder(Object value) {

            return ((SupportToJson)value).toJsonBuilder();

        }

    }

    public static void selfRegister() {

        Json.registerSerializer(new TheJsonSerializer());

    }

}