public void write(DataRaster raster) throws IOException, IllegalArgumentException {
if (null == raster) {
String msg = Logging.getMessage("nullValue.RasterIsNull");
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
if (!(raster.getWidth() > 0)) {
String msg = Logging.getMessage("generic.InvalidWidth", raster.getWidth());
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
if (!(raster.getHeight() > 0)) {
String msg = Logging.getMessage("generic.InvalidHeight", raster.getHeight());
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
if (raster instanceof BufferedImageRaster) {
this.write(((BufferedImageRaster) raster).getBufferedImage(), raster);
} else if (raster instanceof BufferWrapperRaster) {
this.writeRaster((BufferWrapperRaster) raster);
}
}
public GeotiffWriter(File file) throws IOException {
if (null == file) {
String msg = Logging.getMessage("nullValue.FileIsNull");
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
commonInitializer(file);
}
public GeotiffWriter(String filename) throws IOException {
if (null == filename || 0 == filename.trim().length()) {
String msg = Logging.getMessage("generic.FileNameIsMissing");
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
// the initializer does the validity checking...
commonInitializer(new File(filename));
}
private void commonInitializer(File file) throws IOException {
File parent = file.getParentFile();
if (parent == null) parent = new File(System.getProperty("user.dir"));
if (!parent.canWrite()) {
String msg = Logging.getMessage("generic.FolderNoWritePermission", parent.getAbsolutePath());
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
this.targetFile = new RandomAccessFile(file, "rw");
this.theChannel = this.targetFile.getChannel();
}
private byte[] getBytes(short[] array) {
try {
ByteArrayOutputStream bytestream = new ByteArrayOutputStream();
DataOutputStream datastream = new DataOutputStream(bytestream);
for (short n : array) {
datastream.writeShort(n);
}
datastream.flush();
return bytestream.toByteArray();
} catch (IOException ioe) {
Logging.logger().finest(ioe.getMessage());
}
return null;
}
public boolean initialize(MapSource mapSource) throws IOException, WMSServiceException {
if (null == mapSource) {
String msg = Logging.getMessage("nullValue.MapSourceIsNull");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
this.mapSource = mapSource;
this.params = mapSource.getParameters();
if (null == params) {
String msg = Logging.getMessage("nullValue.AVListIsNull");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
if (!params.hasKey(AVKey.FILE_NAME)) {
String msg = Logging.getMessage("nullValue.ParamsIsNull");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
this.sourceFile = new File(params.getStringValue(AVKey.FILE_NAME));
if (!this.sourceFile.exists()) {
String msg = Logging.getMessage("generic.FileNotFound", this.sourceFile.getAbsolutePath());
Logging.logger().severe(msg);
throw new FileNotFoundException(msg);
}
AVList fileParams = this.params.copy();
try {
this.readerFactory =
(DataRasterReaderFactory)
WorldWind.createConfigurationComponent(AVKey.DATA_RASTER_READER_FACTORY_CLASS_NAME);
} catch (Exception e) {
this.readerFactory = new BasicDataRasterReaderFactory();
}
DataRasterReader reader =
this.readerFactory.findReaderFor(this.sourceFile, fileParams, readers);
if (reader == null) {
String msg = Logging.getMessage("nullValue.ReaderIsNull", this.sourceFile);
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
reader.readMetadata(this.sourceFile, fileParams);
this.params.setValues(fileParams);
if (!this.params.hasKey(AVKey.SECTOR)) {
String msg = Logging.getMessage("nullValue.SectorIsNull");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
this.BBOX = (Sector) this.params.getValue(AVKey.SECTOR);
if (0d == this.BBOX.getDeltaLatDegrees() || 0d == this.BBOX.getDeltaLonDegrees()) {
String msg = Logging.getMessage("generic.SectorSizeInvalid");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
int height = 0;
if (!this.params.hasKey(AVKey.HEIGHT)) {
String msg = Logging.getMessage("generic.InvalidHeight", 0);
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
} else {
Object o = this.params.getValue(AVKey.HEIGHT);
double d = Double.parseDouble("" + o);
height = (int) d;
}
if (!this.params.hasKey(AVKey.WIDTH)) {
String msg = Logging.getMessage("generic.InvalidWidth", 0);
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
this.isElevation =
(this.params.hasKey(AVKey.PIXEL_FORMAT)
&& AVKey.ELEVATION.equals(this.params.getValue(AVKey.PIXEL_FORMAT)));
if (this.params.hasKey(AVKey.MISSING_DATA_SIGNAL)) {
try {
Object o = this.params.getValue(AVKey.MISSING_DATA_SIGNAL);
double d = Double.parseDouble("" + o);
this.nodataSignal = (short) d;
} catch (Exception e) {
this.nodataSignal = (this.isElevation) ? Short.MIN_VALUE : 0;
}
} else {
this.nodataSignal = (this.isElevation) ? Short.MIN_VALUE : 0;
}
if (this.params.hasKey(AVKey.MISSING_DATA_REPLACEMENT)) {
try {
Object o = this.params.getValue(AVKey.MISSING_DATA_REPLACEMENT);
double d = Double.parseDouble("" + o);
this.nodataReplacement = (short) d;
} catch (Exception e) {
Logging.logger().finest(e.getMessage());
this.nodataReplacement = (this.isElevation) ? Short.MIN_VALUE : 0;
}
} else {
this.nodataReplacement = (this.isElevation) ? Short.MIN_VALUE : 0;
}
if (this.isElevation) {
if (this.params.hasKey(AVKey.ELEVATION_UNIT)) {
try {
String unit = this.params.getStringValue(AVKey.ELEVATION_UNIT);
this.convertFeetToMeters = "feet".equalsIgnoreCase(unit);
} catch (Exception e) {
Logging.logger().finest(e.getMessage());
}
}
}
// if PIXEL_HEIGHT is specified, we are not overriding it
// because UTM images will have different pixel size
if (!this.params.hasKey(AVKey.PIXEL_HEIGHT)) {
this.pixelHeight = this.BBOX.getDeltaLatDegrees() / (double) height;
} else {
try {
Object o = this.params.getValue(AVKey.PIXEL_HEIGHT);
this.pixelHeight = Double.parseDouble("" + o);
} catch (Exception e) {
Logging.logger().finest(e.getMessage());
}
}
this.rasters = reader.read(this.sourceFile, this.params);
if (null == this.rasters || 0 == this.rasters.length) {
String msg = Logging.getMessage("nullValue.RasterIsNull");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
return true;
}
public ImageFormatter serviceRequest(IMapRequest req) throws IOException, WMSServiceException {
ImageFormatter formatter = null;
try {
Sector reqExtent =
(SingleFileLayer.this.isElevation)
? req.getExtentForElevationRequest()
: req.getExtent();
if (null == this.intersects(reqExtent, SingleFileLayer.this.getBBox())) {
String msg =
Logging.getMessage(
"WMS.Layer.OutOfCoverage",
reqExtent.toString(),
SingleFileLayer.this.getBBox().toString());
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
if (req.getHeight() <= 0 || req.getWidth() <= 0) {
String msg =
Logging.getMessage("generic.InvalidImageSize", req.getWidth(), req.getHeight());
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
DataRaster[] rasters = SingleFileLayer.this.rasters;
if (null == rasters || 0 == rasters.length) {
String msg = Logging.getMessage("nullValue.RasterIsNull");
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
double missingDataReplacement = (double) SingleFileLayer.this.nodataReplacement;
try {
String s = req.getBGColor();
if (null != s) {
missingDataReplacement = Double.parseDouble(s);
}
} catch (Exception e) {
missingDataReplacement = (double) SingleFileLayer.this.nodataReplacement;
}
DataRaster raster = rasters[0];
if (raster instanceof BufferedImageRaster) {
BufferedImageRaster reqRaster =
new BufferedImageRaster(
req.getWidth(), req.getHeight(), Transparency.TRANSLUCENT, reqExtent);
raster.drawOnTo(reqRaster);
BufferedImage img = reqRaster.getBufferedImage();
this.makeNoDataTransparent(
img, SingleFileLayer.this.nodataSignal, (short) missingDataReplacement);
formatter = new BufferedImageFormatter(img);
} else if (raster instanceof ByteBufferRaster) {
AVList reqParams = new AVListImpl();
reqParams.setValue(AVKey.WIDTH, req.getWidth());
reqParams.setValue(AVKey.HEIGHT, req.getHeight());
reqParams.setValue(AVKey.SECTOR, reqExtent);
reqParams.setValue(
AVKey.BYTE_ORDER, AVKey.LITTLE_ENDIAN); // by default BIL is LITTLE ENDIAN
reqParams.setValue(AVKey.PIXEL_FORMAT, AVKey.ELEVATION);
String reqFormat = req.getFormat();
if (null != reqFormat && reqFormat.endsWith("32")) {
reqParams.setValue(AVKey.DATA_TYPE, AVKey.FLOAT32);
} else {
reqParams.setValue(AVKey.DATA_TYPE, AVKey.INT16);
}
reqParams.setValue(AVKey.MISSING_DATA_REPLACEMENT, missingDataReplacement);
ByteBufferRaster reqRaster =
new ByteBufferRaster(req.getWidth(), req.getHeight(), reqExtent, reqParams);
raster.drawOnTo(reqRaster);
if (SingleFileLayer.this.convertFeetToMeters) {
this.convertFeetToMeters(reqRaster);
}
formatter = new DataRasterFormatter(reqRaster);
} else {
String msg =
Logging.getMessage(
"generic.UnrecognizedImageSourceType", raster.getClass().getName());
Logging.logger().severe(msg);
throw new WMSServiceException(msg);
}
} catch (WMSServiceException wmsse) {
throw wmsse;
} catch (Exception ex) {
Logging.logger()
.log(
java.util.logging.Level.SEVERE,
SingleFileLayer.this.getThreadId() + ex.getMessage(),
ex);
// throw new WMSServiceException( s );
} finally {
}
return formatter;
}
protected void validateParameters(AVList list, int srcWidth, int srcHeight)
throws IllegalArgumentException {
if (null == list || 0 == list.getValues().size()) {
String reason = Logging.getMessage("nullValue.AVListIsNull");
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", reason);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
if (!(srcWidth > 0 && srcHeight > 0)) {
String msg = Logging.getMessage("generic.InvalidImageSize", srcWidth, srcHeight);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
if (list.hasKey(AVKey.WIDTH)) {
int width = (Integer) list.getValue(AVKey.WIDTH);
if (width != srcWidth) {
String msg = Logging.getMessage("GeotiffWriter.ImageWidthMismatch", width, srcWidth);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
} else list.setValue(AVKey.WIDTH, srcWidth);
if (list.hasKey(AVKey.HEIGHT)) {
int height = (Integer) list.getValue(AVKey.HEIGHT);
if (height != srcHeight) {
String msg = Logging.getMessage("GeotiffWriter.ImageHeightMismatch", height, srcHeight);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
} else list.setValue(AVKey.HEIGHT, srcHeight);
Sector sector = null;
if (list.hasKey(AVKey.SECTOR)) sector = (Sector) list.getValue(AVKey.SECTOR);
if (null == sector) {
String msg = Logging.getMessage("GeotiffWriter.NoSectorSpecified");
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
if (!list.hasKey(AVKey.COORDINATE_SYSTEM)) {
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", AVKey.COORDINATE_SYSTEM);
Logging.logger().finest(msg);
// throw new IllegalArgumentException(msg);
// assume Geodetic Coordinate System
list.setValue(AVKey.COORDINATE_SYSTEM, AVKey.COORDINATE_SYSTEM_GEOGRAPHIC);
}
if (!list.hasKey(AVKey.PROJECTION_EPSG_CODE)) {
if (isGeographic(list)) {
// assume WGS84
list.setValue(AVKey.PROJECTION_EPSG_CODE, GeoTiff.GCS.WGS_84);
} else {
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", AVKey.PROJECTION_EPSG_CODE);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
}
// if PIXEL_WIDTH is specified, we are not overriding it because UTM images
// will have different pixel size
if (!list.hasKey(AVKey.PIXEL_WIDTH)) {
if (isGeographic(list)) {
double pixelWidth = sector.getDeltaLonDegrees() / (double) srcWidth;
list.setValue(AVKey.PIXEL_WIDTH, pixelWidth);
} else {
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", AVKey.PIXEL_WIDTH);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
}
// if PIXEL_HEIGHT is specified, we are not overriding it
// because UTM images will have different pixel size
if (!list.hasKey(AVKey.PIXEL_HEIGHT)) {
if (isGeographic(list)) {
double pixelHeight = sector.getDeltaLatDegrees() / (double) srcHeight;
list.setValue(AVKey.PIXEL_HEIGHT, pixelHeight);
} else {
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", AVKey.PIXEL_HEIGHT);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
}
if (!list.hasKey(AVKey.PIXEL_FORMAT)) {
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", AVKey.PIXEL_FORMAT);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
} else {
String pixelFormat = list.getStringValue(AVKey.PIXEL_FORMAT);
if (!AVKey.ELEVATION.equals(pixelFormat) && !AVKey.IMAGE.equals(pixelFormat)) {
String msg =
Logging.getMessage("Geotiff.UnknownGeoKeyValue", pixelFormat, AVKey.PIXEL_FORMAT);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
}
// validate elevation parameters
if (AVKey.ELEVATION.equals(list.getValue(AVKey.PIXEL_FORMAT))) {
if (!list.hasKey(AVKey.DATA_TYPE)) {
String msg = Logging.getMessage("GeotiffWriter.GeoKeysMissing", AVKey.DATA_TYPE);
Logging.logger().finest(msg);
throw new IllegalArgumentException(msg);
}
String type = list.getStringValue(AVKey.DATA_TYPE);
if (!AVKey.FLOAT32.equals(type) && !AVKey.INT16.equals(type)) {
String msg = Logging.getMessage("Geotiff.UnknownGeoKeyValue", type, AVKey.DATA_TYPE);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
}
if (!list.hasKey(AVKey.ORIGIN)) {
// set UpperLeft corner as the origin, if not specified
LatLon origin = new LatLon(sector.getMaxLatitude(), sector.getMinLongitude());
list.setValue(AVKey.ORIGIN, origin);
}
if (list.hasKey(AVKey.BYTE_ORDER)
&& !AVKey.BIG_ENDIAN.equals(list.getStringValue(AVKey.BYTE_ORDER))) {
String msg =
Logging.getMessage(
"generic.UnrecognizedByteOrder", list.getStringValue(AVKey.BYTE_ORDER));
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
if (!list.hasKey(AVKey.DATE_TIME)) {
// add NUL (\0) termination as required by TIFF v6 spec (20 bytes length)
String timestamp = String.format("%1$tY:%1$tm:%1$td %tT\0", Calendar.getInstance());
list.setValue(AVKey.DATE_TIME, timestamp);
}
if (!list.hasKey(AVKey.VERSION)) {
list.setValue(AVKey.VERSION, Version.getVersion());
}
}
private void appendGeoTiff(ArrayList<TiffIFDEntry> ifds, AVList params)
throws IOException, IllegalArgumentException {
if (null == params || 0 == params.getEntries().size()) {
String reason = Logging.getMessage("nullValue.AVListIsNull");
Logging.logger().finest(Logging.getMessage("GeotiffWriter.GeoKeysMissing", reason));
return;
}
long offset = this.theChannel.position();
if (params.hasKey(AVKey.DISPLAY_NAME)) {
String value = params.getStringValue(AVKey.DISPLAY_NAME);
if (null != value && 0 < value.trim().length()) {
offset = this.theChannel.position();
byte[] bytes = value.trim().getBytes();
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(new TiffIFDEntry(Tiff.Tag.DOCUMENT_NAME, Tiff.Type.ASCII, bytes.length, offset));
}
}
if (params.hasKey(AVKey.DESCRIPTION)) {
String value = params.getStringValue(AVKey.DESCRIPTION);
if (null != value && 0 < value.trim().length()) {
offset = this.theChannel.position();
byte[] bytes = value.trim().getBytes();
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(
new TiffIFDEntry(Tiff.Tag.IMAGE_DESCRIPTION, Tiff.Type.ASCII, bytes.length, offset));
}
}
if (params.hasKey(AVKey.VERSION)) {
String value = params.getStringValue(AVKey.VERSION);
if (null != value && 0 < value.trim().length()) {
offset = this.theChannel.position();
byte[] bytes = value.trim().getBytes();
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(
new TiffIFDEntry(Tiff.Tag.SOFTWARE_VERSION, Tiff.Type.ASCII, bytes.length, offset));
}
}
if (params.hasKey(AVKey.DATE_TIME)) {
String value = params.getStringValue(AVKey.DATE_TIME);
if (null != value && 0 < value.trim().length()) {
offset = this.theChannel.position();
byte[] bytes = value.getBytes();
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(new TiffIFDEntry(Tiff.Tag.DATE_TIME, Tiff.Type.ASCII, bytes.length, offset));
}
}
if (params.hasKey(AVKey.SECTOR)) {
if (params.hasKey(AVKey.PIXEL_WIDTH) && params.hasKey(AVKey.PIXEL_HEIGHT)) {
offset = this.theChannel.position();
double[] values =
new double[] {
(Double) params.getValue(AVKey.PIXEL_WIDTH),
(Double) params.getValue(AVKey.PIXEL_HEIGHT),
isElevation(params) ? 1d : 0d
};
byte[] bytes = this.getBytes(values);
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(
new TiffIFDEntry(
GeoTiff.Tag.MODEL_PIXELSCALE, Tiff.Type.DOUBLE, values.length, offset));
}
if (params.hasKey(AVKey.WIDTH) && params.hasKey(AVKey.HEIGHT)) {
offset = this.theChannel.position();
double w = (Integer) params.getValue(AVKey.WIDTH);
double h = (Integer) params.getValue(AVKey.HEIGHT);
Sector sec = (Sector) params.getValue(AVKey.SECTOR);
double[] values =
new double[] { // i , j, k=0, x, y, z=0
0d,
0d,
0d,
sec.getMinLongitude().degrees,
sec.getMaxLatitude().degrees,
0d,
w - 1,
0d,
0d,
sec.getMaxLongitude().degrees,
sec.getMaxLatitude().degrees,
0d,
w - 1,
h - 1,
0d,
sec.getMaxLongitude().degrees,
sec.getMinLatitude().degrees,
0d,
0d,
h - 1,
0d,
sec.getMinLongitude().degrees,
sec.getMinLatitude().degrees,
0d,
};
byte[] bytes = this.getBytes(values);
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(
new TiffIFDEntry(GeoTiff.Tag.MODEL_TIEPOINT, Tiff.Type.DOUBLE, values.length, offset));
}
// Tiff.Tag.MODEL_TRANSFORMATION excludes Tiff.Tag.MODEL_TIEPOINT & Tiff.Tag.MODEL_PIXELSCALE
if (params.hasKey(AVKey.MISSING_DATA_SIGNAL)
|| params.hasKey(AVKey.MISSING_DATA_REPLACEMENT)) {
offset = this.theChannel.position();
Object nodata =
params.hasKey(AVKey.MISSING_DATA_SIGNAL)
? params.getValue(AVKey.MISSING_DATA_SIGNAL)
: params.getValue(AVKey.MISSING_DATA_REPLACEMENT);
String value = "" + nodata + "\0";
byte[] bytes = value.getBytes();
this.theChannel.write(ByteBuffer.wrap(bytes));
ifds.add(new TiffIFDEntry(GeoTiff.Tag.GDAL_NODATA, Tiff.Type.ASCII, bytes.length, offset));
}
if (params.hasKey(AVKey.COORDINATE_SYSTEM)) {
String cs = params.getStringValue(AVKey.COORDINATE_SYSTEM);
if (AVKey.COORDINATE_SYSTEM_GEOGRAPHIC.equals(cs)) {
if (isElevation(params)) this.writeGeographicElevationGeoKeys(ifds, params);
else this.writeGeographicImageGeoKeys(ifds, params);
} else if (AVKey.COORDINATE_SYSTEM_PROJECTED.equals(cs)) {
String msg = Logging.getMessage("GeotiffWriter.FeatureNotImplementedd", cs);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
// TODO extract PCS (Projection Coordinate System)
} else {
String msg = Logging.getMessage("GeotiffWriter.UnknownCoordinateSystem", cs);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
}
}
}
public void write(BufferedImage image, AVList params) throws IOException {
if (image == null) {
String msg = Logging.getMessage("nullValue.ImageSource");
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
if (0 == image.getWidth() || 0 == image.getHeight()) {
String msg =
Logging.getMessage("generic.InvalidImageSize", image.getWidth(), image.getHeight());
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
if (null == params || 0 == params.getValues().size()) {
String reason = Logging.getMessage("nullValue.AVListIsNull");
Logging.logger().finest(Logging.getMessage("GeotiffWriter.GeoKeysMissing", reason));
params = new AVListImpl();
} else {
this.validateParameters(params, image.getWidth(), image.getHeight());
}
// how we proceed in part depends upon the image type...
int type = image.getType();
// handle CUSTOM type which comes from our GeoTiffreader (for now)
if (BufferedImage.TYPE_CUSTOM == type) {
int numColorComponents = 0, numComponents = 0, pixelSize = 0, dataType = 0, csType = 0;
boolean hasAlpha = false;
if (null != image.getColorModel()) {
ColorModel cm = image.getColorModel();
numColorComponents = cm.getNumColorComponents();
numComponents = cm.getNumComponents();
pixelSize = cm.getPixelSize();
hasAlpha = cm.hasAlpha();
ColorSpace cs = cm.getColorSpace();
if (null != cs) csType = cs.getType();
}
if (null != image.getSampleModel()) {
SampleModel sm = image.getSampleModel();
dataType = sm.getDataType();
}
if (dataType == DataBuffer.TYPE_FLOAT && pixelSize == Float.SIZE && numComponents == 1) {
type = BufferedImage_TYPE_ELEVATION_FLOAT32;
} else if (dataType == DataBuffer.TYPE_SHORT
&& pixelSize == Short.SIZE
&& numComponents == 1) {
type = BufferedImage_TYPE_ELEVATION_SHORT16;
} else if (ColorSpace.CS_GRAY == csType && pixelSize == Byte.SIZE) {
type = BufferedImage.TYPE_BYTE_GRAY;
} else if (dataType == DataBuffer.TYPE_USHORT
&& ColorSpace.CS_GRAY == csType
&& pixelSize == Short.SIZE) {
type = BufferedImage.TYPE_USHORT_GRAY;
} else if (ColorSpace.TYPE_RGB == csType
&& pixelSize == 3 * Byte.SIZE
&& numColorComponents == 3) {
type = BufferedImage.TYPE_3BYTE_BGR;
} else if (ColorSpace.TYPE_RGB == csType
&& hasAlpha
&& pixelSize == 4 * Byte.SIZE
&& numComponents == 4) {
type = BufferedImage.TYPE_4BYTE_ABGR;
}
}
switch (type) {
case BufferedImage.TYPE_3BYTE_BGR:
case BufferedImage.TYPE_4BYTE_ABGR:
case BufferedImage.TYPE_4BYTE_ABGR_PRE:
case BufferedImage.TYPE_INT_RGB:
case BufferedImage.TYPE_INT_BGR:
case BufferedImage.TYPE_INT_ARGB:
case BufferedImage.TYPE_INT_ARGB_PRE:
{
this.writeColorImage(image, params);
}
break;
case BufferedImage.TYPE_USHORT_GRAY:
case BufferedImage.TYPE_BYTE_GRAY:
{
this.writeGrayscaleImage(image, params);
}
break;
case BufferedImage_TYPE_ELEVATION_SHORT16:
case BufferedImage_TYPE_ELEVATION_FLOAT32:
{
String msg = Logging.getMessage("GeotiffWriter.FeatureNotImplementedd", type);
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
// break;
case BufferedImage.TYPE_CUSTOM:
default:
{
ColorModel cm = image.getColorModel();
SampleModel sm = image.getSampleModel();
StringBuffer sb =
new StringBuffer(Logging.getMessage("GeotiffWriter.UnsupportedType", type));
sb.append("\n");
sb.append("NumBands=").append(sm.getNumBands()).append("\n");
sb.append("NumDataElements=").append(sm.getNumDataElements()).append("\n");
sb.append("NumColorComponents=").append(cm.getNumColorComponents()).append("\n");
sb.append("NumComponents=").append(cm.getNumComponents()).append("\n");
sb.append("PixelSize=").append(cm.getPixelSize()).append("\n");
sb.append("hasAlpha=").append(cm.hasAlpha());
String msg = sb.toString();
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
}
}
public void writeRaster(BufferWrapperRaster raster) throws IOException, IllegalArgumentException {
if (raster == null) {
String msg = Logging.getMessage("nullValue.RasterIsNull");
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
if (0 == raster.getWidth() || 0 == raster.getHeight()) {
String msg =
Logging.getMessage("generic.InvalidImageSize", raster.getWidth(), raster.getHeight());
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
this.validateParameters(raster, raster.getWidth(), raster.getHeight());
int bitsPerSample, samplesPerPixel, sampleFormat, photometric, numBands;
if (AVKey.ELEVATION.equals(raster.getValue(AVKey.PIXEL_FORMAT))) {
if (AVKey.FLOAT32.equals(raster.getValue(AVKey.DATA_TYPE))) {
numBands = 1;
samplesPerPixel = Tiff.SamplesPerPixel.MONOCHROME;
sampleFormat = Tiff.SampleFormat.IEEEFLOAT;
photometric = Tiff.Photometric.Grayscale_BlackIsZero;
bitsPerSample = Tiff.BitsPerSample.ELEVATIONS_FLOAT32;
} else if (AVKey.INT16.equals(raster.getValue(AVKey.DATA_TYPE))) {
numBands = 1;
samplesPerPixel = Tiff.SamplesPerPixel.MONOCHROME;
sampleFormat = Tiff.SampleFormat.SIGNED;
photometric = Tiff.Photometric.Grayscale_BlackIsZero;
bitsPerSample = Tiff.BitsPerSample.ELEVATIONS_INT16;
} else {
String msg =
Logging.getMessage("GeotiffWriter.UnsupportedType", raster.getValue(AVKey.DATA_TYPE));
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
} else if (AVKey.IMAGE.equals(raster.getValue(AVKey.PIXEL_FORMAT))) {
if (AVKey.INT8.equals(raster.getValue(AVKey.DATA_TYPE))) {
numBands = 1;
samplesPerPixel = Tiff.SamplesPerPixel.MONOCHROME;
sampleFormat = Tiff.SampleFormat.UNSIGNED;
photometric = Tiff.Photometric.Grayscale_BlackIsZero;
bitsPerSample = Tiff.BitsPerSample.MONOCHROME_UINT8;
} else if (AVKey.INT16.equals(raster.getValue(AVKey.DATA_TYPE))) {
numBands = 1;
samplesPerPixel = Tiff.SamplesPerPixel.MONOCHROME;
sampleFormat = Tiff.SampleFormat.UNSIGNED;
photometric = Tiff.Photometric.Grayscale_BlackIsZero;
bitsPerSample = Tiff.BitsPerSample.MONOCHROME_UINT16;
} else if (AVKey.INT32.equals(raster.getValue(AVKey.DATA_TYPE))) {
numBands = 3;
// TODO check ALPHA / Transparency
samplesPerPixel = Tiff.SamplesPerPixel.RGB;
sampleFormat = Tiff.SampleFormat.UNSIGNED;
photometric = Tiff.Photometric.Color_RGB;
bitsPerSample = Tiff.BitsPerSample.RGB;
} else {
String msg =
Logging.getMessage("GeotiffWriter.UnsupportedType", raster.getValue(AVKey.DATA_TYPE));
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
} else {
String msg =
Logging.getMessage("GeotiffWriter.UnsupportedType", raster.getValue(AVKey.PIXEL_FORMAT));
Logging.logger().severe(msg);
throw new IllegalArgumentException(msg);
}
int bytesPerSample = numBands * bitsPerSample / Byte.SIZE;
this.writeTiffHeader();
// write the image data...
int numRows = raster.getHeight();
int numCols = raster.getWidth();
int[] stripCounts = new int[numRows];
int[] stripOffsets = new int[numRows];
BufferWrapper srcBuffer = raster.getBuffer();
ByteBuffer dataBuff = ByteBuffer.allocateDirect(numCols * bytesPerSample);
switch (bitsPerSample) {
// case Tiff.BitsPerSample.MONOCHROME_BYTE:
case Tiff.BitsPerSample.MONOCHROME_UINT8:
{
for (int y = 0; y < numRows; y++) {
stripOffsets[y] = (int) this.theChannel.position();
stripCounts[y] = numCols * bytesPerSample;
dataBuff.clear();
for (int x = 0; x < numCols * numBands; x++) {
dataBuff.put(srcBuffer.getByte(x + y * numCols));
}
dataBuff.flip();
this.theChannel.write(dataBuff);
}
}
break;
// case Tiff.BitsPerSample.MONOCHROME_UINT16:
case Tiff.BitsPerSample.ELEVATIONS_INT16:
{
for (int y = 0; y < numRows; y++) {
stripOffsets[y] = (int) this.theChannel.position();
stripCounts[y] = numCols * bytesPerSample;
dataBuff.clear();
for (int x = 0; x < numCols * numBands; x++) {
dataBuff.putShort(srcBuffer.getShort(x + y * numCols));
}
dataBuff.flip();
this.theChannel.write(dataBuff);
}
}
break;
case Tiff.BitsPerSample.ELEVATIONS_FLOAT32:
{
for (int y = 0; y < numRows; y++) {
stripOffsets[y] = (int) this.theChannel.position();
stripCounts[y] = numCols * bytesPerSample;
dataBuff.clear();
for (int x = 0; x < numCols * numBands; x++) {
dataBuff.putFloat(srcBuffer.getFloat(x + y * numCols));
}
dataBuff.flip();
this.theChannel.write(dataBuff);
}
}
break;
case Tiff.BitsPerSample.RGB:
{
for (int y = 0; y < numRows; y++) {
stripOffsets[y] = (int) this.theChannel.position();
stripCounts[y] = numCols * bytesPerSample;
dataBuff.clear();
for (int x = 0; x < numCols; x++) {
int color = srcBuffer.getInt(x + y * numCols);
byte red = (byte) (0xFF & (color >> 16));
byte green = (byte) (0xFF & (color >> 8));
byte blue = (byte) (0xFF & color);
// dataBuff.put(0xFF & (color >> 24)); // alpha
dataBuff.put(red).put(green).put(blue);
}
dataBuff.flip();
this.theChannel.write(dataBuff);
}
}
break;
}
// write out values for the tiff tags and build up the IFD. These are supposed to be sorted; for
// now
// do this manually here.
ArrayList<TiffIFDEntry> ifds = new ArrayList<TiffIFDEntry>(10);
ifds.add(new TiffIFDEntry(Tiff.Tag.IMAGE_WIDTH, Tiff.Type.LONG, 1, numCols));
ifds.add(new TiffIFDEntry(Tiff.Tag.IMAGE_LENGTH, Tiff.Type.LONG, 1, numRows));
long offset = this.theChannel.position();
if (Tiff.BitsPerSample.RGB == bitsPerSample) {
short[] bps = new short[numBands];
for (int i = 0; i < numBands; i++) {
bps[i] = Tiff.BitsPerSample.MONOCHROME_BYTE;
}
this.theChannel.write(ByteBuffer.wrap(this.getBytes(bps)));
ifds.add(new TiffIFDEntry(Tiff.Tag.BITS_PER_SAMPLE, Tiff.Type.SHORT, numBands, offset));
} else ifds.add(new TiffIFDEntry(Tiff.Tag.BITS_PER_SAMPLE, Tiff.Type.SHORT, 1, bitsPerSample));
ifds.add(new TiffIFDEntry(Tiff.Tag.COMPRESSION, Tiff.Type.LONG, 1, Tiff.Compression.NONE));
ifds.add(new TiffIFDEntry(Tiff.Tag.PHOTO_INTERPRETATION, Tiff.Type.SHORT, 1, photometric));
ifds.add(new TiffIFDEntry(Tiff.Tag.SAMPLES_PER_PIXEL, Tiff.Type.SHORT, 1, samplesPerPixel));
ifds.add(new TiffIFDEntry(Tiff.Tag.ORIENTATION, Tiff.Type.SHORT, 1, Tiff.Orientation.DEFAULT));
ifds.add(
new TiffIFDEntry(
Tiff.Tag.PLANAR_CONFIGURATION, Tiff.Type.SHORT, 1, Tiff.PlanarConfiguration.CHUNKY));
ifds.add(new TiffIFDEntry(Tiff.Tag.SAMPLE_FORMAT, Tiff.Type.SHORT, 1, sampleFormat));
offset = this.theChannel.position();
dataBuff = ByteBuffer.allocateDirect(stripOffsets.length * INTEGER_SIZEOF);
for (int stripOffset : stripOffsets) {
dataBuff.putInt(stripOffset);
}
dataBuff.flip();
this.theChannel.write(dataBuff);
ifds.add(new TiffIFDEntry(Tiff.Tag.STRIP_OFFSETS, Tiff.Type.LONG, stripOffsets.length, offset));
ifds.add(new TiffIFDEntry(Tiff.Tag.ROWS_PER_STRIP, Tiff.Type.LONG, 1, 1));
offset = this.theChannel.position();
dataBuff.clear(); // stripOffsets and stripCounts are same length by design; can reuse the
// ByteBuffer...
for (int stripCount : stripCounts) {
dataBuff.putInt(stripCount);
}
dataBuff.flip();
this.theChannel.write(dataBuff);
ifds.add(
new TiffIFDEntry(Tiff.Tag.STRIP_BYTE_COUNTS, Tiff.Type.LONG, stripCounts.length, offset));
this.appendGeoTiff(ifds, raster);
this.writeIFDs(ifds);
}