Implement sidebar preview for voxel units

Author: Rampastring

src/TSMapEditor/UI/Sidebar/AircraftListPanel.cs

@@ -1,4 +1,5 @@
 using System;
+using Microsoft.Xna.Framework.Graphics;
 using Rampastring.XNAUI;
 using TSMapEditor.Models;
 using TSMapEditor.Rendering;
@@ -26,9 +27,19 @@ private void AircraftPlacementAction_ActionExited(object sender, EventArgs e)
 
         private readonly AircraftPlacementAction aircraftPlacementAction;
 
+        private RenderTarget2D renderTarget;
+
+        protected override (Texture2D regular, Texture2D remap) GetObjectTextures<T>(T objectType, ShapeImage[] textures)
+        {
+            const byte facingSouthEast = 64;
+            return GetTextureForVoxel(objectType, TheaterGraphics.AircraftModels, renderTarget, facingSouthEast);
+        }
+
         protected override void InitObjects()
         {
+            renderTarget = new RenderTarget2D(GraphicsDevice, ObjectTreeView.Width, ObjectTreeView.LineHeight, false, SurfaceFormat.Color, DepthFormat.None);
             InitObjectsBase(Map.Rules.AircraftTypes, null);
+            renderTarget.Dispose();
         }
 
         protected override void ObjectSelected()

src/TSMapEditor/UI/Sidebar/ObjectListPanel.cs

@@ -5,10 +5,10 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using TSMapEditor.CCEngine;
 using TSMapEditor.Models;
 using TSMapEditor.Models.ArtConfig;
 using TSMapEditor.Rendering;
-using TSMapEditor.Rendering.ObjectRenderers;
 
 namespace TSMapEditor.UI.Sidebar
 {
@@ -143,6 +143,28 @@ private void SearchBox_TextChanged(object sender, EventArgs e)
 
         protected abstract void InitObjects();
 
+        protected (Texture2D regular, Texture2D remap) GetTextureForVoxel<T>(T objectType, VoxelModel[] typeGraphicsArray, RenderTarget2D renderTarget, byte facing) where T : TechnoType, IArtConfigContainer
+        {
+            Renderer.BeginDraw();
+
+            var frame = typeGraphicsArray[objectType.Index].GetFrame(facing, RampType.None, false);
+            if (frame == null || frame.Texture == null)
+                return (null, null);
+
+            var remapFrame = typeGraphicsArray[objectType.Index].GetRemapFrame(facing, RampType.None, false);
+
+            Renderer.EndDraw();
+
+            // Render them as smaller textures to be independent of the voxel cache
+            Texture2D regularTexture = Helpers.RenderTextureAsSmaller(frame.Texture, renderTarget, GraphicsDevice);
+            Texture2D remapTexture = null;
+
+            if (remapFrame != null && remapFrame.Texture != null)
+                remapTexture = Helpers.RenderTextureAsSmaller(remapFrame.Texture, renderTarget, GraphicsDevice);
+
+            return (regularTexture, remapTexture);
+        }
+
         protected virtual (Texture2D regular, Texture2D remap) GetObjectTextures<T>(T objectType, ShapeImage[] textures) where T : TechnoType, IArtConfigContainer
         {
             Texture2D texture = null;
@@ -218,7 +240,20 @@ protected void InitObjectsBase<T>(List<T> objectTypeList, ShapeImage[] textures,
 
                         House ownerHouse = Map.StandardHouses.Find(h => h.ININame == ownerName);
                         if (ownerHouse != null)
+                        {
                             remapColor = ownerHouse.XNAColor;
+                        }
+                        else
+                        {
+                            // As as last resort, check if EditorRules has a remap color specified for the side
+                            string colorOverrideName = Map.EditorConfig.EditorRulesIni.GetStringValue("ObjectOwnerColors", ownerName, null);
+                            if (!string.IsNullOrWhiteSpace(colorOverrideName))
+                            {
+                                var rulesColor = Map.Rules.Colors.Find(c => c.Name == colorOverrideName);
+                                if (rulesColor != null)
+                                    remapColor = rulesColor.XNAColor;
+                            }
+                        }
 
                         // Prevent duplicates that can occur due to category overrides
                         // (For example, if objects owned by "Soviet1" are overridden to be listed under

src/TSMapEditor/UI/Sidebar/UnitListPanel.cs

@@ -1,4 +1,5 @@
-using Rampastring.XNAUI;
+using Microsoft.Xna.Framework.Graphics;
+using Rampastring.XNAUI;
 using System;
 using TSMapEditor.Models;
 using TSMapEditor.Rendering;
@@ -27,6 +28,19 @@ private void UnitPlacementAction_ActionExited(object sender, System.EventArgs e)
         private readonly bool isNaval;
         private readonly UnitPlacementAction unitPlacementAction;
 
+        private RenderTarget2D renderTarget;
+
+        protected override (Texture2D regular, Texture2D remap) GetObjectTextures<T>(T objectType, ShapeImage[] textures)
+        {
+            const byte facingSouthEast = 64;
+
+            var unitType = objectType as UnitType;
+            if (unitType.ArtConfig.Voxel)
+                return GetTextureForVoxel(objectType, TheaterGraphics.UnitModels, renderTarget, facingSouthEast);
+
+            return base.GetObjectTextures(objectType, textures);
+        }
+
         protected override void InitObjects()
         {
             Func<UnitType, bool> filterFunction = null;
@@ -40,7 +54,9 @@ protected override void InitObjects()
                 filterFunction = u => u.SpeedType != "Float" && u.MovementZone != "Water";
             }
 
+            renderTarget = new RenderTarget2D(GraphicsDevice, ObjectTreeView.Width, ObjectTreeView.LineHeight, false, SurfaceFormat.Color, DepthFormat.None);
             InitObjectsBase(Map.Rules.UnitTypes, TheaterGraphics.UnitTextures, filterFunction);
+            renderTarget.Dispose();
         }
 
         protected override void ObjectSelected()