#include "CtrlGeneralShape.h"

#include "CtrlShapePic.h"
#include "CtrlShapeRect.h"
#include "CtrlShapeLine.h"
#include "CtrlContainer.h"
#include "CtrlShapeEllipse.h"
#include "CtrlShapePolygon.h"
#include "CtrlShapeArc.h"
#include "CtrlShapeCurve.h"
#include "CtrlEqEdit.h"
#include "CtrlShapeOle.h"
#include "CtrlShapeVideo.h"
#include "CtrlShapeTextArt.h"
#include "CtrlCharacter.h"

namespace HWP
{
CCtrlGeneralShape::CCtrlGeneralShape()
{
	InitData();
}

CCtrlGeneralShape::CCtrlGeneralShape(const HWP_STRING& sCtrlID)
	: CCtrlObjElement(sCtrlID)
{
	InitData();
}

CCtrlGeneralShape::CCtrlGeneralShape(const CCtrlGeneralShape& oGeneralShape)
	: CCtrlObjElement(oGeneralShape)
{
	InitData();

	m_pParent = oGeneralShape.m_pParent;

	m_nLineColor = oGeneralShape.m_nLineColor;
	m_nLineThick = oGeneralShape.m_nLineThick;
	m_eLineHead = oGeneralShape.m_eLineHead;
	m_eLineTail = oGeneralShape.m_eLineTail;
	m_eLineHeadSz = oGeneralShape.m_eLineHeadSz;
	m_eLineTailSz = oGeneralShape.m_eLineTailSz;
	m_eLineStyle = oGeneralShape.m_eLineStyle;
	m_chOutline = oGeneralShape.m_chOutline;

	m_nFillType = oGeneralShape.m_nFillType;
	m_pFill = oGeneralShape.m_pFill;

	if (nullptr != m_pFill)
		m_pFill->AddRef();

	m_shLeftSpace = oGeneralShape.m_shLeftSpace;
	m_shRightSpace = oGeneralShape.m_shRightSpace;
	m_shTopSpace = oGeneralShape.m_shTopSpace;
	m_shBottomSpace = oGeneralShape.m_shBottomSpace;
	m_nMaxTxtWidth = oGeneralShape.m_nMaxTxtWidth;
}

CCtrlGeneralShape::CCtrlGeneralShape(const HWP_STRING& sCtrlID, int nSize, CHWPStream& oBuffer, int nOff, int nVersion)
	: CCtrlObjElement(sCtrlID, nSize, oBuffer, nOff, nVersion)
{
	InitData();
}

CCtrlGeneralShape::CCtrlGeneralShape(const HWP_STRING& sCtrlID, CXMLReader& oReader, int nVersion)
    : CCtrlObjElement(sCtrlID, oReader, nVersion)
{
	InitData();
}

CCtrlGeneralShape::~CCtrlGeneralShape()
{
	if (nullptr != m_pFill && 0 == m_pFill->Release())
		m_pFill = nullptr;
}

ECtrlObjectType CCtrlGeneralShape::GetCtrlType() const
{
	return ECtrlObjectType::Shape;
}

EShapeType CCtrlGeneralShape::GetShapeType() const
{
	return EShapeType::GeneralShape;
}

void CCtrlGeneralShape::ParseChildren(CXMLReader& oReader, int nVersion)
{
	bool bHeadFill = false, bTailFill = false;

	const std::string sNodeName{oReader.GetName()};

	if ("hp:lineShape" == sNodeName)
	{
		std::string sHeadStyle, sTailStyle;

		START_READ_ATTRIBUTES(oReader)
		{
			if ("color" == sAttributeName)
				m_nLineColor = oReader.GetInt();
			else if ("width" == sAttributeName)
				m_nLineThick = oReader.GetInt();
			else if ("style" == sAttributeName)
				m_eLineStyle = GetLineStyle2(oReader.GetText());
			else if ("headStyle" == sAttributeName)
				sHeadStyle = oReader.GetTextA();
			else if ("headfill" == sAttributeName)
				bHeadFill = oReader.GetBool();
			else if ("headSz" == sAttributeName)
			{
				const std::string sHeadSz{oReader.GetTextA()};

				if ("SMALL_SMALL" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::SMALL_SMALL;
				else if ("SMALL_MEDIUM" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::SMALL_MEDIUM;
				else if ("SMALL_LARGE" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::SMALL_LARGE;
				else if ("MEDIUM_SMALL" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::MEDIUM_SMALL;
				else if ("MEDIUM_MEDIUM" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::MEDIUM_MEDIUM;
				else if ("MEDIUM_LARGE" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::MEDIUM_LARGE;
				else if ("LARGE_SMALL" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::LARGE_SMALL;
				else if ("LARGE_MEDIUM" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::LARGE_MEDIUM;
				else if ("LARGE_LARGE" == sHeadSz)
					m_eLineHeadSz = ELineArrowSize::LARGE_LARGE;
				else
					m_eLineHeadSz = ELineArrowSize::MEDIUM_MEDIUM;
			}
			else if ("tailStyle" == sAttributeName)
				sTailStyle = oReader.GetTextA();
			else if ("tailfill" == sAttributeName)
				bTailFill = oReader.GetBool();
			else if ("tailSz" == sAttributeName)
			{
				const std::string sType{oReader.GetTextA()};

				if ("SMALL_SMALL" == sType)
					m_eLineTailSz = ELineArrowSize::SMALL_SMALL;
				else if ("SMALL_MEDIUM" == sType)
					m_eLineTailSz = ELineArrowSize::SMALL_MEDIUM;
				else if ("SMALL_LARGE" == sType)
					m_eLineTailSz = ELineArrowSize::SMALL_LARGE;
				else if ("MEDIUM_SMALL" == sType)
					m_eLineTailSz = ELineArrowSize::MEDIUM_SMALL;
				else if ("MEDIUM_MEDIUM" == sType)
					m_eLineTailSz = ELineArrowSize::MEDIUM_MEDIUM;
				else if ("MEDIUM_LARGE" == sType)
					m_eLineTailSz = ELineArrowSize::MEDIUM_LARGE;
				else if ("LARGE_SMALL" == sType)
					m_eLineTailSz = ELineArrowSize::LARGE_SMALL;
				else if ("LARGE_MEDIUM" == sType)
					m_eLineTailSz = ELineArrowSize::LARGE_MEDIUM;
				else if ("LARGE_LARGE" == sType)
					m_eLineTailSz = ELineArrowSize::LARGE_LARGE;
				else
					m_eLineTailSz = ELineArrowSize::MEDIUM_MEDIUM;
			}
		}
		END_READ_ATTRIBUTES(oReader)

		if (sHeadStyle.empty())
			m_eLineHead = ELineArrowStyle::NORMAL;
		else if ("ARROW" == sHeadStyle)
			m_eLineHead = ELineArrowStyle::ARROW;
		else if ("SPEAR" == sHeadStyle)
			m_eLineHead = ELineArrowStyle::SPEAR;
		else if ("CONCAVE_ARROW" == sHeadStyle)
			m_eLineHead = ELineArrowStyle::CONCAVE_ARROW;
		else if ("EMPTY_DIAMOND" == sHeadStyle)
			m_eLineHead = bHeadFill ? ELineArrowStyle::DIAMOND : ELineArrowStyle::EMPTY_DIAMOND;
		else if ("EMPTY_CIRCLE" == sHeadStyle)
			m_eLineHead = bHeadFill ? ELineArrowStyle::CIRCLE : ELineArrowStyle::EMPTY_CIRCLE;
		else if ("EMPTY_BOX" == sHeadStyle)
			m_eLineHead = bHeadFill ? ELineArrowStyle::BOX : ELineArrowStyle::EMPTY_BOX;
		else
			m_eLineHead = ELineArrowStyle::NORMAL;

		if (sTailStyle.empty())
			m_eLineTail = ELineArrowStyle::NORMAL;
		else if ("ARROW" == sTailStyle)
			m_eLineTail = ELineArrowStyle::ARROW;
		else if ("SPEAR" == sTailStyle)
			m_eLineTail = ELineArrowStyle::SPEAR;
		else if ("CONCAVE_ARROW" == sTailStyle)
			m_eLineTail = ELineArrowStyle::CONCAVE_ARROW;
		else if ("EMPTY_DIAMOND" == sTailStyle)
			m_eLineTail = bTailFill ? ELineArrowStyle::DIAMOND : ELineArrowStyle::EMPTY_DIAMOND;
		else if ("EMPTY_CIRCLE" == sTailStyle)
			m_eLineTail = bTailFill ? ELineArrowStyle::CIRCLE : ELineArrowStyle::EMPTY_CIRCLE;
		else if ("EMPTY_BOX" == sTailStyle)
			m_eLineTail = bTailFill ? ELineArrowStyle::BOX : ELineArrowStyle::EMPTY_BOX;
		else
			m_eLineTail = ELineArrowStyle::NORMAL;
	}
	else if ("hc:fillBrush" == sNodeName)
		m_pFill = new CFill(oReader);
	else if ("hp:drawText" == sNodeName)
	{
		m_nMaxTxtWidth = oReader.GetAttributeInt("lastWidth");

		WHILE_READ_NEXT_NODE_WITH_DEPTH_AND_NAME(oReader, Child)
		{
			if ("hp:textMargin" == sNodeChildName)
			{
				START_READ_ATTRIBUTES(oReader)
				{
					if ("left" == sAttributeName)
						m_shLeftSpace = oReader.GetInt();
					else if ("right" == sAttributeName)
						m_shRightSpace =  oReader.GetInt();
					else if ("top" == sAttributeName)
						m_shTopSpace =  oReader.GetInt();
					else if ("bottom" == sAttributeName)
						m_shBottomSpace =  oReader.GetInt();
				}
				END_READ_ATTRIBUTES(oReader)
			}
			else if ("hp:subList" == sNodeChildName)
				ReadSubList(oReader, nVersion);
		}
		END_WHILE
	}
	else
		CCtrlObjElement::ParseChildren(oReader, nVersion);
}

void CCtrlGeneralShape::InitData()
{
	m_eLineStyle = ELineStyle2::NONE;
	m_pFill = nullptr;
}

void CCtrlGeneralShape::ReadSubList(CXMLReader& oReader, int nVersion)
{
	m_eTextVerAlign = GetVertAlign(oReader.GetAttributeInt("vertAlign"));

	// CHWPPargraph* pLatestParagraph = nullptr;

	WHILE_READ_NEXT_NODE_WITH_ONE_NAME(oReader, "hp:p")
	{
		CHWPPargraph* pParagraph = new CHWPPargraph(oReader, nVersion);

		if (nullptr == pParagraph)
			continue;

		m_arParas.push_back(pParagraph);
		// pLatestParagraph = pParagraph;

		// if (nullptr != pLatestParagraph && 0 != pLatestParagraph->GetCountCtrls() && ECtrlObjectType::ParaText == pLatestParagraph->GetCtrls().back()->GetCtrlType())
		// 	pLatestParagraph->AddCtrl(new CCtrlCharacter(L"   _", ECtrlCharType::PARAGRAPH_BREAK));
	}
	END_WHILE
}

void CCtrlGeneralShape::SetParent(CHWPPargraph* pParent)
{
	m_pParent = pParent;
}

CHWPPargraph* CCtrlGeneralShape::GetParent()
{
	return m_pParent;
}

const CFill* CCtrlGeneralShape::GetFill() const
{
	return m_pFill;
}

ELineStyle2 CCtrlGeneralShape::GetLineStyle() const
{
	return m_eLineStyle;
}

int CCtrlGeneralShape::GetLineColor() const
{
	return m_nLineColor;
}

int CCtrlGeneralShape::GetLineThick() const
{
	return m_nLineThick;
}

CCtrlGeneralShape* CCtrlGeneralShape::Parse(CCtrlGeneralShape& oObj, int nSize, CHWPStream& oBuffer, int nOff, int nVersion)
{
	oBuffer.SavePosition();

	HWP_STRING sCtrlId;
	oBuffer.ReadString(sCtrlId, 4, EStringCharacter::ASCII);

	CCtrlGeneralShape *pShape = nullptr;

	#define CREATE_AND_PARSE_SHAPE(type)\
	pShape = new type(oObj); \
	\
	if (nullptr != pShape) \
		type::ParseCtrl((type&)(*pShape), nSize - oBuffer.GetDistanceToLastPos(), oBuffer, 0, nVersion); \
	else \
		oBuffer.Skip(nSize - oBuffer.GetDistanceToLastPos())

	if (L"cip$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapePic);
	}
	else if (L"cer$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeRect);
	}
	else if (L"nil$" == sCtrlId ||
	         L"loc$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeLine);
	}
	else if (L"noc$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlContainer);
	}
	else if (L"lle$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeEllipse);
	}
	else if (L"lop$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapePolygon);
	}
	else if (L"cra$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeArc);
	}
	else if (L"ruc$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeCurve);
	}
	else if (L"deqe" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlEqEdit);
	}
	else if (L"elo$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeOle);
	}
	else if (L"div$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeVideo);
	}
	else if (L"tat$" == sCtrlId)
	{
		CREATE_AND_PARSE_SHAPE(CCtrlShapeTextArt);
	}

	if (nullptr != pShape)
		pShape->SetID(sCtrlId);

	oBuffer.RemoveLastSavedPos();

	return pShape;
}

int CCtrlGeneralShape::ParseListHeaderApend(CCtrlGeneralShape& oObj, int nSize, CHWPStream& oBuffer, int nOff, int nVersion)
{
	oBuffer.SavePosition();

	if (nSize >= 16)
	{
		oBuffer.Skip(2);
		oBuffer.ReadInt(oObj.m_nCaptionAttr);
		oBuffer.ReadInt(oObj.m_nCaptionWidth);
		oObj.m_nCaptionSpacing = oBuffer.ReadShort();
		oBuffer.ReadInt(oObj.m_nCaptionMaxW);
	}

	if ((nSize - oBuffer.GetDistanceToLastPos()) == 8)
		oBuffer.Skip(8);

	oBuffer.Skip(nSize - oBuffer.GetDistanceToLastPos(true));
	return nSize;
}

int CCtrlGeneralShape::ParseCtrl(CCtrlGeneralShape& oObj, int nSize, CHWPStream& oBuffer, int nOff, int nVersion)
{
	oBuffer.SavePosition();

	CCtrlObjElement::ParseCtrl(oObj, nSize, oBuffer, nOff, nVersion);

	oBuffer.ReadColor(oObj.m_nLineColor);
	oObj.m_nLineThick = oBuffer.ReadShort();

	oBuffer.Skip(2);

	int nLineAttr;
	oBuffer.ReadInt(nLineAttr);

	oObj.m_eLineStyle = GetLineStyle2(nLineAttr & 0x3F);
	oObj.m_eLineHead = GetLineArrowStyle(((nLineAttr >> 10) & 0x3F), CHECK_FLAG(nLineAttr >> 30, 0x1));
	oObj.m_eLineHeadSz = GetLineArrowSize((nLineAttr >> 22) & 0x0F);
	oObj.m_eLineTail = GetLineArrowStyle(((nLineAttr >> 16) & 0x3F), CHECK_FLAG(nLineAttr >> 31, 0x1));
	oObj.m_eLineTailSz = GetLineArrowSize((nLineAttr >> 26) & 0x0F);
	oBuffer.ReadByte(oObj.m_chOutline);

	oObj.m_pFill = new CFill(oBuffer, 0, nSize - oBuffer.GetDistanceToLastPos() - 22);

	oBuffer.Skip(22);

	return oBuffer.GetDistanceToLastPos(true);
}
}