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.export.impl.service;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.collections.ListUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.export.ExportAttributes;

import org.gvsig.export.ExportLocator;

import org.gvsig.export.spi.AttributeNamesTranslator;

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

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

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

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

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.tools.util.UnmodifiableBasicListAdapter;

public class DefaultExportAttributes implements ExportAttributes {

    private List<ExportAttribute> exportAttributes;

    private AttributeNamesTranslator namesTranslator;

    private FeatureType sourceFeatureType;

    public DefaultExportAttributes(FeatureType ftype) {

        //TODO

        //rellenar pirmero sin el suggest name

        //rellamar al algorito de suggest

        this.fillExportAttributes(ftype);

        this.setSourceFeatureType(ftype);

    }

    public void fillExportAttributes(FeatureType ftype) {

        ArrayList attrs = new ArrayList();

        for (FeatureAttributeDescriptor fad : ftype) {

            DefaultExportAttribute exportAttribute = new DefaultExportAttribute(fad);

            exportAttribute.setNewName(fad.getName());

            exportAttribute.setNewType(fad.getDataType().getType());

            exportAttribute.setSize(fad.getSize());

            exportAttribute.setExported(true);

            attrs.add(exportAttribute);

        }

        this.exportAttributes = attrs;

    }

    @Override

    public void setNamesTranslator(AttributeNamesTranslator namesTranslator) {

        //TODO tb hay que rehacer todos los suggestname.

        this.namesTranslator = namesTranslator;

    }

    @Override

    public AttributeNamesTranslator getNamesTranslator() {

        return this.namesTranslator;

    }

    @Override

    public List<ExportAttribute> toList() {

        return this.exportAttributes;

    }

    @Override

    public boolean isEmpty() {

        return this.exportAttributes.isEmpty();

    }

    @Override

    public int size() {

        return this.exportAttributes.size();

    }

    @Override

    public Iterator<ExportAttribute> iterator() {

        return this.exportAttributes.iterator();

    }

    @Override

    public ExportAttribute get(int position) {

        return this.exportAttributes.get(position);

    }

    @Override

    public ExportAttribute getExportAttribute(String name) {

        for (ExportAttribute exportAttribute : this.exportAttributes) {

            if (StringUtils.equals(exportAttribute.getName(), name)) {

                return exportAttribute;

            }

        }

        return null;

    }

    @Override

    public FeatureType getTargetFeatureType() {

        EditableFeatureType targetFeatureType = this.getSourceFeatureType().getCopy().getEditable();

        for (FeatureAttributeDescriptor attrSource : this.getSourceFeatureType()) {

            FeatureAttributeDescriptor attr = targetFeatureType.getAttributeDescriptor(attrSource.getName());

            String name = attr.getName();

            ExportAttribute exportAttr = this.getExportAttribute(name);

            if (!exportAttr.isExported()) {

                targetFeatureType.remove(name);

                continue;

            }

            EditableFeatureAttributeDescriptor eAttr = targetFeatureType.getEditableAttributeDescriptor(name);

            if (this.getTargetName(name) == null ? name != null : !this.getTargetName(name).equals(name)) {

                eAttr.setName(this.getTargetName(name));

            }

            int type = attr.getDataType().getType();

            if (type != this.getTargetType(name)) {

                eAttr.setDataType(this.getTargetType(name));

                eAttr.setDefaultValue(null); // TODO: delete default value

            }

            int size = exportAttr.getSize();

            eAttr.setSize(size);

        }

        return targetFeatureType.getNotEditableCopy();

    }

    @Override

    public void setSourceFeatureType(FeatureType sourceFeatureType) {

        this.sourceFeatureType = sourceFeatureType;

    }

    @Override

    public FeatureType getSourceFeatureType() {

        return this.sourceFeatureType;

    }

    @Override

    public String getTargetName(String name) {

        for (ExportAttribute exportAttribute : this.exportAttributes) {

            if (StringUtils.equals(exportAttribute.getName(), name)) {

                return exportAttribute.getNewName();

            }

        }

        return null;

    }

    @Override

    public String getSourceName(String name) {

        for (ExportAttribute exportAttribute : this.exportAttributes) {

            if (StringUtils.equalsIgnoreCase(exportAttribute.getNewName(), name)) {

                return exportAttribute.getName();

            }

        }

        return null;

    }

    @Override

    public int getTargetType(String name) {

        for (ExportAttribute exportAttribute : this.exportAttributes) {

            if (StringUtils.equals(exportAttribute.getName(), name)) {

                return exportAttribute.getNewDataType();

            }

        }

        return 0;

    }

    @Override

    public int getSourceType(String name) {

        for (ExportAttribute exportAttribute : this.exportAttributes) {

            if (StringUtils.equals(exportAttribute.getNewName(), name)) {

                return exportAttribute.getDataType();

            }

        }

        return 0;

    }

}