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


dddbbad58759a6f3897e339740fab024a866fe97
[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 #define wxID_TREECTRL 7000
316 #define wxID_FILTERLISTCTRL 7001
317
318 #if defined(__DOS__)
319
320 bool wxIsDriveAvailable(const wxString& dirName)
321 {
322 // FIXME_MGL - this method leads to hang up under Watcom for some reason
323 #ifndef __WATCOMC__
324 if ( dirName.Len() == 3 && dirName[1u] == wxT(':') )
325 {
326 wxString dirNameLower(dirName.Lower());
327 // VS: always return TRUE for removable media, since Win95 doesn't
328 // like it when MS-DOS app accesses empty floppy drive
329 return (dirNameLower[0u] == wxT('a') ||
330 dirNameLower[0u] == wxT('b') ||
331 wxPathExists(dirNameLower));
332 }
333 else
334 #endif
335 return TRUE;
336 }
337
338 #elif defined(__WINDOWS__) || defined(__WXPM__)
339
340 int setdrive(int drive)
341 {
342 #if defined(__GNUWIN32__) && \
343 (defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
344 return _chdrive(drive);
345 #else
346 wxChar newdrive[3];
347
348 if (drive < 1 || drive > 31)
349 return -1;
350 newdrive[0] = (wxChar)(wxT('A') + drive - 1);
351 newdrive[1] = wxT(':');
352 newdrive[2] = wxT('\0');
353 #if defined(__WXMSW__)
354 #ifdef __WIN16__
355 if (wxSetWorkingDirectory(newdrive))
356 #else
357 if (::SetCurrentDirectory(newdrive))
358 #endif
359 #else
360 // VA doesn't know what LPSTR is and has its own set
361 if (DosSetCurrentDir((PSZ)newdrive))
362 #endif
363 return 0;
364 else
365 return -1;
366 #endif // !GNUWIN32
367 }
368
369 bool wxIsDriveAvailable(const wxString& dirName)
370 {
371 #ifdef __WIN32__
372 UINT errorMode = SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX);
373 #endif
374 bool success = TRUE;
375
376 // Check if this is a root directory and if so,
377 // whether the drive is avaiable.
378 if (dirName.Len() == 3 && dirName[(size_t)1] == wxT(':'))
379 {
380 wxString dirNameLower(dirName.Lower());
381 #if defined(__GNUWIN32__) && \
382 !(defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
383 success = wxPathExists(dirNameLower);
384 #else
385 int currentDrive = _getdrive();
386 int thisDrive = (int) (dirNameLower[(size_t)0] - 'a' + 1) ;
387 int err = setdrive( thisDrive ) ;
388 setdrive( currentDrive );
389
390 if (err == -1)
391 {
392 success = FALSE;
393 }
394 #endif
395 }
396 #ifdef __WIN32__
397 (void) SetErrorMode(errorMode);
398 #endif
399
400 return success;
401 }
402 #endif // __WINDOWS__ || __WXPM__
403
404 // Function which is called by quick sort. We want to override the default wxArrayString behaviour,
405 // and sort regardless of case.
406 static int LINKAGEMODE wxDirCtrlStringCompareFunction(const void *first, const void *second)
407 {
408 wxString *strFirst = (wxString *)first;
409 wxString *strSecond = (wxString *)second;
410
411 return strFirst->CmpNoCase(*strSecond);
412 }
413
414 //-----------------------------------------------------------------------------
415 // wxDirItemData
416 //-----------------------------------------------------------------------------
417
418 wxDirItemData::wxDirItemData(const wxString& path, const wxString& name,
419 bool isDir)
420 {
421 m_path = path;
422 m_name = name;
423 /* Insert logic to detect hidden files here
424 * In UnixLand we just check whether the first char is a dot
425 * For FileNameFromPath read LastDirNameInThisPath ;-) */
426 // m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
427 m_isHidden = FALSE;
428 m_isExpanded = FALSE;
429 m_isDir = isDir;
430 }
431
432 wxDirItemData::~wxDirItemData()
433 {
434 }
435
436 void wxDirItemData::SetNewDirName(const wxString& path)
437 {
438 m_path = path;
439 m_name = wxFileNameFromPath(path);
440 }
441
442 bool wxDirItemData::HasSubDirs() const
443 {
444 if (m_path.IsEmpty())
445 return FALSE;
446
447 wxDir dir;
448 {
449 wxLogNull nolog;
450 if ( !dir.Open(m_path) )
451 return FALSE;
452 }
453
454 return dir.HasSubDirs();
455 }
456
457 bool wxDirItemData::HasFiles(const wxString& WXUNUSED(spec)) const
458 {
459 if (m_path.IsEmpty())
460 return FALSE;
461
462 wxDir dir;
463 {
464 wxLogNull nolog;
465 if ( !dir.Open(m_path) )
466 return FALSE;
467 }
468
469 return dir.HasFiles();
470 }
471
472 //-----------------------------------------------------------------------------
473 // wxGenericDirCtrl
474 //-----------------------------------------------------------------------------
475
476 IMPLEMENT_DYNAMIC_CLASS(wxGenericDirCtrl, wxControl)
477
478 BEGIN_EVENT_TABLE(wxGenericDirCtrl, wxControl)
479 EVT_TREE_ITEM_EXPANDING (-1, wxGenericDirCtrl::OnExpandItem)
480 EVT_TREE_ITEM_COLLAPSED (-1, wxGenericDirCtrl::OnCollapseItem)
481 EVT_TREE_BEGIN_LABEL_EDIT (-1, wxGenericDirCtrl::OnBeginEditItem)
482 EVT_TREE_END_LABEL_EDIT (-1, wxGenericDirCtrl::OnEndEditItem)
483 EVT_SIZE (wxGenericDirCtrl::OnSize)
484 END_EVENT_TABLE()
485
486 wxGenericDirCtrl::wxGenericDirCtrl(void)
487 {
488 Init();
489 }
490
491 bool wxGenericDirCtrl::Create(wxWindow *parent,
492 const wxWindowID id,
493 const wxString& dir,
494 const wxPoint& pos,
495 const wxSize& size,
496 long style,
497 const wxString& filter,
498 int defaultFilter,
499 const wxString& name)
500 {
501 if (!wxControl::Create(parent, id, pos, size, style, wxDefaultValidator, name))
502 return FALSE;
503
504 SetBackgroundColour(wxSystemSettings::GetColour(wxSYS_COLOUR_3DFACE));
505
506 Init();
507
508 long treeStyle = wxTR_HAS_BUTTONS;
509 #ifdef __WXMSW__
510 // VS: Do **NOT** remove this style, ever. MSW native wxTreeCtrl::EditLabel doesn't
511 // work without this style and we need it to be able to create new directories.
512 // Generic wxTreeCtrl can do it even w/o wxTR_EDIT_LABELS, so we only add it
513 // in case of wxMSW (as it is arguably better to not have the style enabled)
514 treeStyle |= wxTR_EDIT_LABELS;
515 #endif
516 #ifndef __WXMSW__
517 // FIXME, doesn't work for some reason
518 treeStyle |= wxTR_HIDE_ROOT;
519 #endif
520
521 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
522 treeStyle |= wxNO_BORDER;
523
524 long filterStyle = 0;
525 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
526 filterStyle |= wxNO_BORDER;
527
528 m_treeCtrl = new wxTreeCtrl(this, wxID_TREECTRL, pos, size, treeStyle);
529
530 if (!filter.IsEmpty() && (style & wxDIRCTRL_SHOW_FILTERS))
531 m_filterListCtrl = new wxDirFilterListCtrl(this, wxID_FILTERLISTCTRL, wxDefaultPosition, wxDefaultSize, filterStyle);
532
533 m_defaultPath = dir;
534 m_filter = filter;
535
536 SetFilterIndex(defaultFilter);
537
538 if (m_filterListCtrl)
539 m_filterListCtrl->FillFilterList(filter, defaultFilter);
540
541 m_imageList = new wxImageList(16, 16, TRUE);
542 m_imageList->Add(wxIcon(icon1_xpm));
543 m_imageList->Add(wxIcon(icon2_xpm));
544 m_imageList->Add(wxIcon(icon3_xpm));
545 m_imageList->Add(wxIcon(icon4_xpm));
546 m_imageList->Add(wxIcon(icon5_xpm));
547 m_imageList->Add(wxIcon(icon6_xpm));
548 m_imageList->Add(wxIcon(icon7_xpm));
549 m_imageList->Add(wxIcon(icon8_xpm));
550 m_treeCtrl->AssignImageList(m_imageList);
551
552 m_showHidden = FALSE;
553 wxDirItemData* rootData = new wxDirItemData(wxT(""), wxT(""), TRUE);
554
555 wxString rootName;
556
557 #if defined(__WINDOWS__) || defined(__WXPM__) || defined(__DOS__)
558 rootName = _("Computer");
559 #else
560 rootName = _("Sections");
561 #endif
562
563 m_rootId = m_treeCtrl->AddRoot( rootName, 3, -1, rootData);
564 m_treeCtrl->SetItemHasChildren(m_rootId);
565 m_treeCtrl->Expand(m_rootId); // automatically expand first level
566
567 // Expand and select the default path
568 if (!m_defaultPath.IsEmpty())
569 ExpandPath(m_defaultPath);
570
571 DoResize();
572
573 return TRUE;
574 }
575
576 wxGenericDirCtrl::~wxGenericDirCtrl()
577 {
578 }
579
580 void wxGenericDirCtrl::Init()
581 {
582 m_showHidden = FALSE;
583 m_imageList = NULL;
584 m_currentFilter = 0;
585 m_currentFilterStr = wxEmptyString; // Default: any file
586 m_treeCtrl = NULL;
587 m_filterListCtrl = NULL;
588 }
589
590 void wxGenericDirCtrl::ShowHidden( bool show )
591 {
592 m_showHidden = show;
593
594 wxString path = GetPath();
595 m_treeCtrl->Collapse(m_treeCtrl->GetRootItem());
596 m_treeCtrl->Expand(m_treeCtrl->GetRootItem());
597 SetPath(path);
598 }
599
600 void wxGenericDirCtrl::AddSection(const wxString& path, const wxString& name, int imageId)
601 {
602 wxDirItemData *dir_item = new wxDirItemData(path,name,TRUE);
603
604 wxTreeItemId id = m_treeCtrl->AppendItem( m_rootId, name, imageId, -1, dir_item);
605
606 m_treeCtrl->SetItemHasChildren(id);
607 }
608
609 void wxGenericDirCtrl::SetupSections()
610 {
611 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
612
613 #ifdef __WIN32__
614 wxChar driveBuffer[256];
615 size_t n = (size_t) GetLogicalDriveStrings(255, driveBuffer);
616 size_t i = 0;
617 while (i < n)
618 {
619 wxString path, name;
620 path.Printf(wxT("%c:\\"), driveBuffer[i]);
621 name.Printf(wxT("(%c:)"), driveBuffer[i]);
622
623 int imageId = 4;
624 int driveType = ::GetDriveType(path);
625 switch (driveType)
626 {
627 case DRIVE_REMOVABLE:
628 if (path == wxT("a:\\") || path == wxT("b:\\"))
629 imageId = 6; // Floppy
630 else
631 imageId = 7;
632 break;
633 case DRIVE_FIXED:
634 imageId = 4;
635 break;
636 case DRIVE_REMOTE:
637 imageId = 4;
638 break;
639 case DRIVE_CDROM:
640 imageId = 5;
641 break;
642 default:
643 imageId = 4;
644 break;
645 }
646
647 AddSection(path, name, imageId);
648
649 while (driveBuffer[i] != wxT('\0'))
650 i ++;
651 i ++;
652 if (driveBuffer[i] == wxT('\0'))
653 break;
654 }
655 #else // !__WIN32__
656 int drive;
657
658 /* If we can switch to the drive, it exists. */
659 for( drive = 1; drive <= 26; drive++ )
660 {
661 wxString path, name;
662 path.Printf(wxT("%c:\\"), (char) (drive + 'a' - 1));
663 name.Printf(wxT("(%c:)"), (char) (drive + 'A' - 1));
664
665 if (wxIsDriveAvailable(path))
666 {
667 AddSection(path, name, (drive <= 2) ? 6/*floppy*/ : 4/*disk*/);
668 }
669 }
670 #endif // __WIN32__/!__WIN32__
671
672 #elif defined(__WXMAC__)
673 FSSpec volume ;
674 short index = 1 ;
675 while(1) {
676 short actualCount = 0 ;
677 if ( OnLine( &volume , 1 , &actualCount , &index ) != noErr || actualCount == 0 )
678 break ;
679
680 wxString name = wxMacFSSpec2MacFilename( &volume ) ;
681 AddSection(name + wxFILE_SEP_PATH, name, 0);
682 }
683 #elif defined(__UNIX__)
684 AddSection(wxT("/"), wxT("/"), 3/*computer icon*/);
685 #else
686 #error "Unsupported platform in wxGenericDirCtrl!"
687 #endif
688 }
689
690 void wxGenericDirCtrl::OnBeginEditItem(wxTreeEvent &event)
691 {
692 // don't rename the main entry "Sections"
693 if (event.GetItem() == m_rootId)
694 {
695 event.Veto();
696 return;
697 }
698
699 // don't rename the individual sections
700 if (m_treeCtrl->GetParent( event.GetItem() ) == m_rootId)
701 {
702 event.Veto();
703 return;
704 }
705 }
706
707 void wxGenericDirCtrl::OnEndEditItem(wxTreeEvent &event)
708 {
709 if ((event.GetLabel().IsEmpty()) ||
710 (event.GetLabel() == _(".")) ||
711 (event.GetLabel() == _("..")) ||
712 (event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
713 {
714 wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
715 dialog.ShowModal();
716 event.Veto();
717 return;
718 }
719
720 wxTreeItemId id = event.GetItem();
721 wxDirItemData *data = (wxDirItemData*)m_treeCtrl->GetItemData( id );
722 wxASSERT( data );
723
724 wxString new_name( wxPathOnly( data->m_path ) );
725 new_name += wxString(wxFILE_SEP_PATH);
726 new_name += event.GetLabel();
727
728 wxLogNull log;
729
730 if (wxFileExists(new_name))
731 {
732 wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
733 dialog.ShowModal();
734 event.Veto();
735 }
736
737 if (wxRenameFile(data->m_path,new_name))
738 {
739 data->SetNewDirName( new_name );
740 }
741 else
742 {
743 wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
744 dialog.ShowModal();
745 event.Veto();
746 }
747 }
748
749 void wxGenericDirCtrl::OnExpandItem(wxTreeEvent &event)
750 {
751 wxTreeItemId parentId = event.GetItem();
752
753 // VS: this is needed because the event handler is called from wxTreeCtrl
754 // ctor when wxTR_HIDE_ROOT was specified
755 if (m_rootId == 0)
756 m_rootId = m_treeCtrl->GetRootItem();
757
758 ExpandDir(parentId);
759 }
760
761 void wxGenericDirCtrl::OnCollapseItem(wxTreeEvent &event )
762 {
763 wxTreeItemId child, parent = event.GetItem();
764
765 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(event.GetItem());
766 if (!data->m_isExpanded)
767 return;
768
769 data->m_isExpanded = FALSE;
770 long cookie;
771 /* Workaround because DeleteChildren has disapeared (why?) and
772 * CollapseAndReset doesn't work as advertised (deletes parent too) */
773 child = m_treeCtrl->GetFirstChild(parent, cookie);
774 while (child.IsOk())
775 {
776 m_treeCtrl->Delete(child);
777 /* Not GetNextChild below, because the cookie mechanism can't
778 * handle disappearing children! */
779 child = m_treeCtrl->GetFirstChild(parent, cookie);
780 }
781 }
782
783 void wxGenericDirCtrl::ExpandDir(wxTreeItemId parentId)
784 {
785 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(parentId);
786
787 if (data->m_isExpanded)
788 return;
789
790 data->m_isExpanded = TRUE;
791
792 if (parentId == m_treeCtrl->GetRootItem())
793 {
794 SetupSections();
795 return;
796 }
797
798 wxASSERT(data);
799
800 wxString search,path,filename;
801
802 wxString dirName(data->m_path);
803
804 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
805 // Check if this is a root directory and if so,
806 // whether the drive is avaiable.
807 if (!wxIsDriveAvailable(dirName))
808 {
809 data->m_isExpanded = FALSE;
810 //wxMessageBox(wxT("Sorry, this drive is not available."));
811 return;
812 }
813 #endif
814
815 // This may take a longish time. Go to busy cursor
816 wxBusyCursor busy;
817
818 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
819 if (dirName.Last() == ':')
820 dirName += wxString(wxFILE_SEP_PATH);
821 #endif
822
823 wxArrayString dirs;
824 wxArrayString filenames;
825
826 wxDir d;
827 wxString eachFilename;
828
829 wxLogNull log;
830 d.Open(dirName);
831
832 if (d.IsOpened())
833 {
834 int style = wxDIR_DIRS;
835 if (m_showHidden) style |= wxDIR_HIDDEN;
836 if (d.GetFirst(& eachFilename, wxEmptyString, style))
837 {
838 do
839 {
840 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
841 {
842 dirs.Add(eachFilename);
843 }
844 }
845 while (d.GetNext(& eachFilename));
846 }
847 }
848 dirs.Sort((wxArrayString::CompareFunction) wxDirCtrlStringCompareFunction);
849
850 // Now do the filenames -- but only if we're allowed to
851 if ((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0)
852 {
853 wxLogNull log;
854
855 d.Open(dirName);
856
857 if (d.IsOpened())
858 {
859 if (d.GetFirst(& eachFilename, m_currentFilterStr, wxDIR_FILES))
860 {
861 do
862 {
863 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
864 {
865 filenames.Add(eachFilename);
866 }
867 }
868 while (d.GetNext(& eachFilename));
869 }
870 }
871 filenames.Sort((wxArrayString::CompareFunction) wxDirCtrlStringCompareFunction);
872 }
873
874 // Add the sorted dirs
875 size_t i;
876 for (i = 0; i < dirs.Count(); i++)
877 {
878 wxString eachFilename(dirs[i]);
879 path = dirName;
880 if (path.Last() != wxFILE_SEP_PATH)
881 path += wxString(wxFILE_SEP_PATH);
882 path += eachFilename;
883
884 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,TRUE);
885 wxTreeItemId id = m_treeCtrl->AppendItem( parentId, eachFilename, 0, -1, dir_item);
886 m_treeCtrl->SetItemImage( id, 1, wxTreeItemIcon_Expanded );
887
888 // Has this got any children? If so, make it expandable.
889 // (There are two situations when a dir has children: either it
890 // has subdirectories or it contains files that weren't filtered
891 // out. The latter only applies to dirctrl with files.)
892 if ( dir_item->HasSubDirs() ||
893 (((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0) &&
894 dir_item->HasFiles(m_currentFilterStr)) )
895 {
896 m_treeCtrl->SetItemHasChildren(id);
897 }
898 }
899
900 // Add the sorted filenames
901 if ((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0)
902 {
903 for (i = 0; i < filenames.Count(); i++)
904 {
905 wxString eachFilename(filenames[i]);
906 path = dirName;
907 if (path.Last() != wxFILE_SEP_PATH)
908 path += wxString(wxFILE_SEP_PATH);
909 path += eachFilename;
910 //path = dirName + wxString(wxT("/")) + eachFilename;
911 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,FALSE);
912 (void)m_treeCtrl->AppendItem( parentId, eachFilename, 2, -1, dir_item);
913 }
914 }
915 }
916
917 // Find the child that matches the first part of 'path'.
918 // E.g. if a child path is "/usr" and 'path' is "/usr/include"
919 // then the child for /usr is returned.
920 wxTreeItemId wxGenericDirCtrl::FindChild(wxTreeItemId parentId, const wxString& path, bool& done)
921 {
922 wxString path2(path);
923
924 // Make sure all separators are as per the current platform
925 path2.Replace(wxT("\\"), wxString(wxFILE_SEP_PATH));
926 path2.Replace(wxT("/"), wxString(wxFILE_SEP_PATH));
927
928 // Append a separator to foil bogus substring matching
929 path2 += wxString(wxFILE_SEP_PATH);
930
931 // In MSW or PM, case is not significant
932 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
933 path2.MakeLower();
934 #endif
935
936 long cookie;
937 wxTreeItemId childId = m_treeCtrl->GetFirstChild(parentId, cookie);
938 while (childId.IsOk())
939 {
940 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
941
942 if (data && !data->m_path.IsEmpty())
943 {
944 wxString childPath(data->m_path);
945 if (childPath.Last() != wxFILE_SEP_PATH)
946 childPath += wxString(wxFILE_SEP_PATH);
947
948 // In MSW and PM, case is not significant
949 #if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
950 childPath.MakeLower();
951 #endif
952
953 if (childPath.Len() <= path2.Len())
954 {
955 wxString path3 = path2.Mid(0, childPath.Len());
956 if (childPath == path3)
957 {
958 if (path3.Len() == path2.Len())
959 done = TRUE;
960 else
961 done = FALSE;
962 return childId;
963 }
964 }
965 }
966
967 childId = m_treeCtrl->GetNextChild(parentId, cookie);
968 }
969 wxTreeItemId invalid;
970 return invalid;
971 }
972
973 // Try to expand as much of the given path as possible,
974 // and select the given tree item.
975 bool wxGenericDirCtrl::ExpandPath(const wxString& path)
976 {
977 bool done = FALSE;
978 wxTreeItemId id = FindChild(m_rootId, path, done);
979 wxTreeItemId lastId = id; // The last non-zero id
980 while (id.IsOk() && !done)
981 {
982 ExpandDir(id);
983
984 id = FindChild(id, path, done);
985 if (id.IsOk())
986 lastId = id;
987 }
988 if (lastId.IsOk())
989 {
990 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(lastId);
991 if (data->m_isDir)
992 {
993 m_treeCtrl->Expand(lastId);
994 }
995 if ((GetWindowStyle() & wxDIRCTRL_SELECT_FIRST) && data->m_isDir)
996 {
997 // Find the first file in this directory
998 long cookie;
999 wxTreeItemId childId = m_treeCtrl->GetFirstChild(lastId, cookie);
1000 bool selectedChild = FALSE;
1001 while (childId.IsOk())
1002 {
1003 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
1004
1005 if (data && data->m_path != "" && !data->m_isDir)
1006 {
1007 m_treeCtrl->SelectItem(childId);
1008 m_treeCtrl->EnsureVisible(childId);
1009 selectedChild = TRUE;
1010 break;
1011 }
1012 childId = m_treeCtrl->GetNextChild(lastId, cookie);
1013 }
1014 if (!selectedChild)
1015 {
1016 m_treeCtrl->SelectItem(lastId);
1017 m_treeCtrl->EnsureVisible(lastId);
1018 }
1019 }
1020 else
1021 {
1022 m_treeCtrl->SelectItem(lastId);
1023 m_treeCtrl->EnsureVisible(lastId);
1024 }
1025
1026 return TRUE;
1027 }
1028 else
1029 return FALSE;
1030 }
1031
1032 wxString wxGenericDirCtrl::GetPath() const
1033 {
1034 wxTreeItemId id = m_treeCtrl->GetSelection();
1035 if (id)
1036 {
1037 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1038 return data->m_path;
1039 }
1040 else
1041 return wxEmptyString;
1042 }
1043
1044 wxString wxGenericDirCtrl::GetFilePath() const
1045 {
1046 wxTreeItemId id = m_treeCtrl->GetSelection();
1047 if (id)
1048 {
1049 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1050 if (data->m_isDir)
1051 return wxEmptyString;
1052 else
1053 return data->m_path;
1054 }
1055 else
1056 return wxEmptyString;
1057 }
1058
1059 void wxGenericDirCtrl::SetPath(const wxString& path)
1060 {
1061 m_defaultPath = path;
1062 if (m_rootId)
1063 ExpandPath(path);
1064 }
1065
1066 // Not used
1067 #if 0
1068 void wxGenericDirCtrl::FindChildFiles(wxTreeItemId id, int dirFlags, wxArrayString& filenames)
1069 {
1070 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(id);
1071
1072 // This may take a longish time. Go to busy cursor
1073 wxBusyCursor busy;
1074
1075 wxASSERT(data);
1076
1077 wxString search,path,filename;
1078
1079 wxString dirName(data->m_path);
1080
1081 #if defined(__WXMSW__) || defined(__WXPM__)
1082 if (dirName.Last() == ':')
1083 dirName += wxString(wxFILE_SEP_PATH);
1084 #endif
1085
1086 wxDir d;
1087 wxString eachFilename;
1088
1089 wxLogNull log;
1090 d.Open(dirName);
1091
1092 if (d.IsOpened())
1093 {
1094 if (d.GetFirst(& eachFilename, m_currentFilterStr, dirFlags))
1095 {
1096 do
1097 {
1098 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
1099 {
1100 filenames.Add(eachFilename);
1101 }
1102 }
1103 while (d.GetNext(& eachFilename)) ;
1104 }
1105 }
1106 }
1107 #endif
1108
1109 void wxGenericDirCtrl::SetFilterIndex(int n)
1110 {
1111 m_currentFilter = n;
1112
1113 wxString f, d;
1114 if (ExtractWildcard(m_filter, n, f, d))
1115 m_currentFilterStr = f;
1116 else
1117 m_currentFilterStr = wxT("*.*");
1118 }
1119
1120 void wxGenericDirCtrl::SetFilter(const wxString& filter)
1121 {
1122 m_filter = filter;
1123
1124 wxString f, d;
1125 if (ExtractWildcard(m_filter, m_currentFilter, f, d))
1126 m_currentFilterStr = f;
1127 else
1128 m_currentFilterStr = wxT("*.*");
1129 }
1130
1131 // Extract description and actual filter from overall filter string
1132 bool wxGenericDirCtrl::ExtractWildcard(const wxString& filterStr, int n, wxString& filter, wxString& description)
1133 {
1134 wxArrayString filters, descriptions;
1135 int count = ParseFilter(filterStr, filters, descriptions);
1136 if (count > 0 && n < count)
1137 {
1138 filter = filters[n];
1139 description = descriptions[n];
1140 return TRUE;
1141 }
1142
1143 return FALSE;
1144 }
1145
1146 // Parses the global filter, returning the number of filters.
1147 // Returns 0 if none or if there's a problem.
1148 // filterStr is in the form:
1149 //
1150 // "All files (*.*)|*.*|JPEG Files (*.jpeg)|*.jpg"
1151
1152 int wxGenericDirCtrl::ParseFilter(const wxString& filterStr, wxArrayString& filters, wxArrayString& descriptions)
1153 {
1154 wxString str(filterStr);
1155
1156 wxString description, filter;
1157 int pos;
1158 bool finished = FALSE;
1159 do
1160 {
1161 pos = str.Find(wxT('|'));
1162 if (pos == -1)
1163 return 0; // Problem
1164 description = str.Left(pos);
1165 str = str.Mid(pos+1);
1166 pos = str.Find(wxT('|'));
1167 if (pos == -1)
1168 {
1169 filter = str;
1170 finished = TRUE;
1171 }
1172 else
1173 {
1174 filter = str.Left(pos);
1175 str = str.Mid(pos+1);
1176 }
1177 descriptions.Add(description);
1178 filters.Add(filter);
1179 }
1180 while (!finished) ;
1181
1182 return filters.Count();
1183 }
1184
1185 void wxGenericDirCtrl::DoResize()
1186 {
1187 wxSize sz = GetClientSize();
1188 int verticalSpacing = 3;
1189 if (m_treeCtrl)
1190 {
1191 wxSize filterSz ;
1192 if (m_filterListCtrl)
1193 {
1194 #ifdef __WXMSW__
1195 // For some reason, this is required in order for the
1196 // correct control height to always be returned, rather
1197 // than the drop-down list height which is sometimes returned.
1198 wxSize oldSize = m_filterListCtrl->GetSize();
1199 m_filterListCtrl->SetSize(-1, -1, oldSize.x+10, -1, wxSIZE_USE_EXISTING);
1200 m_filterListCtrl->SetSize(-1, -1, oldSize.x, -1, wxSIZE_USE_EXISTING);
1201 #endif
1202 filterSz = m_filterListCtrl->GetSize();
1203 sz.y -= (filterSz.y + verticalSpacing);
1204 }
1205 m_treeCtrl->SetSize(0, 0, sz.x, sz.y);
1206 if (m_filterListCtrl)
1207 {
1208 m_filterListCtrl->SetSize(0, sz.y + verticalSpacing, sz.x, filterSz.y);
1209 // Don't know why, but this needs refreshing after a resize (wxMSW)
1210 m_filterListCtrl->Refresh();
1211 }
1212 }
1213 }
1214
1215
1216 void wxGenericDirCtrl::OnSize(wxSizeEvent& WXUNUSED(event))
1217 {
1218 DoResize();
1219 }
1220
1221 //-----------------------------------------------------------------------------
1222 // wxDirFilterListCtrl
1223 //-----------------------------------------------------------------------------
1224
1225 IMPLEMENT_CLASS(wxDirFilterListCtrl, wxChoice)
1226
1227 BEGIN_EVENT_TABLE(wxDirFilterListCtrl, wxChoice)
1228 EVT_CHOICE(-1, wxDirFilterListCtrl::OnSelFilter)
1229 END_EVENT_TABLE()
1230
1231 bool wxDirFilterListCtrl::Create(wxGenericDirCtrl* parent, const wxWindowID id,
1232 const wxPoint& pos,
1233 const wxSize& size,
1234 long style)
1235 {
1236 m_dirCtrl = parent;
1237 return wxChoice::Create(parent, id, pos, size, 0, NULL, style);
1238 }
1239
1240 void wxDirFilterListCtrl::Init()
1241 {
1242 m_dirCtrl = NULL;
1243 }
1244
1245 void wxDirFilterListCtrl::OnSelFilter(wxCommandEvent& WXUNUSED(event))
1246 {
1247 int sel = GetSelection();
1248
1249 wxString currentPath = m_dirCtrl->GetPath();
1250
1251 m_dirCtrl->SetFilterIndex(sel);
1252
1253 // If the filter has changed, the view is out of date, so
1254 // collapse the tree.
1255 m_dirCtrl->GetTreeCtrl()->Collapse(m_dirCtrl->GetRootId());
1256 m_dirCtrl->GetTreeCtrl()->Expand(m_dirCtrl->GetRootId());
1257
1258 // Try to restore the selection, or at least the directory
1259 m_dirCtrl->ExpandPath(currentPath);
1260 }
1261
1262 void wxDirFilterListCtrl::FillFilterList(const wxString& filter, int defaultFilter)
1263 {
1264 Clear();
1265 wxArrayString descriptions, filters;
1266 size_t n = (size_t) m_dirCtrl->ParseFilter(filter, filters, descriptions);
1267
1268 if (n > 0 && defaultFilter < (int) n)
1269 {
1270 size_t i = 0;
1271 for (i = 0; i < n; i++)
1272 Append(descriptions[i]);
1273 SetSelection(defaultFilter);
1274 }
1275 }
1276
1277 #endif // wxUSE_DIRDLG
wxWidgets (Impactor)