www.pudn.com > Property_List_src.zip > ChkComboBox.cpp

// ChkComboBox.cpp : implementation file 
// 
 
#include "stdafx.h" 
#include "ChkComboBox.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
extern void SplitString(LPCTSTR src, LPCTSTR delimiter, CStringList& strList); 
 
///////////////////////////////////////////////////////////////////////////// 
// CChkComboBox 
 
IMPLEMENT_DYNAMIC(CChkComboBox, CComboBox) 
 
static WNDPROC m_pWndProc = 0; 
static CChkComboBox *m_pComboBox = 0; 
 
BEGIN_MESSAGE_MAP(CChkComboBox, CComboBox) 
	//{{AFX_MSG_MAP(CChkComboBox) 
	ON_MESSAGE(WM_CTLCOLORLISTBOX, OnCtlColorListBox) 
	ON_MESSAGE(WM_GETTEXT, OnGetText) 
	ON_MESSAGE(WM_GETTEXTLENGTH, OnGetTextLength) 
	ON_CONTROL_REFLECT(CBN_DROPDOWN, OnDropDown) 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
 
// 
// The subclassed COMBOLBOX message handler 
// 
extern "C" LRESULT FAR PASCAL ComboBoxListBoxProc(HWND hWnd, UINT nMsg, WPARAM wParam, LPARAM lParam) 
{ 
 
	switch (nMsg) { 
 
		 
		case WM_RBUTTONDOWN: { 
			// If you want to select all/unselect all using the 
			// right button, remove this ifdef. Personally, I don't really like it 
//			#if FALSE 
 
				if (m_pComboBox != 0) { 
					INT nCount = m_pComboBox->GetCount(); 
					INT nSelCount = 0; 
 
					for (INT i = 0; i < nCount; i++) { 
						if (m_pComboBox->GetCheck(i)) 
							nSelCount++; 
					} 
 
 
					m_pComboBox->SelectAll(nSelCount != nCount); 
 
					// Make sure to invalidate this window as well 
					InvalidateRect(hWnd, 0, FALSE); 
					m_pComboBox->GetParent()->SendMessage(WM_COMMAND, MAKELONG(GetWindowLong(m_pComboBox->m_hWnd, GWL_ID), CBN_SELCHANGE), (LPARAM)m_pComboBox->m_hWnd); 
 
				} 
//			#endif 
 
			break; 
		} 
 
		// Make the combobox always return -1 as the current selection. This 
		// causes the lpDrawItemStruct->itemID in DrawItem() to be -1 
		// when the always-visible-portion of the combo is drawn 
		case LB_GETCURSEL: { 
			return -1; 
		} 
 
 
		case WM_CHAR: { 
			if (wParam == VK_SPACE) { 
				// Get the current selection 
				INT nIndex = CallWindowProcA(m_pWndProc, hWnd, LB_GETCURSEL, wParam, lParam); 
 
				CRect rcItem; 
				SendMessage(hWnd, LB_GETITEMRECT, nIndex, (LONG)(VOID *)&rcItem); 
				InvalidateRect(hWnd, rcItem, FALSE); 
 
				// Invert the check mark 
				m_pComboBox->SetCheck(nIndex, !m_pComboBox->GetCheck(nIndex)); 
 
				// Notify that selection has changed 
				m_pComboBox->GetParent()->SendMessage(WM_COMMAND, MAKELONG(GetWindowLong(m_pComboBox->m_hWnd, GWL_ID), CBN_SELCHANGE), (LPARAM)m_pComboBox->m_hWnd); 
				return 0; 
			} 
 
			break; 
		} 
 
 
		case WM_LBUTTONDOWN: { 
 
			CRect rcClient; 
			GetClientRect(hWnd, rcClient); 
 
			CPoint pt; 
			pt.x = LOWORD(lParam); 
			pt.y = HIWORD(lParam); 
 
 
			if (PtInRect(rcClient, pt)) { 
				INT nItemHeight = SendMessage(hWnd, LB_GETITEMHEIGHT, 0, 0); 
				INT nTopIndex   = SendMessage(hWnd, LB_GETTOPINDEX, 0, 0); 
 
				// Compute which index to check/uncheck 
				INT nIndex = nTopIndex + pt.y / nItemHeight; 
 
				CRect rcItem; 
				SendMessage(hWnd, LB_GETITEMRECT, nIndex, (LONG)(VOID *)&rcItem); 
 
				if (PtInRect(rcItem, pt)) { 
					// Invalidate this window 
					InvalidateRect(hWnd, rcItem, FALSE); 
					m_pComboBox->SetCheck(nIndex, !m_pComboBox->GetCheck(nIndex)); 
 
					// Notify that selection has changed 
					m_pComboBox->GetParent()->SendMessage(WM_COMMAND, MAKELONG(GetWindowLong(m_pComboBox->m_hWnd, GWL_ID), CBN_SELCHANGE), (LPARAM)m_pComboBox->m_hWnd); 
 
 
				} 
			} 
 
			// Do the default handling now (such as close the popup 
			// window when clicked outside) 
			break; 
		} 
 
		case WM_LBUTTONUP: { 
			// Don't do anything here. This causes the combobox popup 
			// windows to remain open after a selection has been made 
			return 0; 
		} 
	} 
 
	return CallWindowProc(m_pWndProc, hWnd, nMsg, wParam, lParam); 
} 
 
 
void CChkComboBox::SetMe() 
{ 
	OnDropDown() ; 
} 
 
