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