F:/My Projects/LSEngine/Modules/LSCompressionLib/Src/ZLib/LSCZLib.h

#ifndef __LSC_ZLIB_H__
#define __LSC_ZLIB_H__

#include "../LSCCompressionLib.h"
#include "../BitStream/LSCBitStream.h"
#include "Streams/LSSTDExternalByteStream.h"
#include "Vector/LSTLVectorPoD.h"

namespace lsc {

        class CZLib {
        public :
                // == Various constructors.
                LSE_CALLCTOR                                    CZLib();


                // == Types.
                typedef CVectorPoD<LSUINT8, LSUINT32, 0xFFFFUL>
                                                                                LSC_ZLIB_BUFFER;

                // == Functions.
                LSBOOL LSE_CALL                                 DecodeZLib( const LSUINT8 * _pui8Buffer, LSUINTPTR _uiptrLen,
                        LSBOOL _bSkipHeaders, LSC_ZLIB_BUFFER &_zbOut );


        protected :
                // == Enumerations.
                enum {
                        LSC_FAST_BITS                           = 9UL,
                        LSC_MASK                                        = (1UL << LSC_FAST_BITS) - 1UL,
                };


                // == Types.
                typedef struct LSC_HUFFMAN {
                        LSINT32                                         i32MaxCode[17];

                        LSUINT16                                        ui16Fast[1UL<<LSC_FAST_BITS];

                        LSUINT16                                        ui16FirstCode[16];

                        LSUINT16                                        ui16FirstSymbol[16];

                        LSUINT16                                        ui16Values[288];

                        LSUINT8                                         ui8Size[288];
                } * LPLSC_HUFFMAN, * const LPCLSC_HUFFMAN;

                // == Members.
                LSUINT32                                                m_ui32BufferedBits;

                LSUINT32                                                m_ui32TotalBitsBuffered;

                LSC_HUFFMAN                                             m_hLength;

                LSC_HUFFMAN                                             m_hDistance;

                static LSUINT8                                  m_ui8DefaultLengths[288];

                static LSUINT8                                  m_ui8DefaultDistances[32];

                static CCriticalSection                 m_scCrit;


                // == Functions.
                LSBOOL LSE_CALL                                 FillBits( const CExternalByteStream &_ebsStream );

                LSBOOL LSE_CALL                                 Receive( const CExternalByteStream &_ebsStream, LSUINT32 _ui32Total, LSUINT32 &_ui32Return );

                LSBOOL LSE_CALL                                 ComputeHuffmanCodes( const CExternalByteStream &_ebsStream );

                LSBOOL LSE_CALL                                 BuildHuffman( LSC_HUFFMAN &_hHuffman,
                        const LSUINT8 * _pui8Codes, LSUINT32 _ui32Total );

                LSINT32 LSE_CALL                                DecodeHuffman( const CExternalByteStream &_ebsStream, LSC_HUFFMAN &_hHuffman );

                LSBOOL LSE_CALL                                 ParseHuffmanBlock( const CExternalByteStream &_ebsStream, LSC_ZLIB_BUFFER &_zbOut );

                LSBOOL LSE_CALL                                 ParseUncompressed( const CExternalByteStream &_ebsStream, LSC_ZLIB_BUFFER &_zbOut );

                static LSBOOL LSE_CALL                  CheckHeader( const CExternalByteStream &_ebsStream );

                static LSBOOL LSE_CALL                  ReadUi8( const CExternalByteStream &_ebsStream, LSUINT8 &_ui8Dst );

                static LSE_INLINE LSUINT16 LSE_CALL
                                                                                ReverseBits( LSUINT16 _ui16ReverseMe, LSUINT32 _ui32Shift );

                static LSVOID LSE_CALL                  SetDefaults();
        };


        // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
        // DEFINITIONS
        // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
        // == Functions.
         LSE_INLINE LSUINT16 LSE_CALL CZLib::ReverseBits( LSUINT16 _ui16ReverseMe, LSUINT32 _ui32Shift ) {
                 return static_cast<LSUINT16>(CStd::BitSwap16( _ui16ReverseMe ) >> (16 - _ui32Shift));
         }

}       // namespace lsc

#endif  // __LSC_ZLIB_H__