package org.geotools.shapefile;
import java.io.IOException;
import jeql.io.EndianDataInputStream;
import jeql.io.EndianDataOutputStream;
import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Envelope;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.LineString;
import com.vividsolutions.jts.geom.MultiLineString;
import com.vividsolutions.jts.geom.PrecisionModel;
/**
* Wrapper for a Shapefile arc.
*/
public class MultiLineHandler implements ShapeHandler{
int myShapeType= -1;
public MultiLineHandler()
{
myShapeType = 3;
}
public MultiLineHandler(int type) throws InvalidShapefileException
{
if ( (type != 3) && (type != 13) && (type != 23) )
throw new InvalidShapefileException("MultiLineHandler constructor - expected type to be 3,13 or 23");
myShapeType = type;
}
public Geometry read( EndianDataInputStream file , GeometryFactory geometryFactory, int contentLength) throws IOException,InvalidShapefileException
{
double junk;
int actualReadWords = 0; //actual number of words read (word = 16bits)
//file.setLittleEndianMode(true);
int shapeType = file.readIntLE();
actualReadWords += 2;
if (shapeType ==0)
{
return new MultiLineString(null,new PrecisionModel(),0); //null shape
}
if (shapeType != myShapeType)
{
throw new InvalidShapefileException("MultilineHandler.read() - file says its type "+shapeType+" but i'm expecting type "+myShapeType);
}
//read bounding box (not needed)
junk = file.readDoubleLE();
junk =file.readDoubleLE();
junk =file.readDoubleLE();
junk =file.readDoubleLE();
actualReadWords += 4*4;
int numParts = file.readIntLE();
int numPoints = file.readIntLE();//total number of points
actualReadWords += 4;
int[] partOffsets = new int[numParts];
//points = new Coordinate[numPoints];
for ( int i = 0; i < numParts; i++ ){
partOffsets[i]=file.readIntLE();
actualReadWords += 2;
}
LineString lines[] = new LineString[numParts];
Coordinate[] coords = new Coordinate[numPoints];
for (int t =0;t<numPoints; t++)
{
coords[t] = new Coordinate(file.readDoubleLE(),file.readDoubleLE());
actualReadWords += 8;
}
if (myShapeType ==13)
{
junk =file.readDoubleLE(); //z min, max
junk =file.readDoubleLE();
actualReadWords += 8;
for (int t =0;t<numPoints; t++)
{
coords[t].z = file.readDoubleLE(); //z value
actualReadWords += 4;
}
}
if (myShapeType >=13)
{
// int fullLength = 22 + 2*numParts + (numPoints * 8) + 4+4+4*numPoints+ 4+4+4*numPoints;
int fullLength;
if (myShapeType == 13)
{
//polylineZ (with M)
fullLength = 22 + 2*numParts + (numPoints * 8) + 4+4+4*numPoints+ 4+4+4*numPoints;
}
else
{
// polylineM (with M)
fullLength = 22 + 2*numParts + (numPoints * 8) + 4+4+4*numPoints;
}
if (contentLength >= fullLength) //are ms actually there?
{
junk =file.readDoubleLE(); //m min, max
junk =file.readDoubleLE();
actualReadWords += 8;
for (int t =0;t<numPoints; t++)
{
junk =file.readDoubleLE(); //m value
actualReadWords += 4;
}
}
}
//verify that we have read everything we need
while (actualReadWords < contentLength)
{
int junk2 = file.readShortBE();
actualReadWords += 1;
}
int offset = 0;
int start,finish,length;
for(int part=0;part<numParts;part++){
start = partOffsets[part];
if(part == numParts-1)
{
finish = numPoints;
}
else {
finish=partOffsets[part+1];
}
length = finish-start;
Coordinate points[] = new Coordinate[length];
for(int i=0;i<length;i++){
points[i]=coords[offset];
offset++;
}
lines[part] = geometryFactory.createLineString(points);
}
if (numParts ==1)
return lines[0];
else
return geometryFactory.createMultiLineString(lines);
}
public void write(Geometry geometry,EndianDataOutputStream file)throws IOException{
// MultiLineString multi = (MultiLineString)geometry;
Geometry multi = geometry;
int npoints;
Coordinate[] coords;
//file.setLittleEndianMode(true);
file.writeIntLE(getShapeType());
Envelope box = multi.getEnvelopeInternal();
file.writeDoubleLE(box.getMinX());
file.writeDoubleLE(box.getMinY());
file.writeDoubleLE(box.getMaxX());
file.writeDoubleLE(box.getMaxY());
int numParts = multi.getNumGeometries();
file.writeIntLE(numParts);
npoints= multi.getNumPoints();
file.writeIntLE(npoints);
LineString[] lines = new LineString[numParts];
int idx = 0;
for(int i = 0;i<numParts;i++){
lines[i] = (LineString)multi.getGeometryN(i);
file.writeIntLE(idx);
idx = idx + lines[i].getNumPoints();
}
coords = multi.getCoordinates();
for(int t=0;t<npoints;t++)
{
file.writeDoubleLE(coords[t].x);
file.writeDoubleLE(coords[t].y);
}
if (myShapeType == 13)
{
//z
double[] zExtreame = zMinMax(multi);
if (Double.isNaN(zExtreame[0] ))
{
file.writeDoubleLE(0.0);
file.writeDoubleLE(0.0);
}
else
{
file.writeDoubleLE(zExtreame[0]);
file.writeDoubleLE(zExtreame[1]);
}
for (int t=0;t<npoints; t++)
{
double z = coords[t].z;
if (Double.isNaN(z))
file.writeDoubleLE(0.0);
else
file.writeDoubleLE(z);
}
}
if (myShapeType >=13)
{
//m
file.writeDoubleLE(-10E40);
file.writeDoubleLE(-10E40);
for(int t=0;t<npoints;t++)
{
file.writeDoubleLE(-10E40);
}
}
}
/**
* Get the type of shape stored (Shapefile.ARC)
*/
public int getShapeType(){
return myShapeType;
}
public int getLength(Geometry geometry){
// BUG!
//MultiLineString multi = (MultiLineString) geometry;
Geometry multi = geometry;
int numlines, numpoints;
numlines = multi.getNumGeometries();
numpoints = multi.getNumPoints();
if (myShapeType == 3)
{
return 22 + 2*numlines + (numpoints * 8);
}
if (myShapeType == 23)
{
return 22 + 2*numlines + (numpoints * 8) + 4+4+4*numpoints;
}
return 22 + 2*numlines + (numpoints * 8) + 4+4+4*numpoints+ 4+4+4*numpoints;
// return 22 + 2*numlines + (numpoints * 8);
//return (44+(4*((GeometryCollection)geometry).getNumGeometries()));
}
double[] zMinMax(Geometry g)
{
double zmin,zmax;
boolean validZFound = false;
Coordinate[] cs = g.getCoordinates();
double[] result = new double[2];
zmin = Double.NaN;
zmax = Double.NaN;
double z;
for (int t=0;t<cs.length; t++)
{
z= cs[t].z ;
if (!(Double.isNaN( z ) ))
{
if (validZFound)
{
if (z < zmin)
zmin = z;
if (z > zmax)
zmax = z;
}
else
{
validZFound = true;
zmin = z ;
zmax = z ;
}
}
}
result[0] = (zmin);
result[1] = (zmax);
return result;
}
}
/*
* $Log: MultiLineHandler.java,v $
* Revision 1.4 2003/07/25 18:49:15 dblasby
* Allow "extra" data after the content. Fixes the ICI shapefile bug.
*
* Revision 1.3 2003/02/04 02:10:37 jaquino
* Feature: EditWMSQuery dialog
*
* Revision 1.2 2003/01/22 18:31:05 jaquino
* Enh: Make About Box configurable
*
* Revision 1.3 2002/10/30 22:36:11 dblasby
* Line reader now returns LINESTRING(..) if there is only one part to the arc
* polyline.
*
* Revision 1.2 2002/09/09 20:46:22 dblasby
* Removed LEDatastream refs and replaced with EndianData[in/out]putstream
*
* Revision 1.1 2002/08/27 21:04:58 dblasby
* orginal
*
* Revision 1.2 2002/03/05 10:23:59 jmacgill
* made sure geometries were created using the factory methods
*
* Revision 1.1 2002/02/28 00:38:50 jmacgill
* Renamed files to more intuitve names
*
* Revision 1.3 2002/02/13 00:23:53 jmacgill
* First semi working JTS version of Shapefile code
*
* Revision 1.2 2002/02/11 18:42:45 jmacgill
* changed read and write statements so that they produce and take Geometry objects instead of specific MultiLine objects
* changed parts[] array name to partOffsets[] for clarity and consistency with ShapePolygon
*
* Revision 1.1 2002/02/11 16:54:43 jmacgill
* added shapefile code and directories
*
*/