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.format.DateTimeFormatter;

import java.time.temporal.TemporalAccessor;

import java.util.Date;

import java.util.Objects;

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.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataTypeUtils;

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.dal.feature.FeatureStore;

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

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.util.Invocable;

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;

    private boolean modified;

    protected DefaultEditableFeature(DefaultFeature feature) {

        super(feature);

        this.source = feature;

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

        this.modified = false;

    }

    protected DefaultEditableFeature(DefaultEditableFeature feature) {

        super(feature);

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

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

        this.modified = feature.modified;

    }

    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;

        this.modified = false;

    }

    @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());

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void set(int index, Object value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

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

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

        modified = true;

        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.");

        }

        modified = true;

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

    }

    @Override

    public void setBoolean(int index, boolean value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setByte(int index, byte value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setDate(int index, Date value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setDouble(int index, double value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setDecimal(int index, BigDecimal value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setFeature(int index, Feature value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setFloat(int index, float value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setGeometry(int index, Geometry value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setInt(int index, int value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setLong(int index, long value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setString(int index, String value) {

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

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void copyFrom(JsonObject values) {

        this.copyFrom(values, null);

    }

//    private boolean notCopyAttribute(FeatureAttributeDescriptor attr,Predicate<FeatureAttributeDescriptor> filter) {

//        // helper function to use in copyFrom

//        if (attr==null  ) {

//            return true;

//        }

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

//            return true;

//        }

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

//            return true;

//        }

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

//            return true;

//        }

//        return false;

//    }

    @Override

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

      // iterate over the attributes and copy one by one

      if( values == null ) {

          throw new IllegalArgumentException("'values' argument can't be null");

      }

      boolean geometryCopied = false;

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

            if( !canSetValue(attr, filter) ) {

                continue;

            }

            String attrname = attr.getName();

            if( !values.containsKey(attrname) ) {

                continue;

            }

            Object value;

            JsonValue jsonValue = values.get(attrname);

           if(jsonValue == null){

                value = null;

            } else {

                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.BYTEARRAY:

                                    value = DataTypeUtils.coerce(DataTypes.BYTEARRAY, s, null);

                                    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) {

                LOG.trace("Can't assign value "+Objects.toString(value,null)+" to "+attr.getName());// 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, null);

    }

    @Override

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

        copyFrom(source, copy, null);

    }

    @Override

    public void copyFrom(Feature source, Predicate<FeatureAttributeDescriptor> copy, Invocable onerror) {

      // iterate over the attributes and copy one by one

        if( source == null ) {

            throw new IllegalArgumentException("'source' argument can't be null");

        }

        FeatureType sourceType = source.getType();

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

            if( !canSetValue(attrTarget, copy) ) {

                continue;

            }

            String attrname = attrTarget.getName();

            FeatureAttributeDescriptor attrSource = sourceType.getAttributeDescriptorFromAll(attrname);

            if( attrSource==null ) {

              continue;

            }

            Object value;

            if( attrSource.hasDataProfile() ) {

                value = source.getFromProfile(attrname);

                if( value == null ) {

                    value = source.get(attrname);

                }

            } else {

                value = source.get(attrname);

            }

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

                if( onerror != null ) {

                    onerror.call("Can't assign null to field '"+attrname+"'.", null);

                }

                continue;

            }

            try {

                set(attrTarget.getIndex(), value);

            } catch(Throwable th) {

                if( onerror == null ) {

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

                } else {

                    String s = null;

                    try {

                        s = StringUtils.replace(BaseException.getMessageStack(th, 0),"\n"," ");

                    } catch(Throwable th2) {

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

                    }

                    onerror.call(s, 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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setInstant(int index, Instant value) {

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

        modified = true;

        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.");

        }

        modified = true;

        this.set(attribute, value);

    }

    @Override

    public void setInterval(int index, Interval value) {

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

        modified = true;

        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);

    }

    public boolean isModified() {

        return modified;

    }

    @Override

    public Feature getOriginal() {

        FeatureStore store = this.getStore();

        if( store == null ) {

            return null;

        }

        Feature original = store.getOriginalFeature(this);

        return original;

    }

}