CnC_Remastered_Collection/REDALERT/LZOPIPE.CPP
PG-SteveT 03416d24e1 Initial Source Code commit
Initial commit of original Tiberian Dawn and Red Alert source code converted to build as DLLs, and compatible with the release version of Command & Conquer Remastered.
2020-05-27 12:16:20 -07:00

319 lines
15 KiB
C++

//
// Copyright 2020 Electronic Arts Inc.
//
// TiberianDawn.DLL and RedAlert.dll and corresponding source code is free
// software: you can redistribute it and/or modify it under the terms of
// the GNU General Public License as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
// TiberianDawn.DLL and RedAlert.dll and corresponding source code is distributed
// in the hope that it will be useful, but with permitted additional restrictions
// under Section 7 of the GPL. See the GNU General Public License in LICENSE.TXT
// distributed with this program. You should have received a copy of the
// GNU General Public License along with permitted additional restrictions
// with this program. If not, see https://github.com/electronicarts/CnC_Remastered_Collection
/* $Header: /CounterStrike/LZOPIPE.CPP 1 3/03/97 10:25a Joe_bostic $ */
/***********************************************************************************************
*** C O N F I D E N T I A L --- W E S T W O O D S T U D I O S ***
***********************************************************************************************
* *
* Project Name : Command & Conquer *
* *
* File Name : LZWPIPE.CPP *
* *
* Programmer : Joe L. Bostic *
* *
* Start Date : 06/30/96 *
* *
* Last Update : July 4, 1996 [JLB] *
* *
*---------------------------------------------------------------------------------------------*
* Functions: *
* LZWPipe::Flush -- Flushes any partially accumulated block. *
* LZWPipe::LZWPipe -- Constructor for the LZO processor pipe. *
* LZWPipe::Put -- Send some data through the LZO processor pipe. *
* LZWPipe::~LZWPipe -- Deconstructor for the LZO pipe object. *
* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
#include "lzopipe.h"
#include "lzo.h"
#include "buff.h"
#include <string.h>
#include <assert.h>
/***********************************************************************************************
* LZOPipe::LZOPipe -- Constructor for the LZO processor pipe. *
* *
* This will initialize the LZOPipe object so that it is prepared for compression or *
* decompression as indicated. *
* *
* INPUT: decrypt -- Should decompression be performed? *
* *
* blocksize-- The size of the data blocks to process. *
* *
* OUTPUT: none *
* *
* WARNINGS: none *
* *
* HISTORY: *
* 07/04/1996 JLB : Created. *
*=============================================================================================*/
LZOPipe::LZOPipe(CompControl control, int blocksize) :
Control(control),
Counter(0),
Buffer(NULL),
Buffer2(NULL),
BlockSize(blocksize)
{
SafetyMargin = BlockSize;
Buffer = new char[BlockSize+SafetyMargin];
Buffer2 = new char[BlockSize+SafetyMargin];
BlockHeader.CompCount = 0xFFFF;
}
/***********************************************************************************************
* LZOPipe::~LZOPipe -- Deconstructor for the LZO pipe object. *
* *
* This will free any buffers it may have allocated. *
* *
* INPUT: none *
* *
* OUTPUT: none *
* *
* WARNINGS: none *
* *
* HISTORY: *
* 07/04/1996 JLB : Created. *
*=============================================================================================*/
LZOPipe::~LZOPipe(void)
{
delete [] Buffer;
Buffer = NULL;
delete [] Buffer2;
Buffer2 = NULL;
}
/***********************************************************************************************
* LZOPipe::Put -- Send some data through the LZO processor pipe. *
* *
* This routine will take the data requested and process it (decompression or compression). *
* It does this by accumulating the necessary bytes to make a whole block. Then the block *
* is processed and the entire contents are flushed to the next pipe segment in the chain. *
* *
* INPUT: source -- Pointer to the data to be fed to this LZO processor. *
* *
* length -- The number of bytes received. *
* *
* OUTPUT: Returns with the actual number of bytes output at the far distant final link in *
* the pipe chain. *
* *
* WARNINGS: The compression process may be slow as well as consuming two buffers. *
* *
* HISTORY: *
* 07/04/1996 JLB : Created. *
*=============================================================================================*/
int LZOPipe::Put(void const * source, int slen)
{
if (source == NULL || slen < 1) {
return(Pipe::Put(source, slen));
}
assert(Buffer != NULL);
int total = 0;
/*
** Copy as much as can fit into the buffer from the source data supplied.
*/
if (Control == DECOMPRESS) {
while (slen > 0) {
/*
** First check to see if we are in the block header accumulation phase.
** When a whole block header has been accumulated, only then will the regular
** data processing begin for the block.
*/
if (BlockHeader.CompCount == 0xFFFF) {
int len = (slen < ((int)sizeof(BlockHeader)-Counter)) ? slen : (sizeof(BlockHeader)-Counter);
memmove(&Buffer[Counter], source, len);
source = ((char *)source) + len;
slen -= len;
Counter += len;
/*
** A whole block header has been accumulated. Store it for safekeeping.
*/
if (Counter == sizeof(BlockHeader)) {
memmove(&BlockHeader, Buffer, sizeof(BlockHeader));
Counter = 0;
}
}
/*
** Fill the buffer with compressed data until there is enough to make a whole
** data block.
*/
if (slen > 0) {
int len = (slen < (BlockHeader.CompCount-Counter)) ? slen : (BlockHeader.CompCount-Counter);
memmove(&Buffer[Counter], source, len);
slen -= len;
source = ((char *)source) + len;
Counter += len;
/*
** If an entire block has been accumulated, then uncompress it and feed it
** through the pipe.
*/
if (Counter == BlockHeader.CompCount) {
unsigned int length = sizeof (Buffer2);
lzo1x_decompress ((unsigned char*)Buffer, BlockHeader.CompCount, (unsigned char*)Buffer2, &length, NULL);
total += Pipe::Put(Buffer2, BlockHeader.UncompCount);
Counter = 0;
BlockHeader.CompCount = 0xFFFF;
}
}
}
} else {
/*
** If the buffer already contains some data, then any new data must be stored
** into the staging buffer until a full set has been accumulated.
*/
if (Counter > 0) {
int tocopy = (slen < (BlockSize-Counter)) ? slen : (BlockSize-Counter);
memmove(&Buffer[Counter], source, tocopy);
source = ((char *)source) + tocopy;
slen -= tocopy;
Counter += tocopy;
if (Counter == BlockSize) {
unsigned int len = sizeof (Buffer2);
char *dictionary = new char [64*1024];
lzo1x_1_compress ((unsigned char*)Buffer, BlockSize, (unsigned char*)Buffer2, &len, dictionary);
delete [] dictionary;
BlockHeader.CompCount = (unsigned short)len;
BlockHeader.UncompCount = (unsigned short)BlockSize;
total += Pipe::Put(&BlockHeader, sizeof(BlockHeader));
total += Pipe::Put(Buffer2, len);
Counter = 0;
}
}
/*
** Process the source data in whole block chunks until there is insufficient
** source data left for a whole data block.
*/
while (slen >= BlockSize) {
unsigned int len = sizeof (Buffer2);
char *dictionary = new char [64*1024];
lzo1x_1_compress ((unsigned char*)source, BlockSize, (unsigned char*)Buffer2, &len, dictionary);
delete [] dictionary;
source = ((char *)source) + BlockSize;
slen -= BlockSize;
BlockHeader.CompCount = (unsigned short)len;
BlockHeader.UncompCount = (unsigned short)BlockSize;
total += Pipe::Put(&BlockHeader, sizeof(BlockHeader));
total += Pipe::Put(Buffer2, len);
}
/*
** If there is any remaining data, then it is stored into the buffer
** until a full data block has been accumulated.
*/
if (slen > 0) {
memmove(Buffer, source, slen);
Counter = slen;
}
}
return(total);
}
/***********************************************************************************************
* LZOPipe::Flush -- Flushes any partially accumulated block. *
* *
* This routine is called when any buffered data must be flushed out the pipe. For the *
* compression process, this will generate the sub-sized compressed block. For *
* decompression, this routine should not have any data in the buffer. In such a case, it *
* means that the data source was prematurely truncated. In such a case, just dump the *
* accumulated data through the pipe. *
* *
* INPUT: none *
* *
* OUTPUT: Returns with the actual number of data bytes output to the distant final link in *
* the pipe chain. *
* *
* WARNINGS: none *
* *
* HISTORY: *
* 07/04/1996 JLB : Created. *
*=============================================================================================*/
int LZOPipe::Flush(void)
{
assert(Buffer != NULL);
int total = 0;
/*
** If there is accumulated data, then it must processed.
*/
if (Counter > 0) {
if (Control == DECOMPRESS) {
/*
** If the accumulated data is insufficient to make a block header, then
** this means the data has been truncated. Just dump the data through
** as if were already decompressed.
*/
if (BlockHeader.CompCount == 0xFFFF) {
total += Pipe::Put(Buffer, Counter);
Counter = 0;
}
/*
** There appears to be a partial block accumulated in the buffer. It would
** be disastrous to try to decompress the data since there wouldn't be
** the special end of data code that LZO decompression needs. In this
** case, dump the data out as if it were already decompressed.
*/
if (Counter > 0) {
total += Pipe::Put(&BlockHeader, sizeof(BlockHeader));
total += Pipe::Put(Buffer, Counter);
Counter = 0;
BlockHeader.CompCount = 0xFFFF;
}
} else {
/*
** A partial block in the compression process is a normal occurrence. Just
** compress the partial block and output normally.
*/
unsigned int len = sizeof (Buffer2);
char *dictionary = new char [64*1024];
lzo1x_1_compress ((unsigned char*)Buffer, Counter, (unsigned char *)Buffer2, &len, dictionary);
delete [] dictionary;
BlockHeader.CompCount = (unsigned short)len;
BlockHeader.UncompCount = (unsigned short)Counter;
total += Pipe::Put(&BlockHeader, sizeof(BlockHeader));
total += Pipe::Put(Buffer2, len);
Counter = 0;
}
}
total += Pipe::Flush();
return(total);
}