트리 컨트롤 - MFC

CTreeCtrl

on TVN_SELCHANGED

NM_TREEVIEW

.itemNew

on TVN_ITEMEXPANDING

.SetImageList()
.InsertItem()
.DeleteItem()
.GetChildItem()
.GetItemText()
.Expand

class CTreeDlg : public CDialog
{
// Construction
public:
    CTreeDlg(CWnd* pParent = NULL);    // standard constructor
    CImageList m_ImageList;

// Dialog Data
    //{{AFX_DATA(CTreeDlg)
    enum { IDD = IDD_TREE_DIALOG };
    CTreeCtrl    m_ctrlTree;
    //}}AFX_DATA
..
    // Generated message map functions
    //{{AFX_MSG(CTreeDlg)
    virtual BOOL OnInitDialog();
    ..
    afx_msg void OnSelchangedTree1(NMHDR* pNMHDR, LRESULT* pResult);
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()
};


BEGIN_MESSAGE_MAP(CTreeDlg, CDialog)
    //{{AFX_MSG_MAP(CTreeDlg)
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_NOTIFY(TVN_SELCHANGED, IDC_TREE1, OnSelchangedTree1)
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()
..
BOOL CTreeDlg::OnInitDialog()
{
    ..
    SetIcon(m_hIcon, FALSE);        // Set small icon
   
    m_ImageList.Create(IDB_BITMAP1, 16, 2, RGB(255, 255, 255));
    m_ctrlTree.SetImageList(&m_ImageList, TVSIL_NORMAL);
   
    HTREEITEM hLevel1;

    hLevel1 = m_ctrlTree.InsertItem("..", 0, 1);
    m_ctrlTree.InsertItem("..", 0, 1, hLevel1);
    m_ctrlTree.Expand(hLevel1, TVE_EXPAND);
    return TRUE;  // return TRUE  unless you set the focus to a control
}
..
void CTreeDlg::OnSelchangedTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
    NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;

    HTREEITEM hItem = pNMTreeView->itemNew.hItem;
    .. m_ctrlTree.GetItemText(hItem) ..
    *pResult = 0;
}

class CPractice8_2Dlg : public CDialog
{
...
// Dialog Data
    //{{AFX_DATA(CPractice8_2Dlg)
    enum { IDD = IDD_PRACTICE8_2_DIALOG };
    CTreeCtrl    m_treeControl;
...
    //}}AFX_DATA
...
    // Generated message map functions
    //{{AFX_MSG(CPractice8_2Dlg)
    virtual BOOL OnInitDialog();
...
    afx_msg void OnSelchangedTreeControl(NMHDR* pNMHDR, LRESULT* pResult);
    afx_msg void OnButtonInsert();
    afx_msg void OnButtonModify();
    afx_msg void OnButtonDelete();
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()

private:
    HTREEITEM m_hSelectedNode;
    HTREEITEM m_hRoot;
};

CPractice8_2Dlg::CPractice8_2Dlg(CWnd* pParent /*=NULL*/)
    : CDialog(CPractice8_2Dlg::IDD, pParent)
{
...
    // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
// {
    m_hSelectedNode = NULL;
    m_hRoot = NULL;
// }
}
...
BEGIN_MESSAGE_MAP(CPractice8_2Dlg, CDialog)
    //{{AFX_MSG_MAP(CPractice8_2Dlg)
...
    ON_NOTIFY(TVN_SELCHANGED, IDC_TREE_CONTROL, OnSelchangedTreeControl)
    ON_BN_CLICKED(IDC_BUTTON_INSERT, OnButtonInsert)
    ON_BN_CLICKED(IDC_BUTTON_MODIFY, OnButtonModify)
    ON_BN_CLICKED(IDC_BUTTON_DELETE, OnButtonDelete)
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

BOOL CPractice8_2Dlg::OnInitDialog()
{
  ...   
    // TODO: Add extra initialization here
// {
    TV_INSERTSTRUCT tvStruct;

    tvStruct.hParent = 0;
    tvStruct.hInsertAfter = TVI_LAST;
    tvStruct.item.mask = TVIF_TEXT;
    tvStruct.item.pszText = "루트 노드";
    m_hRoot = m_treeControl.InsertItem(&tvStruct);
// }   
    return TRUE;  // return TRUE  unless you set the focus to a control
}

void CPractice8_2Dlg::OnSelchangedTreeControl(NMHDR* pNMHDR, LRESULT* pResult)
{
    NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
    // TODO: Add your control notification handler code here
// {
    // 현재 선택한 아이템의 핸들 값을 멤버변수에 저장한다.
    m_hSelectedNode = pNMTreeView->itemNew.hItem;
...
// }   
    *pResult = 0;
}

