]> git.saurik.com Git - wxWidgets.git/blame_incremental - src/generic/dirctrlg.cpp
Add a WX_APPNAME_DATA_DIR hack for wxStandardPaths::GetDataDir().
[wxWidgets.git] / src / generic / dirctrlg.cpp
... / ...
CommitLineData
1/////////////////////////////////////////////////////////////////////////////
2// Name: src/generic/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// For compilers that support precompilation, includes "wx.h".
13#include "wx/wxprec.h"
14
15#ifdef __BORLANDC__
16 #pragma hdrstop
17#endif
18
19#if wxUSE_DIRDLG || wxUSE_FILEDLG
20
21#include "wx/generic/dirctrlg.h"
22
23#ifndef WX_PRECOMP
24 #include "wx/hash.h"
25 #include "wx/intl.h"
26 #include "wx/log.h"
27 #include "wx/utils.h"
28 #include "wx/button.h"
29 #include "wx/icon.h"
30 #include "wx/settings.h"
31 #include "wx/msgdlg.h"
32 #include "wx/cmndata.h"
33 #include "wx/choice.h"
34 #include "wx/textctrl.h"
35 #include "wx/layout.h"
36 #include "wx/sizer.h"
37 #include "wx/textdlg.h"
38 #include "wx/gdicmn.h"
39 #include "wx/image.h"
40 #include "wx/module.h"
41#endif
42
43#include "wx/filename.h"
44#include "wx/filefn.h"
45#include "wx/imaglist.h"
46#include "wx/tokenzr.h"
47#include "wx/dir.h"
48#include "wx/artprov.h"
49#include "wx/mimetype.h"
50
51#if wxUSE_STATLINE
52 #include "wx/statline.h"
53#endif
54
55#if defined(__WXMAC__)
56 #include "wx/osx/private.h" // includes mac headers
57#endif
58
59#ifdef __WXMSW__
60#include <windows.h>
61#include "wx/msw/winundef.h"
62#include "wx/volume.h"
63
64// FIXME - Mingw32 1.0 has both _getdrive() and _chdrive(). For now, let's assume
65// older releases don't, but it should be verified and the checks modified
66// accordingly.
67#if !defined(__GNUWIN32__) || (defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
68#if !defined(__WXWINCE__)
69 #include <direct.h>
70#endif
71 #include <stdlib.h>
72 #include <ctype.h>
73#endif
74
75#endif
76
77#if defined(__OS2__) || defined(__DOS__)
78 #ifdef __OS2__
79 #define INCL_BASE
80 #include <os2.h>
81 #ifndef __EMX__
82 #include <direct.h>
83 #endif
84 #include <stdlib.h>
85 #include <ctype.h>
86 #endif
87#endif // __OS2__
88
89#if defined(__WXMAC__)
90// #include "MoreFilesX.h"
91#endif
92
93#ifdef __BORLANDC__
94 #include "dos.h"
95#endif
96
97extern WXDLLEXPORT_DATA(const char) wxFileSelectorDefaultWildcardStr[];
98
99// If compiled under Windows, this macro can cause problems
100#ifdef GetFirstChild
101#undef GetFirstChild
102#endif
103
104bool wxIsDriveAvailable(const wxString& dirName);
105
106// ----------------------------------------------------------------------------
107// wxGetAvailableDrives, for WINDOWS, DOS, OS2, MAC, UNIX (returns "/")
108// ----------------------------------------------------------------------------
109
110size_t wxGetAvailableDrives(wxArrayString &paths, wxArrayString &names, wxArrayInt &icon_ids)
111{
112#ifdef wxHAS_FILESYSTEM_VOLUMES
113
114#ifdef __WXWINCE__
115 // No logical drives; return "\"
116 paths.Add(wxT("\\"));
117 names.Add(wxT("\\"));
118 icon_ids.Add(wxFileIconsTable::computer);
119#elif defined(__WIN32__) && wxUSE_FSVOLUME
120 // TODO: this code (using wxFSVolumeBase) should be used for all platforms
121 // but unfortunately wxFSVolumeBase is not implemented everywhere
122 const wxArrayString as = wxFSVolumeBase::GetVolumes();
123
124 for (size_t i = 0; i < as.GetCount(); i++)
125 {
126 wxString path = as[i];
127 wxFSVolume vol(path);
128 int imageId;
129 switch (vol.GetKind())
130 {
131 case wxFS_VOL_FLOPPY:
132 if ( (path == wxT("a:\\")) || (path == wxT("b:\\")) )
133 imageId = wxFileIconsTable::floppy;
134 else
135 imageId = wxFileIconsTable::removeable;
136 break;
137 case wxFS_VOL_DVDROM:
138 case wxFS_VOL_CDROM:
139 imageId = wxFileIconsTable::cdrom;
140 break;
141 case wxFS_VOL_NETWORK:
142 if (path[0] == wxT('\\'))
143 continue; // skip "\\computer\folder"
144 imageId = wxFileIconsTable::drive;
145 break;
146 case wxFS_VOL_DISK:
147 case wxFS_VOL_OTHER:
148 default:
149 imageId = wxFileIconsTable::drive;
150 break;
151 }
152 paths.Add(path);
153 names.Add(vol.GetDisplayName());
154 icon_ids.Add(imageId);
155 }
156#elif defined(__OS2__)
157 APIRET rc;
158 ULONG ulDriveNum = 0;
159 ULONG ulDriveMap = 0;
160 rc = ::DosQueryCurrentDisk(&ulDriveNum, &ulDriveMap);
161 if ( rc == 0)
162 {
163 size_t i = 0;
164 while (i < 26)
165 {
166 if (ulDriveMap & ( 1 << i ))
167 {
168 const wxString path = wxFileName::GetVolumeString(
169 'A' + i, wxPATH_GET_SEPARATOR);
170 const wxString name = wxFileName::GetVolumeString(
171 'A' + i, wxPATH_NO_SEPARATOR);
172
173 // Note: If _filesys is unsupported by some compilers,
174 // we can always replace it by DosQueryFSAttach
175 char filesysname[20];
176#ifdef __WATCOMC__
177 ULONG cbBuffer = sizeof(filesysname);
178 PFSQBUFFER2 pfsqBuffer = (PFSQBUFFER2)filesysname;
179 APIRET rc = ::DosQueryFSAttach(name.fn_str(),0,FSAIL_QUERYNAME,pfsqBuffer,&cbBuffer);
180 if (rc != NO_ERROR)
181 {
182 filesysname[0] = '\0';
183 }
184#else
185 _filesys(name.fn_str(), filesysname, sizeof(filesysname));
186#endif
187 /* FAT, LAN, HPFS, CDFS, NFS */
188 int imageId;
189 if (path == wxT("A:\\") || path == wxT("B:\\"))
190 imageId = wxFileIconsTable::floppy;
191 else if (!strcmp(filesysname, "CDFS"))
192 imageId = wxFileIconsTable::cdrom;
193 else if (!strcmp(filesysname, "LAN") ||
194 !strcmp(filesysname, "NFS"))
195 imageId = wxFileIconsTable::drive;
196 else
197 imageId = wxFileIconsTable::drive;
198 paths.Add(path);
199 names.Add(name);
200 icon_ids.Add(imageId);
201 }
202 i ++;
203 }
204 }
205#else // !__WIN32__, !__OS2__
206 /* If we can switch to the drive, it exists. */
207 for ( char drive = 'A'; drive <= 'Z'; drive++ )
208 {
209 const wxString
210 path = wxFileName::GetVolumeString(drive, wxPATH_GET_SEPARATOR);
211
212 if (wxIsDriveAvailable(path))
213 {
214 paths.Add(path);
215 names.Add(wxFileName::GetVolumeString(drive, wxPATH_NO_SEPARATOR));
216 icon_ids.Add(drive <= 2 ? wxFileIconsTable::floppy
217 : wxFileIconsTable::drive);
218 }
219 }
220#endif // __WIN32__/!__WIN32__
221
222#elif defined(__WXMAC__) && wxOSX_USE_COCOA_OR_CARBON
223
224 ItemCount volumeIndex = 1;
225 OSErr err = noErr ;
226
227 while( noErr == err )
228 {
229 HFSUniStr255 volumeName ;
230 FSRef fsRef ;
231 FSVolumeInfo volumeInfo ;
232 err = FSGetVolumeInfo(0, volumeIndex, NULL, kFSVolInfoFlags , &volumeInfo , &volumeName, &fsRef);
233 if( noErr == err )
234 {
235 wxString path = wxMacFSRefToPath( &fsRef ) ;
236 wxString name = wxMacHFSUniStrToString( &volumeName ) ;
237
238 if ( (volumeInfo.flags & kFSVolFlagSoftwareLockedMask) || (volumeInfo.flags & kFSVolFlagHardwareLockedMask) )
239 {
240 icon_ids.Add(wxFileIconsTable::cdrom);
241 }
242 else
243 {
244 icon_ids.Add(wxFileIconsTable::drive);
245 }
246 // todo other removable
247
248 paths.Add(path);
249 names.Add(name);
250 volumeIndex++ ;
251 }
252 }
253
254#elif defined(__UNIX__) || defined(__WXPALMOS__)
255 paths.Add(wxT("/"));
256 names.Add(wxT("/"));
257 icon_ids.Add(wxFileIconsTable::computer);
258#else
259 #error "Unsupported platform in wxGenericDirCtrl!"
260#endif
261 wxASSERT_MSG( (paths.GetCount() == names.GetCount()), wxT("The number of paths and their human readable names should be equal in number."));
262 wxASSERT_MSG( (paths.GetCount() == icon_ids.GetCount()), wxT("Wrong number of icons for available drives."));
263 return paths.GetCount();
264}
265
266// ----------------------------------------------------------------------------
267// wxIsDriveAvailable
268// ----------------------------------------------------------------------------
269
270#if defined(__DOS__)
271
272bool wxIsDriveAvailable(const wxString& dirName)
273{
274 // FIXME_MGL - this method leads to hang up under Watcom for some reason
275#ifdef __WATCOMC__
276 wxUnusedVar(dirName);
277#else
278 if ( dirName.length() == 3 && dirName[1u] == wxT(':') )
279 {
280 wxString dirNameLower(dirName.Lower());
281 // VS: always return true for removable media, since Win95 doesn't
282 // like it when MS-DOS app accesses empty floppy drive
283 return (dirNameLower[0u] == wxT('a') ||
284 dirNameLower[0u] == wxT('b') ||
285 wxDirExists(dirNameLower));
286 }
287 else
288#endif
289 return true;
290}
291
292#elif defined(__WINDOWS__) || defined(__OS2__)
293
294int setdrive(int WXUNUSED_IN_WINCE(drive))
295{
296#ifdef __WXWINCE__
297 return 0;
298#elif defined(__GNUWIN32__) && \
299 (defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
300 return _chdrive(drive);
301#else
302 wxChar newdrive[4];
303
304 if (drive < 1 || drive > 31)
305 return -1;
306 newdrive[0] = (wxChar)(wxT('A') + drive - 1);
307 newdrive[1] = wxT(':');
308#ifdef __OS2__
309 newdrive[2] = wxT('\\');
310 newdrive[3] = wxT('\0');
311#else
312 newdrive[2] = wxT('\0');
313#endif
314#if defined(__WXMSW__)
315 if (::SetCurrentDirectory(newdrive))
316#else
317 // VA doesn't know what LPSTR is and has its own set
318 if (!DosSetCurrentDir((PSZ)newdrive))
319#endif
320 return 0;
321 else
322 return -1;
323#endif // !GNUWIN32
324}
325
326bool wxIsDriveAvailable(const wxString& WXUNUSED_IN_WINCE(dirName))
327{
328#ifdef __WXWINCE__
329 return false;
330#else
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 available.
338 if (dirName.length() == 3 && dirName[(size_t)1] == wxT(':'))
339 {
340 wxString dirNameLower(dirName.Lower());
341#if defined(__GNUWIN32__) && !(defined(__MINGW32_MAJOR_VERSION) && __MINGW32_MAJOR_VERSION >= 1)
342 success = wxDirExists(dirNameLower);
343#else
344 #if defined(__OS2__)
345 // Avoid changing to drive since no media may be inserted.
346 if (dirNameLower[(size_t)0] == 'a' || dirNameLower[(size_t)0] == 'b')
347 return success;
348 #endif
349 int currentDrive = _getdrive();
350 int thisDrive = (int) (dirNameLower[(size_t)0] - 'a' + 1) ;
351 int err = setdrive( thisDrive ) ;
352 setdrive( currentDrive );
353
354 if (err == -1)
355 {
356 success = false;
357 }
358#endif
359 }
360#ifdef __WIN32__
361 (void) SetErrorMode(errorMode);
362#endif
363
364 return success;
365#endif
366}
367#endif // __WINDOWS__ || __OS2__
368
369#endif // wxUSE_DIRDLG || wxUSE_FILEDLG
370
371
372
373#if wxUSE_DIRDLG
374
375// Function which is called by quick sort. We want to override the default wxArrayString behaviour,
376// and sort regardless of case.
377static int wxCMPFUNC_CONV wxDirCtrlStringCompareFunction(const wxString& strFirst, const wxString& strSecond)
378{
379 return strFirst.CmpNoCase(strSecond);
380}
381
382//-----------------------------------------------------------------------------
383// wxDirItemData
384//-----------------------------------------------------------------------------
385
386wxDirItemData::wxDirItemData(const wxString& path, const wxString& name,
387 bool isDir)
388{
389 m_path = path;
390 m_name = name;
391 /* Insert logic to detect hidden files here
392 * In UnixLand we just check whether the first char is a dot
393 * For FileNameFromPath read LastDirNameInThisPath ;-) */
394 // m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
395 m_isHidden = false;
396 m_isExpanded = false;
397 m_isDir = isDir;
398}
399
400void wxDirItemData::SetNewDirName(const wxString& path)
401{
402 m_path = path;
403 m_name = wxFileNameFromPath(path);
404}
405
406bool wxDirItemData::HasSubDirs() const
407{
408 if (m_path.empty())
409 return false;
410
411 wxDir dir;
412 {
413 wxLogNull nolog;
414 if ( !dir.Open(m_path) )
415 return false;
416 }
417
418 return dir.HasSubDirs();
419}
420
421bool wxDirItemData::HasFiles(const wxString& WXUNUSED(spec)) const
422{
423 if (m_path.empty())
424 return false;
425
426 wxDir dir;
427 {
428 wxLogNull nolog;
429 if ( !dir.Open(m_path) )
430 return false;
431 }
432
433 return dir.HasFiles();
434}
435
436//-----------------------------------------------------------------------------
437// wxGenericDirCtrl
438//-----------------------------------------------------------------------------
439
440
441#if wxUSE_EXTENDED_RTTI
442WX_DEFINE_FLAGS( wxGenericDirCtrlStyle )
443
444wxBEGIN_FLAGS( wxGenericDirCtrlStyle )
445 // new style border flags, we put them first to
446 // use them for streaming out
447 wxFLAGS_MEMBER(wxBORDER_SIMPLE)
448 wxFLAGS_MEMBER(wxBORDER_SUNKEN)
449 wxFLAGS_MEMBER(wxBORDER_DOUBLE)
450 wxFLAGS_MEMBER(wxBORDER_RAISED)
451 wxFLAGS_MEMBER(wxBORDER_STATIC)
452 wxFLAGS_MEMBER(wxBORDER_NONE)
453
454 // old style border flags
455 wxFLAGS_MEMBER(wxSIMPLE_BORDER)
456 wxFLAGS_MEMBER(wxSUNKEN_BORDER)
457 wxFLAGS_MEMBER(wxDOUBLE_BORDER)
458 wxFLAGS_MEMBER(wxRAISED_BORDER)
459 wxFLAGS_MEMBER(wxSTATIC_BORDER)
460 wxFLAGS_MEMBER(wxBORDER)
461
462 // standard window styles
463 wxFLAGS_MEMBER(wxTAB_TRAVERSAL)
464 wxFLAGS_MEMBER(wxCLIP_CHILDREN)
465 wxFLAGS_MEMBER(wxTRANSPARENT_WINDOW)
466 wxFLAGS_MEMBER(wxWANTS_CHARS)
467 wxFLAGS_MEMBER(wxFULL_REPAINT_ON_RESIZE)
468 wxFLAGS_MEMBER(wxALWAYS_SHOW_SB )
469 wxFLAGS_MEMBER(wxVSCROLL)
470 wxFLAGS_MEMBER(wxHSCROLL)
471
472 wxFLAGS_MEMBER(wxDIRCTRL_DIR_ONLY)
473 wxFLAGS_MEMBER(wxDIRCTRL_3D_INTERNAL)
474 wxFLAGS_MEMBER(wxDIRCTRL_SELECT_FIRST)
475 wxFLAGS_MEMBER(wxDIRCTRL_MULTIPLE)
476
477wxEND_FLAGS( wxGenericDirCtrlStyle )
478
479IMPLEMENT_DYNAMIC_CLASS_XTI(wxGenericDirCtrl, wxControl,"wx/dirctrl.h")
480
481wxBEGIN_PROPERTIES_TABLE(wxGenericDirCtrl)
482 wxHIDE_PROPERTY( Children )
483 wxPROPERTY( DefaultPath , wxString , SetDefaultPath , GetDefaultPath , EMPTY_MACROVALUE , 0 /*flags*/ , wxT("Helpstring") , wxT("group"))
484 wxPROPERTY( Filter , wxString , SetFilter , GetFilter , EMPTY_MACROVALUE , 0 /*flags*/ , wxT("Helpstring") , wxT("group") )
485 wxPROPERTY( DefaultFilter , int , SetFilterIndex, GetFilterIndex, EMPTY_MACROVALUE , 0 /*flags*/ , wxT("Helpstring") , wxT("group") )
486 wxPROPERTY_FLAGS( WindowStyle, wxGenericDirCtrlStyle, long, SetWindowStyleFlag, GetWindowStyleFlag, EMPTY_MACROVALUE , 0, wxT("Helpstring"), wxT("group") )
487wxEND_PROPERTIES_TABLE()
488
489wxBEGIN_HANDLERS_TABLE(wxGenericDirCtrl)
490wxEND_HANDLERS_TABLE()
491
492wxCONSTRUCTOR_8( wxGenericDirCtrl , wxWindow* , Parent , wxWindowID , Id , wxString , DefaultPath ,
493 wxPoint , Position , wxSize , Size , long , WindowStyle , wxString , Filter , int , DefaultFilter )
494#else
495IMPLEMENT_DYNAMIC_CLASS(wxGenericDirCtrl, wxControl)
496#endif
497
498BEGIN_EVENT_TABLE(wxGenericDirCtrl, wxControl)
499 EVT_TREE_ITEM_EXPANDING (wxID_TREECTRL, wxGenericDirCtrl::OnExpandItem)
500 EVT_TREE_ITEM_COLLAPSED (wxID_TREECTRL, wxGenericDirCtrl::OnCollapseItem)
501 EVT_TREE_BEGIN_LABEL_EDIT (wxID_TREECTRL, wxGenericDirCtrl::OnBeginEditItem)
502 EVT_TREE_END_LABEL_EDIT (wxID_TREECTRL, wxGenericDirCtrl::OnEndEditItem)
503 EVT_SIZE (wxGenericDirCtrl::OnSize)
504END_EVENT_TABLE()
505
506wxGenericDirCtrl::wxGenericDirCtrl(void)
507{
508 Init();
509}
510
511void wxGenericDirCtrl::ExpandRoot()
512{
513 ExpandDir(m_rootId); // automatically expand first level
514
515 // Expand and select the default path
516 if (!m_defaultPath.empty())
517 {
518 ExpandPath(m_defaultPath);
519 }
520#ifdef __UNIX__
521 else
522 {
523 // On Unix, there's only one node under the (hidden) root node. It
524 // represents the / path, so the user would always have to expand it;
525 // let's do it ourselves
526 ExpandPath( wxT("/") );
527 }
528#endif
529}
530
531bool wxGenericDirCtrl::Create(wxWindow *parent,
532 const wxWindowID id,
533 const wxString& dir,
534 const wxPoint& pos,
535 const wxSize& size,
536 long style,
537 const wxString& filter,
538 int defaultFilter,
539 const wxString& name)
540{
541 if (!wxControl::Create(parent, id, pos, size, style, wxDefaultValidator, name))
542 return false;
543
544 SetBackgroundColour(wxSystemSettings::GetColour(wxSYS_COLOUR_3DFACE));
545 SetForegroundColour(wxSystemSettings::GetColour(wxSYS_COLOUR_WINDOWTEXT));
546
547 Init();
548
549 long treeStyle = wxTR_HAS_BUTTONS;
550
551 // On Windows CE, if you hide the root, you get a crash when
552 // attempting to access data for children of the root item.
553#ifndef __WXWINCE__
554 treeStyle |= wxTR_HIDE_ROOT;
555#endif
556
557#ifdef __WXGTK20__
558 treeStyle |= wxTR_NO_LINES;
559#endif
560
561 if (style & wxDIRCTRL_EDIT_LABELS)
562 treeStyle |= wxTR_EDIT_LABELS;
563
564 if (style & wxDIRCTRL_MULTIPLE)
565 treeStyle |= wxTR_MULTIPLE;
566
567 if ((style & wxDIRCTRL_3D_INTERNAL) == 0)
568 treeStyle |= wxNO_BORDER;
569
570 m_treeCtrl = CreateTreeCtrl(this, wxID_TREECTRL,
571 wxPoint(0,0), GetClientSize(), treeStyle);
572
573 if (!filter.empty())
574 m_filterListCtrl = new wxDirFilterListCtrl(this, wxID_FILTERLISTCTRL);
575
576 m_defaultPath = dir;
577 m_filter = filter;
578
579 if (m_filter.empty())
580 m_filter = wxFileSelectorDefaultWildcardStr;
581
582 SetFilterIndex(defaultFilter);
583
584 if (m_filterListCtrl)
585 m_filterListCtrl->FillFilterList(filter, defaultFilter);
586
587 m_treeCtrl->SetImageList(wxTheFileIconsTable->GetSmallImageList());
588
589 m_showHidden = false;
590 wxDirItemData* rootData = new wxDirItemData(wxEmptyString, wxEmptyString, true);
591
592 wxString rootName;
593
594#if defined(__WINDOWS__) || defined(__OS2__) || defined(__DOS__)
595 rootName = _("Computer");
596#else
597 rootName = _("Sections");
598#endif
599
600 m_rootId = m_treeCtrl->AddRoot( rootName, 3, -1, rootData);
601 m_treeCtrl->SetItemHasChildren(m_rootId);
602
603 ExpandRoot();
604
605 SetInitialSize(size);
606 DoResize();
607
608 return true;
609}
610
611wxGenericDirCtrl::~wxGenericDirCtrl()
612{
613}
614
615void wxGenericDirCtrl::Init()
616{
617 m_showHidden = false;
618 m_currentFilter = 0;
619 m_currentFilterStr = wxEmptyString; // Default: any file
620 m_treeCtrl = NULL;
621 m_filterListCtrl = NULL;
622}
623
624wxTreeCtrl* wxGenericDirCtrl::CreateTreeCtrl(wxWindow *parent, wxWindowID id, const wxPoint& pos, const wxSize& size, long treeStyle)
625{
626 return new wxTreeCtrl(parent, id, pos, size, treeStyle);
627}
628
629void wxGenericDirCtrl::ShowHidden( bool show )
630{
631 if ( m_showHidden == show )
632 return;
633
634 m_showHidden = show;
635
636 if ( HasFlag(wxDIRCTRL_MULTIPLE) )
637 {
638 wxArrayString paths;
639 GetPaths(paths);
640 ReCreateTree();
641 for ( unsigned n = 0; n < paths.size(); n++ )
642 {
643 ExpandPath(paths[n]);
644 }
645 }
646 else
647 {
648 wxString path = GetPath();
649 ReCreateTree();
650 SetPath(path);
651 }
652}
653
654const wxTreeItemId
655wxGenericDirCtrl::AddSection(const wxString& path, const wxString& name, int imageId)
656{
657 wxDirItemData *dir_item = new wxDirItemData(path,name,true);
658
659 wxTreeItemId id = AppendItem( m_rootId, name, imageId, -1, dir_item);
660
661 m_treeCtrl->SetItemHasChildren(id);
662
663 return id;
664}
665
666void wxGenericDirCtrl::SetupSections()
667{
668 wxArrayString paths, names;
669 wxArrayInt icons;
670
671 size_t n, count = wxGetAvailableDrives(paths, names, icons);
672
673#ifdef __WXGTK20__
674 wxString home = wxGetHomeDir();
675 AddSection( home, _("Home directory"), 1);
676 home += wxT("/Desktop");
677 AddSection( home, _("Desktop"), 1);
678#endif
679
680 for (n = 0; n < count; n++)
681 AddSection(paths[n], names[n], icons[n]);
682}
683
684void wxGenericDirCtrl::SetFocus()
685{
686 // we don't need focus ourselves, give it to the tree so that the user
687 // could navigate it
688 if (m_treeCtrl)
689 m_treeCtrl->SetFocus();
690}
691
692void wxGenericDirCtrl::OnBeginEditItem(wxTreeEvent &event)
693{
694 // don't rename the main entry "Sections"
695 if (event.GetItem() == m_rootId)
696 {
697 event.Veto();
698 return;
699 }
700
701 // don't rename the individual sections
702 if (m_treeCtrl->GetItemParent( event.GetItem() ) == m_rootId)
703 {
704 event.Veto();
705 return;
706 }
707}
708
709void wxGenericDirCtrl::OnEndEditItem(wxTreeEvent &event)
710{
711 if (event.IsEditCancelled())
712 return;
713
714 if ((event.GetLabel().empty()) ||
715 (event.GetLabel() == wxT(".")) ||
716 (event.GetLabel() == wxT("..")) ||
717 (event.GetLabel().Find(wxT('/')) != wxNOT_FOUND) ||
718 (event.GetLabel().Find(wxT('\\')) != wxNOT_FOUND) ||
719 (event.GetLabel().Find(wxT('|')) != wxNOT_FOUND))
720 {
721 wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
722 dialog.ShowModal();
723 event.Veto();
724 return;
725 }
726
727 wxTreeItemId id = event.GetItem();
728 wxDirItemData *data = (wxDirItemData*)m_treeCtrl->GetItemData( id );
729 wxASSERT( data );
730
731 wxString new_name( wxPathOnly( data->m_path ) );
732 new_name += wxString(wxFILE_SEP_PATH);
733 new_name += event.GetLabel();
734
735 wxLogNull log;
736
737 if (wxFileExists(new_name))
738 {
739 wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
740 dialog.ShowModal();
741 event.Veto();
742 }
743
744 if (wxRenameFile(data->m_path,new_name))
745 {
746 data->SetNewDirName( new_name );
747 }
748 else
749 {
750 wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
751 dialog.ShowModal();
752 event.Veto();
753 }
754}
755
756void wxGenericDirCtrl::OnExpandItem(wxTreeEvent &event)
757{
758 wxTreeItemId parentId = event.GetItem();
759
760 // VS: this is needed because the event handler is called from wxTreeCtrl
761 // ctor when wxTR_HIDE_ROOT was specified
762
763 if (!m_rootId.IsOk())
764
765 m_rootId = m_treeCtrl->GetRootItem();
766
767 ExpandDir(parentId);
768 if ( m_treeCtrl->GetChildrenCount(parentId, false) == 0 )
769 {
770 m_treeCtrl->SetItemHasChildren(parentId, false);
771 }
772}
773
774void wxGenericDirCtrl::OnCollapseItem(wxTreeEvent &event )
775{
776 CollapseDir(event.GetItem());
777}
778
779void wxGenericDirCtrl::CollapseDir(wxTreeItemId parentId)
780{
781 wxTreeItemId child;
782
783 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(parentId);
784 if (!data->m_isExpanded)
785 return;
786
787 data->m_isExpanded = false;
788
789 m_treeCtrl->Freeze();
790 if (parentId != m_treeCtrl->GetRootItem())
791 m_treeCtrl->CollapseAndReset(parentId);
792 m_treeCtrl->DeleteChildren(parentId);
793 m_treeCtrl->Thaw();
794}
795
796void wxGenericDirCtrl::ExpandDir(wxTreeItemId parentId)
797{
798 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(parentId);
799
800 if (data->m_isExpanded)
801 return;
802
803 data->m_isExpanded = true;
804
805 if (parentId == m_treeCtrl->GetRootItem())
806 {
807 SetupSections();
808 return;
809 }
810
811 wxASSERT(data);
812
813 wxString search,path,filename;
814
815 wxString dirName(data->m_path);
816
817#if (defined(__WINDOWS__) && !defined(__WXWINCE__)) || defined(__DOS__) || defined(__OS2__)
818 // Check if this is a root directory and if so,
819 // whether the drive is avaiable.
820 if (!wxIsDriveAvailable(dirName))
821 {
822 data->m_isExpanded = false;
823 //wxMessageBox(wxT("Sorry, this drive is not available."));
824 return;
825 }
826#endif
827
828 // This may take a longish time. Go to busy cursor
829 wxBusyCursor busy;
830
831#if defined(__WINDOWS__) || defined(__DOS__) || defined(__OS2__)
832 if (dirName.Last() == ':')
833 dirName += wxString(wxFILE_SEP_PATH);
834#endif
835
836 wxArrayString dirs;
837 wxArrayString filenames;
838
839 wxDir d;
840 wxString eachFilename;
841
842 wxLogNull log;
843 d.Open(dirName);
844
845 if (d.IsOpened())
846 {
847 int style = wxDIR_DIRS;
848 if (m_showHidden) style |= wxDIR_HIDDEN;
849 if (d.GetFirst(& eachFilename, wxEmptyString, style))
850 {
851 do
852 {
853 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
854 {
855 dirs.Add(eachFilename);
856 }
857 }
858 while (d.GetNext(&eachFilename));
859 }
860 }
861 dirs.Sort(wxDirCtrlStringCompareFunction);
862
863 // Now do the filenames -- but only if we're allowed to
864 if (!HasFlag(wxDIRCTRL_DIR_ONLY))
865 {
866 d.Open(dirName);
867
868 if (d.IsOpened())
869 {
870 int style = wxDIR_FILES;
871 if (m_showHidden) style |= wxDIR_HIDDEN;
872 // Process each filter (ex: "JPEG Files (*.jpg;*.jpeg)|*.jpg;*.jpeg")
873 wxStringTokenizer strTok;
874 wxString curFilter;
875 strTok.SetString(m_currentFilterStr,wxT(";"));
876 while(strTok.HasMoreTokens())
877 {
878 curFilter = strTok.GetNextToken();
879 if (d.GetFirst(& eachFilename, curFilter, style))
880 {
881 do
882 {
883 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
884 {
885 filenames.Add(eachFilename);
886 }
887 }
888 while (d.GetNext(& eachFilename));
889 }
890 }
891 }
892 filenames.Sort(wxDirCtrlStringCompareFunction);
893 }
894
895 // Add the sorted dirs
896 size_t i;
897 for (i = 0; i < dirs.GetCount(); i++)
898 {
899 eachFilename = dirs[i];
900 path = dirName;
901 if (!wxEndsWithPathSeparator(path))
902 path += wxString(wxFILE_SEP_PATH);
903 path += eachFilename;
904
905 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,true);
906 wxTreeItemId id = AppendItem( parentId, eachFilename,
907 wxFileIconsTable::folder, -1, dir_item);
908 m_treeCtrl->SetItemImage( id, wxFileIconsTable::folder_open,
909 wxTreeItemIcon_Expanded );
910
911 // assume that it does have children by default as it can take a long
912 // time to really check for this (think remote drives...)
913 //
914 // and if we're wrong, we'll correct it later in OnExpandItem() if
915 // the user really tries to open this item
916 m_treeCtrl->SetItemHasChildren(id);
917 }
918
919 // Add the sorted filenames
920 if (!HasFlag(wxDIRCTRL_DIR_ONLY))
921 {
922 for (i = 0; i < filenames.GetCount(); i++)
923 {
924 eachFilename = filenames[i];
925 path = dirName;
926 if (!wxEndsWithPathSeparator(path))
927 path += wxString(wxFILE_SEP_PATH);
928 path += eachFilename;
929 //path = dirName + wxString(wxT("/")) + eachFilename;
930 wxDirItemData *dir_item = new wxDirItemData(path,eachFilename,false);
931 int image_id = wxFileIconsTable::file;
932 if (eachFilename.Find(wxT('.')) != wxNOT_FOUND)
933 image_id = wxTheFileIconsTable->GetIconID(eachFilename.AfterLast(wxT('.')));
934 (void) AppendItem( parentId, eachFilename, image_id, -1, dir_item);
935 }
936 }
937}
938
939void wxGenericDirCtrl::ReCreateTree()
940{
941 CollapseDir(m_treeCtrl->GetRootItem());
942 ExpandRoot();
943}
944
945void wxGenericDirCtrl::CollapseTree()
946{
947 wxTreeItemIdValue cookie;
948 wxTreeItemId child = m_treeCtrl->GetFirstChild(m_rootId, cookie);
949 while (child.IsOk())
950 {
951 CollapseDir(child);
952 child = m_treeCtrl->GetNextChild(m_rootId, cookie);
953 }
954}
955
956// Find the child that matches the first part of 'path'.
957// E.g. if a child path is "/usr" and 'path' is "/usr/include"
958// then the child for /usr is returned.
959wxTreeItemId wxGenericDirCtrl::FindChild(wxTreeItemId parentId, const wxString& path, bool& done)
960{
961 wxString path2(path);
962
963 // Make sure all separators are as per the current platform
964 path2.Replace(wxT("\\"), wxString(wxFILE_SEP_PATH));
965 path2.Replace(wxT("/"), wxString(wxFILE_SEP_PATH));
966
967 // Append a separator to foil bogus substring matching
968 path2 += wxString(wxFILE_SEP_PATH);
969
970 // In MSW or PM, case is not significant
971#if defined(__WINDOWS__) || defined(__DOS__) || defined(__OS2__)
972 path2.MakeLower();
973#endif
974
975 wxTreeItemIdValue cookie;
976 wxTreeItemId childId = m_treeCtrl->GetFirstChild(parentId, cookie);
977 while (childId.IsOk())
978 {
979 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
980
981 if (data && !data->m_path.empty())
982 {
983 wxString childPath(data->m_path);
984 if (!wxEndsWithPathSeparator(childPath))
985 childPath += wxString(wxFILE_SEP_PATH);
986
987 // In MSW and PM, case is not significant
988#if defined(__WINDOWS__) || defined(__DOS__) || defined(__OS2__)
989 childPath.MakeLower();
990#endif
991
992 if (childPath.length() <= path2.length())
993 {
994 wxString path3 = path2.Mid(0, childPath.length());
995 if (childPath == path3)
996 {
997 if (path3.length() == path2.length())
998 done = true;
999 else
1000 done = false;
1001 return childId;
1002 }
1003 }
1004 }
1005
1006 childId = m_treeCtrl->GetNextChild(parentId, cookie);
1007 }
1008 wxTreeItemId invalid;
1009 return invalid;
1010}
1011
1012// Try to expand as much of the given path as possible,
1013// and select the given tree item.
1014bool wxGenericDirCtrl::ExpandPath(const wxString& path)
1015{
1016 bool done = false;
1017 wxTreeItemId id = FindChild(m_rootId, path, done);
1018 wxTreeItemId lastId = id; // The last non-zero id
1019 while (id.IsOk() && !done)
1020 {
1021 ExpandDir(id);
1022
1023 id = FindChild(id, path, done);
1024 if (id.IsOk())
1025 lastId = id;
1026 }
1027 if (!lastId.IsOk())
1028 return false;
1029
1030 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(lastId);
1031 if (data->m_isDir)
1032 {
1033 m_treeCtrl->Expand(lastId);
1034 }
1035 if (HasFlag(wxDIRCTRL_SELECT_FIRST) && data->m_isDir)
1036 {
1037 // Find the first file in this directory
1038 wxTreeItemIdValue cookie;
1039 wxTreeItemId childId = m_treeCtrl->GetFirstChild(lastId, cookie);
1040 bool selectedChild = false;
1041 while (childId.IsOk())
1042 {
1043 data = (wxDirItemData*) m_treeCtrl->GetItemData(childId);
1044
1045 if (data && data->m_path != wxEmptyString && !data->m_isDir)
1046 {
1047 m_treeCtrl->SelectItem(childId);
1048 m_treeCtrl->EnsureVisible(childId);
1049 selectedChild = true;
1050 break;
1051 }
1052 childId = m_treeCtrl->GetNextChild(lastId, cookie);
1053 }
1054 if (!selectedChild)
1055 {
1056 m_treeCtrl->SelectItem(lastId);
1057 m_treeCtrl->EnsureVisible(lastId);
1058 }
1059 }
1060 else
1061 {
1062 m_treeCtrl->SelectItem(lastId);
1063 m_treeCtrl->EnsureVisible(lastId);
1064 }
1065
1066 return true;
1067}
1068
1069
1070bool wxGenericDirCtrl::CollapsePath(const wxString& path)
1071{
1072 bool done = false;
1073 wxTreeItemId id = FindChild(m_rootId, path, done);
1074 wxTreeItemId lastId = id; // The last non-zero id
1075
1076 while ( id.IsOk() && !done )
1077 {
1078 CollapseDir(id);
1079
1080 id = FindChild(id, path, done);
1081
1082 if ( id.IsOk() )
1083 lastId = id;
1084 }
1085
1086 if ( !lastId.IsOk() )
1087 return false;
1088
1089 m_treeCtrl->SelectItem(lastId);
1090 m_treeCtrl->EnsureVisible(lastId);
1091
1092 return true;
1093}
1094
1095
1096wxString wxGenericDirCtrl::GetPath() const
1097{
1098 wxTreeItemId id = m_treeCtrl->GetSelection();
1099 if (id)
1100 {
1101 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1102 return data->m_path;
1103 }
1104 else
1105 return wxEmptyString;
1106}
1107
1108void wxGenericDirCtrl::GetPaths(wxArrayString& paths) const
1109{
1110 paths.clear();
1111
1112 wxArrayTreeItemIds items;
1113 m_treeCtrl->GetSelections(items);
1114 for ( unsigned n = 0; n < items.size(); n++ )
1115 {
1116 wxTreeItemId id = items[n];
1117 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1118 paths.Add(data->m_path);
1119 }
1120}
1121
1122wxString wxGenericDirCtrl::GetFilePath() const
1123{
1124 wxTreeItemId id = m_treeCtrl->GetSelection();
1125 if (id)
1126 {
1127 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1128 if (data->m_isDir)
1129 return wxEmptyString;
1130 else
1131 return data->m_path;
1132 }
1133 else
1134 return wxEmptyString;
1135}
1136
1137void wxGenericDirCtrl::GetFilePaths(wxArrayString& paths) const
1138{
1139 paths.clear();
1140
1141 wxArrayTreeItemIds items;
1142 m_treeCtrl->GetSelections(items);
1143 for ( unsigned n = 0; n < items.size(); n++ )
1144 {
1145 wxTreeItemId id = items[n];
1146 wxDirItemData* data = (wxDirItemData*) m_treeCtrl->GetItemData(id);
1147 if ( !data->m_isDir )
1148 paths.Add(data->m_path);
1149 }
1150}
1151
1152void wxGenericDirCtrl::SetPath(const wxString& path)
1153{
1154 m_defaultPath = path;
1155 if (m_rootId)
1156 ExpandPath(path);
1157}
1158
1159void wxGenericDirCtrl::SelectPath(const wxString& path, bool select)
1160{
1161 bool done = false;
1162 wxTreeItemId id = FindChild(m_rootId, path, done);
1163 wxTreeItemId lastId = id; // The last non-zero id
1164 while ( id.IsOk() && !done )
1165 {
1166 id = FindChild(id, path, done);
1167 if ( id.IsOk() )
1168 lastId = id;
1169 }
1170 if ( !lastId.IsOk() )
1171 return;
1172
1173 if ( done )
1174 {
1175 m_treeCtrl->SelectItem(id, select);
1176 }
1177}
1178
1179void wxGenericDirCtrl::SelectPaths(const wxArrayString& paths)
1180{
1181 if ( HasFlag(wxDIRCTRL_MULTIPLE) )
1182 {
1183 UnselectAll();
1184 for ( unsigned n = 0; n < paths.size(); n++ )
1185 {
1186 SelectPath(paths[n]);
1187 }
1188 }
1189}
1190
1191void wxGenericDirCtrl::UnselectAll()
1192{
1193 m_treeCtrl->UnselectAll();
1194}
1195
1196// Not used
1197#if 0
1198void wxGenericDirCtrl::FindChildFiles(wxTreeItemId id, int dirFlags, wxArrayString& filenames)
1199{
1200 wxDirItemData *data = (wxDirItemData *) m_treeCtrl->GetItemData(id);
1201
1202 // This may take a longish time. Go to busy cursor
1203 wxBusyCursor busy;
1204
1205 wxASSERT(data);
1206
1207 wxString search,path,filename;
1208
1209 wxString dirName(data->m_path);
1210
1211#if defined(__WXMSW__) || defined(__OS2__)
1212 if (dirName.Last() == ':')
1213 dirName += wxString(wxFILE_SEP_PATH);
1214#endif
1215
1216 wxDir d;
1217 wxString eachFilename;
1218
1219 wxLogNull log;
1220 d.Open(dirName);
1221
1222 if (d.IsOpened())
1223 {
1224 if (d.GetFirst(& eachFilename, m_currentFilterStr, dirFlags))
1225 {
1226 do
1227 {
1228 if ((eachFilename != wxT(".")) && (eachFilename != wxT("..")))
1229 {
1230 filenames.Add(eachFilename);
1231 }
1232 }
1233 while (d.GetNext(& eachFilename)) ;
1234 }
1235 }
1236}
1237#endif
1238
1239void wxGenericDirCtrl::SetFilterIndex(int n)
1240{
1241 m_currentFilter = n;
1242
1243 wxString f, d;
1244 if (ExtractWildcard(m_filter, n, f, d))
1245 m_currentFilterStr = f;
1246 else
1247#ifdef __UNIX__
1248 m_currentFilterStr = wxT("*");
1249#else
1250 m_currentFilterStr = wxT("*.*");
1251#endif
1252}
1253
1254void wxGenericDirCtrl::SetFilter(const wxString& filter)
1255{
1256 m_filter = filter;
1257
1258 if (!filter.empty() && !m_filterListCtrl)
1259 m_filterListCtrl = new wxDirFilterListCtrl(this, wxID_FILTERLISTCTRL);
1260 else if (filter.empty() && m_filterListCtrl)
1261 {
1262 m_filterListCtrl->Destroy();
1263 m_filterListCtrl = NULL;
1264 }
1265
1266 wxString f, d;
1267 if (ExtractWildcard(m_filter, m_currentFilter, f, d))
1268 m_currentFilterStr = f;
1269 else
1270#ifdef __UNIX__
1271 m_currentFilterStr = wxT("*");
1272#else
1273 m_currentFilterStr = wxT("*.*");
1274#endif
1275 // current filter index is meaningless after filter change, set it to zero
1276 SetFilterIndex(0);
1277 if (m_filterListCtrl)
1278 m_filterListCtrl->FillFilterList(m_filter, 0);
1279}
1280
1281// Extract description and actual filter from overall filter string
1282bool wxGenericDirCtrl::ExtractWildcard(const wxString& filterStr, int n, wxString& filter, wxString& description)
1283{
1284 wxArrayString filters, descriptions;
1285 int count = wxParseCommonDialogsFilter(filterStr, descriptions, filters);
1286 if (count > 0 && n < count)
1287 {
1288 filter = filters[n];
1289 description = descriptions[n];
1290 return true;
1291 }
1292
1293 return false;
1294}
1295
1296
1297void wxGenericDirCtrl::DoResize()
1298{
1299 wxSize sz = GetClientSize();
1300 int verticalSpacing = 3;
1301 if (m_treeCtrl)
1302 {
1303 wxSize filterSz ;
1304 if (m_filterListCtrl)
1305 {
1306 filterSz = m_filterListCtrl->GetSize();
1307 sz.y -= (filterSz.y + verticalSpacing);
1308 }
1309 m_treeCtrl->SetSize(0, 0, sz.x, sz.y);
1310 if (m_filterListCtrl)
1311 {
1312 m_filterListCtrl->SetSize(0, sz.y + verticalSpacing, sz.x, filterSz.y);
1313 // Don't know why, but this needs refreshing after a resize (wxMSW)
1314 m_filterListCtrl->Refresh();
1315 }
1316 }
1317}
1318
1319
1320void wxGenericDirCtrl::OnSize(wxSizeEvent& WXUNUSED(event))
1321{
1322 DoResize();
1323}
1324
1325wxTreeItemId wxGenericDirCtrl::AppendItem (const wxTreeItemId & parent,
1326 const wxString & text,
1327 int image, int selectedImage,
1328 wxTreeItemData * data)
1329{
1330 wxTreeCtrl *treeCtrl = GetTreeCtrl ();
1331
1332 wxASSERT (treeCtrl);
1333
1334 if (treeCtrl)
1335 {
1336 return treeCtrl->AppendItem (parent, text, image, selectedImage, data);
1337 }
1338 else
1339 {
1340 return wxTreeItemId();
1341 }
1342}
1343
1344
1345//-----------------------------------------------------------------------------
1346// wxDirFilterListCtrl
1347//-----------------------------------------------------------------------------
1348
1349IMPLEMENT_CLASS(wxDirFilterListCtrl, wxChoice)
1350
1351BEGIN_EVENT_TABLE(wxDirFilterListCtrl, wxChoice)
1352 EVT_CHOICE(wxID_ANY, wxDirFilterListCtrl::OnSelFilter)
1353END_EVENT_TABLE()
1354
1355bool wxDirFilterListCtrl::Create(wxGenericDirCtrl* parent,
1356 const wxWindowID id,
1357 const wxPoint& pos,
1358 const wxSize& size,
1359 long style)
1360{
1361 m_dirCtrl = parent;
1362
1363 // by default our border style is determined by the style of our parent
1364 if ( !(style & wxBORDER_MASK) )
1365 {
1366 style |= parent->HasFlag(wxDIRCTRL_3D_INTERNAL) ? wxBORDER_SUNKEN
1367 : wxBORDER_NONE;
1368 }
1369
1370 return wxChoice::Create(parent, id, pos, size, 0, NULL, style);
1371}
1372
1373void wxDirFilterListCtrl::Init()
1374{
1375 m_dirCtrl = NULL;
1376}
1377
1378void wxDirFilterListCtrl::OnSelFilter(wxCommandEvent& WXUNUSED(event))
1379{
1380 int sel = GetSelection();
1381
1382 wxString currentPath = m_dirCtrl->GetPath();
1383
1384 m_dirCtrl->SetFilterIndex(sel);
1385
1386 // If the filter has changed, the view is out of date, so
1387 // collapse the tree.
1388 m_dirCtrl->ReCreateTree();
1389
1390 // Try to restore the selection, or at least the directory
1391 m_dirCtrl->ExpandPath(currentPath);
1392}
1393
1394void wxDirFilterListCtrl::FillFilterList(const wxString& filter, int defaultFilter)
1395{
1396 Clear();
1397 wxArrayString descriptions, filters;
1398 size_t n = (size_t) wxParseCommonDialogsFilter(filter, descriptions, filters);
1399
1400 if (n > 0 && defaultFilter < (int) n)
1401 {
1402 for (size_t i = 0; i < n; i++)
1403 Append(descriptions[i]);
1404 SetSelection(defaultFilter);
1405 }
1406}
1407#endif // wxUSE_DIRDLG
1408
1409#if wxUSE_DIRDLG || wxUSE_FILEDLG
1410
1411// ----------------------------------------------------------------------------
1412// wxFileIconsTable icons
1413// ----------------------------------------------------------------------------
1414
1415#ifndef __WXGTK20__
1416/* Computer (c) Julian Smart */
1417static const char * file_icons_tbl_computer_xpm[] = {
1418/* columns rows colors chars-per-pixel */
1419"16 16 42 1",
1420"r c #4E7FD0",
1421"$ c #7198D9",
1422"; c #DCE6F6",
1423"q c #FFFFFF",
1424"u c #4A7CCE",
1425"# c #779DDB",
1426"w c #95B2E3",
1427"y c #7FA2DD",
1428"f c #3263B4",
1429"= c #EAF0FA",
1430"< c #B1C7EB",
1431"% c #6992D7",
1432"9 c #D9E4F5",
1433"o c #9BB7E5",
1434"6 c #F7F9FD",
1435", c #BED0EE",
1436"3 c #F0F5FC",
1437"1 c #A8C0E8",
1438" c None",
1439"0 c #FDFEFF",
1440"4 c #C4D5F0",
1441"@ c #81A4DD",
1442"e c #4377CD",
1443"- c #E2EAF8",
1444"i c #9FB9E5",
1445"> c #CCDAF2",
1446"+ c #89A9DF",
1447"s c #5584D1",
1448"t c #5D89D3",
1449": c #D2DFF4",
1450"5 c #FAFCFE",
1451"2 c #F5F8FD",
1452"8 c #DFE8F7",
1453"& c #5E8AD4",
1454"X c #638ED5",
1455"a c #CEDCF2",
1456"p c #90AFE2",
1457"d c #2F5DA9",
1458"* c #5282D0",
1459"7 c #E5EDF9",
1460". c #A2BCE6",
1461"O c #8CACE0",
1462/* pixels */
1463" ",
1464" .XXXXXXXXXXX ",
1465" oXO++@#$%&*X ",
1466" oX=-;:>,<1%X ",
1467" oX23=-;:4,$X ",
1468" oX5633789:@X ",
1469" oX05623=78+X ",
1470" oXqq05623=OX ",
1471" oX,,,,,<<<$X ",
1472" wXXXXXXXXXXe ",
1473" XrtX%$$y@+O,, ",
1474" uyiiiiiiiii@< ",
1475" ouiiiiiiiiiip<a",
1476" rustX%$$y@+Ow,,",
1477" dfffffffffffffd",
1478" "
1479};
1480#endif // !GTK+ 2
1481
1482// ----------------------------------------------------------------------------
1483// wxFileIconsTable & friends
1484// ----------------------------------------------------------------------------
1485
1486// global instance of a wxFileIconsTable
1487wxFileIconsTable* wxTheFileIconsTable = NULL;
1488
1489// A module to allow icons table cleanup
1490
1491class wxFileIconsTableModule: public wxModule
1492{
1493DECLARE_DYNAMIC_CLASS(wxFileIconsTableModule)
1494public:
1495 wxFileIconsTableModule() {}
1496 bool OnInit() { wxTheFileIconsTable = new wxFileIconsTable; return true; }
1497 void OnExit()
1498 {
1499 if (wxTheFileIconsTable)
1500 {
1501 delete wxTheFileIconsTable;
1502 wxTheFileIconsTable = NULL;
1503 }
1504 }
1505};
1506
1507IMPLEMENT_DYNAMIC_CLASS(wxFileIconsTableModule, wxModule)
1508
1509class wxFileIconEntry : public wxObject
1510{
1511public:
1512 wxFileIconEntry(int i) { id = i; }
1513
1514 int id;
1515};
1516
1517wxFileIconsTable::wxFileIconsTable()
1518{
1519 m_HashTable = NULL;
1520 m_smallImageList = NULL;
1521}
1522
1523wxFileIconsTable::~wxFileIconsTable()
1524{
1525 if (m_HashTable)
1526 {
1527 WX_CLEAR_HASH_TABLE(*m_HashTable);
1528 delete m_HashTable;
1529 }
1530 if (m_smallImageList) delete m_smallImageList;
1531}
1532
1533// delayed initialization - wait until first use (wxArtProv not created yet)
1534void wxFileIconsTable::Create()
1535{
1536 wxCHECK_RET(!m_smallImageList && !m_HashTable, wxT("creating icons twice"));
1537 m_HashTable = new wxHashTable(wxKEY_STRING);
1538 m_smallImageList = new wxImageList(16, 16);
1539
1540 // folder:
1541 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_FOLDER,
1542 wxART_CMN_DIALOG,
1543 wxSize(16, 16)));
1544 // folder_open
1545 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_FOLDER_OPEN,
1546 wxART_CMN_DIALOG,
1547 wxSize(16, 16)));
1548 // computer
1549#ifdef __WXGTK20__
1550 // GTK24 uses this icon in the file open dialog
1551 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_HARDDISK,
1552 wxART_CMN_DIALOG,
1553 wxSize(16, 16)));
1554#else
1555 m_smallImageList->Add(wxIcon(file_icons_tbl_computer_xpm));
1556#endif
1557 // drive
1558 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_HARDDISK,
1559 wxART_CMN_DIALOG,
1560 wxSize(16, 16)));
1561 // cdrom
1562 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_CDROM,
1563 wxART_CMN_DIALOG,
1564 wxSize(16, 16)));
1565 // floppy
1566 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_FLOPPY,
1567 wxART_CMN_DIALOG,
1568 wxSize(16, 16)));
1569 // removeable
1570 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_REMOVABLE,
1571 wxART_CMN_DIALOG,
1572 wxSize(16, 16)));
1573 // file
1574 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_NORMAL_FILE,
1575 wxART_CMN_DIALOG,
1576 wxSize(16, 16)));
1577 // executable
1578 if (GetIconID(wxEmptyString, wxT("application/x-executable")) == file)
1579 {
1580 m_smallImageList->Add(wxArtProvider::GetBitmap(wxART_EXECUTABLE_FILE,
1581 wxART_CMN_DIALOG,
1582 wxSize(16, 16)));
1583 delete m_HashTable->Get(wxT("exe"));
1584 m_HashTable->Delete(wxT("exe"));
1585 m_HashTable->Put(wxT("exe"), new wxFileIconEntry(executable));
1586 }
1587 /* else put into list by GetIconID
1588 (KDE defines application/x-executable for *.exe and has nice icon)
1589 */
1590}
1591
1592wxImageList *wxFileIconsTable::GetSmallImageList()
1593{
1594 if (!m_smallImageList)
1595 Create();
1596
1597 return m_smallImageList;
1598}
1599
1600#if wxUSE_MIMETYPE && wxUSE_IMAGE && (!defined(__WXMSW__) || wxUSE_WXDIB)
1601// VS: we don't need this function w/o wxMimeTypesManager because we'll only have
1602// one icon and we won't resize it
1603
1604static wxBitmap CreateAntialiasedBitmap(const wxImage& img)
1605{
1606 const unsigned int size = 16;
1607
1608 wxImage smallimg (size, size);
1609 unsigned char *p1, *p2, *ps;
1610 unsigned char mr = img.GetMaskRed(),
1611 mg = img.GetMaskGreen(),
1612 mb = img.GetMaskBlue();
1613
1614 unsigned x, y;
1615 unsigned sr, sg, sb, smask;
1616
1617 p1 = img.GetData(), p2 = img.GetData() + 3 * size*2, ps = smallimg.GetData();
1618 smallimg.SetMaskColour(mr, mr, mr);
1619
1620 for (y = 0; y < size; y++)
1621 {
1622 for (x = 0; x < size; x++)
1623 {
1624 sr = sg = sb = smask = 0;
1625 if (p1[0] != mr || p1[1] != mg || p1[2] != mb)
1626 sr += p1[0], sg += p1[1], sb += p1[2];
1627 else smask++;
1628 p1 += 3;
1629 if (p1[0] != mr || p1[1] != mg || p1[2] != mb)
1630 sr += p1[0], sg += p1[1], sb += p1[2];
1631 else smask++;
1632 p1 += 3;
1633 if (p2[0] != mr || p2[1] != mg || p2[2] != mb)
1634 sr += p2[0], sg += p2[1], sb += p2[2];
1635 else smask++;
1636 p2 += 3;
1637 if (p2[0] != mr || p2[1] != mg || p2[2] != mb)
1638 sr += p2[0], sg += p2[1], sb += p2[2];
1639 else smask++;
1640 p2 += 3;
1641
1642 if (smask > 2)
1643 ps[0] = ps[1] = ps[2] = mr;
1644 else
1645 {
1646 ps[0] = (unsigned char)(sr >> 2);
1647 ps[1] = (unsigned char)(sg >> 2);
1648 ps[2] = (unsigned char)(sb >> 2);
1649 }
1650 ps += 3;
1651 }
1652 p1 += size*2 * 3, p2 += size*2 * 3;
1653 }
1654
1655 return wxBitmap(smallimg);
1656}
1657
1658// This function is currently not unused anymore
1659#if 0
1660// finds empty borders and return non-empty area of image:
1661static wxImage CutEmptyBorders(const wxImage& img)
1662{
1663 unsigned char mr = img.GetMaskRed(),
1664 mg = img.GetMaskGreen(),
1665 mb = img.GetMaskBlue();
1666 unsigned char *dt = img.GetData(), *dttmp;
1667 unsigned w = img.GetWidth(), h = img.GetHeight();
1668
1669 unsigned top, bottom, left, right, i;
1670 bool empt;
1671
1672#define MK_DTTMP(x,y) dttmp = dt + ((x + y * w) * 3)
1673#define NOEMPTY_PIX(empt) if (dttmp[0] != mr || dttmp[1] != mg || dttmp[2] != mb) {empt = false; break;}
1674
1675 for (empt = true, top = 0; empt && top < h; top++)
1676 {
1677 MK_DTTMP(0, top);
1678 for (i = 0; i < w; i++, dttmp+=3)
1679 NOEMPTY_PIX(empt)
1680 }
1681 for (empt = true, bottom = h-1; empt && bottom > top; bottom--)
1682 {
1683 MK_DTTMP(0, bottom);
1684 for (i = 0; i < w; i++, dttmp+=3)
1685 NOEMPTY_PIX(empt)
1686 }
1687 for (empt = true, left = 0; empt && left < w; left++)
1688 {
1689 MK_DTTMP(left, 0);
1690 for (i = 0; i < h; i++, dttmp+=3*w)
1691 NOEMPTY_PIX(empt)
1692 }
1693 for (empt = true, right = w-1; empt && right > left; right--)
1694 {
1695 MK_DTTMP(right, 0);
1696 for (i = 0; i < h; i++, dttmp+=3*w)
1697 NOEMPTY_PIX(empt)
1698 }
1699 top--, left--, bottom++, right++;
1700
1701 return img.GetSubImage(wxRect(left, top, right - left + 1, bottom - top + 1));
1702}
1703#endif // #if 0
1704
1705#endif // wxUSE_MIMETYPE
1706
1707int wxFileIconsTable::GetIconID(const wxString& extension, const wxString& mime)
1708{
1709 if (!m_smallImageList)
1710 Create();
1711
1712#if wxUSE_MIMETYPE
1713 if (!extension.empty())
1714 {
1715 wxFileIconEntry *entry = (wxFileIconEntry*) m_HashTable->Get(extension);
1716 if (entry) return (entry -> id);
1717 }
1718
1719 wxFileType *ft = (mime.empty()) ?
1720 wxTheMimeTypesManager -> GetFileTypeFromExtension(extension) :
1721 wxTheMimeTypesManager -> GetFileTypeFromMimeType(mime);
1722
1723 wxIconLocation iconLoc;
1724 wxIcon ic;
1725
1726 {
1727 wxLogNull logNull;
1728 if ( ft && ft->GetIcon(&iconLoc) )
1729 {
1730 ic = wxIcon( iconLoc );
1731 }
1732 }
1733
1734 delete ft;
1735
1736 if ( !ic.Ok() )
1737 {
1738 int newid = file;
1739 m_HashTable->Put(extension, new wxFileIconEntry(newid));
1740 return newid;
1741 }
1742
1743 wxBitmap bmp;
1744 bmp.CopyFromIcon(ic);
1745
1746 if ( !bmp.Ok() )
1747 {
1748 int newid = file;
1749 m_HashTable->Put(extension, new wxFileIconEntry(newid));
1750 return newid;
1751 }
1752
1753 const unsigned int size = 16;
1754
1755 int id = m_smallImageList->GetImageCount();
1756 if ((bmp.GetWidth() == (int) size) && (bmp.GetHeight() == (int) size))
1757 {
1758 m_smallImageList->Add(bmp);
1759 }
1760#if wxUSE_IMAGE && (!defined(__WXMSW__) || wxUSE_WXDIB)
1761 else
1762 {
1763 wxImage img = bmp.ConvertToImage();
1764
1765 if ((img.GetWidth() != size*2) || (img.GetHeight() != size*2))
1766// m_smallImageList->Add(CreateAntialiasedBitmap(CutEmptyBorders(img).Rescale(size*2, size*2)));
1767 m_smallImageList->Add(CreateAntialiasedBitmap(img.Rescale(size*2, size*2)));
1768 else
1769 m_smallImageList->Add(CreateAntialiasedBitmap(img));
1770 }
1771#endif // wxUSE_IMAGE
1772
1773 m_HashTable->Put(extension, new wxFileIconEntry(id));
1774 return id;
1775
1776#else // !wxUSE_MIMETYPE
1777
1778 wxUnusedVar(mime);
1779 if (extension == wxT("exe"))
1780 return executable;
1781 else
1782 return file;
1783#endif // wxUSE_MIMETYPE/!wxUSE_MIMETYPE
1784}
1785
1786#endif // wxUSE_DIRDLG || wxUSE_FILEDLG