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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图象 > 查看源码
UITreeCtrl.cpp
资源名称：isee_code01231.zip　[点击查看]
上传用户：yatsl7111
上传日期：2007-01-08
资源大小：1433k
文件大小：22k
源码类别：

图形图象

开发平台：

Visual C++

UITreeCtrl.cpp：源码内容
  1. //*******************************************************************************
  2. // COPYRIGHT NOTES
  3. // ---------------
  4. // You may use this source code, compile or redistribute it as part of your application 
  5. // for free. You cannot redistribute it as a part of a software development 
  6. // library without the agreement of the author. If the sources are 
  7. // distributed along with the application, you should leave the original 
  8. // copyright notes in the source code without any changes.
  9. // This code can be used WITHOUT ANY WARRANTIES at your own risk.
  10. // 
  11. // For the latest updates to this code, check this site:
  12. // http://www.masmex.com 
  13. // after Sept 2000
  14. // 
  15. // Copyright(C) 2000 Philip Oldaker <email: philip@masmex.com>
  16. //*******************************************************************************
  18. // URLTreeCtrl.cpp : implementation file
  19. //
  21. #include "stdafx.h"
  22. #include "UITreeCtrl.h"
  23. #include "UIMessages.h"
  24. #include "cbformats.h"
  25. #include "UIres.h"
  27. #ifdef _DEBUG
  28. #define new DEBUG_NEW
  29. #undef THIS_FILE
  30. static char THIS_FILE[] = __FILE__;
  31. #endif
  33. /////////////////////////////////////////////////////////////////////////////
  34. // CUITreeCtrl
  35. IMPLEMENT_DYNAMIC(CUITreeCtrl,CUIDragDropTree)
  36. IMPLEMENT_DYNAMIC(CRefresh,CObject)
  37. IMPLEMENT_DYNAMIC(CRefreshCategory,CRefresh)
  40. CUITreeCtrl::CUITreeCtrl(bool bDragDrop) 
  41. : CUIDragDropTree(bDragDrop)
  42. {
  43. m_idxClosed= 0;
  44. m_idxOpen = 0; 
  45. m_vkeydown = 0;
  46. m_IDTimer = 0;
  47. m_style = TVS_HASLINES | TVS_HASBUTTONS | TVS_LINESATROOT 
  48. | TVS_SHOWSELALWAYS | TVS_DISABLEDRAGDROP;
  49. m_PopupID = 0;
  50. m_pPopupWnd = NULL;
  51. m_hOrigFont = NULL;
  52. m_bMe = false;
  53. }
  55. CUITreeCtrl::~CUITreeCtrl()
  56. {
  57. }
  59. void CUITreeCtrl::OnDeleteItemData(DWORD dwData)
  60. {
  62. }
  64. void CUITreeCtrl::ItemPrePaint(LPNMTVCUSTOMDRAW lptvcd,LRESULT *pResult)
  65. {
  66. *pResult = CDRF_DODEFAULT;
  67. if (lptvcd->nmcd.dwDrawStage & CDDS_ITEM)
  68. {
  69. UINT state = GetItemState((HTREEITEM)lptvcd->nmcd.dwItemSpec,TVIS_SELECTED | TVIS_EXPANDED);
  70. if ((state & TVIS_SELECTED))
  71. return;
  72. if ((state & TVIS_SELECTED) && (state & TVIS_EXPANDED))
  73. return;
  74. }
  75. HTREEITEM hItem = (HTREEITEM)lptvcd->nmcd.dwItemSpec;
  76. CUIListCtrlData* pData = GetListCtrlData(hItem);
  77. if (pData == NULL)
  78. return;
  79. *pResult = CDRF_NOTIFYPOSTPAINT;
  80. if (pData->IsFontSet())
  81. {
  82. CDC *pDC = CDC::FromHandle(lptvcd->nmcd.hdc);
  83. CFont *pOldFont = pDC->SelectObject((CFont*)pData->GetFont());
  84. m_hOrigFont = (HFONT)pOldFont;
  85. *pResult |= CDRF_NEWFONT;
  86. }
  87. lptvcd->clrText = pData->GetTextColor();
  88. lptvcd->clrTextBk = pData->GetBkColor();
  89. }
  91. void CUITreeCtrl::ItemPostPaint(LPNMTVCUSTOMDRAW lptvcd,LRESULT *pResult)
  92. {
  93. CDC *pDC = CDC::FromHandle(lptvcd->nmcd.hdc);
  94. if (m_hOrigFont)
  95. {
  96. pDC->SelectObject(CFont::FromHandle(m_hOrigFont));
  97. m_hOrigFont = NULL;
  98. }
  99. *pResult = CDRF_DODEFAULT;
  100. }
  102. bool CUITreeCtrl::EndLabelEdit(HTREEITEM hItem,LPCTSTR pszText)
  103. {
  104. return false;
  105. }
  107. UINT CUITreeCtrl::GetMenuID()
  108. {
  109. return m_PopupID;
  110. }
  112. CWnd *CUITreeCtrl::GetMenuWnd()
  113. {
  114. return m_pPopupWnd;
  115. }
  117. void CUITreeCtrl::SetPopupID(UINT nID)
  118. {
  119. m_PopupID = nID;
  120. }
  122. void CUITreeCtrl::SetPopupWnd(CWnd *pWnd)
  123. {
  124. m_pPopupWnd = pWnd;
  125. }
  127. bool CUITreeCtrl::DragDrop(CDD_OleDropTargetInfo *pInfo)
  128. {
  129. if (pInfo->GetDataObject() == NULL)
  130. {
  131. HTREEITEM hItem = pInfo->GetTreeItem();
  132. if (pInfo->GetDropEffect() == DROPEFFECT_MOVE)
  133. {
  134. if (TransferItem(GetDragItem(),hItem))
  135. {
  136. DeleteItem(GetDragItem());
  137. }
  138. }
  139. else
  140. {
  141. AfxMessageBox(_T("Copy not supported!"));
  142. }
  143. return true;
  144. }
  145. return false;
  146. }
  148. bool CUITreeCtrl::DragEnter(CDD_OleDropTargetInfo *pInfo)
  149. {
  150. return false;
  151. }
  153. bool CUITreeCtrl::DragLeave(CDD_OleDropTargetInfo *pInfo)
  154. {
  155. return false;
  156. }
  158. bool CUITreeCtrl::DragOver(CDD_OleDropTargetInfo *pInfo)
  159. {
  160. // possible non ole drag & drop
  161. if (pInfo->GetDataObject() == NULL)
  162. {
  163. return true;
  164. }
  165. // ole drag and drop expects the dropeffect to be set
  166. // any valid formats?
  167. if (CWDClipboardData::Instance()->IsDataAvailable(pInfo->GetDataObject()))
  168. {
  169. FORMATETC FormatEtc;
  170. FormatEtc.cfFormat = CWDClipboardData::Instance()->GetClipboardFormat(CWDClipboardData::e_cfFileGroupDesc);
  171. FormatEtc.ptd = NULL;
  172. FormatEtc.dwAspect = DVASPECT_CONTENT;
  173. FormatEtc.lindex = -1;
  174. FormatEtc.tymed = TYMED_HGLOBAL;
  175. HRESULT hr = pInfo->GetDataObject()->m_lpDataObject->QueryGetData(&FormatEtc);
  176. if ((pInfo->GetKeyboardState() & MK_SHIFT))
  177. pInfo->SetDropEffect(DROPEFFECT_MOVE);
  178. else if (hr == S_OK)
  179. {
  180. pInfo->SetDropEffect(DROPEFFECT_COPY);
  181. }
  182. else
  183. pInfo->SetDropEffect(DROPEFFECT_COPY);
  184. return true;
  185. }
  186. return false;
  187. }
  189. DROPEFFECT CUITreeCtrl::DoDragDrop(NM_TREEVIEW* pNMTreeView,COleDataSource *pOleDataSource)
  190. {
  191. // non ole drag and drop only within tree
  192. CCF_String ccfText(GetItemText(pNMTreeView->itemNew.hItem));
  193. CWDClipboardData::Instance()->SetData(pOleDataSource,&ccfText,CWDClipboardData::e_cfString);
  194. return DROPEFFECT_ALL;
  195. }
  197. CRefresh *CUITreeCtrl::CreateRefreshObject(HTREEITEM hItem,LPARAM lParam)
  198. {
  199. return new CRefresh(hItem,lParam);
  200. }
  202. void CUITreeCtrl::UpdateEvent(LPARAM lHint,CObject *pHint)
  203. {
  204. GetParent()->SendMessage(WM_APP_UPDATE_ALL_VIEWS,(WPARAM)lHint,(LPARAM)pHint);
  205. }
  207. void CUITreeCtrl::UpdateCurrentSelection()
  208. {
  209. HTREEITEM hItem = GetSelectedItem();
  210. if (hItem)
  211. {
  212. SelectionChanged(hItem,GetItemData(hItem));
  213. }
  214. }
  216. void CUITreeCtrl::DoubleClick(HTREEITEM hItem)
  217. {
  218. }
  220. void CUITreeCtrl::SelectionChanged(HTREEITEM hItem,LPARAM lParam)
  221. {
  222. CRefresh *pRefresh = CreateRefreshObject(hItem,lParam);
  223. if (pRefresh)
  224. UpdateEvent(HINT_TREE_SEL_CHANGED,pRefresh);
  225. delete pRefresh;
  226. }
  228. HTREEITEM CUITreeCtrl::AddAnItem(HTREEITEM hParent, LPCTSTR szText, LPARAM lParam, HTREEITEM hInsAfter,int iImage,int iSelImage, int nChildren)
  229. {
  230. ASSERT_VALID(this);
  232. TV_ITEM tvI;
  233. TV_INSERTSTRUCT tvIns;
  234. ZeroMemory(&tvI,sizeof(tvI));
  235. ZeroMemory(&tvIns,sizeof(tvIns));
  236. tvI.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
  237. if (nChildren) // get the plus sign
  238. {
  239. tvI.mask |= TVIF_CHILDREN;
  240. tvI.cChildren = nChildren;
  241. }
  242. tvI.pszText = (LPTSTR)szText;
  243. tvI.cchTextMax = lstrlen(szText);
  244. if (iImage == 0)
  245. iImage = m_idxClosed;
  246. tvI.iImage = iImage;
  247. if (iSelImage == 0)
  248. iSelImage = iImage;
  249. tvI.iSelectedImage = iSelImage;
  250. tvI.lParam = (LPARAM)GetNewListCtrlData(lParam,hParent);
  251. tvIns.item = tvI;
  252. tvIns.hInsertAfter = hInsAfter;
  253. tvIns.hParent = hParent;
  255. m_bMe = true;
  256. HTREEITEM hNewItem = InsertItem(&tvIns);
  257. m_bMe = false;
  258. return hNewItem;
  259. }
  261. CUIListCtrlData *CUITreeCtrl::GetNewListCtrlData(DWORD dwData,HTREEITEM hParent)
  262. {
  263. CUIListCtrlData *pListCtrlData = new CUIListCtrlData;
  264. pListCtrlData->SetExtData(dwData); 
  265. return pListCtrlData;
  266. }
  268. DWORD CUITreeCtrl::GetItemData(HTREEITEM hItem) const
  269. {
  270. if (hItem == NULL)
  271. return 0;
  272. const CUIListCtrlData* pData = GetListCtrlData(hItem);
  273. if (pData)
  274. return pData->GetExtData();
  275. return 0;
  276. }
  278. BOOL CUITreeCtrl::SetItemData(HTREEITEM hItem,DWORD dwData,bool bAutoDelete)
  279. {
  280. ASSERT(hItem);
  281. if (hItem == NULL) 
  282. return FALSE;
  283. CUIListCtrlData* pData = GetListCtrlData(hItem);
  284. if (pData == NULL) 
  285. return FALSE;
  286. pData->SetExtData(dwData);
  287. pData->SetAutoDelete(bAutoDelete);
  288. return TRUE;
  289. }
  291. CUIListCtrlData *CUITreeCtrl::GetListCtrlData(HTREEITEM hItem) const
  292. {
  293. ((CUITreeCtrl*)this)->m_bMe = true;
  294. CUIListCtrlData *pData = (CUIListCtrlData*)CTreeCtrl::GetItemData(hItem);
  295. ((CUITreeCtrl*)this)->m_bMe = false;
  296. if (pData == NULL)
  297. return NULL;
  298. ASSERT(pData->IsKindOf(RUNTIME_CLASS(CUIListCtrlData)));
  299. return pData;
  300. }
  302. void CUITreeCtrl::SetTextColor(HTREEITEM hItem,COLORREF TextColor)
  303. {
  304. CUIListCtrlData *pData = GetListCtrlData(hItem);
  305. if (pData)
  306. pData->SetTextColor(TextColor);
  307. };
  309. void CUITreeCtrl::SetBkColor(HTREEITEM hItem,COLORREF BkColor)
  310. {
  311. CUIListCtrlData *pData = GetListCtrlData(hItem);
  312. if (pData)
  313. pData->SetBkColor(BkColor);
  314. }
  316. void CUITreeCtrl::SetDefaultTextColor(HTREEITEM hItem)
  317. {
  318. CUIListCtrlData *pData = GetListCtrlData(hItem);
  319. if (pData)
  320. pData->SetDefaultTextColor();
  321. };
  323. void CUITreeCtrl::SetDefaultBkColor(HTREEITEM hItem)
  324. {
  325. CUIListCtrlData *pData = GetListCtrlData(hItem);
  326. if (pData)
  327. pData->SetDefaultBkColor();
  328. }
  330. void CUITreeCtrl::SetItemFont(HTREEITEM hItem,CFont *pFont)
  331. {
  332. CUIListCtrlData* pData = GetListCtrlData(hItem);
  333. if (pData == NULL)
  334. return;
  335. ASSERT(pFont);
  336. if (pFont)
  337. pData->SetFont(pFont);
  338. }
  340. void CUITreeCtrl::SetItemBold(HTREEITEM hItem,bool bBold)
  341. {
  342. CUIListCtrlData* pData = GetListCtrlData(hItem);
  343. if (pData == NULL)
  344. return;
  345. if (!pData->IsFontSet())
  346. pData->SetFont(GetFont());
  347. const CFont *pFont = pData->GetFont();
  348. LOGFONT lf;
  349. ((CFont*)pFont)->GetLogFont(&lf);
  350. lf.lfWeight = bBold ? FW_BOLD : FW_NORMAL; 
  351. CFont font;
  352. font.CreateFontIndirect(&lf);
  353. pData->SetFont(&font);
  354. CRect rect;
  355. if (GetItemRect(hItem,&rect,FALSE))
  356. {
  357. InvalidateRect(rect);
  358. UpdateWindow();
  359. }
  360. }
  362. HTREEITEM CUITreeCtrl::GetHitItem(CPoint *pt)
  363. {
  364. UINT Flag;
  365. HTREEITEM hItem;
  366. CPoint cpt;
  368. POINT pt1;
  369. GetCursorPos(&pt1);
  370. ScreenToClient(&pt1);
  371. cpt.x = pt1.x;
  372. cpt.y = pt1.y;
  373. hItem = HitTest(cpt, &Flag);
  374. if (hItem && ((Flag & TVHT_ONITEMLABEL) || (Flag & TVHT_ONITEMICON)))
  375. {
  376. if (pt)
  377. *pt = pt1;
  378. return hItem;
  379. }
  380. return NULL;
  381. }
  383. void CUITreeCtrl::Refresh()
  384. {
  385. }
  387. void CUITreeCtrl::DeleteKey(HTREEITEM hItem)
  388. {
  389. }
  391. void CUITreeCtrl::GoBack(HTREEITEM hItem)
  392. {
  393. }
  395. void CUITreeCtrl::ShowProperties(HTREEITEM hItem)
  396. {
  397. }
  399. void CUITreeCtrl::ShowPopupMenu(HTREEITEM hItem,CPoint point)
  400. {
  401. UINT MenuID = GetMenuID();
  402. if (MenuID == 0) 
  403. return;
  404. CWnd *pWndMess = GetMenuWnd();
  405. if (pWndMess == NULL)
  406. pWndMess = this;
  407. CMenu menu;
  408. VERIFY(menu.LoadMenu(MenuID));
  409. CMenu* pPopup = menu.GetSubMenu(0);
  410. ASSERT(pPopup != NULL);
  411. pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x, point.y, pWndMess);
  412. }
  414. BEGIN_MESSAGE_MAP(CUITreeCtrl, CUIDragDropTree)
  415. //{{AFX_MSG_MAP(CUITreeCtrl)
  416. ON_NOTIFY_REFLECT_EX(NM_DBLCLK, OnDblclk)
  417. ON_NOTIFY_REFLECT_EX(NM_RETURN, OnReturn)
  418. ON_NOTIFY_REFLECT_EX(NM_RCLICK, OnRclick)
  419. ON_NOTIFY_REFLECT_EX(NM_CUSTOMDRAW, OnCustomDraw)
  420. ON_NOTIFY_REFLECT_EX(TVN_ITEMEXPANDED, OnItemExpanded)
  421. ON_NOTIFY_REFLECT_EX(TVN_ITEMEXPANDING, OnItemExpanding)
  422. ON_NOTIFY_REFLECT_EX(TVN_SELCHANGED, OnSelchanged)
  423. ON_NOTIFY_REFLECT_EX(TVN_ENDLABELEDIT, OnEndLabelEdit)
  424. ON_NOTIFY_REFLECT_EX(TVN_DELETEITEM, OnDeleteItem)
  425. ON_NOTIFY_REFLECT(TVN_KEYDOWN, OnKeydownFolder)
  426. ON_WM_MOUSEWHEEL()
  427. ON_WM_TIMER()
  428. ON_WM_CREATE()
  429. //}}AFX_MSG_MAP
  430. ON_MESSAGE(WM_APP_ON_PROPERTIES_KEY,OnAppPropertiesKey)
  431. ON_MESSAGE(WM_APP_ON_DELETE_KEY,OnAppDeleteKey)
  432. ON_MESSAGE(WM_APP_ON_CONTEXT_MENU_KEY,OnAppContextMenuKey)
  433. ON_MESSAGE(WM_APP_ON_BACKSPACE_KEY,OnAppBackspaceKey)
  434. ON_MESSAGE(WM_APP_ON_EDIT_KEY,OnAppEditKey)
  435. ON_MESSAGE(WM_APP_ON_REFRESH_KEY,OnAppRefreshKey)
  437. ON_MESSAGE(TVM_GETITEM,OnTVMGetItem)
  438. ON_MESSAGE(TVM_SETITEM,OnTVMSetItem)
  439. ON_MESSAGE(TVM_INSERTITEM,OnTVMInsertItem)
  440. ON_MESSAGE(WM_APP_TIMER_SEL_CHANGE,OnTimerSelChange)
  441. ON_MESSAGE(WM_APP_OLE_DD_DODRAGDROP, OnDDDoDragDrop)
  442. ON_MESSAGE(WM_APP_OLE_DD_DROP, OnDragDrop)
  443. ON_MESSAGE(WM_APP_OLE_DD_OVER, OnDragOver)
  444. ON_MESSAGE(WM_APP_OLE_DD_ENTER, OnDragEnter)
  445. ON_MESSAGE(WM_APP_OLE_DD_LEAVE, OnDragLeave)
  446. END_MESSAGE_MAP()
  448. /////////////////////////////////////////////////////////////////////////////
  449. // CUITreeCtrl message handlers
  450. BOOL CUITreeCtrl::OnCustomDraw(NMHDR* pNMHDR, LRESULT* pResult) 
  451. {
  452. LPNMTVCUSTOMDRAW lptvcd = (LPNMTVCUSTOMDRAW)pNMHDR;
  453. if(lptvcd->nmcd.dwDrawStage == CDDS_PREPAINT)
  454. {
  455. *pResult = CDRF_NOTIFYITEMDRAW;
  456. }
  457. else if(lptvcd->nmcd.dwDrawStage == CDDS_ITEMPREPAINT)
  458. {
  459. ItemPrePaint(lptvcd,pResult);
  460. }
  461. else if(lptvcd->nmcd.dwDrawStage == CDDS_ITEMPOSTPAINT)
  462. {
  463. ItemPostPaint(lptvcd,pResult);
  464. }
  465. return TRUE;
  466. }
  468. BOOL CUITreeCtrl::OnRclick(NMHDR* pNMHDR, LRESULT* pResult) 
  469. {
  470. // TODO: Add your control notification handler code here
  471. CPoint point;
  472. HTREEITEM hItem = GetHitItem(&point);
  473. if (hItem == NULL) 
  474. return FALSE;
  475. ClientToScreen(&point);
  476. ShowPopupMenu(hItem,point);
  477. *pResult = 0;
  478. return TRUE;
  479. }
  481. BOOL CUITreeCtrl::OnDeleteItem(NMHDR* pNMHDR, LRESULT* pResult) 
  482. {
  483. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  484. // TODO: Add your control notification handler code here
  485. CUIListCtrlData* pData = (CUIListCtrlData*)pNMTreeView->itemOld.lParam;
  486. if (!pData->GetAutoDelete())
  487. OnDeleteItemData(pData->GetExtData());
  488. delete pData;
  489. return TRUE;
  490. }
  492. LRESULT CUITreeCtrl::OnTVMGetItem(WPARAM wParam,LPARAM lParam)
  493. {
  494. LPTVITEM pitem = (LPTVITEM)lParam;
  495. if (m_bMe == false && pitem->mask & TVIF_PARAM)
  496. {
  497. CUIListCtrlData* pData = GetListCtrlData(pitem->hItem);
  498. if (pData)
  499. {
  500. pitem->lParam = pData->GetExtData();
  501. pitem->mask &= ~TVIF_PARAM;
  502. }
  503. }
  504. return Default();
  505. }
  507. LRESULT CUITreeCtrl::OnTVMSetItem(WPARAM wParam,LPARAM lParam)
  508. {
  509. LPTVITEM pitem = (LPTVITEM)lParam;
  510. if (m_bMe == false && pitem->mask & TVIF_PARAM)
  511. {
  512. CUIListCtrlData* pData = GetListCtrlData(pitem->hItem);
  513. if (pData)
  514. {
  515. pData->SetExtData(pitem->lParam);
  516. pitem->mask &= ~TVIF_PARAM;
  517. }
  518. }
  519. return Default();
  520. }
  522. LRESULT CUITreeCtrl::OnTVMInsertItem(WPARAM wParam,LPARAM lParam)
  523. {
  524. LPTVINSERTSTRUCT lpis = (LPTVINSERTSTRUCT)lParam;
  525. if (m_bMe == false && lpis->item.mask & TVIF_PARAM)
  526. {
  527. lpis->item.lParam = (LPARAM)GetNewListCtrlData(lpis->item.lParam,lpis->hParent);
  528. }
  529. return Default();
  530. }
  532. int CUITreeCtrl::OnCreate(LPCREATESTRUCT lpCreateStruct) 
  533. {
  534. if (CUIDragDropTree::OnCreate(lpCreateStruct) == -1)
  535. return -1;
  537. Init();
  539. // TODO: Add your specialized creation code here
  540. return 0;
  541. }
  543. UINT CUITreeCtrl::GetIconID()
  544. {
  545. return IDI_FOLDER_CLOSED;
  546. }
  548. UINT CUITreeCtrl::GetOpenIconID()
  549. {
  550. return 0;
  551. }
  553. bool CUITreeCtrl::Expanding(NM_TREEVIEW *nmtvw)
  554. {
  555. return false;
  556. }
  558. bool CUITreeCtrl::Collapsing(NM_TREEVIEW *nmtvw)
  559. {
  560. return true;
  561. }
  563. void CUITreeCtrl::Init()
  564. {
  565. // Create the image list we will need
  566. int nCxSmall = GetSystemMetrics(SM_CXSMICON);
  567. int nCySmall = GetSystemMetrics(SM_CYSMICON);
  568. m_ImageList.Create(nCxSmall,
  569. nCySmall,
  570. TRUE, // list does not include masks
  571. 2,
  572. 0 ); // list won't grow
  573. /* UINT nIconID = GetIconID();
  574. if (nIconID)
  575. {
  576. HINSTANCE hInstIcon = AfxFindResourceHandle(MAKEINTRESOURCE(nIconID),RT_GROUP_ICON);
  577. HICON hIcon = (HICON)LoadImage( hInstIcon, MAKEINTRESOURCE(nIconID), IMAGE_ICON, 
  578. nCxSmall, nCySmall, 0 );
  579. m_idxClosed = m_ImageList.Add(hIcon);
  580. }
  581. UINT nOpenIconID = GetOpenIconID();
  582. if (nOpenIconID)
  583. {
  584. HINSTANCE hInstIcon = AfxFindResourceHandle(MAKEINTRESOURCE(nOpenIconID),RT_GROUP_ICON);
  585. HICON hOpenIcon = (HICON)LoadImage( hInstIcon, MAKEINTRESOURCE(nOpenIconID), IMAGE_ICON, 
  586. nCxSmall, nCySmall, 0 );
  587. m_idxOpen = m_ImageList.Add(hOpenIcon);
  588. }
  589. */
  590. // Associate the image list with the tree
  591. SetImageList(&m_ImageList,TVSIL_NORMAL);
  592. m_ImageList.SetBkColor(CLR_NONE);
  593. ModifyStyle(0,m_style);
  594. }
  596. BOOL CUITreeCtrl::PreCreateWindow(CREATESTRUCT& cs) 
  597. {
  598. // TODO: Add your specialized code here and/or call the base class
  599. cs.style |= m_style;
  600. return CUIDragDropTree::PreCreateWindow(cs);
  601. }
  603. BOOL CUITreeCtrl::OnEndLabelEdit(NMHDR* pNMHDR, LRESULT* pResult) 
  604. {
  605. LPNMTVDISPINFO pDispInfo = (LPNMTVDISPINFO)pNMHDR;
  606. if (pDispInfo->item.pszText)
  607. *pResult = EndLabelEdit(pDispInfo->item.hItem,pDispInfo->item.pszText);
  608. return TRUE;
  609. }
  611. BOOL CUITreeCtrl::OnDblclk(NMHDR* pNMHDR, LRESULT* pResult) 
  612. {
  613. // TODO: Add your control notification handler code here
  614. HTREEITEM hItem = GetSelectedItem();
  615. DoubleClick(hItem);
  616. return TRUE;
  617. }
  619. BOOL CUITreeCtrl::OnReturn(NMHDR* pNMHDR, LRESULT* pResult) 
  620. {
  621. // TODO: Add your control notification handler code here
  622. OnDblclk(pNMHDR,pResult);
  623. return TRUE;
  624. }
  626. BOOL CUITreeCtrl::OnItemExpanding(NMHDR* pNMHDR, LRESULT* pResult) 
  627. {
  628. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  629. CWaitCursor w;
  630. // TODO: Add your control notification handler code here
  631. if (pNMTreeView->action == TVE_EXPAND)
  632. {
  633. Expanding(pNMTreeView);
  634. }
  635. else if (pNMTreeView->action == TVE_COLLAPSE)
  636. {
  637. Collapsing(pNMTreeView);
  638. }
  639. return TRUE;
  640. }
  642. BOOL CUITreeCtrl::OnItemExpanded(NMHDR* pNMHDR, LRESULT* pResult) 
  643. {
  644. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  645. // TODO: Add your control notification handler code here
  646. /* if (pNMTreeView->action == TVE_EXPAND)
  647. {
  648. SetItemImage(pNMTreeView->itemNew.hItem,m_idxClosed,0);
  649. }
  650. else if (pNMTreeView->action == TVE_COLLAPSE)
  651. {
  652. SetItemImage(pNMTreeView->itemNew.hItem,m_idxOpen,0);
  653. }*/
  654. return TRUE;
  655. }
  657. BOOL CUITreeCtrl::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
  658. {
  659. //VK_PRIOR 21 page up key  
  660. //VK_NEXT 22 page down key  
  661. //VK_END 23 end key  
  662. //VK_HOME 24 home key  
  663. //VK_LEFT 25 left arrow key  
  664. //VK_UP 26 up arrow key  
  665. //VK_RIGHT 27 right arrow key  
  666. //VK_DOWN 28 down arrow key  
  668. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  669. // TODO: Add your control notification handler code here
  670. if ((pNMTreeView->itemNew.state & TVIS_SELECTED)) 
  671. {
  672. if (pNMTreeView->action == TVC_BYMOUSE || pNMTreeView->action == TVC_UNKNOWN)
  673. {
  674. CUIListCtrlData *pData = (CUIListCtrlData*)pNMTreeView->itemNew.lParam;
  675. if (pData)
  676. {
  677. ASSERT_KINDOF(CUIListCtrlData,pData);
  678. SelectionChanged(pNMTreeView->itemNew.hItem,pData->GetExtData());
  679. }
  680. else
  681. SelectionChanged(pNMTreeView->itemNew.hItem,pNMTreeView->itemNew.lParam);
  682. }
  683. else if (pNMTreeView->action == TVC_BYKEYBOARD)
  684. {
  685. CopyMemory(&m_NMTreeView,pNMTreeView,sizeof(NM_TREEVIEW));
  686. if (m_vkeydown && m_IDTimer == 0)
  687. {
  688. TRACE0("Timer setn");
  689. m_IDTimer = SetTimer(2,100,NULL);
  690. }
  691. }
  692. }
  693. *pResult = 0;
  694. return TRUE;
  695. }
  697. BOOL CUITreeCtrl::OnMouseWheel(UINT nFlags, short zDelta, CPoint pt) 
  698. {
  699. // TODO: Add your message handler code here and/or call default
  700. return CUIDragDropTree::OnMouseWheel(nFlags, zDelta, pt);
  701. }
  703. BOOL CUITreeCtrl::PreTranslateMessage(MSG* pMsg) 
  704. {
  705. // TODO: Add your specialized code here and/or call the base class
  706. if (pMsg->message == WM_KEYDOWN || pMsg->message == WM_SYSKEYDOWN)
  707. {
  708. if (pMsg->wParam == VK_F10 && (GetKeyState(VK_SHIFT) < 0))
  709. {
  710. PostMessage(WM_APP_ON_CONTEXT_MENU_KEY);
  711. return TRUE;
  712. }
  713. else if (pMsg->wParam == VK_BACK)
  714. {
  715. PostMessage(WM_APP_ON_BACKSPACE_KEY);
  716. return TRUE;
  717. }
  718. else if (pMsg->wParam == VK_F2)
  719. {
  720. PostMessage(WM_APP_ON_EDIT_KEY);
  721. return TRUE;
  722. }
  723. else if (pMsg->wParam == VK_F5)
  724. {
  725. PostMessage(WM_APP_ON_REFRESH_KEY);
  726. return TRUE;
  727. }
  728. else if (pMsg->wParam == VK_DELETE)
  729. {
  730. PostMessage(WM_APP_ON_DELETE_KEY);
  731. return TRUE;
  732. }
  733. else if (pMsg->message == WM_SYSKEYDOWN && pMsg->wParam == VK_RETURN)
  734. {
  735. PostMessage(WM_APP_ON_PROPERTIES_KEY);
  736. return TRUE;
  737. }
  738. }
  740. return CUIDragDropTree::PreTranslateMessage(pMsg);
  741. }
  743. void CUITreeCtrl::OnKeydownFolder(NMHDR* pNMHDR, LRESULT* pResult) 
  744. {
  745. TV_KEYDOWN* pTVKeyDown = (TV_KEYDOWN*)pNMHDR;
  746. // TODO: Add your control notification handler code here
  747. m_vkeydown = pTVKeyDown->wVKey;
  748. *pResult = 0;
  749. }
  751. LRESULT CUITreeCtrl::OnTimerSelChange(WPARAM wParam,LPARAM lParam)
  752. {
  753. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)&m_NMTreeView;
  754. CUIListCtrlData *pData = (CUIListCtrlData*)pNMTreeView->itemNew.lParam;
  755. if (pData)
  756. {
  757. ASSERT_KINDOF(CUIListCtrlData,pData);
  758. SelectionChanged(pNMTreeView->itemNew.hItem,pData->GetExtData());
  759. }
  760. else
  761. SelectionChanged(pNMTreeView->itemNew.hItem,pNMTreeView->itemNew.lParam);
  762. return 1L;
  763. }
  765. void CUITreeCtrl::OnTimer( UINT nIDEvent )
  766. {
  767. CUIDragDropTree::OnTimer(nIDEvent);
  768. if (nIDEvent == m_IDTimer)
  769. {
  770. if (m_vkeydown)
  771. {
  772. SHORT sRet = GetAsyncKeyState(m_vkeydown);
  773. if (sRet >= 0)
  774. {
  775. KillTimer(m_IDTimer);
  776. m_IDTimer = 0;
  777. m_vkeydown = 0;
  778. PostMessage(WM_APP_TIMER_SEL_CHANGE);
  779. }
  780. }
  781. }
  782. }
  784. LRESULT CUITreeCtrl::OnDragDrop(WPARAM wParam,LPARAM lParam)
  785. {
  786. if (GetDragDrop() == false)
  787. return 1L;
  788. // get the info we need
  789. CDD_OleDropTargetInfo *pInfo = (CDD_OleDropTargetInfo*)wParam;
  790. ASSERT(pInfo);
  791. if (pInfo->GetDataObject() != NULL)
  792. return DragDrop(pInfo) ? 1 : 0;
  793. else
  794. return 1;
  795. }
  797. // user is currently over the tree view window
  798. // return 1 if we are interested in the CB format
  799. LRESULT CUITreeCtrl::OnDragOver(WPARAM wParam,LPARAM lParam)
  800. {
  801. if (GetDragDrop() == false)
  802. return 1L;
  803. CDD_OleDropTargetInfo *pInfo = (CDD_OleDropTargetInfo*)wParam;
  804. ASSERT(pInfo);
  805. if (pInfo->GetDataObject() != NULL)
  806. return DragOver(pInfo) ? 1 : 0;
  807. else
  808. return 1;
  809. }
  811. LRESULT CUITreeCtrl::OnDragEnter(WPARAM wParam,LPARAM lParam)
  812. {
  813. if (GetDragDrop() == false)
  814. return 1L;
  815. CDD_OleDropTargetInfo *pInfo = (CDD_OleDropTargetInfo*)wParam;
  816. ASSERT(pInfo);
  817. if (pInfo->GetDataObject() != NULL)
  818. return DragEnter(pInfo) ? 1 : 0;
  819. else
  820. return 1;
  821. }
  823. LRESULT CUITreeCtrl::OnDragLeave(WPARAM wParam,LPARAM lParam)
  824. {
  825. if (GetDragDrop() == false)
  826. return 1L;
  827. CDD_OleDropTargetInfo *pInfo = (CDD_OleDropTargetInfo*)wParam;
  828. ASSERT(pInfo);
  829. if (pInfo->GetDataObject() != NULL)
  830. return DragLeave(pInfo) ? 1 : 0;
  831. else
  832. return 1;
  833. }
  835. // Drag and drop initiated
  836. // Return 1 if processed
  837. LRESULT CUITreeCtrl::OnDDDoDragDrop(WPARAM wParam,LPARAM lParam)
  838. {
  839. if (GetDragDrop() == false)
  840. return 1L;
  841. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)wParam;
  842. COleDataSource *pOleDataSource = (COleDataSource*)lParam;
  843. ASSERT(pOleDataSource);
  844. return DoDragDrop(pNMTreeView,pOleDataSource);
  845. }
  847. LRESULT CUITreeCtrl::OnAppPropertiesKey(WPARAM wParam, LPARAM lParam)
  848. {
  849. ShowProperties(GetSelectedItem());
  850. return 1L;
  851. }
  853. LRESULT CUITreeCtrl::OnAppDeleteKey(WPARAM wParam, LPARAM lParam)
  854. {
  855. DeleteItem(GetSelectedItem());
  856. return 1L;
  857. }
  859. LRESULT CUITreeCtrl::OnAppRefreshKey(WPARAM wParam, LPARAM lParam)
  860. {
  861. Refresh();
  862. return 1L;
  863. }
  865. LRESULT CUITreeCtrl::OnAppEditKey(WPARAM wParam, LPARAM lParam)
  866. {
  867. EditLabel(GetSelectedItem());
  868. return 1L;
  869. }
  871. LRESULT CUITreeCtrl::OnAppContextMenuKey(WPARAM wParam, LPARAM lParam)
  872. {
  873. CRect rc;
  874. GetItemRect(GetSelectedItem(),&rc,FALSE);
  875. ClientToScreen(&rc);
  876. ShowPopupMenu(GetSelectedItem(),rc.CenterPoint());
  877. return 1L;
  878. }
  880. LRESULT CUITreeCtrl::OnAppBackspaceKey(WPARAM wParam, LPARAM lParam)
  881. {
  882. GoBack(GetSelectedItem());
  883. return 1L;
  884. }