Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License as published by the Free Software

 * Foundation; either version 3 of the License, or (at your option) any later

 * version.

 *

 * This program is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more

 * details.

 *

 * You should have received a copy of the GNU General Public License along with

 * this program; if not, write to the Free Software Foundation, Inc., 51

 * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us at info AT

 * gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.dal.feature.spi;

import java.math.BigDecimal;

import org.apache.commons.lang3.ArrayUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.timesupport.Time;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.Coercion;

import org.gvsig.tools.dataTypes.CoercionException;

import org.gvsig.tools.math.BigDecimalUtils;

/**

 * Default implementation for {@link FeatureProvider}

 *

 */

public class DefaultFeatureProvider implements FeatureProvider {

    protected FeatureType featureType;

    protected boolean[] nulls;

    protected Object[] values;

    protected Geometry defaultGeometry;

    protected Envelope envelope;

    private Object oid;

    private boolean isNew = false;

    private String[] extraValuesNames;

    private Object[] extraValues;

    protected boolean broken;

    public DefaultFeatureProvider(FeatureType type) {

//        if (type instanceof EditableFeatureType) {

//            throw new IllegalArgumentException("type can't be editable.");

//        }

        this.featureType = type;

        this.values = new Object[featureType.size()];

        this.nulls = new boolean[featureType.size()];

        this.envelope = null;

        this.defaultGeometry = null;

        this.oid = null;

        this.broken = false;

    }

    public DefaultFeatureProvider(FeatureType type, Object oid) {

        this(type);

        this.oid = oid;

    }

    public DefaultFeatureProvider(FeatureType type, DefaultFeatureProvider data) {

        this(type);

        this.oid = data.oid;

        this.defaultGeometry = data.defaultGeometry;

        this.envelope = data.envelope;

        this.isNew = data.isNew;

    }

    public boolean isReadOnly(int i) {

        FeatureAttributeDescriptor attribute = featureType.getAttributeDescriptor(i);

        if (attribute.getEvaluator() != null) {

            return true;

        }

        return false;

    }

    @Override

    public void set(int i, Object value) {

        FeatureAttributeDescriptor attribute = featureType.getAttributeDescriptor(i);

//        if (this.isReadOnly(i)) {

//            return;

//        }

        if (featureType.getDefaultGeometryAttributeIndex() == i) {

            if (value instanceof Geometry) {

                defaultGeometry = (Geometry) value;

            } else {

                try {

                    defaultGeometry = (Geometry) ToolsLocator.getDataTypesManager().coerce(DataTypes.GEOMETRY, value);

                } catch (CoercionException ex) {

                }

            }

            envelope = null;

        } else if(value != null) {

            Class objectClass = attribute.getObjectClass();

            if (objectClass != null) {

                if (!objectClass.isInstance(value)) {

                    try {

                        Coercion coercer = attribute.getDataType().getCoercion();

                        if (attribute.getType() == DataTypes.STRING && value instanceof Boolean) {

                            value = coercer.coerce(value, attribute.getCoercionContext());

                            value = StringUtils.left((String) value, attribute.getSize());

                        } else {

                            value = coercer.coerce(value, attribute.getCoercionContext());

                        }

                    } catch (CoercionException e) {

                        throw new IllegalArgumentException("Can't convert to "

                                + attribute.getDataType().getName()

                                + " from '"

                                + value.getClass().getName()

                                + "' with value '"

                                + value.toString()

                                + "' and context '"

                                + attribute.getCoercionContext()

                                + "' to assign to '"

                                + attribute.getName()

                                + "'.",e);

                    }

                } else {

                    try {

                        switch(attribute.getType()){

                            case DataTypes.DECIMAL:

                                value = BigDecimalUtils.force((BigDecimal) value, attribute.getScale(), attribute.getPrecision());

                                break;

                        }

                    } catch(Throwable th) {

                            throw new IllegalArgumentException("Can't assign "

                                + value.toString()

                                + " to '"

                                + attribute.getName()

                                + "'.",th);

                    }

                }

            }

        }

        if (value == null) {

            values[i] = attribute.getDefaultValue();

            nulls[i] = true;

        } else {

            values[i] = value;

            nulls[i] = false;

        }

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#set(java.lang.String,

     * java.lang.Object)

     */

