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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
MonitorWnd.cpp
资源名称：GGBT.rar　[点击查看]
上传用户：lds876
上传日期：2013-05-25
资源大小：567k
文件大小：11k
源码类别：

P2P编程

开发平台：

Visual C++

MonitorWnd.cpp：源码内容
  1. // MonitorWnd.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "testbt.h"
  6. #include "MonitorWnd.h"
  9. #include "download.h"
  10. #include "FileDB.h"
  11. #include "FileTorrent.h"
  12. #include "MainFrm.h"
  14. #ifdef _DEBUG
  15. #define new DEBUG_NEW
  16. #undef THIS_FILE
  17. static char THIS_FILE[] = __FILE__;
  18. #endif
  20. const CString CMonitorWnd::m_strSection = "Gui";
  21. const CString CMonitorWnd::m_strStateKey = "splitter";
  23. /////////////////////////////////////////////////////////////////////////////
  24. // CMonitorWnd
  26. CMonitorWnd::CMonitorWnd()
  27. {
  28. m_pFileDBItem = 0;
  29. m_pSetupRegister = 0;
  30. m_bDeleteing = false;
  32. m_lSplitPercent = 200; // (n%100)
  33. }
  35. CMonitorWnd::~CMonitorWnd()
  36. {
  37. }
  40. BEGIN_MESSAGE_MAP(CMonitorWnd, CWnd)
  41. //{{AFX_MSG_MAP(CMonitorWnd)
  42. ON_WM_CREATE()
  43. ON_WM_SIZE()
  44. ON_WM_DESTROY()
  45. //}}AFX_MSG_MAP
  46. ON_NOTIFY(TVN_SELCHANGED, 10568, OnNodeSelect)
  47. ON_MESSAGE(WM_SPLITTER_MOVED, OnSplitterMoved)
  48. END_MESSAGE_MAP()
  51. /////////////////////////////////////////////////////////////////////////////
  52. // CMonitorWnd message handlers
  54. void CMonitorWnd::LoadState()
  55. {
  56. m_lSplitPercent = AfxGetApp()->GetProfileInt(m_strSection, m_strStateKey, m_lSplitPercent);
  57. }
  59. void CMonitorWnd::SaveState()
  60. {
  61. AfxGetApp()->WriteProfileInt(m_strSection, m_strStateKey, m_lSplitPercent);
  62. }
  64. void CMonitorWnd::SetCateGoryBarWnd(CCategoryBarWnd* pCategoryBarWnd)
  65. {
  66. m_MonitorDetailWnd.SetCateGoryBarWnd(pCategoryBarWnd);
  67. }
  70. BOOL CMonitorWnd::CreateEx(CWnd *pParent, CRect &rc, long lParam, CSetupRegister* pSetupRegister)
  71. {
  72. m_pSetupRegister = pSetupRegister;
  73. return Create(0, 0, WS_VISIBLE|WS_CHILD|WS_BORDER, rc, pParent, 21321);
  74. }
  77. int CMonitorWnd::OnCreate(LPCREATESTRUCT lpCreateStruct) 
  78. {
  79. if (CWnd::OnCreate(lpCreateStruct) == -1)
  80. return -1;
  83. // Create the MonitorDetailWnd.
  84. m_MonitorDetailWnd.CreateEx(this, CRect(0, 0, 100, 100), 32645, 0, m_pSetupRegister);
  86. // Create the treeCategory.
  87. m_treeCategory.Create(WS_VISIBLE|WS_CHILD|WS_BORDER|TVS_HASBUTTONS | TVS_LINESATROOT | TVS_HASLINES|TVS_SHOWSELALWAYS,
  88. CRect(0, 0, 100, 100), this, 10568);
  92. //
  93. // Create BitmapList
  94. //
  95. // m_ctlImage.Create(IDB_IMAGELIST,16,0,RGB(255,0,255));
  96. // m_ctlImage.SetBkColor(GetSysColor(COLOR_WINDOW));
  98. HBITMAP hbm = (HBITMAP)::LoadImage(AfxGetInstanceHandle(),
  99. MAKEINTRESOURCE(IDB_IMAGELIST),
  100. IMAGE_BITMAP,
  101. 0,0, // cx,cy
  102. LR_CREATEDIBSECTION); // | LR_LOADMAP3DCOLORS );
  103. CBitmap bm;
  104. bm.Attach(hbm);
  105. m_ctlImage.Create(16,15, ILC_COLOR8|ILC_MASK, 4, 4);
  106. m_ctlImage.SetBkColor(::GetSysColor(COLOR_WINDOW)); 
  107. m_ctlImage.Add(&bm, (COLORREF)RGB(255,0,255));
  108. m_treeCategory.SetImageList(&m_ctlImage);
  111. // add cursor.
  112. CTreeCursor curRoot = m_treeCategory.GetRootItem();
  113. m_curInfomation = curRoot.AddTail("详细信息", IID_INFOMATION);
  114. m_curSubFiles = curRoot.AddTail("包含文件", IID_SUBFILES);
  116. // m_curStatics = curRoot.AddTail("运行信息", IID_RUN_STATICS);
  117. m_curAllPeers = curRoot.AddTail("Peers", IID_ALL_PEERS);
  118. m_curGraph = curRoot.AddTail("图形", IID_RUN_GRAPH);
  119. m_curLog = curRoot.AddTail("日志", IID_LOG);
  121. // init
  122. m_MonitorDetailWnd.ShowNode(m_curInfomation, false);
  125. // Create splitter.
  126. CRect rect(0,0,0,0);
  127. m_wndHSplitterBar.Create(WS_CHILD|WS_BORDER|WS_DLGFRAME|WS_VISIBLE,rect,this, 112998, FALSE);
  128. m_wndHSplitterBar.SetPanes(&m_treeCategory, &m_MonitorDetailWnd);
  130. LoadState();
  132. return 0;
  133. }
  135. void CMonitorWnd::OnSize(UINT nType, int cx, int cy) 
  136. {
  137. long lTreeCX = min(cx, m_lSplitPercent); // cx*m_lSplitPercent/100;
  138. long lMonitorCX = max((cx - lTreeCX), 0); //cx*(100 - m_lSplitPercent)/100;
  140. m_treeCategory.MoveWindow(CRect(0, 0, lTreeCX, cy));
  141. m_MonitorDetailWnd.MoveWindow(CRect(lMonitorCX, 0, cx, cy));
  143. m_wndHSplitterBar.SetPanes(&m_treeCategory,&m_MonitorDetailWnd);
  144. CWnd::OnSize(nType, cx, cy);
  145. }
  147. void CMonitorWnd::OnSplitterMoved(long lParam, long wParam)
  148. {
  149. CRect rc;
  150. m_treeCategory.GetClientRect(rc);
  151. long lTreeCX = rc.Width();
  152. m_lSplitPercent = lTreeCX;
  153. }
  155. void CMonitorWnd::OnNodeSelect(NMHDR *pNotifyStruct,LRESULT *result)
  156. {
  157. if (m_bDeleteing)
  158. return;
  159. CTreeCursor curSelected  = m_treeCategory.GetSelectedItem();
  160. if (!curSelected) return;
  162. m_MonitorDetailWnd.ShowNode(curSelected, true);
  163. }
  166. void CMonitorWnd::OnDestroy() 
  167. {
  168. m_bDeleteing = true;
  169. while (m_curAllPeers.HasChildren())
  170. {
  171. CTreeCursor curNext;
  172. CTreeCursor curChild = m_curAllPeers.GetChild();
  173. while (curChild)
  174. {
  176. CAddrPort* pData = (CAddrPort*)curChild.GetData();
  177. assert(pData);
  178. delete pData;
  180. curNext = curChild.GetNext(TVGN_NEXT);
  181. curChild.Delete();
  182. curChild = curNext;
  183. }
  184. }
  185. m_bDeleteing = false;
  187. SaveState();
  189. CWnd::OnDestroy();
  190. }
  192. /////////////////////////////////////////////////////////////////////////////
  193. // CMonitorWnd function
  195. void CMonitorWnd::Refresh()
  196. {
  197. if (!m_pFileDBItem || !m_pFileDBItem->m_pDownload)
  198. return;
  200. // refresh peers in tree.
  201. // RefreshPeers();
  203. // show the infomation for each category.
  204. CTreeCursor curSelected = m_treeCategory.GetSelectedItem();
  205. if (!curSelected)
  206. curSelected = m_curInfomation;
  208. m_MonitorDetailWnd.ShowNode(curSelected, false);
  209. }
  212. void CMonitorWnd::OnContextQuery(CMenu& menubar)
  213. {
  214. menubar.AppendMenu(MF_STRING, ID_MENUITEM_MONITOR_INFORMATION, "详细信息");
  215. menubar.AppendMenu(MF_STRING, ID_MENUITEM_MONITOR_SUBFILES, "包含文件");
  216. menubar.AppendMenu(MF_STRING, ID_MENUITEM_MONITOR_PEERS, "Peers");
  217. menubar.AppendMenu(MF_STRING, ID_MENUITEM_MONITOR_RUNSTATICS, "图形");
  218. menubar.AppendMenu(MF_STRING, ID_MENUITEM_MONITOR_LOGS, "日志");
  220. menubar.EnableMenuItem(ID_MENUITEM_MONITOR_INFORMATION, MF_BYCOMMAND | MF_ENABLED);
  221. menubar.EnableMenuItem(ID_MENUITEM_MONITOR_SUBFILES, MF_BYCOMMAND | MF_ENABLED);
  222. menubar.EnableMenuItem(ID_MENUITEM_MONITOR_RUNSTATICS, MF_BYCOMMAND | MF_ENABLED);
  223. menubar.EnableMenuItem(ID_MENUITEM_MONITOR_PEERS, MF_BYCOMMAND | MF_ENABLED);
  224. menubar.EnableMenuItem(ID_MENUITEM_MONITOR_LOGS, MF_BYCOMMAND | MF_ENABLED);
  226. }
  228. void CMonitorWnd::OnShowNode(UINT uID)
  229. {
  230. long lShowNode = uID - ID_MENUITEM_MONITOR_INFORMATION;
  233. CTreeCursor curSelected = m_treeCategory.GetSelectedItem();
  234. // if called by menu monitor command, show the required node.
  235. switch (lShowNode)
  236. {
  237. case 0:
  238. curSelected = m_curInfomation;
  239. break;
  240. case 1:
  241. curSelected = m_curSubFiles;
  242. break;
  243. case 2:
  244. curSelected = m_curAllPeers;
  245. break;
  246. case 3:
  247. curSelected = m_curGraph;
  248. break;
  249. case 4:
  250. curSelected = m_curLog;
  251. break;
  252. }
  254. m_bDeleteing = true;
  255. curSelected.Select();
  256. m_MonitorDetailWnd.ShowNode(curSelected, true);
  257. m_bDeleteing = false;
  258. }
  260. void CMonitorWnd::Monitor(CFileDBItem* pFileDBItem, long lShowNode)
  261. {
  262. m_pFileDBItem = pFileDBItem;
  264. m_MonitorDetailWnd.Monitor(pFileDBItem);
  266. // show peers in tree.
  267. // ShowPeers();
  270. // show the infomation for node category.
  271. m_bDeleteing = true;
  272. CTreeCursor curSelected = m_treeCategory.GetSelectedItem();
  273. if (!curSelected)
  274. curSelected = m_curInfomation;
  276. if (lShowNode != -1)
  277. {
  278. // if not visible monitorbar, show it.
  279. CMainFrame* pMainFrame = (CMainFrame*)AfxGetMainWnd();
  280. if (!pMainFrame)
  281. {
  282. assert(false);
  283. return;
  284. }
  285. pMainFrame->ShowControlBar(&pMainFrame->m_MonitorBar, true, false);
  287. // if called by menu monitor command, show the required node.
  288. switch (lShowNode)
  289. {
  290. case 0:
  291. curSelected = m_curInfomation;
  292. break;
  293. case 1:
  294. curSelected = m_curSubFiles;
  295. break;
  296. case 2:
  297. curSelected = m_curAllPeers;
  298. break;
  299. case 3:
  300. curSelected = m_curGraph;
  301. break;
  302. case 4:
  303. curSelected = m_curLog;
  304. break;
  305. }
  306. }
  308. curSelected.Select();
  309. // m_curInfomation.Expand();
  310. // m_curStatics.Expand();
  311. m_MonitorDetailWnd.ShowNode(curSelected, true);
  313. m_bDeleteing = false;
  314. }
  316. void CMonitorWnd::ShowPeers()
  317. {
  318. // clear
  319. m_bDeleteing = true;
  320. if (m_curAllPeers.HasChildren())
  321. {
  322. CTreeCursor curNext;
  323. CTreeCursor curChild = m_curAllPeers.GetChild();
  324. while (curChild)
  325. {
  326. CAddrPort* pData = (CAddrPort*)curChild.GetData();
  327. if (pData)
  328. delete pData;
  329. else
  330. assert(false);
  332. curNext = curChild.GetNext(TVGN_NEXT);
  333. curChild.Delete();
  334. curChild = curNext;
  335. }
  336. }
  337. m_bDeleteing = false;
  340. // show peers.
  341. if (!m_pFileDBItem || !m_pFileDBItem->m_pDownload || !m_pFileDBItem->m_pDownload->IsDownloading()) return;
  343. vector<CConncetionData> v = m_pFileDBItem->m_pDownload->GetConnections();
  344. for (int i=0; i<v.size(); i++)
  345. {
  346. in_addr inaddress;
  347. inaddress.S_un.S_addr = v[i].m_lAddr;
  349. char szText[100] = {0};
  350. sprintf(szText, "%s:%d(%d)", 
  351. inet_ntoa(inaddress), 
  352. ntohs(v[i].m_sPort), 
  353. v[i].m_lDownRate); 
  355. CTreeCursor curItem = m_curAllPeers.AddTail(szText, IID_PEER);
  357. CAddrPort* pData = new CAddrPort(v[i].m_lAddr, v[i].m_sPort);
  358. curItem.SetData((long)pData);
  359. }
  361. m_curAllPeers.Expand();
  362. }
  364. void CMonitorWnd::RefreshPeers()
  365. {
  366. if (!m_pFileDBItem || !m_pFileDBItem->m_pDownload || !m_pFileDBItem->m_pDownload->IsDownloading())
  367. return;
  369. m_bDeleteing = true;
  370. vector<CConncetionData> vConnections = m_pFileDBItem->m_pDownload->GetConnections();
  372. if (m_curAllPeers.HasChildren())
  373. {
  374. CTreeCursor curNext;
  375. CTreeCursor curChild = m_curAllPeers.GetChild();
  376. while (curChild)
  377. {
  378. CAddrPort* pData = (CAddrPort*)curChild.GetData();
  379. if (!pData)
  380. assert(false);
  382. bool bFind = false;
  383. for (int i=0; i<vConnections.size(); i++)
  384. {
  385. if (vConnections[i].m_lAddr == pData->m_lAddr && 
  386. vConnections[i].m_sPort == pData->m_sPort)
  387. {
  388. vConnections.erase(vConnections.begin() + i);
  389. bFind = true;
  390. break;
  391. }
  392. }
  394. curNext = curChild.GetNext(TVGN_NEXT);
  396. if (!bFind)
  397. {
  398. delete pData;
  399. curChild.Delete();
  400. }
  402. curChild = curNext;
  403. }
  404. }
  406. for (int i=0; i<vConnections.size(); i++)
  407. {
  408. in_addr inaddress;
  409. inaddress.S_un.S_addr = vConnections[i].m_lAddr;
  411. char szText[100] = {0};
  412. sprintf(szText, "%s:%d(%d)", 
  413. inet_ntoa(inaddress), 
  414. ntohs(vConnections[i].m_sPort), 
  415. vConnections[i].m_lDownRate); 
  417. CTreeCursor curItem = m_curAllPeers.AddTail(szText, IID_PEER);
  419. CAddrPort* pData = new CAddrPort(vConnections[i].m_lAddr, vConnections[i].m_sPort);
  420. curItem.SetData((long)pData);
  421. }
  423. m_bDeleteing = false;
  424. }
  426. void CMonitorWnd::SetFocusCategory()
  427. {
  428. m_treeCategory.SetFocus();
  429. }
  431. void CMonitorWnd::SwitchMonitorCategory(bool bNext)
  432. {
  433. CTreeCursor curSelected  = m_treeCategory.GetSelectedItem();
  434. if (!curSelected) 
  435. {
  436. m_curInfomation.Select();
  437. return;
  438. }
  440. if (curSelected == m_curInfomation)
  441. {
  442. if (bNext)
  443. curSelected = m_curSubFiles;
  444. else
  445. curSelected = m_curLog;
  446. }
  447. else if (curSelected == m_curSubFiles) 
  448. {
  449. if (bNext)
  450. curSelected = m_curAllPeers;
  451. else
  452. curSelected = m_curInfomation;
  453. }
  454. else if (curSelected == m_curAllPeers) 
  455. {
  456. if (bNext)
  457. curSelected = m_curGraph;
  458. else
  459. curSelected = m_curSubFiles;
  460. }
  461. else if (curSelected == m_curGraph) 
  462. {
  463. if (bNext)
  464. curSelected = m_curLog;
  465. else
  466. curSelected = m_curAllPeers;
  467. }
  468. else if (curSelected == m_curLog) 
  469. {
  470. if (bNext)
  471. curSelected = m_curInfomation;
  472. else
  473. curSelected = m_curGraph;
  474. }
  476. curSelected.Select();
  477. }