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


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