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.Json;

import javax.json.JsonArray;

import javax.json.JsonArrayBuilder;

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;

/**

 *

 * @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.equals(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()) {

            this.extraColumns.add(col);

        }

    }

    @Override

    public void clear() {

        this.extraColumns.clear();

    }

    public FeatureExtraColumns getDeepCopy() {

        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 FeatureExtraColumns getCopy() {

        DefaultFeatureExtraColumns other = new DefaultFeatureExtraColumns();

        ArrayList otherCols = new ArrayList();

        otherCols.addAll(extraColumns);

        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 JsonArray toJson() {

        return this.toJsonBuilder().build();

    }

    public JsonArrayBuilder toJsonBuilder() {

        JsonArrayBuilder builder = Json.createArrayBuilder();

        for (EditableFeatureAttributeDescriptor extraColumn : extraColumns) {

            builder.add(((DefaultFeatureAttributeDescriptor)extraColumns).toJsonBuilder());

        }

        return builder;

    }

    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;

            }

        }

    }

}