/*
* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of the License "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: 
* @internalComponent * @released
*
*/

#include "utf16string.h"
#ifdef __LINUX__
#include <iconv.h>
#elif defined(WIN32)
#ifdef _STLP_INTERNAL_WINDOWS_H
#define __INTERLOCKED_DECLARED
#endif
#include <windows.h>
#endif
UTF16String::UTF16String() :iData(0), iLength(0){
}
UTF16String::UTF16String(const UTF16String& aRight){
	iLength = aRight.iLength ;
	iData = new TUint16[iLength + 1];
	memcpy(iData,aRight.iData, iLength << 1);
	iData[iLength] = 0;
}
UTF16String::UTF16String(const string& aUtf8Str){
	iData = 0 ;
	iLength = 0 ;
	Assign(aUtf8Str.c_str(),aUtf8Str.length());
}
UTF16String::UTF16String(const TUint16* aUtf16Str,TInt aLength /* = -1*/){
	
	if(aLength < 0){
		const TUint16* p = aUtf16Str ;
		while(*p){
			p++ ;
			aLength ++ ;
		}
	}
	if(aLength > 0){		
		iLength = aLength ;
		aLength <<= 1 ;
		iData = new TUint16[iLength + 1] ;
		memcpy(iData,aUtf16Str,aLength);
		iData[iLength] = 0;
	}else{
		iData = 0 ;
		iLength = 0 ;
	}
	
}
UTF16String::UTF16String(const char* aUtf8Str,TInt aLength /*= -1 */){
	iData = 0 ;
	iLength = 0 ;	
	Assign(aUtf8Str,aLength);
}	
UTF16String::~UTF16String(){
	if(iData)
		delete []iData ;
}	

