Skip to content

Commit

Permalink
Draw editor terrain/resource preview as part of the world.
Browse files Browse the repository at this point in the history
  • Loading branch information
pchote committed Jan 12, 2020
1 parent c4033ff commit 173d8ce
Show file tree
Hide file tree
Showing 8 changed files with 127 additions and 71 deletions.
8 changes: 8 additions & 0 deletions OpenRA.Game/Map/Map.cs
Expand Up @@ -824,6 +824,14 @@ public WDist DistanceAboveTerrain(WPos pos)
return new WDist(delta.Z);
}

public WVec Offset(CVec delta, int dz)
{
if (Grid.Type == MapGridType.Rectangular)
return new WVec(1024 * delta.X, 1024 * delta.Y, 0);

return new WVec(724 * (delta.X - delta.Y), 724 * (delta.X + delta.Y), 724 * dz);
}

/// <summary>
/// The size of the map Height step in world units
/// </summary>
Expand Down
38 changes: 11 additions & 27 deletions OpenRA.Mods.Common/EditorBrushes/EditorResourceBrush.cs
Expand Up @@ -10,7 +10,6 @@
#endregion

using System.Collections.Generic;
using System.Linq;
using OpenRA.Graphics;
using OpenRA.Mods.Common.Traits;

Expand All @@ -23,8 +22,9 @@ public sealed class EditorResourceBrush : IEditorBrush
readonly WorldRenderer worldRenderer;
readonly World world;
readonly EditorViewportControllerWidget editorWidget;
readonly SpriteWidget preview;
readonly EditorActionManager editorActionManager;
readonly EditorCursorLayer editorCursor;
readonly int cursorToken;

AddResourcesEditorAction action;
bool resourceAdded;
Expand All @@ -36,21 +36,10 @@ public EditorResourceBrush(EditorViewportControllerWidget editorWidget, Resource
worldRenderer = wr;
world = wr.World;
editorActionManager = world.WorldActor.Trait<EditorActionManager>();
editorCursor = world.WorldActor.Trait<EditorCursorLayer>();
action = new AddResourcesEditorAction(world.Map, ResourceType);

preview = editorWidget.Get<SpriteWidget>("DRAG_LAYER_PREVIEW");
preview.Palette = resource.Palette;
preview.GetScale = () => worldRenderer.Viewport.Zoom;
preview.IsVisible = () => editorWidget.CurrentBrush == this;

var variant = resource.Sequences.FirstOrDefault();
var sequence = wr.World.Map.Rules.Sequences.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();
cursorToken = editorCursor.SetResource(wr, resource);
}

public bool HandleMouseInput(MouseInput mi)
Expand All @@ -70,6 +59,9 @@ public bool HandleMouseInput(MouseInput mi)
return false;
}

if (editorCursor.CurrentToken != cursorToken)
return false;

var cell = worldRenderer.Viewport.ViewToWorld(mi.Location);

if (mi.Button == MouseButton.Left && mi.Event != MouseInputEvent.Up && AllowResourceAt(cell))
Expand Down Expand Up @@ -111,20 +103,12 @@ public bool AllowResourceAt(CPos cell)
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);
public void Tick() { }

preview.Bounds.X = cellScreenPixel.X;
preview.Bounds.Y = cellScreenPixel.Y;
public void Dispose()
{
editorCursor.Clear(cursorToken);
}

public void Dispose() { }
}

struct CellResource
Expand Down
47 changes: 16 additions & 31 deletions OpenRA.Mods.Common/EditorBrushes/EditorTileBrush.cs
Expand Up @@ -14,7 +14,6 @@
using System.Linq;
using OpenRA.Graphics;
using OpenRA.Mods.Common.Traits;
using OpenRA.Primitives;

