Fix for assert after left click on tree in generic dir dialog in native MSW build...
[wxWidgets.git] / src / generic / dirdlgg.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: dirdlg.cpp
3 // Purpose: wxDirDialog
4 // Author: Harm van der Heijden, Robert Roebling & Julian Smart
5 // Modified by:
6 // Created: 12/12/98
7 // RCS-ID: $Id$
8 // Copyright: (c) Harm van der Heijden, Robert Roebling, Julian Smart
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 #if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
13 #pragma implementation "dirdlgg.h"
14 #endif
15
16
17 // For compilers that support precompilation, includes "wx.h".
18 #include "wx/wxprec.h"
19
20 #ifdef __BORLANDC__
21 #pragma hdrstop
22 #endif
23
24 #include "wx/defs.h"
25
26 #if wxUSE_DIRDLG
27
28 #ifndef WX_PRECOMP
29 #include "wx/textctrl.h"
30 #include "wx/button.h"
31 #include "wx/checkbox.h"
32 #include "wx/sizer.h"
33 #include "wx/intl.h"
34 #include "wx/log.h"
35 #include "wx/msgdlg.h"
36 #endif
37
38 #include "wx/statline.h"
39 #include "wx/dirctrl.h"
40 #include "wx/generic/dirdlgg.h"
41 #include "wx/artprov.h"
42 #include "wx/bmpbuttn.h"
43
44 // ----------------------------------------------------------------------------
45 // constants
46 // ----------------------------------------------------------------------------
47
48 static const int ID_DIRCTRL = 1000;
49 static const int ID_TEXTCTRL = 1001;
50 static const int ID_OK = 1002;
51 static const int ID_CANCEL = 1003;
52 static const int ID_NEW = 1004;
53 static const int ID_SHOW_HIDDEN = 1005;
54 static const int ID_GO_HOME = 1006;
55
56 // ---------------------------------------------------------------------------
57 // macros
58 // ---------------------------------------------------------------------------
59
60 /* Macro for avoiding #ifdefs when value have to be different depending on size of
61 device we display on - take it from something like wxDesktopPolicy in the future
62 */
63
64 #if defined(__SMARTPHONE__)
65 #define wxLARGESMALL(large,small) small
66 #else
67 #define wxLARGESMALL(large,small) large
68 #endif
69
70 //-----------------------------------------------------------------------------
71 // wxGenericDirDialog
72 //-----------------------------------------------------------------------------
73
74 IMPLEMENT_DYNAMIC_CLASS(wxGenericDirDialog, wxDialog)
75
76 BEGIN_EVENT_TABLE(wxGenericDirDialog, wxDialog)
77 EVT_CLOSE (wxGenericDirDialog::OnCloseWindow)
78 EVT_BUTTON (wxID_OK, wxGenericDirDialog::OnOK)
79 EVT_BUTTON (ID_NEW, wxGenericDirDialog::OnNew)
80 EVT_BUTTON (ID_GO_HOME, wxGenericDirDialog::OnGoHome)
81 EVT_TREE_KEY_DOWN (wxID_ANY, wxGenericDirDialog::OnTreeKeyDown)
82 EVT_TREE_SEL_CHANGED (wxID_ANY, wxGenericDirDialog::OnTreeSelected)
83 EVT_TEXT_ENTER (ID_TEXTCTRL, wxGenericDirDialog::OnOK)
84 EVT_CHECKBOX (ID_SHOW_HIDDEN, wxGenericDirDialog::OnShowHidden)
85 END_EVENT_TABLE()
86
87 wxGenericDirDialog::wxGenericDirDialog(wxWindow* parent, const wxString& title,
88 const wxString& defaultPath, long style,
89 const wxPoint& pos, const wxSize& sz,
90 const wxString& name):
91 wxDialog(parent, ID_DIRCTRL, title, pos, sz, style, name)
92 {
93 wxBusyCursor cursor;
94
95 m_path = defaultPath;
96 if (m_path == wxT("~"))
97 wxGetHomeDir(&m_path);
98 if (m_path == wxT("."))
99 m_path = wxGetCwd();
100
101 wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );
102
103 // smart phones does not support or do not waste space for wxButtons
104 #if defined(__SMARTPHONE__)
105
106 wxMenu *dirMenu = new wxMenu;
107 dirMenu->Append(ID_GO_HOME, _("Home"));
108
109 if (style & wxDD_NEW_DIR_BUTTON)
110 {
111 dirMenu->Append(ID_NEW, _("New directory"));
112 }
113
114 dirMenu->AppendCheckItem(ID_SHOW_HIDDEN, _("Show hidden directories"));
115 dirMenu->AppendSeparator();
116 dirMenu->Append(wxID_CANCEL, _("Cancel"));
117
118 SetRightMenu(wxID_ANY, _("Options"), dirMenu);
119
120 #else
121
122 // 0) 'New' and 'Home' Buttons
123 wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
124
125 // VS: 'Home directory' concept is unknown to MS-DOS
126 #if !defined(__DOS__)
127 wxBitmapButton* homeButton =
128 new wxBitmapButton(this, ID_GO_HOME,
129 wxArtProvider::GetBitmap(wxART_GO_HOME, wxART_BUTTON));
130 buttonsizer->Add( homeButton, 0, wxLEFT|wxRIGHT, 10 );
131 #endif
132
133 // I'm not convinced we need a New button, and we tend to get annoying
134 // accidental-editing with label editing enabled.
135 if (style & wxDD_NEW_DIR_BUTTON)
136 {
137 wxBitmapButton* newButton =
138 new wxBitmapButton(this, ID_NEW,
139 wxArtProvider::GetBitmap(wxART_NEW_DIR, wxART_BUTTON));
140 buttonsizer->Add( newButton, 0, wxRIGHT, 10 );
141 #if wxUSE_TOOLTIPS
142 newButton->SetToolTip(_("Create new directory"));
143 #endif
144 }
145
146 #if wxUSE_TOOLTIPS
147 homeButton->SetToolTip(_("Go to home directory"));
148 #endif
149
150 topsizer->Add( buttonsizer, 0, wxTOP | wxALIGN_RIGHT, 10 );
151
152 #endif // __SMARTPHONE__/!__SMARTPHONE__
153
154 // 1) dir ctrl
155 m_dirCtrl = NULL; // this is neccessary, event handler called from
156 // wxGenericDirCtrl would crash otherwise!
157 long dirStyle = wxDIRCTRL_DIR_ONLY|wxSUNKEN_BORDER;
158
159 #ifdef __WXMSW__
160 if (style & wxDD_NEW_DIR_BUTTON)
161 {
162 // Only under Windows do we need the wxTR_EDIT_LABEL tree control style
163 // before we can call EditLabel (required for "New directory")
164 dirStyle |= wxDIRCTRL_EDIT_LABELS;
165 }
166 #endif
167
168 m_dirCtrl = new wxGenericDirCtrl(this, ID_DIRCTRL,
169 m_path, wxDefaultPosition,
170 wxSize(300, 200),
171 dirStyle);
172
173 topsizer->Add( m_dirCtrl, 1, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, wxLARGESMALL(10,0) );
174
175 #ifndef __SMARTPHONE__
176 // Make the an option depending on a flag?
177 wxCheckBox* check = new wxCheckBox( this, ID_SHOW_HIDDEN, _("Show hidden directories") );
178 topsizer->Add( check, 0, wxLEFT|wxRIGHT|wxTOP | wxALIGN_RIGHT, 10 );
179 #endif // !__SMARTPHONE__
180
181 // 2) text ctrl
182 m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition );
183 topsizer->Add( m_input, 0, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, wxLARGESMALL(10,0) );
184
185 #ifndef __SMARTPHONE__
186
187 #if wxUSE_STATLINE
188 // 3) Static line
189 topsizer->Add( new wxStaticLine( this, wxID_ANY ), 0, wxEXPAND | wxLEFT|wxRIGHT|wxTOP, 10 );
190 #endif
191
192 // 4) Buttons
193 buttonsizer = new wxBoxSizer( wxHORIZONTAL );
194
195 // OK and Cancel button should be at the right bottom
196 wxButton* okButton = new wxButton(this, wxID_OK);
197 buttonsizer->Add( okButton, 0, wxLEFT|wxRIGHT, 10 );
198 wxButton* cancelButton = new wxButton(this, wxID_CANCEL);
199 buttonsizer->Add( cancelButton, 0, wxLEFT|wxRIGHT, 10 );
200
201 topsizer->Add( buttonsizer, 0, wxLEFT|wxTOP|wxBOTTOM | wxALIGN_RIGHT, 10 );
202
203 okButton->SetDefault();
204
205 #endif // !__SMARTPHONE__
206
207 m_input->SetFocus();
208
209 SetAutoLayout( true );
210 SetSizer( topsizer );
211
212 topsizer->SetSizeHints( this );
213 topsizer->Fit( this );
214
215 Centre( wxBOTH );
216 }
217
218 void wxGenericDirDialog::OnCloseWindow(wxCloseEvent& WXUNUSED(event))
219 {
220 EndModal(wxID_CANCEL);
221 }
222
223 void wxGenericDirDialog::OnOK(wxCommandEvent& WXUNUSED(event))
224 {
225 m_path = m_input->GetValue();
226 // Does the path exist? (User may have typed anything in m_input)
227 if (wxPathExists(m_path)) {
228 // OK, path exists, we're done.
229 EndModal(wxID_OK);
230 return;
231 }
232 // Interact with user, find out if the dir is a typo or to be created
233 wxString msg;
234 msg.Printf(_("The directory '%s' does not exist\nCreate it now?"),
235 m_path.c_str());
236 wxMessageDialog dialog(this, msg, _("Directory does not exist"),
237 wxYES_NO | wxICON_WARNING);
238
239 if ( dialog.ShowModal() == wxID_YES ) {
240 // Okay, let's make it
241 wxLogNull log;
242 if (wxMkdir(m_path)) {
243 // The new dir was created okay.
244 EndModal(wxID_OK);
245 return;
246 }
247 else {
248 // Trouble...
249 msg.Printf(_("Failed to create directory '%s'\n(Do you have the required permissions?)"),
250 m_path.c_str());
251 wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK | wxICON_ERROR);
252 errmsg.ShowModal();
253 // We still don't have a valid dir. Back to the main dialog.
254 }
255 }
256 // User has answered NO to create dir.
257 }
258
259 void wxGenericDirDialog::SetPath(const wxString& path)
260 {
261 m_dirCtrl->SetPath(path);
262 m_path = path;
263 }
264
265 wxString wxGenericDirDialog::GetPath(void) const
266 {
267 return m_path;
268 }
269
270 int wxGenericDirDialog::ShowModal()
271 {
272 m_input->SetValue( m_path );
273 return wxDialog::ShowModal();
274 }
275
276 void wxGenericDirDialog::OnTreeSelected( wxTreeEvent &event )
277 {
278 if (!m_dirCtrl)
279 return;
280
281 wxTreeItemId item = event.GetItem();
282
283 wxDirItemData *data = NULL;
284
285 if(item.IsOk())
286 data = (wxDirItemData*)m_dirCtrl->GetTreeCtrl()->GetItemData(item);
287
288 if (data)
289 m_input->SetValue( data->m_path );
290 };
291
292 void wxGenericDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
293 {
294 if (!m_dirCtrl)
295 return;
296
297 wxDirItemData *data = (wxDirItemData*)m_dirCtrl->GetTreeCtrl()->GetItemData(m_dirCtrl->GetTreeCtrl()->GetSelection());
298 if (data)
299 m_input->SetValue( data->m_path );
300 };
301
302 void wxGenericDirDialog::OnShowHidden( wxCommandEvent& event )
303 {
304 if (!m_dirCtrl)
305 return;
306
307 m_dirCtrl->ShowHidden( event.GetInt() != 0 );
308 }
309
310 void wxGenericDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
311 {
312 wxTreeItemId id = m_dirCtrl->GetTreeCtrl()->GetSelection();
313 if ((id == m_dirCtrl->GetTreeCtrl()->GetRootItem()) ||
314 (m_dirCtrl->GetTreeCtrl()->GetItemParent(id) == m_dirCtrl->GetTreeCtrl()->GetRootItem()))
315 {
316 wxMessageDialog msg(this, _("You cannot add a new directory to this section."),
317 _("Create directory"), wxOK | wxICON_INFORMATION );
318 msg.ShowModal();
319 return;
320 }
321
322 wxTreeItemId parent = id ; // m_dirCtrl->GetTreeCtrl()->GetItemParent( id );
323 wxDirItemData *data = (wxDirItemData*)m_dirCtrl->GetTreeCtrl()->GetItemData( parent );
324 wxASSERT( data );
325
326 wxString new_name( _("NewName") );
327 wxString path( data->m_path );
328 if (!wxEndsWithPathSeparator(path))
329 path += wxFILE_SEP_PATH;
330 path += new_name;
331 if (wxFileExists(path))
332 {
333 // try NewName0, NewName1 etc.
334 int i = 0;
335 do {
336 new_name = _("NewName");
337 wxString num;
338 num.Printf( wxT("%d"), i );
339 new_name += num;
340
341 path = data->m_path;
342 if (!wxEndsWithPathSeparator(path))
343 path += wxFILE_SEP_PATH;
344 path += new_name;
345 i++;
346 } while (wxFileExists(path));
347 }
348
349 wxLogNull log;
350 if (!wxMkdir(path))
351 {
352 wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
353 dialog.ShowModal();
354 return;
355 }
356
357 wxDirItemData *new_data = new wxDirItemData( path, new_name, true );
358
359 // TODO: THIS CODE DOESN'T WORK YET. We need to avoid duplication of the first child
360 // of the parent.
361 wxTreeItemId new_id = m_dirCtrl->GetTreeCtrl()->AppendItem( parent, new_name, 0, 0, new_data );
362 m_dirCtrl->GetTreeCtrl()->EnsureVisible( new_id );
363 m_dirCtrl->GetTreeCtrl()->EditLabel( new_id );
364 }
365
366 void wxGenericDirDialog::OnGoHome(wxCommandEvent& WXUNUSED(event))
367 {
368 SetPath(wxGetUserHome());
369 }
370
371 #endif // wxUSE_DIRDLG