#pragma once

class CModifiedButton : 
	public CButton
{
	bool   m_Changed;
	bool   m_Reset;
	HFONT  m_BoldFont;
	HFONT  m_OriginalFont;


public:
	// Constructors
	CModifiedButton(HWND hWnd = nullptr) : 
	    CButton(hWnd),
		m_Changed(false),
		m_Reset(false),
		m_BoldFont(nullptr),
		m_OriginalFont(nullptr)
	{ 		
	}
	
	~CModifiedButton()
	{
		if (m_BoldFont)
		{
			DeleteObject(m_BoldFont);
		}
	}

	void SetReset (bool Reset)
	{
		m_Reset = Reset;
		if (m_Reset)
		{
			SetChanged(false);
		}
	}

	void SetChanged (bool Changed)
	{
		m_Changed = Changed;
		if (m_Changed)
		{
			SetReset(false);
			if (m_BoldFont == nullptr)
			{
				m_OriginalFont = (HFONT)SendMessage(WM_GETFONT); 

				LOGFONT lfSystemVariableFont;
				GetObject ( m_OriginalFont, sizeof(LOGFONT), &lfSystemVariableFont );
				lfSystemVariableFont.lfWeight = FW_BOLD;

				m_BoldFont = CreateFontIndirect ( &lfSystemVariableFont );
			}
			SendMessage(WM_SETFONT,(WPARAM)m_BoldFont);
			InvalidateRect(nullptr);
		} else {
			if (m_OriginalFont)
			{
				SendMessage(WM_SETFONT,(WPARAM)m_OriginalFont);
				InvalidateRect(nullptr);
			}
		}
	}

	inline bool IsChanged ( void ) const 
	{
		return m_Changed;
	}
	inline bool IsReset ( void ) const 
	{
		return m_Reset;
	}
};