1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
|
// SelectDrivesDlg.h - Declaration of CDriveItem, CDrivesList and CSelectDrivesDlg
//
// WinDirStat - Directory Statistics
// Copyright (C) 2003-2005 Bernhard Seifert
// Copyright (C) 2004-2017 WinDirStat Team (windirstat.net)
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
#ifndef __WDS_SELECTDRIVESDLG_H__
#define __WDS_SELECTDRIVESDLG_H__
#pragma once
#include "ownerdrawnlistcontrol.h"
#include "layout.h"
#include "resource.h"
#include "set.h"
//
// The dialog has these three radio buttons.
//
enum RADIO
{
RADIO_ALLLOCALDRIVES,
RADIO_SOMEDRIVES,
RADIO_AFOLDER
};
class CDrivesList;
//
// CDriveItem. An item in the CDrivesList Control.
// All methods are called by the gui thread.
//
class CDriveItem: public COwnerDrawnListItem
{
public:
CDriveItem(CDrivesList *list, LPCTSTR pszPath);
void StartQuery(HWND dialog, UINT serial);
void SetDriveInformation(bool success, LPCTSTR name, ULONGLONG total, ULONGLONG free);
virtual int Compare(const CSortingListItem *other, int subitem) const;
CString GetPath() const;
CString GetDrive() const;
bool IsRemote() const;
bool IsSUBSTed() const;
virtual bool DrawSubitem(int subitem, CDC *pdc, CRect rc, UINT state, int *width, int *focusLeft) const;
virtual CString GetText(int subitem) const;
int GetImage() const;
private:
CDrivesList *m_list; // Backpointer
CString m_path; // e.g. "C:\"
bool m_isRemote; // Whether the drive type is DRIVE_REMOTE (network drive)
bool m_querying; // Information thread is running.
bool m_success; // Drive is accessible. false while m_querying is true.
CString m_name; // e.g. "BOOT (C:)"
ULONGLONG m_totalBytes; // Capacity
ULONGLONG m_freeBytes; // Free space
double m_used; // used space / total space
};
//
// CDriveInformationThread. Does the GetVolumeInformation() call, which
// may hang for ca. 30 sec, it a network drive is not accessible.
//
class CDriveInformationThread: public CWinThread
{
// Set of all running CDriveInformationThreads.
// Used by InvalidateDialogHandle().
static CSet<CDriveInformationThread *, CDriveInformationThread *> _runningThreads;
static CCriticalSection _csRunningThreads;
// The objects register and unregister themselves in _runningThreads
void AddRunningThread();
void RemoveRunningThread();
public:
static void InvalidateDialogHandle();
static void OnAppExit();
CDriveInformationThread(LPCTSTR path, LPARAM driveItem, HWND dialog, UINT serial);
virtual BOOL InitInstance();
LPARAM GetDriveInformation(bool& success, CString& name, ULONGLONG& total, ULONGLONG& free);
private:
const CString m_path; // Path like "C:\"
const LPARAM m_driveItem; // The list item, we belong to
CCriticalSection m_cs; // for m_dialog
HWND m_dialog; // synchronized by m_cs
const UINT m_serial; // serial number of m_dialog
// "[out]"-parameters
CString m_name; // Result: name like "BOOT (C:)", valid if m_success
ULONGLONG m_totalBytes; // Result: capacity of the drive, valid if m_success
ULONGLONG m_freeBytes; // Result: free space on the drive, valid if m_success
bool m_success; // Result: false, iff drive is unaccessible.
};
//
// CDrivesList.
//
class CDrivesList: public COwnerDrawnListControl
{
DECLARE_DYNAMIC(CDrivesList)
public:
CDrivesList();
CDriveItem *GetItem(int i);
void SelectItem(CDriveItem *item);
bool IsItemSelected(int i);
virtual bool HasImages();
DECLARE_MESSAGE_MAP()
afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
afx_msg void OnLvnDeleteitem(NMHDR *pNMHDR, LRESULT *pResult);
afx_msg void MeasureItem(LPMEASUREITEMSTRUCT lpMeasureItemStruct);
afx_msg void OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult);
};
//
// CSelectDrivesDlg. The initial dialog, where the user can select
// one or more drives or a folder for scanning.
//
class CSelectDrivesDlg : public CDialog
{
DECLARE_DYNAMIC(CSelectDrivesDlg)
enum { IDD = IDD_SELECTDRIVES };
static CString getFullPathName_(LPCTSTR relativePath);
public:
CSelectDrivesDlg(CWnd* pParent = NULL);
virtual ~CSelectDrivesDlg();
// Dialog Data
int m_radio; // out.
CString m_folderName; // out. Valid if m_radio = RADIO_AFOLDER
CStringArray m_drives; // out. Valid if m_radio != RADIO_AFOLDER
protected:
virtual void DoDataExchange(CDataExchange* pDX);
virtual BOOL OnInitDialog();
virtual void OnOK();
void UpdateButtons();
static UINT _serial; // Each Instance of this dialog gets a serial number
CDrivesList m_list;
CButton m_okButton;
CStringArray m_selectedDrives;
CLayout m_layout;
// Callback function for the dialog shown by SHBrowseForFolder()
// MUST be static!
static int CALLBACK BrowseCallbackProc(HWND hWnd, UINT uMsg, LPARAM lParam, LPARAM lpData);
DECLARE_MESSAGE_MAP()
afx_msg void OnBnClickedBrowsefolder();
// afx_msg void OnLbnSelchangeDrives();
afx_msg void OnBnClickedAlllocaldrives();
afx_msg void OnBnClickedAfolder();
afx_msg void OnBnClickedSomedrives();
afx_msg void OnEnChangeFoldername();
afx_msg void OnMeasureItem(int nIDCtl, LPMEASUREITEMSTRUCT lpMeasureItemStruct);
afx_msg void OnLvnItemchangedDrives(NMHDR *pNMHDR, LRESULT *pResult);
afx_msg void OnSize(UINT nType, int cx, int cy);
afx_msg void OnGetMinMaxInfo(MINMAXINFO* lpMMI);
afx_msg void OnDestroy();
afx_msg LRESULT OnWmuOk(WPARAM, LPARAM);
afx_msg LRESULT OnWmuThreadFinished(WPARAM, LPARAM lparam);
afx_msg void OnSysColorChange();
};
#endif // __WDS_SELECTDRIVESDLG_H__
|
