zhangengru/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2025-03-20 07:40:02 +08:00
Code Issues Projects Wiki Activity

Add LZ4HC support to FileWrite.

Browse Source
This commit is contained in:
Bartosz Taudul
2018-08-26 16:25:43 +02:00
parent 0f0528ca3d
commit b3b12f76f3
7 changed files with 52 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/TracyFileWrite.hpp
+28 -6
View File
@@ -7,6 +7,7 @@
#include "TracyFileHeader.hpp"
#include "../common/tracy_lz4.hpp"
#include "../common/tracy_lz4hc.hpp"
#include "../common/TracyForceInline.hpp"
namespace tracy
@@ -15,10 +16,10 @@ namespace tracy
class FileWrite
{
public:
static FileWrite* Open( const char* fn )
static FileWrite* Open( const char* fn, bool hc = false )
{
auto f = fopen( fn, "wb" );
return f ? new FileWrite( f ) : nullptr;
return f ? new FileWrite( f, hc ) : nullptr;
}
~FileWrite()
@@ -28,7 +29,9 @@ public:
WriteLz4Block();
}
fclose( m_file );
LZ4_freeStream( m_stream );
if( m_stream ) LZ4_freeStream( m_stream );
if( m_streamHC ) LZ4_freeStreamHC( m_streamHC );
}
tracy_force_inline void Write( const void* ptr, size_t size )
@@ -44,13 +47,23 @@ public:
}
private:
FileWrite( FILE* f )
: m_stream( LZ4_createStream() )
FileWrite( FILE* f, bool hc )
: m_stream( nullptr )
, m_streamHC( nullptr )
, m_file( f )
, m_buf( m_bufData[0] )
, m_second( m_bufData[1] )
, m_offset( 0 )
{
if( hc )
{
m_streamHC = LZ4_createStreamHC();
}
else
{
m_stream = LZ4_createStream();
}
fwrite( Lz4Header, 1, sizeof( Lz4Header ), m_file );
}
@@ -81,7 +94,15 @@ private:
void WriteLz4Block()
{
char lz4[LZ4Size];
const uint32_t sz = LZ4_compress_fast_continue( m_stream, m_buf, lz4, m_offset, LZ4Size, 1 );
uint32_t sz;
if( m_stream )
{
sz = LZ4_compress_fast_continue( m_stream, m_buf, lz4, m_offset, LZ4Size, 1 );
}
else
{
sz = LZ4_compress_HC_continue( m_streamHC, m_buf, lz4, m_offset, LZ4Size );
}
fwrite( &sz, 1, sizeof( sz ), m_file );
fwrite( lz4, 1, sz, m_file );
m_offset = 0;
@@ -92,6 +113,7 @@ private:
enum { LZ4Size = LZ4_COMPRESSBOUND( BufSize ) };
LZ4_stream_t* m_stream;
LZ4_streamHC_t* m_streamHC;
FILE* m_file;
char m_bufData[2][BufSize];
char* m_buf;