java读取DDS图片文件类库

package org.zsword.util; import java.awt.image.BufferedImage; import java.io.IOException; import java.nio.ByteBuffer; import java.util.logging.Level; import java.util.logging.Logger; /** * @author JemiZhuu(周士淳, zsword) * @version 1.1 * @since 2009-09-13 * DDS(DIRECTDRAWSURFACE) File Reader */ public class PluginDDS { private DDSHEADER header; private int bpp; private int rowBlockSize; private BufferedImage image; private static final Logger logger = Logger.getLogger(PluginDDS.class.getName()); static { logger.setLevel(Level.SEVERE); } public DDSHEADER getHeader() { return header; } private static final int DDPF_ALPHAPIXELS = 0x00000001; private static final int DDPF_ALPHA = 0x00000002; private static final int DDPF_FOURCC = 0x00000004; private static final int DDPF_RGB = 0x00000040; private static final String FOURCC_DXT1 = "DXT1"; private static final String FOURCC_DXT2 = "DXT2"; private static final String FOURCC_DXT3 = "DXT3"; private static final String FOURCC_DXT4 = "DXT4"; private static final String FOURCC_DXT5 = "DXT5"; private static final int FI_RGBA_RED = 2; private static final int FI_RGBA_GREEN = 1; private static final int FI_RGBA_BLUE = 0; private static final int FI_RGBA_ALPHA = 3; private static final int FI_RGBA_RED_MASK = 0x00FF0000; private static final int FI_RGBA_GREEN_MASK = 0x0000FF00; private static final int FI_RGBA_BLUE_MASK = 0x000000FF; private static final int FI_RGBA_ALPHA_MASK = 0xFF000000; private static final int FI_RGBA_RED_SHIFT = 16; private static final int FI_RGBA_GREEN_SHIFT = 8; private static final int FI_RGBA_BLUE_SHIFT = 0; private static final int FI_RGBA_ALPHA_SHIFT = 24; private static final int FIT_BITMAP = 1; /** * read a DDS Texture File * @param io * @return BufferedImage * @throws IOException */ public BufferedImage readDDSFile(ByteFileReader io) throws IOException { this.readHeader(io); this.readPixeData(io); return image; } /** * read DDS header * @param io * @throws IOException */ protected void readHeader(ByteFileReader io) throws IOException { if (io == null || io.length() <= 0) { return; } header = new DDSHEADER(); byte[] buf = new byte[4]; io.read(buf); header.dwMagic = new String(buf); header.dwSize = io.read4BytesAsInt(); header.dwFlags = io.read4BytesAsInt(); header.dwHeight = io.read4BytesAsInt(); header.dwWidth = io.read4BytesAsInt(); header.dwPitchOrLinearSize = io.read4BytesAsInt(); header.dwDepth = io.read4BytesAsInt(); header.dwMipMapCount = io.read4BytesAsInt(); for (int i = 0; i < header.dwReserved1.length; i++) { header.dwReserved1[i] = io.read4BytesAsInt(); } header.ddpfPixelFormat.dwSize = io.read4BytesAsInt(); header.ddpfPixelFormat.dwFlags = io.read4BytesAsInt(); io.read(buf); header.ddpfPixelFormat.dwFourCC = new String(buf); header.ddpfPixelFormat.dwRGBBitCount = io.read4BytesAsInt(); header.ddpfPixelFormat.dwRBitMask = io.read4BytesAsInt(); header.ddpfPixelFormat.dwGBitMask = io.read4BytesAsInt(); header.ddpfPixelFormat.dwBBitMask = io.read4BytesAsInt(); header.ddpfPixelFormat.dwRGBAlphaBitMask = io.read4BytesAsInt(); header.ddsCaps.dwCaps1 = io.read4BytesAsInt(); header.ddsCaps.dwCaps2 = io.read4BytesAsInt(); for (int i = 0; i < header.ddsCaps.dwReserved.length; i++) { header.ddsCaps.dwReserved[i] = io.read4BytesAsInt(); } header.dwReserved2 = io.read4BytesAsInt(); } /** * read DDS pixel datas * @param io * @throws IOException */ protected void readPixeData(ByteFileReader io) throws IOException { if (header == null) { throw new IOException("DDS Header is NULL"); } image = new BufferedImage(header.dwWidth, header.dwHeight, BufferedImage.TYPE_INT_ARGB_PRE); if (0 != (header.ddpfPixelFormat.dwFlags & DDPF_RGB)) { LoadRGB(io); } else if (0 != (header.ddpfPixelFormat.dwFlags & DDPF_FOURCC)) { this.LoadDXT(io); } else { throw new IOException("Unkown DDS Format"); } } /** * load DXT data by buffer * @param io * @throws IOException */ protected void LoadDXT(ByteFileReader io) throws IOException { if (header == null) { throw new IOException("DDS Header is NULL"); } rowBlockSize = header.dwWidth * 4; int width = header.dwWidth & ~3; int height = header.dwHeight & ~3; int bytesPerBlock = 8; if (FOURCC_DXT3.equals(header.ddpfPixelFormat.dwFourCC)) { bytesPerBlock = 16; } else if (FOURCC_DXT5.equals(header.ddpfPixelFormat.dwFourCC)) { bytesPerBlock = 16; } //allocate a 32-bit dib bpp = 32; int line = ((width * bpp) + 7) / 8; int widthRest = width & 3; int heightRest = height & 3; int inputLine = (width + 3) / 4; ByteBuffer buff = ByteBuffer.allocate(inputLine * bytesPerBlock); if (height >= 4) { for (int yheight = 0; yheight < height; yheight += 4) { buff.clear(); io.read(buff.array()); if (width >= 4) { for (int xwidth = 0; xwidth < width; xwidth += 4) { this.DecodeDXTBlock(image, buff, xwidth, yheight, 4, 4); } } if (widthRest != 0) { this.DecodeDXTBlock(image, buff, widthRest, yheight, widthRest, 4); } } } if (heightRest != 0) { buff.clear(); io.read(buff.array()); if (width >= 4) { for (int xwidth = 0; xwidth < width; xwidth += 4) { this.DecodeDXTBlock(image, buff, xwidth, heightRest, 4, heightRest); } } if (widthRest != 0) { this.DecodeDXTBlock(image, buff, widthRest, heightRest, widthRest, heightRest); } } } /** * dacode DXT pixel data * @param img * @param io * @param width * @param height * @param bw * @param bh */ protected void DecodeDXTBlock(BufferedImage img, ByteBuffer io, int width, int height, int bw, int bh) { int alpha[] = null; int aphrow[] = null; int data[] = null; //read alpha data if (FOURCC_DXT3.equals(header.ddpfPixelFormat.dwFourCC)) { aphrow = new int[4]; for (int i = 0; i < 4; i++) { aphrow[i] = BufferUtil.readBytesAsInt(io, 2); } } else if (FOURCC_DXT5.equals(header.ddpfPixelFormat.dwFourCC)) { alpha = new int[8]; for (int i = 0; i < 2; i++) { alpha[i] = BufferUtil.readBytesAsInt(io, 1); } if (alpha[0] > alpha[1]) { //8 alpha block for (int i = 0; i < 6; i++) { alpha[i + 2] = (((6 - i) * alpha[0] + (1 + i) * alpha[1] + 3) / 7); } } else { //6 alpha block for (int i = 0; i < 4; i++) { alpha[i + 2] = (((4 - i) * alpha[0] + (1 + i) * alpha[1] + 2) / 5); } alpha[6] = 0x00; alpha[7] = 0xFF; } data = new int[6]; for (int i = 0; i < 6; i++) { data[i] = BufferUtil.readBytesAsInt(io, 1); } } //read color data Color8888[] colors = new Color8888[4]; for (int i = 0; i < 4; i++) { colors[i] = new Color8888(); } int colval[] = new int[2]; for (int i = 0; i < 2; i++) { colval[i] = BufferUtil.readBytesAsInt(io, 2); } GetBlockColors(colval, colors, true); int row[] = new int[4]; for (int i = 0; i < 4; i++) { row[i] = BufferUtil.readBytesAsInt(io, 1); } //decode color data for (int y = 0; y < bh; y++) { int m_colorRow = row[y]; int m_alphaRow = 0; int m_alphaBits = 0; int m_offset = 0; if (FOURCC_DXT3.equals(header.ddpfPixelFormat.dwFourCC)) { m_alphaRow = (short) aphrow[y]; } else if (FOURCC_DXT5.equals(header.ddpfPixelFormat.dwFourCC)) { int ti = y / 2; m_alphaBits = (data[0 + ti * 3] | (data[1 + ti * 3] << 8) | (data[2 + ti * 3] << 16)); m_offset = (y & 1) * 12; } for (int x = 0; x < bw; x++) { Color8888 getCol = new Color8888(); int bits = ((m_colorRow >> (x * 2)) & 3); getCol.Copy(colors[bits]); if (FOURCC_DXT3.equals(header.ddpfPixelFormat.dwFourCC)) { bits = ((m_alphaRow >> (x *