SelectDrivesDlg.h

// SelectDrivesDlg.h	- Declaration of CDriveItem, CDrivesList and CSelectDrivesDlg
//
// WinDirStat - Directory Statistics
// Copyright (C) 2003-2004 Bernhard Seifert
//
// 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
//
// Author: bseifert@users.sourceforge.net, bseifert@daccord.net
//
// Last modified: $Date: 2004/11/14 21:50:44 $

#pragma once

#include "ownerdrawnlistcontrol.h"
#include "layout.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, LONGLONG total, LONGLONG 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:)"
	LONGLONG m_totalBytes;	// Capacity
	LONGLONG 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 deregister 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, LONGLONG& total, LONGLONG& 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
	LONGLONG m_totalBytes;	// Result: capacity of the drive, valid if m_success
	LONGLONG 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 };

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();
};

// $Log: SelectDrivesDlg.h,v $
// Revision 1.11  2004/11/14 21:50:44  assarbad
// - Pre-select the last used folder
//
// Revision 1.10  2004/11/13 08:17:07  bseifert
// Remove blanks in Unicode Configuration names.
//
// Revision 1.9  2004/11/12 22:14:16  bseifert
// Eliminated CLR_NONE. Minor corrections.
//
// Revision 1.8  2004/11/12 00:47:42  assarbad
// - Fixed the code for coloring of compressed/encrypted items. Now the coloring spans the full row!
//
// Revision 1.7  2004/11/08 00:46:26  assarbad
// - Added feature to distinguish compressed and encrypted files/folders by color as in the Windows 2000/XP explorer.
//   Same rules apply. (Green = encrypted / Blue = compressed)
//
// Revision 1.6  2004/11/05 16:53:06  assarbad
// Added Date and History tag where appropriate.
//