void CPractice8_2Dlg::OnButtonInsert()
{
    // TODO: Add your control notification handler code here
    TV_INSERTSTRUCT tvStruct;
   
    // 컨트롤로부터 데이터를 읽어온다.
    UpdateData(TRUE);

    // 에러처리 - 입력할 텍스트가 비어있나 검사한다.
    if( !m_edText.IsEmpty() )
    {
        // 현재 선택된 아이템을 부모노드로 한다.
        tvStruct.hParent = m_hSelectedNode;
        tvStruct.hInsertAfter = TVI_LAST;
        tvStruct.item.mask = TVIF_TEXT;
        // 에디트 상자에 입력된 텍스트를 노드 이름으로 한다.
        tvStruct.item.pszText = (LPTSTR)(LPCSTR)m_edText;
        // 위에서 지정한 정보를 노드를 추가한다.
        m_treeControl.InsertItem(&tvStruct);
       
        m_treeControl.Expand(m_hSelectedNode, TVE_EXPAND);
    }
    else
        AfxMessageBox("입력 노드의 텍스트를 입력하세요");
}

void CPractice8_2Dlg::OnButtonModify()
{
    // TODO: Add your control notification handler code here
    UpdateData(TRUE);

    // 입력할 텍스트가 비어 있는지 검사
    if ( !m_edText.IsEmpty() )
    {
        // 선택된 아이템의 텍스트를 수정한다.
        m_treeControl.SetItemText(m_hSelectedNode, m_edText);
    }
    else
    {
        AfxMessageBox("수정 노드의 텍스트를 입력하세요");
    }
}

void CPractice8_2Dlg::OnButtonDelete()
{
    // TODO: Add your control notification handler code here

        if ( MessageBox("정말 삭제하시겠습니까?", "삭제 경고", MB_YESNO) == IDYES )
            m_treeControl.DeleteItem(m_hSelectedNode);
}

CTreeView

.GetTreeCtrl()

SystemImageList.zip

#include "SystemImageList.h"

class CLeftView : public CTreeView
{
...
// Attributes
public:
    CFileTamDoc* GetDocument();

// Operations
public:

// Overrides
    // ClassWizard generated virtual function overrides
    //{{AFX_VIRTUAL(CLeftView)
    public:
    virtual void OnDraw(CDC* pDC);  // overridden to draw this view
    virtual BOOL PreCreateWindow(CREATESTRUCT& cs);
    protected:
    ...
    virtual void OnInitialUpdate(); // called first time after construct
    //}}AFX_VIRTUAL

// Implementation
public:
    virtual ~CLeftView();
...
protected:
private:
    CString GetSelectFolder(HTREEITEM hItem);
    void SetFolderInfo(CString strParentName, HTREEITEM hItemParent);
    void DeleteAllSubItem(HTREEITEM hItemParent);
    void SetDriveInfo(CString strDrive);

    CSystemImageList m_imgTree;

// Generated message map functions
protected:
    //{{AFX_MSG(CLeftView)
    afx_msg void OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult);
    afx_msg void OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult);
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()
};


BEGIN_MESSAGE_MAP(CLeftView, CTreeView)
    //{{AFX_MSG_MAP(CLeftView)
    ON_NOTIFY_REFLECT(TVN_ITEMEXPANDING, OnItemexpanding)
    ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
    //}}AFX_MSG_MAP
    // Standard printing commands
    ...
END_MESSAGE_MAP()
...
BOOL CLeftView::PreCreateWindow(CREATESTRUCT& cs)
{
    // TODO: Modify the Window class or styles here by modifying
    //  the CREATESTRUCT cs
// {
    // 트리부의 모양 정의
    cs.style |= TVS_HASLINES | TVS_LINESATROOT | TVS_HASBUTTONS | TVS_SHOWSELALWAYS ;   
// }
    return CTreeView::PreCreateWindow(cs);
}
...
void CLeftView::OnInitialUpdate()
{
    CTreeView::OnInitialUpdate();

    // TODO: You may populate your TreeView with items by directly accessing
    //  its tree control through a call to GetTreeCtrl().
    // 트리부에 시스템 이미지 리스트 연결
    GetTreeCtrl().SetImageList(m_imgTree.GetImageList(), TVSIL_NORMAL);


    char pTmp[256];
    memset(pTmp, '\0', 256);
    // 논리적 드라이브 정보 얻기
    GetLogicalDriveStrings(256, pTmp);

    CString strDrive;

    // pTmp 에 얻어진 논리적 드라이브 정보를 strDrive 를 이용해 정보 출력
    for(int i=0 ; i<256 ; i++)
    {
        if(pTmp[i] == '\0' && pTmp[i+1] == '\0')
            break;
       
        strDrive.Empty();
        while(pTmp[i] != '\0')
            strDrive += pTmp[i++];
        SetDriveInfo(strDrive);
    }
}
...

