projects / wxWidgets.git / blob
? search:


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