]> git.saurik.com Git - wxWidgets.git/blame_incremental - src/generic/dirctrlg.cpp
Some more fiddling with two-window approach and
[wxWidgets.git] / src / generic / dirctrlg.cpp
... / ...
CommitLineData
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 */
93static 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 */
122static 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 */
151static 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 */
177static 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 */
204static 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 */
231static 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 */
261static 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 */
288static 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
320bool 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
340int 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
369bool 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.
406static 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
418wxDirItemData::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
432wxDirItemData::~wxDirItemData()
433{
434}
435
436void wxDirItemData::SetNewDirName(const wxString& path)
437{
438 m_path = path;
439 m_name = wxFileNameFromPath(path);
440}
441
442bool 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
457bool 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
476IMPLEMENT_DYNAMIC_CLASS(wxGenericDirCtrl, wxControl)
477
478BEGIN_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)
484END_EVENT_TABLE()
485
486wxGenericDirCtrl::wxGenericDirCtrl(void)
487{
488 Init();
489}
490
491bool 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 | wxTR_EDIT_LABELS | wxTR_HIDE_ROOT;
509 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
510 treeStyle |= wxNO_BORDER;
511
512 long filterStyle = 0;
513 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
514 filterStyle |= wxNO_BORDER;
515
516 m_treeCtrl = new wxTreeCtrl(this, wxID_TREECTRL, pos, size, treeStyle);
517
518 if (!filter.IsEmpty() && (style & wxDIRCTRL_SHOW_FILTERS))
519 m_filterListCtrl = new wxDirFilterListCtrl(this, wxID_FILTERLISTCTRL, wxDefaultPosition, wxDefaultSize, filterStyle);
520
521 m_defaultPath = dir;
522 m_filter = filter;
523
524 SetFilterIndex(defaultFilter);
525
526 if (m_filterListCtrl)
527 m_filterListCtrl->FillFilterList(filter, defaultFilter);
528
529 m_imageList = new wxImageList(16, 16, TRUE);
530 m_imageList->Add(wxIcon(icon1_xpm));
531 m_imageList->Add(wxIcon(icon2_xpm));
532 m_imageList->Add(wxIcon(icon3_xpm));
533 m_imageList->Add(wxIcon(icon4_xpm));
534 m_imageList->Add(wxIcon(icon5_xpm));
535 m_imageList->Add(wxIcon(icon6_xpm));
536 m_imageList->Add(wxIcon(icon7_xpm));
537 m_imageList->Add(wxIcon(icon8_xpm));
538 m_treeCtrl->AssignImageList(m_imageList);
539
540 m_showHidden = FALSE;
541 wxDirItemData* rootData = new wxDirItemData(wxT(""), wxT(""), TRUE);
542
543 wxString rootName;
544
545#if defined(__WINDOWS__) || defined(__WXPM__) || defined(__DOS__)
546 rootName = _("Computer");
547#else
548 rootName = _("Sections");
549#endif
550
551 m_rootId = m_treeCtrl->AddRoot( rootName, 3, -1, rootData);
552 m_treeCtrl->SetItemHasChildren(m_rootId);
553 m_treeCtrl->Expand(m_rootId); // automatically expand first level
554
555 // Expand and select the default path
556 if (!m_defaultPath.IsEmpty())
557 ExpandPath(m_defaultPath);
558
559 DoResize();
560
561 return TRUE;
562}
563
564wxGenericDirCtrl::~wxGenericDirCtrl()
565{
566}
567
568void wxGenericDirCtrl::Init()
569{
570 m_showHidden = FALSE;
571 m_imageList = NULL;
572 m_currentFilter = 0;
573 m_currentFilterStr = wxEmptyString; // Default: any file
574 m_treeCtrl = NULL;
575 m_filterListCtrl = NULL;
576}
577
578void wxGenericDirCtrl::ShowHidden( bool show )
579{
580 m_showHidden = show;
581
582 wxString path = GetPath();
583 m_treeCtrl->Collapse(m_treeCtrl->GetRootItem());
584 m_treeCtrl->Expand(m_treeCtrl->GetRootItem());
585 SetPath(path);
586}
587
588void wxGenericDirCtrl::AddSection(const wxString& path, const wxString& name, int imageId)
589{
590 wxDirItemData *dir_item = new wxDirItemData(path,name,TRUE);
591
592 wxTreeItemId id = m_treeCtrl->AppendItem( m_rootId, name, imageId, -1, dir_item);
593
594 m_treeCtrl->SetItemHasChildren(id);
595}
596
597void wxGenericDirCtrl::SetupSections()
598{
599#if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
600
601#ifdef __WIN32__
602 wxChar driveBuffer[256];
603 size_t n = (size_t) GetLogicalDriveStrings(255, driveBuffer);
604 size_t i = 0;
605 while (i < n)
606 {
607 wxString path, name;
608 path.Printf(wxT("%c:\\"), driveBuffer[i]);
609 name.Printf(wxT("(%c:)"), driveBuffer[i]);
610
611 int imageId = 4;
612 int driveType = ::GetDriveType(path);
613 switch (driveType)
614 {
615 case DRIVE_REMOVABLE:
616 if (path == wxT("a:\\") || path == wxT("b:\\"))
617 imageId = 6; // Floppy
618 else
619 imageId = 7;
620 break;
621 case DRIVE_FIXED:
622 imageId = 4;
623 break;
624 case DRIVE_REMOTE:
625 imageId = 4;
626 break;
627 case DRIVE_CDROM:
628 imageId = 5;
629 break;
630 default:
631 imageId = 4;
632 break;
633 }
634
635 AddSection(path, name, imageId);
636
637 while (driveBuffer[i] != wxT('\0'))
638 i ++;
639 i ++;
640 if (driveBuffer[i] == wxT('\0'))
641 break;
642 }
643#else // !__WIN32__
644 int drive;
645
646 /* If we can switch to the drive, it exists. */
647 for( drive = 1; drive <= 26; drive++ )
648 {
649 wxString path, name;
650 path.Printf(wxT("%c:\\"), (char) (drive + 'a' - 1));
651 name.Printf(wxT("(%c:)"), (char) (drive + 'A' - 1));
652
653 if (wxIsDriveAvailable(path))
654 {
655 AddSection(path, name, (drive <= 2) ? 6/*floppy*/ : 4/*disk*/);
656 }
657 }
658#endif // __WIN32__/!__WIN32__
659
660#elif defined(__WXMAC__)
661 FSSpec volume ;
662 short index = 1 ;
663 while(1) {
664 short actualCount = 0 ;
665 if ( OnLine( &volume , 1 , &actualCount , &index ) != noErr || actualCount == 0 )
666 break ;
667
668 wxString name = wxMacFSSpec2MacFilename( &volume ) ;
669 AddSection(name + wxFILE_SEP_PATH, name, 0);
670 }
671#elif defined(__UNIX__)
672 AddSection(wxT("/"), wxT("/"), 3/*computer icon*/);
673#else
674 #error "Unsupported platform in wxGenericDirCtrl!"
675#endif
676}
677
678void wxGenericDirCtrl::OnBeginEditItem(wxTreeEvent &event)
679{
680 // don't rename the main entry "Sections"
681 if (event.GetItem() == m_rootId)
682 {
683 event.Veto();
684 return;
685 }
686
687 // don't rename the individual sections
688 if (m_treeCtrl->GetParent( event.GetItem() ) == m_rootId)
689 {
690 event.Veto();
691 return;
692 }
693}
694
695void wxGenericDirCtrl::OnEndEditItem(wxTreeEvent &event)
696{
697 if ((event.GetLabel().IsEmpty()) ||
698 (event.GetLabel() == _(".")) ||
699 (event.GetLabel() == _("..")) ||
700 (event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
701 {
702 wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
703 dialog.ShowModal();
704 event.Veto();
705 return;
706 }
707
708 wxTreeItemId id = event.GetItem();
709 wxDirItemData *data = (wxDirItemData*)m_treeCtrl->GetItemData( id );
710 wxASSERT( data );
711
712 wxString new_name( wxPathOnly( data->m_path ) );
713 new_name += wxString(wxFILE_SEP_PATH);
714 new_name += event.GetLabel();
715
716 wxLogNull log;
717
718 if (wxFileExists(new_name))
719 {
720 wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
721 dialog.ShowModal();
722 event.Veto();
723 }
724
725 if (wxRenameFile(data->m_path,new_name))
726 {
727 data->SetNewDirName( new_name );
728 }
729 else
730 {
731 wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
732 dialog.ShowModal();
733 event.Veto();
734 }
735}
736
737void wxGenericDirCtrl::OnExpandItem(wxTreeEvent &event)
738{
739 wxTreeItemId parentId = event.GetItem();
740
741 // VS: this is needed because the event handler is called from wxTreeCtrl
742 // ctor when wxTR_HIDE_ROOT was specified
743 if (m_rootId == 0)
744 m_rootId = m_treeCtrl->GetRootItem();
745
746 ExpandDir(parentId);
747}
748
749void wxGenericDirCtrl::OnCollapseItem(wxTreeEvent &event )
750{
751 wxTreeItemId child, parent = event.GetItem();
752
753 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(event.GetItem());
754 if (!data->m_isExpanded)
755 return;
756
757 data->m_isExpanded = FALSE;
758 long cookie;
759 /* Workaround because DeleteChildren has disapeared (why?) and
760 * CollapseAndReset doesn't work as advertised (deletes parent too) */
761 child = m_treeCtrl->GetFirstChild(parent, cookie);
762 while (child.IsOk())
763 {
764 m_treeCtrl->Delete(child);
765 /* Not GetNextChild below, because the cookie mechanism can't
766 * handle disappearing children! */
767 child = m_treeCtrl->GetFirstChild(parent, cookie);
768 }
769}
770
771void wxGenericDirCtrl::ExpandDir(wxTreeItemId parentId)
772{
773 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(parentId);
774
775 if (data->m_isExpanded)
776 return;
777
778 data->m_isExpanded = TRUE;
779
780 if (parentId == m_treeCtrl->GetRootItem())
781 {
782 SetupSections();
783 return;
784 }
785
786 wxASSERT(data);
787
788 wxString search,path,filename;
789
790 wxString dirName(data->m_path);
791
792#if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
793 // Check if this is a root directory and if so,
794 // whether the drive is avaiable.
795 if (!wxIsDriveAvailable(dirName))
796 {
797 data->m_isExpanded = FALSE;
798 //wxMessageBox(wxT("Sorry, this drive is not available."));
799 return;
800 }
801#endif
802
803 // This may take a longish time. Go to busy cursor
804 wxBusyCursor busy;
805
806#if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
807 if (dirName.Last() == ':')
808 dirName += wxString(wxFILE_SEP_PATH);
809#endif
810
811 wxArrayString dirs;
812 wxArrayString filenames;
813
814 wxDir d;
815 wxString eachFilename;
816
817 wxLogNull log;
818 d.Open(dirName);
819
820 if (d.IsOpened())
821 {
822 int style = wxDIR_DIRS;
823 if (m_showHidden) style |= wxDIR_HIDDEN;
824 if (d.GetFirst(& eachFilename, wxEmptyString, style))
825 {
826 do
827 {
828 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
829 {
830 dirs.Add(eachFilename);
831 }
832 }
833 while (d.GetNext(& eachFilename));
834 }
835 }
836 dirs.Sort((wxArrayString::CompareFunction) wxDirCtrlStringCompareFunction);
837
838 // Now do the filenames -- but only if we're allowed to
839 if ((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0)
840 {
841 wxLogNull log;
842
843 d.Open(dirName);
844
845 if (d.IsOpened())
846 {
847 if (d.GetFirst(& eachFilename, m_currentFilterStr, wxDIR_FILES))
848 {
849 do
850 {
851 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
852 {
853 filenames.Add(eachFilename);
854 }
855 }
856 while (d.GetNext(& eachFilename));
857 }
858 }
859 filenames.Sort((wxArrayString::CompareFunction) wxDirCtrlStringCompareFunction);
860 }
861
862 // Add the sorted dirs
863 size_t i;
864 for (i = 0; i < dirs.Count(); i++)
865 {
866 wxString eachFilename(dirs[i]);
867 path = dirName;
868 if (path.Last() != wxFILE_SEP_PATH)
869 path += wxString(wxFILE_SEP_PATH);
870 path += eachFilename;
871
872 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,TRUE);
873 wxTreeItemId id = m_treeCtrl->AppendItem( parentId, eachFilename, 0, -1, dir_item);
874 m_treeCtrl->SetItemImage( id, 1, wxTreeItemIcon_Expanded );
875
876 // Has this got any children? If so, make it expandable.
877 // (There are two situations when a dir has children: either it
878 // has subdirectories or it contains files that weren't filtered
879 // out. The latter only applies to dirctrl with files.)
880 if ( dir_item->HasSubDirs() ||
881 (((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0) &&
882 dir_item->HasFiles(m_currentFilterStr)) )
883 {
884 m_treeCtrl->SetItemHasChildren(id);
885 }
886 }
887
888 // Add the sorted filenames
889 if ((GetWindowStyle() & wxDIRCTRL_DIR_ONLY) == 0)
890 {
891 for (i = 0; i < filenames.Count(); i++)
892 {
893 wxString eachFilename(filenames[i]);
894 path = dirName;
895 if (path.Last() != wxFILE_SEP_PATH)
896 path += wxString(wxFILE_SEP_PATH);
897 path += eachFilename;
898 //path = dirName + wxString(wxT("/")) + eachFilename;
899 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,FALSE);
900 (void)m_treeCtrl->AppendItem( parentId, eachFilename, 2, -1, dir_item);
901 }
902 }
903}
904
905// Find the child that matches the first part of 'path'.
906// E.g. if a child path is "/usr" and 'path' is "/usr/include"
907// then the child for /usr is returned.
908wxTreeItemId wxGenericDirCtrl::FindChild(wxTreeItemId parentId, const wxString& path, bool& done)
909{
910 wxString path2(path);
911
912 // Make sure all separators are as per the current platform
913 path2.Replace(wxT("\\"), wxString(wxFILE_SEP_PATH));
914 path2.Replace(wxT("/"), wxString(wxFILE_SEP_PATH));
915
916 // Append a separator to foil bogus substring matching
917 path2 += wxString(wxFILE_SEP_PATH);
918
919 // In MSW or PM, case is not significant
920#if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
921 path2.MakeLower();
922#endif
923
924 long cookie;
925 wxTreeItemId childId = m_treeCtrl->GetFirstChild(parentId, cookie);
926 while (childId.IsOk())
927 {
928 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
929
930 if (data && !data->m_path.IsEmpty())
931 {
932 wxString childPath(data->m_path);
933 if (childPath.Last() != wxFILE_SEP_PATH)
934 childPath += wxString(wxFILE_SEP_PATH);
935
936 // In MSW and PM, case is not significant
937#if defined(__WINDOWS__) || defined(__DOS__) || defined(__WXPM__)
938 childPath.MakeLower();
939#endif
940
941 if (childPath.Len() <= path2.Len())
942 {
943 wxString path3 = path2.Mid(0, childPath.Len());
944 if (childPath == path3)
945 {
946 if (path3.Len() == path2.Len())
947 done = TRUE;
948 else
949 done = FALSE;
950 return childId;
951 }
952 }
953 }
954
955 childId = m_treeCtrl->GetNextChild(parentId, cookie);
956 }
957 wxTreeItemId invalid;
958 return invalid;
959}
960
961// Try to expand as much of the given path as possible,
962// and select the given tree item.
963bool wxGenericDirCtrl::ExpandPath(const wxString& path)
964{
965 bool done = FALSE;
966 wxTreeItemId id = FindChild(m_rootId, path, done);
967 wxTreeItemId lastId = id; // The last non-zero id
968 while (id.IsOk() && !done)
969 {
970 ExpandDir(id);
971
972 id = FindChild(id, path, done);
973 if (id.IsOk())
974 lastId = id;
975 }
976 if (lastId.IsOk())
977 {
978 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(lastId);
979 if (data->m_isDir)
980 {
981 m_treeCtrl->Expand(lastId);
982 }
983 if ((GetWindowStyle() & wxDIRCTRL_SELECT_FIRST) && data->m_isDir)
984 {
985 // Find the first file in this directory
986 long cookie;
987 wxTreeItemId childId = m_treeCtrl->GetFirstChild(lastId, cookie);
988 bool selectedChild = FALSE;
989 while (childId.IsOk())
990 {
991 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
992
993 if (data && data->m_path != "" && !data->m_isDir)
994 {
995 m_treeCtrl->SelectItem(childId);
996 m_treeCtrl->EnsureVisible(childId);
997 selectedChild = TRUE;
998 break;
999 }
1000 childId = m_treeCtrl->GetNextChild(lastId, cookie);
1001 }
1002 if (!selectedChild)
1003 {
1004 m_treeCtrl->SelectItem(lastId);
1005 m_treeCtrl->EnsureVisible(lastId);
1006 }
1007 }
1008 else
1009 {
1010 m_treeCtrl->SelectItem(lastId);
1011 m_treeCtrl->EnsureVisible(lastId);
1012 }
1013
1014 return TRUE;
1015 }
1016 else
1017 return FALSE;
1018}
1019
1020wxString wxGenericDirCtrl::GetPath() const
1021{
1022 wxTreeItemId id = m_treeCtrl->GetSelection();
1023 if (id)
1024 {
1025 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1026 return data->m_path;
1027 }
1028 else
1029 return wxEmptyString;
1030}
1031
1032wxString wxGenericDirCtrl::GetFilePath() const
1033{
1034 wxTreeItemId id = m_treeCtrl->GetSelection();
1035 if (id)
1036 {
1037 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1038 if (data->m_isDir)
1039 return wxEmptyString;
1040 else
1041 return data->m_path;
1042 }
1043 else
1044 return wxEmptyString;
1045}
1046
1047void wxGenericDirCtrl::SetPath(const wxString& path)
1048{
1049 m_defaultPath = path;
1050 if (m_rootId)
1051 ExpandPath(path);
1052}
1053
1054// Not used
1055#if 0
1056void wxGenericDirCtrl::FindChildFiles(wxTreeItemId id, int dirFlags, wxArrayString& filenames)
1057{
1058 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(id);
1059
1060 // This may take a longish time. Go to busy cursor
1061 wxBusyCursor busy;
1062
1063 wxASSERT(data);
1064
1065 wxString search,path,filename;
1066
1067 wxString dirName(data->m_path);
1068
1069#if defined(__WXMSW__) || defined(__WXPM__)
1070 if (dirName.Last() == ':')
1071 dirName += wxString(wxFILE_SEP_PATH);
1072#endif
1073
1074 wxDir d;
1075 wxString eachFilename;
1076
1077 wxLogNull log;
1078 d.Open(dirName);
1079
1080 if (d.IsOpened())
1081 {
1082 if (d.GetFirst(& eachFilename, m_currentFilterStr, dirFlags))
1083 {
1084 do
1085 {
1086 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
1087 {
1088 filenames.Add(eachFilename);
1089 }
1090 }
1091 while (d.GetNext(& eachFilename)) ;
1092 }
1093 }
1094}
1095#endif
1096
1097void wxGenericDirCtrl::SetFilterIndex(int n)
1098{
1099 m_currentFilter = n;
1100
1101 wxString f, d;
1102 if (ExtractWildcard(m_filter, n, f, d))
1103 m_currentFilterStr = f;
1104 else
1105 m_currentFilterStr = wxT("*.*");
1106}
1107
1108void wxGenericDirCtrl::SetFilter(const wxString& filter)
1109{
1110 m_filter = filter;
1111
1112 wxString f, d;
1113 if (ExtractWildcard(m_filter, m_currentFilter, f, d))
1114 m_currentFilterStr = f;
1115 else
1116 m_currentFilterStr = wxT("*.*");
1117}
1118
1119// Extract description and actual filter from overall filter string
1120bool wxGenericDirCtrl::ExtractWildcard(const wxString& filterStr, int n, wxString& filter, wxString& description)
1121{
1122 wxArrayString filters, descriptions;
1123 int count = ParseFilter(filterStr, filters, descriptions);
1124 if (count > 0 && n < count)
1125 {
1126 filter = filters[n];
1127 description = descriptions[n];
1128 return TRUE;
1129 }
1130
1131 return FALSE;
1132}
1133
1134// Parses the global filter, returning the number of filters.
1135// Returns 0 if none or if there's a problem.
1136// filterStr is in the form:
1137//
1138// "All files (*.*)|*.*|JPEG Files (*.jpeg)|*.jpg"
1139
1140int wxGenericDirCtrl::ParseFilter(const wxString& filterStr, wxArrayString& filters, wxArrayString& descriptions)
1141{
1142 wxString str(filterStr);
1143
1144 wxString description, filter;
1145 int pos;
1146 bool finished = FALSE;
1147 do
1148 {
1149 pos = str.Find(wxT('|'));
1150 if (pos == -1)
1151 return 0; // Problem
1152 description = str.Left(pos);
1153 str = str.Mid(pos+1);
1154 pos = str.Find(wxT('|'));
1155 if (pos == -1)
1156 {
1157 filter = str;
1158 finished = TRUE;
1159 }
1160 else
1161 {
1162 filter = str.Left(pos);
1163 str = str.Mid(pos+1);
1164 }
1165 descriptions.Add(description);
1166 filters.Add(filter);
1167 }
1168 while (!finished) ;
1169
1170 return filters.Count();
1171}
1172
1173void wxGenericDirCtrl::DoResize()
1174{
1175 wxSize sz = GetClientSize();
1176 int verticalSpacing = 3;
1177 if (m_treeCtrl)
1178 {
1179 wxSize filterSz ;
1180 if (m_filterListCtrl)
1181 {
1182 filterSz = m_filterListCtrl->GetSize();
1183 sz.y -= (filterSz.y + verticalSpacing);
1184 }
1185 m_treeCtrl->SetSize(0, 0, sz.x, sz.y);
1186 if (m_filterListCtrl)
1187 {
1188 m_filterListCtrl->SetSize(0, sz.y + verticalSpacing, sz.x, filterSz.y);
1189 // Don't know why, but this needs refreshing after a resize (wxMSW)
1190 m_filterListCtrl->Refresh();
1191 }
1192 }
1193}
1194
1195
1196void wxGenericDirCtrl::OnSize(wxSizeEvent& WXUNUSED(event))
1197{
1198 DoResize();
1199}
1200
1201//-----------------------------------------------------------------------------
1202// wxDirFilterListCtrl
1203//-----------------------------------------------------------------------------
1204
1205IMPLEMENT_CLASS(wxDirFilterListCtrl, wxChoice)
1206
1207BEGIN_EVENT_TABLE(wxDirFilterListCtrl, wxChoice)
1208 EVT_CHOICE(-1, wxDirFilterListCtrl::OnSelFilter)
1209END_EVENT_TABLE()
1210
1211bool wxDirFilterListCtrl::Create(wxGenericDirCtrl* parent, const wxWindowID id,
1212 const wxPoint& pos,
1213 const wxSize& size,
1214 long style)
1215{
1216 m_dirCtrl = parent;
1217 return wxChoice::Create(parent, id, pos, size, 0, NULL, style);
1218}
1219
1220void wxDirFilterListCtrl::Init()
1221{
1222 m_dirCtrl = NULL;
1223}
1224
1225void wxDirFilterListCtrl::OnSelFilter(wxCommandEvent& WXUNUSED(event))
1226{
1227 int sel = GetSelection();
1228
1229 wxString currentPath = m_dirCtrl->GetPath();
1230
1231 m_dirCtrl->SetFilterIndex(sel);
1232
1233 // If the filter has changed, the view is out of date, so
1234 // collapse the tree.
1235 m_dirCtrl->GetTreeCtrl()->Collapse(m_dirCtrl->GetRootId());
1236 m_dirCtrl->GetTreeCtrl()->Expand(m_dirCtrl->GetRootId());
1237
1238 // Try to restore the selection, or at least the directory
1239 m_dirCtrl->ExpandPath(currentPath);
1240}
1241
1242void wxDirFilterListCtrl::FillFilterList(const wxString& filter, int defaultFilter)
1243{
1244 Clear();
1245 wxArrayString descriptions, filters;
1246 size_t n = (size_t) m_dirCtrl->ParseFilter(filter, filters, descriptions);
1247
1248 if (n > 0 && defaultFilter < (int) n)
1249 {
1250 size_t i = 0;
1251 for (i = 0; i < n; i++)
1252 Append(descriptions[i]);
1253 SetSelection(defaultFilter);
1254 }
1255}
1256
1257#endif // wxUSE_DIRDLG