hl2_src-leak-2017/src/hammer/FilteredComboBox.cpp

880 lines
20 KiB
C++
Raw Normal View History

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================
#include "stdafx.h"
#include "FilteredComboBox.h"
BEGIN_MESSAGE_MAP(CFilteredComboBox, CComboBox)
//{{AFX_MSG_MAP(CFilteredComboBox)
ON_CONTROL_REFLECT_EX(CBN_SELCHANGE, OnSelChange)
ON_CONTROL_REFLECT_EX(CBN_EDITCHANGE, OnEditChange)
ON_CONTROL_REFLECT_EX(CBN_CLOSEUP, OnCloseUp)
ON_CONTROL_REFLECT_EX(CBN_DROPDOWN, OnDropDown)
ON_CONTROL_REFLECT_EX(CBN_SELENDOK, OnSelEndOK)
ON_WM_CTLCOLOR()
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
static const char *s_pStringToMatch = NULL;
static int s_iStringToMatchLen;
// This can help debug events in the combo box.
static int g_iFunctionMarkerEvent = 1;
class CFunctionMarker
{
public:
CFunctionMarker( const char *p )
{
#if 0
m_iEvent = g_iFunctionMarkerEvent++;
char str[512];
Q_snprintf( str, sizeof( str ), "enter %d: %s\n", m_iEvent, p );
OutputDebugString( str );
m_p = p;
#endif
}
~CFunctionMarker()
{
#if 0
char str[512];
Q_snprintf( str, sizeof( str ), "exit %d: %s\n", m_iEvent, m_p );
OutputDebugString( str );
#endif
}
const char *m_p;
int m_iEvent;
};
// ------------------------------------------------------------------------------------------------------------ //
// CFilteredComboBox implementation.
// ------------------------------------------------------------------------------------------------------------ //
CFilteredComboBox::CFilteredComboBox( CFilteredComboBox::ICallbacks *pCallbacks )
: m_pCallbacks( pCallbacks )
{
m_hQueuedFont = NULL;
m_bInSelChange = false;
m_bNotifyParent = true;
m_dwTextColor = RGB(0, 0, 0);
m_bOnlyProvideSuggestions = true;
m_hEditControlFont = NULL;
m_bInEnterKeyPressedHandler = false;
}
void CFilteredComboBox::SetSuggestions( CUtlVector<CString> &suggestions, int flags )
{
CreateFonts();
// Verify some of the window styles. This class requires these, and it doesn't get a change to set them
// unless you call Create on it.
// If we use owner draw variable, we get the bug described here: http://support.microsoft.com/kb/813791.
Assert( GetStyle() & CBS_OWNERDRAWFIXED );
Assert( GetStyle() & CBS_HASSTRINGS );
Assert( !( GetStyle() & CBS_SORT ) );
// Copy the list.
m_Suggestions = suggestions;
CString str;
GetWindowText( str );
DWORD sel = GetEditSel();
FillDropdownList( NULL, false );
// Force it to provide the first one if they only want suggestions and the current text in there is not valid.
bool bSelectFirst = ((flags & SETSUGGESTIONS_SELECTFIRST) != 0);
bool bCallback = ((flags & SETSUGGESTIONS_CALLBACK) != 0);
bool bForceFirst = (m_bOnlyProvideSuggestions && FindSuggestion( str ) == -1);
if ( bSelectFirst || bForceFirst )
{
SetCurSel( 0 );
if ( GetCount() > 0 )
{
CString strLB;
GetLBText( 0, strLB );
if ( bCallback )
DoTextChangedCallback( strLB );
}
else
{
m_LastTextChangedValue = "";
}
}
else
{
SetWindowText( str );
SetEditSel( LOWORD( sel ), HIWORD( sel ) );
if ( bCallback )
DoTextChangedCallback( str );
}
SetRedraw( true );
Invalidate();
}
void CFilteredComboBox::AddSuggestion( const CString &suggestion )
{
if ( FindSuggestion( suggestion ) == -1 )
m_Suggestions.AddToTail( suggestion );
}
void CFilteredComboBox::Clear()
{
m_Suggestions.Purge();
SetWindowText( "" );
}
void CFilteredComboBox::ForceEditControlText( const char *pStr )
{
SetWindowText( pStr );
}
void CFilteredComboBox::SelectItem( const char *pStr )
{
if ( !pStr )
{
SetEditControlText( "" );
return;
}
// See if we already have this item selected. If so, don't do anything.
int iCurSel = GetCurSel();
if ( iCurSel != CB_ERR )
{
CString str;
GetLBText( iCurSel, str );
if ( Q_stricmp( pStr, str ) == 0 )
{
// Make sure the edit control has the right text in there. If they called ForceEditControlText,
// then it might not.
CString strWindow;
GetWindowText( strWindow );
if ( Q_stricmp( strWindow, pStr ) != 0 )
{
SetWindowText( pStr );
}
m_LastTextChangedValue = pStr;
return;
}
}
if ( m_bOnlyProvideSuggestions && FindSuggestion( pStr ) == -1 )
{
// This item doesn't match any suggestion. We can get rid of this assert
// if it becomes a nuissance, but for now it's good to note that this
// is a weird situation.
Assert( false );
SetEditControlText( pStr );
return;
}
FillDropdownList( pStr );
}
CString CFilteredComboBox::GetCurrentItem()
{
return m_LastTextChangedValue;
}
void CFilteredComboBox::SetEditControlFont( HFONT hFont )
{
if ( !hFont )
return;
if ( m_bInSelChange )
{
m_hQueuedFont = hFont;
return;
}
CString str;
GetWindowText( str );
DWORD sel = GetEditSel();
InternalSetEditControlFont( hFont, str, sel );
}
void CFilteredComboBox::InternalSetEditControlFont( HFONT hFont, const char *pEditText, DWORD sel )
{
if ( hFont != m_hEditControlFont )
{
CFunctionMarker marker( "InternalSetEditControlFont" );
// Don't let it mess with everything here.
SetRedraw( false );
CRect rcMyRect;
GetWindowRect( rcMyRect );
CWnd *pParent = GetParent();
if ( pParent )
pParent->ScreenToClient( &rcMyRect );
BOOL bWasDropped = GetDroppedState();
m_hEditControlFont = hFont;
SetFont( CFont::FromHandle( m_hEditControlFont ), false );
SetWindowText( pEditText );
SetEditSel( LOWORD( sel ), HIWORD( sel ) );
if ( pParent )
MoveWindow( rcMyRect );
if ( bWasDropped )
ShowDropDown( true );
SetRedraw( true );
Invalidate();
}
}
HFONT CFilteredComboBox::GetEditControlFont() const
{
return m_hEditControlFont;
}
void CFilteredComboBox::SetEditControlTextColor(COLORREF dwColor)
{
m_dwTextColor = dwColor;
}
COLORREF CFilteredComboBox::GetEditControlTextColor() const
{
return m_dwTextColor;
}
void CFilteredComboBox::SetEditControlText( const char *pText )
{
SetWindowText( pText );
}
CString CFilteredComboBox::GetEditControlText() const
{
CString ret;
GetWindowText( ret );
return ret;
}
bool CFilteredComboBox::IsWindowEnabled() const
{
return (BaseClass::IsWindowEnabled() == TRUE);
}
void CFilteredComboBox::EnableWindow( bool bEnable )
{
BaseClass::EnableWindow( bEnable );
}
void CFilteredComboBox::SetOnlyProvideSuggestions( bool bOnlyProvideSuggestions )
{
m_bOnlyProvideSuggestions = bOnlyProvideSuggestions;
}
void CFilteredComboBox::FillDropdownList( const char *pInitialSel, bool bEnableRedraw )
{
CFunctionMarker marker( "FillDropdownList" );
SetRedraw( FALSE );
ResetContent();
// Fill the box with the initial set of values.
CUtlVector<CString> items;
GetItemsMatchingString( "", items );
for ( int i=0; i < items.Count(); i++ )
AddString( items[i] );
if ( pInitialSel )
{
CString str = pInitialSel;
if ( m_bOnlyProvideSuggestions )
{
str = GetBestSuggestion( pInitialSel );
if ( !InternalSelectItemByName( pInitialSel) )
{
Assert( false );
}
}
else
{
// Make sure we're putting the item they requested in there.
if ( !InternalSelectItemByName( str ) )
{
// Add the typed text to the combobox here otherwise it'll select the nearest match when they drop it down with the mouse.
AddString( str );
InternalSelectItemByName( str );
}
}
DoTextChangedCallback( str );
}
if ( bEnableRedraw )
{
SetRedraw( TRUE );
Invalidate();
}
}
LRESULT CFilteredComboBox::DefWindowProc(
UINT message,
WPARAM wParam,
LPARAM lParam
)
{
// We handle the enter key specifically because the default combo box behavior is to
// reset the text and all this stuff we don't want.
if ( message == WM_KEYDOWN )
{
if ( wParam == '\r' )
{
OnEnterKeyPressed( NULL );
return 0;
}
else if ( wParam == 27 )
{
// Escape..
OnEscapeKeyPressed();
return 0;
}
}
return BaseClass::DefWindowProc( message, wParam, lParam );
}
BOOL CFilteredComboBox::PreCreateWindow( CREATESTRUCT& cs )
{
// We need these styles in order for owner draw to work.
// If we use CBS_OWNERDRAWVARIABLE, then we run into this bug: http://support.microsoft.com/kb/813791.
cs.style |= CBS_OWNERDRAWFIXED | CBS_HASSTRINGS;
cs.style &= ~CBS_SORT;
return BaseClass::PreCreateWindow( cs );
}
void CFilteredComboBox::OnEnterKeyPressed( const char *pForceText )
{
if ( m_bInEnterKeyPressedHandler )
return;
CFunctionMarker marker( "OnEnterKeyPressed" );
m_bInEnterKeyPressedHandler = true;
// Must do this before ShowDropDown because that will change these variables underneath us.
CString szTypedText;
DWORD sel;
if ( pForceText )
{
szTypedText = pForceText;
sel = 0;
}
else
{
GetWindowText( szTypedText );
sel = GetEditSel();
}
CRect rcMyRect;
GetWindowRect( rcMyRect );
CWnd *pParent = GetParent();
if ( pParent )
pParent->ScreenToClient( &rcMyRect );
SetRedraw( false );
ShowDropDown( FALSE );
// They can get into here a variety of ways. Editing followed by enter. Editing+arrow keys, followed by enter, etc.
if ( m_bOnlyProvideSuggestions )
{
CString str;
if ( FindSuggestion( szTypedText ) == -1 && m_pCallbacks->OnUnknownEntry( szTypedText ) )
{
// They want us to KEEP this unknown entry, so add it to our list and select it.
m_Suggestions.AddToTail( szTypedText );
str = szTypedText;
}
else
{
// They returned false, so do the default behavior: go to the best match we can find.
str = GetBestSuggestion( szTypedText );
}
DoTextChangedCallback( str );
FillDropdownList( str, false );
if ( GetCurSel() == CB_ERR )
SetCurSel( 0 );
}
else
{
FillDropdownList( szTypedText, false );
SetWindowText( szTypedText );
SetEditSel( LOWORD(sel), HIWORD(sel) );
}
// Restore our window if necessary.
if ( pParent )
MoveWindow( &rcMyRect );
SetRedraw( true );
Invalidate();
DoTextChangedCallback( GetEditControlText() );
m_bInEnterKeyPressedHandler = false;
}
void CFilteredComboBox::OnEscapeKeyPressed()
{
// Fill it with everything and force it to select whatever we last selected.
m_bInEnterKeyPressedHandler = true;
ShowDropDown( FALSE );
m_bInEnterKeyPressedHandler = false;
FillDropdownList( m_LastTextChangedValue, true );
}
BOOL CFilteredComboBox::OnDropDown()
{
CFunctionMarker marker( "OnDropDown" );
// This is necessary to keep the cursor from disappearing.
SendMessage( WM_SETCURSOR, 0, 0 );
return !m_bNotifyParent;
}
//-----------------------------------------------------------------------------
// Purpose: Attaches this object to the given dialog item.
//-----------------------------------------------------------------------------
void CFilteredComboBox::SubclassDlgItem(UINT nID, CWnd *pParent)
{
//
// Disable parent notifications for CControlBar-derived classes. This is
// necessary because these classes result in multiple message reflections
// unless we return TRUE from our message handler.
//
if (pParent->IsKindOf(RUNTIME_CLASS(CControlBar)))
{
m_bNotifyParent = false;
}
else
{
m_bNotifyParent = true;
}
BaseClass::SubclassDlgItem(nID, pParent);
}
BOOL CFilteredComboBox::OnSelChange()
{
if ( !m_bInSelChange )
{
CFunctionMarker marker( "OnSelChange" );
CString strOriginalText;
GetWindowText( strOriginalText );
DWORD dwOriginalEditSel = GetEditSel();
m_bInSelChange = true;
int iSel = GetCurSel();
if ( iSel != CB_ERR )
{
CString str;
GetLBText( iSel, str );
strOriginalText = str;
DoTextChangedCallback( str );
}
m_bInSelChange = false;
if ( m_hQueuedFont )
{
HFONT hFont = m_hQueuedFont;
m_hQueuedFont = NULL;
m_bInSelChange = false;
InternalSetEditControlFont( hFont, strOriginalText, dwOriginalEditSel );
}
}
//
// Despite MSDN's lies, returning FALSE here allows the parent
// window to hook the notification message as well, not TRUE.
//
return !m_bNotifyParent;
}
BOOL CFilteredComboBox::OnCloseUp()
{
if ( !m_bInEnterKeyPressedHandler )
{
CFunctionMarker marker( "OnCloseUp" );
CString str;
if ( GetCurSel() == CB_ERR || GetCount() == 0 )
str = m_LastTextChangedValue;
else
GetLBText( GetCurSel(), str );
OnEnterKeyPressed( str );
}
//
// Despite MSDN's lies, returning FALSE here allows the parent
// window to hook the notification message as well, not TRUE.
//
return !m_bNotifyParent;
}
BOOL CFilteredComboBox::OnSelEndOK()
{
//
// Despite MSDN's lies, returning FALSE here allows the parent
// window to hook the notification message as well, not TRUE.
//
return !m_bNotifyParent;
}
BOOL CFilteredComboBox::OnEditChange()
{
CFunctionMarker marker( "OnEditChange" );
// Remember the text in the edit control because we're going to slam the
// contents of the list and we'll want to put the text back in.
CString szTypedText;
DWORD dwEditSel;
GetWindowText( szTypedText );
dwEditSel = GetEditSel();
// Show all the matching autosuggestions.
CUtlVector<CString> items;
GetItemsMatchingString( szTypedText, items );
SetRedraw( FALSE );
ResetContent();
for ( int i=0; i < items.Count(); i++ )
{
AddString( items[i] );
}
// Add the typed text to the combobox here otherwise it'll select the nearest match when they drop it down with the mouse.
if ( !m_bOnlyProvideSuggestions && FindSuggestion( szTypedText ) == -1 )
AddString( szTypedText );
// Note: for arcane and unspeakable MFC reasons, the placement of this call is VERY sensitive.
// For example, if CTargetNameComboBox changes from a bold font to a normal font, then if this
// call comes before ResetContent(), it will resize the dropdown listbox to a small size and not
// size it back until it is cloesd and opened again.
ShowDropDown();
SetRedraw( TRUE );
Invalidate();
// Possibly tell the app about this change.
if ( m_bOnlyProvideSuggestions )
{
if ( FindSuggestion( szTypedText ) != -1 )
DoTextChangedCallback( szTypedText );
}
else
{
DoTextChangedCallback( szTypedText );
}
// Put the text BACK in there.
SetWindowText( szTypedText );
SetEditSel( LOWORD( dwEditSel ), HIWORD( dwEditSel ) );
//
// Despite MSDN's lies, returning FALSE here allows the parent
// window to hook the notification message as well, not TRUE.
//
return !m_bNotifyParent;
}
int CFilteredComboBox::FindSuggestion( const char *pTest ) const
{
for ( int i=0; i < m_Suggestions.Count(); i++ )
{
if ( Q_stricmp( m_Suggestions[i], pTest ) == 0 )
return i;
}
return -1;
}
CString CFilteredComboBox::GetBestSuggestion( const char *pTest )
{
// If it's an exact match, use that.
if ( FindSuggestion( pTest ) != -1 )
return pTest;
// Look for the first autocomplete suggestion.
CUtlVector<CString> matches;
GetItemsMatchingString( pTest, matches );
if ( matches.Count() > 0 )
return matches[0];
// Ok, fall back to the last known good one.
return m_LastTextChangedValue;
}
CFont& CFilteredComboBox::GetNormalFont()
{
CreateFonts();
return m_NormalFont;
}
void CFilteredComboBox::GetItemsMatchingString( const char *pStringToMatch, CUtlVector<CString> &matchingItems )
{
for ( int i=0; i < m_Suggestions.Count(); i++ )
{
if ( MatchString( pStringToMatch, m_Suggestions[i] ) )
matchingItems.AddToTail( m_Suggestions[i] );
}
s_pStringToMatch = pStringToMatch;
s_iStringToMatchLen = V_strlen( pStringToMatch );
matchingItems.Sort( &CFilteredComboBox::SortFn );
s_pStringToMatch = NULL;
}
int CFilteredComboBox::SortFn( const CString *pItem1, const CString *pItem2 )
{
// If one of them matches the prefix we're looking at, then that one should be listed first.
// Otherwise, just do an alphabetical sort.
bool bPrefixMatch1=false, bPrefixMatch2=false;
if ( s_pStringToMatch )
{
bPrefixMatch1 = ( V_strnistr( *pItem1, s_pStringToMatch, s_iStringToMatchLen ) != NULL );
bPrefixMatch2 = ( V_strnistr( *pItem2, s_pStringToMatch, s_iStringToMatchLen ) != NULL );
}
if ( bPrefixMatch1 == bPrefixMatch2 )
{
return Q_stricmp( *pItem1, *pItem2 );
}
else
{
return bPrefixMatch1 ? -1 : 1;
}
}
bool CFilteredComboBox::MatchString( const char *pStringToMatchStart, const char *pTestStringStart )
{
if ( !pStringToMatchStart || pStringToMatchStart[0] == 0 )
return true;
while ( *pTestStringStart )
{
const char *pStringToMatch = pStringToMatchStart;
const char *pTestString = pTestStringStart;
while ( 1 )
{
// Skip underscores in both strings.
while ( *pStringToMatch == '_' )
++pStringToMatch;
while ( *pTestString == '_' )
++pTestString;
// If we're at the end of pStringToMatch with no mismatch, then treat this as a prefix match.
// If we're at the end of pTestString, but pStringToMatch has more to go, then it's not a match.
if ( *pStringToMatch == 0 )
return true;
else if ( *pTestString == 0 )
break;
// Match this character.
if ( toupper( *pStringToMatch ) != toupper( *pTestString ) )
break;
++pStringToMatch;
++pTestString;
}
++pTestStringStart;
}
return false;
}
//-----------------------------------------------------------------------------
// Purpose: Called before painting to override default colors.
// Input : pDC - DEvice context being painted into.
// pWnd - Control asking for color.
// nCtlColor - Type of control asking for color.
// Output : Returns the handle of a brush to use as the background color.
//-----------------------------------------------------------------------------
HBRUSH CFilteredComboBox::OnCtlColor(CDC *pDC, CWnd *pWnd, UINT nCtlColor)
{
HBRUSH hBrush = CComboBox::OnCtlColor(pDC, pWnd, nCtlColor);
if (nCtlColor == CTLCOLOR_EDIT)
{
pDC->SetTextColor(m_dwTextColor);
}
return(hBrush);
}
void CFilteredComboBox::DoTextChangedCallback( const char *pText )
{
// Sometimes it'll call here from a few places in a row. Only pass the result
// to the owner once.
if ( Q_stricmp( pText, m_LastTextChangedValue ) == 0 )
return;
m_LastTextChangedValue = pText;
m_pCallbacks->OnTextChanged( pText );
}
void CFilteredComboBox::CreateFonts()
{
//
// Create a normal and bold font.
//
if (!m_NormalFont.m_hObject)
{
CFont *pFont = GetFont();
if (pFont)
{
LOGFONT LogFont;
pFont->GetLogFont(&LogFont);
m_NormalFont.CreateFontIndirect(&LogFont);
}
}
}
void CFilteredComboBox::MeasureItem(LPMEASUREITEMSTRUCT pStruct)
{
HFONT hFont;
CFont *pFont = GetFont();
if ( pFont )
hFont = *pFont;
else
hFont = (HFONT)GetStockObject( DEFAULT_GUI_FONT );
CFont *pActualFont = CFont::FromHandle( hFont );
if ( pActualFont )
{
LOGFONT logFont;
pActualFont->GetLogFont( &logFont );
pStruct->itemHeight = abs( logFont.lfHeight ) + 5;
}
else
{
pStruct->itemHeight = 16;
}
}
void CFilteredComboBox::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
{
if ( GetCount() == 0 )
return;
CString str;
GetLBText( lpDrawItemStruct->itemID, str );
CDC dc;
dc.Attach( lpDrawItemStruct->hDC );
// Save these values to restore them when done drawing.
COLORREF crOldTextColor = dc.GetTextColor();
COLORREF crOldBkColor = dc.GetBkColor();
// If this item is selected, set the background color
// and the text color to appropriate values. Erase
// the rect by filling it with the background color.
// The left side of this expression was originally
// "(lpDrawItemStruct->itemAction | ODA_SELECT)", which is always true.
// To suppress the associated /analyze warning without changing
// behavior the expression was fixed but commented out.
if ( /*(lpDrawItemStruct->itemAction & ODA_SELECT) &&*/ (lpDrawItemStruct->itemState & ODS_SELECTED) )
{
dc.SetTextColor( ::GetSysColor(COLOR_HIGHLIGHTTEXT) );
dc.SetBkColor( ::GetSysColor(COLOR_HIGHLIGHT) );
dc.FillSolidRect( &lpDrawItemStruct->rcItem, ::GetSysColor(COLOR_HIGHLIGHT) );
}
else
{
dc.FillSolidRect(&lpDrawItemStruct->rcItem, crOldBkColor);
}
CFont *pOldFont = dc.SelectObject( &m_NormalFont );
// Draw the text.
RECT rcDraw = lpDrawItemStruct->rcItem;
rcDraw.left += 1;
dc.DrawText( str, -1, &rcDraw, DT_LEFT|DT_SINGLELINE|DT_VCENTER );
// Restore stuff.
dc.SelectObject( pOldFont );
dc.SetTextColor(crOldTextColor);
dc.SetBkColor(crOldBkColor);
dc.Detach();
}
bool CFilteredComboBox::InternalSelectItemByName( const char *pName )
{
int i = FindStringExact( -1, pName );
if ( i == CB_ERR )
{
return false;
}
else
{
SetCurSel( i );
CString str;
GetWindowText( str );
if ( Q_stricmp( str, pName ) != 0 )
SetWindowText( pName );
return true;
}
}