吕鑫MFC学习系列十

继续吕鑫VS2015的MFC类封装原理学习：

前面已经介绍过了，这一节实现一个员工管理系统中的数据管理和显示。



看一下代码：

// workDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "work.h"
#include "workDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialogEx
{
public:
CAboutDlg();

// 对话框数据
enum { IDD = IDD_ABOUTBOX };

protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持

// 实现
protected:
DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()

// CworkDlg 对话框

CworkDlg::CworkDlg(CWnd* pParent /*=NULL*/)
: CDialogEx(CworkDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
m_bModify = FALSE;//被修改false
}

void CworkDlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CworkDlg, CDialogEx)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDOK, &CworkDlg::OnBnClickedOk)
ON_BN_CLICKED(IDCANCEL, &CworkDlg::OnBnClickedCancel)
ON_BN_CLICKED(IDC_ADD, &CworkDlg::OnBnClickedAdd)
ON_BN_CLICKED(IDC_DEL, &CworkDlg::OnBnClickedDel)
ON_BN_CLICKED(IDC_MODIFY, &CworkDlg::OnBnClickedModify)

ON_BN_CLICKED(IDC_LOAD, &CworkDlg::OnBnClickedLoad)
ON_BN_CLICKED(IDC_SAVE, &CworkDlg::OnBnClickedSave)
END_MESSAGE_MAP()

// CworkDlg 消息处理程序

BOOL CworkDlg::OnInitDialog()
{
CDialogEx::OnInitDialog();

// 将“关于...”菜单项添加到系统菜单中。

// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);

CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
BOOL bNameValid;
CString strAboutMenu;
bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
ASSERT(bNameValid);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}

SetIcon(m_hIcon, TRUE); // 设置大图标
SetIcon(m_hIcon, FALSE); // 设置小图标
CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);

pList->InsertColumn(0, _T("工号"), LVCFMT_LEFT, 120);
pList->InsertColumn(1, _T("姓名"), LVCFMT_LEFT, 150);
pList->InsertColumn(2, _T("工资"), LVCFMT_LEFT, 200);
pList->InsertColumn(3, _T("入职时间"), LVCFMT_LEFT, 200);
pList->SetTextBkColor(RGB(205, 226, 252));
pList->SetBkColor(RGB(205,226,252));
pList->SetTextColor(RGB(0, 0,255));
pList->SetExtendedStyle(LVS_EX_CHECKBOXES | LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES);

OnBnClickedLoad();
return TRUE; // 除非将焦点设置到控件，否则返回 TRUE
}

void CworkDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialogEx::OnSysCommand(nID, lParam);
}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
// 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序，
// 这将由框架自动完成。

void CworkDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用于绘制的设备上下文

SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

// 使图标在工作区矩形中居中
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;

// 绘制图标
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialogEx::OnPaint();
}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CworkDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}

void CworkDlg::OnBnClickedOk()
{
// TODO: 在此添加控件通知处理程序代码
//CDialogEx::OnOK();
}

void CworkDlg::OnBnClickedCancel()
{
if (m_bModify)
{
if (IDYES == AfxMessageBox(_T("是否保存改动过的数据？"), MB_YESNO | MB_ICONQUESTION))
OnBnClickedSave();
}
CDialogEx::OnCancel();

}

// 检查工号是否存在,ture代表不可以插入，false可以插入
BOOL CworkDlg::CheckNumber(LPCTSTR sznumber)
{
CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
int i = 0, nCount = pList->GetItemCount();
while (i < nCount)
{
if (pList->GetItemText(i, 0) == sznumber)
{
AfxMessageBox(_T("该工号已存在，请重新输入！"));
this->SetFocus();
SetDlgItemText(IDC_NUMB, NULL);
SetDlgItemText(IDC_NAME,NULL);
SetDlgItemText(IDC_SALA, NULL);
return TRUE;
}
++i;
}
return FALSE;
}

void CworkDlg::OnBnClickedAdd()
{//排除重复工号的功能，当重复的工号禁止插入（已存在！）
CString str;
GetDlgItemText(IDC_NUMB, str);
if (str == "")
{
return;
}
if (CheckNumber(str))
return;
CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
int i = pList->GetItemCount();

pList->InsertItem(i, str);
GetDlgItemText(IDC_NAME, str);
pList->SetItemText(i,1,str);
GetDlgItemText(IDC_SALA, str);
pList->SetItemText(i, 2, str);
GetDlgItemText(IDC_DATE, str);
int ii=str.Find(_T("日"));//如果后面有星期，通过字符串切割
pList->SetItemText(i, 3, str.Left(ii+1));
m_bModify = TRUE;
}

//pList->GetNextSelectedItem();
//pList->GetFirstSelectedItemPosition();

