From bfd7b1930cff3652fa1822eb4fc0e43a91f82a26 Mon Sep 17 00:00:00 2001
From: Chris Danford <chrisdanford@users.sourceforge.net>
Date: Mon, 3 May 2004 00:34:54 +0000
Subject: [PATCH] simplify name update logic

---
 stepmania/src/GameState.cpp                   |  4 +--
 stepmania/src/MemoryCardManager.cpp           | 27 +++----------------
 stepmania/src/MemoryCardManager.h             |  3 +--
 stepmania/src/ScreenBranch.cpp                |  2 +-
 .../src/arch/MemoryCard/MemoryCardDriver.h    | 16 ++++++-----
 .../MemoryCardDriverThreaded_Windows.cpp      | 13 +++++++++
 .../MemoryCardDriverThreaded_Windows.h        |  4 ++-
 7 files changed, 32 insertions(+), 37 deletions(-)

diff --git a/stepmania/src/GameState.cpp b/stepmania/src/GameState.cpp
index df2b15ce37..27dc1e9ed0 100644
--- a/stepmania/src/GameState.cpp
+++ b/stepmania/src/GameState.cpp
@@ -368,11 +368,9 @@ void GameState::EndGame()
 		PROFILEMAN->UnloadProfile( pn );
 	}
 
-	MEMCARDMAN->RefreshNames();
-
 	// Reset the USB storage device numbers -after- saving
 	CHECKPOINT;
-	MEMCARDMAN->FlushAllDisks();
+	MEMCARDMAN->FlushAndReset();
 	CHECKPOINT;
 
 	SONGMAN->FreeAllLoadedFromProfiles();
diff --git a/stepmania/src/MemoryCardManager.cpp b/stepmania/src/MemoryCardManager.cpp
index 257cbe929e..10ccf63ab5 100644
--- a/stepmania/src/MemoryCardManager.cpp
+++ b/stepmania/src/MemoryCardManager.cpp
@@ -176,8 +176,8 @@ void MemoryCardManager::AssignUnassignedCards()
 				continue;
 
 			vector<UsbStorageDevice>::iterator it = find(vUnassignedDevices.begin(),vUnassignedDevices.end(),m_Device[p]);
-			ASSERT( it != vUnassignedDevices.end() )	// the assigned card better be connected!
-			vUnassignedDevices.erase( it );
+			if( it != vUnassignedDevices.end() )
+				vUnassignedDevices.erase( it );
 		}
 	}
 
@@ -259,7 +259,7 @@ match:
 	}
 }
 
-void MemoryCardManager::FlushAllDisks()
+void MemoryCardManager::FlushAndReset()
 {
 	FOREACH_PlayerNumber( p )
 	{
@@ -270,7 +270,7 @@ void MemoryCardManager::FlushAllDisks()
 		m_pDriver->Flush(&m_Device[p]);
 	}
 
-	m_pDriver->ResetUsbStorage();
+	m_pDriver->ResetUsbStorage();	// forces cards to be re-detected
 }
 
 bool MemoryCardManager::PathIsMemCard( CString sDir ) const
@@ -285,22 +285,3 @@ CString MemoryCardManager::GetName( PlayerNumber pn ) const
 {
 	return m_Device[pn].sName;
 }
-
-void MemoryCardManager::RefreshNames()
-{
-	// HACK: UsbStorageDevice::sName is only read when the device is connected.
-	// If the player leaves their memory card in after the game ends and they
-	// immediate start another game, then sName isn't re-read automatically.
-	// It's much harder to add logic to read the name again to MemoryCardDriver
-	// than to add a small hack here.
-
-	FOREACH_PlayerNumber( p )
-	{
-		if( m_Device[p].IsBlank() )	// no card assigned
-			continue;	// skip
-		if( m_bWriteError[p] || m_bTooLate[p] )
-			continue;	// skip
-		CString sProfileDir = MEM_CARD_MOUNT_POINT[p] + PREFSMAN->m_sMemoryCardProfileSubdir + '/'; 
-		m_Device[p].sName = Profile::GetProfileDisplayNameFromDir( sProfileDir );
-	}
-}
diff --git a/stepmania/src/MemoryCardManager.h b/stepmania/src/MemoryCardManager.h
index babb94309a..27a2359be0 100644
--- a/stepmania/src/MemoryCardManager.h
+++ b/stepmania/src/MemoryCardManager.h
@@ -37,12 +37,11 @@ public:
 
 	void LockCards( bool bLock );	// prevent removing or changing of memory cards
 	
