ddnet/src/engine/shared/compression.cpp

104 lines
2.3 KiB
C++
Raw Normal View History

2010-11-20 10:37:14 +00:00
/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */
/* If you are missing that file, acquire a complete release at teeworlds.com. */
2010-05-29 07:25:38 +00:00
#include <base/system.h>
#include "compression.h"
// Format: ESDDDDDD EDDDDDDD EDD... Extended, Data, Sign
unsigned char *CVariableInt::Pack(unsigned char *pDst, int i, int DstSize)
{
if(DstSize <= 0)
return 0;
DstSize--;
2021-12-27 12:17:39 +00:00
*pDst = 0;
if(i < 0)
{
*pDst |= 0x40; // set sign bit
i = ~i;
}
2010-05-29 07:25:38 +00:00
*pDst |= i & 0x3F; // pack 6bit into dst
2010-05-29 07:25:38 +00:00
i >>= 6; // discard 6 bits
2021-12-27 12:17:39 +00:00
while(i)
2010-05-29 07:25:38 +00:00
{
if(DstSize <= 0)
return 0;
2010-05-29 07:25:38 +00:00
*pDst |= 0x80; // set extend bit
DstSize--;
2021-12-27 12:17:39 +00:00
pDst++;
*pDst = i & 0x7F; // pack 7bit
i >>= 7; // discard 7 bits
2010-05-29 07:25:38 +00:00
}
pDst++;
return pDst;
}
const unsigned char *CVariableInt::Unpack(const unsigned char *pSrc, int *pInOut, int SrcSize)
{
if(SrcSize <= 0)
return 0;
const int Sign = (*pSrc >> 6) & 1;
*pInOut = *pSrc & 0x3F;
SrcSize--;
2010-05-29 07:25:38 +00:00
const static int s_aMasks[] = {0x7F, 0x7F, 0x7F, 0x0F};
const static int s_aShifts[] = {6, 6 + 7, 6 + 7 + 7, 6 + 7 + 7 + 7};
2010-05-29 07:25:38 +00:00
for(unsigned i = 0; i < sizeof(s_aMasks) / sizeof(int); i++)
{
if(!(*pSrc & 0x80))
break;
if(SrcSize <= 0)
return 0;
SrcSize--;
2010-05-29 07:25:38 +00:00
pSrc++;
*pInOut |= (*pSrc & s_aMasks[i]) << s_aShifts[i];
}
2010-05-29 07:25:38 +00:00
pSrc++;
*pInOut ^= -Sign; // if(sign) *i = ~(*i)
return pSrc;
}
2010-05-29 07:25:38 +00:00
long CVariableInt::Decompress(const void *pSrc_, int SrcSize, void *pDst_, int DstSize)
2010-05-29 07:25:38 +00:00
{
dbg_assert(DstSize % sizeof(int) == 0, "invalid bounds");
2010-05-29 07:25:38 +00:00
const unsigned char *pSrc = (unsigned char *)pSrc_;
const unsigned char *pSrcEnd = pSrc + SrcSize;
2010-05-29 07:25:38 +00:00
int *pDst = (int *)pDst_;
const int *pDstEnd = pDst + DstSize / sizeof(int);
while(pSrc < pSrcEnd)
2010-05-29 07:25:38 +00:00
{
2017-09-16 17:30:08 +00:00
if(pDst >= pDstEnd)
return -1;
pSrc = CVariableInt::Unpack(pSrc, pDst, pSrcEnd - pSrc);
if(!pSrc)
return -1;
2010-05-29 07:25:38 +00:00
pDst++;
}
return (long)((unsigned char *)pDst - (unsigned char *)pDst_);
2010-05-29 07:25:38 +00:00
}
long CVariableInt::Compress(const void *pSrc_, int SrcSize, void *pDst_, int DstSize)
2010-05-29 07:25:38 +00:00
{
dbg_assert(SrcSize % sizeof(int) == 0, "invalid bounds");
const int *pSrc = (int *)pSrc_;
2010-05-29 07:25:38 +00:00
unsigned char *pDst = (unsigned char *)pDst_;
const unsigned char *pDstEnd = pDst + DstSize;
SrcSize /= sizeof(int);
while(SrcSize)
2010-05-29 07:25:38 +00:00
{
pDst = CVariableInt::Pack(pDst, *pSrc, pDstEnd - pDst);
if(!pDst)
2017-09-16 17:30:08 +00:00
return -1;
SrcSize--;
2010-05-29 07:25:38 +00:00
pSrc++;
}
return (long)(pDst - (unsigned char *)pDst_);
2010-05-29 07:25:38 +00:00
}