void CworkDlg::OnBnClickedDel()
{
//同时删除选中的多个列表项
CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
if (!pList->GetSelectedCount())
{
AfxMessageBox(_T("请选中一行再修删除"));
return;
}
POSITION pos = pList->GetFirstSelectedItemPosition();
while (pos)
{
int nSelItem = pList->GetNextSelectedItem(pos);
CString str = pList->GetItemText(nSelItem, 0);
if (IDNO == AfxMessageBox(_T("确定要删除") + str + _T("的信息吗？"), MB_YESNO))
{
continue;
}
else
{
m_bModify = TRUE;
pList->DeleteItem(nSelItem);
pos = pList->GetFirstSelectedItemPosition();
}
}

//单行删除代码：
/*
CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
if (!pList->GetSelectedCount())
{
AfxMessageBox(_T("请选中一行再修删除"));
return;
}
POSITION pos = pList->GetFirstSelectedItemPosition();
int nSel = pList->GetNextSelectedItem(pos);
CString str = pList->GetItemText(nSel, 0);
if (IDNO == AfxMessageBox(_T("确定要删除") + str + _T("的信息吗？"), MB_YESNO))
return;
pList->DeleteItem(nSel);*/

}
//CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
//int nSel = pList->GetSelectionMark();
//if (nSel < 0)
//{
// AfxMessageBox(_T("请选择一行再删除！"));
// return;
//}
//CString str = pList->GetItemText(nSel, 0);
//if (AfxMessageBox(_T("你确定要删除") + str + _T("吗？"), MB_YESNO) == IDYES)
//pList->DeleteItem(nSel);
void CworkDlg::OnBnClickedModify()
{
CListCtrl *pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
POSITION pos = pList->GetFirstSelectedItemPosition();
if (!pos)
{
AfxMessageBox(_T("请选中一行再修改"));
return;
}
while (pos)
{
int nSel = pList->GetNextSelectedItem(pos);
CString str = pList->GetItemText(nSel, 0);
if (IDNO == AfxMessageBox(_T("确定要修改") + str + _T("的信息吗？"), MB_YESNO))
continue;
GetDlgItemText(IDC_NAME, str);
pList->SetItemText(nSel, 1, str);
GetDlgItemText(IDC_SALA, str);
pList->SetItemText(nSel, 2, str);
GetDlgItemText(IDC_DATE, str);
int ii = str.Find(_T("日"));//如果后面有星期，通过字符串切割
pList->SetItemText(nSel, 3, str.Left(ii + 1));
m_bModify = TRUE;
}
//单选代码如下：
/*POSITION pos = pList->GetFirstSelectedItemPosition();
if (!pos)
{
AfxMessageBox(_T("请选中一行再修改"));
return;
}
int nSel=pList->GetNextSelectedItem(pos);
CString str=pList->GetItemText(nSel, 0);
if (IDNO==AfxMessageBox(_T("确定要修改") + str + _T("的信息吗？"),MB_YESNO))
return;
GetDlgItemText(IDC_NAME,str);
pList->SetItemText(nSel,1,str);
GetDlgItemText(IDC_SALA, str);
pList->SetItemText(nSel, 2, str);
GetDlgItemText(IDC_DATE, str);
int ii = str.Find(_T("日"));//如果后面有星期，通过字符串切割
pList->SetItemText(nSel, 3, str.Left(ii+1));*/

}
//if (!pList->GetSelectedCount())
//{
// AfxMessageBox(_T("请选中一行再修改"));
// return;
//}
void CworkDlg::OnBnClickedLoad()
{
CFile file;
if (!file.Open(_T("./worker.dat"), CFile::modeRead ))
return;
SInfo info;
CListCtrl * pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
int i = 0;
CString str;
while (file.Read(&info, sizeof(info)) == sizeof(info))
{
str.Format(_T("%d"), info.nNumb);
pList->InsertItem(i,str);
pList->SetItemText(i, 1, info.sName);
str.Format(_T("%0.2f"),info.fSala);
pList->SetItemText(i, 2, str);
//str = info.tDate.Format(VAR_DATEVALUEONLY);
//str = info.tDate.Format(_T("%Y年%m月%d日"));
str.Format(_T("%d年%d月%d日"), info.tDate.GetYear(), info.tDate.GetMonth(), info.tDate.GetDay() );
pList->SetItemText(i,3,str);
++i;
}
file.Close();
}
void CworkDlg::OnBnClickedSave()
{
CFile file;
if (!file.Open(_T("./worker.dat"), CFile::modeCreate | CFile::modeWrite))
{
AfxMessageBox(_T("保存文件失败！"));
return;
}
SInfo info;
CListCtrl * pList = (CListCtrl *)GetDlgItem(IDC_LIST1);
int i = 0, nCount = pList->GetItemCount();
while (i<nCount)
{
info.nNumb=_tstoi(pList->GetItemText(i, 0));
_tcscpy(info.sName, pList->GetItemText(i, 1));
info.fSala = (float)_tstof(pList->GetItemText(i, 2));
info.tDate.ParseDateTime(pList->GetItemText(i, 3), VAR_DATEVALUEONLY);
file.Write(&info, sizeof(info));
++i;
}
file.Close();
m_bModify = FALSE;
}

// workDlg.h : 头文件
//

#pragma once

// CworkDlg 对话框
class CworkDlg : public CDialogEx
{
// 构造
struct SInfo
{
int nNumb;
TCHAR sName[16];
float fSala;
COleDateTime tDate;
};
BOOL m_bModify;
public:
CworkDlg(CWnd* pParent = NULL);	// 标准构造函数

// 对话框数据
enum { IDD = IDD_WORK_DIALOG };

protected:
virtual void DoDataExchange(CDataExchange* pDX);	// DDX/DDV 支持

// 实现
protected:
HICON m_hIcon;

// 生成的消息映射函数
virtual BOOL OnInitDialog();
afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
afx_msg void OnPaint();
afx_msg HCURSOR OnQueryDragIcon();
DECLARE_MESSAGE_MAP()
public:
afx_msg void OnBnClickedOk();
afx_msg void OnBnClickedCancel();
afx_msg void OnBnClickedAdd();
afx_msg void OnBnClickedDel();
afx_msg void OnBnClickedModify();
afx_msg void OnDtnDatetimechangeDate(NMHDR *pNMHDR, LRESULT *pResult);
afx_msg void OnBnClickedLoad();
afx_msg void OnBnClickedSave();
// 检查工号是否存在
BOOL CheckNumber(LPCTSTR sznumber);
};

其中视频中没有实现的几个作业，我做了一个简单实现，如发现BUG，请留言非常感谢。对于MFC窗口的细节就不详细说明里面的按钮添加顺序和按钮名称了，可以详细看我代码。建议按照视频的代码一步步敲一遍。