GifManager refactor

LEGO1/lego/legoomni/include/gifmanager.h

@@ -3,73 +3,135 @@
 
 #include "compat.h"
 #include "decomp.h"
+#include "mxstl/stlcompat.h"
 #include "mxtypes.h"
 
 #include <d3drmobj.h>
 #include <ddraw.h>
 
+#pragma warning(disable : 4237)
+
 struct GifData {
 public:
-	const char* m_name;
+	char* m_name;
 	LPDIRECTDRAWSURFACE m_surface;
 	LPDIRECTDRAWPALETTE m_palette;
 	LPDIRECT3DRMTEXTURE2 m_texture;
 	MxU8* m_data;
+
+	~GifData();
 };
 
-struct GifMapEntry {
+struct GifMapComparator {
-public:
+	bool operator()(const char* const& p_key0, const char* const& p_key1) const { return strcmp(p_key0, p_key1) > 0; }
-	GifMapEntry* m_right;
-	GifMapEntry* m_parent;
-	GifMapEntry* m_left;
-	const char* m_key;
-	GifData* m_value;
 };
 
-class GifMap {
+class GifMap : public map<const char*, GifData*, GifMapComparator> {
-public:
+	// SYNTHETIC: LEGO1 0x1005a400
-	GifMapEntry* FindNode(const char*& p_string);
+	// GifMap::~GifMap
-
-	inline GifData* Get(const char* p_string)
-	{
-		GifData* ret = NULL;
-		GifMapEntry* entry = FindNode(p_string);
-		if (((m_unk0x4 == entry || strcmp(p_string, entry->m_key) > 0) ? m_unk0x4 : entry) != entry)
-			ret = entry->m_value;
-		return ret;
-	}
-
-	undefined4 m_unk0x0;
-	GifMapEntry* m_unk0x4;
 };
 
+typedef list<GifData*> GifList;
+
 // VTABLE: LEGO1 0x100d86d4
 class GifManagerBase {
 public:
-	// STUB: LEGO1 0x1005b660
+	// FUNCTION: LEGO1 0x1005b660
-	virtual ~GifManagerBase() {}
+	virtual ~GifManagerBase()
+	{
+		GifMap::iterator it;
+		for (it = m_map.begin(); it != m_map.end(); it++) {
+			// DECOMP: Use of const_cast here matches ~ViewLODListManager from 96 source.
+			const char* const& key = (*it).first;
+			delete[] const_cast<char*>(key);
+
+			if (m_ownership) {
+				delete (*it).second; // GifData*
+			}
+		}
+	}
+
+	inline GifData* Get(const char* p_name)
+	{
+		GifMap::iterator it = m_map.find(p_name);
+		if (it != m_map.end()) {
+			return (*it).second;
+		}
+
+		return NULL;
+	}
 
 	// SYNTHETIC: LEGO1 0x1005a310
 	// GifManagerBase::`scalar deleting destructor'
 
-	inline GifData* Get(const char* p_name) { return m_unk0x8.Get(p_name); }
-
 protected:
-	undefined4 m_unk0x0;
+	MxBool m_ownership;
-	undefined4 m_unk0x4;
+	GifMap m_map;
-	GifMap m_unk0x8;
 };
 
 // VTABLE: LEGO1 0x100d86fc
 class GifManager : public GifManagerBase {
 public:
+	GifManager() { m_ownership = TRUE; };
 	virtual ~GifManager() override;
 
 	// SYNTHETIC: LEGO1 0x1005a580
 	// GifManager::`scalar deleting destructor'
 
+	void FUN_10099cc0(GifData* p_data);
+
 protected:
-	undefined m_unk0x14[0x1c];
+	GifList m_list;
 };
 
+// TEMPLATE: LEGO1 0x10059c50
+// allocator<GifData *>::_Charalloc
+
+// clang-format off
+// TEMPLATE: LEGO1 0x10001cc0
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Lbound
+
+// TEMPLATE: LEGO1 0x1004f9b0
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Insert
+
+// TEMPLATE: LEGO1 0x10059c70
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Color
+
+// TEMPLATE: LEGO1 0x10059c80
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Left
+
+// TEMPLATE: LEGO1 0x10059c90
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Parent
+
+// TEMPLATE: LEGO1 0x10059ca0
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Right
+
+// TEMPLATE: LEGO1 0x10059cb0
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::~_Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >
+
+// TEMPLATE: LEGO1 0x10059d80
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::iterator::_Inc
+
+// TEMPLATE: LEGO1 0x10059dc0
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::erase
+
+// TEMPLATE: LEGO1 0x1005a210
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Erase
+
+// TEMPLATE: LEGO1 0x1005a250
+// list<GifData *,allocator<GifData *> >::~list<GifData *,allocator<GifData *> >
+
+// TEMPLATE: LEGO1 0x1005a2c0
+// map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::~map<char const *,GifData *,GifMapComparator,allocator<GifData *> >
+
+// TEMPLATE: LEGO1 0x1005a450
+// Map<char const *,GifData *,GifMapComparator>::~Map<char const *,GifData *,GifMapComparator>
+
+// TEMPLATE: LEGO1 0x1005a5a0
+// List<GifData *>::~List<GifData *>
+
+// GLOBAL: LEGO1 0x100f0100
+// _Tree<char const *,pair<char const * const,GifData *>,map<char const *,GifData *,GifMapComparator,allocator<GifData *> >::_Kfn,GifMapComparator,allocator<GifData *> >::_Nil
+// clang-format on
+
 #endif // GIFMANAGER_H

LEGO1/lego/legoomni/src/common/gifmanager.cpp

@@ -1,32 +1,66 @@
 #include "gifmanager.h"
 
 DECOMP_SIZE_ASSERT(GifData, 0x14);
-DECOMP_SIZE_ASSERT(GifMapEntry, 0x14);
+DECOMP_SIZE_ASSERT(GifMap, 0x10);
-DECOMP_SIZE_ASSERT(GifMap, 0x08);
+DECOMP_SIZE_ASSERT(GifManagerBase, 0x18);
-DECOMP_SIZE_ASSERT(GifManagerBase, 0x14);
+DECOMP_SIZE_ASSERT(GifManager, 0x24);
-DECOMP_SIZE_ASSERT(GifManager, 0x30);
 
-// GLOBAL: LEGO1 0x100f0100
+// FUNCTION: LEGO1 0x10065c00
-GifMapEntry* g_unk0x100f0100;
+GifData::~GifData()
-
-// FUNCTION: LEGO1 0x10001cc0
-GifMapEntry* GifMap::FindNode(const char*& p_string)
 {
-	GifMapEntry* ret = m_unk0x4;
+	if (m_name) {
-	GifMapEntry* current = ret->m_parent;
+		delete[] m_name;
-	while (current != g_unk0x100f0100) {
+		m_name = NULL;
-		if (strcmp(current->m_key, p_string) <= 0) {
+	}
-			ret = current;
+
-			current = current->m_right;
+	if (m_palette) {
-		}
+		m_palette->Release();
-		else
+		m_palette = NULL;
-			current = current->m_left;
+	}
+
+	if (m_surface) {
+		m_surface->Release();
+		m_surface = NULL;
+	}
+
+	if (m_texture) {
+		m_texture->Release();
+		m_texture = NULL;
 	}
-	return ret;
 }
 
-// STUB: LEGO1 0x10099870
+// FUNCTION: LEGO1 0x10099870
 GifManager::~GifManager()
 {
-	// TODO
+}
+
+// FUNCTION: LEGO1 0x10099cc0
+void GifManager::FUN_10099cc0(GifData* p_data)
+{
+	if (p_data == NULL)
+		return;
+
+#ifdef COMPAT_MODE
+	GifList::iterator it;
+	for (it = m_list.begin(); it != m_list.end(); it++) {
+#else
+	for (GifList::iterator it = m_list.begin(); it != m_list.end(); it++) {
+#endif
+		if (*it == p_data)
+			goto found;
+	}
+
+	// TODO: Maybe a function from <algorithm> would make this more idiomatic
+	// and not require a goto? This is not the only place where we iterate on
+	// a <list> and return early if there is no match.
+	return;
+
+found:
+	// TODO: This is wrong, but what is at +0xc on the iterator?
+	*it = NULL;
+
+	if (p_data->m_texture->Release() == TRUE) {
+		delete p_data;
+		m_list.erase(it);
+	}
 }

LEGO1/lego/legoomni/src/infocenter/score.cpp

@@ -227,8 +227,7 @@ void Score::VTable0x68(MxBool p_add)
 // FUNCTION: LEGO1 0x100019d0
 void Score::Paint()
 {
-	GifManager* gm = GetGifManager();
+	GifData* gd = GetGifManager()->Get("bigcube.gif");
-	GifData* gd = gm->Get("bigcube.gif");
 
 	if (gd) {
 		RaceState* l78 = (RaceState*) GameState()->GetState("JetskiRaceState");