/**
 * @file XTPFlowGraphImage.cpp
 *
 * @copyright
 * (c) 1998-2025 Codejock Software, All Rights Reserved.
 *
 * This source file is the property of Codejock Software and must not be
 * redistributed by any means without the explicit written permission of
 * Codejock Software.
 *
 * The use of this source code is governed by the terms and conditions specified
 * in the Toolkit Pro license agreement. Codejock Software grants you, as a
 * single software developer, the limited right to use this software on one
 * computer only.
 *
 * Contact Information:
 * support@codejock.com
 * http://www.codejock.com
 *
 */

#include "stdafx.h"

#include "Common/Base/Diagnostic/XTPDisableAdvancedWarnings.h"
#include <comdef.h>
#include "Common/Base/Diagnostic/XTPEnableAdvancedWarnings.h"

#include "GraphicLibrary/GdiPlus/XTPGdiPlus.h"
#include "GraphicLibrary/XTGraphicUtils.h"

#include "Common/XTPTypeId.h"
#include "Common/XTPCasting.h"
#include "Common/XTPFramework.h"
#include "Common/XTPSystemHelpers.h"
#include "Common/XTPSynchro.h"
#include "Common/XTPApplication.h"
#include "Common/XTPSingleton.h"
#include "Common/ScrollBar/XTPScrollBase.h"

#include "Common/XTPResourceManager.h"
#include "Common/XTPVC80Helpers.h"
#include "Common/XTPImageManager.h"

#include "FlowGraph/XTPFlowGraphImage.h"
#include "FlowGraph/XTPFlowGraphControl.h"
#include "FlowGraph/XTPFlowGraphIIDs.h"

#include <atlbase.h>
#include "Markup/XTPMarkupObject.h"
#include "Markup/XTPMarkupContext.h"

#include "Common/Base/Diagnostic/XTPDisableNoisyWarnings.h"

#ifdef _DEBUG
#	define new DEBUG_NEW
#	undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

//////////////////////////////////////////////////////////////////////////
// CXTPFlowGraphImage

CXTPFlowGraphImage::CXTPFlowGraphImage()
	: m_pImage(NULL)
	, m_pIcon(NULL)
	, m_pParent(NULL)
	, m_nCommandId((UINT)-1)
{
	m_szImage = CSize(0, 0);

#ifdef _XTP_ACTIVEX
	EnableAutomation();
	EnableTypeLib();
#endif
}

CXTPFlowGraphImage::~CXTPFlowGraphImage()
{
	SAFE_DELETE(m_pImage);
	CMDTARGET_RELEASE(m_pIcon);
}

void CXTPFlowGraphImage::OnRemoved()
{
	SAFE_DELETE(m_pImage);
	CMDTARGET_RELEASE(m_pIcon);
}

void CXTPFlowGraphImage::LoadFromResource(HINSTANCE hInstance, UINT nResourceName)
{
	SAFE_DELETE(m_pImage);
#ifdef _DEBUG
#	pragma push_macro("new")
#	undef new
#endif
	m_pImage = new Gdiplus::Bitmap(hInstance, MAKEINTRESOURCEW(nResourceName));
#ifdef _DEBUG
#	pragma pop_macro("new")
#endif
	ImageChanged();
}

void CXTPFlowGraphImage::LoadFromResource(HINSTANCE hInstance, LPCTSTR lpszResourceName,
										  LPCTSTR lpszResourceType)
{
	SAFE_DELETE(m_pImage);

	IStreamPtr pStream;
	if (SUCCEEDED(XTPCreateReadOnlyResourceStream(hInstance, lpszResourceName, lpszResourceType,
												  FALSE, &pStream)))
	{
#ifdef _DEBUG
#	pragma push_macro("new")
#	undef new
#endif
		m_pImage = new Gdiplus::Bitmap(pStream.GetInterfacePtr(), FALSE);
#ifdef _DEBUG
#	pragma pop_macro("new")
#endif
	}
	ImageChanged();
}

