【正文】 pSetFocus()。 return。 } if( m_strStudentName == ) { AfxMessageBox(請?zhí)钏奚犷愋?)。 p = (CEdit*)GetDlgItem(IDC_EDIT_ROOMTYPE)。 pSetFocus()。 return。 } if( m_nBuildingNum == 0 ) { AfxMessageBox(樓號不能為零 )。 p = (CEdit*)GetDlgItem(IDC_EDIT_BUILDINGNUM)。 pSetFocus()。 return。 } if( m_nRoomNum == 0 ) { AfxMessageBox(房間號不能為零 )。 p = (CEdit*)GetDlgItem(IDC_EDIT_ROOMNUM)。 pSetFocus()。 return。 } if( m_nBedNum == 0 ) 14 { AfxMessageBox(床鋪號不能為零 )。 p = (CEdit*)GetDlgItem(IDC_EDIT_BED)。 pSetFocus()。 return。 } CMainFrame* pFrame = (CMainFrame*)AfxGetApp()m_pMainWnd。 CInhabitantsDoc* pDoc = (CInhabitantsDoc*)pFrameGetActiveDocument()。 //讓用戶確認操作 if(thisMessageBox(你真的要添加這條記錄 ?, 中南民族大學宿舍管理系統(tǒng) ,MB_YESNO)==IDNO) { CDialog::OnCancel()。 return。 } //增加學生到數(shù)據(jù)庫中 USER user。 = m_strStudentName。 = m_nBuildingNum。 = m_nRoomNum。 = m_nBedNum。 = m_strStudentname。 15 = m_strHomeTel。 = m_strSex。 = m_strNationNal。 = m_strMajor。 = m_strId。 = m_strRemarks。 switch(pDocAddUser(user)) { case 0: { //已經(jīng)存在該學生了 CString str。 ( %s%d 號樓 %d 號房間 %d 號床鋪已有學生入住 !, m_strStudentName,m_nBuildingNum,m_nRoomNum,m_nBedNum )。 AfxMessageBox( str )。 CEdit* p = (CEdit*)GetDlgItem(IDC_EDIT_BED)。 pSetFocus()。 return。 break。 } case 1: { //增加成功后刷新左邊的樹視圖 pFramem_pLeftViewAddUserToTree(m_strStudentName, m_nBuildingNum,m_nRoomNum,m_nBedNum)。 break。 16 } default: return。 break。 } CDialog::OnOK()。 } ENDING of Beganing of // : implementation file // include include include include include include include ifdef _DEBUG define new DEBUG_NEW undef THIS_FILE 17 static char THIS_FILE[] = __FILE__。 endif ///////////////////////////////////////////////////////////////////////////// // CLeftTreeView IMPLEMENT_DYNCREATE(CLeftTreeView, CTreeView) CLeftTreeView::CLeftTreeView() { } CLeftTreeView::~CLeftTreeView() { } BEGIN_MESSAGE_MAP(CLeftTreeView, CTreeView) //{{AFX_MSG_MAP(CLeftTreeView) ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged) ON_COMMAND(ID_OPERATE_DELETE, OnOperateDelete) ON_UPDATE_COMMAND_UI(ID_OPERATE_DELETE, OnUpdateOperateDelete) ON_WM_LBUTTONDOWN() 18 //}}AFX_MSG_MAP END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CLeftTreeView drawing void CLeftTreeView::OnDraw(CDC* pDC) { CDocument* pDoc = GetDocument()。 // TODO: add draw code here } ///////////////////////////////////////////////////////////////////////////// // CLeftTreeView diagnostics ifdef _DEBUG void CLeftTreeView::AssertValid() const { CTreeView::AssertValid()。 } void CLeftTreeView::Dump(CDumpContextamp。 dc) const { CTreeView::Dump(dc)。 } 19 endif //_DEBUG ///////////////////////////////////////////////////////////////////////////// // CLeftTreeView message handlers //將數(shù)據(jù)庫中的所有學生到加到樹中 void CLeftTreeView::AddUsersToTree() { CInhabitantsDoc* pDoc = (CInhabitantsDoc*)GetDocument()。 if(pDoc==NULL) return。 //清空樹 GetTreeCtrl().DeleteAllItems()。 //取數(shù)據(jù)庫中得所有記錄 CString strSql = select * from house。 CCommandCAccessorCHouseAccessor dbHouse。 long* pCount = new long。 if((pDoc,strSql,NULL,pCount) != S_OK) { AfxMessageBox(error)。 delete pCount。 return。 } delete pCount。 20 if(() == S_OK) { do { AddUserToTree(, ,)。 } while( () == S_OK )。 } ()。 } void CLeftTreeView::OnInitialUpdate() { CTreeView::OnInitialUpdate()。 //設置樹風格 ::SetWindowLong(m_hWnd,GWL_STYLE,WS_VISIBLE | WS_TABSTOP | WS_CHILD | WS_BORDER| TVS_HASBUTTONS | TVS_LINESATROOT | TVS_HASLINES | TVS_DISABLEDRAGDROP|TVS_SHOWSELALWAYS)。 //為樹視圖創(chuàng)建圖標連表 CInhabitantsApp* pApp = (CInhabitantsApp*)AfxGetApp()。 21 CTreeCtrl* pTreeCtrl = amp。GetTreeCtrl()。 (16,16,ILC_COLOR16,4,4)。 (pAppLoadIcon(IDI_ICON_SECTION))。 (pAppLoadIcon(IDI_ICON_BUILDING))。 (pAppLoadIcon(IDI_ICON_CELL))。 (pAppLoadIcon(IDI_ICON_USER))。 pTreeCtrlSetImageList(amp。m_ImageList,TVSIL_NORMAL)。 //將學生加入樹中 AddUsersToTree()。 pTreeCtrlSortChildren(TVI_ROOT)。 } void CLeftTreeView::AddUserToTree(CString strStudentName,int nBuildingNum,int nRoomNum,int nBedNum) { HTREEITEM hSectionItem = AddSectionToTree(strStudentName)。 HTREEITEM hBuildingItem = AddBuildingToTree(hSectionItem,nBuildingNum)。 HTREEITEM hCellItem = AddCellToTree(hBuildingItem,nRoomNum)。 if(hCellItem == NULL) return。 CTreeCtrl* pCtrl = amp。GetTreeCtrl()。 TV_INSERTSTRUCT TCItem。//插入數(shù)據(jù)項數(shù)據(jù)結(jié)構(gòu) = hCellItem。 22 = TVI_LAST。 =TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE。//設屏蔽 =0。//序號 =3。//正常圖標 =3。//選中時圖標 CString str。 (%d 號床鋪 ,nBedNum)。 = (LPTSTR)(LPCTSTR)str。 pCtrlInsertItem(amp。TCItem)。 pCtrlSortChildren(hCellItem)。 } //將宿舍類型加到樹視圖中 HTREEITEM CLeftTreeView::AddSectionToTree(CString strStudentName) { CTreeCtrl* pCtrl = amp。GetTreeCtrl()。 HTREEITEM hRootItem = pCtrlGetRootItem()。 if(hRootItem) { while(hRootItem) { CString strItemText = pCtrlGetItemText(hRootItem)。 if(strItemText == strStudentName) 23 return hRootItem。 hRootItem = pCtrlGetNextSiblingItem(hRootItem)。 } } TV_INSERTSTRUCT TCItem。//插入數(shù)據(jù)項數(shù)據(jù)結(jié)構(gòu) = TVI_ROOT。 = TVI_LAST。 =TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE。//設屏蔽 =0。//序號 =0。//正常圖標 =0。//選中時圖標 = (LPTSTR)(LPCTSTR)strStudentName。 HTREEITEM hSectionItem = pCtrlInsertItem(amp。TCItem)