public static void add(Object tag, Geometry geom, String msg) {
if (!Debug.isDebugging()) return;
FeatureDataset fd = getDebugFeatureDataset(tag);
Feature feature = new BasicFeature(fd.getFeatureSchema());
feature.setGeometry(geom);
feature.setAttribute(MESG_ATTR_NAME, msg);
fd.add(feature);
}
public static Feature createLineSegmentFeature(
FeatureSchema fs, Coordinate p0, Coordinate p1, String msg) {
Feature feature = new BasicFeature(fs);
LineString lineSeg = fact.createLineString(new Coordinate[] {p0, p1});
feature.setGeometry(lineSeg);
feature.setAttribute(MESG_ATTR_NAME, msg);
return feature;
}
/**
* look at all the data in the column of the featurecollection, and find the largest string!
*
* @param fc features to look at
* @param attributeNumber which of the column to test.
*/
int findMaxStringLength(FeatureCollection fc, int attributeNumber) {
int l;
int maxlen = 0;
Feature f;
for (Iterator i = fc.iterator(); i.hasNext(); ) {
f = (Feature) i.next();
l = f.getString(attributeNumber).length();
if (l > maxlen) {
maxlen = l;
}
}
return maxlen;
}
/**
* Write a dbf file with the information from the featureCollection.
*
* @param featureCollection column data from collection
* @param fname name of the dbf file to write to
*/
void writeDbf(FeatureCollection featureCollection, String fname) throws Exception {
DbfFileWriter dbf;
FeatureSchema fs;
int t;
int f;
int u;
int num;
fs = featureCollection.getFeatureSchema();
// -1 because one of the columns is geometry
DbfFieldDef[] fields = new DbfFieldDef[fs.getAttributeCount() - 1];
// dbf column type and size
f = 0;
for (t = 0; t < fs.getAttributeCount(); t++) {
AttributeType columnType = fs.getAttributeType(t);
String columnName = fs.getAttributeName(t);
if (columnType == AttributeType.INTEGER) {
fields[f] = new DbfFieldDef(columnName, 'N', 16, 0);
f++;
} else if (columnType == AttributeType.DOUBLE) {
fields[f] = new DbfFieldDef(columnName, 'N', 33, 16);
f++;
} else if (columnType == AttributeType.STRING) {
int maxlength = findMaxStringLength(featureCollection, t);
if (maxlength > 255) {
throw new Exception(
"ShapefileWriter does not support strings longer than 255 characters");
}
fields[f] = new DbfFieldDef(columnName, 'C', maxlength, 0);
f++;
} else if (columnType == AttributeType.DATE) {
fields[f] = new DbfFieldDef(columnName, 'D', 8, 0);
f++;
} else if (columnType == AttributeType.GEOMETRY) {
// do nothing - the .shp file handles this
} else {
throw new Exception("Shapewriter: unsupported AttributeType found in featurecollection.");
}
}
// write header
dbf = new DbfFileWriter(fname);
dbf.writeHeader(fields, featureCollection.size());
// write rows
num = featureCollection.size();
List features = featureCollection.getFeatures();
for (t = 0; t < num; t++) {
// System.out.println("dbf: record "+t);
Feature feature = (Feature) features.get(t);
Vector DBFrow = new Vector();
// make data for each column in this feature (row)
for (u = 0; u < fs.getAttributeCount(); u++) {
AttributeType columnType = fs.getAttributeType(u);
if (columnType == AttributeType.INTEGER) {
Object a = feature.getAttribute(u);
if (a == null) {
DBFrow.add(new Integer(0));
} else {
DBFrow.add((Integer) a);
}
} else if (columnType == AttributeType.DOUBLE) {
Object a = feature.getAttribute(u);
if (a == null) {
DBFrow.add(new Double(0.0));
} else {
DBFrow.add((Double) a);
}
} else if (columnType == AttributeType.DATE) {
Object a = feature.getAttribute(u);
if (a == null) {
DBFrow.add("");
} else {
DBFrow.add(DbfFile.DATE_PARSER.format((Date) a));
}
} else if (columnType == AttributeType.STRING) {
Object a = feature.getAttribute(u);
if (a == null) {
DBFrow.add(new String(""));
} else {
// MD 16 jan 03 - added some defensive programming
if (a instanceof String) {
DBFrow.add(a);
} else {
DBFrow.add(a.toString());
}
}
}
}
dbf.writeRecord(DBFrow);
}
dbf.close();
}