Merge pull request #5505 from pavlos256/map-preview-generation

Map preview generation

OpenRA.Game/Graphics/Sheet.cs

@@ -21,6 +21,7 @@ namespace OpenRA.Graphics
 		ITexture texture;
 		bool dirty;
 		byte[] data;
+		readonly object dirtyLock = new object();
 
 		public readonly Size Size;
 		public byte[] Data { get { return data ?? texture.GetData(); } }
@@ -68,6 +69,8 @@ namespace OpenRA.Graphics
 
 		public ITexture Texture
 		{
+			// This is only called from the main thread but 'dirty'
+			// is set from other threads too via CommitData().
 			get
 			{
 				if (texture == null)
@@ -76,11 +79,14 @@ namespace OpenRA.Graphics
 					dirty = true;
 				}
 
+				lock (dirtyLock)
+				{
 					if (dirty)
 					{
 						texture.SetData(data, Size.Width, Size.Height);
 						dirty = false;
 					}
+				}
 
 				return texture;
 			}
@@ -140,7 +146,10 @@ namespace OpenRA.Graphics
 			if (data == null)
 				throw new InvalidOperationException("Texture-wrappers are read-only");
 
+			lock (dirtyLock)
+			{
 				dirty = true;
 			}
 		}
 	}
+}

OpenRA.Game/Map/MapCache.cs

@@ -131,25 +131,35 @@ namespace OpenRA
 
 		void LoadAsyncInternal()
 		{
-			for (;;)
+			Log.Write("debug", "MapCache.LoadAsyncInternal started");
+
+			// Milliseconds to wait on one loop when nothing to do
+			var emptyDelay = 50;
+			// Keep the thread alive for at least 5 seconds after the last minimap generation
+			var maxKeepAlive = 5000 / emptyDelay;
+			var keepAlive = maxKeepAlive;
+
+			while (keepAlive-- > 0)
 			{
-				MapPreview p;
+				List<MapPreview> todo;
 				lock (syncRoot)
 				{
-					if (generateMinimap.Count == 0)
-						break;
-
-					p = generateMinimap.Peek();
-
-					// Preview already exists
-					if (p.Minimap != null)
+					todo = generateMinimap.Where(p => p.Minimap == null).ToList();
+					generateMinimap.Clear();
+				}
+				if (todo.Count == 0)
 				{
-						generateMinimap.Dequeue();
+					Thread.Sleep(emptyDelay);
 					continue;
 				}
-				}
+				else
+					keepAlive = maxKeepAlive;
 
 				// Render the minimap into the shared sheet
+				foreach (var p in todo)
+				{
+					// The rendering is thread safe because it only reads from the passed instances and writes to a new bitmap
+					var bitmap = p.CustomPreview ?? Minimap.RenderMapPreview(modData.DefaultRules.TileSets[p.Map.Tileset], p.Map, modData.DefaultRules, true);
 					// Note: this is not generally thread-safe, but it works here because:
 					//   (a) This worker is the only thread writing to this sheet
 					//   (b) The main thread is the only thread reading this sheet
@@ -158,15 +168,13 @@ namespace OpenRA
 					//       the next render cycle.
 					//   (d) Any partially written bytes from the next minimap is in an
 					//       unallocated area, and will be committed in the next cycle.
-				var bitmap = p.CustomPreview ?? Minimap.RenderMapPreview(modData.DefaultRules.TileSets[p.Map.Tileset], p.Map, modData.DefaultRules, true);
 					p.Minimap = sheetBuilder.Add(bitmap);
 
-				lock (syncRoot)
-					generateMinimap.Dequeue();
-
 					// Yuck... But this helps the UI Jank when opening the map selector significantly.
-				Thread.Sleep(50);
+					Thread.Sleep(Environment.ProcessorCount == 1 ? 25 : 5);
+				};
 			}
+			Log.Write("debug", "MapCache.LoadAsyncInternal ended");
 		}
 
 		public void CacheMinimap(MapPreview preview)
@@ -177,6 +185,7 @@ namespace OpenRA
 			if (previewLoaderThread == null || !previewLoaderThread.IsAlive)
 			{
 				previewLoaderThread = new Thread(LoadAsyncInternal);
+				previewLoaderThread.IsBackground = true;
 				previewLoaderThread.Start();
 			}
 		}