modifyclasstreenode.cpp

深入浅出Visual C++入门进阶与应用实例 随书光盘 作者 何志丹

// ModifyClassTreeNode.cpp : implementation file
//

#include "stdafx.h"
#include "ModifyClassTreeNode.h"

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

/////////////////////////////////////////////////////////////////////////////
// CModifyClassTreeNodeDlg dialog


CModifyClassTreeNodeDlg::CModifyClassTreeNodeDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CModifyClassTreeNodeDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CModifyClassTreeNodeDlg)
	m_strNodeName = _T("");
	m_strDir = _T("");
	//}}AFX_DATA_INIT
	m_treectrl.m_bReadOnly = true;
	m_pMainData = NULL ;
}


void CModifyClassTreeNodeDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CModifyClassTreeNodeDlg)
	DDX_Control(pDX, IDC_PARENTID, m_treectrl);
	DDX_Text(pDX, IDC_NODENAME, m_strNodeName);
	DDX_Text(pDX, IDC_URL_DIR, m_strDir);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CModifyClassTreeNodeDlg, CDialog)
	//{{AFX_MSG_MAP(CModifyClassTreeNodeDlg)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CModifyClassTreeNodeDlg message handlers

BOOL CModifyClassTreeNodeDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	//初始化树控件,并选中节点
	m_treectrl.SetMainData(m_pMainData);
	m_treectrl.InitTree();
	HTREEITEM hItem = m_treectrl.GetRootItem();
	hItem = m_treectrl.GetItemByID(hItem,m_ParentID);
	if(NULL != hItem)
	{
		m_treectrl.Expand(hItem,TVE_EXPAND);
		m_treectrl.SelectItem(hItem);
	}
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CModifyClassTreeNodeDlg::OnOK() 
{
	HTREEITEM hItem = m_treectrl.GetSelectedItem();
	if(NULL == hItem)
		return ;
	m_ParentID = m_treectrl.GetItemData(hItem);
	
	CDialog::OnOK();
}

bool CModifyClassTreeNodeDlg::GetNewTreeNode(CString &strContent, CString &strDir, int &nParentID)
{

	m_strNodeName = strContent ;
	m_ParentID	  = nParentID ;
	m_strDir      = strDir ;
	if(IDOK != DoModal())
		return false ;
	
	strContent = m_strNodeName;
	nParentID  = m_ParentID;
	strDir     = m_strDir ;
	return true ;
}

CMainData* CModifyClassTreeNodeDlg::SetMainData(CMainData *pData)
{
	m_pMainData = pData ;
	return m_pMainData ;
}