namespace OpenRA.Mods.Common.Widgets
{
Expand All @@ -25,32 +24,26 @@ public sealed class EditorTileBrush : IEditorBrush
readonly WorldRenderer worldRenderer;
readonly World world;
readonly EditorViewportControllerWidget editorWidget;
readonly TerrainTemplatePreviewWidget preview;
readonly Rectangle bounds;
readonly EditorActionManager editorActionManager;
readonly EditorCursorLayer editorCursor;
readonly int cursorToken;

bool painting;

public EditorTileBrush(EditorViewportControllerWidget editorWidget, ushort template, WorldRenderer wr)
public EditorTileBrush(EditorViewportControllerWidget editorWidget, ushort id, WorldRenderer wr)
{
this.editorWidget = editorWidget;
Template = template;
worldRenderer = wr;
world = wr.World;

editorActionManager = world.WorldActor.Trait<EditorActionManager>();
editorCursor = world.WorldActor.Trait<EditorCursorLayer>();

preview = editorWidget.Get<TerrainTemplatePreviewWidget>("DRAG_TILE_PREVIEW");
preview.GetScale = () => worldRenderer.Viewport.Zoom;
preview.IsVisible = () => editorWidget.CurrentBrush == this;

preview.Template = world.Map.Rules.TileSet.Templates.First(t => t.Value.Id == template).Value;
var grid = world.Map.Grid;
bounds = worldRenderer.Theater.TemplateBounds(preview.Template, grid.TileSize, grid.Type);
Template = id;
worldRenderer = wr;
world = wr.World;

// 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();
var template = world.Map.Rules.TileSet.Templates.First(t => t.Value.Id == id).Value;
cursorToken = editorCursor.SetTerrainTemplate(wr, template);
}

public bool HandleMouseInput(MouseInput mi)
Expand Down Expand Up @@ -84,6 +77,9 @@ public bool HandleMouseInput(MouseInput mi)
if (mi.Event != MouseInputEvent.Down && mi.Event != MouseInputEvent.Move)
return true;

if (editorCursor.CurrentToken != cursorToken)
return false;

var cell = worldRenderer.Viewport.ViewToWorld(mi.Location);
var isMoving = mi.Event == MouseInputEvent.Move;

Expand Down Expand Up @@ -205,23 +201,12 @@ bool PlacementOverlapsSameTemplate(TerrainTemplateInfo template, CPos cell)
return false;
}

public void Tick()
public void Tick() { }

public void Dispose()
{
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);
var zoom = worldRenderer.Viewport.Zoom;

preview.Bounds.X = cellScreenPixel.X + (int)(zoom * bounds.X);
preview.Bounds.Y = cellScreenPixel.Y + (int)(zoom * bounds.Y);
preview.Bounds.Width = (int)(zoom * bounds.Width);
preview.Bounds.Height = (int)(zoom * bounds.Height);
editorCursor.Clear(cursorToken);
}

public void Dispose() { }
}

class PaintTileEditorAction : IEditorAction
Expand Down
88 changes: 85 additions & 3 deletions OpenRA.Mods.Common/Traits/World/EditorCursorLayer.cs
Expand Up @@ -16,7 +16,7 @@

