Application: gvSIG desktop

package com.iver.cit.gvsig.fmap.drivers.shp;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.nio.ByteBuffer;

import java.nio.ByteOrder;

import java.nio.channels.FileChannel;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.GeneralPathX;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.core.ShapeFactory;

import com.iver.cit.gvsig.fmap.drivers.BoundedShapes;

import com.iver.cit.gvsig.fmap.drivers.ExternalData;

import com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver;

/**

 * DOCUMENT ME!

 *

 * @author $author$

 */

public class DemoSHPDriver implements VectorialFileDriver, BoundedShapes, ExternalData {

    private File file;

    private ByteBuffer bb;

    private FileChannel channel;

    private FileInputStream fin;

    private int type;

    private int[] m_posShapes;

    private int numReg;

    private Rectangle2D extent;

    /**

     * DOCUMENT ME!

     *

     * @throws IOException

     *

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#close()

     */

    public void close() throws IOException {

        IOException ret = null;

        try {

            channel.close();

        } catch (IOException e) {

            ret = e;

        } finally {

            try {

                fin.close();

            } catch (IOException e1) {

                ret = e1;

            }

        }

        if (ret != null) {

            throw ret;

        }

    }

    /**

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#open(java.io.File)

     */

    public void open(File f) throws IOException {

        file = f;

        fin = new FileInputStream(f);

        // Open the file and then get a channel from the stream

        channel = fin.getChannel();

        long size = channel.size();

        // Get the file's size and then map it into memory

        bb = channel.map(FileChannel.MapMode.READ_ONLY, 0, size);

    }

    /**

     * DOCUMENT ME!

     *

     * @param index DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws IOException

     *

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getShape(int)

     */

    /* public FShape getShapeByID(int ID) {

       Point2D.Double p = new Point2D.Double();

       Point2D.Double pAnt = null;

       int numParts;

       int numPoints;

       int i;

       int j;

       int numReg;

       int numeroPuntos;

       int hasta;

       int desde;

       Rectangle2D.Double BoundingBox = new Rectangle2D.Double();

               SHPShape shapeShp=null;

       FShape resulShape = null;

       try {

           bb.position(m_posShapes[ID]);

           bb.order(ByteOrder.LITTLE_ENDIAN);

           int tipoShape = bb.getInt();

           m_shapeType = tipoShape;

           // retrieve that shape.

           // tempRecord.setShape(readShape(tempShapeType, tempContentLength, in));

           if (tipoShape == FConstant.SHAPE_TYPE_POINT) {

               p = readPoint(bb);

               resulShape = new FShape(new FPoint(p.getX(),p.getY()),FConstant.SHAPE_TYPE_POINT);

               //Comprobaci?n punto.

               //System.err.println("p.x = "+p.x);

               //System.err.println("p.y = "+p.y);

           } else if ((tipoShape == FConstant.SHAPE_TYPE_POLYLINE) ||

                   (tipoShape == FConstant.SHAPE_TYPE_POLYGON)) {

               // BoundingBox

               BoundingBox = readRectangle(bb);

               numParts = bb.getInt();

               numPoints = bb.getInt();

               // part indexes.

               // Geometry geom = GeometryFactory.toGeometryArray();

               GeneralPathX elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD,

                       numPoints);

               int[] tempParts = new int[numParts];

               for (i = 0; i < numParts; i++) {

                   tempParts[i] = bb.getInt();

               }

               j = 0;

               ///Line2D.Double line2D;

               FPoint[] points=new FPoint[numPoints];

               for (i = 0; i < numPoints; i++) {

                       p=readPoint(bb);

                   points[i] = new FPoint(p.x,p.y);

                   // System.out.println("x= " + p.x + " y=" + p.y);

                   // System.out.println("x= " + (float) p.x + " y=" + (float) p.y);

                   if (i == tempParts[j]) {

                       elShape.moveTo(p.x, p.y);

                       if (j < (numParts - 1)) {

                           j++;

                       }

                   } else {

                       elShape.lineTo(p.x, p.y);

                   }

               }

               //FGeometry pol=new FPolyLine(points,tempParts,BoundingBox);

               resulShape = new FShape(tipoShape,elShape);

           } else if (tipoShape == FConstant.SHAPE_TYPE_MULTIPOINT) {

               // BoundingBox

               BoundingBox = readRectangle(bb);

               numPoints = bb.getInt();

               FPoint[] tempPoints = new FPoint[numPoints];

               for (i = 0; i < numPoints; i++) {

                       Point2D p2=readPoint(bb);

                   tempPoints[i] = new FPoint(p2.getX(),p2.getY(),0);

               }

               FMultiPoint multipoint = new FMultiPoint(tempPoints,BoundingBox);

               resulShape = new FShape(multipoint,tipoShape);

           } else if (tipoShape == FConstant.SHAPE_TYPE_POINTZ) {

               FPoint p3d = new FPoint();

               p3d.read(bb);

               resulShape = new FShape(p3d,tipoShape);

           } else if ((tipoShape == FConstant.SHAPE_TYPE_POLYLINEZ) ||

                   (tipoShape == FConstant.SHAPE_TYPE_POLYGONZ)) {

               // BoundingBox

               BoundingBox = readRectangle(bb);

               numParts = bb.getInt();

               numPoints = bb.getInt();

               // part indexes.

               // Geometry geom = GeometryFactory.toGeometryArray();

               GeneralPathX elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD,

                       numPoints);

               int[] tempParts = new int[numParts];

               for (i = 0; i < numParts; i++) {

                   tempParts[i] = bb.getInt();

               }

               j = 0;

              //Line2D.Double line2D;

               FPoint[] points=new FPoint[numPoints];

               for (i = 0; i < numPoints; i++) {

                   p = readPoint(bb);

                   points[i]=new FPoint(p.x,p.y);

                   if (i == tempParts[j]) {

                       elShape.moveTo(p.x, p.y);

                       if (j < (numParts - 1)) {

                           j++;

                       }

                   } else {

                       elShape.lineTo(p.x, p.y);

                   }

               }

               double[] boxZ = new double[2];

               boxZ[0] = bb.getDouble();

               boxZ[1] = bb.getDouble();

               double[] pZ = new double[numPoints];

               for (i = 0; i < numPoints; i++) {

                   pZ[i] = bb.getDouble();

               }

               //FGeometry pol=new FPolyLine(points,tempParts,BoundingBox);

               resulShape = new FShape(tipoShape, elShape, pZ);

           } else if (tipoShape == FConstant.SHAPE_TYPE_MULTIPOINTZ) {

               // BoundingBox

               BoundingBox = readRectangle(bb);

               numPoints = bb.getInt();

               FPoint[] tempPoints3D = new FPoint[numPoints];

               for (i = 0; i < numPoints; i++) {

                   tempPoints3D[i] = new FPoint();

                   tempPoints3D[i].read(bb);

               }

               FMultiPoint multipoint3D = new FMultiPoint(tempPoints3D,BoundingBox);

               resulShape = new FShape(multipoint3D,tipoShape);

           }

       } catch (Exception e) {

           System.err.println("Fallo en getShapeByID. ID=" + ID +

               " m_posShapes[ID]=" + m_posShapes[ID]);

           System.err.println("getShapeByID: " + e.getMessage());

           e.printStackTrace();

       }

       return resulShape;

       }

     */