void CChkComboBox::SetCallback(LONG fnCallback) 
{ 
    m_fnComboBoxCallBack = fnCallback; 
     
} 
 
CChkComboBox::CChkComboBox() 
{ 
	m_hListBox       = 0; 
	m_bTextUpdated   = FALSE; 
	m_bItemHeightSet = FALSE; 
    m_fnComboBoxCallBack = 0L; 
} 
 
 
CChkComboBox::~CChkComboBox() 
{ 
} 
 
 
BOOL CChkComboBox::Create(DWORD dwStyle, const RECT& rect, CWnd* pParentWnd, UINT nID) 
{ 
 
	// Remove the CBS_SIMPLE and CBS_DROPDOWN styles and add the one I'm designed for 
	dwStyle &= ~0xF; 
	dwStyle |= CBS_DROPDOWNLIST; 
 
	// Make sure to use the CBS_OWNERDRAWVARIABLE style 
	dwStyle |= CBS_OWNERDRAWVARIABLE; 
 
	// Use default strings. We need the itemdata to store checkmarks 
	dwStyle |= CBS_HASSTRINGS; 
 
	return CComboBox::Create(dwStyle, rect, pParentWnd, nID); 
} 
 
 
LRESULT CChkComboBox::OnCtlColorListBox(WPARAM wParam, LPARAM lParam)  
{ 
	// If the listbox hasn't been subclassed yet, do so... 
	if (m_hListBox == 0) { 
		HWND hWnd = (HWND)lParam; 
 
		if (hWnd != 0 && hWnd != m_hWnd) { 
			// Save the listbox handle 
			m_hListBox = hWnd; 
 
			// Do the subclassing 
			m_pWndProc = (WNDPROC)GetWindowLong(m_hListBox, GWL_WNDPROC); 
            SetWindowLong(m_hListBox, GWL_WNDPROC, m_fnComboBoxCallBack == 0  ? (LONG) ComboBoxListBoxProc : m_fnComboBoxCallBack); 
		} 
	} 
 
	 
	return DefWindowProc(WM_CTLCOLORLISTBOX, wParam, lParam); 
} 
 
 
void CChkComboBox::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)  
{ 
	HDC dc = lpDrawItemStruct->hDC; 
 
	CRect rcBitmap = lpDrawItemStruct->rcItem; 
	CRect rcText   = lpDrawItemStruct->rcItem; 
 
	CString strText; 
 
	// 0 - No check, 1 - Empty check, 2 - Checked 
	INT nCheck = 0; 
 
	// Check if we are drawing the static portion of the combobox 
	if ((LONG)lpDrawItemStruct->itemID < 0) { 
		// Make sure the m_strText member is updated 
		RecalcText(); 
 
		// Get the text 
		strText = m_strText; 
 
		// Don't draw any boxes on this item 
		nCheck = 0; 
	} 
 
	// Otherwise it is one of the items 
	else { 
		GetLBText(lpDrawItemStruct->itemID, strText); 
		nCheck = 1 + (GetItemData(lpDrawItemStruct->itemID) != 0); 
 
		TEXTMETRIC metrics; 
		GetTextMetrics(dc, &metrics); 
 
		rcBitmap.left    = 0; 
		rcBitmap.right   = rcBitmap.left + metrics.tmHeight + metrics.tmExternalLeading + 6; 
		rcBitmap.top    += 1; 
		rcBitmap.bottom -= 1; 
 
		rcText.left = rcBitmap.right; 
	} 
	 
	  
 
	if (nCheck > 0) { 
		SetBkColor(dc, GetSysColor(COLOR_WINDOW)); 
		SetTextColor(dc, GetSysColor(COLOR_WINDOWTEXT)); 
 
		UINT nState = DFCS_BUTTONCHECK; 
 
		if (nCheck > 1) 
			nState |= DFCS_CHECKED; 
 
		// Draw the checkmark using DrawFrameControl 
		DrawFrameControl(dc, rcBitmap, DFC_BUTTON, nState | DFCS_FLAT); 
	} 
 
	if (lpDrawItemStruct->itemState & ODS_SELECTED) { 
		SetBkColor(dc, GetSysColor(COLOR_HIGHLIGHT)); 
		SetTextColor(dc, GetSysColor(COLOR_HIGHLIGHTTEXT)); 
	} 
	else { 
		SetBkColor(dc, GetSysColor(COLOR_WINDOW)); 
		SetTextColor(dc, GetSysColor(COLOR_WINDOWTEXT)); 
	} 
 
	// Erase and draw 
	ExtTextOut(dc, 0, 0, ETO_OPAQUE, &rcText, 0, 0, 0); 
	DrawText(dc, ' ' + strText, strText.GetLength() + 1, &rcText, DT_SINGLELINE|DT_VCENTER|DT_END_ELLIPSIS); 
 
	if ((lpDrawItemStruct->itemState & (ODS_FOCUS|ODS_SELECTED)) == (ODS_FOCUS|ODS_SELECTED)) 
		DrawFocusRect(dc, &rcText); 
	 
} 
 
 
void CChkComboBox::MeasureItem(LPMEASUREITEMSTRUCT lpMeasureItemStruct)  
{ 
	CClientDC dc(this); 
	CFont *pFont = dc.SelectObject(GetFont()); 
 
	if (pFont != 0) { 
 
		TEXTMETRIC metrics; 
		dc.GetTextMetrics(&metrics); 
 
		lpMeasureItemStruct->itemHeight = metrics.tmHeight + metrics.tmExternalLeading; 
 
		// An extra height of 2 looks good I think.  
		// Otherwise the list looks a bit crowded... 
		lpMeasureItemStruct->itemHeight += 2; 
 
 
		// This is needed since the WM_MEASUREITEM message is sent before 
		// MFC hooks everything up if used in i dialog. So adjust the 
		// static portion of the combo box now 
		if (!m_bItemHeightSet) { 
			m_bItemHeightSet = TRUE; 
			SetItemHeight(-1, lpMeasureItemStruct->itemHeight); 
		} 
 
		dc.SelectObject(pFont); 
	} 
} 
 
 
// 
// Make sure the combobox window handle is updated since 
// there may be many CChkComboBox windows active 
// 
void CChkComboBox::OnDropDown()  
{ 
	m_pComboBox = this; 
	 
} 
 
 
// 
// Selects/unselects all items in the list 
// 
void CChkComboBox::SelectAll(BOOL bCheck) 
{ 
	INT nCount = GetCount(); 
 
	for (INT i = 0; i < nCount; i++) 
		SetCheck(i, bCheck); 
 
} 
 
 
// 
// By adding this message handler, we may use CWnd::GetText() 
// 
LRESULT CChkComboBox::OnGetText(WPARAM wParam, LPARAM lParam) 
{ 
	// Make sure the text is updated 
	RecalcText(); 
 
	if (lParam == 0) 
		return 0; 
 
	// Copy the 'fake' window text 
	lstrcpyn((LPSTR)lParam, m_strText, (INT)wParam); 
	return m_strText.GetLength(); 
} 
 
 
// 
// By adding this message handler, we may use CWnd::GetTextLength() 
// 
LRESULT CChkComboBox::OnGetTextLength(WPARAM, LPARAM) 
{ 
	// Make sure the text is updated 
	RecalcText(); 
	return m_strText.GetLength(); 
} 
 
 
// 
// This routine steps thru all the items and builds 
// a string containing the checked items 
// 
void CChkComboBox::RecalcText() 
{ 
	if (!m_bTextUpdated) { 
		CString strText; 
		 
		// Get the list count 
		INT nCount    = GetCount(); 
 
		CString strSeparator; 
 
		// If none found, the the '|' 
		if (strSeparator.GetLength() == 0) 
			strSeparator = '|'; 
 
		// Trim extra spaces 
		strSeparator.TrimRight(); 
 
		// And one... 
//		strSeparator += ' ';//no separator 
 
		for (INT i = 0; i < nCount; i++) { 
 
			if (GetItemData(i)) { 
				CString strItem; 
				GetLBText(i, strItem); 
 
				if (!strText.IsEmpty()) 
					strText += strSeparator; 
 
				strText += strItem; 
			} 
		} 
 
		// Set the text 
		m_strText = strText; 
 
		m_bTextUpdated = TRUE; 
	} 
} 
 
