/* * Copyright (c) 2005 Sun Microsystems, Inc. All Rights Reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are * met: * * - Redistribution of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * - Redistribution in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * Neither the name of Sun Microsystems, Inc. or the names of * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * * This software is provided "AS IS," without a warranty of any kind. ALL * EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, * INCLUDING ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A * PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE HEREBY EXCLUDED. SUN * MICROSYSTEMS, INC. ("SUN") AND ITS LICENSORS SHALL NOT BE LIABLE FOR * ANY DAMAGES SUFFERED BY LICENSEE AS A RESULT OF USING, MODIFYING OR * DISTRIBUTING THIS SOFTWARE OR ITS DERIVATIVES. IN NO EVENT WILL SUN OR * ITS LICENSORS BE LIABLE FOR ANY LOST REVENUE, PROFIT OR DATA, OR FOR * DIRECT, INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL OR PUNITIVE * DAMAGES, HOWEVER CAUSED AND REGARDLESS OF THE THEORY OF LIABILITY, * ARISING OUT OF THE USE OF OR INABILITY TO USE THIS SOFTWARE, EVEN IF * SUN HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES. * * You acknowledge that this software is not designed or intended for use * in the design, construction, operation or maintenance of any nuclear * facility. * * Sun gratefully acknowledges that this software was originally authored * and developed by Kenneth Bradley Russell and Christopher John Kline. */ package com.sun.opengl.util.texture.awt; import java.awt.Graphics; import java.awt.image.*; import java.io.*; import java.net.*; import java.nio.*; import java.util.*; import javax.imageio.*; import javax.media.opengl.*; import javax.media.opengl.glu.*; import com.sun.opengl.impl.Debug; import com.sun.opengl.util.awt.*; import com.sun.opengl.util.io.*; import com.sun.opengl.util.texture.*; import com.sun.opengl.util.texture.spi.*; /**

Provides input and output facilities for both loading OpenGL textures from disk and streams as well as writing textures already in memory back to disk.

The TextureIO class supports an arbitrary number of plug-in readers and writers via TextureProviders and TextureWriters. TextureProviders know how to produce TextureData objects from files, InputStreams and URLs. TextureWriters know how to write TextureData objects to disk in various file formats. The TextureData class represents the raw data of the texture before it has been converted to an OpenGL texture object. The Texture class represents the OpenGL texture object and provides easy facilities for using the texture.

There are several built-in TextureProviders and TextureWriters supplied with the TextureIO implementation. The most basic provider uses the platform's Image I/O facilities to read in a BufferedImage and convert it to a texture. This is the baseline provider and is registered so that it is the last one consulted. All others are asked first to open a given file.

There are three other providers registered by default as of the time of this writing. One handles SGI RGB (".sgi", ".rgb") images from both files and streams. One handles DirectDraw Surface (".dds") images read from files, though can not read these images from streams. One handles Targa (".tga") images read from both files and streams. These providers are executed in an arbitrary order. Some of these providers require the file's suffix to either be specified via the newTextureData methods or for the file to be named with the appropriate suffix. In general a file suffix should be provided to the newTexture and newTextureData methods if at all possible.

Note that additional TextureProviders, if reading images from InputStreams, must use the mark()/reset() methods on InputStream when probing for e.g. magic numbers at the head of the file to make sure not to disturb the state of the InputStream for downstream TextureProviders.

There are analogous TextureWriters provided for writing textures back to disk if desired. As of this writing, there are four TextureWriters registered by default: one for Targa files, one for SGI RGB files, one for DirectDraw surface (.dds) files, and one for ImageIO-supplied formats such as .jpg and .png. Some of these writers have certain limitations such as only being able to write out textures stored in GL_RGB or GL_RGBA format. The DDS writer supports fetching and writing to disk of texture data in DXTn compressed format. Whether this will occur is dependent on whether the texture's internal format is one of the DXTn compressed formats and whether the target file is .dds format. */ public class AWTTextureWriter { private static final boolean DEBUG = Debug.debug("TextureReader"); //---------------------------------------------------------------------- // methods that *do not* require a current context // These methods assume RGB or RGBA textures. // Some texture providers may not recognize the file format unless // the fileSuffix is specified, so it is strongly recommended to // specify it wherever it is known. // Some texture providers may also only support one kind of input, // i.e., reading from a file as opposed to a stream. /** * Writes the given texture to a file. The type of the file is * inferred from its suffix. An OpenGL context must be current in * order to fetch the texture data back from the OpenGL pipeline. * This method causes the specified Texture to be bound to the * GL_TEXTURE_2D state. If no suitable writer for the requested file * format was found, throws an IOException.

* * Reasonable attempts are made to produce good results in the * resulting images. The Targa, SGI and ImageIO writers produce * results in the correct vertical orientation for those file * formats. The DDS writer performs no vertical flip of the data, * even in uncompressed mode. (It is impossible to perform such a * vertical flip with compressed data.) Applications should keep * this in mind when using this routine to save textures to disk for * later re-loading.

* * Any mipmaps for the specified texture are currently discarded * when it is written to disk, regardless of whether the underlying * file format supports multiple mipmaps in a given file. * * @throws IOException if an error occurred during writing or no * suitable writer was found * @throws GLException if no OpenGL context was current or an * OpenGL-related error occurred */ public static void write(Texture texture, File file) throws IOException, GLException { if (texture.getTarget() != GL.GL_TEXTURE_2D) { throw new GLException("Only GL_TEXTURE_2D textures are supported"); } // First fetch the texture data GL _gl = GLU.getCurrentGL(); if(!_gl.isGL2()) { throw new GLException("Only GL2 supports fetching compressed images, GL: "+_gl); } GL2 gl = _gl.getGL2(); texture.bind(); int internalFormat = glGetTexLevelParameteri(gl, GL.GL_TEXTURE_2D, 0, GL2.GL_TEXTURE_INTERNAL_FORMAT); int width = glGetTexLevelParameteri(gl, GL.GL_TEXTURE_2D, 0, GL2.GL_TEXTURE_WIDTH); int height = glGetTexLevelParameteri(gl, GL.GL_TEXTURE_2D, 0, GL2.GL_TEXTURE_HEIGHT); int border = glGetTexLevelParameteri(gl, GL.GL_TEXTURE_2D, 0, GL2.GL_TEXTURE_BORDER); TextureData data = null; if (internalFormat == GL.GL_COMPRESSED_RGB_S3TC_DXT1_EXT || internalFormat == GL.GL_COMPRESSED_RGBA_S3TC_DXT1_EXT || internalFormat == GL.GL_COMPRESSED_RGBA_S3TC_DXT3_EXT || internalFormat == GL.GL_COMPRESSED_RGBA_S3TC_DXT5_EXT) { // Fetch using glGetCompressedTexImage int size = glGetTexLevelParameteri(gl, GL.GL_TEXTURE_2D, 0, GL2.GL_TEXTURE_COMPRESSED_IMAGE_SIZE); ByteBuffer res = ByteBuffer.allocate(size); gl.glGetCompressedTexImage(GL.GL_TEXTURE_2D, 0, res); data = new AWTTextureData(internalFormat, width, height, border, internalFormat, GL.GL_UNSIGNED_BYTE, false, true, true, res, null); } else { int bytesPerPixel = 0; int fetchedFormat = 0; switch (internalFormat) { case GL.GL_RGB: case GL2.GL_BGR: case GL.GL_RGB8: bytesPerPixel = 3; fetchedFormat = GL.GL_RGB; break; case GL.GL_RGBA: case GL2.GL_BGRA: case GL2.GL_ABGR_EXT: case GL.GL_RGBA8: bytesPerPixel = 4; fetchedFormat = GL.GL_RGBA; break; default: throw new IOException("Unsupported texture internal format 0x" + Integer.toHexString(internalFormat)); } // Fetch using glGetTexImage int packAlignment = glGetInteger(GL.GL_PACK_ALIGNMENT); int packRowLength = glGetInteger(GL2.GL_PACK_ROW_LENGTH); int packSkipRows = glGetInteger(GL2.GL_PACK_SKIP_ROWS); int packSkipPixels = glGetInteger(GL2.GL_PACK_SKIP_PIXELS); int packSwapBytes = glGetInteger(GL2.GL_PACK_SWAP_BYTES); gl.glPixelStorei(GL.GL_PACK_ALIGNMENT, 1); gl.glPixelStorei(GL2.GL_PACK_ROW_LENGTH, 0); gl.glPixelStorei(GL2.GL_PACK_SKIP_ROWS, 0); gl.glPixelStorei(GL2.GL_PACK_SKIP_PIXELS, 0); gl.glPixelStorei(GL2.GL_PACK_SWAP_BYTES, 0); ByteBuffer res = ByteBuffer.allocate((width + (2 * border)) * (height + (2 * border)) * bytesPerPixel); if (DEBUG) { System.out.println("Allocated buffer of size " + res.remaining() + " for fetched image (" + ((fetchedFormat == GL.GL_RGB) ? "GL_RGB" : "GL_RGBA") + ")"); } gl.glGetTexImage(GL.GL_TEXTURE_2D, 0, fetchedFormat, GL.GL_UNSIGNED_BYTE, res); gl.glPixelStorei(GL.GL_PACK_ALIGNMENT, packAlignment); gl.glPixelStorei(GL2.GL_PACK_ROW_LENGTH, packRowLength); gl.glPixelStorei(GL2.GL_PACK_SKIP_ROWS, packSkipRows); gl.glPixelStorei(GL2.GL_PACK_SKIP_PIXELS, packSkipPixels); gl.glPixelStorei(GL2.GL_PACK_SWAP_BYTES, packSwapBytes); data = new AWTTextureData(internalFormat, width, height, border, fetchedFormat, GL.GL_UNSIGNED_BYTE, false, false, false, res, null); if (DEBUG) { System.out.println("data.getPixelFormat() = " + ((data.getPixelFormat() == GL.GL_RGB) ? "GL_RGB" : "GL_RGBA")); } } for (Iterator iter = textureWriters.iterator(); iter.hasNext(); ) { TextureWriter writer = (TextureWriter) iter.next(); if (writer.write(file, data)) { return; } } throw new IOException("No suitable texture writer found"); } //---------------------------------------------------------------------- // SPI support // /** Adds a TextureWriter to support writing of a new file format. */ public static void addTextureWriter(TextureWriter writer) { // Must always add at the front so the ImageIO writer is last, // so we don't accidentally use it instead of a user's possibly // more optimal writer textureWriters.add(0, writer); } //--------------------------------------------------------------------------- // Global disabling of texture rectangle extension // //---------------------------------------------------------------------- // Internals only below this point // private static List/**/ textureWriters = new ArrayList/**/(); static { // ImageIO writer, the fall-back, must be the first one added textureWriters.add(new IIOTextureWriter()); // Other special-case writers addTextureWriter(new DDSTextureWriter()); addTextureWriter(new SGITextureWriter()); addTextureWriter(new TGATextureWriter()); } //---------------------------------------------------------------------- // ImageIO texture writer // static class IIOTextureWriter implements TextureWriter { public boolean write(File file, TextureData data) throws IOException { int pixelFormat = data.getPixelFormat(); int pixelType = data.getPixelType(); if ((pixelFormat == GL.GL_RGB || pixelFormat == GL.GL_RGBA) && (pixelType == GL.GL_BYTE || pixelType == GL.GL_UNSIGNED_BYTE)) { // Convert TextureData to appropriate BufferedImage // FIXME: almost certainly not obeying correct pixel order BufferedImage image = new BufferedImage(data.getWidth(), data.getHeight(), (pixelFormat == GL.GL_RGB) ? BufferedImage.TYPE_3BYTE_BGR : BufferedImage.TYPE_4BYTE_ABGR); byte[] imageData = ((DataBufferByte) image.getRaster().getDataBuffer()).getData(); ByteBuffer buf = (ByteBuffer) data.getBuffer(); if (buf == null) { buf = (ByteBuffer) data.getMipmapData()[0]; } buf.rewind(); buf.get(imageData); buf.rewind(); // Swizzle image components to be correct if (pixelFormat == GL.GL_RGB) { for (int i = 0; i < imageData.length; i += 3) { byte red = imageData[i + 0]; byte blue = imageData[i + 2]; imageData[i + 0] = blue; imageData[i + 2] = red; } } else { for (int i = 0; i < imageData.length; i += 4) { byte red = imageData[i + 0]; byte green = imageData[i + 1]; byte blue = imageData[i + 2]; byte alpha = imageData[i + 3]; imageData[i + 0] = alpha; imageData[i + 1] = blue; imageData[i + 2] = green; imageData[i + 3] = red; } } // Flip image vertically for the user's convenience ImageUtil.flipImageVertically(image); // Happened to notice that writing RGBA images to JPEGS is broken if (TextureUtil.JPG.equals(FileUtil.getFileSuffix(file)) && image.getType() == BufferedImage.TYPE_4BYTE_ABGR) { BufferedImage tmpImage = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_3BYTE_BGR); Graphics g = tmpImage.getGraphics(); g.drawImage(image, 0, 0, null); g.dispose(); image = tmpImage; } return ImageIO.write(image, FileUtil.getFileSuffix(file), file); } throw new IOException("ImageIO writer doesn't support this pixel format / type (only GL_RGB/A + bytes)"); } } //---------------------------------------------------------------------- // DDS texture writer // static class DDSTextureWriter implements TextureWriter { public boolean write(File file, TextureData data) throws IOException { if (TextureUtil.DDS.equals(FileUtil.getFileSuffix(file))) { // See whether the DDS writer can handle this TextureData int pixelFormat = data.getPixelFormat(); int pixelType = data.getPixelType(); if (pixelType != GL.GL_BYTE && pixelType != GL.GL_UNSIGNED_BYTE) { throw new IOException("DDS writer only supports byte / unsigned byte textures"); } int d3dFormat = 0; // FIXME: some of these are probably not completely correct and would require swizzling switch (pixelFormat) { case GL.GL_RGB: d3dFormat = DDSImage.D3DFMT_R8G8B8; break; case GL.GL_RGBA: d3dFormat = DDSImage.D3DFMT_A8R8G8B8; break; case GL.GL_COMPRESSED_RGB_S3TC_DXT1_EXT: d3dFormat = DDSImage.D3DFMT_DXT1; break; case GL.GL_COMPRESSED_RGBA_S3TC_DXT1_EXT: throw new IOException("RGBA DXT1 not yet supported"); case GL.GL_COMPRESSED_RGBA_S3TC_DXT3_EXT: d3dFormat = DDSImage.D3DFMT_DXT3; break; case GL.GL_COMPRESSED_RGBA_S3TC_DXT5_EXT: d3dFormat = DDSImage.D3DFMT_DXT5; break; default: throw new IOException("Unsupported pixel format 0x" + Integer.toHexString(pixelFormat) + " by DDS writer"); } ByteBuffer[] mipmaps = null; if (data.getMipmapData() != null) { mipmaps = new ByteBuffer[data.getMipmapData().length]; for (int i = 0; i < mipmaps.length; i++) { mipmaps[i] = (ByteBuffer) data.getMipmapData()[i]; } } else { mipmaps = new ByteBuffer[] { (ByteBuffer) data.getBuffer() }; } DDSImage image = DDSImage.createFromData(d3dFormat, data.getWidth(), data.getHeight(), mipmaps); image.write(file); return true; } return false; } } //---------------------------------------------------------------------- // SGI (rgb) texture writer // static class SGITextureWriter implements TextureWriter { public boolean write(File file, TextureData data) throws IOException { String fileSuffix = FileUtil.getFileSuffix(file); if (TextureUtil.SGI.equals(fileSuffix) || TextureUtil.SGI_RGB.equals(fileSuffix)) { // See whether the SGI writer can handle this TextureData int pixelFormat = data.getPixelFormat(); int pixelType = data.getPixelType(); if ((pixelFormat == GL.GL_RGB || pixelFormat == GL.GL_RGBA) && (pixelType == GL.GL_BYTE || pixelType == GL.GL_UNSIGNED_BYTE)) { ByteBuffer buf = ((data.getBuffer() != null) ? (ByteBuffer) data.getBuffer() : (ByteBuffer) data.getMipmapData()[0]); byte[] bytes; if (buf.hasArray()) { bytes = buf.array(); } else { buf.rewind(); bytes = new byte[buf.remaining()]; buf.get(bytes); buf.rewind(); } SGIImage image = SGIImage.createFromData(data.getWidth(), data.getHeight(), (pixelFormat == GL.GL_RGBA), bytes); image.write(file, false); return true; } throw new IOException("SGI writer doesn't support this pixel format / type (only GL_RGB/A + bytes)"); } return false; } } //---------------------------------------------------------------------- // TGA (Targa) texture writer static class TGATextureWriter implements TextureWriter { public boolean write(File file, TextureData data) throws IOException { if (TextureUtil.TGA.equals(FileUtil.getFileSuffix(file))) { // See whether the TGA writer can handle this TextureData int pixelFormat = data.getPixelFormat(); int pixelType = data.getPixelType(); if ((pixelFormat == GL.GL_RGB || pixelFormat == GL.GL_RGBA) && (pixelType == GL.GL_BYTE || pixelType == GL.GL_UNSIGNED_BYTE)) { ByteBuffer buf = ((data.getBuffer() != null) ? (ByteBuffer) data.getBuffer() : (ByteBuffer) data.getMipmapData()[0]); // Must reverse order of red and blue channels to get correct results int skip = ((pixelFormat == GL.GL_RGB) ? 3 : 4); for (int i = 0; i < buf.remaining(); i += skip) { byte red = buf.get(i + 0); byte blue = buf.get(i + 2); buf.put(i + 0, blue); buf.put(i + 2, red); } TGAImage image = TGAImage.createFromData(data.getWidth(), data.getHeight(), (pixelFormat == GL.GL_RGBA), false, ((data.getBuffer() != null) ? (ByteBuffer) data.getBuffer() : (ByteBuffer) data.getMipmapData()[0])); image.write(file); return true; } throw new IOException("TGA writer doesn't support this pixel format / type (only GL_RGB/A + bytes)"); } return false; } } //---------------------------------------------------------------------- // Helper routines // private static int glGetInteger(int pname) { int[] tmp = new int[1]; GL gl = GLU.getCurrentGL(); gl.glGetIntegerv(pname, tmp, 0); return tmp[0]; } private static int glGetTexLevelParameteri(GL2 gl, int target, int level, int pname) { int[] tmp = new int[1]; gl.glGetTexLevelParameteriv(target, 0, pname, tmp, 0); return tmp[0]; } }