]> git.saurik.com Git - wxWidgets.git/blob - src/generic/dirctrlg.cpp
projects / wxWidgets.git / blob
? search:


6084a44bd23c88958c70ee91de0f4cf906ac6314
[wxWidgets.git] / src / generic / dirctrlg.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: dirctrlg.cpp
3 // Purpose: wxGenericDirCtrl
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 and Julian Smart
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 #ifdef __GNUG__
13 #pragma implementation "dirctrlg.h"
14 #endif
15
16 // For compilers that support precompilation, includes "wx.h".
17 #include "wx/wxprec.h"
18
19 #ifdef __BORLANDC__
20 #pragma hdrstop
21 #endif
22
23 #if wxUSE_DIRDLG
24
25 #include "wx/utils.h"
26 #include "wx/dialog.h"
27 #include "wx/button.h"
28 #include "wx/layout.h"
29 #include "wx/msgdlg.h"
30 #include "wx/textctrl.h"
31 #include "wx/textdlg.h"
32 #include "wx/filefn.h"
33 #include "wx/cmndata.h"
34 #include "wx/gdicmn.h"
35 #include "wx/intl.h"
36 #include "wx/imaglist.h"
37 #include "wx/icon.h"
38 #include "wx/log.h"
39 #include "wx/sizer.h"
40 #include "wx/tokenzr.h"
41 #include "wx/dir.h"
42 #include "wx/settings.h"
43
44 #if wxUSE_STATLINE
45 #include "wx/statline.h"
46 #endif
47
48 #include "wx/generic/dirctrlg.h"
49
50 #if defined(__WXMAC__)
51 #include "wx/mac/private.h" // includes mac headers
52 #endif
53
54 #ifdef __WXMSW__
55 #include <windows.h>
56
57 // FIXME - Mingw32 1.0 has both _getdrive() and _chdrive(). For now, let's assume
58 // older releases don't, but it should be verified and the checks modified
59 // accordingly.
60 #if !defined(__GNUWIN32__) || \
61 (defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
62 #include <direct.h>
63 #include <stdlib.h>
64 #include <ctype.h>
65 #endif
66
67 #endif
68
69 #ifdef __WXPM__
70
71 #define INCL_BASE
72 #include <os2.h>
73 #include <direct.h>
74 #include <stdlib.h>
75 #include <ctype.h>
76
77 #endif // __WXPM__
78
79 #if defined(__WXMAC__)
80 # include "MoreFilesExtras.h"
81 #endif
82
83 #ifdef __BORLANDC__
84 #include "dos.h"
85 #endif
86
87 // If compiled under Windows, this macro can cause problems
88 #ifdef GetFirstChild
89 #undef GetFirstChild
90 #endif
91
92 /* Closed folder */
93 static const char * icon1_xpm[] = {
94 /* width height ncolors chars_per_pixel */
95 "16 16 6 1",
96 /* colors */
97 " s None c None",
98 ". c #000000",
99 "+ c #c0c0c0",
100 "@ c #808080",
101 "# c #ffff00",
102 "$ c #ffffff",
103 /* pixels */
104 " ",
105 " @@@@@ ",
106 " @#+#+#@ ",
107 " @#+#+#+#@@@@@@ ",
108 " @$$$$$$$$$$$$@.",
109 " @$#+#+#+#+#+#@.",
110 " @$+#+#+#+#+#+@.",
111 " @$#+#+#+#+#+#@.",
112 " @$+#+#+#+#+#+@.",
113 " @$#+#+#+#+#+#@.",
114 " @$+#+#+#+#+#+@.",
115 " @$#+#+#+#+#+#@.",
116 " @@@@@@@@@@@@@@.",
117 " ..............",
118 " ",
119 " "};
120
121 /* Open folder */
122 static const char * icon2_xpm[] = {
123 /* width height ncolors chars_per_pixel */
124 "16 16 6 1",
125 /* colors */
126 " s None c None",
127 ". c #000000",
128 "+ c #c0c0c0",
129 "@ c #808080",
130 "# c #ffff00",
131 "$ c #ffffff",
132 /* pixels */
133 " ",
134 " @@@@@ ",
135 " @$$$$$@ ",
136 " @$#+#+#$@@@@@@ ",
137 " @$+#+#+$$$$$$@.",
138 " @$#+#+#+#+#+#@.",
139 "@@@@@@@@@@@@@#@.",
140 "@$$$$$$$$$$@@+@.",
141 "@$#+#+#+#+##.@@.",
142 " @$#+#+#+#+#+.@.",
143 " @$+#+#+#+#+#.@.",
144 " @$+#+#+#+##@..",
145 " @@@@@@@@@@@@@.",
146 " .............",
147 " ",
148 " "};
149
150 /* File */
151 static const char * icon3_xpm[] = {
152 /* width height ncolors chars_per_pixel */
153 "16 16 3 1",
154 /* colors */
155 " s None c None",
156 ". c #000000",
157 "+ c #ffffff",
158 /* pixels */
159 " ",
160 " ........ ",
161 " .++++++.. ",
162 " .+.+.++.+. ",
163 " .++++++.... ",
164 " .+.+.+++++. ",
165 " .+++++++++. ",
166 " .+.+.+.+.+. ",
167 " .+++++++++. ",
168 " .+.+.+.+.+. ",
169 " .+++++++++. ",
170 " .+.+.+.+.+. ",
171 " .+++++++++. ",
172 " ........... ",
173 " ",
174 " "};
175
176 /* Computer */
177 static const char * icon4_xpm[] = {
178 "16 16 7 1",
179 " s None c None",
180 ". c #808080",
181 "X c #c0c0c0",
182 "o c Black",
183 "O c Gray100",
184 "+ c #008080",
185 "@ c Blue",
186 " ........... ",
187 " .XXXXXXXXXX.o",
188 " .OOOOOOOOO..o",
189 " .OoooooooX..o",
190 " .Oo+...@+X..o",
191 " .Oo+XXX.+X..o",
192 " .Oo+....+X..o",
193 " .Oo++++++X..o",
194 " .OXXXXXXXX.oo",
195 " ..........o.o",
196 " ...........Xo",
197 " .XXXXXXXXXX.o",
198 " .o.o.o.o.o...o",
199 " .oXoXoXoXoXo.o ",
200 ".XOXXXXXXXXX.o ",
201 "............o "};
202
203 /* Drive */
204 static const char * icon5_xpm[] = {
205 "16 16 7 1",
206 " s None c None",
207 ". c #808080",
208 "X c #c0c0c0",
209 "o c Black",
210 "O c Gray100",
211 "+ c Green",
212 "@ c #008000",
213 " ",
214 " ",
215 " ",
216 " ",
217 " ............. ",
218 " .XXXXXXXXXXXX.o",
219 ".OOOOOOOOOOOO..o",
220 ".XXXXXXXXX+@X..o",
221 ".XXXXXXXXXXXX..o",
222 ".X..........X..o",
223 ".XOOOOOOOOOOX..o",
224 "..............o ",
225 " ooooooooooooo ",
226 " ",
227 " ",
228 " "};
229
230 /* CD-ROM */
231 static const char *icon6_xpm[] = {
232 "16 16 10 1",
233 " s None c None",
234 ". c #808080",
235 "X c #c0c0c0",
236 "o c Yellow",
237 "O c Blue",
238 "+ c Black",
239 "@ c Gray100",
240 "# c #008080",
241 "$ c Green",
242 "% c #008000",
243 " ... ",
244 " ..XoX.. ",
245 " .O.XoXXX+ ",
246 " ...O.oXXXX+ ",
247 " .O..X.XXXX+ ",
248 " ....X.+..XXX+",
249 " .XXX.+@+.XXX+",
250 " .X@XX.+.X@@X+",
251 " .....X...#XX@+ ",
252 ".@@@...XXo.O@X+ ",
253 ".@XXX..XXoXOO+ ",
254 ".@++++..XoX+++ ",
255 ".@$%@@XX+++X.+ ",
256 ".............+ ",
257 " ++++++++++++ ",
258 " "};
259
260 /* Floppy */
261 static const char * icon7_xpm[] = {
262 "16 16 7 1",
263 " s None c None",
264 ". c #808080",
265 "X c Gray100",
266 "o c #c0c0c0",
267 "O c Black",
268 "+ c Cyan",
269 "@ c Red",
270 " ......X",
271 " .ooooooO",
272 " .+++++OO",
273 " .++++++O",
274 " .++++++O",
275 " .ooooooO",
276 " .......o....oO",
277 " .oooooo.o.O.XoO",
278 ".XXXXXXXXOOOOOO ",
279 ".ooooooooo@o..O ",
280 ".ooo....oooo..O ",
281 ".o..OOOO...o..O ",
282 ".oooXXXXoooo..O ",
283 ".............O ",
284 " OOOOOOOOOOOO ",
285 " "};
286
287 /* Removeable */
288 static const char * icon8_xpm[] = {
289 "16 16 7 1",
290 " s None c None",
291 ". c #808080",
292 "X c #c0c0c0",
293 "o c Black",
294 "O c Gray100",
295 "+ c Red",
296 "@ c #800000",
297 " ",
298 " ",
299 " ",
300 " ............. ",
301 " .XXXXXXXXXXXX.o",
302 ".OOOOOOOOOOOO..o",
303 ".OXXXXXXXXXXX..o",
304 ".O+@.oooooo.X..o",
305 ".OXXOooooooOX..o",
306 ".OXXXOOOOOOXX..o",
307 ".OXXXXXXXXXXX..o",
308 ".O............o ",
309 " ooooooooooooo ",
310 " ",
311 " ",
312 " "};
313
314
315 #if defined(__DOS__)
316
317 bool wxIsDriveAvailable(const wxString& dirName)
318 {
319 // FIXME_MGL - this method leads to hang up under Watcom for some reason
320 #ifndef __WATCOMC__
321 if ( dirName.Len() == 3 && dirName[1u] == wxT(':') )
322 {
323 wxString dirNameLower(dirName.Lower());
324 // VS: always return TRUE for removable media, since Win95 doesn't
325 // like it when MS-DOS app accesses empty floppy drive
326 return (dirNameLower[0u] == wxT('a') ||
327 dirNameLower[0u] == wxT('b') ||
328 wxPathExists(dirNameLower));
329 }
330 else
331 #endif
332 return TRUE;
333 }
334
335 #elif defined(__WINDOWS__) || defined(__WXPM__)
336
337 int setdrive(int drive)
338 {
339 #if defined(__GNUWIN32__) && \
340 (defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
341 return _chdrive(drive);
342 #else
343 wxChar newdrive[3];
344
345 if (drive < 1 || drive > 31)
346 return -1;
347 newdrive[0] = (wxChar)(wxT('A') + drive - 1);
348 newdrive[1] = wxT(':');
349 newdrive[2] = wxT('\0');
350 #if defined(__WXMSW__)
351 #ifdef __WIN16__
352 if (wxSetWorkingDirectory(newdrive))
353 #else
354 if (::SetCurrentDirectory(newdrive))
355 #endif
356 #else
357 // VA doesn't know what LPSTR is and has its own set
358 if (DosSetCurrentDir((PSZ)newdrive))
359 #endif
360 return 0;
361 else
362 return -1;
363 #endif // !GNUWIN32
364 }
365
366 bool wxIsDriveAvailable(const wxString& dirName)
367 {
368 #ifdef __WIN32__
369 UINT errorMode = SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX);
370 #endif
371 bool success = TRUE;
372
373 // Check if this is a root directory and if so,
374 // whether the drive is avaiable.
375 if (dirName.Len() == 3 && dirName[(size_t)1] == wxT(':'))
376 {
377 wxString dirNameLower(dirName.Lower());
378 #if defined(__GNUWIN32__) && \
379 !(defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
380 success = wxPathExists(dirNameLower);
381 #else
382 int currentDrive = _getdrive();
383 int thisDrive = (int) (dirNameLower[(size_t)0] - 'a' + 1) ;
384 int err = setdrive( thisDrive ) ;
385 setdrive( currentDrive );
386
387 if (err == -1)
388 {
389 success = FALSE;
390 }
391 #endif
392 }
393 #ifdef __WIN32__
394 (void) SetErrorMode(errorMode);
395 #endif
396
397 return success;
398 }
399 #endif // __WINDOWS__ || __WXPM__
400
401 // Function which is called by quick sort. We want to override the default wxArrayString behaviour,
402 // and sort regardless of case.
403 static int LINKAGEMODE wxDirCtrlStringCompareFunction(const void *first, const void *second)
404 {
405 wxString *strFirst = (wxString *)first;
406 wxString *strSecond = (wxString *)second;
407
408 return strFirst->CmpNoCase(*strSecond);
409 }
410
411 //-----------------------------------------------------------------------------
412 // wxDirItemData
413 //-----------------------------------------------------------------------------
414
415 wxDirItemData::wxDirItemData(const wxString& path, const wxString& name,
416 bool isDir)
417 {
418 m_path = path;
419 m_name = name;
420 /* Insert logic to detect hidden files here
421 * In UnixLand we just check whether the first char is a dot
422 * For FileNameFromPath read LastDirNameInThisPath ;-) */
423 // m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
424 m_isHidden = FALSE;
425 m_isExpanded = FALSE;
426 m_isDir = isDir;
427 }
428
429 wxDirItemData::~wxDirItemData()
430 {
431 }
432
433 void wxDirItemData::SetNewDirName(const wxString& path)
434 {
435 m_path = path;
436 m_name = wxFileNameFromPath(path);
437 }
438
439 bool wxDirItemData::HasSubDirs() const
440 {
441 if (m_path.IsEmpty())
442 return FALSE;
443
444 wxDir dir;
445 {
446 wxLogNull nolog;
447 if ( !dir.Open(m_path) )
448 return FALSE;
449 }
450
451 return dir.HasSubDirs();
452 }
453
454 bool wxDirItemData::HasFiles(const wxString& WXUNUSED(spec)) const
455 {
456 if (m_path.IsEmpty())
457 return FALSE;
458
459 wxDir dir;
460 {
461 wxLogNull nolog;
462 if ( !dir.Open(m_path) )
463 return FALSE;
464 }
465
466 return dir.HasFiles();
467 }
468
469 //-----------------------------------------------------------------------------
470 // wxGenericDirCtrl
471 //-----------------------------------------------------------------------------
472
473 IMPLEMENT_DYNAMIC_CLASS(wxGenericDirCtrl, wxControl)
474
475 BEGIN_EVENT_TABLE(wxGenericDirCtrl, wxControl)
476 EVT_TREE_ITEM_EXPANDING (-1, wxGenericDirCtrl::OnExpandItem)
477 EVT_TREE_ITEM_COLLAPSED (-1, wxGenericDirCtrl::OnCollapseItem)
478 EVT_TREE_BEGIN_LABEL_EDIT (-1, wxGenericDirCtrl::OnBeginEditItem)
479 EVT_TREE_END_LABEL_EDIT (-1, wxGenericDirCtrl::OnEndEditItem)
480 EVT_SIZE (wxGenericDirCtrl::OnSize)
481 END_EVENT_TABLE()
482
483 wxGenericDirCtrl::wxGenericDirCtrl(void)
484 {
485 Init();
486 }
487
488 bool wxGenericDirCtrl::Create(wxWindow *parent,
489 const wxWindowID id,
490 const wxString& dir,
491 const wxPoint& pos,
492 const wxSize& size,
493 long style,
494 const wxString& filter,
495 int defaultFilter,
496 const wxString& name)
497 {
498 if (!wxControl::Create(parent, id, pos, size, style, wxDefaultValidator, name))
499 return FALSE;
500
501 SetBackgroundColour(wxSystemSettings::GetColour(wxSYS_COLOUR_3DFACE));
502
503 Init();
504
505 long treeStyle = wxTR_HAS_BUTTONS;
506 #ifdef __WXMSW__
507 // VS: Do **NOT** remove this style, ever. MSW native wxTreeCtrl::EditLabel doesn't
508 // work without this style and we need it to be able to create new directories.
509 // Generic wxTreeCtrl can do it even w/o wxTR_EDIT_LABELS, so we only add it
510 // in case of wxMSW (as it is arguably better to not have the style enabled)
511 treeStyle |= wxTR_EDIT_LABELS;
512 #endif
513 #ifndef __WXMSW__
514 // FIXME, doesn't work for some reason
515 treeStyle |= wxTR_HIDE_ROOT;
516 #endif
517
518 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
519 treeStyle |= wxNO_BORDER;
520
521 long filterStyle = 0;
522 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
523 filterStyle |= wxNO_BORDER;
524
525 m_treeCtrl = new wxTreeCtrl(this, wxID_TREECTRL, pos, size, treeStyle);
526
527 if (!filter.IsEmpty() && (style & wxDIRCTRL_SHOW_FILTERS))
528 m_filterListCtrl = new wxDirFilterListCtrl(this, wxID_FILTERLISTCTRL, wxDefaultPosition, wxDefaultSize, filterStyle);
529
530 m_defaultPath = dir;
531 m_filter = filter;
532
533 SetFilterIndex(defaultFilter);
534
535 if (m_filterListCtrl)
536 m_filterListCtrl->FillFilterList(filter, defaultFilter);
537
538 m_imageList = new wxImageList(16, 16, TRUE);
539 m_imageList->Add(wxIcon(icon1_xpm));
540 m_imageList->Add(wxIcon(icon2_xpm));
541 m_imageList->Add(wxIcon(icon3_xpm));
542 m_imageList->Add(wxIcon(icon4_xpm));
543 m_imageList->Add(wxIcon(icon5_xpm));
544 m_imageList->Add(wxIcon(icon6_xpm));
545 m_imageList->Add(wxIcon(icon7_xpm));
546 m_imageList->Add(wxIcon(icon8_xpm));
547 m_treeCtrl->AssignImageList(m_imageList);
548
549 m_showHidden = FALSE;
550 wxDirItemData* rootData = new wxDirItemData(wxT(""), wxT(""), TRUE);
551
552 wxString rootName;
553
554 #if defined(__WINDOWS__) || defined(__WXPM__) || defined(__DOS__)
555 rootName = _("Computer");
556 #else
557 rootName = _("Sections");
558 #endif
559
560 m_rootId = m_treeCtrl->AddRoot( rootName, 3, -1, rootData);
561 m_treeCtrl->SetItemHasChildren(m_rootId);
562 m_treeCtrl->Expand(m_rootId); // automatically expand first level
563
564 // Expand and select the default path
565 if (!m_defaultPath.IsEmpty())
566 ExpandPath(m_defaultPath);
567
568 DoResize();
569
570 return TRUE;
571 }
572
573 wxGenericDirCtrl::~wxGenericDirCtrl()
574 {
575 }
576
577 void wxGenericDirCtrl::Init()
578 {
579 m_showHidden = FALSE;
580 m_imageList = NULL;
581 m_currentFilter = 0;
582 m_currentFilterStr = wxEmptyString; // Default: any file
583 m_treeCtrl = NULL;
584 m_filterListCtrl = NULL;
585 }
586
587 void wxGenericDirCtrl::ShowHidden( bool show )
588 {
589 m_showHidden = show;
590
591 wxString path = GetPath();
592 m_treeCtrl->Collapse(m_treeCtrl->GetRootItem());
593 m_treeCtrl->Expand(m_treeCtrl->GetRootItem());
594 SetPath(path);
595 }
596
597 void wxGenericDirCtrl::AddSection(const wxString& path, const wxString& name, int imageId)
598 {
599 wxDirItemData *dir_item = new wxDirItemData(path,name,TRUE);
600
601 wxTreeItemId id = m_treeCtrl->AppendItem( m_rootId, name, imageId, -1, dir_item);
602
603 m_treeCtrl->SetItemHasChildren(id);
604 }
605
606 void wxGenericDirCtrl::SetupSections()
607 {
608 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
609
610 #ifdef __WIN32__
611 wxChar driveBuffer[256];
612 size_t n = (size_t) GetLogicalDriveStrings(255, driveBuffer);
613 size_t i = 0;
614 while (i < n)
615 {
616 wxString path, name;
617 path.Printf(wxT("%c:\\"), driveBuffer[i]);
618 name.Printf(wxT("(%c:)"), driveBuffer[i]);
619
620 int imageId = 4;
621 int driveType = ::GetDriveType(path);
622 switch (driveType)
623 {
624 case DRIVE_REMOVABLE:
625 if (path == wxT("a:\\") || path == wxT("b:\\"))
626 imageId = 6; // Floppy
627 else
628 imageId = 7;
629 break;
630 case DRIVE_FIXED:
631 imageId = 4;
632 break;
633 case DRIVE_REMOTE:
634 imageId = 4;
635 break;
636 case DRIVE_CDROM:
637 imageId = 5;
638 break;
639 default:
640 imageId = 4;
641 break;
642 }
643
644 AddSection(path, name, imageId);
645
646 while (driveBuffer[i] != wxT('\0'))
647 i ++;
648 i ++;
649 if (driveBuffer[i] == wxT('\0'))
650 break;
651 }
652 #else // !__WIN32__
653 int drive;
654
655 /* If we can switch to the drive, it exists. */
656 for( drive = 1; drive <= 26; drive++ )
657 {
658 wxString path, name;
659 path.Printf(wxT("%c:\\"), (char) (drive + 'a' - 1));
660 name.Printf(wxT("(%c:)"), (char) (drive + 'A' - 1));
661
662 if (wxIsDriveAvailable(path))
663 {
664 AddSection(path, name, (drive <= 2) ? 6/*floppy*/ : 4/*disk*/);
665 }
666 }
667 #endif // __WIN32__/!__WIN32__
668
669 #elif defined(__WXMAC__)
670 FSSpec volume ;
671 short index = 1 ;
672 while(1) {
673 short actualCount = 0 ;
674 if ( OnLine( &volume , 1 , &actualCount , &index ) != noErr || actualCount == 0 )
675 break ;
676
677 wxString name = wxMacFSSpec2MacFilename( &volume ) ;
678 AddSection(name + wxFILE_SEP_PATH, name, 0);
679 }
680 #elif defined(__UNIX__)
681 AddSection(wxT("/"), wxT("/"), 3/*computer icon*/);
682 #else
683 #error "Unsupported platform in wxGenericDirCtrl!"
684 #endif
685 }
686
687 void wxGenericDirCtrl::OnBeginEditItem(wxTreeEvent &event)
688 {
689 // don't rename the main entry "Sections"
690 if (event.GetItem() == m_rootId)
691 {
692 event.Veto();
693 return;
694 }
695
696 // don't rename the individual sections
697 if (m_treeCtrl->GetParent( event.GetItem() ) == m_rootId)
698 {
699 event.Veto();
700 return;
701 }
702 }
703
704 void wxGenericDirCtrl::OnEndEditItem(wxTreeEvent &event)
705 {
706 if ((event.GetLabel().IsEmpty()) ||
707 (event.GetLabel() == _(".")) ||
708 (event.GetLabel() == _("..")) ||
709 (event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
710 {
711 wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
712 dialog.ShowModal();
713 event.Veto();
714 return;
715 }
716
717 wxTreeItemId id = event.GetItem();
718 wxDirItemData *data = (wxDirItemData*)m_treeCtrl->GetItemData( id );
719 wxASSERT( data );
720
721 wxString new_name( wxPathOnly( data->m_path ) );
722 new_name += wxString(wxFILE_SEP_PATH);
723 new_name += event.GetLabel();
724
725 wxLogNull log;
726
727 if (wxFileExists(new_name))
728 {
729 wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
730 dialog.ShowModal();
731 event.Veto();
732 }
733
734 if (wxRenameFile(data->m_path,new_name))
735 {
736 data->SetNewDirName( new_name );
737 }
738 else
739 {
740 wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
741 dialog.ShowModal();
742 event.Veto();
743 }
744 }
745
746 void wxGenericDirCtrl::OnExpandItem(wxTreeEvent &event)
747 {
748 wxTreeItemId parentId = event.GetItem();
749
750 // VS: this is needed because the event handler is called from wxTreeCtrl
751 // ctor when wxTR_HIDE_ROOT was specified
752 if (m_rootId == 0)
753 m_rootId = m_treeCtrl->GetRootItem();
754
755 ExpandDir(parentId);
756 }
757
758 void wxGenericDirCtrl::OnCollapseItem(wxTreeEvent &event )
759 {
760 wxTreeItemId child, parent = event.GetItem();
761
762 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(event.GetItem());
763 if (!data->m_isExpanded)
764 return;
765
766 data->m_isExpanded = FALSE;
767 long cookie;
768 /* Workaround because DeleteChildren has disapeared (why?) and
769 * CollapseAndReset doesn't work as advertised (deletes parent too) */
770 child = m_treeCtrl->GetFirstChild(parent, cookie);
771 while (child.IsOk())
772 {
773 m_treeCtrl->Delete(child);
774 /* Not GetNextChild below, because the cookie mechanism can't
775 * handle disappearing children! */
776 child = m_treeCtrl->GetFirstChild(parent, cookie);
777 }
778 }
779
780 void wxGenericDirCtrl::ExpandDir(wxTreeItemId parentId)
781 {
782 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(parentId);
783
784 if (data->m_isExpanded)
785 return;
786
787 data->m_isExpanded = TRUE;
788
789 if (parentId == m_treeCtrl->GetRootItem())
790 {
791 SetupSections();
792 return;
793 }
794
795 wxASSERT(data);
796
797 wxString search,path,filename;
798
799 wxString dirName(data->m_path);
800
801 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
802 // Check if this is a root directory and if so,
803 // whether the drive is avaiable.
804 if (!wxIsDriveAvailable(dirName))
805 {
806 data->m_isExpanded = FALSE;
807 //wxMessageBox(wxT("Sorry, this drive is not available."));
808 return;
809 }
810 #endif
811
812 // This may take a longish time. Go to busy cursor
813 wxBusyCursor busy;
814
815 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
816 if (dirName.Last() == ':')
817 dirName += wxString(wxFILE_SEP_PATH);
818 #endif
819
820 wxArrayString dirs;
821 wxArrayString filenames;
822
823 wxDir d;
824 wxString eachFilename;
825
826 wxLogNull log;
827 d.Open(dirName);
828
829 if (d.IsOpened())
830 {
831 int style = wxDIR_DIRS;
832 if (m_showHidden) style |= wxDIR_HIDDEN;
833 if (d.GetFirst(& eachFilename, wxEmptyString, style))
834 {
835 do
836 {
837 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
838 {
839 dirs.Add(eachFilename);
840 }
841 }
842 while (d.GetNext(& eachFilename));
843 }
844 }
845 dirs.Sort((wxArrayString::CompareFunction) wxDirCtrlStringCompareFunction);
846
847 // Now do the filenames -- but only if we're allowed to
848 if ((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0)
849 {
850 wxLogNull log;
851
852 d.Open(dirName);
853
854 if (d.IsOpened())
855 {
856 if (d.GetFirst(& eachFilename, m_currentFilterStr, wxDIR_FILES))
857 {
858 do
859 {
860 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
861 {
862 filenames.Add(eachFilename);
863 }
864 }
865 while (d.GetNext(& eachFilename));
866 }
867 }
868 filenames.Sort((wxArrayString::CompareFunction) wxDirCtrlStringCompareFunction);
869 }
870
871 // Add the sorted dirs
872 size_t i;
873 for (i = 0; i < dirs.Count(); i++)
874 {
875 wxString eachFilename(dirs[i]);
876 path = dirName;
877 if (path.Last() != wxFILE_SEP_PATH)
878 path += wxString(wxFILE_SEP_PATH);
879 path += eachFilename;
880
881 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,TRUE);
882 wxTreeItemId id = m_treeCtrl->AppendItem( parentId, eachFilename, 0, -1, dir_item);
883 m_treeCtrl->SetItemImage( id, 1, wxTreeItemIcon_Expanded );
884
885 // Has this got any children? If so, make it expandable.
886 // (There are two situations when a dir has children: either it
887 // has subdirectories or it contains files that weren't filtered
888 // out. The latter only applies to dirctrl with files.)
889 if ( dir_item->HasSubDirs() ||
890 (((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0) &&
891 dir_item->HasFiles(m_currentFilterStr)) )
892 {
893 m_treeCtrl->SetItemHasChildren(id);
894 }
895 }
896
897 // Add the sorted filenames
898 if ((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0)
899 {
900 for (i = 0; i < filenames.Count(); i++)
901 {
902 wxString eachFilename(filenames[i]);
903 path = dirName;
904 if (path.Last() != wxFILE_SEP_PATH)
905 path += wxString(wxFILE_SEP_PATH);
906 path += eachFilename;
907 //path = dirName + wxString(wxT("/")) + eachFilename;
908 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,FALSE);
909 (void)m_treeCtrl->AppendItem( parentId, eachFilename, 2, -1, dir_item);
910 }
911 }
912 }
913
914 // Find the child that matches the first part of 'path'.
915 // E.g. if a child path is "/usr" and 'path' is "/usr/include"
916 // then the child for /usr is returned.
917 wxTreeItemId wxGenericDirCtrl::FindChild(wxTreeItemId parentId, const wxString& path, bool& done)
918 {
919 wxString path2(path);
920
921 // Make sure all separators are as per the current platform
922 path2.Replace(wxT("\\"), wxString(wxFILE_SEP_PATH));
923 path2.Replace(wxT("/"), wxString(wxFILE_SEP_PATH));
924
925 // Append a separator to foil bogus substring matching
926 path2 += wxString(wxFILE_SEP_PATH);
927
928 // In MSW or PM, case is not significant
929 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
930 path2.MakeLower();
931 #endif
932
933 long cookie;
934 wxTreeItemId childId = m_treeCtrl->GetFirstChild(parentId, cookie);
935 while (childId.IsOk())
936 {
937 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
938
939 if (data && !data->m_path.IsEmpty())
940 {
941 wxString childPath(data->m_path);
942 if (childPath.Last() != wxFILE_SEP_PATH)
943 childPath += wxString(wxFILE_SEP_PATH);
944
945 // In MSW and PM, case is not significant
946 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
947 childPath.MakeLower();
948 #endif
949
950 if (childPath.Len() <= path2.Len())
951 {
952 wxString path3 = path2.Mid(0, childPath.Len());
953 if (childPath == path3)
954 {
955 if (path3.Len() == path2.Len())
956 done = TRUE;
957 else
958 done = FALSE;
959 return childId;
960 }
961 }
962 }
963
964 childId = m_treeCtrl->GetNextChild(parentId, cookie);
965 }
966 wxTreeItemId invalid;
967 return invalid;
968 }
969
970 // Try to expand as much of the given path as possible,
971 // and select the given tree item.
972 bool wxGenericDirCtrl::ExpandPath(const wxString& path)
973 {
974 bool done = FALSE;
975 wxTreeItemId id = FindChild(m_rootId, path, done);
976 wxTreeItemId lastId = id; // The last non-zero id
977 while (id.IsOk() && !done)
978 {
979 ExpandDir(id);
980
981 id = FindChild(id, path, done);
982 if (id.IsOk())
983 lastId = id;
984 }
985 if (lastId.IsOk())
986 {
987 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(lastId);
988 if (data->m_isDir)
989 {
990 m_treeCtrl->Expand(lastId);
991 }
992 if ((GetWindowStyle() & wxDIRCTRL_SELECT_FIRST) && data->m_isDir)
993 {
994 // Find the first file in this directory
995 long cookie;
996 wxTreeItemId childId = m_treeCtrl->GetFirstChild(lastId, cookie);
997 bool selectedChild = FALSE;
998 while (childId.IsOk())
999 {
1000 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
1001
1002 if (data && data->m_path != "" && !data->m_isDir)
1003 {
1004 m_treeCtrl->SelectItem(childId);
1005 m_treeCtrl->EnsureVisible(childId);
1006 selectedChild = TRUE;
1007 break;
1008 }
1009 childId = m_treeCtrl->GetNextChild(lastId, cookie);
1010 }
1011 if (!selectedChild)
1012 {
1013 m_treeCtrl->SelectItem(lastId);
1014 m_treeCtrl->EnsureVisible(lastId);
1015 }
1016 }
1017 else
1018 {
1019 m_treeCtrl->SelectItem(lastId);
1020 m_treeCtrl->EnsureVisible(lastId);
1021 }
1022
1023 return TRUE;
1024 }
1025 else
1026 return FALSE;
1027 }
1028
1029 wxString wxGenericDirCtrl::GetPath() const
1030 {
1031 wxTreeItemId id = m_treeCtrl->GetSelection();
1032 if (id)
1033 {
1034 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1035 return data->m_path;
1036 }
1037 else
1038 return wxEmptyString;
1039 }
1040
1041 wxString wxGenericDirCtrl::GetFilePath() const
1042 {
1043 wxTreeItemId id = m_treeCtrl->GetSelection();
1044 if (id)
1045 {
1046 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1047 if (data->m_isDir)
1048 return wxEmptyString;
1049 else
1050 return data->m_path;
1051 }
1052 else
1053 return wxEmptyString;
1054 }
1055
1056 void wxGenericDirCtrl::SetPath(const wxString& path)
1057 {
1058 m_defaultPath = path;
1059 if (m_rootId)
1060 ExpandPath(path);
1061 }
1062
1063 // Not used
1064 #if 0
1065 void wxGenericDirCtrl::FindChildFiles(wxTreeItemId id, int dirFlags, wxArrayString& filenames)
1066 {
1067 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(id);
1068
1069 // This may take a longish time. Go to busy cursor
1070 wxBusyCursor busy;
1071
1072 wxASSERT(data);
1073
1074 wxString search,path,filename;
1075
1076 wxString dirName(data->m_path);
1077
1078 #if defined(__WXMSW__) || defined(__WXPM__)
1079 if (dirName.Last() == ':')
1080 dirName += wxString(wxFILE_SEP_PATH);
1081 #endif
1082
1083 wxDir d;
1084 wxString eachFilename;
1085
1086 wxLogNull log;
1087 d.Open(dirName);
1088
1089 if (d.IsOpened())
1090 {
1091 if (d.GetFirst(& eachFilename, m_currentFilterStr, dirFlags))
1092 {
1093 do
1094 {
1095 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
1096 {
1097 filenames.Add(eachFilename);
1098 }
1099 }
1100 while (d.GetNext(& eachFilename)) ;
1101 }
1102 }
1103 }
1104 #endif
1105
1106 void wxGenericDirCtrl::SetFilterIndex(int n)
1107 {
1108 m_currentFilter = n;
1109
1110 wxString f, d;
1111 if (ExtractWildcard(m_filter, n, f, d))
1112 m_currentFilterStr = f;
1113 else
1114 m_currentFilterStr = wxT("*.*");
1115 }
1116
1117 void wxGenericDirCtrl::SetFilter(const wxString& filter)
1118 {
1119 m_filter = filter;
1120
1121 wxString f, d;
1122 if (ExtractWildcard(m_filter, m_currentFilter, f, d))
1123 m_currentFilterStr = f;
1124 else
1125 m_currentFilterStr = wxT("*.*");
1126 }
1127
1128 // Extract description and actual filter from overall filter string
1129 bool wxGenericDirCtrl::ExtractWildcard(const wxString& filterStr, int n, wxString& filter, wxString& description)
1130 {
1131 wxArrayString filters, descriptions;
1132 int count = ParseFilter(filterStr, filters, descriptions);
1133 if (count > 0 && n < count)
1134 {
1135 filter = filters[n];
1136 description = descriptions[n];
1137 return TRUE;
1138 }
1139
1140 return FALSE;
1141 }
1142
1143 // Parses the global filter, returning the number of filters.
1144 // Returns 0 if none or if there's a problem.
1145 // filterStr is in the form:
1146 //
1147 // "All files (*.*)|*.*|JPEG Files (*.jpeg)|*.jpg"
1148
1149 int wxGenericDirCtrl::ParseFilter(const wxString& filterStr, wxArrayString& filters, wxArrayString& descriptions)
1150 {
1151 wxString str(filterStr);
1152
1153 wxString description, filter;
1154 int pos;
1155 bool finished = FALSE;
1156 do
1157 {
1158 pos = str.Find(wxT('|'));
1159 if (pos == -1)
1160 return 0; // Problem
1161 description = str.Left(pos);
1162 str = str.Mid(pos+1);
1163 pos = str.Find(wxT('|'));
1164 if (pos == -1)
1165 {
1166 filter = str;
1167 finished = TRUE;
1168 }
1169 else
1170 {
1171 filter = str.Left(pos);
1172 str = str.Mid(pos+1);
1173 }
1174 descriptions.Add(description);
1175 filters.Add(filter);
1176 }
1177 while (!finished) ;
1178
1179 return filters.Count();
1180 }
1181
1182 void wxGenericDirCtrl::DoResize()
1183 {
1184 wxSize sz = GetClientSize();
1185 int verticalSpacing = 3;
1186 if (m_treeCtrl)
1187 {
1188 wxSize filterSz ;
1189 if (m_filterListCtrl)
1190 {
1191 #ifdef __WXMSW__
1192 // For some reason, this is required in order for the
1193 // correct control height to always be returned, rather
1194 // than the drop-down list height which is sometimes returned.
1195 wxSize oldSize = m_filterListCtrl->GetSize();
1196 m_filterListCtrl->SetSize(-1, -1, oldSize.x+10, -1, wxSIZE_USE_EXISTING);
1197 m_filterListCtrl->SetSize(-1, -1, oldSize.x, -1, wxSIZE_USE_EXISTING);
1198 #endif
1199 filterSz = m_filterListCtrl->GetSize();
1200 sz.y -= (filterSz.y + verticalSpacing);
1201 }
1202 m_treeCtrl->SetSize(0, 0, sz.x, sz.y);
1203 if (m_filterListCtrl)
1204 {
1205 m_filterListCtrl->SetSize(0, sz.y + verticalSpacing, sz.x, filterSz.y);
1206 // Don't know why, but this needs refreshing after a resize (wxMSW)
1207 m_filterListCtrl->Refresh();
1208 }
1209 }
1210 }
1211
1212
1213 void wxGenericDirCtrl::OnSize(wxSizeEvent& WXUNUSED(event))
1214 {
1215 DoResize();
1216 }
1217
1218 //-----------------------------------------------------------------------------
1219 // wxDirFilterListCtrl
1220 //-----------------------------------------------------------------------------
1221
1222 IMPLEMENT_CLASS(wxDirFilterListCtrl, wxChoice)
1223
1224 BEGIN_EVENT_TABLE(wxDirFilterListCtrl, wxChoice)
1225 EVT_CHOICE(-1, wxDirFilterListCtrl::OnSelFilter)
1226 END_EVENT_TABLE()
1227
1228 bool wxDirFilterListCtrl::Create(wxGenericDirCtrl* parent, const wxWindowID id,
1229 const wxPoint& pos,
1230 const wxSize& size,
1231 long style)
1232 {
1233 m_dirCtrl = parent;
1234 return wxChoice::Create(parent, id, pos, size, 0, NULL, style);
1235 }
1236
1237 void wxDirFilterListCtrl::Init()
1238 {
1239 m_dirCtrl = NULL;
1240 }
1241
1242 void wxDirFilterListCtrl::OnSelFilter(wxCommandEvent& WXUNUSED(event))
1243 {
1244 int sel = GetSelection();
1245
1246 wxString currentPath = m_dirCtrl->GetPath();
1247
1248 m_dirCtrl->SetFilterIndex(sel);
1249
1250 // If the filter has changed, the view is out of date, so
1251 // collapse the tree.
1252 m_dirCtrl->GetTreeCtrl()->Collapse(m_dirCtrl->GetRootId());
1253 m_dirCtrl->GetTreeCtrl()->Expand(m_dirCtrl->GetRootId());
1254
1255 // Try to restore the selection, or at least the directory
1256 m_dirCtrl->ExpandPath(currentPath);
1257 }
1258
1259 void wxDirFilterListCtrl::FillFilterList(const wxString& filter, int defaultFilter)
1260 {
1261 Clear();
1262 wxArrayString descriptions, filters;
1263 size_t n = (size_t) m_dirCtrl->ParseFilter(filter, filters, descriptions);
1264
1265 if (n > 0 && defaultFilter < (int) n)
1266 {
1267 size_t i = 0;
1268 for (i = 0; i < n; i++)
1269 Append(descriptions[i]);
1270 SetSelection(defaultFilter);
1271 }
1272 }
1273
1274 #endif // wxUSE_DIRDLG
wxWidgets (Impactor)