]> git.saurik.com Git - wxWidgets.git/blobdiff - src/generic/dirdlgg.cpp
Ensure there is valid context for DrawRectangle
[wxWidgets.git] / src / generic / dirdlgg.cpp
index b8bab14d3ff93815b0fa73ee47eb188b99188f78..14cf31b5a0bf257f335bbb5ea7cacef11226b0c4 100644 (file)
 /////////////////////////////////////////////////////////////////////////////
-// Name:        dirdlg.cpp
+// Name:        src/generic/dirdlgg.cpp
 // Purpose:     wxDirDialog
-// Author:      Harm van der Heijden and Robert Roebling
+// Author:      Harm van der Heijden, Robert Roebling & Julian Smart
 // Modified by:
 // Created:     12/12/98
 // RCS-ID:      $Id$
-// Copyright:   (c) Harm van der Heijden and Robert Roebling
+// Copyright:   (c) Harm van der Heijden, Robert Roebling, Julian Smart
 // Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
-#ifdef __GNUG__
-#pragma implementation "dirdlgg.h"
-#endif
-
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
-#pragma hdrstop
+    #pragma hdrstop
 #endif
 
-#include "wx/defs.h"
-
 #if wxUSE_DIRDLG
 
-#include "wx/utils.h"
-#include "wx/dialog.h"
-#include "wx/button.h"
-#include "wx/layout.h"
-#include "wx/msgdlg.h"
-#include "wx/textctrl.h"
-#include "wx/textdlg.h"
-#include "wx/filefn.h"
-#include "wx/cmndata.h"
-#include "wx/gdicmn.h"
-#include "wx/intl.h"
-#include "wx/imaglist.h"
-#include "wx/icon.h"
-#include "wx/log.h"
-#include "wx/sizer.h"
-#include "wx/tokenzr.h"
-#include "wx/dir.h"
-
-#if wxUSE_STATLINE
-    #include "wx/statline.h"
+#ifndef WX_PRECOMP
+    #include "wx/textctrl.h"
+    #include "wx/button.h"
+    #include "wx/checkbox.h"
+    #include "wx/sizer.h"
+    #include "wx/intl.h"
+    #include "wx/log.h"
+    #include "wx/msgdlg.h"
+    #include "wx/bmpbuttn.h"
 #endif
 
+#include "wx/statline.h"
+#include "wx/dirctrl.h"
 #include "wx/generic/dirdlgg.h"
+#include "wx/artprov.h"
+#include "wx/menu.h"
 
-// If compiled under Windows, this macro can cause problems
-#ifdef GetFirstChild
-#undef GetFirstChild
-#endif
-
-#ifndef __WXMSW__
-/* XPM */
-static char * icon1_xpm[] = {
-/* width height ncolors chars_per_pixel */
-"16 16 6 1",
-/* colors */
-"   s None  c None",
-".  c #000000",
-"+  c #c0c0c0",
-"@  c #808080",
-"#  c #ffff00",
-"$  c #ffffff",
-/* pixels */
-"                ",
-"   @@@@@        ",
-"  @#+#+#@       ",
-" @#+#+#+#@@@@@@ ",
-" @$$$$$$$$$$$$@.",
-" @$#+#+#+#+#+#@.",
-" @$+#+#+#+#+#+@.",
-" @$#+#+#+#+#+#@.",
-" @$+#+#+#+#+#+@.",
-" @$#+#+#+#+#+#@.",
-" @$+#+#+#+#+#+@.",
-" @$#+#+#+#+#+#@.",
-" @@@@@@@@@@@@@@.",
-"  ..............",
-"                ",
-"                "};
-
-/* XPM */
-static char * icon2_xpm[] = {
-/* width height ncolors chars_per_pixel */
-"16 16 6 1",
-/* colors */
-"   s None  c None",
-".  c #000000",
-"+  c #c0c0c0",
-"@  c #808080",
-"#  c #ffff00",
-"$  c #ffffff",
-/* pixels */
-"                ",
-"   @@@@@        ",
-"  @$$$$$@       ",
-" @$#+#+#$@@@@@@ ",
-" @$+#+#+$$$$$$@.",
-" @$#+#+#+#+#+#@.",
-"@@@@@@@@@@@@@#@.",
-"@$$$$$$$$$$@@+@.",
-"@$#+#+#+#+##.@@.",
-" @$#+#+#+#+#+.@.",
-" @$+#+#+#+#+#.@.",
-"  @$+#+#+#+##@..",
-"  @@@@@@@@@@@@@.",
-"   .............",
-"                ",
-"                "};
-
-#endif // !wxMSW
+// ----------------------------------------------------------------------------
+// constants
+// ----------------------------------------------------------------------------
 
 static const int ID_DIRCTRL = 1000;
 static const int ID_TEXTCTRL = 1001;
