#include "../zstd/zstd.h" #include "TracyEvent.hpp" #include "TracyTextureCompression.hpp" namespace tracy { TextureCompression::TextureCompression() : m_buf( nullptr ) , m_bufSize( 0 ) , m_cctx( ZSTD_createCCtx() ) , m_dctx( ZSTD_createDCtx() ) { } TextureCompression::~TextureCompression() { delete[] m_buf; ZSTD_freeCCtx( m_cctx ); ZSTD_freeDCtx( m_dctx ); } uint32_t TextureCompression::Pack( struct ZSTD_CCtx_s* ctx, char*& buf, size_t& bufsz, const char* image, uint32_t inBytes ) { const auto maxout = ZSTD_COMPRESSBOUND( inBytes ); if( bufsz < maxout ) { bufsz = maxout; delete[] buf; buf = new char[maxout]; } assert( ctx ); auto ret = (uint32_t)ZSTD_compressCCtx( ctx, buf, maxout, image, inBytes, 3 ); #ifndef TRACY_NO_STATISTICS m_inputBytes.fetch_add( inBytes, std::memory_order_relaxed ); m_outputBytes.fetch_add( ret, std::memory_order_relaxed ); #endif return ret; } const char* TextureCompression::Unpack( const FrameImage& image ) { const auto outsz = size_t( image.w ) * size_t( image.h ) / 2; if( m_bufSize < outsz ) { m_bufSize = outsz; delete[] m_buf; m_buf = new char[outsz]; } assert( m_dctx ); ZSTD_decompressDCtx( m_dctx, m_buf, outsz, image.ptr, image.csz ); return m_buf; } static constexpr uint8_t Dxtc4To3Table[256] = { 85, 84, 86, 86, 81, 80, 82, 82, 89, 88, 90, 90, 89, 88, 90, 90, 69, 68, 70, 70, 65, 64, 66, 66, 73, 72, 74, 74, 73, 72, 74, 74, 101, 100, 102, 102, 97, 96, 98, 98, 105, 104, 106, 106, 105, 104, 106, 106, 101, 100, 102, 102, 97, 96, 98, 98, 105, 104, 106, 106, 105, 104, 106, 106, 21, 20, 22, 22, 17, 16, 18, 18, 25, 24, 26, 26, 25, 24, 26, 26, 5, 4, 6, 6, 1, 0, 2, 2, 9, 8, 10, 10, 9, 8, 10, 10, 37, 36, 38, 38, 33, 32, 34, 34, 41, 40, 42, 42, 41, 40, 42, 42, 37, 36, 38, 38, 33, 32, 34, 34, 41, 40, 42, 42, 41, 40, 42, 42, 149, 148, 150, 150, 145, 144, 146, 146, 153, 152, 154, 154, 153, 152, 154, 154, 133, 132, 134, 134, 129, 128, 130, 130, 137, 136, 138, 138, 137, 136, 138, 138, 165, 164, 166, 166, 161, 160, 162, 162, 169, 168, 170, 170, 169, 168, 170, 170, 165, 164, 166, 166, 161, 160, 162, 162, 169, 168, 170, 170, 169, 168, 170, 170, 149, 148, 150, 150, 145, 144, 146, 146, 153, 152, 154, 154, 153, 152, 154, 154, 133, 132, 134, 134, 129, 128, 130, 130, 137, 136, 138, 138, 137, 136, 138, 138, 165, 164, 166, 166, 161, 160, 162, 162, 169, 168, 170, 170, 169, 168, 170, 170, 165, 164, 166, 166, 161, 160, 162, 162, 169, 168, 170, 170, 169, 168, 170, 170 }; void TextureCompression::Rdo( char* data, size_t blocks ) { assert( blocks > 0 ); do { uint32_t idx; memcpy( &idx, data+4, 4 ); if( idx == 0x55555555 ) { data += 8; continue; } uint16_t c0, c1; memcpy( &c0, data, 2 ); memcpy( &c1, data+2, 2 ); const int r0b = c0 & 0xF800; const int g0b = c0 & 0x07E0; const int b0b = c0 & 0x001F; const int r1b = c1 & 0xF800; const int g1b = c1 & 0x07E0; const int b1b = c1 & 0x001F; const int r0 = ( r0b >> 8 ) | ( r0b >> 13 ); const int g0 = ( g0b >> 3 ) | ( g0b >> 9 ); const int b0 = ( b0b << 3 ) | ( b0b >> 2 ); const int r1 = ( r1b >> 8 ) | ( r1b >> 13 ); const int g1 = ( g1b >> 3 ) | ( g1b >> 9 ); const int b1 = ( b1b << 3 ) | ( b1b >> 2 ); const int dr = abs( r0 - r1 ); const int dg = abs( g0 - g1 ); const int db = abs( b0 - b1 ); const int maxChan1 = std::max( { r0-1, g0, b0-2 } ); const int maxDelta1 = std::max( { dr-1, dg, db-2 } ); const int tr1a = 16; const int tr1b = 45; int tr1; if( maxChan1 < tr1a ) { tr1 = 12; } else if( maxChan1 < tr1b ) { tr1 = 6; } else { tr1 = 3; } if( maxDelta1 <= tr1 ) { uint64_t blk = ( ( ( r0b + r1b ) >> 1 ) & 0xF800 ) | ( ( ( g0b + g1b ) >> 1 ) & 0x07E0 ) | ( ( ( b0b + b1b ) >> 1 ) ); memcpy( data, &blk, 8 ); } else { const int maxChan23 = std::max( { r0-2, g0, b0-5 } ); const int maxDelta23 = std::max( { dr-2, dg, db-5 } ); const int tr2a = 32; const int tr2b = 48; int tr2 = 0; if( maxChan23 < tr2a ) { tr2 = 12; } else if( maxChan23 < tr2b ) { tr2 = 6; } else { tr2 = 3; } if( maxDelta23 <= tr2 ) { idx &= 0x55555555; memcpy( data+4, &idx, 4 ); } else { const int tr3a = 48; const int tr3b = 64; int tr3; if( maxChan23 < tr3a ) { tr3 = 48; } else if( maxChan23 < tr3b ) { tr3 = 32; } else { tr3 = 24; } if( maxDelta23 <= tr3 ) { memcpy( data, &c1, 2 ); memcpy( data+2, &c0, 2 ); uint8_t tmp[4]; memcpy( tmp, &idx, 4 ); for( int k=0; k<4; k++ ) tmp[k] = Dxtc4To3Table[tmp[k]]; memcpy( data+4, tmp, 4 ); } } } data += 8; } while( --blocks ); } static constexpr uint8_t DxtcIndexTable[256] = { 85, 87, 86, 84, 93, 95, 94, 92, 89, 91, 90, 88, 81, 83, 82, 80, 117, 119, 118, 116, 125, 127, 126, 124, 121, 123, 122, 120, 113, 115, 114, 112, 101, 103, 102, 100, 109, 111, 110, 108, 105, 107, 106, 104, 97, 99, 98, 96, 69, 71, 70, 68, 77, 79, 78, 76, 73, 75, 74, 72, 65, 67, 66, 64, 213, 215, 214, 212, 221, 223, 222, 220, 217, 219, 218, 216, 209, 211, 210, 208, 245, 247, 246, 244, 253, 255, 254, 252, 249, 251, 250, 248, 241, 243, 242, 240, 229, 231, 230, 228, 237, 239, 238, 236, 233, 235, 234, 232, 225, 227, 226, 224, 197, 199, 198, 196, 205, 207, 206, 204, 201, 203, 202, 200, 193, 195, 194, 192, 149, 151, 150, 148, 157, 159, 158, 156, 153, 155, 154, 152, 145, 147, 146, 144, 181, 183, 182, 180, 189, 191, 190, 188, 185, 187, 186, 184, 177, 179, 178, 176, 165, 167, 166, 164, 173, 175, 174, 172, 169, 171, 170, 168, 161, 163, 162, 160, 133, 135, 134, 132, 141, 143, 142, 140, 137, 139, 138, 136, 129, 131, 130, 128, 21, 23, 22, 20, 29, 31, 30, 28, 25, 27, 26, 24, 17, 19, 18, 16, 53, 55, 54, 52, 61, 63, 62, 60, 57, 59, 58, 56, 49, 51, 50, 48, 37, 39, 38, 36, 45, 47, 46, 44, 41, 43, 42, 40, 33, 35, 34, 32, 5, 7, 6, 4, 13, 15, 14, 12, 9, 11, 10, 8, 1, 3, 2, 0 }; void TextureCompression::FixOrder( char* data, size_t blocks ) { assert( blocks > 0 ); do { uint32_t res = 0; uint32_t tmp; memcpy( &tmp, data+4, 4 ); for( int k=0; k<4; k++ ) res |= DxtcIndexTable[(tmp >> (k*8)) & 0xFF] << (k*8); memcpy( data+4, &res, 4 ); data += 8; } while( --blocks ); } }