//========= Copyright ?1996-2005, Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
// $NoKeywords: $
//=============================================================================//

#include <stdio.h>

#include <vgui/IBorder.h>
#include <vgui/ISurface.h>
#include <vgui/IScheme.h>
#include <vgui/IBorder.h>
#include <tier1/KeyValues.h>

#include <vgui_controls/ImagePanel.h>
#include <vgui_controls/Image.h>
#include <vgui_controls/Controls.h>

// memdbgon must be the last include file in a .cpp file!!!
#include <tier0/memdbgon.h>

using namespace vgui2;

DECLARE_BUILD_FACTORY(ImagePanel);

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
ImagePanel::ImagePanel(Panel *parent, const char *name) : Panel(parent, name)
{
	m_pImage = NULL;
	m_pszImageName = NULL;
	m_pszColorName = NULL;
	m_bScaleImage = false;
	m_bCenterImage = false;
	m_bTileImage = false;
	m_bTileHorizontally = false;
	m_bTileVertically = false;
	m_fScaleAmount = 0.0f;
	m_FillColor = Color(0, 0, 0, 0);
	m_DrawColor = Color(255, 255, 255, 255);

	SetImage(m_pImage);
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
ImagePanel::~ImagePanel()
{
	delete[] m_pszImageName;
	delete[] m_pszColorName;
}

//-----------------------------------------------------------------------------
// Purpose: handles size changing
//-----------------------------------------------------------------------------
void ImagePanel::OnSizeChanged(int newWide, int newTall)
{
	BaseClass::OnSizeChanged(newWide, newTall);
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void ImagePanel::SetImage(IImage *image)
{
	m_pImage = image;
	Repaint();
}

//-----------------------------------------------------------------------------
// Purpose: sets an image by file name
//-----------------------------------------------------------------------------
void ImagePanel::SetImage(const char *imageName)
{
	int len = Q_strlen(imageName) + 1;
	delete[] m_pszImageName;
	m_pszImageName = new char[len];
	Q_strncpy(m_pszImageName, imageName, len);
	m_pImage = NULL;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
IImage *ImagePanel::GetImage()
{
	return m_pImage;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
Color ImagePanel::GetDrawColor(void)
{
	return m_DrawColor;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void ImagePanel::SetDrawColor(Color drawColor)
{
	m_DrawColor = drawColor;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void ImagePanel::PaintBackground()
{
	if (m_FillColor[3] > 0)
	{
		// draw the specified fill color
		int wide, tall;
		GetSize(wide, tall);
		surface()->DrawSetColor(m_FillColor);
		surface()->DrawFilledRect(0, 0, wide, tall);
	}
	if (m_pImage)
	{
		surface()->DrawSetColor(m_DrawColor);
		m_pImage->SetPos(0, 0);

		if (m_bScaleImage)
		{
			// Image size is stored in the bitmap, so temporarily set its size
			// to our panel size and then restore after we draw it.

			int imageWide, imageTall;
			m_pImage->GetSize(imageWide, imageTall);

			int wide, tall;
			GetSize(wide, tall);

			int newWide = imageWide;
			int newTall = imageTall;

			if (m_fScaleAmount > 0.0f)
			{
				newWide = static_cast<int>(static_cast<float>(imageWide) * m_fScaleAmount);
				newTall = static_cast<int>(static_cast<float>(imageTall) * m_fScaleAmount);
			}
			else
			{
				if (m_bCenterImage)
				{
					float wideRate = static_cast<float>(wide) / newWide;
					float tallRate = static_cast<float>(tall) / newTall;
					float scaleAmout = min(wideRate, tallRate);

					newWide = static_cast<int>(newWide * scaleAmout);
					newTall = static_cast<int>(newTall * scaleAmout);
				}
				else
				{
					newWide = wide;
					newTall = tall;
				}
			}

			if (m_bCenterImage)
			{
				int x = (wide - newWide) / 2;
				int y = (tall - newTall) / 2;

				m_pImage->SetPos(x, y);
			}

			m_pImage->SetSize(newWide, newTall);

			m_pImage->SetColor(GetDrawColor());
			m_pImage->Paint();

			m_pImage->SetSize(imageWide, imageTall);
		}
		else if (m_bCenterImage)
		{
			int imageWide, imageTall;
			m_pImage->GetSize(imageWide, imageTall);

			int wide, tall;
			GetSize(wide, tall);

			int x = (wide - imageWide) / 2;
			int y = (tall - imageTall) / 2;

			m_pImage->SetPos(x, y);
			m_pImage->SetColor(GetDrawColor());
			m_pImage->Paint();

			m_pImage->SetSize(imageWide, imageTall);
		}
		else if (m_bTileImage || m_bTileHorizontally || m_bTileVertically)
		{
			int wide, tall;
			GetSize(wide, tall);
			int imageWide, imageTall;
			m_pImage->GetSize(imageWide, imageTall);

			int y = 0;
			while (y < tall)
			{
				int x = 0;
				while (x < wide)
				{
					m_pImage->SetPos(x, y);
					m_pImage->Paint();

					x += imageWide;

					if (!m_bTileHorizontally)
						break;
				}

				y += imageTall;

				if (!m_bTileVertically)
					break;
			}
			m_pImage->SetPos(0, 0);
		}
		else
		{
			m_pImage->Paint();
		}
	}
	else
	{
		if (m_pszImageName && strlen(m_pszImageName) > 0)
		{
			SetImage(scheme()->GetImage(m_pszImageName, m_bScaleImage));
		}
	}
}

//-----------------------------------------------------------------------------
// Purpose: Gets control settings for editing
//-----------------------------------------------------------------------------
void ImagePanel::GetSettings(KeyValues *outResourceData)
{
	BaseClass::GetSettings(outResourceData);
	if (m_pszImageName)
	{
		outResourceData->SetString("image", m_pszImageName);
	}
	if (m_pszColorName)
	{
		outResourceData->SetString("fillcolor", m_pszColorName);
	}
	if (GetBorder())
	{
		outResourceData->SetString("border", GetBorder()->GetName());
	}

	outResourceData->SetInt("scaleImage", m_bScaleImage);
	outResourceData->SetInt("centerImage", m_bCenterImage);
	outResourceData->SetFloat("scaleAmount", m_fScaleAmount);
	outResourceData->SetInt("tileImage", m_bTileImage);
	outResourceData->SetInt("tileHorizontally", m_bTileHorizontally);
	outResourceData->SetInt("tileVertically", m_bTileVertically);
}

//-----------------------------------------------------------------------------
// Purpose: Applies designer settings from res file
//-----------------------------------------------------------------------------
void ImagePanel::ApplySettings(KeyValues *inResourceData)
{
	delete[] m_pszImageName;
	delete[] m_pszColorName;
	m_pszImageName = NULL;
	m_pszColorName = NULL;

	m_bScaleImage = inResourceData->GetInt("scaleImage", m_bScaleImage);
	m_bCenterImage = inResourceData->GetInt("centerImage", m_bCenterImage);
	m_fScaleAmount = inResourceData->GetFloat("scaleAmount", 0.0);
	m_bTileImage = inResourceData->GetInt("tileImage", 0);
	m_bTileHorizontally = inResourceData->GetInt("tileHorizontally", m_bTileImage);
	m_bTileVertically = inResourceData->GetInt("tileVertically", m_bTileImage);
	const char *imageName = inResourceData->GetString("image", "");
	if (*imageName)
	{
		SetImage(imageName);
	}

	const char *pszFillColor = inResourceData->GetString("fillcolor", "");
	if (*pszFillColor)
	{
		int r = 0, g = 0, b = 0, a = 255;
		int len = Q_strlen(pszFillColor) + 1;
		m_pszColorName = new char[len];
		Q_strncpy(m_pszColorName, pszFillColor, len);

		if (sscanf(pszFillColor, "%d %d %d %d", &r, &g, &b, &a) >= 3)
		{
			// it's a direct color
			m_FillColor = Color(r, g, b, a);
		}
		else
		{
			IScheme *pScheme = scheme()->GetIScheme(GetScheme());
			m_FillColor = pScheme->GetColor(pszFillColor, Color(0, 0, 0, 0));
		}
	}

	const char *pszBorder = inResourceData->GetString("border", "");
	if (*pszBorder)
	{
		IScheme *pScheme = scheme()->GetIScheme(GetScheme());
		SetBorder(pScheme->GetBorder(pszBorder));
	}

	BaseClass::ApplySettings(inResourceData);
}

//-----------------------------------------------------------------------------
// Purpose:  load the image, this is done just before this control is displayed
//-----------------------------------------------------------------------------
void ImagePanel::ApplySchemeSettings(IScheme *pScheme)
{
	BaseClass::ApplySchemeSettings(pScheme);
	if (m_pszImageName && strlen(m_pszImageName) > 0)
	{
		SetImage(scheme()->GetImage(m_pszImageName, m_bScaleImage));
	}
}

//-----------------------------------------------------------------------------
// Purpose: Describes editing details
//-----------------------------------------------------------------------------
const char *ImagePanel::GetDescription()
{
	static char buf[1024];
	_snprintf(buf, sizeof(buf), "%s, string image, string border, string fillcolor, bool scaleImage", BaseClass::GetDescription());
	return buf;
}

//-----------------------------------------------------------------------------
// Purpose: sets whether or not the image should scale to fit the size of the ImagePanel (defaults to false)
//-----------------------------------------------------------------------------
void ImagePanel::SetShouldScaleImage(bool state)
{
	m_bScaleImage = state;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void ImagePanel::SetShouldCenterImage(bool state)
{
	m_bCenterImage = state;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
bool ImagePanel::IsShouldScaleImage(void)
{
	return m_bScaleImage;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
bool ImagePanel::IsShouldCenterImage(void)
{
	return m_bCenterImage;
}

//-----------------------------------------------------------------------------
// Purpose: used in conjunction with setting that the image should scale and defines an absolute scale amount
//-----------------------------------------------------------------------------
void ImagePanel::SetScaleAmount(float scale)
{
	m_fScaleAmount = scale;
}

float ImagePanel::GetScaleAmount(void)
{
	return m_fScaleAmount;
}

//-----------------------------------------------------------------------------
// Purpose: set the color to fill with, if no Image is specified
//-----------------------------------------------------------------------------
void ImagePanel::SetFillColor(Color col)
{
	m_FillColor = col;
}

//-----------------------------------------------------------------------------
// Purpose: data accessor
//-----------------------------------------------------------------------------
Color ImagePanel::GetFillColor()
{
	return m_FillColor;
}

char *ImagePanel::GetImageName()
{
	return m_pszImageName;
}