protected void write(FeatureCollectionResponse featureCollection, OutputStream output,
Operation getFeature) throws IOException ,ServiceException {
// Create the workbook
Workbook wb = getNewWorkbook();
CreationHelper helper = wb.getCreationHelper();
ExcelCellStyles styles = new ExcelCellStyles(wb);
for (Iterator it = featureCollection.getFeature().iterator(); it.hasNext();) {
SimpleFeatureCollection fc = (SimpleFeatureCollection) it.next();
// create the sheet for this feature collection
Sheet sheet = wb.createSheet(fc.getSchema().getTypeName());
// write out the header
Row header = sheet.createRow(0);
SimpleFeatureType ft = fc.getSchema();
Cell cell;
cell = header.createCell(0);
cell.setCellValue(helper.createRichTextString("FID"));
for (int i = 0; i < ft.getAttributeCount() && i < colLimit; i++) {
AttributeDescriptor ad = ft.getDescriptor(i);
cell = header.createCell(i + 1);
cell.setCellValue(helper.createRichTextString(ad.getLocalName()));
cell.setCellStyle(styles.getHeaderStyle());
}
// write out the features
SimpleFeatureIterator i = fc.features();
int r = 0; // row index
try {
Row row;
while (i.hasNext()) {
r++; // start at 1, since header is at 0
row = sheet.createRow(r);
cell = row.createCell(0);
if (r == (rowLimit - 1) && i.hasNext()) {
// there are more features than rows available in this
// Excel format. write out a warning line and break
RichTextString rowWarning = helper.createRichTextString(TRUNCATE_WARNING
+ ": ROWS " + r + " - " + fc.size() + " NOT SHOWN");
cell.setCellValue(rowWarning);
cell.setCellStyle(styles.getWarningStyle());
break;
}
SimpleFeature f = i.next();
cell.setCellValue(helper.createRichTextString(f.getID()));
for (int j = 0; j < f.getAttributeCount() && j < colLimit; j++) {
Object att = f.getAttribute(j);
if (att != null) {
cell = row.createCell(j + 1);
if (att instanceof Number) {
cell.setCellValue(((Number) att).doubleValue());
} else if (att instanceof Date) {
cell.setCellValue((Date) att);
cell.setCellStyle(styles.getDateStyle());
} else if (att instanceof Calendar) {
cell.setCellValue((Calendar) att);
cell.setCellStyle(styles.getDateStyle());
} else if (att instanceof Boolean) {
cell.setCellValue((Boolean) att);
} else {
// ok, it seems we have no better way than dump it as a string
String stringVal = att.toString();
// if string length > excel cell limit, truncate it and warn the
// user, otherwise excel workbook will be corrupted
if (stringVal.length() > CELL_CHAR_LIMIT) {
stringVal = TRUNCATE_WARNING
+ " "
+ stringVal.substring(0, CELL_CHAR_LIMIT
- TRUNCATE_WARNING.length() - 1);
cell.setCellStyle(styles.getWarningStyle());
}
cell.setCellValue(helper.createRichTextString(stringVal));
}
}
}
}