BOOL CXTTreeBase::PreTranslateMessage(MSG* pMsg)
{
    if (pMsg->message == WM_KEYDOWN)
    {
        // ensure that keystrokes are handled by the edit control.
        if (HasEditLabels() && m_pTreeCtrl->GetEditControl())
        {
            ::TranslateMessage(pMsg);
            ::DispatchMessage(pMsg);

            return TRUE;
        }

        // toggle expand / contract when return key is hit.
        if (pMsg->wParam == VK_RETURN)
        {
            HTREEITEM htItem = m_pTreeCtrl->GetSelectedItem();
            if (htItem != NULL)
            {
                m_pTreeCtrl->Expand(htItem, TVE_TOGGLE);
                return TRUE;
            }
        }
    }

    return m_pTreeCtrl->CTreeCtrl::PreTranslateMessage(pMsg);
}