Streamlined colormap netsyncing so duplicates are not saved to $$$.sav per sector

src/p_saveg.c

@@ -474,50 +474,144 @@ static void P_NetUnArchivePlayers(void)
 	}
 }
 
-static void SaveExtraColormap(UINT8 *put, extracolormap_t *exc)
+///
+/// Colormaps
+///
+
+static extracolormap_t *net_colormaps = NULL;
+static UINT32 num_net_colormaps = 0;
+
+// Copypasta from r_data.c AddColormapToList
+// But also check for equality and return the matching index
+static UINT32 CheckAddNetColormapToList(extracolormap_t *extra_colormap)
 {
-	if (!exc) // Just give it default values, we done goofed. (or sector->extra_colormap was intentionally set to default (NULL))
+	extracolormap_t *exc;
-		exc = R_GetDefaultColormap();
+	UINT32 i = 0;
 
-	WRITEUINT8(put, exc->fadestart);
+	if (!net_colormaps)
-	WRITEUINT8(put, exc->fadeend);
+	{
-	WRITEUINT8(put, exc->fog);
+		net_colormaps = extra_colormap;
+		net_colormaps->next = 0;
+		net_colormaps->prev = 0;
+		return i;
+	}
 
-	WRITEINT32(put, exc->rgba);
+	for (exc = net_colormaps; exc->next; exc = exc->next)
-	WRITEINT32(put, exc->fadergba);
+	{
+		if (R_CheckEqualColormaps(exc, extra_colormap, true, true, true))
+			return i;
+		i++;
+	}
 
-#ifdef EXTRACOLORMAPLUMPS
+	exc->next = extra_colormap;
-	WRITESTRINGN(put, exc->lumpname, 9);
+	extra_colormap->prev = exc;
-#endif
+	extra_colormap->next = 0;
+
+	num_net_colormaps = i;
+	return i;
 }
 
-static extracolormap_t *LoadExtraColormap(UINT8 *get)
+static extracolormap_t *GetNetColormapFromList(UINT32 index)
 {
-	extracolormap_t *exc, *exc_exist;
+	// For loading, we have to be tricky:
+	// We load the sectors BEFORE knowing the colormap values
+	// So if an index doesn't exist, fill our list with dummy colormaps
+	// until we get the index we want
+	// Then when we load the color data, we set up the dummy colormaps
 
-	UINT8 fadestart = READUINT8(get),
+	extracolormap_t *exc, *last_exc = NULL;
-		fadeend = READUINT8(get),
+	UINT32 i = 0;
-		fog = READUINT8(get);
 
-	INT32 rgba = READINT32(get),
+	for (exc = net_colormaps; exc; last_exc = exc, exc = exc->next)
-		fadergba = READINT32(get);
+	{
+		if (i++ == index)
+			return exc;
+	}
+
+	// our index doesn't exist, so just make the entry
+	for (; i <= index && i < num_net_colormaps; i++)
+	{
+		exc = R_CreateDefaultColormap(false);
+		if (last_exc)
+			last_exc->next = exc;
+		exc->prev = last_exc;
+		exc->next = NULL;
+		last_exc = exc;
+	}
+	return exc;
+}
+
+static void ClearNetColormaps(void)
+{
+	// We're actually Z_Freeing each entry here,
+	// so don't call this in P_NetUnArchiveColormaps (where entries will be used in-game)
+	extracolormap_t *exc, *exc_next;
+
+	for (exc = net_colormaps; exc; exc = exc_next)
+	{
+		exc_next = exc->next;
+		Z_Free(exc);
+	}
+	num_net_colormaps = 0;
+	net_colormaps = NULL;
+}
+
+static void P_NetArchiveColormaps()
+{
+	// We save and then we clean up our colormap mess
+	extracolormap_t *exc, *exc_next;
+
+	WRITEUINT32(save_p, num_net_colormaps); // save for safety
+
+	for (exc = net_colormaps; exc; exc = exc_next)
+	{
+		WRITEUINT8(save_p, exc->fadestart);
+		WRITEUINT8(save_p, exc->fadeend);
+		WRITEUINT8(save_p, exc->fog);
+
+		WRITEINT32(save_p, exc->rgba);
+		WRITEINT32(save_p, exc->fadergba);
 
 #ifdef EXTRACOLORMAPLUMPS
-	char lumpname[9];
+		WRITESTRINGN(save_p, exc->lumpname, 9);
-	READSTRINGN(get, lumpname, 9);
-
-	if (lumpname[0])
-		return R_ColormapForName(lumpname);
 #endif
-	exc = R_GetColormapFromListByValues(rgba, fadergba, fadestart, fadeend, fog);
 
-	if (!exc)
+		exc_next = exc->next;
+		Z_Free(exc); // don't need anymore
+	}
+
+	num_net_colormaps = 0;
+}
+
+static void P_NetUnArchiveColormaps()
+{
+	// When we reach this point, we already populated our list with
+	// dummy colormaps. Now that we are loading the color data,
+	// set up the dummies.
+	extracolormap_t *exc, *existing_exc;
+	num_net_colormaps = READUINT32(save_p);
+
+	for (exc = net_colormaps; exc; exc = exc->next)
 	{
-		// CONS_Debug(DBG_RENDER, "Creating Colormap: rgba(%d,%d,%d,%d) fadergba(%d,%d,%d,%d)\n",
+		UINT8 fadestart = READUINT8(save_p),
-		// 	R_GetRgbaR(rgba), R_GetRgbaG(rgba), R_GetRgbaB(rgba), R_GetRgbaA(rgba),
+			fadeend = READUINT8(save_p),
-		//	R_GetRgbaR(fadergba), R_GetRgbaG(fadergba), R_GetRgbaB(fadergba), R_GetRgbaA(fadergba));
+			fog = READUINT8(save_p);
 
-		exc = Z_Calloc(sizeof (*exc), PU_LEVEL, NULL);
+		INT32 rgba = READINT32(save_p),
+			fadergba = READINT32(save_p);
+
+#ifdef EXTRACOLORMAPLUMPS
+		char lumpname[9];
+		READSTRINGN(save_p, lumpname, 9);
+
+		if (lumpname[0])
+		{
+			existing_exc = R_ColormapForName(lumpname);
+			*exc = *existing_exc;
+			R_AddColormapToList(exc); // see HACK note below on why we're adding duplicates
+			continue;
+		}
+#endif
 
 		exc->fadestart = fadestart;
 		exc->fadeend = fadeend;
@@ -531,21 +625,32 @@ static extracolormap_t *LoadExtraColormap(UINT8 *get)
 		exc->lumpname[0] = 0;
 #endif
 
-		if (!(exc_exist = R_GetColormapFromList(exc)))
+		existing_exc = R_GetColormapFromListByValues(rgba, fadergba, fadestart, fadeend, fog);
-		{
+
-			exc->colormap = R_CreateLightTable(exc);
+		if (existing_exc)
-			R_AddColormapToList(exc);
+			exc->colormap = existing_exc->colormap;
-		}
 		else
-		{
+			// CONS_Debug(DBG_RENDER, "Creating Colormap: rgba(%d,%d,%d,%d) fadergba(%d,%d,%d,%d)\n",
-			Z_Free(exc);
+			// 	R_GetRgbaR(rgba), R_GetRgbaG(rgba), R_GetRgbaB(rgba), R_GetRgbaA(rgba),
-			exc = R_CheckDefaultColormap(exc_exist, true, true, true) ? NULL : exc_exist;
+			//	R_GetRgbaR(fadergba), R_GetRgbaG(fadergba), R_GetRgbaB(fadergba), R_GetRgbaA(fadergba));
-		}
+			exc->colormap = R_CreateLightTable(exc);
+
+		// HACK: If this dummy is a duplicate, we're going to add it
+		// to the extra_colormaps list anyway. I think this is faster
+		// than going through every loaded sector and correcting their
+		// colormap address to the pre-existing one, PER net_colormap entry
+		R_AddColormapToList(exc);
 	}
 
-	return exc;
+	// Don't need these anymore
+	num_net_colormaps = 0;
+	net_colormaps = NULL;
 }
 
