From 5a2c4504f21267b43441fa592595777d81d3d3fb Mon Sep 17 00:00:00 2001
From: Chris Danford <chrisdanford@users.sourceforge.net>
Date: Sun, 4 Dec 2005 09:02:40 +0000
Subject: [PATCH] add volume size

---
 stepmania/src/arch/MemoryCard/MemoryCardDriver.h  |  2 ++
 .../MemoryCardDriverThreaded_Windows.cpp          | 15 +++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/stepmania/src/arch/MemoryCard/MemoryCardDriver.h b/stepmania/src/arch/MemoryCard/MemoryCardDriver.h
index be0bce1b66..2d2ea57935 100644
--- a/stepmania/src/arch/MemoryCard/MemoryCardDriver.h
+++ b/stepmania/src/arch/MemoryCard/MemoryCardDriver.h
@@ -22,6 +22,7 @@ struct UsbStorageDevice
 		sVendor = "";
 		sProduct = "";
 		sVolumeLabel = "";
+		iVolumeSizeMB = 0;
 	};
 	int iBus;
 	int iPort;
@@ -60,6 +61,7 @@ struct UsbStorageDevice
 	CString sVendor;
 	CString sProduct;
 	CString sVolumeLabel;
+	int iVolumeSizeMB;
 
 	bool IsBlank() const { return m_State == STATE_NONE; }
 	void SetOsMountDir( const CString &s );
diff --git a/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp b/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp
index 1424dbf0a9..e0b7412a57 100644
--- a/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp
+++ b/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp
@@ -101,6 +101,21 @@ bool MemoryCardDriverThreaded_Windows::DoOneUpdate( bool bMount, vector<UsbStora
 			else
 				usbd.SetError( "MountFailed" );
 
+			// find volume size
+			DWORD dwSectorsPerCluster;
+			DWORD dwBytesPerSector;
+			DWORD dwNumberOfFreeClusters;
+			DWORD dwTotalNumberOfClusters;
+			if( GetDiskFreeSpace(
+					sDrive,
+					&dwSectorsPerCluster,
+					&dwBytesPerSector,
+					&dwNumberOfFreeClusters,
+					&dwTotalNumberOfClusters ) )
+			{
+				usbd.iVolumeSizeMB = (int)roundf( dwTotalNumberOfClusters * (float)dwSectorsPerCluster * dwBytesPerSector / (1024*1024) );
+			}
+
 			// read name
 			this->Mount( &usbd );
 			FILEMAN->Mount( "dir", usbd.sOsMountDir, TEMP_MOUNT_POINT_INTERNAL );