-static const int ID_OK = 1002;
-static const int ID_CANCEL = 1003;
 static const int ID_NEW = 1004;
-//static const int ID_CHECK = 1005;
-
-//-----------------------------------------------------------------------------
-// wxDirItemData
-//-----------------------------------------------------------------------------
-
-wxDirItemData::wxDirItemData(wxString& path, wxString& name)
-{
-    m_path = path;
-    m_name = name;
-    /* Insert logic to detect hidden files here
-     * In UnixLand we just check whether the first char is a dot
-     * For FileNameFromPath read LastDirNameInThisPath ;-) */
-    // m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
-    m_isHidden = FALSE;
-    m_hasSubDirs = HasSubDirs();
-}
-
-wxDirItemData::~wxDirItemData()
-{
-}
-
-void wxDirItemData::SetNewDirName( wxString path )
-{
-    m_path = path;
-    m_name = wxFileNameFromPath( path );
-}
-
-bool wxDirItemData::HasSubDirs()
-{
-    return wxDir(m_path).HasSubDirs();
-}
+static const int ID_SHOW_HIDDEN = 1005;
+static const int ID_GO_HOME = 1006;
 
 //-----------------------------------------------------------------------------
-// wxDirCtrl
+// wxGenericDirDialog
 //-----------------------------------------------------------------------------
 
-IMPLEMENT_DYNAMIC_CLASS(wxDirCtrl,wxTreeCtrl)
-
-BEGIN_EVENT_TABLE(wxDirCtrl,wxTreeCtrl)
-  EVT_TREE_ITEM_EXPANDING     (-1, wxDirCtrl::OnExpandItem)
-  EVT_TREE_ITEM_COLLAPSED     (-1, wxDirCtrl::OnCollapseItem)
-  EVT_TREE_BEGIN_LABEL_EDIT   (-1, wxDirCtrl::OnBeginEditItem)
-  EVT_TREE_END_LABEL_EDIT     (-1, wxDirCtrl::OnEndEditItem)
+IMPLEMENT_DYNAMIC_CLASS(wxGenericDirDialog, wxDialog)
+
+BEGIN_EVENT_TABLE(wxGenericDirDialog, wxDialog)
+    EVT_CLOSE                (wxGenericDirDialog::OnCloseWindow)
+    EVT_BUTTON               (wxID_OK,        wxGenericDirDialog::OnOK)
+    EVT_BUTTON               (ID_NEW,         wxGenericDirDialog::OnNew)
+    EVT_BUTTON               (ID_GO_HOME,     wxGenericDirDialog::OnGoHome)
+    EVT_TREE_KEY_DOWN        (wxID_ANY,       wxGenericDirDialog::OnTreeKeyDown)
+    EVT_TREE_SEL_CHANGED     (wxID_ANY,       wxGenericDirDialog::OnTreeSelected)
+    EVT_TEXT_ENTER           (ID_TEXTCTRL,    wxGenericDirDialog::OnOK)
+    EVT_CHECKBOX             (ID_SHOW_HIDDEN, wxGenericDirDialog::OnShowHidden)
 END_EVENT_TABLE()
 
-wxDirCtrl::wxDirCtrl(void)
+wxGenericDirDialog::wxGenericDirDialog(wxWindow* parent, const wxString& title,
+                                       const wxString& defaultPath, long style,
+                                       const wxPoint& pos, const wxSize& sz,
+                                       const wxString& name)
 {
-    m_showHidden = FALSE;
+    Create(parent, title, defaultPath, style, pos, sz, name);
 }
 