    public void set(String name, Object value) {

        set(featureType.getIndex(name), value);

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#get(int)

     */

    public Object get(int i) {

        return values[i];

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#get(java.lang.String)

     */

    public Object get(String name) {

        FeatureAttributeDescriptor featureAttributeDescriptor = featureType.getAttributeDescriptor(name);

        if(featureAttributeDescriptor == null){

            throw new IllegalStateException("Can't get attribute '"+name+"'");

        }

        return values[featureAttributeDescriptor.getIndex()];

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#getType()

     */

    public FeatureType getType() {

        return featureType;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#getCopy()

     */

    public FeatureProvider getCopy() {

        DefaultFeatureProvider data = new DefaultFeatureProvider(

                this.getType());

        return getCopy(data);

    }

    /**

     * Copy values from current instance to <code>data</code>

     */

    protected FeatureProvider getCopy(DefaultFeatureProvider data) {

        data.oid = this.oid;

        System.arraycopy(this.values, 0, data.values, 0, this.values.length);

        data.defaultGeometry = this.defaultGeometry;

        data.envelope = this.envelope;

        data.isNew = this.isNew;

        if (this.extraValues!=null) {

            data.extraValues = new Object[this.extraValues.length];

            System.arraycopy(this.extraValues, 0, data.extraValues, 0, this.extraValues.length);

        } else {

            data.extraValues = null;

        }

        if (this.extraValuesNames!=null) {

            data.extraValuesNames = new String[this.extraValuesNames.length];

            System.arraycopy(this.extraValuesNames, 0, data.extraValuesNames, 0, this.extraValuesNames.length);

        } else {

            data.extraValuesNames = null;

        }

        data.broken = this.broken;

        return data;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#getDefaultEnvelope()

     */

    public Envelope getDefaultEnvelope() {

        if (envelope == null && getDefaultGeometry() != null) {

            envelope = getDefaultGeometry().getEnvelope();

        }

        return envelope;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#getDefaultGeometry()

     */

    public Geometry getDefaultGeometry() {

        return this.defaultGeometry;

    }

    @Override

    public Time getDefaultTime() {

        int index = featureType.getDefaultTimeAttributeIndex();

        Object value = this.get(index);

        if (! (value instanceof Time) ) {

            try {

                value = (Time) ToolsLocator.getDataTypesManager().coerce(DataTypes.INSTANT, value);

            } catch (CoercionException ex) {

            }

        }

        return (Time) value;

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.gvsig.fmap.dal.feature.spi.FeatureProvider#setDefaultEnvelope(org

     * .gvsig.fmap.geom.primitive.Envelope)

     */

    public void setDefaultEnvelope(Envelope envelope) {

        this.envelope = envelope;

    }

    @Override

    public void setDefaultGeometry(Geometry geom) {

        int i = featureType.getDefaultGeometryAttributeIndex();

        this.set(i, geom);

    }

    @Override

    public void setDefaultTime(Time time) {

        int i = featureType.getDefaultTimeAttributeIndex();

        this.set(i, time);

    }

    public boolean isNull(int i) {

        return nulls[i];

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.gvsig.fmap.dal.feature.spi.FeatureProvider#isNull(java.lang.String)

     */

    public boolean isNull(String name) {

        int i = featureType.getIndex(name);

        return isNull(i);

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#getOID()

     */

    public Object getOID() {

        return this.oid;

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.gvsig.fmap.dal.feature.spi.FeatureProvider#setOID(java.lang.Object)

     */

    public void setOID(Object oid) {

        this.oid = oid;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#isNew()

     */

    public boolean isNew() {

        return isNew;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.fmap.dal.feature.spi.FeatureProvider#setNew(boolean)

     */

    public void setNew(boolean isNew) {

        this.isNew = isNew;

    }

    @Override

    public void setExtraValue(int index, Object value) {

        this.extraValues[index] = value;

    }

    @Override

    public Object getExtraValue(int index) {

        if( this.extraValues == null ) {

            return null;

        }

        return this.extraValues[index];

    }

    @Override

    public Object getExtraValue(String name) {

        if( this.extraValuesNames == null || this.extraValues == null) {

            return null;

        }

        for (int i = 0; i < this.extraValuesNames.length; i++) {

            if( StringUtils.equalsIgnoreCase(this.extraValuesNames[i], name) ) {

                return this.extraValues[i];

            }

        }

        return null;

    }

    @Override

    public boolean hasExtraValue(String name) {

        if( this.extraValuesNames == null ) {

            return false;

        }

        for (int i = 0; i < this.extraValuesNames.length; i++) {

            if( StringUtils.equalsIgnoreCase(this.extraValuesNames[i], name) ) {

                return true;

            }

        }

        return false;

    }

    @Override

    public void setExtraValueNames(String[] extraValueNames) {

        this.extraValuesNames = extraValueNames;

        this.extraValues = new Object[this.extraValuesNames.length];

    }

    @Override

    public boolean isBroken() {

        return this.broken;

    }

}