]> git.saurik.com Git - wxWidgets.git/blob - src/msw/dir.cpp
wxFileDialog cleanup, extracted common code to fldlgcmn.cpp (patch 754187)
[wxWidgets.git] / src / msw / dir.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: msw/dir.cpp
3 // Purpose: wxDir implementation for Win32
4 // Author: Vadim Zeitlin
5 // Modified by:
6 // Created: 08.12.99
7 // RCS-ID: $Id$
8 // Copyright: (c) 1999 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 // ============================================================================
13 // declarations
14 // ============================================================================
15
16 // ----------------------------------------------------------------------------
17 // headers
18 // ----------------------------------------------------------------------------
19
20 #ifdef __GNUG__
21 #pragma implementation "dir.h"
22 #endif
23
24 // For compilers that support precompilation, includes "wx.h".
25 #include "wx/wxprec.h"
26
27 #ifdef __BORLANDC__
28 #pragma hdrstop
29 #endif
30
31 // For _A_SUBDIR, etc.
32 #if defined(__BORLANDC__) && defined(__WIN16__)
33 #include <dos.h>
34 #endif
35
36 #ifndef WX_PRECOMP
37 #include "wx/intl.h"
38 #include "wx/log.h"
39 #endif // PCH
40
41 #include "wx/dir.h"
42 #include "wx/filefn.h" // for wxPathExists()
43
44 // ----------------------------------------------------------------------------
45 // define the types and functions used for file searching
46 // ----------------------------------------------------------------------------
47
48 // under Win16 use compiler-specific functions
49 #ifdef __WIN16__
50 #ifdef __VISUALC__
51 #include <dos.h>
52 #include <errno.h>
53
54 typedef struct _find_t FIND_STRUCT;
55 #elif defined(__BORLANDC__)
56 #include <dir.h>
57
58 typedef struct ffblk FIND_STRUCT;
59 #else
60 #error "No directory searching functions for this compiler"
61 #endif
62
63 typedef FIND_STRUCT *FIND_DATA;
64 typedef char FIND_ATTR;
65
66 static inline FIND_DATA InitFindData() { return (FIND_DATA)NULL; }
67 static inline bool IsFindDataOk(FIND_DATA fd) { return fd != NULL; }
68 static inline void FreeFindData(FIND_DATA fd) { free(fd); }
69
70 static inline FIND_DATA FindFirst(const wxString& spec,
71 FIND_STRUCT * WXUNUSED(finddata))
72 {
73 // attribute to find all files
74 static const FIND_ATTR attr = 0x3F;
75
76 FIND_DATA fd = (FIND_DATA)malloc(sizeof(FIND_STRUCT));
77
78 if (
79 #ifdef __VISUALC__
80 _dos_findfirst(spec, attr, fd) == 0
81 #else // Borland
82 findfirst(spec, fd, attr) == 0
83 #endif
84 )
85 {
86 return fd;
87 }
88 else
89 {
90 free(fd);
91
92 return NULL;
93 }
94 }
95
96 static inline bool FindNext(FIND_DATA fd, FIND_STRUCT * WXUNUSED(finddata))
97 {
98 #ifdef __VISUALC__
99 return _dos_findnext(fd) == 0;
100 #else // Borland
101 return findnext(fd) == 0;
102 #endif
103 }
104
105 static const wxChar *GetNameFromFindData(FIND_STRUCT *finddata)
106 {
107 #ifdef __VISUALC__
108 return finddata->name;
109 #else // Borland
110 return finddata->ff_name;
111 #endif
112 }
113
114 static const FIND_ATTR GetAttrFromFindData(FIND_STRUCT *finddata)
115 {
116 #ifdef __VISUALC__
117 return finddata->attrib;
118 #else // Borland
119 return finddata->ff_attrib;
120 #endif
121 }
122
123 static inline bool IsDir(FIND_ATTR attr)
124 {
125 return (attr & _A_SUBDIR) != 0;
126 }
127
128 static inline bool IsHidden(FIND_ATTR attr)
129 {
130 return (attr & (_A_SYSTEM | _A_HIDDEN)) != 0;
131 }
132 #else // Win32
133 #include <windows.h>
134
135 typedef WIN32_FIND_DATA FIND_STRUCT;
136 typedef HANDLE FIND_DATA;
137 typedef DWORD FIND_ATTR;
138
139 static inline FIND_DATA InitFindData() { return INVALID_HANDLE_VALUE; }
140
141 static inline bool IsFindDataOk(FIND_DATA fd)
142 {
143 return fd != INVALID_HANDLE_VALUE;
144 }
145
146 static inline void FreeFindData(FIND_DATA fd)
147 {
148 if ( !::FindClose(fd) )
149 {
150 wxLogLastError(_T("FindClose"));
151 }
152 }
153
154 static inline FIND_DATA FindFirst(const wxString& spec,
155 FIND_STRUCT *finddata)
156 {
157 return ::FindFirstFile(spec, finddata);
158 }
159
160 static inline bool FindNext(FIND_DATA fd, FIND_STRUCT *finddata)
161 {
162 return ::FindNextFile(fd, finddata) != 0;
163 }
164
165 static const wxChar *GetNameFromFindData(FIND_STRUCT *finddata)
166 {
167 return finddata->cFileName;
168 }
169
170 static const FIND_ATTR GetAttrFromFindData(FIND_STRUCT *finddata)
171 {
172 return finddata->dwFileAttributes;
173 }
174
175 static inline bool IsDir(FIND_ATTR attr)
176 {
177 return (attr & FILE_ATTRIBUTE_DIRECTORY) != 0;
178 }
179
180 static inline bool IsHidden(FIND_ATTR attr)
181 {
182 return (attr & (FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM)) != 0;
183 }
184 #endif // __WIN16__
185
186 // ----------------------------------------------------------------------------
187 // constants
188 // ----------------------------------------------------------------------------
189
190 #ifndef MAX_PATH
191 #define MAX_PATH 260 // from VC++ headers
192 #endif
193
194 // ----------------------------------------------------------------------------
195 // macros
196 // ----------------------------------------------------------------------------
197
198 #define M_DIR ((wxDirData *)m_data)
199
200 // ----------------------------------------------------------------------------
201 // private classes
202 // ----------------------------------------------------------------------------
203
204 // this class stores everything we need to enumerate the files
205 class wxDirData
206 {
207 public:
208 wxDirData(const wxString& dirname);
209 ~wxDirData();
210
211 void SetFileSpec(const wxString& filespec) { m_filespec = filespec; }
212 void SetFlags(int flags) { m_flags = flags; }
213
214 void Close();
215 void Rewind();
216 bool Read(wxString *filename);
217
218 const wxString& GetName() const { return m_dirname; }
219
220 private:
221 FIND_DATA m_finddata;
222
223 wxString m_dirname;
224 wxString m_filespec;
225
226 int m_flags;
227
228 DECLARE_NO_COPY_CLASS(wxDirData)
229 };
230
231 // ============================================================================
232 // implementation
233 // ============================================================================
234
235 // ----------------------------------------------------------------------------
236 // wxDirData
237 // ----------------------------------------------------------------------------
238
239 wxDirData::wxDirData(const wxString& dirname)
240 : m_dirname(dirname)
241 {
242 m_finddata = InitFindData();
243 }
244
245 wxDirData::~wxDirData()
246 {
247 Close();
248 }
249
250 void wxDirData::Close()
251 {
252 if ( IsFindDataOk(m_finddata) )
253 {
254 FreeFindData(m_finddata);
255
256 m_finddata = InitFindData();
257 }
258 }
259
260 void wxDirData::Rewind()
261 {
262 Close();
263 }
264
265 bool wxDirData::Read(wxString *filename)
266 {
267 bool first = FALSE;
268
269 #ifdef __WIN32__
270 WIN32_FIND_DATA finddata;
271 #define PTR_TO_FINDDATA (&finddata)
272 #else // Win16
273 #define PTR_TO_FINDDATA (m_finddata)
274 #endif
275
276 if ( !IsFindDataOk(m_finddata) )
277 {
278 // open first
279 wxString filespec = m_dirname;
280 if ( !wxEndsWithPathSeparator(filespec) )
281 {
282 filespec += _T('\\');
283 }
284 filespec += (!m_filespec ? _T("*.*") : m_filespec.c_str());
285
286 m_finddata = FindFirst(filespec, PTR_TO_FINDDATA);
287
288 first = TRUE;
289 }
290
291 if ( !IsFindDataOk(m_finddata) )
292 {
293 #ifdef __WIN32__
294 DWORD err = ::GetLastError();
295
296 if ( err != ERROR_FILE_NOT_FOUND )
297 {
298 wxLogSysError(err, _("Can not enumerate files in directory '%s'"),
299 m_dirname.c_str());
300 }
301 #endif // __WIN32__
302 //else: not an error, just no (such) files
303
304 return FALSE;
305 }
306
307 const wxChar *name;
308 FIND_ATTR attr;
309
310 for ( ;; )
311 {
312 if ( first )
313 {
314 first = FALSE;
315 }
316 else
317 {
318 if ( !FindNext(m_finddata, PTR_TO_FINDDATA) )
319 {
320 #ifdef __WIN32__
321 DWORD err = ::GetLastError();
322
323 if ( err != ERROR_NO_MORE_FILES )
324 {
325 wxLogLastError(_T("FindNext"));
326 }
327 #endif // __WIN32__
328 //else: not an error, just no more (such) files
329
330 return FALSE;
331 }
332 }
333
334 name = GetNameFromFindData(PTR_TO_FINDDATA);
335 attr = GetAttrFromFindData(PTR_TO_FINDDATA);
336
337 // don't return "." and ".." unless asked for
338 if ( name[0] == _T('.') &&
339 ((name[1] == _T('.') && name[2] == _T('\0')) ||
340 (name[1] == _T('\0'))) )
341 {
342 if ( !(m_flags & wxDIR_DOTDOT) )
343 continue;
344 }
345
346 // check the type now
347 if ( !(m_flags & wxDIR_FILES) && !IsDir(attr) )
348 {
349 // it's a file, but we don't want them
350 continue;
351 }
352 else if ( !(m_flags & wxDIR_DIRS) && IsDir(attr) )
353 {
354 // it's a dir, and we don't want it
355 continue;
356 }
357
358 // finally, check whether it's a hidden file
359 if ( !(m_flags & wxDIR_HIDDEN) )
360 {
361 if ( IsHidden(attr) )
362 {
363 // it's a hidden file, skip it
364 continue;
365 }
366 }
367
368 *filename = name;
369
370 break;
371 }
372
373 return TRUE;
374 }
375
376 // ----------------------------------------------------------------------------
377 // wxDir helpers
378 // ----------------------------------------------------------------------------
379
380 /* static */
381 bool wxDir::Exists(const wxString& dir)
382 {
383 return wxPathExists(dir);
384 }
385
386 // ----------------------------------------------------------------------------
387 // wxDir construction/destruction
388 // ----------------------------------------------------------------------------
389
390 wxDir::wxDir(const wxString& dirname)
391 {
392 m_data = NULL;
393
394 (void)Open(dirname);
395 }
396
397 bool wxDir::Open(const wxString& dirname)
398 {
399 delete M_DIR;
400 m_data = new wxDirData(dirname);
401
402 return TRUE;
403 }
404
405 bool wxDir::IsOpened() const
406 {
407 return m_data != NULL;
408 }
409
410 wxString wxDir::GetName() const
411 {
412 wxString name;
413 if ( m_data )
414 {
415 name = M_DIR->GetName();
416 if ( !name.empty() )
417 {
418 // bring to canonical Windows form
419 name.Replace(_T("/"), _T("\\"));
420
421 if ( name.Last() == _T('\\') )
422 {
423 // chop off the last (back)slash
424 name.Truncate(name.length() - 1);
425 }
426 }
427 }
428
429 return name;
430 }
431
432 wxDir::~wxDir()
433 {
434 delete M_DIR;
435 }
436
437 // ----------------------------------------------------------------------------
438 // wxDir enumerating
439 // ----------------------------------------------------------------------------
440
441 bool wxDir::GetFirst(wxString *filename,
442 const wxString& filespec,
443 int flags) const
444 {
445 wxCHECK_MSG( IsOpened(), FALSE, _T("must wxDir::Open() first") );
446
447 M_DIR->Rewind();
448
449 M_DIR->SetFileSpec(filespec);
450 M_DIR->SetFlags(flags);
451
452 return GetNext(filename);
453 }
454
455 bool wxDir::GetNext(wxString *filename) const
456 {
457 wxCHECK_MSG( IsOpened(), FALSE, _T("must wxDir::Open() first") );
458
459 wxCHECK_MSG( filename, FALSE, _T("bad pointer in wxDir::GetNext()") );
460
461 return M_DIR->Read(filename);
462 }
463
464 // ----------------------------------------------------------------------------
465 // wxGetDirectoryTimes: used by wxFileName::GetTimes()
466 // ----------------------------------------------------------------------------
467
468 #ifdef __WIN32__
469
470 extern bool
471 wxGetDirectoryTimes(const wxString& dirname,
472 FILETIME *ftAccess, FILETIME *ftCreate, FILETIME *ftMod)
473 {
474 // FindFirst() is going to fail
475 wxASSERT_MSG( !dirname.empty() && dirname.Last() != _T('\\'),
476 _T("incorrect directory name format in wxGetDirectoryTimes") );
477
478 FIND_STRUCT fs;
479 FIND_DATA fd = FindFirst(dirname, &fs);
480 if ( !IsFindDataOk(fd) )
481 {
482 return FALSE;
483 }
484
485 *ftAccess = fs.ftLastAccessTime;
486 *ftCreate = fs.ftCreationTime;
487 *ftMod = fs.ftLastWriteTime;
488
489 FindClose(fd);
490
491 return TRUE;
492 }
493
494 #endif // __WIN32__
495