-wxDirCtrl::wxDirCtrl(wxWindow *parent,
-                     const wxWindowID id,
-                     const wxString &WXUNUSED(dir),
-                     const wxPoint& pos,
-                     const wxSize& size,
-                     const long style,
-                     const wxString& name )
-         : wxTreeCtrl( parent, id, pos, size, style, wxDefaultValidator, name )
-{
-#ifndef __WXMSW__
-    m_imageListNormal = new wxImageList(16, 16, TRUE);
-    m_imageListNormal->Add(wxICON(icon1));
-    m_imageListNormal->Add(wxICON(icon2));
-    SetImageList(m_imageListNormal);
-#endif // !MSW
-
-    m_showHidden = FALSE;
-    m_rootId = AddRoot( _("Sections") );
-    SetItemHasChildren(m_rootId);
-    Expand(m_rootId); // automatically expand first level
-}
-
-/* Quick macro. Don't worry, I'll #undef it later */
-#define ADD_SECTION(a,b) \
-  if (wxPathExists((a))) { m_paths.Add( (a) ); m_names.Add( (b) ); };
-
-void wxDirCtrl::SetupSections()
+bool wxGenericDirDialog::Create(wxWindow* parent,
+                                const wxString& title,
+                                const wxString& defaultPath, long style,
+                                const wxPoint& pos,
+                                const wxSize& sz,
+                                const wxString& name)
 {
-  wxString home;
+    wxBusyCursor cursor;
 
-  m_paths.Clear();
-  m_names.Clear();
-#ifdef __WXMSW__
-  // better than nothing
-  ADD_SECTION(wxT("c:\\"), _("My Harddisk") )
-#else
-  ADD_SECTION(wxT("/"), _("The Computer") )
-  wxGetHomeDir(&home);
-  ADD_SECTION(home, _("My Home") )
-  ADD_SECTION(wxT("/mnt"), _("Mounted Devices") )
-  ADD_SECTION(wxT("/usr/local"), _("User Local") )
-  ADD_SECTION(wxT("/usr"), _("User") )
-  ADD_SECTION(wxT("/var"), _("Variables") )
-  ADD_SECTION(wxT("/etc"), _("Etcetera") )
-  ADD_SECTION(wxT("/tmp"), _("Temporary") )
-#endif
-}
-#undef ADD_SECTION
+    parent = GetParentForModalDialog(parent, style);
 
-void wxDirCtrl::CreateItems(const wxTreeItemId &parent)
-{
-    wxTreeItemId id;
-    wxDirItemData *dir_item;
+    if (!wxDirDialogBase::Create(parent, title, defaultPath, style, pos, sz, name))
+        return false;
 
-//  wxASSERT(m_paths.Count() == m_names.Count());  ?
+    m_path = defaultPath;
+    if (m_path == wxT("~"))
+        wxGetHomeDir(&m_path);
+    if (m_path == wxT("."))
+        m_path = wxGetCwd();
 
-    size_t count = m_paths.GetCount();
-    for ( size_t i=0; i<count; i++)
-    {
-        dir_item = new wxDirItemData(m_paths[i],m_names[i]);
-#ifdef __WXMSW__
-        id = AppendItem( parent, m_names[i], -1, -1, dir_item);
-#else
-        id = AppendItem( parent, m_names[i], 0, -1, dir_item);
-        SetItemImage( id, 1, wxTreeItemIcon_Expanded );
-#endif
-        if (dir_item->m_hasSubDirs) SetItemHasChildren(id);
-    }
-}
+    wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );
 
-void wxDirCtrl::OnBeginEditItem(wxTreeEvent &event)
-{
-    // don't rename the main entry "Sections"
-    if (event.GetItem() == m_rootId)
-    {
-        event.Veto();
-        return;
-    }
+    // smartphones does not support or do not waste space for wxButtons
+#if defined(__SMARTPHONE__)
 
-    // don't rename the individual sections
-    if (GetParent( event.GetItem() ) == m_rootId)
-    {
-        event.Veto();
-        return;
-    }
-}
+    wxMenu *dirMenu = new wxMenu;
+    dirMenu->Append(ID_GO_HOME, _("Home"));
 