namespace {

// 선택 아이템의 서브 아이템을 모두 제거한다.

void CLeftView::DeleteAllSubItem(HTREEITEM hItemParent)
{
    HTREEITEM hItem;
   
    while(hItem = GetTreeCtrl().GetChildItem(hItemParent))
        GetTreeCtrl().DeleteItem(hItem);
}

}

void CLeftView::SetDriveInfo(CString strDrive)
{
    // 트리뷰에 논리적드라이브 정보 출력
    strDrive.MakeUpper();
    HTREEITEM hItem = GetTreeCtrl().InsertItem(strDrive, m_imgTree.GetIcon(strDrive), m_imgTree.GetIcon(strDrive, TRUE));

    // 만약 서브폴더가 있다면 ??? 라는 임시 아이템을 삽입
    if(HasSubFolder(strDrive))
        GetTreeCtrl().InsertItem("???", hItem);
}

void CLeftView::OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult)
{
    NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
    // TODO: Add your control notification handler code here
// {
    if (pNMTreeView->action == TVE_EXPAND)
    {
        SetFolderInfo(GetSelectFolder(pNMTreeView->itemNew.hItem), pNMTreeView->itemNew.hItem);
    }
// }   
    *pResult = 0;
}

CString CLeftView::GetSelectFolder(HTREEITEM hItem)
{
    // 선택한 아이템의 전체경로를 얻는다.

    HTREEITEM hItemParent;
    CString strFolderName, strTmp;
   
    // 선택한 아이템의 텍스트 얻기
    strFolderName = GetTreeCtrl().GetItemText(hItem);
    // 선택 아이템의 부모 아이템을 검사해서 전체경로 얻기
    for(hItemParent = GetTreeCtrl().GetParentItem(hItem) ; hItemParent ; hItemParent = GetTreeCtrl().GetParentItem(hItemParent))
    {
        strTmp = GetTreeCtrl().GetItemText(hItemParent);
        if(strTmp.Right(1) != "\\")
            strTmp += "\\";

        strFolderName = strTmp + strFolderName;
    }
    return strFolderName;
}

// strParentName의 서브폴더를 찾아서 트리뷰에 출력하는 함수
void CLeftView::SetFolderInfo(CString strParentName, HTREEITEM hItemParent)
{
    DeleteAllSubItem(hItemParent);

    CFileFind cfile;
    CString strFolderName, strFolderPath;
    if(strParentName.Right(1) != "\\")
        strParentName += "\\";

    // 파일 찾기 시작
    BOOL b = cfile.FindFile(strParentName + "*.*");
    HTREEITEM hItem;

    while(b)
    {
        b = cfile.FindNextFile();
       
        // '.' '..' 이면
        if(cfile.IsDots())
            continue;

        // 폴더명 출력
        if(cfile.IsDirectory())
        {
            strFolderName = cfile.GetFileName();
            strFolderPath = cfile.GetFilePath();
            hItem = GetTreeCtrl().InsertItem(strFolderName, m_imgTree.GetIcon(strFolderPath), m_imgTree.GetIcon(strFolderPath, TRUE), hItemParent);
            if(HasSubFolder(strFolderPath))
                GetTreeCtrl().InsertItem("???", hItem);
        }
    }
}

void CLeftView::DeleteAllSubItem(HTREEITEM hItemParent)
{
    // 선택 아이템의 서브 아이템을 모두 제거한다.
    HTREEITEM hItem;
   
    while(hItem = GetTreeCtrl().GetChildItem(hItemParent))
        GetTreeCtrl().DeleteItem(hItem);
}

#include "FileTamView.h"
#include "MainFrm.h"

void CLeftView::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult)
{
    NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
    // TODO: Add your control notification handler code here
// {
    // CFileView의 포인터를 이용해 직접 컨트롤
    CFileTamView *pView = (CFileTamView*)((CMainFrame*)(AfxGetApp()->m_pMainWnd))->m_wndSplitter.GetPane(0, 1);
//    pView->SetFileInfo(GetSelectFolder(pNMTreeView->itemNew.hItem));
// }   
    *pResult = 0;
}

참조 사이트:

http://www.codeguru.com/cpp/controls/controls/article.php/c2169