namespace OpenRA.Mods.Common.Traits
{
public enum EditorCursorType { None, Actor }
public enum EditorCursorType { None, Actor, TerrainTemplate, Resource }

[Desc("Required for the map editor to work. Attach this to the world actor.")]
public class EditorCursorLayerInfo : ITraitInfo, Requires<EditorActorLayerInfo>
Expand All @@ -40,6 +40,12 @@ public class EditorCursorLayer : ITickRender, IRenderAboveShroud, IRenderAnnotat
WVec actorCenterOffset;
bool actorSharesCell;

public TerrainTemplateInfo TerrainTemplate { get; private set; }
public ResourceTypeInfo Resource { get; private set; }
CPos terrainOrResourceCell;
bool terrainOrResourceDirty;
readonly List<IRenderable> terrainOrResourcePreview = new List<IRenderable>();

public EditorCursorLayer(Actor self, EditorCursorLayerInfo info)
{
this.info = info;
Expand All @@ -54,7 +60,50 @@ void ITickRender.TickRender(WorldRenderer wr, Actor self)
if (wr.World.Type != WorldType.Editor)
return;

if (Actor != null)
if (Type == EditorCursorType.TerrainTemplate || Type == EditorCursorType.Resource)
{
var cell = wr.Viewport.ViewToWorld(Viewport.LastMousePos);
if (terrainOrResourceCell != cell || terrainOrResourceDirty)
{
terrainOrResourceCell = cell;
terrainOrResourceDirty = false;
terrainOrResourcePreview.Clear();
var pos = world.Map.CenterOfCell(cell);

if (Type == EditorCursorType.TerrainTemplate)
{
var i = 0;
for (var y = 0; y < TerrainTemplate.Size.Y; y++)
{
for (var x = 0; x < TerrainTemplate.Size.X; x++)
{
var tile = new TerrainTile(TerrainTemplate.Id, (byte)i++);
var tileInfo = world.Map.Rules.TileSet.GetTileInfo(tile);

// Empty tile
if (tileInfo == null)
continue;

var sprite = wr.Theater.TileSprite(tile, 0);
var offset = world.Map.Offset(new CVec(x, y), tileInfo.Height);
var palette = wr.Palette(TerrainTemplate.Palette ?? TileSet.TerrainPaletteInternalName);

terrainOrResourcePreview.Add(new SpriteRenderable(sprite, pos, offset, 0, palette, 1, false));
}
}
}
else
{
var variant = Resource.Sequences.FirstOrDefault();
var sequence = wr.World.Map.Rules.Sequences.GetSequence("resources", variant);
var sprite = sequence.GetSprite(Resource.MaxDensity - 1);
var palette = wr.Palette(Resource.Palette);

terrainOrResourcePreview.Add(new SpriteRenderable(sprite, pos, WVec.Zero, 0, palette, 1, false));
}
}
}
else if (Type == EditorCursorType.Actor)
{
// Offset mouse position by the center offset (in world pixels)
var worldPx = wr.Viewport.ViewToWorldPx(Viewport.LastMousePos) - wr.ScreenPxOffset(actorCenterOffset);
Expand Down Expand Up @@ -99,6 +148,9 @@ IEnumerable<IRenderable> IRenderAboveShroud.RenderAboveShroud(Actor self, WorldR
if (wr.World.Type != WorldType.Editor)
return NoRenderables;

if (Type == EditorCursorType.TerrainTemplate || Type == EditorCursorType.Resource)
return terrainOrResourcePreview;

if (Type == EditorCursorType.Actor)
return Actor.Render().OrderBy(WorldRenderer.RenderableScreenZPositionComparisonKey);

Expand All @@ -112,7 +164,7 @@ public IEnumerable<IRenderable> RenderAnnotations(Actor self, WorldRenderer wr)
if (wr.World.Type != WorldType.Editor)
return NoRenderables;

return Actor != null ? Actor.RenderAnnotations() : NoRenderables;
return Type == EditorCursorType.Actor ? Actor.RenderAnnotations() : NoRenderables;
}

bool IRenderAnnotations.SpatiallyPartitionable { get { return false; } }
Expand Down Expand Up @@ -154,6 +206,34 @@ public int SetActor(WorldRenderer wr, ActorInfo actor, PlayerReference owner)

Type = EditorCursorType.Actor;
Actor = new EditorActorPreview(wr, null, reference, owner);
TerrainTemplate = null;
Resource = null;

return ++CurrentToken;
}

public int SetTerrainTemplate(WorldRenderer wr, TerrainTemplateInfo template)
{
terrainOrResourceCell = wr.Viewport.ViewToWorld(wr.Viewport.WorldToViewPx(Viewport.LastMousePos));

Type = EditorCursorType.TerrainTemplate;
TerrainTemplate = template;
Actor = null;
Resource = null;
terrainOrResourceDirty = true;

return ++CurrentToken;
}

public int SetResource(WorldRenderer wr, ResourceTypeInfo resource)
{
terrainOrResourceCell = wr.Viewport.ViewToWorld(wr.Viewport.WorldToViewPx(Viewport.LastMousePos));

Type = EditorCursorType.Resource;
Resource = resource;
Actor = null;
TerrainTemplate = null;
terrainOrResourceDirty = true;

return ++CurrentToken;
}
Expand All @@ -165,6 +245,8 @@ public void Clear(int token)

Type = EditorCursorType.None;
Actor = null;
TerrainTemplate = null;
Resource = null;
}
}
}
Expand Up @@ -20,6 +20,7 @@ public class LayerSelectorLogic : ChromeLogic
{
readonly EditorViewportControllerWidget editor;
readonly WorldRenderer worldRenderer;
readonly EditorCursorLayer editorCursor;

readonly ScrollPanelWidget layerTemplateList;
readonly ScrollItemWidget layerPreviewTemplate;
Expand All @@ -29,6 +30,7 @@ public LayerSelectorLogic(Widget widget, WorldRenderer worldRenderer)
{
this.worldRenderer = worldRenderer;
editor = widget.Parent.Get<EditorViewportControllerWidget>("MAP_EDITOR");
editorCursor = worldRenderer.World.WorldActor.Trait<EditorCursorLayer>();

layerTemplateList = widget.Get<ScrollPanelWidget>("LAYERTEMPLATE_LIST");
layerTemplateList.Layout = new GridLayout(layerTemplateList);
Expand All @@ -46,7 +48,7 @@ void IntializeLayerPreview(Widget widget)
foreach (var resource in resources)
{
var newResourcePreviewTemplate = ScrollItemWidget.Setup(layerPreviewTemplate,
() => { var brush = editor.CurrentBrush as EditorResourceBrush; return brush != null && brush.ResourceType == resource; },
() => editorCursor.Type == EditorCursorType.Resource && editorCursor.Resource == resource,
() => editor.SetBrush(new EditorResourceBrush(editor, resource, worldRenderer)));

newResourcePreviewTemplate.Bounds.X = 0;
Expand Down
5 changes: 4 additions & 1 deletion OpenRA.Mods.Common/Widgets/Logic/Editor/TileSelectorLogic.cs
Expand Up @@ -12,6 +12,7 @@
using System;
using System.Linq;
using OpenRA.Graphics;
using OpenRA.Mods.Common.Traits;
using OpenRA.Widgets;

namespace OpenRA.Mods.Common.Widgets.Logic
Expand All @@ -36,13 +37,15 @@ public TileSelectorTemplate(TerrainTemplateInfo template)

readonly TileSet tileset;
readonly TileSelectorTemplate[] allTemplates;
readonly EditorCursorLayer editorCursor;

[ObjectCreator.UseCtor]
public TileSelectorLogic(Widget widget, World world, WorldRenderer worldRenderer)
: base(widget, world, worldRenderer, "TILETEMPLATE_LIST", "TILEPREVIEW_TEMPLATE")
{
tileset = world.Map.Rules.TileSet;
allTemplates = tileset.Templates.Values.Select(t => new TileSelectorTemplate(t)).ToArray();
editorCursor = world.WorldActor.Trait<EditorCursorLayer>();

allCategories = allTemplates.SelectMany(t => t.Categories)
.Distinct()
Expand Down Expand Up @@ -98,7 +101,7 @@ protected override void InitializePreviews()

var tileId = t.Template.Id;
var item = ScrollItemWidget.Setup(ItemTemplate,
() => { var brush = Editor.CurrentBrush as EditorTileBrush; return brush != null && brush.Template == tileId; },
() => editorCursor.Type == EditorCursorType.TerrainTemplate && editorCursor.TerrainTemplate.Id == tileId,
() => Editor.SetBrush(new EditorTileBrush(Editor, tileId, WorldRenderer)));

var preview = item.Get<TerrainTemplatePreviewWidget>("TILE_PREVIEW");
Expand Down
4 changes: 0 additions & 4 deletions mods/cnc/chrome/editor.yaml
Expand Up @@ -220,10 +220,6 @@ Container@EDITOR_WORLD_ROOT:
TooltipContainer: TOOLTIP_CONTAINER
TooltipTemplate: SIMPLE_TOOLTIP
Children:
TerrainTemplatePreview@DRAG_TILE_PREVIEW:
Visible: false
Sprite@DRAG_LAYER_PREVIEW:
Visible: false
Background@ACTOR_EDIT_PANEL:
Background: panel-black
Width: 269
Expand Down
4 changes: 0 additions & 4 deletions mods/common/chrome/editor.yaml
Expand Up @@ -211,10 +211,6 @@ Container@EDITOR_WORLD_ROOT:
TooltipContainer: TOOLTIP_CONTAINER
TooltipTemplate: SIMPLE_TOOLTIP
Children:
TerrainTemplatePreview@DRAG_TILE_PREVIEW:
Visible: false
Sprite@DRAG_LAYER_PREVIEW:
Visible: false
Background@ACTOR_EDIT_PANEL:
X: 32
Y: 32
Expand Down

0 comments on commit 173d8ce

Please sign in to comment.