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.store.simplereader;

import javax.json.JsonArray;

import javax.json.JsonObject;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.aggregate.MultiPoint;

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

import org.gvsig.json.Json;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.Coercion;

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.tools.dataTypes.DataTypesManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

public class SimpleReaderPointAttributeEmulator implements FeatureAttributeEmulator {

    private static final Logger logger = LoggerFactory.getLogger(SimpleReaderPointAttributeEmulator.class);

    private static final int XNAME = 0;

    private static final int YNAME = 1;

    private static final int ZNAME = 2;

    private GeometryManager geommgr;

    private String[] fieldNames;

    private Coercion toDouble;

    private DataType dataType;

    private int errorcount = 0;

    public SimpleReaderPointAttributeEmulator() {

    }

    public SimpleReaderPointAttributeEmulator(String[] pointDimensionNames) {

        if (pointDimensionNames.length > 2) {

            this.fieldNames = new String[3];

            this.fieldNames[ZNAME] = pointDimensionNames[2];

        } else {

            this.fieldNames = new String[2];

        }

        this.fieldNames[XNAME] = pointDimensionNames[0];

        this.fieldNames[YNAME] = pointDimensionNames[1];

        this.geommgr = GeometryLocator.getGeometryManager();

        DataTypesManager datatypeManager = ToolsLocator.getDataTypesManager();

        this.toDouble = datatypeManager.getCoercion(DataTypes.DOUBLE);

        this.dataType = datatypeManager.get(DataTypes.GEOMETRY);

    }

    public String[] getFieldNames() {

        return fieldNames;

    }

    @Override

    public Object get(Feature feature) {

        try {

            Object valueX = feature.get(this.fieldNames[XNAME]);

            valueX = toDouble.coerce(valueX);

            if (valueX == null) {

                return null;

            }

            Object valueY = feature.get(this.fieldNames[YNAME]);

            valueY = toDouble.coerce(valueY);

            if (valueY == null) {

                return null;

            }

            Object valueZ = null;

            if (this.fieldNames.length > 2) {

                valueZ = toDouble.coerce(feature.get(this.fieldNames[ZNAME]));

                if (valueZ == null) {

                    return null;

                }

            }

            double x = (Double) valueX;

            double y = (Double) valueY;

            Point point = geommgr.createPoint(x, y, Geometry.SUBTYPES.GEOM3D);

            if (this.fieldNames.length > 2) {

                double z = (Double) valueZ;

                point.setCoordinateAt(2, z);

            }

            return point;

        } catch (Exception ex) {

            if (++errorcount < 5) {

                logger.warn("[" + errorcount + "] Can't create point in CSV provider. XNAME='" + this.fieldNames[XNAME] + "', YNAME='" + this.fieldNames[XNAME] + "' feature=" + feature.toString(), ex);

            }

            return null;

        }

    }

    @Override

    public void set(EditableFeature feature, Object value) {

        if (value == null) {

            return;

        }

        Point point;

        if (value instanceof MultiPoint) {

            point = (Point) ((MultiPoint) value).getPrimitiveAt(0);

        } else {

            point = (Point) value;

        }

        feature.set(this.fieldNames[XNAME], point.getX());

        feature.set(this.fieldNames[YNAME], point.getY());

        if (this.fieldNames.length > 2) {

            feature.set(this.fieldNames[ZNAME], point.getCoordinateAt(2));

        }

    }

    @Override

    public boolean allowSetting() {

        return true;

    }

    @Override

    public String[] getRequiredFieldNames() {

        return this.fieldNames;

    }

    @Override

    public JsonObject toJson() {

        return this.toJsonBuilder().build();

    }

    @Override

    public JsonObjectBuilder toJsonBuilder() {

        JsonObjectBuilder builder = Json.createObjectBuilder();

        builder.add_class(this);

        builder.add("fieldNames", fieldNames);

        return builder;

    }

    @Override

    public void fromJson(JsonObject json) {

        JsonArray jsonFieldNames = json.getJsonArray("fieldNames");

        this.fieldNames = new String[jsonFieldNames.size()];

        for (int i = 0; i < jsonFieldNames.size(); i++) {

            this.fieldNames[XNAME] = jsonFieldNames.getString(i);

        }

        this.geommgr = GeometryLocator.getGeometryManager();

        DataTypesManager datatypeManager = ToolsLocator.getDataTypesManager();

        this.toDouble = datatypeManager.getCoercion(DataTypes.DOUBLE);

        this.dataType = datatypeManager.get(DataTypes.GEOMETRY);

    }

}