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                          const wxSize& WXUNUSED(size),
  86                          const wxString& WXUNUSED(name))
  87 {
  88     m_message = message;
  89     m_parent = parent;
  90
  91     SetPath(defaultPath);
  92 }
  93
  94 void wxDirDialog::SetPath(const wxString& path)
  95 {
  96     m_path = path;
  97
  98     // SHBrowseForFolder doesn't like '/'s nor the trailing backslashes
  99     m_path.Replace(_T("/"), _T("\\"));
 100     if ( !m_path.empty() )
 101     {
 102         while ( *(m_path.end() - 1) == _T('\\') )
 103         {
 104             m_path.erase(m_path.length() - 1);
 105         }
 106
 107         // but the root drive should have a trailing slash (again, this is just
 108         // the way the native dialog works)
 109         if ( *(m_path.end() - 1) == _T(':') )
 110         {
 111             m_path += _T('\\');
 112         }
 113     }
 114 }
 115
 116 int wxDirDialog::ShowModal()
 117 {
 118     wxWindow *parent = GetParent();
 119
 120     BROWSEINFO bi;
 121     bi.hwndOwner      = parent ? GetHwndOf(parent) : NULL;
 122     bi.pidlRoot       = NULL;
 123     bi.pszDisplayName = NULL;
 124     bi.lpszTitle      = m_message.c_str();
 125     bi.ulFlags        = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT;
 126     bi.lpfn           = BrowseCallbackProc;
 127     bi.lParam         = (LPARAM)m_path.c_str();    // param for the callback
 128
 129     LPITEMIDLIST pidl = SHBrowseForFolder(&bi);
 130
 131     if ( bi.pidlRoot )
 132     {
 133         ItemListFree((LPITEMIDLIST)bi.pidlRoot);
 134     }
 135
 136     if ( !pidl )
 137     {
 138         // Cancel button pressed
 139         return wxID_CANCEL;
 140     }
 141
 142     BOOL ok = SHGetPathFromIDList(pidl, m_path.GetWriteBuf(MAX_PATH));
 143     m_path.UngetWriteBuf();
 144
 145     ItemListFree(pidl);
 146
 147     if ( !ok )
 148     {
 149         wxLogLastError(wxT("SHGetPathFromIDList"));
 150
 151         return wxID_CANCEL;
 152     }
 153
 154     return wxID_OK;
 155 }
 156
 157 // ----------------------------------------------------------------------------
 158 // private functions
 159 // ----------------------------------------------------------------------------
 160
 161 static int CALLBACK
 162 BrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lp, LPARAM pData)
 163 {
 164     switch(uMsg)
 165     {
 166         case BFFM_INITIALIZED:
 167             // sent immediately after initialisation and so we may set the
 168             // initial selection here
 169             //
 170             // wParam = TRUE => lParam is a string and not a PIDL
 171             SendMessage(hwnd, BFFM_SETSELECTION, TRUE, pData);
 172             break;
 173
 174         case BFFM_SELCHANGED:
 175             {
 176                 // Set the status window to the currently selected path.
 177                 TCHAR szDir[MAX_PATH];
 178                 if ( SHGetPathFromIDList((LPITEMIDLIST)lp, szDir) )
 179                 {
 180                     wxString strDir(szDir);
 181                     int maxChars = 40; // Have to truncate string else it displays incorrectly
 182                     if (strDir.Len() > (size_t) (maxChars - 3))
 183                     {
 184                         strDir = strDir.Right(maxChars - 3);
 185                         strDir = wxString(wxT("...")) + strDir;
 186                     }
 187                     SendMessage(hwnd, BFFM_SETSTATUSTEXT, 0, (LPARAM) (const wxChar*) strDir);
 188                 }
 189             }
 190             break;
 191
 192         //case BFFM_VALIDATEFAILED: -- might be used to provide custom message
 193         //                             if the user types in invalid dir name
 194     }
 195
 196     return 0;
 197 }
 198
 199
 200 static void ItemListFree(LPITEMIDLIST pidl)
 201 {
 202     if ( pidl )
 203     {
 204         LPMALLOC pMalloc;
 205         SHGetMalloc(&pMalloc);
 206         if ( pMalloc )
 207         {
 208             pMalloc->Free(pidl);
 209             pMalloc->Release();
 210         }
 211         else
 212         {
 213             wxLogLastError(wxT("SHGetMalloc"));
 214         }
 215     }
 216 }
 217
 218 #else
 219     #include "../generic/dirdlgg.cpp"
 220 #endif // compiler/platform on which the code here compiles
 221
 222 #endif // wxUSE_DIRDLG