< 트리 뷰 >

< 트리 뷰 >

이하에 나타내는 것은 트리뷰 컨트롤을 베이스로 한 뷰 윈도우를 작성하는 예입니다. 뷰 윈도우에는 아이템을 추가하여 아이템을 더블 클릭 하면 그 아이템에 관한 정보를 메시지 박스에 표시합니다.

// stdafx.h

#include <atlbase.h>

#include <atlapp.h>

extern CAppModule _Module;

#include <atlwin.h>

#include <atlcrack.h>

#include <atlmisc.h>

#include <atlctrls.h>

#include <atlframe.h>

// view.h

class CTreeView : public CWindowImpl<CTreeView, CTreeViewCtrl>

{

public:

DECLARE_WND_SUPERCLASS(NULL, CTreeViewCtrl::GetWndClassName())

BOOL PreTranslateMessage(MSG* pMsg){

return FALSE;

}

private:

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

// 메세지 맵

BEGIN_MSG_MAP_EX(CTreeView)

MSG_WM_CREATE(OnCreate)

REFLECTED_NOTIFY_CODE_HANDLER_EX(NM_DBLCLK, OnTreeDblClick)

DEFAULT_REFLECTION_HANDLER()

END_MSG_MAP()

LRESULT OnCreate(LPCREATESTRUCT lpcs){

LRESULT lRet = DefWindowProc();

SetFont(AtlGetDefaultGuiFont());

// 트리뷰에 아이템 삽입

HTREEITEM hSysInfo =

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

HTREEITEM hDllVer =

InsertItem(_T("DLL버젼"), hSysInfo, TVI_LAST);

HTREEITEM hItem;

hItem = InsertItem(_T("shell32.dll"), hDllVer, TVI_LAST);

SetItemData(hItem, SHELL);

hItem = InsertItem(_T("comctl32.dll"), hDllVer, TVI_LAST);

SetItemData(hItem, COMCTL);

HTREEITEM hSysDir =

InsertItem(_T("디렉토리"), hSysInfo, TVI_LAST);

hItem = InsertItem(_T("Windows"), hSysDir, TVI_LAST);

SetItemData(hItem, WINDIR);

hItem = InsertItem(_T("시스템"), hSysDir, TVI_LAST);

SetItemData(hItem, SYSDIR);

return lRet;

}

LRESULT OnTreeDblClick(LPNMHDR pnmh){

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

UINT uFlag;

CPoint pt = ::GetMessagePos();

ScreenToClient(&pt);

HTREEITEM hItem = HitTest(pt, &uFlag);

if(hItem == NULL || !(uFlag & TVHT_ONITEM))

return 0;

// 아이템 데이터 취득

CString strInfo;

DWORD dwData = GetItemData(hItem);

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;

}

if(!strInfo.IsEmpty())

MessageBox(strInfo, _T("아이템"));

return 0;

}

};

// mainfrm.h

class CMainFrame : public CFrameWindowImpl<CMainFrame>,

public CUpdateUI<CMainFrame>, public CMessageFilter, public CIdleHandler

{

public:

DECLARE_FRAME_WND_CLASS(NULL, IDR_MAINFRAME)

CTreeView m_view;

virtual BOOL PreTranslateMessage(MSG* pMsg){

if(CFrameWindowImpl<CMainFrame>::PreTranslateMessage(pMsg))

return TRUE;

return m_view.PreTranslateMessage(pMsg);

}

virtual BOOL OnIdle(){

UIUpdateStatusBar();

return FALSE;

}

BEGIN_UPDATE_UI_MAP(CMainFrame)

// 엔트리 없음

END_UPDATE_UI_MAP()

BEGIN_MSG_MAP_EX(CMainFrame)

MSG_WM_CREATE(OnCreate)

COMMAND_ID_HANDLER_EX(ID_APP_EXIT, OnMenuExit)

CHAIN_MSG_MAP(CUpdateUI<CMainFrame>)

CHAIN_MSG_MAP(CFrameWindowImpl<CMainFrame>)

CHAIN_CLIENT_COMMANDS() // 뷰크라스에 커멘드 체인

REFLECT_NOTIFICATIONS() // 메세지 리플렉션

END_MSG_MAP()

LRESULT OnCreate(LPCREATESTRUCT lpcs){

// 상태 바를 작성

CreateSimpleStatusBar();

UIAddStatusBar(m_hWndStatusBar);

// 뷰를 작성

m_hWndClient = m_view.Create(m_hWnd, rcDefault, NULL,

WS_CHILD | WS_VISIBLE | WS_CLIPSIBLINGS | WS_CLIPCHILDREN |

TVS_HASBUTTONS | TVS_HASLINES | TVS_LINESATROOT | TVS_SHOWSELALWAYS,

WS_EX_CLIENTEDGE);

// 메시지 루프에 메세지 필터와 아이돌 핸들러를 추가

CMessageLoop* pLoop = _Module.GetMessageLoop();

pLoop->AddMessageFilter(this);

pLoop->AddIdleHandler(this);

return 0;

}

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

PostMessage(WM_CLOSE);

}

};

// app.cpp

#include "stdafx.h"

#include "resource.h"

#include "view.h"

#include "mainfrm.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: 어플리케이션 실행

{

CMessageLoop theLoop;

_Module.AddMessageLoop(&theLoop);

CMainFrame wnd;

wnd.CreateEx();

wnd.ShowWindow(nCmdShow);

wnd.UpdateWindow();

nRet = theLoop.Run();

_Module.RemoveMessageLoop();

}

_Module.Term();

::CoUninitialize();

return nRet;

}

　우선 stdafx.h 에서는 컨트롤용 클래스를 사용하기 위해서 atlctrls.h 헤더를 인클루드 합니다.

　다음에 뷰 윈도우를 위한 클래스를 정의합니다. 위의 예에서는 프로젝트에 view.h 라고 하는 헤더 파일을 추가하여 거기에 CWindowImpl 클래스로부터 파생한 CTreeView 라고 하는 클래스를 정의하고 있습니다. CWindowImpl 클래스의 제2인수에는 베이스가 되는CTreeViewCtrl 클래스를 지정합니다.

　CTreeView 클래스 내에서는 우선 WM_CREATE 메세지 핸들러를 추가하여 처음에DefWindowProc()를 호출하고 나서 디폴트의 GUI 폰트를 설정해 뷰에 트리 아이템을 삽입합니다.

　다음에 통지 코드가 NM_DBLCLK의 통지 메세지 핸들러를 추가합니다. 여기에서는 선택되고 있는 아이템에 관련 지을 수 있었던 수치에 의해 적절한 시스템 정보를 취득하여 메시지 박스로 표시하고 있습니다. 덧붙여 이 함수는 메세지 리플렉션에 의해 불려 갑니다.

　이번 예에서는 뷰 윈도우의 베이스로서 CTreeViewCtrl 클래스를 사용했습니다만 WTL이 트리뷰 컨트롤을 위해서 준비해 있는 CTreeViewCtrlEx 클래스를 사용할 수도 있습니다.