#include "stdafx.h"

CModifiedEditBox::CModifiedEditBox(bool bString /* = true */, HWND hWnd /* = nullptr */) :
CEdit(hWnd),
m_Changed(false),
m_Reset(false),
m_BoldFont(nullptr),
m_OriginalFont(nullptr),
m_TextField(nullptr),
m_bString(bString)
{
}

CModifiedEditBox::~CModifiedEditBox()
{
    if (m_BoldFont)
    {
        DeleteObject(m_BoldFont);
    }
}

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

void CModifiedEditBox::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);
        if (m_TextField)
        {
            ::SendMessage(m_TextField, WM_SETFONT, (WPARAM)m_BoldFont, 0);
            ::InvalidateRect(m_TextField, nullptr, true);
        }
    }
    else
    {
        if (m_OriginalFont)
        {
            SendMessage(WM_SETFONT, (WPARAM)m_OriginalFont);
            InvalidateRect(nullptr);
            if (m_TextField)
            {
                ::SendMessage(m_TextField, WM_SETFONT, (WPARAM)m_OriginalFont, 0);
                ::InvalidateRect(m_TextField, nullptr, true);
            }
        }
    }
}

void CModifiedEditBox::SetTextField(HWND hWnd)
{
    if (m_TextField && m_OriginalFont)
    {
        ::SendMessage(m_TextField, WM_SETFONT, (WPARAM)m_OriginalFont, 0);
    }
    m_TextField = hWnd;
    if (m_Changed && m_BoldFont)
    {
        ::SendMessage(m_TextField, WM_SETFONT, (WPARAM)m_BoldFont, 0);
    }
}