-void wxDirCtrl::OnEndEditItem(wxTreeEvent &event)
-{
-    if ((event.GetLabel().IsEmpty()) ||
-        (event.GetLabel() == _(".")) ||
-        (event.GetLabel() == _("..")) ||
-        (event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
+    if (!HasFlag(wxDD_DIR_MUST_EXIST))
     {
-        wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
-        dialog.ShowModal();
-        event.Veto();
-        return;
+        dirMenu->Append(ID_NEW, _("New directory"));
     }
 
-    wxTreeItemId id = event.GetItem();
-    wxDirItemData *data = (wxDirItemData*)GetItemData( id );
-    wxASSERT( data );
-
-    wxString new_name( wxPathOnly( data->m_path ) );
-    new_name += wxT("/");
-    new_name += event.GetLabel();
+    dirMenu->AppendCheckItem(ID_SHOW_HIDDEN, _("Show hidden directories"));
+    dirMenu->AppendSeparator();
+    dirMenu->Append(wxID_CANCEL, _("Cancel"));
 
-    wxLogNull log;
+#else
 
-    if (wxFileExists(new_name))
-    {
-        wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
-        dialog.ShowModal();
-        event.Veto();
-    }
+    // 0) 'New' and 'Home' Buttons
+    wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
 
-    if (wxRenameFile(data->m_path,new_name))
-    {
-        data->SetNewDirName( new_name );
-    }
-    else
-    {
-        wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
-        dialog.ShowModal();
-        event.Veto();
-    }
-}
+    // VS: 'Home directory' concept is unknown to MS-DOS
+#if !defined(__DOS__)
+    wxBitmapButton* homeButton =
+        new wxBitmapButton(this, ID_GO_HOME,
+                           wxArtProvider::GetBitmap(wxART_GO_HOME, wxART_BUTTON));
+    buttonsizer->Add( homeButton, 0, wxLEFT|wxRIGHT, 10 );
+#endif
 
-void wxDirCtrl::OnExpandItem(wxTreeEvent &event)
-{
-    if (event.GetItem() == m_rootId)
+    // I'm not convinced we need a New button, and we tend to get annoying
+    // accidental-editing with label editing enabled.
+    if (!HasFlag(wxDD_DIR_MUST_EXIST))
     {
-        SetupSections();
-        CreateItems(m_rootId);
-        return;
+        wxBitmapButton* newButton =
+            new wxBitmapButton(this, ID_NEW,
+                            wxArtProvider::GetBitmap(wxART_NEW_DIR, wxART_BUTTON));
+        buttonsizer->Add( newButton, 0, wxRIGHT, 10 );
+#if wxUSE_TOOLTIPS
+        newButton->SetToolTip(_("Create new directory"));
+#endif
     }
 
-    // This may take a longish time. Go to busy cursor
-    wxBeginBusyCursor();
-
-    wxDirItemData *data = (wxDirItemData *)GetItemData(event.GetItem());
-    wxASSERT(data);
-
-    m_paths.Clear();
-    m_names.Clear();
-
-    wxDir dir(data->m_path);
-
-    wxString filename;
-    bool cont = dir.GetFirst(&filename, "", wxDIR_DIRS | wxDIR_HIDDEN);
-    while ( cont )
-    {
-        m_paths.Add(data->m_path);
-        m_names.Add(filename);
+#if wxUSE_TOOLTIPS
+    homeButton->SetToolTip(_("Go to home directory"));
+#endif
 
-        cont = dir.GetNext(&filename);
-    }
+    topsizer->Add( buttonsizer, 0, wxTOP | wxALIGN_RIGHT, 10 );
 
-    CreateItems( event.GetItem() );
-    SortChildren( event.GetItem() );
+#endif // __SMARTPHONE__/!__SMARTPHONE__
 
-    wxEndBusyCursor();
-}
+    // 1) dir ctrl
+    m_dirCtrl = NULL; // this is necessary, event handler called from
+                      // wxGenericDirCtrl would crash otherwise!
+    long dirStyle = wxDIRCTRL_DIR_ONLY | wxDEFAULT_CONTROL_BORDER;
 
-void wxDirCtrl::OnCollapseItem(wxTreeEvent &event )
-{
-    wxTreeItemId child, parent = event.GetItem();
-    long cookie;
-    /* Workaround because DeleteChildren has disapeared (why?) and
-     * CollapseAndReset doesn't work as advertised (deletes parent too) */
-    child = GetFirstChild(parent, cookie);
-    while (child.IsOk())
+#ifdef __WXMSW__
+    if (!HasFlag(wxDD_DIR_MUST_EXIST))
     {
-        Delete(child);
-        /* Not GetNextChild below, because the cookie mechanism can't
-         * handle disappearing children! */
-        child = GetFirstChild(parent, cookie);
+        // Only under Windows do we need the wxTR_EDIT_LABEL tree control style
+        // before we can call EditLabel (required for "New directory")
+        dirStyle |= wxDIRCTRL_EDIT_LABELS;
     }
-}
-
-//-----------------------------------------------------------------------------
-// wxDirDialog
-//-----------------------------------------------------------------------------
-
-
-IMPLEMENT_DYNAMIC_CLASS( wxDirDialog, wxDialog )
-
-BEGIN_EVENT_TABLE( wxDirDialog, wxDialog )
-  EVT_TREE_KEY_DOWN        (ID_DIRCTRL,   wxDirDialog::OnTreeKeyDown)
-  EVT_TREE_SEL_CHANGED     (ID_DIRCTRL,   wxDirDialog::OnTreeSelected)
-  EVT_SIZE                 (              wxDirDialog::OnSize)
-  EVT_BUTTON               (ID_OK,        wxDirDialog::OnOK)
-  EVT_BUTTON               (ID_CANCEL,    wxDirDialog::OnCancel)
-  EVT_BUTTON               (ID_NEW,       wxDirDialog::OnNew)
-  EVT_TEXT_ENTER           (ID_TEXTCTRL,  wxDirDialog::OnOK)
-  //  EVT_CHECKBOX             (ID_CHECK,     wxDirDialog::OnCheck)
-END_EVENT_TABLE()
+#endif
 
-wxDirDialog::wxDirDialog(wxWindow *parent,
-                         const wxString& message,
-                         const wxString& defaultPath,
-                         long style,
-                         const wxPoint& pos)
-           : wxDialog(parent, -1, message, pos, wxSize(300,300),
-                      wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER)
-{
-    m_message = message;
-    m_dialogStyle = style;
-    m_parent = parent;
+    m_dirCtrl = new wxGenericDirCtrl(this, ID_DIRCTRL,
+                                     m_path, wxDefaultPosition,
+                                     wxSize(300, 200),
+                                     dirStyle);
 
-    m_path = defaultPath;
+    wxSizerFlags flagsBorder2;
+    flagsBorder2.DoubleBorder(wxTOP | wxLEFT | wxRIGHT);
 
-    wxBeginBusyCursor();
+    topsizer->Add(m_dirCtrl, wxSizerFlags(flagsBorder2).Proportion(1).Expand());
 
-    wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );
-
-    // 1) dir ctrl
-    m_dir = new wxDirCtrl( this, ID_DIRCTRL, _T("/"),
-                           wxDefaultPosition,
-                           wxSize(200,200),
-                           wxTR_HAS_BUTTONS |
-                           wxSUNKEN_BORDER |
-                           wxTR_EDIT_LABELS );
-    topsizer->Add( m_dir, 1, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, 10 );
+#ifndef __SMARTPHONE__
+    // TODO: Make this an option depending on a flag?
+    wxCheckBox *
+        check = new wxCheckBox(this, ID_SHOW_HIDDEN, _("Show &hidden directories"));
+    topsizer->Add(check, wxSizerFlags(flagsBorder2).Right());
+#endif // !__SMARTPHONE__
 
     // 2) text ctrl
     m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition );
-    topsizer->Add( m_input, 0, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, 10 );
+    topsizer->Add(m_input, wxSizerFlags(flagsBorder2).Expand());
 
-    // m_check = new wxCheckBox( this, ID_CHECK, _("Show hidden") );
-    // m_check->SetValue(TRUE);
+    // 3) buttons if any
+    wxSizer *buttonSizer = CreateSeparatedButtonSizer(wxOK | wxCANCEL);
+    if ( buttonSizer )
+    {
+        topsizer->Add(buttonSizer, wxSizerFlags().Expand().DoubleBorder());
+    }
 
-#if wxUSE_STATLINE
-    // 3) static line
-    topsizer->Add( new wxStaticLine( this, -1 ), 0, wxEXPAND | wxLEFT|wxRIGHT|wxTOP, 10 );
+#ifdef __SMARTPHONE__
+    // overwrite menu set by CreateSeparatedButtonSizer() call above
+    SetRightMenu(wxID_ANY, _("Options"), dirMenu);
 #endif
 
-    // 4) buttons
-    wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
-    m_ok = new wxButton( this, ID_OK, _("OK") );
-    buttonsizer->Add( m_ok, 0, wxLEFT|wxRIGHT, 10 );
-    m_cancel = new wxButton( this, ID_CANCEL, _("Cancel") );
-    buttonsizer->Add( m_cancel, 0, wxLEFT|wxRIGHT, 10 );
-    m_new = new wxButton( this, ID_NEW, _("New...") );
-    buttonsizer->Add( m_new, 0, wxLEFT|wxRIGHT, 10 );
-
-    topsizer->Add( buttonsizer, 0, wxALL | wxCENTER, 10 );
-
-    m_ok->SetDefault();
-    m_dir->SetFocus();
+    m_input->SetFocus();
 
-    SetAutoLayout( TRUE );
+    SetAutoLayout( true );
     SetSizer( topsizer );
 
     topsizer->SetSizeHints( this );
@@ -430,133 +198,121 @@ wxDirDialog::wxDirDialog(wxWindow *parent,
 
     Centre( wxBOTH );
 
-    if (m_path == wxT("~"))
-        wxGetHomeDir( &m_path );
+    return true;
+}
 
-    // choose the directory corresponding to defaultPath in the tree
-    // VZ: using wxStringTokenizer is probably unsafe here (escaped slashes
-    //     will not be processed correctly...)
-    wxStringTokenizer tk(m_path, wxFILE_SEP_PATH, wxTOKEN_STRTOK);
+void wxGenericDirDialog::EndModal(int retCode)
+{
+    // before proceeding, change the current working directory if user asked so
+    if (retCode == wxID_OK && HasFlag(wxDD_CHANGE_DIR))
+        wxSetWorkingDirectory(m_path);
 
-    wxString path;
+    wxDialog::EndModal(retCode);
+}
 
-    long cookie = 0;
-    // default to root dir
-    wxTreeItemId item = m_dir->GetFirstChild(m_dir->GetRootItem(), cookie);
-    
-    if (!m_path.IsEmpty() && (m_path != wxT("/")) && (m_dir->m_paths.Count() > 1))
+void wxGenericDirDialog::OnCloseWindow(wxCloseEvent& WXUNUSED(event))
+{
+    EndModal(wxID_CANCEL);
+}
+
+void wxGenericDirDialog::OnOK(wxCommandEvent& WXUNUSED(event))
+{
+    m_path = m_input->GetValue();
+
+    // Does the path exist? (User may have typed anything in m_input)
+    if (wxDirExists(m_path))
     {
-        size_t count = m_dir->m_paths.GetCount();
-        for ( size_t i=1; i<count; i++)
-        {
-            if (m_path.Find( m_dir->m_paths[i] ) == 0)
-            {
-                path = m_dir->m_paths[i];
-                
-                for (size_t j = 0; j < i; j++)
-                   item = m_dir->GetNextChild(m_dir->GetRootItem(), cookie);
-                
-                wxStringTokenizer tk2(path, wxFILE_SEP_PATH, wxTOKEN_STRTOK);
-                for (size_t h = 0; h < tk2.CountTokens(); h++)
-                   tk.GetNextToken();
-                
-                break;
-            }
-        }
+        // OK, path exists, we're done.
+        EndModal(wxID_OK);
+        return;
     }
-    while ( tk.HasMoreTokens() && item.IsOk() )
-    {
-        path << wxFILE_SEP_PATH << tk.GetNextToken();
 
-        m_dir->Expand(item);
+    // Interact with user, find out if the dir is a typo or to be created
+    wxString msg;
+    msg.Printf(_("The directory '%s' does not exist\nCreate it now?"),
+               m_path.c_str());
+    wxMessageDialog dialog(this, msg, _("Directory does not exist"),
+                           wxYES_NO | wxICON_WARNING);
 
-        wxTreeItemId child = m_dir->GetFirstChild(item, cookie);
-        while ( child.IsOk() )
+    if ( dialog.ShowModal() == wxID_YES )
+    {
+        // Okay, let's make it
+        wxLogNull log;
+        if (wxMkdir(m_path))
         {
-            wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(child);
-            if ( data->m_path == path )
-                break;
-
-            child = m_dir->GetNextChild(item, cookie);
+            // The new dir was created okay.
+            EndModal(wxID_OK);
+            return;
+        }
+        else
+        {
+            // Trouble...
+            msg.Printf(_("Failed to create directory '%s'\n(Do you have the required permissions?)"),
+                       m_path.c_str());
+            wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK | wxICON_ERROR);
+            errmsg.ShowModal();
+            // We still don't have a valid dir. Back to the main dialog.
         }
-
-        item = child;
     }
+    // User has answered NO to create dir.
+}
 
-    if ( item.IsOk() )
-    {
-        m_dir->Expand(item);
-        m_dir->SelectItem(item);
-        m_dir->EnsureVisible(item);
-    }
+void wxGenericDirDialog::SetPath(const wxString& path)
+{
+    m_dirCtrl->SetPath(path);
+    m_path = path;
+}
 
-    wxEndBusyCursor();
+wxString wxGenericDirDialog::GetPath(void) const
+{
+    return m_path;
 }
 
-int wxDirDialog::ShowModal()
+int wxGenericDirDialog::ShowModal()
 {
     m_input->SetValue( m_path );
     return wxDialog::ShowModal();
 }
 
-void wxDirDialog::OnTreeSelected( wxTreeEvent &event )
+void wxGenericDirDialog::OnTreeSelected( wxTreeEvent &event )
 {
-    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(event.GetItem());
+    if (!m_dirCtrl)
+        return;
+
+    wxTreeItemId item = event.GetItem();
+
+    wxDirItemData *data = NULL;
+
+    if(item.IsOk())
+        data = (wxDirItemData*)m_dirCtrl->GetTreeCtrl()->GetItemData(item);
+
     if (data)
        m_input->SetValue( data->m_path );
-};
+}
 
