116 lines
3.3 KiB
C#
116 lines
3.3 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2015 The OpenRA Developers (see AUTHORS)
|
|
* 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. For more information,
|
|
* see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System.Linq;
|
|
using OpenRA.Graphics;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Widgets
|
|
{
|
|
public sealed class EditorResourceBrush : IEditorBrush
|
|
{
|
|
public readonly ResourceTypeInfo ResourceType;
|
|
|
|
readonly WorldRenderer worldRenderer;
|
|
readonly World world;
|
|
readonly EditorViewportControllerWidget editorWidget;
|
|
readonly SpriteWidget preview;
|
|
|
|
public EditorResourceBrush(EditorViewportControllerWidget editorWidget, ResourceTypeInfo resource, WorldRenderer wr)
|
|
{
|
|
this.editorWidget = editorWidget;
|
|
ResourceType = resource;
|
|
worldRenderer = wr;
|
|
world = wr.World;
|
|
|
|
preview = editorWidget.Get<SpriteWidget>("DRAG_LAYER_PREVIEW");
|
|
preview.Palette = resource.Palette;
|
|
preview.GetScale = () => worldRenderer.Viewport.Zoom;
|
|
preview.IsVisible = () => editorWidget.CurrentBrush == this;
|
|
|
|
var variant = resource.Variants.FirstOrDefault();
|
|
var sequenceProvider = wr.World.Map.Rules.Sequences[world.TileSet.Id];
|
|
var sequence = sequenceProvider.GetSequence("resources", variant);
|
|
var sprite = sequence.GetSprite(resource.MaxDensity - 1);
|
|
preview.GetSprite = () => sprite;
|
|
|
|
// The preview widget may be rendered by the higher-level code before it is ticked.
|
|
// Force a manual tick to ensure the bounds are set correctly for this first draw.
|
|
Tick();
|
|
}
|
|
|
|
public bool HandleMouseInput(MouseInput mi)
|
|
{
|
|
// Exclusively uses left and right mouse buttons, but nothing else
|
|
if (mi.Button != MouseButton.Left && mi.Button != MouseButton.Right)
|
|
return false;
|
|
|
|
if (mi.Button == MouseButton.Right)
|
|
{
|
|
if (mi.Event == MouseInputEvent.Up)
|
|
{
|
|
editorWidget.ClearBrush();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
var cell = worldRenderer.Viewport.ViewToWorld(mi.Location);
|
|
|
|
if (mi.Button == MouseButton.Left && AllowResourceAt(cell))
|
|
{
|
|
var type = (byte)ResourceType.ResourceType;
|
|
var index = (byte)ResourceType.MaxDensity;
|
|
world.Map.MapResources.Value[cell] = new ResourceTile(type, index);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public bool AllowResourceAt(CPos cell)
|
|
{
|
|
var mapResources = world.Map.MapResources.Value;
|
|
if (!mapResources.Contains(cell))
|
|
return false;
|
|
|
|
var tile = world.Map.MapTiles.Value[cell];
|
|
var tileInfo = world.TileSet.GetTileInfo(tile);
|
|
if (tileInfo == null)
|
|
return false;
|
|
|
|
var terrainType = world.TileSet.TerrainInfo[tileInfo.TerrainType];
|
|
|
|
if (mapResources[cell].Type == ResourceType.ResourceType)
|
|
return false;
|
|
|
|
if (!ResourceType.AllowedTerrainTypes.Contains(terrainType.Type))
|
|
return false;
|
|
|
|
return ResourceType.AllowOnRamps || tileInfo.RampType == 0;
|
|
}
|
|
|
|
public void Tick()
|
|
{
|
|
var cell = worldRenderer.Viewport.ViewToWorld(Viewport.LastMousePos);
|
|
var offset = WVec.Zero;
|
|
var location = world.Map.CenterOfCell(cell) + offset;
|
|
|
|
var cellScreenPosition = worldRenderer.ScreenPxPosition(location);
|
|
var cellScreenPixel = worldRenderer.Viewport.WorldToViewPx(cellScreenPosition);
|
|
|
|
preview.Bounds.X = cellScreenPixel.X;
|
|
preview.Bounds.Y = cellScreenPixel.Y;
|
|
}
|
|
|
|
public void Dispose() { }
|
|
}
|
|
}
|