+///
+/// World Archiving
+///
+
 #define SD_FLOORHT  0x01
 #define SD_CEILHT   0x02
 #define SD_FLOORPIC 0x04
@@ -595,6 +700,9 @@ static void P_NetArchiveWorld(void)
 	const side_t *si;
 	UINT8 *put;
 
+	// initialize colormap vars because paranoia
+	ClearNetColormaps();
+
 	// reload the map just to see difference
 	mapsector_t *ms;
 	mapsidedef_t *msd;
@@ -730,7 +838,8 @@ static void P_NetArchiveWorld(void)
 			}
 
 			if (diff3 & SD_COLORMAP)
-				SaveExtraColormap(put, ss->extra_colormap);
+				WRITEUINT32(put, CheckAddNetColormapToList(ss->extra_colormap));
+					// returns existing index if already added, or appends to net_colormaps and returns new index
 
 			// Special case: save the stats of all modified ffloors along with their ffloor "number"s
 			// we don't bother with ffloors that haven't changed, that would just add to savegame even more than is really needed
@@ -865,6 +974,9 @@ static void P_NetUnArchiveWorld(void)
 	if (READUINT32(save_p) != ARCHIVEBLOCK_WORLD)
 		I_Error("Bad $$$.sav at archive block World");
 
+	// initialize colormap vars because paranoia
+	ClearNetColormaps();
+
 	get = save_p;
 
 	for (;;)
@@ -927,7 +1039,7 @@ static void P_NetUnArchiveWorld(void)
 		}
 
 		if (diff3 & SD_COLORMAP)
-			sectors[i].extra_colormap = LoadExtraColormap(get);
+			sectors[i].extra_colormap = GetNetColormapFromList(READUINT32(get));
 
 		if (diff & SD_FFLOORS)
 		{
@@ -3536,6 +3648,7 @@ void P_SaveNetGame(void)
 #endif
 		P_NetArchiveThinkers();
 		P_NetArchiveSpecials();
+		P_NetArchiveColormaps();
 	}
 #ifdef HAVE_BLUA
 	LUA_Archive();
@@ -3578,6 +3691,7 @@ boolean P_LoadNetGame(void)
 #endif
 		P_NetUnArchiveThinkers();
 		P_NetUnArchiveSpecials();
+		P_NetUnArchiveColormaps();
 		P_RelinkPointers();
 		P_FinishMobjs();
 	}