void CXTPFlowGraphImage::LoadFromFile(LPCTSTR lpszFileName)
{
	SAFE_DELETE(m_pImage);
#ifdef _DEBUG
#	pragma push_macro("new")
#	undef new
#endif
	m_pImage = new Gdiplus::Bitmap(XTP_CT2CW(lpszFileName));
#ifdef _DEBUG
#	pragma pop_macro("new")
#endif

	ImageChanged();
}

void CXTPFlowGraphImage::SetBitmap(HBITMAP hBitmap)
{
	SAFE_DELETE(m_pImage);
#ifdef _DEBUG
#	pragma push_macro("new")
#	undef new
#endif
	m_pImage = new Gdiplus::Bitmap(hBitmap, 0);
#ifdef _DEBUG
#	pragma pop_macro("new")
#endif

	ImageChanged();
}

void CXTPFlowGraphImage::SetIcon(HICON hIcon)
{
	SAFE_DELETE(m_pImage);

	if (CXTPGraphicUtils::GdipCreateBitmapFromHICON_Fixed(hIcon, &m_pImage)
		!= Gdiplus::Ok) // Now 'image' has the icon, with transparency
	{
		_ASSERTE(FALSE);
	}

	ImageChanged();
}

void CXTPFlowGraphImage::ImageChanged()
{
	CMDTARGET_RELEASE(m_pIcon);
	m_szImage = CSize(0, 0);

	if (NULL == m_pImage || Gdiplus::Ok != m_pImage->GetLastStatus())
	{
		SAFE_DELETE(m_pImage);
		return;
	}

	m_szImage = CSize(XTPToIntChecked(m_pImage->GetWidth()),
					  XTPToIntChecked(m_pImage->GetHeight()));

	Gdiplus::PixelFormat eSrcPixelFormat = m_pImage->GetPixelFormat();
	BOOL bAlpha							 = FALSE;
	if (Gdiplus::IsAlphaPixelFormat(eSrcPixelFormat))
	{
		bAlpha = TRUE;
	}

	if (m_pParent)
	{
		HBITMAP hBitmap;
		m_pImage->GetHBITMAP(RGB(255, 255, 255), &hBitmap);

		CBitmap bitmap;
		bitmap.Attach(hBitmap);

		m_pParent->GetImageManager()->SetIcons(bitmap, &m_nCommandId, 1, m_szImage, xtpImageNormal,
											   bAlpha);

		m_pIcon = m_pParent->GetImageManager()->GetImage(m_nCommandId, m_szImage.cx);

		m_pIcon->InternalAddRef();
	}
	else
	{
		HBITMAP hBitmap;
		m_pImage->GetHBITMAP(RGB(255, 255, 255), &hBitmap);

		CXTPImageManagerIconHandle ih(hBitmap);

		CXTPImageManagerIcon* pIcon = new CXTPImageManagerIcon(1, m_szImage.cx, m_szImage.cy);
		pIcon->SetIcon(ih);

		m_pIcon = pIcon;

		// we don't have to do InternalAddRef();
		// because image manager doesn't present
	}
}

Gdiplus::Image* CXTPFlowGraphImage::GetImage() const
{
	return m_pImage;
}

CXTPImageManagerIcon* CXTPFlowGraphImage::GetIcon() const
{
	return m_pIcon;
}

//////////////////////////////////////////////////////////////////////////
// CXTPFlowGraphImages

CXTPFlowGraphImages::CXTPFlowGraphImages(CXTPFlowGraphControl* pControl)
{
	m_pControl = pControl;

	_ASSERTE(pControl->GetMarkupContext());

	if (pControl->GetMarkupContext())
	{
		// each control must have its own image manager (not library-shared)
		pControl->GetMarkupContext()->SetImageManager(new CXTPImageManager());

		m_pImageManager = pControl->GetMarkupContext()->GetImageManager();
	}

#ifdef _XTP_ACTIVEX
	EnableAutomation();
	EnableTypeLib();
#endif
}

CXTPFlowGraphImages::~CXTPFlowGraphImages()
{
	RemoveAll();
}

void CXTPFlowGraphImages::RemoveAll()
{
	if (m_arrImages.GetSize() == 0)
		return;

	for (int i = 0; i < m_arrImages.GetSize(); i++)
	{
		m_arrImages[i]->OnRemoved();
		m_arrImages[i]->InternalRelease();
	}
	m_arrImages.RemoveAll();

	if (m_pImageManager)
		m_pImageManager->RemoveAll();
}

