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