    public IGeometry getShape(int index) throws IOException {

        Point2D.Double p = new Point2D.Double();

        Point2D.Double pAnt = null;

        int numParts;

        int numPoints;

        int i;

        int j;

        int numReg;

        int numeroPuntos;

        int hasta;

        int desde;

        int shapeType;

        //Rectangle2D.Double BoundingBox;

        bb.position(m_posShapes[index]);

        bb.order(ByteOrder.LITTLE_ENDIAN);

        ///bb.position(bb.position()+4);

        shapeType = bb.getInt();

        // retrieve that shape.

        // tempRecord.setShape(readShape(tempShapeType, tempContentLength, in));

        switch (type) {

            case (SHP.POINT2D):

                p = readPoint(bb);

                return ShapeFactory.createPoint2D(p.getX(), p.getY());

            case (SHP.POLYLINE2D):

                //BoundingBox = readRectangle(bb);

                //bb.getDouble();

                //bb.getDouble();

                //bb.getDouble();

                //bb.getDouble();

                bb.position(bb.position() + 32);

                numParts = bb.getInt();

                numPoints = bb.getInt();

                // part indexes.

                // Geometry geom = GeometryFactory.toGeometryArray();

                GeneralPathX elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD,

                        numPoints);

                int[] tempParts = new int[numParts];

                for (i = 0; i < numParts; i++) {

                    tempParts[i] = bb.getInt();

                }

                j = 0;

                for (i = 0; i < numPoints; i++) {

                    p = readPoint(bb);

                    if (i == tempParts[j]) {

                        elShape.moveTo(p.x, p.y);

                        if (j < (numParts - 1)) {

                            j++;

                        }

                    } else {

                        elShape.lineTo(p.x, p.y);

                    }

                }

                return ShapeFactory.createPolyline2D(elShape);

            case (SHP.POLYGON2D):

                //                            BoundingBox = readRectangle(bb);

                bb.getDouble();

                bb.getDouble();

                bb.getDouble();

                bb.getDouble();

                numParts = bb.getInt();

                numPoints = bb.getInt();

                // part indexes.

                // Geometry geom = GeometryFactory.toGeometryArray();

                elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD, numPoints);