CXTPFlowGraphImage* CXTPFlowGraphImages::AddImage(CXTPFlowGraphImage* pImage)
{
	m_arrImages.Add(pImage);
	pImage->m_pParent	= this;
	pImage->m_nCommandId = (UINT)(m_arrImages.GetSize() - 1);

	return pImage;
}

void CXTPFlowGraphImages::Remove(CXTPFlowGraphImage* pImage)
{
	for (int i = 0; i < m_arrImages.GetSize(); i++)
	{
		if (m_arrImages[i] == pImage)
		{
			if (m_pImageManager)
				m_pImageManager->RemoveIcon(pImage->m_nCommandId);

			RemoveAt(i);
			return;
		}
	}
}

void CXTPFlowGraphImages::RemoveAt(int nIndex)
{
	if (nIndex < 0 || nIndex >= m_arrImages.GetSize())
		return;

	CXTPFlowGraphImage* pImage = m_arrImages[nIndex];

	m_arrImages.RemoveAt(nIndex);

	pImage->OnRemoved();

	pImage->InternalRelease();

	if (m_pImageManager)
		m_pImageManager->RemoveIcon(pImage->m_nCommandId);
}

#ifdef _XTP_ACTIVEX

//////////////////////////////////////////////////////////////////////////

BEGIN_DISPATCH_MAP(CXTPFlowGraphImage, CXTPCmdTarget)
	DISP_FUNCTION_ID(CXTPFlowGraphImages, "LoadFromFile", 1, LoadFromFile, VT_EMPTY, VTS_BSTR)
END_DISPATCH_MAP()

BEGIN_INTERFACE_MAP(CXTPFlowGraphImage, CXTPCmdTarget)
	INTERFACE_PART(CXTPFlowGraphImage, XTPDIID_FlowGraphImage, Dispatch)
END_INTERFACE_MAP()

IMPLEMENT_OLETYPELIB_EX(CXTPFlowGraphImage, XTPDIID_FlowGraphImage)

//////////////////////////////////////////////////////////////////////////

BEGIN_DISPATCH_MAP(CXTPFlowGraphImages, CXTPCmdTarget)
	DISP_FUNCTION_ID(CXTPFlowGraphImages, "Count", 1, OleGetItemCount, VT_I4, VTS_NONE)
	DISP_FUNCTION_ID(CXTPFlowGraphImages, "Image", DISPID_VALUE, OleGetItem, VT_DISPATCH, VTS_I4)
	DISP_FUNCTION_ID(CXTPFlowGraphImages, "_NewEnum", DISPID_NEWENUM, OleNewEnum, VT_UNKNOWN,
					 VTS_NONE)

	DISP_FUNCTION_ID(CXTPFlowGraphImages, "AddImage", 2, OleAddImage, VT_DISPATCH, VTS_NONE)

	DISP_FUNCTION_ID(CXTPFlowGraphImages, "RemoveAll", 3, RemoveAll, VT_EMPTY, VTS_NONE)
	DISP_FUNCTION_ID(CXTPFlowGraphImages, "RemoveAt", 4, RemoveAt, VT_EMPTY, VTS_I4)
END_DISPATCH_MAP()

BEGIN_INTERFACE_MAP(CXTPFlowGraphImages, CXTPCmdTarget)
	INTERFACE_PART(CXTPFlowGraphImages, XTPDIID_FlowGraphImages, Dispatch)
END_INTERFACE_MAP()

IMPLEMENT_OLETYPELIB_EX(CXTPFlowGraphImages, XTPDIID_FlowGraphImages)
IMPLEMENT_ENUM_VARIANT(CXTPFlowGraphImages)

LPDISPATCH CXTPFlowGraphImages::OleGetItem(int nIndex)
{
	return XTPGetDispatch(GetAt(nIndex));
}

int CXTPFlowGraphImages::OleGetItemCount()
{
	return GetCount();
}

LPDISPATCH CXTPFlowGraphImages::OleAddImage()
{
	return XTPGetDispatch(AddImage(new CXTPFlowGraphImage()));
}

#endif