/* LZMA Compress Implementation

Copyright (c) 2012, Nikolaj Schlej. All rights reserved.
This program and the accompanying materials
are licensed and made available under the terms and conditions of the BSD License
which accompanies this distribution.  The full text of the license may be found at
http://opensource.org/licenses/bsd-license.php

THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
WITHWARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.

*/

#include "LzmaCompress.h"
#include "SDK/C/7zVersion.h"
#include "SDK/C/LzmaEnc.h"

#include <stdlib.h>

#define LZMA_HEADER_SIZE (LZMA_PROPS_SIZE + 8)

static void * AllocForLzma(void *p, size_t size) { return malloc(size); }
static void FreeForLzma(void *p, void *address) { free(address); }
static ISzAlloc SzAllocForLzma = { &AllocForLzma, &FreeForLzma };

SRes OnProgress(void *p, UInt64 inSize, UInt64 outSize)
{
	return SZ_OK;
}

static ICompressProgress g_ProgressCallback = { &OnProgress };

STATIC
	UINT64
	EFIAPI
	RShiftU64 (
	UINT64                    Operand,
	UINT32                     Count
	)
{
	return Operand >> Count;
}

VOID
	SetEncodedSizeOfBuf(
	UINT64 EncodedSize,
	UINT8 *EncodedData
	)
{
	INT32   Index;

	EncodedData[LZMA_PROPS_SIZE]  = EncodedSize & 0xFF;
	for (Index = LZMA_PROPS_SIZE+1; Index <= LZMA_PROPS_SIZE + 7; Index++)
	{
		EncodedSize = RShiftU64(EncodedSize, 8);
		EncodedData[Index] = EncodedSize & 0xFF;
	}
}

INT32
	EFIAPI
	LzmaCompress (
	CONST UINT8  *Source,
	UINT32       SourceSize,
	UINT8    *Destination,
	UINT32   *DestinationSize
	)
{
	SRes              LzmaResult;
	CLzmaEncProps     props;
	SizeT propsSize = LZMA_PROPS_SIZE;
	SizeT destLen = SourceSize + SourceSize / 3 + 128;

	if (*DestinationSize < destLen)
	{
		*DestinationSize = destLen;
		return ERR_BUFFER_TOO_SMALL;
	}

	LzmaEncProps_Init(&props);
	props.dictSize = LZMA_DICTIONARY_SIZE;
	props.level = 9;
	props.fb = 273;

	LzmaResult = LzmaEncode(
		(Byte*)((UINT8*)Destination + LZMA_HEADER_SIZE), 
		&destLen,
		Source, 
		SourceSize,
		&props, 
		(UINT8*)Destination, 
		&propsSize, 
		props.writeEndMark,
		&g_ProgressCallback, 
		&SzAllocForLzma, 
		&SzAllocForLzma);

	*DestinationSize = destLen + LZMA_HEADER_SIZE;

	SetEncodedSizeOfBuf((UINT64)SourceSize, Destination);

	if (LzmaResult == SZ_OK) {
		return ERR_SUCCESS;
	} else {
		return ERR_INVALID_PARAMETER;
	}
}