Fix 332+ maps aswell as 359+

src/Components/Modules/Zones.cpp

@@ -1655,6 +1655,24 @@ namespace Components
 				image->delayLoadPixels = image359.loaded;
 				image->name = image359.name;
 
+				FixImageCategory(image);
+
+				// Used for later stuff
+				(&image->delayLoadPixels)[1] = image359.pad3[1];
+			}
+			else
+			{
+				std::memcpy(buffer + 28, buffer + (size - 4), 4);
+
+				Game::GfxImage* image = reinterpret_cast<Game::GfxImage*>(buffer);
+				FixImageCategory(image);
+			}
+		}
+
+		return result;
+	}
+
+	void Zones::FixImageCategory(Game::GfxImage* image) {
 		// CODO makes use of additional enumerator values (9, 10, 11) that don't exist in IW4
 		// We have to translate them. 9 is for Reflection probes,  11 is for Compass,  10 is for Lightmap
 		switch (image->category)
@@ -1670,16 +1688,13 @@ namespace Components
 				break;
 		}
 
-				// Used for later stuff
-				(&image->delayLoadPixels)[1] = image359.pad3[1];
-			}
-			else
-			{
-				std::memcpy(buffer + 28, buffer + (size - 4), 4);
-			}
-		}
 
-		return result;
+		if (image->category > 7 || image->category < 0) {
+
+#ifdef DEBUG
+			if (IsDebuggerPresent()) __debugbreak();
+#endif
+		}
 	}
 
 	bool Zones::LoadXAsset(bool atStreamStart, char* buffer, int size)

src/Components/Modules/Zones.hpp

@@ -100,5 +100,6 @@ namespace Components
 		static void LoadMaterialAsset(Game::Material** asset);
 		static void LoadTracerDef(bool atStreamStart, Game::TracerDef* tracer, int size);
 		static void LoadTracerDefFxEffect();
+		static void FixImageCategory(Game::GfxImage* image);
 	};
 }