Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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 2

 * 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.

 *

 */

/*

 * AUTHORS (In addition to CIT):

 * 2008 IVER T.I. S.A.   {{Task}}

 */

package org.gvsig.fmap.dal.store.shp;

import java.io.File;

import java.io.IOException;

import java.nio.channels.FileChannel;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

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.store.dbf.DBFFeatureWriter;

import org.gvsig.fmap.dal.store.dbf.DBFStoreParameters;

import org.gvsig.fmap.dal.store.shp.utils.SHPFileWrite;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

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

public class SHPFeatureWriter extends DBFFeatureWriter {

    private static final GeometryManager geomManager = GeometryLocator

        .getGeometryManager();

    private static final Logger logger = LoggerFactory

        .getLogger(GeometryManager.class);

    private SHPFileWrite shpWrite;

    private Envelope envelope = null;

    private int[] supportedGeometryTypes;

    private Geometry defaultGeometry;

    private int fileSize;

    private FeatureType shpFeatureType;

    protected SHPFeatureWriter(String name) {

        super(name);

        try {

            this.defaultGeometry =

                geomManager.createNullGeometry(SUBTYPES.GEOM2D);

        } catch (CreateGeometryException e) {

            logger.error("Creating a NullGeometry", e);

        }

    }

    public void begin(DBFStoreParameters dbfParameters,

        FeatureType featureType, FeatureType dbfFeatureType, long numRows)

        throws DataException {

        SHPStoreParameters shpParameters = (SHPStoreParameters) dbfParameters;

        File shpFile = shpParameters.getSHPFile();

        File shxFile = shpParameters.getSHXFile();

        FileChannel shpChannel = null;

        FileChannel shxChannel = null;

        try {

            shpChannel =

                (FileChannel) getWriteChannel(shpFile.getAbsolutePath());

            shxChannel =

                (FileChannel) getWriteChannel(shxFile.getAbsolutePath());

        } catch (IOException e) {

            throw new WriteException(this.name, e);

        }

        shpWrite = new SHPFileWrite(shpChannel, shxChannel);

        int shapeType = getShapeTypeAndSetSupportedGeometries(featureType);

        try {

            shpWrite.writeHeaders(

                geomManager.createEnvelope(0, 0, 0, 0, SUBTYPES.GEOM2D),

                shapeType, 0, 0);

        } catch (CreateEnvelopeException e) {

            logger.error("Error creating the envelope", e);

        }

        this.shpFeatureType = featureType;

        super.begin(dbfParameters, dbfFeatureType, numRows);

    }

    private int getShapeTypeAndSetSupportedGeometries(FeatureType featureType) {

        FeatureAttributeDescriptor geometryAttr =

            featureType.getAttributeDescriptor(featureType

                .getDefaultGeometryAttributeIndex());

        int gvSIG_geometryType = geometryAttr.getGeometryType();

        int gvSIG_geometrySubType = geometryAttr.getGeometrySubType();

        this.setSupportedGeometryTypes(gvSIG_geometryType);

        int shapeType = 0;

        shapeType =

            shpWrite.getShapeType(gvSIG_geometryType, gvSIG_geometrySubType);

        return shapeType;

    }

    public void dispose() {

        super.dispose();

        this.envelope = null;

        this.shpWrite = null;

    }

    public void end() throws DataException {

        if (envelope == null) {

            try {

                envelope =

                    geomManager.createEnvelope(0, 0, 0, 0, SUBTYPES.GEOM2D);

            } catch (CreateEnvelopeException e) {

                logger.error("Error creating the envelope", e);

            }

        }

        int shapeType = getShapeTypeAndSetSupportedGeometries(shpFeatureType);

        shpWrite.writeHeaders(envelope, shapeType, super.getRowCount(),

            fileSize);

        super.end();

        shpWrite.close();

    }

    public void append(Feature feature) throws DataException {

        Geometry theGeom = feature.getDefaultGeometry();

        if (theGeom == null) {

            theGeom = this.defaultGeometry;

        }

        if (!canWriteGeometry(theGeom.getType())) {

            throw new WriteException(this.name, // FIXME Excepcion correcta

                new RuntimeException("UnsupportedGeometryType: "

                    + theGeom.getGeometryType().getName()));

        }

        // numRows++;

        super.append(feature);

        // fileSize = shpWrite.writeIGeometry(theGeom);

        fileSize = shpWrite.writeIGeometry(theGeom);

        Envelope envelope = theGeom.getEnvelope();

        if (envelope != null) {

            if (this.envelope != null) {

                this.envelope.add(envelope);

            } else {

                this.envelope = envelope;

            }

        }

    }

    private void setSupportedGeometryTypes(int gvSIG_geometryType) {

        switch (gvSIG_geometryType) {

        case Geometry.TYPES.POINT:

            supportedGeometryTypes =

                new int[] { Geometry.TYPES.POINT, Geometry.TYPES.NULL };

            break;

        case Geometry.TYPES.MULTIPOINT:

            supportedGeometryTypes =

                new int[] { Geometry.TYPES.MULTIPOINT, Geometry.TYPES.NULL };

            break;

        case Geometry.TYPES.MULTICURVE:

            supportedGeometryTypes =

                new int[] { Geometry.TYPES.CURVE, Geometry.TYPES.ELLIPSE,

                    Geometry.TYPES.ARC, Geometry.TYPES.CIRCLE,

                    Geometry.TYPES.ELLIPSE, Geometry.TYPES.MULTICURVE,

                    Geometry.TYPES.SPLINE, Geometry.TYPES.NULL };

            break;

        case Geometry.TYPES.MULTISURFACE:

            supportedGeometryTypes =

                new int[] { Geometry.TYPES.SURFACE, Geometry.TYPES.CIRCLE,

                    Geometry.TYPES.ELLIPSE, Geometry.TYPES.MULTISURFACE,

                    Geometry.TYPES.SPLINE, Geometry.TYPES.NULL };

            break;

        default:

            supportedGeometryTypes = new int[] {};

        }

    }

    public boolean canWriteGeometry(int gvSIGgeometryType) {

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

            if (gvSIGgeometryType == supportedGeometryTypes[i]) {

                return true;

            }

        }

        return false;

    }

    public void begin(DBFStoreParameters storeParameters,

        FeatureType featureType, long numRows) throws DataException {

        throw new UnsupportedOperationException();

    }

}