-void wxDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
+void wxGenericDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
 {
-    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(m_dir->GetSelection());
+    if (!m_dirCtrl)
+        return;
+
+    wxDirItemData *data = (wxDirItemData*)m_dirCtrl->GetTreeCtrl()->GetItemData(m_dirCtrl->GetTreeCtrl()->GetSelection());
     if (data)
         m_input->SetValue( data->m_path );
-};
-
-void wxDirDialog::OnOK( wxCommandEvent& WXUNUSED(event) )
-{
-  m_path = m_input->GetValue();
-  // Does the path exist? (User may have typed anything in m_input)
-  if (wxPathExists(m_path)) {
-    // OK, path exists, we're done.
-    EndModal(wxID_OK);
-    return;
-  }
-  // Interact with user, find out if the dir is a typo or to be created
-  wxString msg( _("The directory ") );
-  msg = msg + m_path;
-  msg = msg + _("\ndoes not exist\nCreate it now?") ;
-  wxMessageDialog dialog(this, msg, _("Directory does not exist"), wxYES_NO | wxICON_WARNING );
-  if ( dialog.ShowModal() == wxID_YES ) {
-    // Okay, let's make it
-    wxLogNull log;
-    if (wxMkdir(m_path)) {
-      // The new dir was created okay.
-      EndModal(wxID_OK);
-      return;
-    }
-    else {
-      // Trouble...
-      msg = _("Failed to create directory ")+m_path+
-  _("\n(Do you have the required permissions?)");
-      wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK | wxICON_ERROR);
-      errmsg.ShowModal();
-      // We still don't have a valid dir. Back to the main dialog.
-    }
-  }
-  // User has answered NO to create dir.
 }
 
