src/msw/dirdlg.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        dirdlg.cpp
   3 // Purpose:     wxDirDialog
   4 // Author:      Julian Smart
   5 // Modified by:
   6 // Created:     01/02/97
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Julian Smart and Markus Holzem
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 #ifdef __GNUG__
  21     #pragma implementation "dirdlg.h"
  22 #endif
  23
  24 // For compilers that support precompilation, includes "wx.h".
  25 #include "wx/wxprec.h"
  26
  27 #ifdef __BORLANDC__
  28     #pragma hdrstop
  29 #endif
  30
  31 #if wxUSE_DIRDLG
  32
  33 #if defined(__WIN95__) && !defined(__GNUWIN32_OLD__) && wxUSE_OLE
  34
  35 #ifndef WX_PRECOMP
  36     #include "wx/utils.h"
  37     #include "wx/dialog.h"
  38     #include "wx/dirdlg.h"
  39     #include "wx/log.h"
  40 #endif
  41
  42 #include "wx/msw/private.h"
  43
  44 #include <shlobj.h> // Win95 shell
  45
  46 // ----------------------------------------------------------------------------
  47 // constants
  48 // ----------------------------------------------------------------------------
  49
  50 #ifndef MAX_PATH
  51     #define MAX_PATH 4096      // be generous
  52 #endif
  53
  54 // ----------------------------------------------------------------------------
  55 // wxWindows macros
  56 // ----------------------------------------------------------------------------
  57
  58 IMPLEMENT_CLASS(wxDirDialog, wxDialog)
  59
  60 // ----------------------------------------------------------------------------
  61 // private functions prototypes
  62 // ----------------------------------------------------------------------------
  63
  64 // free the parameter
  65 static void ItemListFree(LPITEMIDLIST pidl);
  66
  67 // the callback proc for the dir dlg
  68 static int CALLBACK BrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lp,
  69                                        LPARAM pData);
  70
  71
  72 // ============================================================================
  73 // implementation
  74 // ============================================================================
  75
  76 // ----------------------------------------------------------------------------
  77 // wxDirDialog
  78 // ----------------------------------------------------------------------------
  79
  80 wxDirDialog::wxDirDialog(wxWindow *parent,
  81                          const wxString& message,
  82                          const wxString& defaultPath,
  83                          long WXUNUSED(style),
  84                          const wxPoint& WXUNUSED(pos))
  85 {
  86     m_message = message;
  87     m_parent = parent;
  88     m_path = defaultPath;
  89
  90     // SHBrowseForFolder doesn't like '/'s nor the trailing backslashes
  91     m_path.Replace(_T("/"), _T("\\"));
  92     if ( *m_path.end() == _T('\\') )
  93     {
  94         m_path.erase(m_path.length() - 1);
  95     }
  96 }
  97
  98 int wxDirDialog::ShowModal()
  99 {
 100     BROWSEINFO bi;
 101     bi.hwndOwner      = m_parent ? GetHwndOf(m_parent) : NULL;
 102     bi.pidlRoot       = NULL;
 103     bi.pszDisplayName = NULL;
 104     bi.lpszTitle      = m_message.c_str();
 105     bi.ulFlags        = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT;
 106     bi.lpfn           = BrowseCallbackProc;
 107     bi.lParam         = (LPARAM)m_path.c_str();    // param for the callback
 108
 109     LPITEMIDLIST pidl = SHBrowseForFolder(&bi);
 110
 111     if ( bi.pidlRoot )
 112     {
 113         ItemListFree((LPITEMIDLIST)bi.pidlRoot);
 114     }
 115
 116     if ( !pidl )
 117     {
 118         // Cancel button pressed
 119         return wxID_CANCEL;
 120     }
 121
 122     BOOL ok = SHGetPathFromIDList(pidl, m_path.GetWriteBuf(MAX_PATH));
 123     m_path.UngetWriteBuf();
 124
 125     ItemListFree(pidl);
 126
 127     if ( !ok )
 128     {
 129         wxLogLastError(wxT("SHGetPathFromIDList"));
 130
 131         return wxID_CANCEL;
 132     }
 133
 134     return wxID_OK;
 135 }
 136
 137 // ----------------------------------------------------------------------------
 138 // private functions
 139 // ----------------------------------------------------------------------------
 140
 141 static int CALLBACK
 142 BrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lp, LPARAM pData)
 143 {
 144     switch(uMsg)
 145     {
 146         case BFFM_INITIALIZED:
 147             // sent immediately after initialisation and so we may set the
 148             // initial selection here
 149             //
 150             // wParam = TRUE => lParam is a string and not a PIDL
 151             SendMessage(hwnd, BFFM_SETSELECTION, TRUE, pData);
 152             break;
 153
 154         case BFFM_SELCHANGED:
 155             {
 156                 // Set the status window to the currently selected path.
 157                 TCHAR szDir[MAX_PATH];
 158                 if ( SHGetPathFromIDList((LPITEMIDLIST)lp, szDir) )
 159                 {
 160                     wxString strDir(szDir);
 161                     int maxChars = 40; // Have to truncate string else it displays incorrectly
 162                     if (strDir.Len() > (size_t) (maxChars - 3))
 163                     {
 164                         strDir = strDir.Right(maxChars - 3);
 165                         strDir = wxString(wxT("...")) + strDir;
 166                     }
 167                     SendMessage(hwnd, BFFM_SETSTATUSTEXT, 0, (LPARAM) (const wxChar*) strDir);
 168                 }
 169             }
 170             break;
 171
 172         //case BFFM_VALIDATEFAILED: -- might be used to provide custom message
 173         //                             if the user types in invalid dir name
 174     }
 175
 176     return 0;
 177 }
 178
 179
 180 static void ItemListFree(LPITEMIDLIST pidl)
 181 {
 182     if ( pidl )
 183     {
 184         LPMALLOC pMalloc;
 185         SHGetMalloc(&pMalloc);
 186         if ( pMalloc )
 187         {
 188             pMalloc->Free(pidl);
 189             pMalloc->Release();
 190         }
 191         else
 192         {
 193             wxLogLastError(wxT("SHGetMalloc"));
 194         }
 195     }
 196 }
 197
 198 #else
 199     #include "../generic/dirdlgg.cpp"
 200 #endif // compiler/platform on which the code here compiles
 201
 202 #endif // wxUSE_DIRDLG