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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 行业应用 > 查看源码
LeftTreeView.cpp
资源名称：xiaoquzhuhuguanglixitong.rar　[点击查看]
上传用户：yuxuan88
上传日期：2022-05-09
资源大小：2290k
文件大小：12k
源码类别：

行业应用

开发平台：

Visual C++

LeftTreeView.cpp：源码内容
  1. // LeftTreeView.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Inhabitants.h"
  6. #include "InhabitantsDoc.h"
  7. #include "LeftTreeView.h"
  8. #include "MainFrm.h"
  9. #include "UsersListView.h"
  10. #include "UserinfoView.h"
  14. #ifdef _DEBUG
  15. #define new DEBUG_NEW
  16. #undef THIS_FILE
  17. static char THIS_FILE[] = __FILE__;
  18. #endif
  20. /////////////////////////////////////////////////////////////////////////////
  21. // CLeftTreeView
  23. IMPLEMENT_DYNCREATE(CLeftTreeView, CTreeView)
  25. CLeftTreeView::CLeftTreeView()
  26. {
  28. }
  30. CLeftTreeView::~CLeftTreeView()
  31. {
  33. }
  36. BEGIN_MESSAGE_MAP(CLeftTreeView, CTreeView)
  37. //{{AFX_MSG_MAP(CLeftTreeView)
  38. ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
  39. ON_COMMAND(ID_OPERATE_DELETE, OnOperateDelete)
  40. ON_UPDATE_COMMAND_UI(ID_OPERATE_DELETE, OnUpdateOperateDelete)
  41. ON_WM_LBUTTONDOWN()
  42. //}}AFX_MSG_MAP
  43. END_MESSAGE_MAP()
  45. /////////////////////////////////////////////////////////////////////////////
  46. // CLeftTreeView drawing
  48. void CLeftTreeView::OnDraw(CDC* pDC)
  49. {
  50. CDocument* pDoc = GetDocument();
  51. // TODO: add draw code here
  52. }
  54. /////////////////////////////////////////////////////////////////////////////
  55. // CLeftTreeView diagnostics
  57. #ifdef _DEBUG
  58. void CLeftTreeView::AssertValid() const
  59. {
  60. CTreeView::AssertValid();
  61. }
  63. void CLeftTreeView::Dump(CDumpContext& dc) const
  64. {
  65. CTreeView::Dump(dc);
  66. }
  67. #endif //_DEBUG
  69. /////////////////////////////////////////////////////////////////////////////
  70. // CLeftTreeView message handlers
  72. //将数据库中的所有住户到加到树中
  73. void CLeftTreeView::AddUsersToTree()
  74. {
  75. CInhabitantsDoc* pDoc = (CInhabitantsDoc*)GetDocument();
  76. if(pDoc==NULL)
  77. return;
  78. //清空树
  79. GetTreeCtrl().DeleteAllItems();
  81. //取数据库中得所有记录
  82. CString strSql = "select * from house";
  83. CCommand<CAccessor<CHouseAccessor> > dbHouse;
  84. long* pCount = new long;
  85. if(dbHouse.Open(pDoc->m_dbHouse.m_session,strSql,NULL,pCount) != S_OK)
  86. {
  87. AfxMessageBox("error");
  88. delete pCount;
  89. return;
  90. }
  91. delete pCount;
  93. if(dbHouse.MoveFirst() == S_OK)
  94. {
  95. do
  96. {
  97. AddUserToTree(dbHouse.m_sectionname,dbHouse.m_buildingnum,
  98. dbHouse.m_cellnum,dbHouse.m_roomnum);
  99. }
  100. while( dbHouse.MoveNext() == S_OK );
  101. }
  102. dbHouse.Close();
  103. }
  106. void CLeftTreeView::OnInitialUpdate() 
  107. {
  108. CTreeView::OnInitialUpdate();
  109. //设置树风格
  110. ::SetWindowLong(m_hWnd,GWL_STYLE,WS_VISIBLE | WS_TABSTOP 
  111. | WS_CHILD | WS_BORDER| TVS_HASBUTTONS 
  112. | TVS_LINESATROOT | TVS_HASLINES
  113. | TVS_DISABLEDRAGDROP|TVS_SHOWSELALWAYS);
  115. //为树视图创建图标连表
  116. CInhabitantsApp* pApp = (CInhabitantsApp*)AfxGetApp();
  117. CTreeCtrl* pTreeCtrl = &GetTreeCtrl();
  118. m_ImageList.Create(16,16,ILC_COLOR16,4,4);
  119. m_ImageList.Add(pApp->LoadIcon(IDI_ICON_SECTION));
  120. m_ImageList.Add(pApp->LoadIcon(IDI_ICON_BUILDING));
  121. m_ImageList.Add(pApp->LoadIcon(IDI_ICON_CELL));
  122. m_ImageList.Add(pApp->LoadIcon(IDI_ICON_USER));
  123. pTreeCtrl->SetImageList(&m_ImageList,TVSIL_NORMAL);
  125. //将住户加入树中
  126. AddUsersToTree();
  127. pTreeCtrl->SortChildren(TVI_ROOT);
  128. }
  131. void CLeftTreeView::AddUserToTree(CString strSectionName,int nBuildingNum,int nCellNum,int nRoomNum)
  132. {
  133. HTREEITEM hSectionItem = AddSectionToTree(strSectionName);
  134. HTREEITEM hBuildingItem = AddBuildingToTree(hSectionItem,nBuildingNum);
  135. HTREEITEM hCellItem = AddCellToTree(hBuildingItem,nCellNum);
  136. if(hCellItem == NULL)
  137. return;
  138. CTreeCtrl* pCtrl = &GetTreeCtrl();
  139. TV_INSERTSTRUCT TCItem;//插入数据项数据结构
  140. TCItem.hParent = hCellItem;
  141. TCItem.hInsertAfter = TVI_LAST;
  142. TCItem.item.mask=TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
  143. TCItem.item.lParam=0;//序号
  144. TCItem.item.iImage=3;//正常图标
  145. TCItem.item.iSelectedImage=3;//选中时图标
  146. CString str;
  147. str.Format("%d",nRoomNum);
  148. TCItem.item.pszText = (LPTSTR)(LPCTSTR)str;
  149. pCtrl->InsertItem(&TCItem);
  151. pCtrl->SortChildren(hCellItem);
  152. }
  154. //将小区加到树视图中
  155. HTREEITEM CLeftTreeView::AddSectionToTree(CString strSectionName)
  156. {
  157. CTreeCtrl* pCtrl = &GetTreeCtrl();
  158. HTREEITEM hRootItem = pCtrl->GetRootItem();
  159. if(hRootItem)
  160. {
  161. while(hRootItem)
  162. {
  163. CString strItemText = pCtrl->GetItemText(hRootItem);
  164. if(strItemText == strSectionName)
  165. return hRootItem;
  166. hRootItem = pCtrl->GetNextSiblingItem(hRootItem);
  167. }
  168. }
  169. TV_INSERTSTRUCT TCItem;//插入数据项数据结构
  170. TCItem.hParent = TVI_ROOT;
  171. TCItem.hInsertAfter = TVI_LAST;
  172. TCItem.item.mask=TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
  173. TCItem.item.lParam=0;//序号
  174. TCItem.item.iImage=0;//正常图标
  175. TCItem.item.iSelectedImage=0;//选中时图标
  177. TCItem.item.pszText = (LPTSTR)(LPCTSTR)strSectionName;
  178. HTREEITEM hSectionItem = pCtrl->InsertItem(&TCItem);
  179. pCtrl->SortChildren(TVI_ROOT);
  180. return hSectionItem;
  181. }
  183. HTREEITEM CLeftTreeView::AddBuildingToTree(HTREEITEM hSectionItem, int nBuildingnum)
  184. {
  185. if(hSectionItem == NULL)
  186. return NULL;
  187. CTreeCtrl* pCtrl = &GetTreeCtrl();
  188. CString strBuildingNum;
  189. strBuildingNum.Format("%d号楼",nBuildingnum);
  190. if(pCtrl->ItemHasChildren(hSectionItem))
  191. {
  192. HTREEITEM hItem = pCtrl->GetChildItem(hSectionItem);
  193. while(hItem)
  194. {
  195. CString strItemText = pCtrl->GetItemText(hItem);
  196. if(strItemText == strBuildingNum)
  197. return hItem;
  198. hItem = pCtrl->GetNextSiblingItem(hItem);
  199. }
  200. }
  201. TV_INSERTSTRUCT TCItem;//插入数据项数据结构
  202. TCItem.hParent = hSectionItem;
  203. TCItem.hInsertAfter = TVI_LAST;
  204. TCItem.item.mask=TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
  205. TCItem.item.lParam=0;//序号
  206. TCItem.item.iImage=1;//正常图标
  207. TCItem.item.iSelectedImage=1;//选中时图标
  209. TCItem.item.pszText = (LPTSTR)(LPCTSTR)strBuildingNum;
  210. HTREEITEM hBuildingItem = pCtrl->InsertItem(&TCItem);
  212. pCtrl->SortChildren(hSectionItem);
  213. return hBuildingItem;
  214. }
  216. HTREEITEM CLeftTreeView::AddCellToTree(HTREEITEM hBuildingItem, int nCellnum)
  217. {
  218. if(hBuildingItem == NULL)
  219. return NULL;
  220. CTreeCtrl* pCtrl = &GetTreeCtrl();
  221. CString strCellNum;
  222. strCellNum.Format("%d单元",nCellnum);
  223. if(pCtrl->ItemHasChildren(hBuildingItem))
  224. {
  225. HTREEITEM hItem = pCtrl->GetChildItem(hBuildingItem);
  226. while(hItem)
  227. {
  228. CString strItemText = pCtrl->GetItemText(hItem);
  229. if(strItemText == strCellNum)
  230. return hItem;
  231. hItem = pCtrl->GetNextSiblingItem(hItem);
  232. }
  233. }
  234. TV_INSERTSTRUCT TCItem;//插入数据项数据结构
  235. TCItem.hParent = hBuildingItem;
  236. TCItem.hInsertAfter = TVI_LAST;
  237. TCItem.item.mask=TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
  238. TCItem.item.lParam=0;//序号
  239. TCItem.item.iImage=2;//正常图标
  240. TCItem.item.iSelectedImage=2;//选中时图标
  242. TCItem.item.pszText = (LPTSTR)(LPCTSTR)strCellNum;
  243. HTREEITEM hCellItem = pCtrl->InsertItem(&TCItem);
  245. pCtrl->SortChildren(hBuildingItem);
  246. return hCellItem;
  247. }
  249. void CLeftTreeView::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
  250. {
  251. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  253. *pResult = 0;
  254. //获得被选择项
  255. CTreeCtrl* pCtrl = &GetTreeCtrl();
  256. HTREEITEM hSelItem = pCtrl->GetSelectedItem();
  257. m_hHitItem = hSelItem;
  259. //如果没有项被选择，则将右边的列表视图清空
  260. CMainFrame* pFrame = (CMainFrame*)AfxGetApp()->m_pMainWnd;
  261. if(hSelItem == NULL)
  262. {
  263. pFrame->SwitchToView(USERSVIEW);
  264. pFrame->m_pUsersView->ShowUsers("");
  265. return;
  266. }
  268. //判断选择项在树中的位置
  269. HTREEITEM hParentItem = pCtrl->GetParentItem(hSelItem);
  270. CString strSql,strSectionName,strBuildingNum,strCellNum,strRoomNum;
  271. if(hParentItem == NULL)
  272. { //小区
  273. CString strSectionName = pCtrl->GetItemText(hSelItem);
  274. strSql.Format("select * from house where sectionname = '%s'",strSectionName);
  275. pFrame->SwitchToView(USERSVIEW);
  276. pFrame->m_pUsersView->ShowUsers(strSql);
  277. return;
  278. }
  279. hParentItem = pCtrl->GetParentItem(hParentItem);
  280. if(hParentItem == NULL)
  281. { //楼
  282. strSectionName = pCtrl->GetItemText(pCtrl->GetParentItem(hSelItem));
  283. CString str = pCtrl->GetItemText(hSelItem);
  284. int n = str.Find("号楼");
  285. strBuildingNum = str.Left(n);
  286. strSql.Format("select * from house where sectionname = '%s' and buildingnum = %s",
  287. strSectionName,strBuildingNum);
  288. pFrame->SwitchToView(USERSVIEW);
  289. pFrame->m_pUsersView->ShowUsers(strSql);
  290. return;
  291. }
  292. hParentItem = pCtrl->GetParentItem(hParentItem);
  293. if(hParentItem == NULL)
  294. { //单元
  295. strSectionName = pCtrl->GetItemText(pCtrl->GetParentItem
  296. (pCtrl->GetParentItem(hSelItem)));
  297. CString str = pCtrl->GetItemText(pCtrl->GetParentItem(hSelItem));
  298. int n = str.Find("号楼");
  299. strBuildingNum = str.Left(n);
  300. str = pCtrl->GetItemText(hSelItem);
  301. n = str.Find("单元");
  302. strCellNum = str.Left( n );
  303. strSql.Format("select * from house where sectionname = '%s' and buildingnum = %s and cellnum = %s",
  304. strSectionName,strBuildingNum,strCellNum);
  305. pFrame->SwitchToView(USERSVIEW);
  306. pFrame->m_pUsersView->ShowUsers(strSql);
  307. return;
  308. }
  310. //选择项是房间号，则右边视图显示该住户的详细信息
  311. strSectionName = pCtrl->GetItemText(pCtrl->GetParentItem
  312. (pCtrl->GetParentItem(pCtrl->GetParentItem(hSelItem))));
  313. CString str = pCtrl->GetItemText(pCtrl->GetParentItem
  314. (pCtrl->GetParentItem(hSelItem)));
  315. int n = str.Find("号楼");
  316. strBuildingNum = str.Left(n);
  317. str = pCtrl->GetItemText(pCtrl->GetParentItem(hSelItem));
  318. n = str.Find("单元");
  319. strCellNum = str.Left( n );
  320. strRoomNum = pCtrl->GetItemText(hSelItem);
  321. pFrame->SwitchToView(USERINFOVIEW);
  322. pFrame->m_pUserinfoView->UpdateUserInfo(strSectionName,
  323. atoi(strBuildingNum.GetBuffer(0)),
  324. atoi(strCellNum.GetBuffer(0)),
  325. atoi(strRoomNum.GetBuffer(0)));
  326. }
  328. void CLeftTreeView::OnOperateDelete() 
  329. {
  330. if( m_hHitItem == NULL )
  331. return;
  332. CMainFrame* pFrame = (CMainFrame*)AfxGetApp()->m_pMainWnd;
  333. CInhabitantsDoc* pDoc = (CInhabitantsDoc*)GetDocument();
  334. CTreeCtrl* pCtrl = &GetTreeCtrl();
  335. HTREEITEM hParentItem = pCtrl->GetParentItem( m_hHitItem );
  337. CString strSql;
  338. if( hParentItem == NULL && m_hHitItem != NULL )
  339. { //小区
  340. CString strSectionName = pCtrl->GetItemText(m_hHitItem);
  341. strSql.Format("delete * from house where sectionname = '%s'",strSectionName);
  342. if(this->MessageBox("你真的要删除该小区中的所有住户吗?",
  343. "小区居民管理系统",MB_YESNO)==IDNO)
  344. return;
  345. }
  346. else if( hParentItem != NULL && pCtrl->GetParentItem(hParentItem) == NULL )
  347. { //楼
  348. CString strSectionName = pCtrl->GetItemText(pCtrl->GetParentItem(m_hHitItem));
  349. CString str = pCtrl->GetItemText(m_hHitItem);
  350. int n = str.Find("号楼");
  351. CString strBuildingNum = str.Left(n);
  352. strSql.Format("delete * from house where sectionname = '%s' and buildingnum = %s",
  353. strSectionName,strBuildingNum);
  354. if(this->MessageBox("你真的要删除该楼中的所有住户吗?",
  355. "小区居民管理系统",MB_YESNO)==IDNO)
  356. return;
  357. }
  358. else if( pCtrl->GetParentItem(hParentItem) != NULL 
  359. && pCtrl->GetParentItem(pCtrl->GetParentItem(hParentItem)) == NULL )
  360. { //单元
  361. CString strSectionName = pCtrl->GetItemText(
  362. pCtrl->GetParentItem(pCtrl->GetParentItem(m_hHitItem)));
  363. CString str = pCtrl->GetItemText(pCtrl->GetParentItem(m_hHitItem));
  364. int n = str.Find("号楼");
  365. CString strBuildingNum = str.Left(n);
  366. str = pCtrl->GetItemText(m_hHitItem);
  367. n = str.Find("单元");
  368. CString strCellNum = str.Left( n );
  369. strSql.Format("delete * from house where sectionname = '%s' and buildingnum = %s and cellnum = %s",
  370. strSectionName,strBuildingNum,strCellNum);
  371. if(this->MessageBox("你真的要删除该单元中的所有住户吗?",
  372. "小区居民管理系统",MB_YESNO)==IDNO)
  373. return;
  374. }
  375. else
  376. { //用户
  377. CString strSectionName = pCtrl->GetItemText(pCtrl->GetParentItem(
  378. pCtrl->GetParentItem(pCtrl->GetParentItem(m_hHitItem))));
  379. CString str = pCtrl->GetItemText(pCtrl->GetParentItem(
  380. pCtrl->GetParentItem(m_hHitItem)));
  381. int n = str.Find("号楼");
  382. CString strBuildingNum = str.Left(n);
  383. str = pCtrl->GetItemText(pCtrl->GetParentItem(m_hHitItem));
  384. n = str.Find("单元");
  385. CString strCellNum = str.Left( n );
  386. CString strRoomNum = pCtrl->GetItemText(m_hHitItem);
  387. strSql.Format("delete * from house where sectionname = '%s' and buildingnum = %s and cellnum = %s and roomnum = %s",
  388. strSectionName,strBuildingNum,strCellNum,strRoomNum);
  389. if(this->MessageBox("你真的要删除该住户吗?","小区居民管理系统",MB_YESNO)==IDNO)
  390. return;
  391. }
  392. pDoc->DeleteUser(strSql);
  393. AddUsersToTree();
  394. }
  398. void CLeftTreeView::OnUpdateOperateDelete(CCmdUI* pCmdUI) 
  399. {
  400. pCmdUI->Enable(m_hHitItem != NULL);
  401. }