OpenRA/OpenRA.Mods.Common/Widgets/TerrainTemplatePreviewWidget.cs

#region Copyright & License Information
/*
 * Copyright (c) The OpenRA Developers and Contributors
 * This file is part of OpenRA, which is free software. It is made
 * available to you under the terms of the GNU General Public License
 * as published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version. For more
 * information, see COPYING.
 */
#endregion

using OpenRA.Graphics;
using OpenRA.Mods.Common.Terrain;
using OpenRA.Mods.Common.Traits;
using OpenRA.Widgets;

namespace OpenRA.Mods.Common.Widgets
{
	public class TerrainTemplatePreviewWidget : Widget
	{
		public float Scale = 1f;

		readonly ITiledTerrainRenderer terrainRenderer;
		readonly WorldRenderer worldRenderer;
		readonly WorldViewportSizes viewportSizes;

		TerrainTemplateInfo template;

		public int2 PreviewPos { get; private set; }
		public int2 IdealPreviewSize { get; private set; }

		[ObjectCreator.UseCtor]
		public TerrainTemplatePreviewWidget(ModData modData, WorldRenderer worldRenderer, World world)
		{
			this.worldRenderer = worldRenderer;
			viewportSizes = modData.GetOrCreate<WorldViewportSizes>();

			terrainRenderer = world.WorldActor.TraitOrDefault<ITiledTerrainRenderer>();
			if (terrainRenderer == null)
				throw new YamlException("TerrainTemplatePreviewWidget requires a tile-based terrain renderer.");
		}

		protected TerrainTemplatePreviewWidget(TerrainTemplatePreviewWidget other)
			: base(other)
		{
			worldRenderer = other.worldRenderer;
			viewportSizes = other.viewportSizes;
			terrainRenderer = other.terrainRenderer;
			template = other.template;
			Scale = other.Scale;
		}

		public override TerrainTemplatePreviewWidget Clone() { return new TerrainTemplatePreviewWidget(this); }

		public void SetTemplate(TerrainTemplateInfo template)
		{
			this.template = template;
			var b = terrainRenderer.TemplateBounds(template);
			IdealPreviewSize = new int2((int)(b.Width * viewportSizes.DefaultScale), (int)(b.Height * viewportSizes.DefaultScale));

			// Measured from the middle of the widget to the middle of the top-left cell of the template
			PreviewPos = new int2((int)(b.Left * viewportSizes.DefaultScale), (int)(b.Top * viewportSizes.DefaultScale));
		}

		public override void Draw()
		{
			if (template == null)
				return;

			var scale = Scale * viewportSizes.DefaultScale;
			var origin = RenderOrigin - new int2((int)(PreviewPos.X * scale), (int)(PreviewPos.Y * scale));

			foreach (var r in terrainRenderer.RenderUIPreview(worldRenderer, template, origin, scale))
				r.PrepareRender(worldRenderer).Render(worldRenderer);
		}
	}
}