-	void FlushAllDisks();	// force all files to be written to mounted memory cards
+	void FlushAndReset();	// force all files to be written to mounted memory cards
 
 	bool PathIsMemCard( CString sDir ) const;
 	
 	CString GetName( PlayerNumber pn ) const;
-	void RefreshNames();
 
 protected:
 	void AssignUnassignedCards();	// do our best to assign a Device to each player
diff --git a/stepmania/src/ScreenBranch.cpp b/stepmania/src/ScreenBranch.cpp
index 58daf1aa36..98fac2f332 100644
--- a/stepmania/src/ScreenBranch.cpp
+++ b/stepmania/src/ScreenBranch.cpp
@@ -30,7 +30,7 @@ ScreenBranch::ScreenBranch( CString sClassName ) : Screen( sClassName )
 	for( unsigned i=0; i<as.size(); i++ )
 	{
 		CString sChoice = Capitalize( as[i] );
-			CString sCondition = CONDITION(sChoice);
+		CString sCondition = CONDITION(sChoice);
 
 		if( Lua::RunExpression(sCondition) )
 		{
diff --git a/stepmania/src/arch/MemoryCard/MemoryCardDriver.h b/stepmania/src/arch/MemoryCard/MemoryCardDriver.h
index a3d2c8d5c8..50da4652ba 100644
--- a/stepmania/src/arch/MemoryCard/MemoryCardDriver.h
+++ b/stepmania/src/arch/MemoryCard/MemoryCardDriver.h
@@ -15,6 +15,7 @@ struct UsbStorageDevice
 		sScsiDevice = "";
 		sSerial = "";
 		sOsMountDir = "";
+		sName = "";
 	};
 	int iBus;
 	int iPort;
@@ -30,13 +31,14 @@ struct UsbStorageDevice
 
 	bool operator==(const UsbStorageDevice& other) const
 	{
-	  if( (iBus!=-1 || other.iBus!=-1) && iBus != other.iBus )
-	      return false;
-	  if( (iPort!=-1 || other.iPort!=-1) && iPort != other.iPort )
-	      return false;
-	  if( (iLevel!=-1 || other.iLevel!=-1) && iLevel != other.iLevel )
-	      return false;
-	  return sOsMountDir==other.sOsMountDir;  // every time a device is plugged in, it gets a unique device number
+#define COMPARE(x) if( x != other.x ) return false;
+		COMPARE( iBus );
+		COMPARE( iPort );
+		COMPARE( iLevel );
+		COMPARE( sName );
+		COMPARE( sOsMountDir );
+		return true;
+#undef COMPARE
 	}
   bool operator!=(const UsbStorageDevice& other) const
   {
diff --git a/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp b/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp
index 22cba65312..dc7fa932ad 100644
--- a/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp
+++ b/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.cpp
@@ -14,6 +14,8 @@ const CString TEMP_MOUNT_POINT = "@mctemp/";
 
 MemoryCardDriverThreaded_Windows::MemoryCardDriverThreaded_Windows()
 {
+	m_bReset = false;
+
 	StartThread();
 }
 
@@ -59,6 +61,12 @@ void MemoryCardDriverThreaded_Windows::MountThreadMain()
 	
 	while( !m_bShutdown )
 	{
+		if( m_bReset )
+		{
+			dwLastLogicalDrives = 0;
+			m_bReset = false;
+		}
+
 		DWORD dwNewLogicalDrives = ::GetLogicalDrives();
 		if( dwNewLogicalDrives != dwLastLogicalDrives )
 		{
@@ -141,6 +149,11 @@ void MemoryCardDriverThreaded_Windows::Flush( UsbStorageDevice* pDevice )
 	// Windows flushes automatically every ~2 seconds. -Chris
 }
 
+void MemoryCardDriverThreaded_Windows::ResetUsbStorage()
+{
+	m_bReset = true;
+}
+
 /*
  * Copyright (c) 2003 by the person(s) listed below.  All rights reserved.
  *	Chris Danford
diff --git a/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.h b/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.h
index 74d9434bb1..5ff0dd5570 100644
--- a/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.h
+++ b/stepmania/src/arch/MemoryCard/MemoryCardDriverThreaded_Windows.h
@@ -10,10 +10,12 @@ public:
 
 	virtual void Unmount( UsbStorageDevice* pDevice, CString sMountPoint );
 	virtual void Flush( UsbStorageDevice* pDevice );
-	virtual void ResetUsbStorage() {}
+	virtual void ResetUsbStorage();
 protected:
 	virtual void MountThreadMain();
 	virtual void Mount( UsbStorageDevice* pDevice, CString sMountPoint );
+
+	bool m_bReset;
 };
 
 #endif