depmandlg.cpp

visual c++与sql Server数据库开发考勤管理系统

// DepManDlg.cpp : implementation file
//

#include "stdafx.h"
#include "CheckManage.h"
#include "DepManDlg.h"
#include "DepEditDlg.h"
#include "DepInfo.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

extern CUserInfo curUser;

/////////////////////////////////////////////////////////////////////////////
// CDepManDlg dialog


CDepManDlg::CDepManDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CDepManDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CDepManDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
}


void CDepManDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CDepManDlg)
	DDX_Control(pDX, IDC_TREE1, m_tree);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CDepManDlg, CDialog)
	//{{AFX_MSG_MAP(CDepManDlg)
	ON_BN_CLICKED(IDC_ADD_BUTTON, OnAddButton)
	ON_BN_CLICKED(IDC_MODI_BUTTON, OnModiButton)
	ON_BN_CLICKED(IDC_DEL_BUTTON, OnDelButton)
	ON_NOTIFY(TVN_SELCHANGED, IDC_TREE1, OnSelchangedTree1)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CDepManDlg message handlers

BOOL CDepManDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// TODO: Add extra initialization here
	//权限控制，如果不是管理员则添加、修改和删除按钮置灰
	if (curUser.UserType != 1)
	{
		GetDlgItem(IDC_ADD_BUTTON)->EnableWindow(FALSE);
		GetDlgItem(IDC_MODI_BUTTON)->EnableWindow(FALSE);
		GetDlgItem(IDC_DEL_BUTTON)->EnableWindow(FALSE);
	}
	//装入部门信息
	dep.LoadDep();
	
	//添加根节点
	m_treeImageList.Create(16,16,FALSE,1,0);  //创建CImageList控件

	//装入ICON资源
	HICON hIcon = ::LoadIcon(AfxGetResourceHandle(), MAKEINTRESOURCE(IDI_ICON1));
	m_treeImageList.Add(hIcon);  //将ICON资源添加到CImageList控件中

	//将CImageList控件与Tree控件相关联	
	m_tree.SetImageList(&m_treeImageList, LVSIL_NORMAL);
	m_root = m_tree.InsertItem("部门信息");  //添加根节点

	//调用AddtoTree() 函数，将所有部门信息添加到Tree控件中	
	AddtoTree(m_root, 0);
	m_tree.Expand(m_root, TVE_EXPAND); //展开根节点

	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CDepManDlg::OnSelchangedTree1(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
	// TODO: Add your control notification handler code here
	
	*pResult = 0;
}

void CDepManDlg::AddtoTree(HTREEITEM m_node, int UpperId)
{
	//使用递归方法将部门添加到TreeView控件中
	int i;
	HTREEITEM m_child;
	//依次访问dep对象中的部门数组，查找满足条件的记录添加到Tree控件中
	for(i=0; i<dep.a_UID.GetSize(); i++)
	{
		//@@@@ 打开a_UID数组中等于UpperId的元素
		//@@@@ 表明此元素为当前根节点的子节点，因此将其添加到Tree控件中，新节点为m_child
		//@@@@ 使用SetItemData()函数将部门编号值也保存在新节点中
		//@@@@ 以新节点为根节点，再次递归调用AddtoTree()函数，将其子节点添加到Tree控件中
		if (UpperId == atoi(dep.a_UID.GetAt(i)))
		{
			m_child = m_tree.InsertItem(dep.a_DepName.GetAt(i), m_node);
			m_tree.SetItemData(m_child, atol(dep.a_DepId.GetAt(i)));
			AddtoTree(m_child, atoi(dep.a_DepId.GetAt(i)));
		}
	}
}

// 添加按钮
void CDepManDlg::OnAddButton() 
{
	// TODO: Add your control notification handler code here
	HTREEITEM node;  //用于保存当前选择的节点
	node = m_tree.GetSelectedItem();  //取得当前选择的节点

	// 节点数最多只能有两级
	if(m_tree.GetItemData(node) != 0)		// 不是根节点
	{
		if(m_tree.GetItemData(m_tree.GetParentItem(node)) != 0)
		{
			MessageBox("不能在此级部门下再添加部门");
			return;
		}
	}

	CDepEditDlg dlg;
	//设置编辑对话框中变量的值
	dlg.m_Upper = m_tree.GetItemText(node);
	dlg.UpperId = m_tree.GetItemData(node);
	dlg.DepId = 0;  //表示插入记录

	//如果在编辑对话框中单击“确定”按钮，则在TreeView中添加新节点
	if(dlg.DoModal() == IDOK)
	{
		HTREEITEM child;  //表示插入的新节点，它是node的子节点
		child = m_tree.InsertItem(dlg.m_DepName, node);  //插入节点
		m_tree.SetItemData(child, dlg.DepId);  //设置新节点的值
		m_tree.Expand(node, TVE_EXPAND); //展开node节点
	}
}

// 修改按钮
void CDepManDlg::OnModiButton() 
{
	// TODO: Add your control notification handler code here
	HTREEITEM node;  //用于保存当前节点
	node = m_tree.GetSelectedItem(); //取得当前节点值
	if (m_tree.GetItemData(node) == 0)  //不能修改根节点
	{
		MessageBox("不能修改此项");
		return;
	}
	HTREEITEM parent;	//读取父节点
	parent = m_tree.GetParentItem(node);

	CDepEditDlg dlg;
	//设置编辑对话框中变量的值
	dlg.m_Upper = m_tree.GetItemText(parent);
	dlg.UpperId = m_tree.GetItemData(parent);
	dlg.DepId = m_tree.GetItemData(node);
	dlg.DepName = m_tree.GetItemText(node);
	dlg.m_DepName = m_tree.GetItemText(node);

	//如果在编辑对话框中单击“确定”按钮，则在TreeView中修改节点信息
	if(dlg.DoModal() == IDOK)
	{
		m_tree.SetItemText(node, dlg.m_DepName);  //设置当前节点的显示标题
		m_tree.Expand(node, TVE_EXPAND);  //展开当前节点
	}	
}

// 删除按钮
void CDepManDlg::OnDelButton() 
{
	// TODO: Add your control notification handler code here
	HTREEITEM node;  //用于保存当前节点
	node = m_tree.GetSelectedItem();  //取得当前节点
	if (m_tree.GetItemData(node) == 0) //不能删除根节点
	{
		MessageBox("不能删除此项");
		return;
	}

	//取得当前节点的值，并转换为CString类型
	CString cDepId;
	cDepId.Format("%d", m_tree.GetItemData(node));
	if (dep.HaveSon(cDepId) == 1)
	{
		MessageBox("此部门包含下级部门，不能删除！");
		return;
	}
	if (dep.HaveEmp(cDepId) == 1)
	{
		MessageBox("此部门包含员工，不能删除！");
		return;
	}

	//弹出对话框，要求用户确认是否删除。如果用户单击“是”按钮，则删除
	if (MessageBox("是否删除当前部门？", "请确认", MB_YESNO) == IDYES)
	{
		dep.SqlDelete(cDepId);  //删除表Departments中的指定记录
		m_tree.DeleteItem(node);  //删除Tree控件中的当前节点
	}
}