INT CChkComboBox::SetCheck(INT nIndex, BOOL bFlag) 
{ 
	INT nResult = SetItemData(nIndex, bFlag); 
 
	if (nResult < 0) 
		return nResult; 
 
	// Signal that the text need updating 
	m_bTextUpdated = FALSE; 
 
	// Redraw the window 
	Invalidate(FALSE); 
 
	return nResult; 
} 
 
BOOL CChkComboBox::GetCheck(INT nIndex) 
{ 
	return GetItemData(nIndex); 
} 
 
void CChkComboBox::SetOptions(LPCTSTR szOptions) 
{ 
	CStringList strList; 
 
	SplitString(szOptions, "|", strList); 
 
	for (POSITION pos=strList.GetHeadPosition(); pos!=NULL; ) 
	{ 
		CString str = strList.GetNext(pos); 
		AddString(str); 
	} 
} 
 
void CChkComboBox::SetValues(LPCTSTR szValue) 
{ 
	CStringList strList; 
	 
	SplitString(szValue, "|", strList); 
	 
	for (POSITION pos=strList.GetHeadPosition(); pos!=NULL; ) 
	{ 
		CString strSubValue = strList.GetNext(pos); 
		UINT nIndex = FindString(0, strSubValue); 
		SetCheck(nIndex, true); 
	} 
}