开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > SQL Server > 查看源码
ListCtrlEx.cpp
资源名称：ISQL_src.zip　[点击查看]
上传用户：jsxglz
上传日期：2007-01-03
资源大小：117k
文件大小：12k
源码类别：

SQL Server

开发平台：

Visual C++

ListCtrlEx.cpp：源码内容
  1. // ListCtrlEx.cpp : implementation of the CListCtrlEx class
  3. #include "stdafx.h"
  4. #include "ListCtrlEx.h"
  6. #ifdef _DEBUG
  7. #define new DEBUG_NEW
  8. #undef THIS_FILE
  9. static char THIS_FILE[] = __FILE__;
  10. #endif
  12. /////////////////////////////////////////////////////////////////////////////
  13. // CListCtrlEx
  15. IMPLEMENT_DYNCREATE(CListCtrlEx, CListCtrl)
  17. BEGIN_MESSAGE_MAP(CListCtrlEx, CListCtrl)
  18. //{{AFX_MSG_MAP(CListCtrlEx)
  19. ON_WM_SIZE()
  20. ON_WM_PAINT()
  21. ON_WM_SETFOCUS()
  22. ON_WM_KILLFOCUS()
  23. //}}AFX_MSG_MAP
  24. ON_MESSAGE(LVM_SETIMAGELIST, OnSetImageList)
  25. ON_MESSAGE(LVM_SETTEXTCOLOR, OnSetTextColor)
  26. ON_MESSAGE(LVM_SETTEXTBKCOLOR, OnSetTextBkColor)
  27. ON_MESSAGE(LVM_SETBKCOLOR, OnSetBkColor)
  28. ON_NOTIFY_EX_RANGE(TTN_NEEDTEXTW, 0, 0xFFFF, OnToolTipText)
  29. ON_NOTIFY_EX_RANGE(TTN_NEEDTEXTA, 0, 0xFFFF, OnToolTipText)
  30. END_MESSAGE_MAP()
  32. /////////////////////////////////////////////////////////////////////////////
  33. // CListCtrlEx construction/destruction
  35. CListCtrlEx::CListCtrlEx()
  36. {
  37. m_bFullRowSel=FALSE;
  38. m_bClientWidthSel=TRUE;
  40. m_cxClient=0;
  41. m_cxStateImageOffset=0;
  43. m_clrText=::GetSysColor(COLOR_WINDOWTEXT);
  44. m_clrTextBk=::GetSysColor(COLOR_WINDOW);
  45. m_clrBkgnd=::GetSysColor(COLOR_WINDOW);
  46. }
  48. CListCtrlEx::~CListCtrlEx()
  49. {
  50. }
  52. BOOL CListCtrlEx::PreCreateWindow(CREATESTRUCT& cs)
  53. {
  54. // default is report view and full row selection
  55. cs.style &= ~LVS_TYPEMASK;
  56. cs.style |= LVS_REPORT | LVS_OWNERDRAWFIXED;
  58. m_bFullRowSel=TRUE;
  60. return(CListCtrl::PreCreateWindow(cs));
  61. }
  63. BOOL CListCtrlEx::SetFullRowSel(BOOL bFullRowSel)
  64. {
  65. // no painting during change
  66. LockWindowUpdate();
  68. m_bFullRowSel=bFullRowSel;
  70. BOOL bRet;
  72. if(m_bFullRowSel)
  73. bRet=ModifyStyle(0L,LVS_OWNERDRAWFIXED);
  74. else
  75. bRet=ModifyStyle(LVS_OWNERDRAWFIXED,0L);
  77. // repaint window if we are not changing view type
  78. if(bRet && (GetStyle() & LVS_TYPEMASK)==LVS_REPORT)
  79. Invalidate();
  81. // repaint changes
  82. UnlockWindowUpdate();
  84. return(bRet);
  85. }
  87. BOOL CListCtrlEx::GetFullRowSel()
  88. {
  89. return(m_bFullRowSel);
  90. }
  92. /////////////////////////////////////////////////////////////////////////////
  93. // CListCtrlEx drawing
  95. // offsets for first and other columns
  96. #define OFFSET_FIRST 2
  97. #define OFFSET_OTHER 6
  99. void CListCtrlEx::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
  100. {
  101. CDC* pDC=CDC::FromHandle(lpDrawItemStruct->hDC);
  102. CRect rcItem(lpDrawItemStruct->rcItem);
  103. UINT uiFlags=ILD_TRANSPARENT;
  104. CImageList* pImageList;
  105. int nItem=lpDrawItemStruct->itemID;
  106. BOOL bFocus=(GetFocus()==this);
  107. COLORREF clrTextSave, clrBkSave;
  108. COLORREF clrImage=m_clrBkgnd;
  109. static _TCHAR szBuff[MAX_PATH];
  110. LPCTSTR pszText;
  112. // get item data
  114. LV_ITEM lvi;
  115. lvi.mask=LVIF_TEXT | LVIF_IMAGE | LVIF_STATE;
  116. lvi.iItem=nItem;
  117. lvi.iSubItem=0;
  118. lvi.pszText=szBuff;
  119. lvi.cchTextMax=sizeof(szBuff);
  120. lvi.stateMask=0xFFFF; // get all state flags
  121. GetItem(&lvi);
  123. BOOL bSelected=(bFocus || (GetStyle() /*& LVS_SHOWSELALWAYS*/)) && lvi.state & LVIS_SELECTED;
  124. bSelected=bSelected || (lvi.state & LVIS_DROPHILITED);
  126. // set colors if item is selected
  128. CRect rcAllLabels;
  129. GetItemRect(nItem,rcAllLabels,LVIR_BOUNDS);
  130. CRect rcLabel;
  131. GetItemRect(nItem,rcLabel,LVIR_LABEL);
  132. rcAllLabels.left=rcLabel.left;
  133. if(m_bClientWidthSel && rcAllLabels.right<m_cxClient)
  134. rcAllLabels.right=m_cxClient;
  136. if(bSelected)
  137. {
  138. clrTextSave=pDC->SetTextColor(::GetSysColor(COLOR_HIGHLIGHTTEXT));
  139. clrBkSave=pDC->SetBkColor(::GetSysColor(COLOR_HIGHLIGHT));
  141. pDC->FillRect(rcAllLabels,&CBrush(::GetSysColor(COLOR_HIGHLIGHT)));
  142. }
  143. else
  144. pDC->FillRect(rcAllLabels,&CBrush(m_clrTextBk));
  146. // set color and mask for the icon
  148. if(lvi.state & LVIS_CUT)
  149. {
  150. clrImage=m_clrBkgnd;
  151. uiFlags|=ILD_BLEND50;
  152. }
  153. else if(bSelected)
  154. {
  155. clrImage=::GetSysColor(COLOR_HIGHLIGHT);
  156. uiFlags|=ILD_BLEND50;
  157. }
  159. // draw state icon
  161. UINT nStateImageMask=lvi.state & LVIS_STATEIMAGEMASK;
  162. if(nStateImageMask)
  163. {
  164. int nImage=(nStateImageMask>>12)-1;
  165. pImageList=GetImageList(LVSIL_STATE);
  166. if(pImageList)
  167. pImageList->Draw(pDC,nImage,CPoint(rcItem.left,rcItem.top),ILD_TRANSPARENT);
  168. }
  170. // draw normal and overlay icon
  172. CRect rcIcon;
  173. GetItemRect(nItem,rcIcon,LVIR_ICON);
  175. pImageList=GetImageList(LVSIL_SMALL);
  176. if(pImageList)
  177. {
  178. UINT nOvlImageMask=lvi.state & LVIS_OVERLAYMASK;
  179. if(rcItem.left<rcItem.right-1)
  180. ImageList_DrawEx(pImageList->m_hImageList,lvi.iImage,pDC->m_hDC,rcIcon.left,rcIcon.top,16,16,m_clrBkgnd,clrImage,uiFlags | nOvlImageMask);
  181. }
  183. // draw item label
  185. GetItemRect(nItem,rcItem,LVIR_LABEL);
  186. rcItem.right-=m_cxStateImageOffset;
  188. pszText=MakeShortString(pDC,szBuff,rcItem.right-rcItem.left,2*OFFSET_FIRST);
  190. rcLabel=rcItem;
  191. rcLabel.left+=OFFSET_FIRST;
  192. rcLabel.right-=OFFSET_FIRST;
  194. pDC->DrawText(pszText,-1,rcLabel,DT_LEFT | DT_SINGLELINE | DT_NOPREFIX | DT_NOCLIP | DT_VCENTER);
  196. // draw labels for extra columns
  198. LV_COLUMN lvc;
  199. lvc.mask=LVCF_FMT | LVCF_WIDTH;
  201. for(int nColumn=1; GetColumn(nColumn,&lvc); nColumn++)
  202. {
  203. rcItem.left=rcItem.right;
  204. rcItem.right+=lvc.cx;
  206.   int nRetLen=GetItemText(nItem,nColumn,szBuff,sizeof(szBuff));
  207. if(nRetLen==0) continue;
  209. pszText=MakeShortString(pDC,szBuff,rcItem.right-rcItem.left,2*OFFSET_OTHER);
  211. UINT nJustify=DT_LEFT;
  213. if(pszText==szBuff)
  214. {
  215. switch(lvc.fmt & LVCFMT_JUSTIFYMASK)
  216. {
  217. case LVCFMT_RIGHT:
  218. nJustify=DT_RIGHT;
  219. break;
  220. case LVCFMT_CENTER:
  221. nJustify=DT_CENTER;
  222. break;
  223. default:
  224. break;
  225. }
  226. }
  228. rcLabel=rcItem;
  229. rcLabel.left+=OFFSET_OTHER;
  230. rcLabel.right-=OFFSET_OTHER;
  232. pDC->DrawText(pszText,-1,rcLabel,nJustify | DT_SINGLELINE | DT_NOPREFIX | DT_NOCLIP | DT_VCENTER);
  233. }
  235. // draw focus rectangle if item has focus
  237. if(lvi.state & LVIS_FOCUSED && bFocus)
  238. pDC->DrawFocusRect(rcAllLabels);
  240. // set original colors if item was selected
  242. if(bSelected)
  243. {
  244.         pDC->SetTextColor(clrTextSave);
  245. pDC->SetBkColor(clrBkSave);
  246. }
  247. }
  249. LPCTSTR CListCtrlEx::MakeShortString(CDC* pDC, LPCTSTR lpszLong, int nColumnLen, int nOffset)
  250. {
  251. static const _TCHAR szThreeDots[]=_T("...");
  253. int nStringLen=lstrlen(lpszLong);
  255. if(nStringLen==0 || pDC->GetTextExtent(lpszLong,nStringLen).cx+nOffset<=nColumnLen)
  256. return(lpszLong);
  258. static _TCHAR szShort[MAX_PATH];
  260. lstrcpy(szShort,lpszLong);
  261. int nAddLen=pDC->GetTextExtent(szThreeDots,sizeof(szThreeDots)).cx;
  263. for(int i=nStringLen-1; i>0; i--)
  264. {
  265. szShort[i]=0;
  266. if(pDC->GetTextExtent(szShort,i).cx+nOffset+nAddLen<=nColumnLen)
  267. break;
  268. }
  270. lstrcat(szShort,szThreeDots);
  272. return(szShort);
  273. }
  275. void CListCtrlEx::RepaintSelectedItems()
  276. {
  277. CRect rcItem, rcLabel;
  279. // invalidate focused item so it can repaint properly
  281. int nItem=GetNextItem(-1,LVNI_FOCUSED);
  283. if(nItem!=-1)
  284. {
  285. GetItemRect(nItem,rcItem,LVIR_BOUNDS);
  286. GetItemRect(nItem,rcLabel,LVIR_LABEL);
  287. rcItem.left=rcLabel.left;
  289. InvalidateRect(rcItem,FALSE);
  290. }
  292. // if selected items should not be preserved, invalidate them
  294. if(!(GetStyle() & LVS_SHOWSELALWAYS))
  295. {
  296. for(nItem=GetNextItem(-1,LVNI_SELECTED);
  297. nItem!=-1; nItem=GetNextItem(nItem,LVNI_SELECTED))
  298. {
  299. GetItemRect(nItem,rcItem,LVIR_BOUNDS);
  300. GetItemRect(nItem,rcLabel,LVIR_LABEL);
  301. rcItem.left=rcLabel.left;
  303. InvalidateRect(rcItem,FALSE);
  304. }
  305. }
  307. // update changes 
  309. UpdateWindow();
  310. }
  312. /////////////////////////////////////////////////////////////////////////////
  313. // CListCtrlEx diagnostics
  315. #ifdef _DEBUG
  317. void CListCtrlEx::Dump(CDumpContext& dc) const
  318. {
  319. CListCtrl::Dump(dc);
  321. dc << "m_bFullRowSel = " << (UINT)m_bFullRowSel;
  322. dc << "n";
  323. dc << "m_cxStateImageOffset = " << m_cxStateImageOffset;
  324. dc << "n";
  325. }
  327. #endif //_DEBUG
  329. /////////////////////////////////////////////////////////////////////////////
  330. // CListCtrlEx message handlers
  332. LRESULT CListCtrlEx::OnSetImageList(WPARAM wParam, LPARAM lParam)
  333. {
  334. if((int)wParam==LVSIL_STATE)
  335. {
  336. int cx, cy;
  338. if(::ImageList_GetIconSize((HIMAGELIST)lParam,&cx,&cy))
  339. m_cxStateImageOffset=cx;
  340. else
  341. m_cxStateImageOffset=0;
  342. }
  344. return(Default());
  345. }
  347. LRESULT CListCtrlEx::OnSetTextColor(WPARAM wParam, LPARAM lParam)
  348. {
  349. m_clrText=(COLORREF)lParam;
  350. return(Default());
  351. }
  353. LRESULT CListCtrlEx::OnSetTextBkColor(WPARAM wParam, LPARAM lParam)
  354. {
  355. m_clrTextBk=(COLORREF)lParam;
  356. return(Default());
  357. }
  359. LRESULT CListCtrlEx::OnSetBkColor(WPARAM wParam, LPARAM lParam)
  360. {
  361. m_clrBkgnd=(COLORREF)lParam;
  362. return(Default());
  363. }
  365. void CListCtrlEx::OnSize(UINT nType, int cx, int cy) 
  366. {
  367. m_cxClient=cx;
  368. CListCtrl::OnSize(nType, cx, cy);
  369. }
  371. void CListCtrlEx::OnPaint() 
  372. {
  373. // in full row select mode, we need to extend the clipping region
  374. // so we can paint a selection all the way to the right
  375. if(m_bClientWidthSel && (GetStyle() & LVS_TYPEMASK)==LVS_REPORT && GetFullRowSel())
  376. {
  377. CRect rcAllLabels;
  378. GetItemRect(0,rcAllLabels,LVIR_BOUNDS);
  380. if(rcAllLabels.right<m_cxClient)
  381. {
  382. // need to call BeginPaint (in CPaintDC c-tor)
  383. // to get correct clipping rect
  384. CPaintDC dc(this);
  386. CRect rcClip;
  387. dc.GetClipBox(rcClip);
  389. rcClip.left=min(rcAllLabels.right-1,rcClip.left);
  390. rcClip.right=m_cxClient;
  392. InvalidateRect(rcClip,FALSE);
  393. // EndPaint will be called in CPaintDC d-tor
  394. }
  395. }
  397. CListCtrl::OnPaint();
  398. }
  400. void CListCtrlEx::OnSetFocus(CWnd* pOldWnd) 
  401. {
  402. CListCtrl::OnSetFocus(pOldWnd);
  404. // check if we are getting focus from label edit box
  405. if(pOldWnd!=NULL && pOldWnd->GetParent()==this)
  406. return;
  408. // repaint items that should change appearance
  409. if(m_bFullRowSel && (GetStyle() & LVS_TYPEMASK)==LVS_REPORT)
  410. RepaintSelectedItems();
  411. }
  413. void CListCtrlEx::OnKillFocus(CWnd* pNewWnd) 
  414. {
  415. CListCtrl::OnKillFocus(pNewWnd);
  417. // check if we are losing focus to label edit box
  418. if(pNewWnd!=NULL && pNewWnd->GetParent()==this)
  419. return;
  421. // repaint items that should change appearance
  422. if(m_bFullRowSel && (GetStyle() & LVS_TYPEMASK)==LVS_REPORT)
  423. RepaintSelectedItems();
  424. }
  426. void CListCtrlEx::PreSubclassWindow() 
  427. {
  428. CListCtrl::PreSubclassWindow();
  429. EnableToolTips(TRUE);
  430. }
  432. int CListCtrlEx::CellRectFromPoint(CPoint& point, CRect* pRectCell, int* pCol) const
  433. {
  434.     if((::GetWindowLong(m_hWnd, GWL_STYLE) & LVS_TYPEMASK) != LVS_REPORT )
  435.    return -1;
  437.     int nRow = GetTopIndex();
  438.     int nBottom = nRow + GetCountPerPage();
  439.     if(nBottom > GetItemCount())
  440. nBottom = GetItemCount();
  441.         
  442.     CHeaderCtrl* pHeader = (CHeaderCtrl*)GetDlgItem(0);
  443.     int nColumnCount = pHeader->GetItemCount();
  445.     for(; nRow <= nBottom; nRow++)
  446.     {
  447.         CRect rect;
  448.         GetItemRect(nRow, &rect, LVIR_BOUNDS);
  449.         if(rect.PtInRect(point))
  450.         {
  451. for(int nColumn = 0; nColumn < nColumnCount; nColumn++)
  452. {
  453. int nColWidth = GetColumnWidth(nColumn);
  454. if(point.x >= rect.left && point.x <= (rect.left + nColWidth))
  455. {
  456. CRect rectClient;
  457. GetClientRect(&rectClient);
  458. if(pCol)
  459. *pCol = nColumn;
  461. rect.right = rect.left + nColWidth;
  463. if(rect.right > rectClient.right) 
  464. rect.right = rectClient.right;
  466. *pRectCell = rect;
  468. return nRow;
  469. }
  471. rect.left += nColWidth;
  472. }
  473. }
  474. }
  476. return -1;
  477. }
  479. BOOL CListCtrlEx::OnToolTipText(UINT /*uID*/, NMHDR* pNMHDR, LRESULT* pResult)
  480. {
  481.     BOOL bRet = TRUE;
  482. #ifdef _TOOL_TIP_ENABLE
  483. TOOLTIPTEXTA* pTTTA = (TOOLTIPTEXTA*)pNMHDR;
  484.     TOOLTIPTEXTW* pTTTW = (TOOLTIPTEXTW*)pNMHDR;
  485.     UINT nID = pNMHDR->idFrom;
  486.     if(!nID)
  487.      bRet = FALSE;
  489. if(bRet)
  490. {
  491. int nRow = ((nID-1) >> 10) & 0x3fffff ;
  492. int nCol = (nID-1) & 0x3ff;
  493. CString strTipText = GetItemText(nRow, nCol);
  494. strTipText.TrimRight();
  495. strTipText.TrimLeft();
  496. #ifndef _UNICODE
  497. if (pNMHDR->code == TTN_NEEDTEXTA)
  498. lstrcpyn(pTTTA->szText, strTipText, 80);
  499. else
  500. _mbstowcsz(pTTTW->szText, strTipText, 80);
  501. #else
  502. if(pNMHDR->code == TTN_NEEDTEXTA)
  503. _wcstombsz(pTTTA->szText, strTipText, 80);
  504. else
  505. lstrcpyn(pTTTW->szText, strTipText, 80);
  506. #endif
  507. }
  508.     
  509. *pResult = 0;
  510. #endif
  512.     return bRet;
  513. }
  515. int CListCtrlEx::OnToolHitTest(CPoint point, TOOLINFO* pTI) const
  516. {
  517. int nID = -1;
  518. #ifdef _TOOL_TIP_ENABLE
  519. int nCol;
  520. CRect rect;
  521. int nRow = CellRectFromPoint(point, &rect, &nCol);
  522. if(nRow == -1 ) 
  523. return -1;
  525. pTI->hwnd = m_hWnd;
  526. pTI->uId = (UINT)((nRow << 10) + (nCol & 0x3ff) + 1);
  527. nID = pTI->uId
  528. pTI->lpszText = LPSTR_TEXTCALLBACK;
  530. pTI->rect = rect;
  531. #endif
  533. return nID;
  534. }