UTF16String& UTF16String::operator = (const UTF16String& aRight){
	if(&aRight != this){
		if(iData) 
			delete []iData ; 
		iLength = aRight.iLength ;
		iData = new TUint16[iLength + 1];
		memcpy(iData,aRight.iData, iLength << 1);
		iData[iLength] = 0;
	}
	return *this;
}
bool UTF16String::FromFile(const char* aFileName){
	if(!aFileName || !(*aFileName))
		return false ;
	ifstream ifs(aFileName,ios_base::in + ios_base::binary); 	
	if(!ifs.is_open())
		return false ;
	
	ifs.seekg(0,ios_base::end);
	size_t length = ifs.tellg();
	if((length % 2) == 1 ){
		ifs.close() ;
		return false ;
	}
	ifs.seekg(0,ios_base::beg);
	TUint16 hdr ;
	size_t readLen = length - sizeof(hdr) ;
	length >>= 1 ;	
	TUint16 *newData = new TUint16[length + 1];
	ifs.read(reinterpret_cast<char*>(&hdr),sizeof(hdr));		
	if(hdr == 0xFEFF){ 
		ifs.read(reinterpret_cast<char*>(newData),readLen);
		length -- ;
	}
	else{		 
		*newData = hdr ;
		ifs.read(reinterpret_cast<char*>(&newData[1]),readLen);
	} 
	ifs.close();
	iLength = length ;
	if(iData)
		delete []iData ;
	iData = newData ;
	iData[iLength] = 0;	
	return true ;
}
/**
* aResult will not changed on error
*/
bool UTF16String::ToUTF8(string& aResult) const {
	if(IsEmpty()){
		aResult = "";
		return true;
	}
	size_t bufLen = (iLength + 1) << 2 ;
	char* buffer = new char[bufLen] ;
#ifdef WIN32
	int r = WideCharToMultiByte(CP_UTF8,0,reinterpret_cast<WCHAR*>(iData),iLength,buffer,bufLen,NULL,NULL);
	if(r < 0){
		delete []buffer ;
		return false ;
	}
	buffer[r] = 0;
	aResult.assign(buffer,r);
#else
	iconv_t it = iconv_open("UTF-8","UTF-16");
	if((iconv_t)(-1) == it){ 
		return  false;
	}
	char* bufferEnd = buffer ;
	char* in_ptr = reinterpret_cast<char*>(iData);
	size_t in_len = iLength << 1 ;
	iconv(it,&in_ptr,&in_len ,&bufferEnd,&bufLen);
	iconv_close(it);
	*bufferEnd = 0 ;
	size_t length = bufferEnd - buffer ;
	aResult.assign(buffer,length);
#endif
	delete []buffer ;
	return true ;
}
bool UTF16String::Assign(const char* aUtf8Str,TInt aLength /* = -1*/) {
	if(0 == aUtf8Str) return false ;	
	if(aLength < 0) aLength = strlen(aUtf8Str); 
		
#ifdef WIN32
	size_t newLength = aLength + 1;
	TUint16* newData = new TUint16[newLength];
	int r = MultiByteToWideChar(CP_UTF8,0,aUtf8Str,aLength,reinterpret_cast<WCHAR*>(newData),newLength);
	if(r < 0){
		delete []newData ;
		return false ;
	}
	iLength = r ;	
#else
	char* forFree = 0 ;
	if(aUtf8Str[aLength - 1] != 0){
		forFree = new char[aLength + 1];
		memcpy(forFree,aUtf8Str,aLength );
		forFree[aLength] = 0;
		aUtf8Str = forFree ;
	}
	iconv_t it = iconv_open("UTF-16","UTF-8");
	if((iconv_t)(-1) == it){ 
		 
		return  false;
	}
	size_t newLength = aLength + 2;
	TUint16* newData = new TUint16[newLength];
	newLength <<= 1;
	char* out_ptr = reinterpret_cast<char*>(newData);
	size_t in_len = aLength ;
	char* in_ptr = const_cast<char*>(aUtf8Str);
	iconv(it,&in_ptr,&in_len ,&out_ptr ,&newLength);
	newLength = out_ptr - reinterpret_cast<char*>(newData);
	iconv_close(it);
	if(newLength % 2 == 1){ //should not be possible
		delete []newData;
		return false ;
	}
	iLength = (newLength >> 1) ;
	if(forFree)
		delete []forFree;
#endif
	newData[iLength] = 0 ;
	if(iData){
		delete []iData ;
	}
	iData = newData ;	
	if(*iData == 0xFEFF)
		iLength -- ;
	
	return true ;
}
static const TUint16 NullUInt16Str[1] = {0};
const TUint16* UTF16String::c_str() const {
	if(0 == iData)
		return NullUInt16Str;
	else if(0xFEFF != *iData)
		return iData ;
	else
		return iData + 1;
}
///
/// aUtf16Str must end with '\0'
/// 
int UTF16String::Compare(const TUint16* aUtf16Str) const {
	if(!aUtf16Str || !(*aUtf16Str))
		return (iLength > 0) ? 1 : 0; 
	size_t i ;
	for(i = 0 ; aUtf16Str[i] != 0 ; i++) { 
		if( iData[i] > aUtf16Str[i])
			return 1;
		else if(iData[i] < aUtf16Str[i])
			return -1 ;
	}
	return (i < iLength) ? 1 : 0 ; 
}
///
/// aUtf16Str must end with '\0'
/// 
int UTF16String::CompareNoCase(const TUint16* aUtf16Str) const {
	if(!aUtf16Str || !(*aUtf16Str))
		return (iLength > 0) ? 1 : 0; 
	size_t i ;
	TUint16 a, b;
	for(i = 0 ; aUtf16Str[i] != 0 ; i++) { 
		a = iData[i];
		b = aUtf16Str[i] ;
		if( a >= 'A' && a <= 'Z') a |= 0x20 ;
		if( b >= 'A' && b <= 'Z') b |= 0x20 ;
			
		if( a > b )
			return 1;
		else if( a < b )
			return -1 ;
	}
	return (i < iLength) ? 1 : 0 ; 
}
TUint16* UTF16String::Alloc(size_t aNewLen) {
	TUint16* newData = new TUint16[aNewLen + 1] ;
	if(!newData) return 0;
	if(iData) delete []iData ;
	
	iLength = aNewLen ;
	iData = newData ;
	*iData = 0 ;
	iData[aNewLen] = 0;
	return iData;
}