core/ASCOfficeDocFile/DocxDocConverter/STTB.h

/*
 * (c) Copyright Ascensio System SIA 2010-2017
 *
 * This program is a free software product. You can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License (AGPL)
 * version 3 as published by the Free Software Foundation. In accordance with
 * Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
 * that Ascensio System SIA expressly excludes the warranty of non-infringement
 * of any third-party rights.
 *
 * This program is distributed WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For
 * details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
 *
 * You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia,
 * EU, LV-1021.
 *
 * The  interactive user interfaces in modified source and object code versions
 * of the Program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU AGPL version 3.
 *
 * Pursuant to Section 7(b) of the License you must retain the original Product
 * logo when distributing the program. Pursuant to Section 7(e) we decline to
 * grant you any rights under trademark law for use of our trademarks.
 *
 * All the Product's GUI elements, including illustrations and icon sets, as
 * well as technical writing content are licensed under the terms of the
 * Creative Commons Attribution-ShareAlike 4.0 International. See the License
 * terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
 *
 */
#pragma once

#include "Constants.h"

namespace ASCDocFileFormat
{
	class EmptyExtraData
	{
	public:
		unsigned int Size() const
		{
			return 0;
		}

		operator unsigned char*() const
		{
			return (unsigned char*)NULL;
		}
	};

	template<class D, class E = EmptyExtraData> class STTB: public IOperand
	{
	private:
		unsigned char* bytes;
		unsigned int sizeInBytes;

	public:
		STTB() : bytes(NULL), sizeInBytes(0)
		{
		}

		explicit STTB( bool fExtend, const vector<D>* _data, const vector<E>* _extraData = NULL ) : bytes(NULL), sizeInBytes(0)
		{
			//!!!TODO!!!
			if ( ( _data == NULL ) || ( ( _data != NULL ) && ( _extraData != NULL ) && ( _data->size() != _extraData->size() ) ) )
			{
				return;
			}

			if ( fExtend )
			{
				this->sizeInBytes += sizeof(wchar_t); //fExtend
				this->sizeInBytes += ( 2 * _data->size() ); //cchData
			}
			else
			{
				this->sizeInBytes += ( 1 * _data->size() ); //cchData  
			}

			for ( vector<D>::const_iterator iter = _data->begin(); iter != _data->end(); iter++ )
			{
				this->sizeInBytes += iter->Size();
			}

			if ( _data->size() < 0xFFFF )
			{
				this->sizeInBytes += sizeof(unsigned short); //cData
			}
			else
			{
				this->sizeInBytes += sizeof(int); //cData
			}

			this->sizeInBytes += 2; //cbExtra

			if ( _extraData != NULL )
			{
				for ( vector<E>::const_iterator iter = _extraData->begin(); iter != _extraData->end(); iter++ )
				{
					this->sizeInBytes += iter->Size();
				}  
			}

			this->bytes = new unsigned char[this->sizeInBytes];
			memset( this->bytes, 0, this->sizeInBytes );

			unsigned int offset = 0;

			if ( fExtend )
			{
				FormatUtils::SetBytes( ( this->bytes + offset ), (short)(0xFFFF) );
				offset += sizeof(short);
			}

			if ( _data->size() < 0xFFFF )
			{
				FormatUtils::SetBytes( ( this->bytes + offset ), (short)(_data->size()) );
				offset += sizeof(unsigned short);
			}
			else
			{
				FormatUtils::SetBytes( ( this->bytes + offset ), (int)(_data->size()) );
				offset += sizeof(int);
			}

			if ( _extraData != NULL )
			{
				FormatUtils::SetBytes( ( this->bytes + offset ), (unsigned short)(_extraData->front().Size()) );
			}
			else
			{
				FormatUtils::SetBytes( ( this->bytes + offset ), (unsigned short)0 );
			}

			offset += sizeof(unsigned short);  

			if ( fExtend )
			{
				for ( unsigned int i = 0; i < _data->size(); i++ )
				{
					FormatUtils::SetBytes( ( this->bytes + offset ), (unsigned short)( _data->at( i ).Size() / 2 ) );
					offset += sizeof(unsigned short);
					memcpy( ( this->bytes + offset ), (unsigned char*)(_data->at( i )), _data->at( i ).Size() );
					offset += _data->at( i ).Size();

					if ( _extraData != NULL )
					{
						memcpy( ( this->bytes + offset ), (unsigned char*)(_extraData->at( i )), _extraData->at( i ).Size() );
						offset += _extraData->at( i ).Size();  
					}
				}
			}
			else
			{
				for ( unsigned int i = 0; i < _data->size(); i++ )
				{
					FormatUtils::SetBytes( ( this->bytes + offset ), (unsigned char)( _data->at( i ).Size()) );
					offset += sizeof(unsigned char);
					memcpy( ( this->bytes + offset ), (unsigned char*)(_data->at( i )), _data->at( i ).Size() );
					offset += _data->at( i ).Size();

					if ( _extraData != NULL )
					{
						memcpy( ( this->bytes + offset ), (unsigned char*)(_extraData->at( i )), _extraData->at( i ).Size() );
						offset += _extraData->at( i ).Size();  
					}
				}
			}
		}

		STTB( const STTB& _sttb ) : bytes(NULL), sizeInBytes(_sttb.sizeInBytes)
		{
			this->bytes = new unsigned char[this->sizeInBytes];

			if ( this->bytes != NULL )
			{
				memset( this->bytes, 0, this->sizeInBytes );

				memcpy( this->bytes, _sttb.bytes, this->sizeInBytes );
			}
		}

		bool operator == ( const STTB& _sttb )
		{
			return ( ( this->sizeInBytes == _sttb.sizeInBytes ) && 
				( memcmp( this->bytes, _sttb.bytes, this->sizeInBytes ) == 0 ) );
		}

		bool operator != ( const STTB& _sttb )
		{
			return !( this->operator == ( _sttb ) );
		}

		STTB& operator = ( const STTB& _sttb )
		{
			if ( *this != _sttb )
			{
				RELEASEARRAYOBJECTS (bytes);

				this->sizeInBytes = _sttb.sizeInBytes;

				this->bytes = new unsigned char[this->sizeInBytes];

				if ( this->bytes != NULL )
				{
					memcpy( this->bytes, _sttb.bytes, this->sizeInBytes );
				}  
			}

			return *this;
		}

		virtual ~STTB()
		{
			RELEASEARRAYOBJECTS( bytes );
		}

		virtual operator unsigned char*() const
		{
			return this->bytes;
		}

		virtual operator const unsigned char*() const
		{
			return (const unsigned char*)this->bytes;
		}

		virtual unsigned int Size() const
		{
			return this->sizeInBytes;
		}
	};
}