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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > ICQ/即时通讯 > 查看源码
MyTreeView.cpp
资源名称：QQ示例源码.rar　[点击查看]
上传用户：power_led
上传日期：2013-04-11
资源大小：373k
文件大小：14k
源码类别：

ICQ/即时通讯

开发平台：

Visual C++

MyTreeView.cpp：源码内容
  1. // MyTreeView.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "ChatClient.h"
  6. #include "MyTreeView.h"
  7. #include "ClientLog.h"
  8. #include "SendMessageDlg.h"
  9. #include "ClientFormView.h"
  10. #include "MainFrm.h"
  11. #include "ShowMessage.h"
  13. #ifdef _DEBUG
  14. #define new DEBUG_NEW
  15. #undef THIS_FILE
  16. static char THIS_FILE[] = __FILE__;
  17. #endif
  19. /////////////////////////////////////////////////////////////////////////////
  20. // CMyTreeView
  22. IMPLEMENT_DYNCREATE(CMyTreeView, CTreeView)
  24. CMyTreeView::CMyTreeView():m_Tree(GetTreeCtrl())
  25. {
  26. }
  28. CMyTreeView::~CMyTreeView()
  29. {
  30. POSITION pos;
  31. pos = m_ClientNameList.GetHeadPosition();
  32. POSITION tempos = pos;
  33. if(pos)
  34. {
  35. int iCount = m_ClientNameList.GetCount();
  36. for(int i = 0;i < iCount; i ++)
  37. {
  38. tempos = pos;
  39. Client * pTempClient = (Client *)m_ClientNameList.GetNext(pos);
  40. delete pTempClient;
  41. m_ClientNameList.RemoveAt(tempos);
  42. }
  43. }
  45. }
  48. BEGIN_MESSAGE_MAP(CMyTreeView, CTreeView)
  49. //{{AFX_MSG_MAP(CMyTreeView)
  50. ON_WM_CREATE()
  51. ON_NOTIFY_REFLECT(NM_RCLICK, OnRclick)
  52. ON_COMMAND(ID_JOININROOM, OnJoininroom)
  53. ON_COMMAND(ID_MENU_SEND, OnMenuSend)
  54. ON_WM_LBUTTONDBLCLK()
  55. ON_NOTIFY_REFLECT(NM_RDBLCLK, OnRclick)
  56. ON_NOTIFY_REFLECT(NM_DBLCLK, OnRclick)
  57. ON_WM_TIMER()
  58. //}}AFX_MSG_MAP
  59. ON_MESSAGE(WM_ADDNEWROOM,AddNewRoom)
  60. ON_MESSAGE(WM_NEWCLIENTCOMEIN,OnNewClientIn)
  61. ON_MESSAGE(WM_USERLOGOUT,OnClientOut)
  62. ON_MESSAGE(WM_SAYINPRIVATE,OnSayInPrivate)
  64. END_MESSAGE_MAP()
  66. /////////////////////////////////////////////////////////////////////////////
  67. // CMyTreeView drawing
  69. void CMyTreeView::OnDraw(CDC* pDC)
  70. {
  71. CDocument* pDoc = GetDocument();
  72. // TODO: add draw code here
  73. }
  75. /////////////////////////////////////////////////////////////////////////////
  76. // CMyTreeView diagnostics
  78. #ifdef _DEBUG
  79. void CMyTreeView::AssertValid() const
  80. {
  81. CTreeView::AssertValid();
  82. }
  84. void CMyTreeView::Dump(CDumpContext& dc) const
  85. {
  86. CTreeView::Dump(dc);
  87. }
  88. #endif //_DEBUG
  89. extern CChatClientApp theApp;
  90. void CMyTreeView::AddNewRoom(Message wParam,LPARAM lParam)
  91. {
  92. //CString
  94. CMainFrame * m_pMainFrame = (CMainFrame *)AfxGetMainWnd();
  95. ASSERT_VALID(m_pMainFrame);
  96. ASSERT(m_pMainFrame != NULL);
  97. static int iTimes = 1;
  98. if(iTimes == 1)
  99. {
  100. AfxGetMainWnd()->ShowWindow(SW_SHOWNORMAL);
  101. m_pMainFrame ->GetFormView()->SetDlgItemText(IDC_SELFNAME,theApp.m_skMainSocket.GetUserName());
  102. iTimes ++;
  103. }
  105. Message msObj = wParam;
  106. CString strRoom = msObj.strRoom;
  107. m_Tree.InsertItem(strRoom,2,3,m_ParentItem);
  108. // m_Tree.InsertItem("strRoom",2,3,m_ParentItem);
  109. m_Tree.Expand(m_ParentItem,TVE_EXPAND);
  111. }
  113. /////////////////////////////////////////////////////////////////////////////
  114. // CMyTreeView message handlers
  116. int CMyTreeView::OnCreate(LPCREATESTRUCT lpCreateStruct) 
  117. {
  118. if (CTreeView::OnCreate(lpCreateStruct) == -1)
  119. return -1;
  122. // TODO: Add your specialized creation code here
  123. m_ImageList.Create(IDB_IMAGELIST,16,5,RGB(255,0,255));
  124. m_Tree.SetImageList(&m_ImageList,TVSIL_NORMAL);
  125. m_ParentItem = m_Tree.InsertItem("聊天室列表",2,3);
  127. return 0;
  128. }
  129. extern CChatClientApp theApp;
  131. void CMyTreeView::OnRclick(NMHDR* pNMHDR, LRESULT* pResult) 
  132. {
  134. CMenu menu;
  136. CMenu * pContextMenu; 
  137. POINT point;
  138. GetCursorPos(&point);
  139. // CTreeView::OnLButtonDown(0,point);
  140. ScreenToClient(&point);
  142. HTREEITEM hItem = m_Tree.GetSelectedItem();
  143. CRect rect;
  144. m_Tree.GetItemRect(hItem,&rect,TRUE);
  146. if(rect.PtInRect(point))
  147. {
  148. if(m_Tree.GetParentItem(hItem) == m_ParentItem)
  149. {
  150. menu.LoadMenu(IDR_JOINROOM);
  151. pContextMenu = menu.GetSubMenu(0);
  152. ClientToScreen(&point);
  154. pContextMenu->TrackPopupMenu(TPM_LEFTBUTTON   ,point.x,point.y ,this);
  155. }
  156. else
  157. {
  158. if(hItem != m_ParentItem && m_Tree.GetItemText(hItem) != theApp.m_skMainSocket.GetUserName() && theApp.m_skMainSocket.GetState() == HAVELOGIN)
  159. {
  160. HTREEITEM hParent = m_Tree.GetParentItem(hItem);
  161. CString strRoom = m_Tree.GetItemText(hParent);
  162. if(strRoom == theApp.m_skMainSocket.GetRoomName())
  163. {
  164. menu.LoadMenu(IDR_CLIENT);
  165. pContextMenu = menu.GetSubMenu(0);
  166. ClientToScreen(&point);
  168. pContextMenu->TrackPopupMenu(TPM_LEFTBUTTON   ,point.x,point.y ,this);
  170. }
  171. else
  172. AfxMessageBox("处于不同的房间内，不能执行此项操作");
  173. }
  175. }
  176. }
  179. *pResult = 0;
  180. }
  181. extern CChatClientApp theApp;
  182. void CMyTreeView::OnJoininroom() 
  183. {
  184. // TODO: Add your command handler code here
  185. Message msObj;
  186. HTREEITEM hItem;
  188. if(theApp.m_skMainSocket.GetState() == NOTLOGIN)
  189. {
  190. //构造msobj
  193. hItem = m_Tree.GetSelectedItem();
  194. CString strRoomName = m_Tree.GetItemText(hItem);
  195. CString strUsrName = theApp.m_skMainSocket.GetUserName();
  196. // theApp.m_skMainSocket.SetUserName(strUsrName);
  197. theApp.m_skMainSocket.SetRoomName(strRoomName);
  198. theApp.m_skMainSocket.SetState(HAVELOGIN);
  201. msObj.iType = USERLOG;
  202. msObj.iSubType = USERLOGIN;
  203. int iLen = strRoomName.GetLength();
  204. iLen > 20 ? 20: iLen;
  205. lstrcpy(msObj.strRoom,_T(strRoomName.GetBuffer(iLen)));
  206. strRoomName.ReleaseBuffer();
  208. iLen = strUsrName.GetLength();
  209. iLen > 20 ? 20 : iLen;
  210. lstrcpy(msObj.strName,_T(strUsrName.GetBuffer(iLen)));
  211. strUsrName.ReleaseBuffer();
  213. iLen = theApp.m_skMainSocket.Send(&msObj,sizeof(msObj));
  214. if(iLen < sizeof(msObj))
  215. {
  216. AfxMessageBox("发送失败");
  217. return;
  218. }
  219. }
  220. else
  221. if(theApp.m_skMainSocket.GetState() == HAVELOGIN)
  222. {
  223. HTREEITEM hItem = m_Tree.GetSelectedItem();
  224. CString strRoomName = m_Tree.GetItemText(hItem);
  225. if(strRoomName != theApp.m_skMainSocket.GetRoomName())
  226. {
  228. HTREEITEM hItem = m_Tree.GetChildItem(m_ParentItem);
  230. msObj.iType = USERSESSION;
  231. msObj.iSubType = CHANGEROOM;
  232. //用户名
  233. CString strTemp;
  234. strTemp =  theApp.m_skMainSocket.GetUserName();
  235. int iLen = strTemp.GetLength();
  236. iLen > 20 ? 20:iLen;
  237. lstrcpy(msObj.strName,_T(strTemp.GetBuffer(iLen)));
  238. strTemp.ReleaseBuffer();
  239. //旧房件名
  240. strTemp = theApp.m_skMainSocket.GetRoomName();
  241. iLen = strTemp.GetLength();
  242. iLen > 20 ? 20 : iLen;
  243. lstrcpy(msObj.strRoom,_T(strTemp.GetBuffer(iLen)));
  244. strTemp.ReleaseBuffer();
  245. //新房间名
  246. hItem = m_Tree.GetSelectedItem();
  247. strTemp = m_Tree.GetItemText(hItem);
  248. iLen = strTemp.GetLength();
  249. iLen > 20 ? 20 :iLen;
  250. lstrcpy(msObj.strContent,_T(strTemp.GetBuffer(iLen)));
  251. strTemp.ReleaseBuffer();
  253. if(theApp.m_skMainSocket.Send(&msObj,sizeof(msObj)) != sizeof(msObj))
  254. {
  255. AfxMessageBox("发送错误");
  256. }
  257. theApp.m_skMainSocket.SetRoomName(m_Tree.GetItemText(hItem));
  258. }
  259. }
  260. return;
  262. }
  263. void CMyTreeView::OnClientOut(Message wParam, LPARAM lParam)
  264. {
  265. Message msObj = wParam;
  266. CString strRoom = msObj.strRoom;
  267. CString strUsrName = msObj.strName;
  268. HTREEITEM hItem = m_Tree.GetChildItem(m_ParentItem);
  269. BOOL bFind = FALSE;
  270. while(hItem && !bFind)
  271. {
  272. CString strNodeName = m_Tree.GetItemText(hItem);
  273. if(strNodeName == strRoom)
  274. {
  275. //查找用户名的那一项
  276. HTREEITEM hChildItem = m_Tree.GetChildItem(hItem);
  277. while(hChildItem)
  278. {
  279. CString strUserNode = m_Tree.GetItemText(hChildItem);
  280. if(strUserNode == strUsrName)
  281. {
  282. m_Tree.DeleteItem(hChildItem);
  283. m_Tree.Expand(hItem,TVE_EXPAND);
  284. bFind = TRUE;
  285. break;
  286. }
  287. hChildItem = m_Tree.GetNextItem(hChildItem,1);
  288. }
  291. }
  292. hItem = m_Tree.GetNextItem(hItem,1);
  293. }
  294. }
  295. void CMyTreeView::OnNewClientIn(Message wParam, LPARAM lParam)
  296. {
  297. Message msObj = wParam;
  298. CString strRoom = msObj.strRoom;
  299. CString strUsrName = msObj.strName;
  300. //自己加入的消息传入进来
  301. if(strRoom == theApp.m_skMainSocket.GetRoomName() && strUsrName == theApp.m_skMainSocket.GetUserName())
  302. {
  303. //改变FormView中的数据
  304. CClientFormView * m_pFormView =(CClientFormView *) ((CMainFrame * )AfxGetMainWnd())->GetFormView();
  305. ASSERT(m_pFormView != NULL);
  306. ASSERT_VALID(m_pFormView);
  307. m_pFormView ->SetDlgItemText(IDC_SELFNAME,strUsrName);
  308. m_pFormView ->SetDlgItemText(IDC_ROOMNAME,strRoom);
  309. //改变任务栏图标的提醒信息
  310. lstrcpy(theApp.TaskIcon.szTip,_T(strUsrName.GetBuffer(strUsrName.GetLength())));
  311. Shell_NotifyIcon(NIM_MODIFY,&theApp.TaskIcon);
  312. //显示正常的窗口
  315. return;
  316. }
  318. HTREEITEM hItem = m_Tree.GetChildItem(m_ParentItem);
  319. while(hItem)
  320. {
  321. CString strNodeName = m_Tree.GetItemText(hItem);
  322. CString strTemp = theApp.m_skMainSocket.GetUserName();
  323. if(strNodeName == strRoom)
  324. {
  325. if(strUsrName != strTemp)
  326. {
  327. m_Tree.InsertItem(strUsrName,0,1,hItem);
  328. m_Tree.Expand(hItem,TVE_EXPAND);
  329. break;
  330. }
  331. }
  332. hItem = m_Tree.GetNextItem(hItem,1);
  333. }
  336. }
  338. void CMyTreeView::OnMenuSend() 
  339. {
  340. // TODO: Add your command handler code here
  341. CSendMessageDlg Dlg;
  343. if(Dlg.DoModal() == IDOK)
  344. {
  345. if(Dlg.m_strSentence.GetLength())
  346. {
  347. if(Dlg.m_strSentence.GetLength == 0)
  348. return;
  349. Message msObj;
  350. memset(&msObj,0,sizeof(Message));
  351. msObj.iType = USERSESSION;
  352. msObj.iSubType = SAYINPRIVATE;
  353. CString strTemp  = theApp.m_skMainSocket.GetRoomName();
  354. int iLen = strTemp.GetLength();
  355. lstrcpy(msObj.strRoom,_T(strTemp.GetBuffer(iLen)));
  356. strTemp.ReleaseBuffer();
  357. strTemp = Dlg.m_strSentence;
  358. iLen = strTemp.GetLength();
  359. iLen > 1024 ? 1024 : iLen;
  360. lstrcpy(msObj.strContent,_T(strTemp.GetBuffer(iLen)));
  361. strTemp.ReleaseBuffer();
  362. //自己的名字
  363. strTemp = theApp.m_skMainSocket.GetUserName();
  364. iLen = strTemp.GetLength();
  365. iLen > 20 ? 20:iLen;
  366. lstrcpy(msObj.strName,_T(strTemp.GetBuffer(iLen)));
  367. strTemp.ReleaseBuffer();
  368. //说话对象的名字
  369. HTREEITEM hItem = m_Tree.GetSelectedItem();
  370. strTemp = m_Tree.GetItemText(hItem);
  371. iLen = strTemp.GetLength();
  372. iLen > 20 ? 20 : iLen;
  373. lstrcpy(msObj.strClientName,_T(strTemp.GetBuffer(iLen)));
  374. strTemp.ReleaseBuffer();
  376.  iLen = theApp.m_skMainSocket.Send(&msObj,sizeof(Message));
  379. }
  381. }
  383. return;
  384. }
  385. //接收到信息
  386. //整个的处理过程是这样：
  387. //首先，接收到信息以后，在m_ClientNameList中添加一项，这样，在ontimer中，可以检查链表中都是有谁在发送信息，而且
  388. //你还没有回复.这样OnTimer中就可以实现多个人的头像一起晃动。
  389. //当你回复或者取消以后，相应的应该在m_ClientNameList中删除已经恢复或者取消的那一项。
  390. void CMyTreeView::OnSayInPrivate(Message wParam,LPARAM lParam)
  391. {
  392. Message msObj = wParam;
  394. CString strClientName = msObj.strName;
  395. Client * pNewClient;
  396. //设置每一个Client对应的ontimer中IDEVENT
  397. pNewClient = new Client();
  398. int iCount = m_ClientNameList.GetCount();
  399. pNewClient ->SetUserName(strClientName);
  400. int iMaxId = 1;
  401. if(iCount == 0)
  402. {
  403. pNewClient ->SetTimerId(iMaxId);
  405. }
  406. else
  407. {
  408. //查找出最大的iTimerId;
  410. POSITION pos;
  411. pos = m_ClientNameList.GetHeadPosition();
  412. if(pos)
  413. {
  414. for(int i = 0; i< iCount; i ++)
  415. {
  416. Client * pTempClient = (Client *)m_ClientNameList.GetNext(pos);
  417. if(iMaxId > pTempClient ->GetTimerId())
  418. iMaxId = pTempClient ->GetTimerId();
  420. }
  422. iMaxId ++;
  423. }
  425. pNewClient ->SetTimerId(iMaxId);
  427. }
  428. m_ClientNameList.AddTail(pNewClient);
  429. //启动Ontimer
  430. SetTimer(iMaxId,800,NULL);
  431. //显示对话框
  432. CString strContent  = msObj.strContent;
  433. CShowMessage Dlg(strContent,strClientName,iMaxId);
  434. Dlg.DoModal();
  435. SetIcon(strClientName,0,1);
  436. ////
  438. return;
  439. }
  441. BOOL CMyTreeView::PreCreateWindow(CREATESTRUCT& cs) 
  442. {
  443. // TODO: Add your specialized code here and/or call the base class
  444. cs.style |= TVS_HASLINES;
  445. cs.style |= TVS_LINESATROOT;
  446. cs.style |= TVS_HASBUTTONS;
  447. return CTreeView::PreCreateWindow(cs);
  448. }
  450. //启动ONtimer。主要是让用户的图标晃动，并且发出报警的声音
  451. void CMyTreeView::OnTimer(UINT nIDEvent) 
  452. {
  453. // TODO: Add your message handler code here and/or call default
  454. POSITION pos;
  455. pos = m_ClientNameList.GetHeadPosition();
  456. int iCount = m_ClientNameList.GetCount();
  458. if(pos)
  459. {
  462. for(int i = 0 ;i < iCount; i++)
  463. {
  464. Client * pTempClient =(Client *) m_ClientNameList.GetNext(pos);
  465. if(nIDEvent ==(UINT) pTempClient->GetTimerId())
  466. {
  467. int i = pTempClient ->GetTimes();
  468. pTempClient->AddTimes();
  469. i %= 2;
  470. if(i == 1)
  471. {
  472. theApp.TaskIcon.hIcon = theApp.LoadIcon(IDI_EMPTY);
  473. Shell_NotifyIcon(NIM_MODIFY,&theApp.TaskIcon);
  474. PlaySound("msg.wav",NULL,SND_ASYNC);
  475. SetIcon(pTempClient ->GetUserName(),0,0);
  476. }
  477. else
  478. {
  479. theApp.TaskIcon.hIcon = theApp.LoadIcon(IDI_TASKICON);
  480. Shell_NotifyIcon(NIM_MODIFY,&theApp.TaskIcon);
  482. SetIcon(pTempClient ->GetUserName(),5,5);
  483. }
  485. }//end if IDEvent == 
  487. }//endfor
  489. }//end if pos
  490. CTreeView::OnTimer(nIDEvent);
  491. }
  492. //在创建的链表中删除一项
  493. void CMyTreeView::DeleteOneClient(const CString & strName,const int iID)
  494. {
  495. POSITION pos;
  496. pos = m_ClientNameList.GetHeadPosition();
  497. int iCount = m_ClientNameList.GetCount();
  498. POSITION tempos = pos;
  499. if(pos && iCount)
  500. {
  502. for(int i = 0;i < iCount; i ++)
  503. {
  504. tempos = pos;
  505. Client * pTempClient = (Client *)m_ClientNameList.GetNext(pos);
  506. if(pTempClient->GetUserName() == strName && pTempClient->GetTimerId() == iID)
  507. {
  508. delete pTempClient;
  509. m_ClientNameList.RemoveAt(tempos);
  510. break;
  511. }
  512. }
  513. }
  518. return;
  519. }
  520. //设置树型控件中某一项的图标
  521. void CMyTreeView::SetIcon(const CString &strClientName,const int iNormalIcon,const int iSelectedIcon)
  522. {
  524. CString strRoom;
  525. strRoom = theApp.m_skMainSocket.GetRoomName();
  526. HTREEITEM hItem = m_Tree.GetChildItem(m_ParentItem);
  527. BOOL bFind = FALSE;
  528. while(hItem && !bFind)
  529. {
  530. CString strNodeName = m_Tree.GetItemText(hItem);
  531. if(strNodeName == strRoom)
  532. {
  533. //查找用户名的那一项
  534. HTREEITEM hChildItem = m_Tree.GetChildItem(hItem);
  535. while(hChildItem)
  536. {
  537. CString strUserNode = m_Tree.GetItemText(hChildItem);
  538. if(strUserNode == strClientName)
  539. {
  540. m_Tree.SetItemImage(hChildItem,iNormalIcon,iSelectedIcon);
  542. }
  543. hChildItem = m_Tree.GetNextItem(hChildItem,1);
  544. }
  547. }
  548. hItem = m_Tree.GetNextItem(hItem,1);
  549. }
  550. }