< 확장 트리 뷰 >

WTL에서 고용 컨트롤의 트리 뷰 컨트롤을 캡슐화하기 위해서 CTreeViewCtrlT 클래스를 준비해 있습니다만 CTreeViewCtrlExT 이라고 하는 클래스도 준비해 있습니다. 이 클래스는 CTreeViewCtrlT 클래스와 같은 템플릿 클래스입니다. CTreeViewCtrlExT 클래스는 CTreeViewCtrlT 클래스의 상속 클래스 이며 CTreeViewCtrlExT 클래스의 템플릿 인수는 base class인 CTreeViewCtrlT 클래스에 건네 받습니다. atlctrls.h헤더에서는 다음과 같이 선언되고 있습니다.

// atlctrls.h

template <class TBase>

class CTreeViewCtrlExT : public CTreeViewCtrlT< TBase >

{

...

...

typedef CTreeViewCtrlExT<ATL::CWindow> CTreeViewCtrlEx;

CTreeViewCtrlExT 클래스는 CTreeItemT 이라고 하는 템플릿 클래스를 사용합니다. CTreeItemT 클래스는 트리 아이템 핸들HTREEITEM을 캡슐화합니다. 그리고 CTreeItemT 클래스는 CTreeViewCtrlExT 클래스에의 포인터를 멤버 변수로서 가집니다. CTreeItemT 클래스의 템플릿 인수는 이 멤버 변수인 CTreeViewCtrlExT 클래스에 건네 받습니다. atlctrls.h 헤더에서는 다음과 같이 선언되고 있습니다.

// atlctrls.h

template <class TBase>

class CTreeItemT

{

public:

HTREEITEM m_hTreeItem;

CTreeViewCtrlExT<TBase>* m_pTreeView;

...

...

typedef CTreeItemT<ATL::CWindow> CTreeItem;

　CTreeViewCtrlEx 클래스는 CTreeViewCtrl클래스와 동등의 기능을 갖추고 있습니다. 이하에 나타내는 것은 CTreeViewCtrlEx 클래스를 사용하는 예 입니다. 전회 트리 뷰 컨트롤을 사용한 프로그램의 CTreeViewCtrl을 CTreeViewCtrlEx로 변경해 고쳐 씁니다.

// stdafx.h

#include <atlbase.h>

#include <atlapp.h>

extern CAppModule _Module;

#include <atlwin.h>

#include <atlcrack.h>

#include <atlmisc.h>

#include <atlctrls.h> // 컨트롤용 클래스를 사용하기 위해

// maindlg.h

class CMainDlg : public CDialogImpl<CMainDlg>

{

public:

enum { IDD = IDD_MAINDLG };

enum {SHELL = 1, COMCTL, WINDIR, SYSDIR}; // 트리 아이템 식별자

CTreeViewCtrlEx m_tree_sysinfo; // 확장 트리뷰

CStatic m_static_display; // 정보 표시용

// 메세지 맵

BEGIN_MSG_MAP_EX(CMainDlg)

MSG_WM_INITDIALOG(OnInitDialog)

NOTIFY_HANDLER_EX(IDC_TREE_SYSINFO, NM_CLICK, OnTreeClick)

COMMAND_ID_HANDLER_EX(IDOK, OnOK)

COMMAND_ID_HANDLER_EX(IDCANCEL, OnCancel)

END_MSG_MAP()

LRESULT OnInitDialog(HWND hWnd, LPARAM lParam){

// 스크린의 중앙에 배치

CenterWindow();

// 큰 아이콘 설정

HICON hIcon = AtlLoadIconImage(IDR_MAINFRAME, LR_DEFAULTCOLOR,

::GetSystemMetrics(SM_CXICON), ::GetSystemMetrics(SM_CYICON));

SetIcon(hIcon, TRUE);

// 작은 아이콘 설정

HICON hIconSmall = AtlLoadIconImage(IDR_MAINFRAME, LR_DEFAULTCOLOR,

::GetSystemMetrics(SM_CXSMICON), ::GetSystemMetrics(SM_CYSMICON));

SetIcon(hIconSmall, FALSE);

// 컨트롤 설정

m_tree_sysinfo = GetDlgItem(IDC_TREE_SYSINFO);

m_static_display = GetDlgItem(IDC_STATIC_DISPLAY);

CTreeItem tiSysInfo =

m_tree_sysinfo.InsertItem(_T("시스템 정보"), TVI_ROOT, TVI_LAST);

CTreeItem tiDllVer = tiSysInfo.AddTail(_T("DLL버젼"), -1);

CTreeItem tiItem;

tiItem = tiDllVer.AddTail(_T("shell32.dll"), -1);

tiItem.SetData(SHELL);

tiItem = tiDllVer.AddTail(_T("comctl32.dll"), -1);

tiItem.SetData(COMCTL);

CTreeItem tiSysDir = tiSysInfo.AddTail(_T("디렉토리"), -1);

tiItem = tiSysDir.AddTail(_T("Windows"), -1);

tiItem.SetData(WINDIR);

tiItem = tiSysDir.AddTail(_T("시스템"), -1);

tiItem.SetData(SYSDIR);

return TRUE;

}

LRESULT OnTreeClick(LPNMHDR pnmh){

// 클릭된 아이템 핸들 취득

UINT uFlag;

CPoint pt = ::GetMessagePos();

m_tree_sysinfo.ScreenToClient(&pt);

CTreeItem tiItem = m_tree_sysinfo.HitTest(pt, &uFlag);

if(tiItem.IsNull() || !(uFlag & TVHT_ONITEM))

return 0;

// 아이템 데이터 취득

CString strInfo;

DWORD dwData = tiItem.GetData();

switch(dwData){

case SHELL:

{

// shell32.dll의 버젼 취득

DWORD dwMajor, dwMinor;

AtlGetShellVersion(&dwMajor, &dwMinor);

strInfo.Format(_T("Version %d.%02d"), dwMajor, dwMinor);

break;

}

case COMCTL:

{

// comctl32.dll의 버젼 취득

DWORD dwMajor, dwMinor;

AtlGetCommCtrlVersion(&dwMajor, &dwMinor);

strInfo.Format(_T("Version %d.%02d"), dwMajor, dwMinor);

break;

}

case WINDIR:

{

// Windows폴더 패스 취득

TCHAR szPath[MAX_PATH];

GetWindowsDirectory(szPath, MAX_PATH);

strInfo = szPath;

break;

}

case SYSDIR:

{

// 시스템 폴더 패스 취득

TCHAR szPath[MAX_PATH];

GetSystemDirectory(szPath, MAX_PATH);

strInfo = szPath;

break;

}

default:

break;

}

m_static_display.SetWindowText(strInfo);

return 0;

}

void OnOK(UINT uNotifyCode, int nID, HWND hWndCtl){

EndDialog(nID);

}

void OnCancel(UINT uNotifyCode, int nID, HWND hWndCtl){

EndDialog(nID);

}

};

// Control.cpp

#include "stdafx.h"

#include "resource.h"

#include "maindlg.h"

CAppModule _Module;

int WINAPI _tWinMain(HINSTANCE hInstance, HINSTANCE, LPTSTR lpCmdLine, int nCmdShow)

{

HRESULT hRes = ::CoInitialize(NULL);

ATLASSERT(SUCCEEDED(hRes));

::DefWindowProc(NULL, 0, 0, 0L);

AtlInitCommonControls(ICC_COOL_CLASSES | ICC_WIN95_CLASSES);

hRes = _Module.Init(NULL, hInstance);

ATLASSERT(SUCCEEDED(hRes));

int nRet = 0;

// BLOCK: 어플리케이션 실행

{

CMainDlg dlgMain;

nRet = dlgMain.DoModal();

}

_Module.Term();

::CoUninitialize();

return nRet;

}

　우선 CMainDlg 클래스의 CTreeViewCtrl을 CTreeViewCtrlEx에 옮겨놓습니다.

　다음에 WM_INITDIALOG 메세지 핸들러로 트리 뷰 컨트롤에 아이템을 삽입합니다만 전회는 InsertItem()의 리턴 값은 트리 아이템 핸들인 것에 대해 이번은CTreeViewCtrlEx 클래스를 사용하고 있으므로 리턴 값은 CTreeItem 클래스의 인스턴스입니다. 이 인스턴스 안에는 이미 트리 아이템 핸들과 m_tree_sysinfo 에의 포인터가 설정되어 있습니다. 삽입한 아이템 아래에 한층 더 아이 아이템을 추가하려면 CTreeItem의 인스턴스의 AddTail()을 호출합니다.

　OnTreeClick()에서는 HitTest()의 리턴 값으로부터 CTreeItem 클래스의 인스턴스를 취득한 후 IsNull()에 의해 트리 아이템 핸들인가 유효한지 어떤지를 판단해서 유효한 경우는 GetData()에 의해 그 아이템으로부터 데이터를 취득하고 있습니다.

이와 같이 ,CTreeViewCtrlEx클래스를 사용하면(자) , 특정의 트리 아이템을 기점으로서 트리뷰콘트로르를 조작할 수 있습니다.