TnbDirectorySelectDlg.h

[詳解]

#pragma once
#include <shlobj.h>
 
 
 
//TNB Library
namespace TNB
{
    
    
    
class CDirectorySelectDlg
{
public:
 
    struct IListener
    {
        virtual ~IListener(void) {}
 
        virtual void OnDirectorySelectDlgInitialized(HWND hWnd) = 0;
 
        virtual bool OnDirectorySelectDlgSelectChanged(HWND hWnd, LPCTSTR lpszPath) const = 0;
    };
    
    CDirectorySelectDlg(void) : m_piListener(NULL)
    {
        m_path[0] = 0;
    }
 
    void SetListener(IListener* I)
    {
        m_piListener = I;
    }
 
    LPCTSTR GetSelectedPath(void) const
    {
        return m_path;
    }
 
    int DoModal(LPCTSTR lpszDirectory, LPCTSTR lpszTitle, UINT uFlag = 0, HWND hWnd = NULL)
    {
        m_path[0] = 0;
        LPMALLOC lpMalloc = NULL;
        HRESULT hr = ::SHGetMalloc(&lpMalloc);
        if ( FAILED(hr) )
        {
            return IDCANCEL;
        }
        // ブラウズ情報受け取りバッファ領域の確保
        LPTSTR lpBuffer = static_cast<LPTSTR>(lpMalloc->Alloc(MAX_PATH));
        if ( lpBuffer == NULL )
        {
            return IDCANCEL;
        }
        int r = IDCANCEL;
        HRESULT hCom = ::CoInitialize(NULL);
        // BROWSEINFO構造体の初期値設定
        BROWSEINFO bi = { 0 };
        bi.hwndOwner = hWnd;
        bi.pidlRoot = NULL;
        bi.pszDisplayName = lpBuffer;
        bi.lpszTitle = lpszTitle;
        bi.ulFlags = uFlag;
        bi.lpfn = ms_BrowseCallbackProc; // コールバック関数のアドレスを設定
        TParam param;
        param.lpszDefaultDir = lpszDirectory;
        param.piListener = m_piListener;
        bi.lParam = reinterpret_cast<LPARAM>(&param);
        // フォルダ選択ダイアログの表示 
        LPITEMIDLIST pidlBrowse = ::SHBrowseForFolder(&bi);
        if ( pidlBrowse != NULL )
        {
            // PIDL形式の戻り値のファイルシステムのパスに変換
            if ( ::SHGetPathFromIDList(pidlBrowse, lpBuffer) )
            {
                // 取得成功
                _tcscpy(m_path, lpBuffer);
                r = IDOK;
            }
            // SHBrowseForFolderの戻り値PIDLを解放
            lpMalloc->Free(pidlBrowse);
        }
        // クリーンアップ処理
        lpMalloc->Free(lpBuffer);
        if ( hCom == S_OK )
        {
            ::CoUninitialize();
        }
        //
        return r;
    }
 
private:
    IListener*  m_piListener;           
    TCHAR       m_path[MAX_PATH];       
    struct TParam
    {
        LPCTSTR     lpszDefaultDir; 
        IListener*  piListener;     
    };
    static int CALLBACK ms_BrowseCallbackProc(HWND hWnd, UINT uMsg, LPARAM lParam, LPARAM lpData)
    {
        const TParam* P = reinterpret_cast<const TParam*>(lpData);
        if ( uMsg == BFFM_INITIALIZED )
        {
            ::SendMessage(hWnd, BFFM_SETSELECTION, TRUE, reinterpret_cast<LPARAM>(P->lpszDefaultDir));
            if ( P->piListener != NULL )
            {
                P->piListener->OnDirectorySelectDlgInitialized(hWnd);
                bool r = P->piListener->OnDirectorySelectDlgSelectChanged(hWnd, P->lpszDefaultDir);
                ::SendMessage(hWnd, BFFM_ENABLEOK, 0, r);
            }
        }
        else if ( uMsg == BFFM_SELCHANGED ) 
        {
            if ( P->piListener != NULL )
            {
                TCHAR path[MAX_PATH];
                bool r = false;
                if ( ::SHGetPathFromIDList(reinterpret_cast<LPCITEMIDLIST>(lParam), path) )
                {
                    ::SendMessage(hWnd, BFFM_SETSTATUSTEXT, 0, reinterpret_cast<LPARAM>(path));
                    r = P->piListener->OnDirectorySelectDlgSelectChanged(hWnd, path);
                }
                ::SendMessage(hWnd, BFFM_ENABLEOK, 0, r);
            }
        }
        else if( uMsg == BFFM_VALIDATEFAILED )
        {
        }
        return 0;
    }
};
 
 
 
};//TNB