-void wxDirDialog::OnCancel( wxCommandEvent& WXUNUSED(event) )
+void wxGenericDirDialog::OnShowHidden( wxCommandEvent& event )
 {
-  EndModal(wxID_CANCEL);
+    if (!m_dirCtrl)
+        return;
+
+    m_dirCtrl->ShowHidden( event.GetInt() != 0 );
 }
 
-void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
+void wxGenericDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
 {
-    wxTreeItemId id = m_dir->GetSelection();
-    if ((id == m_dir->GetRootItem()) ||
-        (m_dir->GetParent(id) == m_dir->GetRootItem()))
+    wxTreeItemId id = m_dirCtrl->GetTreeCtrl()->GetSelection();
+    if ((id == m_dirCtrl->GetTreeCtrl()->GetRootItem()) ||
+        (m_dirCtrl->GetTreeCtrl()->GetItemParent(id) == m_dirCtrl->GetTreeCtrl()->GetRootItem()))
     {
         wxMessageDialog msg(this, _("You cannot add a new directory to this section."),
                             _("Create directory"), wxOK | wxICON_INFORMATION );
@@ -564,29 +320,31 @@ void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
         return;
     }
 
-    wxTreeItemId parent = m_dir->GetParent( id );
-    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData( parent );
+    wxTreeItemId parent = id ; // m_dirCtrl->GetTreeCtrl()->GetItemParent( id );
+    wxDirItemData *data = (wxDirItemData*)m_dirCtrl->GetTreeCtrl()->GetItemData( parent );
     wxASSERT( data );
 
-    wxString new_name( wxT("NewName") );
+    wxString new_name( _("NewName") );
     wxString path( data->m_path );
-    path += wxT("/");
+    if (!wxEndsWithPathSeparator(path))
+        path += wxFILE_SEP_PATH;
     path += new_name;
-    if (wxFileExists(path))
+    if (wxDirExists(path))
     {
         // try NewName0, NewName1 etc.
         int i = 0;
         do {
-            new_name = wxT("NewName");
+            new_name = _("NewName");
             wxString num;
             num.Printf( wxT("%d"), i );
             new_name += num;
 
             path = data->m_path;
-            path += wxT("/");
+            if (!wxEndsWithPathSeparator(path))
+                path += wxFILE_SEP_PATH;
             path += new_name;
             i++;
-        } while (wxFileExists(path));
+        } while (wxDirExists(path));
     }
 
     wxLogNull log;
