Hi Robert, I guess I already sent you a modified DDS readerwriter, but it was never integrated... Or maybe I really really thought about sending it without doing it really :-D ? Anyway I've cleaned a few little things and it's now worth integrating, if you agree.
- Vertically flipping DDS files upon reading (with the corresponding string option) now also sets the origin of the image (Image::setOrigin()) accordingly. - Writing DDS files now uses image's origin to automatically vertically flip if needed. This can be avoided using the "ddsNoAutoFlipWrite" option. Actually, it seems the vertical flip should be the "standard" reading way (at least for multiple 3rd-party software I use) but this would break code, so I didn't change the default behaviour (I only "standardized" the option name as you'll see). File BASE rev = 11941. Cheers, Sukender PVLE - Lightweight cross-platform game engine - http://pvle.sourceforge.net/
/********************************************************************** * * FILE: ReaderWriterdds.cpp * * DESCRIPTION: Class for reading a DDS file into an osg::Image. * * Example on reading a DDS file code can be found at: * http://developer.nvidia.com/docs/IO/1327/ATT/ * ARB_texture_compression.pdf * Author: Sebastien Domine, NVIDIA Corporation * * CREATED BY: Rune Schmidt Jensen, rsj@uni-dk * * HISTORY: Created 31.03.2003 * Modified 13.05.2004 * by George Tarantilis, [email protected] * Modified 22.05.2009 * Wojtek Lewandowski, [email protected] * * WARNING: * Bit Masks in the WrtiteDDS are set for 8 bit components * write with 4 or 16 bit components will * probably produce corrupted file * Wojtek Lewandowski 2009-05-22 * **********************************************************************/ #include <osg/Texture> #include <osg/Notify> #include <osgDB/Registry> #include <osgDB/FileNameUtils> #include <osgDB/FileUtils> #include <osgDB/fstream> #include <iomanip> #include <stdio.h> #include <string.h> #if defined(OSG_GLES1_AVAILABLE) || defined(OSG_GLES2_AVAILABLE) #define GL_RED 0x1903 #define GL_LUMINANCE4_ALPHA4 0x8043 #endif #if defined(OSG_GL3_AVAILABLE) #define GL_LUMINANCE4_ALPHA4 0x8043 #endif // NOTICE ON WIN32: // typedef DWORD unsigned long; // sizeof(DWORD) = 4 typedef unsigned int UI32; typedef int I32; struct DDCOLORKEY { DDCOLORKEY(): dwColorSpaceLowValue(0), dwColorSpaceHighValue(0) {} UI32 dwColorSpaceLowValue; UI32 dwColorSpaceHighValue; }; struct DDPIXELFORMAT { DDPIXELFORMAT(): dwSize(0), dwFlags(0), dwFourCC(0), dwRGBBitCount(0), dwRBitMask(0), dwGBitMask(0), dwBBitMask(0), dwRGBAlphaBitMask(0) {} UI32 dwSize; UI32 dwFlags; UI32 dwFourCC; union { UI32 dwRGBBitCount; UI32 dwYUVBitCount; UI32 dwZBufferBitDepth; UI32 dwAlphaBitDepth; UI32 dwLuminanceBitDepth; }; union { UI32 dwRBitMask; UI32 dwYBitMask; }; union { UI32 dwGBitMask; UI32 dwUBitMask; }; union { UI32 dwBBitMask; UI32 dwVBitMask; }; union { UI32 dwRGBAlphaBitMask; UI32 dwYUVAlphaBitMask; UI32 dwRGBZBitMask; UI32 dwYUVZBitMask; }; }; struct DDSCAPS2 { DDSCAPS2(): dwCaps(0), dwCaps2(0), dwCaps3(0), dwCaps4(0) {} UI32 dwCaps; UI32 dwCaps2; UI32 dwCaps3; union { UI32 dwCaps4; UI32 dwVolumeDepth; }; }; struct DDSURFACEDESC2 { DDSURFACEDESC2(): dwSize(0), dwFlags(0), dwHeight(0), dwWidth(0), lPitch(0), dwBackBufferCount(0), dwMipMapCount(0), dwAlphaBitDepth(0), dwReserved(0), lpSurface(0), dwTextureStage(0) {} UI32 dwSize; UI32 dwFlags; UI32 dwHeight; UI32 dwWidth; union { I32 lPitch; UI32 dwLinearSize; }; union { UI32 dwBackBufferCount; UI32 dwDepth; }; union { UI32 dwMipMapCount; UI32 dwRefreshRate; }; UI32 dwAlphaBitDepth; UI32 dwReserved; UI32 lpSurface; //Fred Marmond: removed from pointer type to UI32 for 64bits compatibility. it is unused data DDCOLORKEY ddckCKDestOverlay; DDCOLORKEY ddckCKDestBlt; DDCOLORKEY ddckCKSrcOverlay; DDCOLORKEY ddckCKSrcBlt; DDPIXELFORMAT ddpfPixelFormat; DDSCAPS2 ddsCaps; UI32 dwTextureStage; }; // // Structure of a DXT-1 compressed texture block // see http://msdn.microsoft.com/library/default.asp?url=/library/en-us/directx9_c/Opaque_and_1_Bit_Alpha_Textures.asp // struct DXT1TexelsBlock { unsigned short color_0; // colors at their unsigned short color_1; // extreme unsigned int texels4x4; // interpolated colors (2 bits per texel) }; // // DDSURFACEDESC2 flags that mark the validity of the struct data // #define DDSD_CAPS 0x00000001l // default #define DDSD_HEIGHT 0x00000002l // default #define DDSD_WIDTH 0x00000004l // default #define DDSD_PIXELFORMAT 0x00001000l // default #define DDSD_PITCH 0x00000008l // For uncompressed formats #define DDSD_MIPMAPCOUNT 0x00020000l #define DDSD_LINEARSIZE 0x00080000l // For compressed formats #define DDSD_DEPTH 0x00800000l // Volume Textures // // DDPIXELFORMAT flags // #define DDPF_ALPHAPIXELS 0x00000001l #define DDPF_FOURCC 0x00000004l // Compressed formats #define DDPF_RGB 0x00000040l // Uncompressed formats #define DDPF_ALPHA 0x00000002l #define DDPF_COMPRESSED 0x00000080l #define DDPF_LUMINANCE 0x00020000l #define DDPF_BUMPLUMINANCE 0x00040000l // L,U,V #define DDPF_BUMPDUDV 0x00080000l // U,V // // DDSCAPS flags // #define DDSCAPS_TEXTURE 0x00001000l // default #define DDSCAPS_COMPLEX 0x00000008l #define DDSCAPS_MIPMAP 0x00400000l #define DDSCAPS2_VOLUME 0x00200000l #ifndef MAKEFOURCC #define MAKEFOURCC(ch0, ch1, ch2, ch3) \ ((UI32)(char)(ch0) | ((UI32)(char)(ch1) << 8) | \ ((UI32)(char)(ch2) << 16) | ((UI32)(char)(ch3) << 24 )) #endif //defined(MAKEFOURCC) /* * FOURCC codes for DX compressed-texture pixel formats */ #define FOURCC_DXT1 (MAKEFOURCC('D','X','T','1')) #define FOURCC_DXT2 (MAKEFOURCC('D','X','T','2')) #define FOURCC_DXT3 (MAKEFOURCC('D','X','T','3')) #define FOURCC_DXT4 (MAKEFOURCC('D','X','T','4')) #define FOURCC_DXT5 (MAKEFOURCC('D','X','T','5')) /* * FOURCC codes for 3dc compressed-texture pixel formats */ #define FOURCC_ATI1 (MAKEFOURCC('A','T','I','1')) #define FOURCC_ATI2 (MAKEFOURCC('A','T','I','2')) static unsigned int ComputeImageSizeInBytes ( int width, int height, int depth, unsigned int pixelFormat, unsigned int pixelType, int packing = 1, int slice_packing = 1, int image_packing = 1 ) { if( width < 1 ) width = 1; if( height < 1 ) height = 1; if( depth < 1 ) depth = 1; // Taking advantage of the fact that // DXT formats are defined as 4 successive numbers: // GL_COMPRESSED_RGB_S3TC_DXT1_EXT 0x83F0 // GL_COMPRESSED_RGBA_S3TC_DXT1_EXT 0x83F1 // GL_COMPRESSED_RGBA_S3TC_DXT3_EXT 0x83F2 // GL_COMPRESSED_RGBA_S3TC_DXT5_EXT 0x83F3 if( pixelFormat >= GL_COMPRESSED_RGB_S3TC_DXT1_EXT && pixelFormat <= GL_COMPRESSED_RGBA_S3TC_DXT5_EXT ) { width = (width + 3) & ~3; height = (height + 3) & ~3; } // 3dc ATI formats // GL_COMPRESSED_RED_RGTC1_EXT 0x8DBB // GL_COMPRESSED_SIGNED_RED_RGTC1_EXT 0x8DBC // GL_COMPRESSED_RED_GREEN_RGTC2_EXT 0x8DBD // GL_COMPRESSED_SIGNED_RED_GREEN_RGTC2_EXT 0x8DBE if( pixelFormat >= GL_COMPRESSED_RED_RGTC1_EXT && pixelFormat <= GL_COMPRESSED_SIGNED_RED_GREEN_RGTC2_EXT ) { width = (width + 3) & ~3; height = (height + 3) & ~3; } // compute size of one row unsigned int size = osg::Image::computeRowWidthInBytes ( width, pixelFormat, pixelType, packing ); // now compute size of slice size *= height; size += slice_packing - 1; size -= size % slice_packing; // compute size of whole image size *= depth; size += image_packing - 1; size -= size % image_packing; return size; } osg::Image* ReadDDSFile(std::istream& _istream, bool flipDDSRead) { DDSURFACEDESC2 ddsd; char filecode[4]; _istream.read(filecode, 4); if (strncmp(filecode, "DDS ", 4) != 0) { return NULL; } // Get the surface desc. _istream.read((char*)(&ddsd), sizeof(ddsd)); osg::ref_ptr<osg::Image> osgImage = new osg::Image(); //Check valid structure sizes if(ddsd.dwSize != 124 && ddsd.ddpfPixelFormat.dwSize != 32) { return NULL; } bool is3dImage = false; int depth = 1; // Check for volume image if( ddsd.dwDepth > 0 && (ddsd.dwFlags & DDSD_DEPTH)) { is3dImage = true; depth = ddsd.dwDepth; } // Retreive image properties. int s = ddsd.dwWidth; int t = ddsd.dwHeight; int r = depth; unsigned int dataType = GL_UNSIGNED_BYTE; unsigned int pixelFormat = 0; unsigned int internalFormat = 0; // Handle some esoteric formats if(ddsd.ddpfPixelFormat.dwFlags & DDPF_BUMPDUDV) { OSG_WARN << "ReadDDSFile warning: DDPF_BUMPDUDV format is not supported" << std::endl; return NULL; // ddsd.ddpfPixelFormat.dwFlags = // DDPF_LUMINANCE + DDPF_ALPHAPIXELS; // // handle V8U8 as A8L8 // // handle V16U16 as A16L16 // // but Q8W8U8L8 as RGB? // // A2W10U10V10 as RGBA (dwFlags == DDPF_BUMPDUDV + DDPF_ALPHAPIXELS) } if(ddsd.ddpfPixelFormat.dwFlags & DDPF_BUMPLUMINANCE) { OSG_WARN << "ReadDDSFile warning: DDPF_BUMPLUMINANCE format is not supported" << std::endl; return NULL; // ddsd.ddpfPixelFormat.dwFlags = DDPF_RGB; // // handle as RGB // // L6V5U5 -- 655 is not supported data type in GL // // X8L8V8U8 -- just as RGB } // Uncompressed formats will usually use DDPF_RGB to indicate an RGB format, // while compressed formats will use DDPF_FOURCC with a four-character code. bool usingAlpha = ddsd.ddpfPixelFormat.dwFlags & DDPF_ALPHAPIXELS; bool checkIfUsingOneBitAlpha = false; // Uncompressed formats. if(ddsd.ddpfPixelFormat.dwFlags & DDPF_RGB) { struct RGBFormat { const char* name; UI32 bitCount; UI32 rBitMask; UI32 gBitMask; UI32 bBitMask; UI32 aBitMask; unsigned int internalFormat; unsigned int pixelFormat; unsigned int dataType; }; const unsigned int UNSUPPORTED = 0; static const RGBFormat rgbFormats[] = { { "R3G3B2" , 8, 0xe0, 0x1c, 0x03, 0x00, GL_RGB , GL_RGB , GL_UNSIGNED_BYTE_3_3_2 }, { "R5G6B5" , 16, 0xf800, 0x07e0, 0x001f, 0x0000, GL_RGB , GL_RGB , GL_UNSIGNED_SHORT_5_6_5 }, { "A1R5G5B5" , 16, 0x7c00, 0x03e0, 0x001f, 0x8000, GL_RGBA, GL_BGRA, GL_UNSIGNED_SHORT_1_5_5_5_REV }, { "X1R5G5B5" , 16, 0x7c00, 0x03e0, 0x001f, 0x0000, GL_RGB , GL_BGRA, GL_UNSIGNED_SHORT_1_5_5_5_REV }, { "A4R4G4B4" , 16, 0x0f00, 0x00f0, 0x000f, 0xf000, GL_RGBA, GL_BGRA, GL_UNSIGNED_SHORT_4_4_4_4_REV }, { "X4R4G4B4" , 16, 0x0f00, 0x00f0, 0x000f, 0x0000, GL_RGB , GL_BGRA, GL_UNSIGNED_SHORT_4_4_4_4_REV }, { "A8R3G3B2" , 16, 0x00e0, 0x001c, 0x0003, 0xff00, GL_RGBA, GL_BGRA, UNSUPPORTED }, { "R8G8B8", 24, 0xff0000, 0x00ff00, 0x0000ff, 0x000000, GL_RGB , GL_BGR , GL_UNSIGNED_BYTE }, { "B8G8R8", 24, 0x0000ff, 0x00ff00, 0xff0000, 0x000000, GL_RGB , GL_RGB , GL_UNSIGNED_BYTE }, { "A8R8G8B8", 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000, GL_RGBA, GL_BGRA, GL_UNSIGNED_BYTE }, { "X8R8G8B8", 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0x00000000, GL_RGB , GL_BGRA, GL_UNSIGNED_BYTE }, { "A8B8G8R8", 32, 0x000000ff, 0x0000ff00, 0x00ff0000, 0xff000000, GL_RGBA, GL_RGBA, GL_UNSIGNED_BYTE }, { "X8B8G8R8", 32, 0x000000ff, 0x0000ff00, 0x00ff0000, 0x00000000, GL_RGB , GL_RGBA, GL_UNSIGNED_BYTE }, { "A2R10G10B10", 32, 0x000003ff, 0x000ffc00, 0x3ff00000, 0xc0000000, GL_RGBA, GL_BGRA, GL_UNSIGNED_INT_2_10_10_10_REV }, { "A2B10G10R10", 32, 0x3ff00000, 0x000ffc00, 0x000003ff, 0xc0000000, GL_RGBA, GL_RGBA, GL_UNSIGNED_INT_2_10_10_10_REV }, { "G16R16", 32, 0x0000ffff, 0xffff0000, 0x00000000, 0x00000000, GL_RGB, UNSUPPORTED, GL_UNSIGNED_SHORT }, }; bool found = false; for ( unsigned int i = 0; i < sizeof ( rgbFormats ) / sizeof ( RGBFormat ); i++ ) { const RGBFormat& f = rgbFormats[ i ]; if ( ddsd.ddpfPixelFormat.dwRGBBitCount == f.bitCount && ddsd.ddpfPixelFormat.dwRBitMask == f.rBitMask && ddsd.ddpfPixelFormat.dwGBitMask == f.gBitMask && ddsd.ddpfPixelFormat.dwBBitMask == f.bBitMask && ddsd.ddpfPixelFormat.dwRGBAlphaBitMask == f.aBitMask ) { if ( f.internalFormat != UNSUPPORTED && f.pixelFormat != UNSUPPORTED && f.dataType != UNSUPPORTED ) { OSG_INFO << "ReadDDSFile info : format = " << f.name << std::endl; internalFormat = f.internalFormat; pixelFormat = f.pixelFormat; dataType = f.dataType; found = true; break; } else { OSG_INFO << "ReadDDSFile info : " << f.name << " format is not supported" << std::endl; return NULL; } } } if ( !found ) { OSG_WARN << "ReadDDSFile warning: unhandled RGB pixel format in dds file, image not loaded" << std::endl; OSG_INFO << "ReadDDSFile info : ddsd.ddpfPixelFormat.dwRGBBitCount = " << ddsd.ddpfPixelFormat.dwRGBBitCount << std::endl; OSG_INFO << "ReadDDSFile info : ddsd.ddpfPixelFormat.dwRBitMask = 0x" << std::hex << std::setw(8) << std::setfill('0') << ddsd.ddpfPixelFormat.dwRBitMask << std::endl; OSG_INFO << "ReadDDSFile info : ddsd.ddpfPixelFormat.dwGBitMask = 0x" << std::hex << std::setw(8) << std::setfill('0') << ddsd.ddpfPixelFormat.dwGBitMask << std::endl; OSG_INFO << "ReadDDSFile info : ddsd.ddpfPixelFormat.dwBBitMask = 0x" << std::hex << std::setw(8) << std::setfill('0') << ddsd.ddpfPixelFormat.dwBBitMask << std::endl; OSG_INFO << "ReadDDSFile info : ddsd.ddpfPixelFormat.dwRGBAlphaBitMask = 0x" << std::hex << std::setw(8) << std::setfill('0') << ddsd.ddpfPixelFormat.dwRGBAlphaBitMask << std::dec << std::endl; return NULL; } } else if(ddsd.ddpfPixelFormat.dwFlags & DDPF_LUMINANCE) { internalFormat = usingAlpha ? GL_LUMINANCE_ALPHA : GL_LUMINANCE; pixelFormat = usingAlpha ? GL_LUMINANCE_ALPHA : GL_LUMINANCE; if ( usingAlpha && ddsd.ddpfPixelFormat.dwLuminanceBitDepth == 8 ) { OSG_INFO << "ReadDDSFile info : format = L4A4" << std::endl; pixelFormat = GL_LUMINANCE4_ALPHA4; // invalid enumerant? } else if ( usingAlpha && ddsd.ddpfPixelFormat.dwLuminanceBitDepth == 32 ) { OSG_INFO << "ReadDDSFile info : format = L16A16" << std::endl; dataType = GL_UNSIGNED_SHORT; } else if ( !usingAlpha && ddsd.ddpfPixelFormat.dwLuminanceBitDepth == 16 ) { OSG_INFO << "ReadDDSFile info : format = L16" << std::endl; dataType = GL_UNSIGNED_SHORT; } else if ( usingAlpha ) { OSG_INFO << "ReadDDSFile info : format = L8A8" << std::endl; } else { OSG_INFO << "ReadDDSFile info : format = L8" << std::endl; } // else if ( ddsd.ddpfPixelFormat.dwLuminanceBitDepth == (usingAlpha ? 64 : 32) ) // { // dataType = GL_UNSIGNED_INT; // } } else if(ddsd.ddpfPixelFormat.dwFlags & DDPF_ALPHA) { OSG_INFO << "ReadDDSFile info : format = ALPHA" << std::endl; internalFormat = GL_ALPHA; pixelFormat = GL_ALPHA; } // Compressed formats else if(ddsd.ddpfPixelFormat.dwFlags & DDPF_FOURCC) { // TODO: Image::isImageTranslucent() doesn't work with S3TC compressed files switch(ddsd.ddpfPixelFormat.dwFourCC) { case FOURCC_DXT1: OSG_INFO << "ReadDDSFile info : format = DXT1" << std::endl; if (usingAlpha) { internalFormat = GL_COMPRESSED_RGBA_S3TC_DXT1_EXT; pixelFormat = GL_COMPRESSED_RGBA_S3TC_DXT1_EXT; } else { internalFormat = GL_COMPRESSED_RGB_S3TC_DXT1_EXT; pixelFormat = GL_COMPRESSED_RGB_S3TC_DXT1_EXT; checkIfUsingOneBitAlpha = true; } break; case FOURCC_DXT3: OSG_INFO << "ReadDDSFile info : format = DXT3" << std::endl; internalFormat = GL_COMPRESSED_RGBA_S3TC_DXT3_EXT; pixelFormat = GL_COMPRESSED_RGBA_S3TC_DXT3_EXT; break; case FOURCC_DXT5: OSG_INFO << "ReadDDSFile info : format = DXT5" << std::endl; internalFormat = GL_COMPRESSED_RGBA_S3TC_DXT5_EXT; pixelFormat = GL_COMPRESSED_RGBA_S3TC_DXT5_EXT; break; case FOURCC_ATI1: OSG_INFO << "ReadDDSFile info : format = ATI1" << std::endl; internalFormat = GL_COMPRESSED_RED_RGTC1_EXT; pixelFormat = GL_COMPRESSED_RED_RGTC1_EXT; break; case FOURCC_ATI2: OSG_INFO << "ReadDDSFile info : format = ATI2" << std::endl; internalFormat = GL_COMPRESSED_RED_GREEN_RGTC2_EXT; pixelFormat = GL_COMPRESSED_RED_GREEN_RGTC2_EXT; break; case 0x00000024: // A16B16G16R16 OSG_INFO << "ReadDDSFile info : format = A16B16G16R16" << std::endl; internalFormat = GL_RGBA; pixelFormat = GL_RGBA; dataType = GL_UNSIGNED_SHORT; break; case 0x00000071: // A16B16G16R16F OSG_INFO << "ReadDDSFile info : format = A16B16G16R16F" << std::endl; internalFormat = GL_RGBA; // why no transparency? pixelFormat = GL_RGBA; dataType = GL_HALF_FLOAT_NV; break; case 0x0000006E: // Q16W16V16U16 OSG_INFO << "ReadDDSFile info : format = Q16W16V16U16" << std::endl; internalFormat = GL_RGBA; pixelFormat = GL_RGBA; dataType = GL_UNSIGNED_SHORT; break; case 0x00000070: // G16R16F OSG_INFO << "ReadDDSFile info : G16R16F format is not supported" << std::endl; return NULL; // internalFormat = GL_RGB; // pixelFormat = must be GL_RED and GL_GREEN // dataType = GL_HALF_FLOAT_NV; break; case 0x00000073: // G32R32F OSG_INFO << "ReadDDSFile info : G32R32F format is not supported" << std::endl; return NULL; // internalFormat = GL_RGB; // pixelFormat = must be GL_RED and GL_GREEN // dataType = GL_FLOAT; break; case 0x00000072: // R32F OSG_INFO << "ReadDDSFile info : format = R32F" << std::endl; internalFormat = GL_RGB; pixelFormat = GL_RED; dataType = GL_FLOAT; break; case 0x0000006F: // R16F OSG_INFO << "ReadDDSFile info : format = R16F" << std::endl; internalFormat = GL_RGB; pixelFormat = GL_RED; dataType = GL_HALF_FLOAT_NV; break; case 0x00000074: // A32B32G32R32F OSG_INFO << "ReadDDSFile info : format = A32B32G32R32F" << std::endl; internalFormat = GL_RGBA; pixelFormat = GL_RGBA; dataType = GL_FLOAT; break; case 0x00000075: // CxV8U8 OSG_INFO << "ReadDDSFile info : CxV8U8 format is not supported" << std::endl; return NULL; case MAKEFOURCC( 'U', 'Y', 'V', 'Y' ): // not supported in OSG case MAKEFOURCC( 'U', 'Y', 'V', '2' ): // not supported in OSG case MAKEFOURCC( 'R', 'G', 'B', 'G' ): // R8G8_B8G8 -- what is it? case MAKEFOURCC( 'G', 'R', 'G', 'B' ): // G8R8_G8B8 -- what is it? //break; default: OSG_WARN << "ReadDDSFile warning: unhandled FOURCC pixel format (" << (char)((ddsd.ddpfPixelFormat.dwFourCC & 0x000000ff)) << (char)((ddsd.ddpfPixelFormat.dwFourCC & 0x0000ff00) >> 8) << (char)((ddsd.ddpfPixelFormat.dwFourCC & 0x00ff0000) >> 16) << (char)((ddsd.ddpfPixelFormat.dwFourCC & 0xff000000) >> 24) << " = 0x" << std::hex << std::setw(8) << std::setfill('0') << ddsd.ddpfPixelFormat.dwFourCC << std::dec << ") in dds file, image not loaded." << std::endl; return NULL; } } else { OSG_WARN << "ReadDDSFile warning: unhandled pixel format (ddsd.ddpfPixelFormat.dwFlags" << " = 0x" << std::hex << std::setw(8) << std::setfill('0') << ddsd.ddpfPixelFormat.dwFlags << std::dec << ") in dds file, image not loaded."<<std::endl; return NULL; } unsigned int size = ComputeImageSizeInBytes( s, t, r, pixelFormat, dataType ); // Take care of mipmaps if any. unsigned int sizeWithMipmaps = size; osg::Image::MipmapDataType mipmap_offsets; if ( ddsd.dwMipMapCount>1 ) { unsigned numMipmaps = osg::Image::computeNumberOfMipmapLevels( s, t, r ); if( numMipmaps > ddsd.dwMipMapCount ) numMipmaps = ddsd.dwMipMapCount; // array starts at 1 level offset, 0 level skipped mipmap_offsets.resize( numMipmaps - 1 ); int width = s; int height = t; int depth = r; for( unsigned int k = 0; k < mipmap_offsets.size(); ++k ) { mipmap_offsets[k] = sizeWithMipmaps; width = osg::maximum( width >> 1, 1 ); height = osg::maximum( height >> 1, 1 ); depth = osg::maximum( depth >> 1, 1 ); sizeWithMipmaps += ComputeImageSizeInBytes( width, height, depth, pixelFormat, dataType ); } } unsigned char* imageData = new unsigned char [sizeWithMipmaps]; if(!imageData) { OSG_WARN << "ReadDDSFile warning: imageData == NULL" << std::endl; return NULL; } // Read pixels in two chunks. First main image, next mipmaps. if ( !_istream.read( (char*)imageData, size ) ) { delete [] imageData; OSG_WARN << "ReadDDSFile warning: couldn't read imageData" << std::endl; return NULL; } // If loading mipmaps in second chunk fails we may still use main image if ( size < sizeWithMipmaps && !_istream.read( (char*)imageData + size, sizeWithMipmaps - size ) ) { sizeWithMipmaps = size; mipmap_offsets.resize( 0 ); OSG_WARN << "ReadDDSFile warning: couldn't read mipmapData" << std::endl; // if mipmaps read failed we leave some not used overhead memory allocated past main image // this memory will not be used but it will not cause leak in worst meaning of this word. } // Check if alpha information embedded in the 8-byte encoding blocks if (checkIfUsingOneBitAlpha) { const DXT1TexelsBlock *texelsBlock = reinterpret_cast<const DXT1TexelsBlock*>(imageData); // Only do the check on the first mipmap level, and stop when we // see the first alpha texel int i = size / sizeof(DXT1TexelsBlock); bool foundAlpha = false; while ((!foundAlpha) && (i>0)) { // See if this block might contain transparent texels if (texelsBlock->color_0<=texelsBlock->color_1) { // Scan the texels block for the '11' bit pattern that // indicates a transparent texel int j = 0; while ((!foundAlpha) && (j < 32)) { // Check for the '11' bit pattern on this texel if ( ((texelsBlock->texels4x4 >> j) & 0x03) == 0x03) { // Texture is using the 1-bit alpha encoding, so we // need to update the assumed pixel format internalFormat = GL_COMPRESSED_RGBA_S3TC_DXT1_EXT; pixelFormat = GL_COMPRESSED_RGBA_S3TC_DXT1_EXT; foundAlpha = true; } // Next texel j += 2; } } // Next block --i; ++texelsBlock; } } osgImage->setImage(s,t,r, internalFormat, pixelFormat, dataType, imageData, osg::Image::USE_NEW_DELETE); if (mipmap_offsets.size()>0) osgImage->setMipmapLevels(mipmap_offsets); if (flipDDSRead) { osgImage->setOrigin(osg::Image::BOTTOM_LEFT); osgImage->flipVertical(); } // Return Image. return osgImage.release(); } bool WriteDDSFile(const osg::Image *img, std::ostream& fout, bool autoFlipDDSWrite) { // Initialize ddsd structure and its members DDSURFACEDESC2 ddsd; memset( &ddsd, 0, sizeof( ddsd ) ); DDPIXELFORMAT ddpf; memset( &ddpf, 0, sizeof( ddpf ) ); //DDCOLORKEY ddckCKDestOverlay; //DDCOLORKEY ddckCKDestBlt; //DDCOLORKEY ddckCKSrcOverlay; //DDCOLORKEY ddckCKSrcBlt; DDSCAPS2 ddsCaps; memset( &ddsCaps, 0, sizeof( ddsCaps ) ); ddsd.dwSize = sizeof(ddsd); ddpf.dwSize = sizeof(ddpf); // Default values and initialization of structures' flags unsigned int SD_flags = DDSD_CAPS | DDSD_HEIGHT | DDSD_WIDTH | DDSD_PIXELFORMAT; unsigned int CAPS_flags = DDSCAPS_TEXTURE; unsigned int PF_flags = 0; unsigned int CAPS2_flags = 0; // Get image properties unsigned int dataType = img->getDataType(); unsigned int pixelFormat = img->getPixelFormat(); //unsigned int internalFormat = img->getInternalTextureFormat(); //unsigned int components = osg::Image::computeNumComponents(pixelFormat); unsigned int pixelSize = osg::Image::computePixelSizeInBits(pixelFormat, dataType); unsigned int imageSize = img->getImageSizeInBytes(); bool is3dImage = false; ddsd.dwWidth = img->s(); ddsd.dwHeight = img->t(); int r = img->r(); if(r > 1) /* check for 3d image */ { is3dImage = true; ddsd.dwDepth = r; SD_flags |= DDSD_DEPTH; CAPS_flags |= DDSCAPS_COMPLEX; CAPS2_flags |= DDSCAPS2_VOLUME; } // Determine format - set flags and ddsd, ddpf properties switch (pixelFormat) { //Uncompressed case GL_RGBA: { ddpf.dwRBitMask = 0x000000ff; ddpf.dwGBitMask = 0x0000ff00; ddpf.dwBBitMask = 0x00ff0000; ddpf.dwRGBAlphaBitMask = 0xff000000; PF_flags |= (DDPF_ALPHAPIXELS | DDPF_RGB); ddpf.dwRGBBitCount = pixelSize; ddsd.lPitch = img->getRowSizeInBytes(); SD_flags |= DDSD_PITCH; } break; case GL_BGRA: { ddpf.dwBBitMask = 0x000000ff; ddpf.dwGBitMask = 0x0000ff00; ddpf.dwRBitMask = 0x00ff0000; ddpf.dwRGBAlphaBitMask = 0xff000000; PF_flags |= (DDPF_ALPHAPIXELS | DDPF_RGB); ddpf.dwRGBBitCount = pixelSize; ddsd.lPitch = img->getRowSizeInBytes(); SD_flags |= DDSD_PITCH; } break; case GL_LUMINANCE_ALPHA: { ddpf.dwRBitMask = 0x000000ff; ddpf.dwRGBAlphaBitMask = 0x0000ff00; PF_flags |= (DDPF_ALPHAPIXELS | DDPF_LUMINANCE); ddpf.dwRGBBitCount = pixelSize; ddsd.lPitch = img->getRowSizeInBytes(); SD_flags |= DDSD_PITCH; } break; case GL_RGB: { ddpf.dwRBitMask = 0x000000ff; ddpf.dwGBitMask = 0x0000ff00; ddpf.dwBBitMask = 0x00ff0000; PF_flags |= DDPF_RGB; ddpf.dwRGBBitCount = pixelSize; ddsd.lPitch = img->getRowSizeInBytes(); SD_flags |= DDSD_PITCH; } break; case GL_LUMINANCE: { ddpf.dwRBitMask = 0x000000ff; PF_flags |= DDPF_LUMINANCE; ddpf.dwRGBBitCount = pixelSize; ddsd.lPitch = img->getRowSizeInBytes(); SD_flags |= DDSD_PITCH; } break; case GL_ALPHA: { ddpf.dwRGBAlphaBitMask = 0x000000ff; PF_flags |= DDPF_ALPHA; ddpf.dwRGBBitCount = pixelSize; ddsd.lPitch = img->getRowSizeInBytes(); SD_flags |= DDSD_PITCH; } break; //Compressed case GL_COMPRESSED_RGBA_S3TC_DXT1_EXT: { ddpf.dwFourCC = FOURCC_DXT1; PF_flags |= (DDPF_ALPHAPIXELS | DDPF_FOURCC); ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_RGBA_S3TC_DXT3_EXT: { ddpf.dwFourCC = FOURCC_DXT3; PF_flags |= (DDPF_ALPHAPIXELS | DDPF_FOURCC); ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_RGBA_S3TC_DXT5_EXT: { ddpf.dwFourCC = FOURCC_DXT5; PF_flags |= (DDPF_ALPHAPIXELS | DDPF_FOURCC); ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_RGB_S3TC_DXT1_EXT: { ddpf.dwFourCC = FOURCC_DXT1; PF_flags |= DDPF_FOURCC; /* No alpha here */ ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_SIGNED_RED_RGTC1_EXT: { ddpf.dwFourCC = FOURCC_ATI1; PF_flags |= DDPF_FOURCC; /* No alpha here */ ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_RED_RGTC1_EXT: { ddpf.dwFourCC = FOURCC_ATI1; PF_flags |= DDPF_FOURCC; /* No alpha here */ ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_SIGNED_RED_GREEN_RGTC2_EXT: { ddpf.dwFourCC = FOURCC_ATI2; PF_flags |= DDPF_FOURCC; /* No alpha here */ ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; case GL_COMPRESSED_RED_GREEN_RGTC2_EXT: { ddpf.dwFourCC = FOURCC_ATI2; PF_flags |= DDPF_FOURCC; /* No alpha here */ ddsd.dwLinearSize = imageSize; SD_flags |= DDSD_LINEARSIZE; } break; default: OSG_WARN<<"Warning:: unhandled pixel format in image, file cannot be written."<<std::endl; return false; } int size = img->getTotalSizeInBytes(); // set even more flags if( !img->isMipmap() ) { OSG_INFO<<"no mipmaps to write out."<<std::endl; } else if( img->getPacking() > 1 ) { OSG_WARN<<"Warning: mipmaps not written. DDS requires packing == 1."<<std::endl; } else { // image contains mipmaps and has 1 byte alignment SD_flags |= DDSD_MIPMAPCOUNT; CAPS_flags |= DDSCAPS_COMPLEX | DDSCAPS_MIPMAP; ddsd.dwMipMapCount = img->getNumMipmapLevels(); size = img->getTotalSizeInBytesIncludingMipmaps(); OSG_INFO<<"writing out with mipmaps ddsd.dwMipMapCount"<<ddsd.dwMipMapCount<<std::endl; } // Assign flags and structure members of ddsd ddsd.dwFlags = SD_flags; ddpf.dwFlags = PF_flags; ddsCaps.dwCaps = CAPS_flags; ddsCaps.dwCaps2 = CAPS2_flags; ddsd.ddpfPixelFormat = ddpf; ddsd.ddsCaps = ddsCaps; osg::ref_ptr<const osg::Image> source; // int isize = img->getTotalSizeInBytesIncludingMipmaps(); if (autoFlipDDSWrite && img->getOrigin() == osg::Image::BOTTOM_LEFT) { osg::ref_ptr<osg::Image> copy = new osg::Image(*img,osg::CopyOp::DEEP_COPY_ALL); copy->flipVertical(); source = copy; } else { source = img; } // Write DDS file fout.write("DDS ", 4); /* write FOURCC */ fout.write(reinterpret_cast<char*>(&ddsd), sizeof(ddsd)); /* write file header */ fout.write(reinterpret_cast<const char*>(source->data()), size ); // Check for correct saving if ( fout.fail() ) return false; // If we get that far the file was saved properly // return true; } class ReaderWriterDDS : public osgDB::ReaderWriter { public: ReaderWriterDDS() { supportsExtension("dds","DDS image format"); supportsOption("ddsFlipRead", "(Read option) Flips the image vertically (this is not the default for backward compatibility reasons)."); supportsOption("dds_flip", "(Read option) Synonym for 'ddsFlipRead'."); //supportsOption("ddsNoFlipRead", "(Read option) Avoid flipping the image vertically."); supportsOption("ddsNoAutoFlipWrite", "(Write option) Avoid automatically flipping the image vertically when writing, depending on the origin (Image::getOrigin())."); } virtual const char* className() const { return "DDS Image Reader/Writer"; } virtual ReadResult readObject(const std::string& file, const osgDB::ReaderWriter::Options* options) const { return readImage(file,options); } virtual ReadResult readObject(std::istream& fin, const Options* options) const { return readImage(fin,options); } virtual ReadResult readImage(const std::string& file, const osgDB::ReaderWriter::Options* options) const { std::string ext = osgDB::getLowerCaseFileExtension(file); if (!acceptsExtension(ext)) return ReadResult::FILE_NOT_HANDLED; std::string fileName = osgDB::findDataFile( file, options ); if (fileName.empty()) return ReadResult::FILE_NOT_FOUND; osgDB::ifstream stream(fileName.c_str(), std::ios::in | std::ios::binary); if(!stream) return ReadResult::FILE_NOT_HANDLED; ReadResult rr = readImage(stream, options); if(rr.validImage()) rr.getImage()->setFileName(file); return rr; } virtual ReadResult readImage(std::istream& fin, const Options* options) const { // Sukender: The default for DDS should be to flip image (seems the convention for several "heavy" 3D engines), but this surely breaks backward compatibility. bool flipDDSRead(false); if (options) { std::istringstream iss(options->getOptionString()); std::string opt; while (iss >> opt) { if (opt == "dds_flip" || opt == "ddsFlipRead") flipDDSRead = true; if (opt == "ddsNoFlipRead") flipDDSRead = false; } } osg::Image* osgImage = ReadDDSFile(fin, flipDDSRead); if (osgImage==NULL) return ReadResult::FILE_NOT_HANDLED; return osgImage; } virtual WriteResult writeObject(const osg::Object& object,const std::string& file, const osgDB::ReaderWriter::Options* options) const { const osg::Image* image = dynamic_cast<const osg::Image*>(&object); if (!image) return WriteResult::FILE_NOT_HANDLED; return writeImage(*image,file,options); } virtual WriteResult writeObject(const osg::Object& object,std::ostream& fout,const Options* options) const { const osg::Image* image = dynamic_cast<const osg::Image*>(&object); if (!image) return WriteResult::FILE_NOT_HANDLED; return writeImage(*image,fout,options); } virtual WriteResult writeImage(const osg::Image &image,const std::string& file, const osgDB::ReaderWriter::Options* options) const { std::string ext = osgDB::getFileExtension(file); if (!acceptsExtension(ext)) return WriteResult::FILE_NOT_HANDLED; osgDB::ofstream fout(file.c_str(), std::ios::out | std::ios::binary); if(!fout) return WriteResult::ERROR_IN_WRITING_FILE; return writeImage(image,fout,options); } virtual WriteResult writeImage(const osg::Image& image,std::ostream& fout,const Options* options) const { bool noAutoFlipDDSWrite = options && options->getOptionString().find("ddsNoAutoFlipWrite")!=std::string::npos; bool success = WriteDDSFile(&image, fout, !noAutoFlipDDSWrite); if(success) return WriteResult::FILE_SAVED; else return WriteResult::ERROR_IN_WRITING_FILE; } }; // now register with Registry to instantiate the above // reader/writer. REGISTER_OSGPLUGIN(dds, ReaderWriterDDS)
_______________________________________________ osg-submissions mailing list [email protected] http://lists.openscenegraph.org/listinfo.cgi/osg-submissions-openscenegraph.org