                tempParts = new int[numParts];

                for (i = 0; i < numParts; i++) {

                    tempParts[i] = bb.getInt();

                }

                j = 0;

                for (i = 0; i < numPoints; i++) {

                    p = readPoint(bb);

                    if (i == tempParts[j]) {

                        elShape.moveTo(p.x, p.y);

                        if (j < (numParts - 1)) {

                            j++;

                        }

                    } else {

                        elShape.lineTo(p.x, p.y);

                    }

                }

                return ShapeFactory.createPolygon2D(elShape);

            case (SHP.POINT3D):

                double x = bb.getDouble();

                double y = bb.getDouble();

                double z = bb.getDouble();

                return ShapeFactory.createPoint3D(x, y, z);

            case (SHP.POLYLINE3D):

                bb.position(bb.position() + 32);

                numParts = bb.getInt();

                numPoints = bb.getInt();

                elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD, numPoints);

                tempParts = new int[numParts];

                for (i = 0; i < numParts; i++) {

                    tempParts[i] = bb.getInt();

                }

                j = 0;

                for (i = 0; i < numPoints; i++) {

                    p = readPoint(bb);

                    if (i == tempParts[j]) {

                        elShape.moveTo(p.x, p.y);

                        if (j < (numParts - 1)) {

                            j++;

                        }

                    } else {

                        elShape.lineTo(p.x, p.y);

                    }

                }

                double[] boxZ = new double[2];

                boxZ[0] = bb.getDouble();

                boxZ[1] = bb.getDouble();

                double[] pZ = new double[numPoints];

                for (i = 0; i < numPoints; i++) {

                    pZ[i] = bb.getDouble();

                }

                return ShapeFactory.createPolyline3D(elShape, pZ);

            case (SHP.MULTIPOINT2D):

                bb.position(bb.position() + 32);

                numPoints = bb.getInt();

                double[] tempX = new double[numPoints];

                double[] tempY = new double[numPoints];

                for (i = 0; i < numPoints; i++) {

                    tempX[i] = bb.getDouble();

                    tempY[i] = bb.getDouble();

                }

                return ShapeFactory.createMultipoint2D(tempX, tempY);

            case (SHP.MULTIPOINT3D):

                bb.position(bb.position() + 32);

                numPoints = bb.getInt();

                double[] temX = new double[numPoints];

                double[] temY = new double[numPoints];

                double[] temZ = new double[numPoints];

                for (i = 0; i < numPoints; i++) {

                    temX[i] = bb.getDouble();

                    temY[i] = bb.getDouble();

                    temZ[i] = bb.getDouble();

                }

                return ShapeFactory.createMultipoint3D(temX, temY, temZ);

        }

        return null;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getShapeCount()

     */

    public int getShapeCount() {

        return numReg;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getShapeType()

     */

    public int getShapeType() {

        int auxType = 0;

        switch (type) {

            case (SHP.POINT2D):

                auxType = auxType | FShape.POINT;

                break;

            case (SHP.POLYLINE2D):

                auxType = auxType | FShape.LINE;

                break;

            case (SHP.POLYGON2D):

                auxType = auxType | FShape.POLYGON;

                break;

            case (SHP.POINT3D):

                auxType = auxType | FShape.POINT;

                    break;

            case (SHP.POLYLINE3D):

                auxType = auxType | FShape.LINE;

                    break;

            case (SHP.MULTIPOINT2D):

                auxType = auxType | FShape.MULTI;

                    break;

            case (SHP.MULTIPOINT3D):

                auxType = auxType | FShape.MULTI;

                    break;

        }

        return auxType;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#initialize()

     */

    public void initialize() {

        try {

            // create a new header.

            ShapeFileHeader myHeader = new ShapeFileHeader();

            // read the header

            myHeader.readHeader(bb);

            extent = new Rectangle2D.Double(myHeader.myXmin, myHeader.myYmin,

                    myHeader.myXmax - myHeader.myXmin,

                    myHeader.myYmax - myHeader.myYmin);

            type = myHeader.myShapeType;

            double x = myHeader.myXmin;

            double y = myHeader.myYmin;

            double w = myHeader.myXmax - myHeader.myXmin;

            double h = myHeader.myYmax - myHeader.myYmin;

            if (w == 0) {

                x -= 0.1;

                w = 0.2;

            }

            if (h == 0) {

                y -= 0.1;

                h = 0.2;

            }

            // String strFichDbf = m_Path.toLowerCase().replaceAll("\\.shp", ".dbf");

            String strFichDbf = file.getAbsolutePath().replaceAll("\\.shp",

                    ".dbf");

            strFichDbf = strFichDbf.replaceAll("\\.SHP", ".DBF");

            DbaseFileNIO m_FichDbf = new DbaseFileNIO();

            m_FichDbf.open(new File(strFichDbf));

            numReg = m_FichDbf.getRecordCount();

            m_posShapes = new int[numReg];

            // read the records.

            int tempCurrentLength = myHeader.getHeaderLength();

            int numReg = 0;

            int pos1;

            while (tempCurrentLength < myHeader.myFileLength) {

                // read the record header

                // ShapeFileRecord tempRecord = new ShapeFileRecord();

                // Bytes 0 to 4 represent the record number in the file, these may be out of order.

                bb.order(ByteOrder.BIG_ENDIAN);

                // tempRecord.setIndex(in.readInt());

                bb.getInt();

                // read the content length of this record in 16 bit words, excluding the index.

                // in.setLittleEndianMode(false);

                int tempContentLength = bb.getInt();

                pos1 = bb.position();

                m_posShapes[numReg] = bb.position();

                // Posicionamos

                bb.position((pos1 + (2 * tempContentLength)));

                numReg = numReg + 1;

                // update the current length the 4 is for the index, and content length.

                tempCurrentLength = tempCurrentLength + 4 + tempContentLength;

            }

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

    /**

     * Reads the Point from the shape file.

     *

     * @param in DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    private Point2D.Double readPoint(ByteBuffer in) {

        // create a new point

        Point2D.Double tempPoint = new Point2D.Double();

        // bytes 1 to 4 are the type and have already been read.

        // bytes 4 to 12 are the X coordinate

        in.order(ByteOrder.LITTLE_ENDIAN);

        tempPoint.setLocation(in.getDouble(), in.getDouble());

        return tempPoint;

    }

    /**

     * Lee un rect?ngulo del fichero.

     *

     * @param in DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws IOException DOCUMENT ME!

     */

    private Rectangle2D.Double readRectangle(ByteBuffer in)

        throws IOException {

        Rectangle2D.Double tempRect = new Rectangle2D.Double();

        in.order(ByteOrder.LITTLE_ENDIAN);

        tempRect.x = in.getDouble();

        tempRect.y = in.getDouble();

        tempRect.width = in.getDouble() - tempRect.x;

        if (tempRect.width == 0) {

            tempRect.width = 0.2;

            tempRect.x -= 0.1;

        }

        tempRect.height = in.getDouble() - tempRect.y;

        if (tempRect.height == 0) {

            tempRect.height = 0.2;

            tempRect.y -= 0.1;

        }

        return tempRect;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getFullExtent()

     */

    public Rectangle2D getFullExtent() throws IOException {

        return extent;

    }

    /**

     * DOCUMENT ME!

     *

     * @param index DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws IOException

     *

     * @see com.iver.cit.gvsig.fmap.drivers.BoundedShapes#getShapeBounds()

     */

    public Rectangle2D getShapeBounds(int index) throws IOException {

        Point2D p = new Point2D.Double();

        Rectangle2D BoundingBox = new Rectangle2D.Double();

        bb.position(m_posShapes[index]);

        bb.order(ByteOrder.LITTLE_ENDIAN);

        int tipoShape = bb.getInt();

        if (tipoShape != SHP.NULL) {

            type = tipoShape;

        }

        // retrieve that shape.

        // tempRecord.setShape(readShape(tempShapeType, tempContentLength, in));

        switch (tipoShape) {

            case (SHP.POINT2D):

            case (SHP.POINT3D):

                p = readPoint(bb);

                BoundingBox = new Rectangle2D.Double(p.getX() - 0.1,

                        p.getY() - 0.1, 0.2, 0.2);

                break;

            case (SHP.POLYLINE2D):

            case (SHP.POLYGON2D):

            case (SHP.MULTIPOINT2D):

            case (SHP.POLYLINE3D):

            case (SHP.POLYGON3D):

            case (SHP.MULTIPOINT3D):

                // BoundingBox

                BoundingBox = readRectangle(bb);

                break;

        }

        return BoundingBox;

    }

    /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#accept(java.io.File)

         */

        public boolean accept(File f) {

                return (f.getName().toUpperCase().endsWith("SHP"));

        }

        /**

         * @see com.hardcode.driverManager.Driver#getType()

         */

        public String getName() {

                return "gvSIG shp driver";

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getDataDriverName()

         */

        public String getDataDriverName() {

                return "gdbms dbf driver";

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getDataFile(java.io.File)

         */

        public File getDataFile(File f) {

                String str = f.getAbsolutePath();

                return new File(str.substring(0, str.length() - 3)+ "dbf");

        }

        /* (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.BoundedShapes#getShapeType(int)

         */

        public int getShapeType(int index) {

                // Por ahora todos los fichero .shp contienen

                // entidades del mismo tipo. Si trabajamos con

                // alguno mixto, tendremos que cambiar esta funci?n.

                return getShapeType();

        }

}