@@ -597,17 +355,18 @@ void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
         return;
     }
 
-    wxDirItemData *new_data = new wxDirItemData( path, new_name );
-    wxTreeItemId new_id = m_dir->AppendItem( parent, new_name, 0, 1, new_data );
-    m_dir->EnsureVisible( new_id );
-    m_dir->EditLabel( new_id );
+    wxDirItemData *new_data = new wxDirItemData( path, new_name, true );
+
+    // TODO: THIS CODE DOESN'T WORK YET. We need to avoid duplication of the first child
+    // of the parent.
+    wxTreeItemId new_id = m_dirCtrl->GetTreeCtrl()->AppendItem( parent, new_name, 0, 0, new_data );
+    m_dirCtrl->GetTreeCtrl()->EnsureVisible( new_id );
+    m_dirCtrl->GetTreeCtrl()->EditLabel( new_id );
 }
 
-/*
-void wxDirDialog::OnCheck( wxCommandEvent& WXUNUSED(event) )
+void wxGenericDirDialog::OnGoHome(wxCommandEvent& WXUNUSED(event))
 {
-  printf("Checkbox clicked: %s\n", ( m_check->GetValue() ? "on" : "off" ) );
+    SetPath(wxGetUserHome());
 }
-*/
 
 #endif // wxUSE_DIRDLG