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   {{Task}}

*/

/**

 *

 */

package org.gvsig.data.datastores.vectorial.file.shp;

import java.awt.geom.Point2D;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.nio.ByteOrder;

import java.nio.channels.FileChannel;

import org.gvsig.data.datastores.vectorial.file.FileResource;

import org.gvsig.data.datastores.vectorial.file.shp.utils.SHP;

import org.gvsig.data.datastores.vectorial.file.shp.utils.ShapeFileHeader2;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.spatialprovisional.Extent;

import org.gvsig.data.spatialprovisional.IExtent;

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

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.utiles.bigfile.BigByteBuffer2;

/**

 * @author jmvivo

 *

 */

public class SHPResource extends FileResource {

        private boolean firstOpen = true;

        private Extent extent;

        private int type;

        private SHPStoreParameters params;

        private FileInputStream fin;

        private FileChannel channel;

        private BigByteBuffer2 bb;

        private File fileShx;

        private FileInputStream finShx;

        private FileChannel channelShx;

        private BigByteBuffer2 bbShx;

        SHPResource (SHPStoreParameters params){

                super(params.getSHPFile());

                this.params = params;

                this.fileShx = params.getSHXFile();

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.Resource#description()

         */

        public String description() {

                if (this.getFile()!= null){

                        return "SHP File resource: " + this.getFile().getAbsolutePath();

                } else{

                        return "SHP File resource: {No file}";

                }

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.Resource#getKey()

         */

        public String getKey() {

                return "SHP:"+this.getFile().getAbsolutePath();

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.datastores.vectorial.file.IFileResource#getTotalFiles()

         */

        public int getTotalFiles() {

                return 2;

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.Resource#doOpen()

         */

        protected boolean doOpen() throws OpenException {

                if (this.firstOpen){

                        this.updateLastModified();

                } else{

                        this.checkChanged();

                }

                this.openShp();

                this.setOpened();

                return true;

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.Resource#doClose()

         */

        protected boolean doClose() throws CloseException{

                CloseException ret = null;

                try {

                        channel.close();

                        channelShx.close();

                } catch (IOException e) {

                        ret = new CloseException(this.description(),e);

                } finally {

                        try {

                                fin.close();

                                finShx.close();

                        } catch (IOException e1) {

                                ret = new CloseException(this.description(),e1);

                        }

                }

                if (ret != null) {

                        throw ret;

                }

                bb = null;

                bbShx = null;

                return super.doClose();

        }

        private void openShp() throws OpenException{

                try {

                        fin = new FileInputStream(this.getFile());

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

                        bb = new BigByteBuffer2(channel, FileChannel.MapMode.READ_ONLY);

                        finShx = new FileInputStream(this.fileShx);

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

                        channelShx = finShx.getChannel();

//                        long sizeShx = channelShx.size();

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

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

                        // bbShx = channelShx.map(FileChannel.MapMode.READ_ONLY, 0, sizeShx);

                        bbShx = new BigByteBuffer2(channelShx, FileChannel.MapMode.READ_ONLY);

                        bbShx.order(ByteOrder.BIG_ENDIAN);

//                        create a new header.

                        ShapeFileHeader2 myHeader = new ShapeFileHeader2();

                        bb.position(0);

                        // read the header

                        myHeader.readHeader(bb);

                        extent = new Extent(myHeader.myXmin, myHeader.myYmin,myHeader.myXmax,myHeader.myYmax);

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

                        }

                } catch (FileNotFoundException e) {

                        throw new OpenException(this.description(),e);

                } catch (IOException e) {

                        throw new OpenException(this.description(),e);

                }

        }

        protected IExtent getFullExtent() throws ReadException{

                this.checkOpen();

                return this.extent;

        }

        protected boolean isEditable(){

                return this.getFile().canWrite() && this.fileShx.canWrite();

        }

        protected int getGeometryType() throws ReadException {

                this.checkOpen();

                int auxType = 0;

                switch (type) {

                case (SHP.POINT2D):

                case (SHP.POINT3D):

                        auxType = auxType | FShape.POINT;

                break;

                case (SHP.POLYLINE2D):

                case (SHP.POLYLINE3D):

                        auxType = auxType | FShape.LINE;

                break;

                case (SHP.POLYGON2D):

                case (SHP.POLYGON3D):

                        auxType = auxType | FShape.POLYGON;

                break;

                case (SHP.MULTIPOINT2D):

                case (SHP.MULTIPOINT3D):

                        auxType = auxType | FShape.MULTIPOINT;

                break;

                }

                return auxType;

        }

        protected synchronized IGeometry getGeometry(long position) throws ReadException {

                this.checkOpen();

                Point2D p = new Point2D.Double();

                int numParts;

                int numPoints;

                int i;

                int j;

                int shapeType;

                bb.position(getPositionForRecord(position));

                bb.order(ByteOrder.LITTLE_ENDIAN);

                shapeType = bb.getInt();

                //el shape tal con tema tal y n?mro tal es null

                if (shapeType==SHP.NULL){

                        return new FNullGeometry();

                }

                // 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):

                        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.getX(), p.getY());

                                if (j < (numParts - 1)) {

                                        j++;

                                }

                        } else {

                                elShape.lineTo(p.getX(), p.getY());

                        }

                }

                return ShapeFactory.createPolyline2D(elShape);

                case (SHP.POLYGON2D):

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

                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.getX(), p.getY());

                                if (j < (numParts - 1)) {

                                        j++;

                                }

                        } else {

                                if (i==numPoints-1){

                                        elShape.closePath();

                                }else{

                                        elShape.lineTo(p.getX(), p.getY());

                                }

                        }

                }

                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.getX(), p.getY());

                                if (j < (numParts - 1)) {

                                        j++;

                                }

                        } else {

                                elShape.lineTo(p.getX(), p.getY());

                        }

                }

                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.POLYGON3D):

                        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.getX(), p.getY());

                                if (j < (numParts - 1)) {

                                        j++;

                                }

                        } else {

                                if (i==numPoints-1){

                                        elShape.closePath();

                                }else{

                                        elShape.lineTo(p.getX(), p.getY());

                                }

                        }

                }

                double[] boxpoZ = new double[2];

                boxpoZ[0] = bb.getDouble();

                boxpoZ[1] = bb.getDouble();

                double[] poZ = new double[numPoints];

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

                        poZ[i] = bb.getDouble();

                }

                return ShapeFactory.createPolygon3D(elShape, poZ);

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

                }

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

                        temZ[i] = bb.getDouble();

                }

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

                }

                return null;

        }

        private long getPositionForRecord(long numRec)

        {

                // shx file has a 100 bytes header, then, records

                // 8 bytes length, one for each entity.

                // first 4 bytes are the offset

                // next 4 bytes are length

                int posIndex = 100 + ((int)numRec * 8);

                // bbShx.position(posIndex);

                long pos = 8 + 2* bbShx.getInt(posIndex);

                return pos;

        }

        /**

         * Reads the Point from the shape file.

         *

         * @param in ByteBuffer.

         *

         * @return Point2D.

         */

        private synchronized Point2D readPoint(BigByteBuffer2 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 ByteBuffer.

         *

         * @return Rect?ngulo.

         *

         * @throws IOException

         */

        private synchronized IExtent readRectangle(BigByteBuffer2 in){

                in.order(ByteOrder.LITTLE_ENDIAN);

                double x = in.getDouble();

                double y = in.getDouble();

                double x2 = in.getDouble();

                if (x2-x == 0) {

                        x2 = 0.2;

                        x -= 0.1;

                }

                double y2 = in.getDouble();

                if (y2-y == 0) {

                        y2 = 0.2;

                        y -= 0.1;

                }

                IExtent tempRect = new Extent(x,y,x2,y2);

                return tempRect;

        }

        public IExtent getBoundingBox(long featureIndex) throws ReadException {

                this.checkOpen();

                Point2D p = new Point2D.Double();

                IExtent BoundingBox = null;

                try {

                        bb.position(getPositionForRecord(featureIndex));

                }catch (Exception e) {

                        e.printStackTrace();

//                        logger.error(" Shapefile is corrupted. Drawing aborted. ="+e+ "  "+"index = "+index);

                }

                bb.order(ByteOrder.LITTLE_ENDIAN);

                int tipoShape = bb.getInt();

                //AZABALA: si tipoShape viene con valores erroneos deja de funcionar

                //el metodo getShape(i)

//                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 Extent(p.getX() - 0.1, p.getY() - 0.1,p.getX()+0.2,p.getY()+0.2);

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

        }

        /**

         * @return

         */

        public String getSRS() {

                // TODO Auto-generated method stub

                return null;

        }

}