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

import java.math.BigDecimal;

import java.time.LocalDate;

import java.time.LocalDateTime;

import java.time.LocalTime;

import java.time.ZoneOffset;

import java.time.format.DateTimeFormatter;

import java.time.temporal.TemporalAccessor;

import java.util.Date;

import java.util.Set;

import java.util.function.Predicate;

import javax.json.JsonNumber;

import javax.json.JsonObject;

import javax.json.JsonString;

import javax.json.JsonValue;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryUtils;

import org.gvsig.timesupport.Instant;

import org.gvsig.timesupport.Interval;

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

import org.gvsig.fmap.geom.type.GeometryType;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultEditableFeature extends DefaultFeature implements

        EditableFeature {

    private static final Logger LOG = LoggerFactory.getLogger(DefaultEditableFeature.class);

    private DefaultFeature source;

//    private boolean inserted;

    protected DefaultEditableFeature(DefaultFeature feature) {

        super(feature);

        this.source = feature;

//        this.inserted = this.source!=null;

    }

    protected DefaultEditableFeature(DefaultEditableFeature feature) {

        super(feature);

        this.source = (DefaultFeature) feature.getSource();

//        this.inserted = this.source!=null;

    }

    public DefaultEditableFeature(DefaultFeatureStore store, FeatureProvider data) {

        // Se trata de un editable feature sobre una ya existente

        super(store, data);

        this.source = null;

//        this.inserted = this.source!=null;

    }

    @Override

    public Feature getSource() {

        return this.source;

    }

    @Override

    public boolean isUpdatable() {

        return super.isInserted();

    }

    @Override

    public void setUpdatable(boolean updatable) {

        super.setInserted(updatable);

    }

    @Override

    public EditableFeature getEditable() {

        return this;

    }

    @Override

    public Feature getCopy() {

        return new DefaultEditableFeature(this);

    }

    @Override

    public Feature getNotEditableCopy() {

        return new DefaultFeature(this);

    }

    @Override

    public void setDefaultGeometry(Geometry geometry) {

        FeatureAttributeDescriptor attribute = this.getType()

                .getAttributeDescriptor(

                        this.getType().getDefaultGeometryAttributeIndex());

        this.set(attribute, geometry);

    }

    public void __setitem__(String name, Object value) {

        this.set(name, value);

    }

    @Override

    public void set(String name, Object value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void set(int index, Object value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setArray(String name, Object[] value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setArray(int index, Object[] value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setBoolean(String name, boolean value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, Boolean.valueOf(value));

    }

    @Override

    public void setBoolean(int index, boolean value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setByte(String name, byte value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setByte(int index, byte value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setDate(String name, Date value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setDate(int index, Date value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setDouble(String name, double value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setDouble(int index, double value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setDecimal(String name, BigDecimal value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setDecimal(int index, BigDecimal value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setFeature(String name, Feature value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setFeature(int index, Feature value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setFloat(String name, float value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setFloat(int index, float value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setGeometry(String name, Geometry value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setGeometry(int index, Geometry value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setInt(String name, int value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setInt(int index, int value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setLong(String name, long value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setLong(int index, long value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setString(String name, String value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setString(int index, String value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void copyFrom(JsonObject values) {

        this.copyFrom(values, null);    

    }

    @Override

    public void copyFrom(JsonObject values, Predicate<FeatureAttributeDescriptor> filter) {

      // iterate over the attributes and copy one by one

      boolean geometryCopied = false;

        for (FeatureAttributeDescriptor attr : this.getType()) {

            if (attr==null  ) {

                continue;

            }

            if( filter!=null && !filter.test(attr) ) {

                continue;

            }

            if (attr.isAutomatic()  || attr.isComputed() ) {

                continue;

            }

            if( this.isInserted() &&  attr.isReadOnly()) {

                continue;

            }

            String attrname = attr.getName();

            if( !values.containsKey(attrname) ) {

                continue;

            }           

            Object value;

            JsonValue jsonValue = values.get(attrname);

            try {

                switch( jsonValue.getValueType() ) {

                    case STRING:

                        String s = ((JsonString)jsonValue).getString();

                        switch(attr.getType()) {

                            case DataTypes.GEOMETRY:

                                value = GeometryUtils.createFrom(s);

                                geometryCopied = true;

                                break;

                            case DataTypes.TIMESTAMP:

                                try {

                                    TemporalAccessor x = DateTimeFormatter.ISO_DATE_TIME.parse(s);

                                    LocalDateTime date = LocalDateTime.from(x);

                                    value = java.sql.Timestamp.valueOf(date);

                                } catch(Exception ex) {

                                    value = s;

                                }

                                break;

                            case DataTypes.DATE:

                                try {

                                    TemporalAccessor x = DateTimeFormatter.ISO_DATE.parse(s);

                                    LocalDate date = LocalDate.from(x);

                                    value = java.sql.Date.valueOf(date);

                                } catch(Exception ex) {

                                    value = s;

                                }

                                break;

                            case DataTypes.TIME:

                                try {

                                    TemporalAccessor x = DateTimeFormatter.ISO_TIME.parse(s);

                                    LocalTime date = LocalTime.from(x);

                                    value = java.sql.Time.valueOf(date);

                                } catch(Exception ex) {

                                    value = s;

                                }

                                break;

                            case DataTypes.STRING:

                            default:

                                value = s;

                                break;

                        }

                        break;

                    case NUMBER:

                        JsonNumber n =  (JsonNumber) jsonValue;

                        switch(attr.getType()) {

                            case DataTypes.DECIMAL:

                                value = n.bigDecimalValue();

                                break;

                            case DataTypes.BYTE:

                                value = n.bigDecimalValue();

                                break;

                            case DataTypes.FLOAT:

                            case DataTypes.DOUBLE:

                                value = n.doubleValue();

                                break;

                            case DataTypes.INT:

                                value = n.intValue();

                                break;

                            case DataTypes.LONG:

                                value = n.longValue();

                                break;

                            default:

                                value = n.toString();

                                break;

                        }

                        break;

                    case TRUE:

                        value = true;

                        break;

                    case FALSE:

                        value = false;

                        break;

                    case NULL:

                        value = null;

                        break;

                    default:

                        value = values.getString(attrname);

                        break;

                }

            } catch(Exception ex) {

                continue;

            }

            if (value == null && !attr.allowNull()) {

                continue;

            }

            try {

                set(attr.getIndex(), value);

            } catch(Throwable th) {

                // Ignore

            }

        }

        FeatureAttributeDescriptor attr = this.getType().getDefaultGeometryAttribute();

        if (attr!=null && !geometryCopied) {

            if( filter==null || filter.test(attr) ) {

                GeometryType geomType = this.getType().getDefaultGeometryAttribute().getGeomType();

                Set<String> keys = values.keySet();

                for (String key : keys) {

                    try {

                        String wkt = values.getString(key);

                        Geometry geometry = GeometryUtils.createFrom(wkt);

                        if (GeometryUtils.isSubtype(geomType.getType(), geometry.getType()) || GeometryUtils.canAggregate(geomType.getType(), geometry.getType())) {

                            this.setDefaultGeometry(geometry);

                            break;

                        }

                    } catch (Throwable tr) {

                        LOG.trace("Can't get geometry from field {}.", new Object[]{key});

                    };

                }

            }

        }

    }

    @Override

    public void copyFrom(Feature source) {

        this.copyFrom(source, null);

    }

    @Override

    public void copyFrom(Feature source, Predicate<FeatureAttributeDescriptor> filter) {

      // iterate over the attributes and copy one by one

        FeatureType sourceType = source.getType();

        for (FeatureAttributeDescriptor attr : this.getType()) {

            if (attr==null  ) {

                continue;

            }

            if( filter!=null && !filter.test(attr) ) {

                continue;

            }

            if (attr.isAutomatic()  || attr.isComputed() ) {

                continue;

            }

            if( this.isInserted() &&  attr.isReadOnly()) {

                continue;

            }

            String attrname = attr.getName();

            if( sourceType.getAttributeDescriptor(attrname)==null ) {

              continue;

            }

            Object value = source.get(attrname);

            if (value == null && !attr.allowNull()) {

                continue;

            }

            try {

                set(attr.getIndex(), value);

            } catch(Throwable th) {

                LOG.trace("The exception thrown is: ", th);

            }

        }

    }

    @Override

    public void setInstant(String name, Instant value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setInstant(int index, Instant value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public void setInterval(String name, Interval value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(name);

        if ( attribute == null ) {

            throw new IllegalArgumentException("Attribute '" + name + "' does not exits in the feature.");

        }

        this.set(attribute, value);

    }

    @Override

    public void setInterval(int index, Interval value) {

        FeatureAttributeDescriptor attribute = this.getType().getAttributeDescriptor(index);

        this.set(attribute, value);

    }

    @Override

    public Object getExtraValue(String name) {

        Object value;

        FeatureExtraColumns columns = this.getType().getExtraColumns();

        int index = columns.getIndexOf(name);

        if (index >= 0) {

            EditableFeatureAttributeDescriptor attrdesc = columns.get(index);

            value = attrdesc.getFeatureAttributeEmulator().get(this);

            return value;

        }

        value = this.data.getExtraValue(name);

        return value;

    }

    @Override

    public void validate(int check) throws DataException {

        ((DefaultFeatureType) this.data.getType()).validateFeature(this, check);

    }

}