Make brush rendering self-contained
This commit is contained in:
@@ -9,30 +9,65 @@
|
||||
*/
|
||||
#endregion
|
||||
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using OpenRA.Graphics;
|
||||
using OpenRA.Mods.Common.Traits;
|
||||
using OpenRA.Traits;
|
||||
|
||||
namespace OpenRA.Mods.Common.Widgets
|
||||
{
|
||||
public sealed class EditorActorBrush : IEditorBrush
|
||||
{
|
||||
public EditorActorPreview Preview;
|
||||
|
||||
readonly World world;
|
||||
readonly EditorActorLayer editorLayer;
|
||||
readonly EditorCursorLayer editorCursor;
|
||||
readonly EditorActionManager editorActionManager;
|
||||
readonly EditorViewportControllerWidget editorWidget;
|
||||
readonly int cursorToken;
|
||||
readonly WVec centerOffset;
|
||||
readonly bool sharesCell;
|
||||
|
||||
CPos cell;
|
||||
SubCell subcell = SubCell.Invalid;
|
||||
|
||||
public EditorActorBrush(EditorViewportControllerWidget editorWidget, ActorInfo actor, PlayerReference owner, WorldRenderer wr)
|
||||
{
|
||||
this.editorWidget = editorWidget;
|
||||
world = wr.World;
|
||||
editorLayer = world.WorldActor.Trait<EditorActorLayer>();
|
||||
editorCursor = world.WorldActor.Trait<EditorCursorLayer>();
|
||||
editorActionManager = world.WorldActor.Trait<EditorActionManager>();
|
||||
|
||||
cursorToken = editorCursor.SetActor(wr, actor, owner);
|
||||
var ios = actor.TraitInfoOrDefault<IOccupySpaceInfo>();
|
||||
centerOffset = (ios as BuildingInfo)?.CenterOffset(world) ?? WVec.Zero;
|
||||
sharesCell = ios != null && ios.SharesCell;
|
||||
|
||||
// Enforce first entry of ValidOwnerNames as owner if the actor has RequiresSpecificOwners.
|
||||
var ownerName = owner.Name;
|
||||
var specificOwnerInfo = actor.TraitInfoOrDefault<RequiresSpecificOwnersInfo>();
|
||||
if (specificOwnerInfo != null && !specificOwnerInfo.ValidOwnerNames.Contains(ownerName))
|
||||
ownerName = specificOwnerInfo.ValidOwnerNames.First();
|
||||
|
||||
var reference = new ActorReference(actor.Name)
|
||||
{
|
||||
new OwnerInit(ownerName),
|
||||
new FactionInit(owner.Faction)
|
||||
};
|
||||
|
||||
var worldPx = wr.Viewport.ViewToWorldPx(Viewport.LastMousePos) - wr.ScreenPxOffset(centerOffset);
|
||||
cell = wr.Viewport.ViewToWorld(wr.Viewport.WorldToViewPx(worldPx));
|
||||
reference.Add(new LocationInit(cell));
|
||||
if (sharesCell)
|
||||
{
|
||||
subcell = editorLayer.FreeSubCellAt(cell);
|
||||
if (subcell != SubCell.Invalid)
|
||||
reference.Add(new SubCellInit(subcell));
|
||||
}
|
||||
|
||||
if (actor.HasTraitInfo<IFacingInfo>())
|
||||
reference.Add(new FacingInit(editorLayer.Info.DefaultActorFacing));
|
||||
|
||||
Preview = new EditorActorPreview(wr, null, reference, owner);
|
||||
}
|
||||
|
||||
public bool HandleMouseInput(MouseInput mi)
|
||||
@@ -52,29 +87,56 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return false;
|
||||
}
|
||||
|
||||
if (editorCursor.CurrentToken != cursorToken)
|
||||
return false;
|
||||
|
||||
if (mi.Button == MouseButton.Left && mi.Event == MouseInputEvent.Down)
|
||||
{
|
||||
// Check the actor is inside the map
|
||||
var actor = editorCursor.Actor;
|
||||
if (!actor.Footprint.All(c => world.Map.Tiles.Contains(c.Key)))
|
||||
if (!Preview.Footprint.All(c => world.Map.Tiles.Contains(c.Key)))
|
||||
return true;
|
||||
|
||||
var action = new AddActorAction(editorLayer, actor.Export());
|
||||
var action = new AddActorAction(editorLayer, Preview.Export());
|
||||
editorActionManager.Add(action);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void IEditorBrush.TickRender(WorldRenderer wr, Actor self)
|
||||
{
|
||||
// Offset mouse position by the center offset (in world pixels)
|
||||
var worldPx = wr.Viewport.ViewToWorldPx(Viewport.LastMousePos) - wr.ScreenPxOffset(centerOffset);
|
||||
var currentCell = wr.Viewport.ViewToWorld(wr.Viewport.WorldToViewPx(worldPx));
|
||||
var currentSubcell = sharesCell ? editorLayer.FreeSubCellAt(currentCell) : SubCell.Invalid;
|
||||
if (cell != currentCell || subcell != currentSubcell)
|
||||
{
|
||||
cell = currentCell;
|
||||
Preview.ReplaceInit(new LocationInit(cell));
|
||||
|
||||
if (sharesCell)
|
||||
{
|
||||
subcell = editorLayer.FreeSubCellAt(cell);
|
||||
if (subcell == SubCell.Invalid)
|
||||
Preview.RemoveInit<SubCellInit>();
|
||||
else
|
||||
Preview.ReplaceInit(new SubCellInit(subcell));
|
||||
}
|
||||
|
||||
Preview.UpdateFromMove();
|
||||
}
|
||||
}
|
||||
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAboveShroud(Actor self, WorldRenderer wr)
|
||||
{
|
||||
return Preview.Render().OrderBy(WorldRenderer.RenderableZPositionComparisonKey);
|
||||
}
|
||||
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAnnotations(Actor self, WorldRenderer wr)
|
||||
{
|
||||
return Preview.RenderAnnotations();
|
||||
}
|
||||
|
||||
public void Tick() { }
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
editorCursor.Clear(cursorToken);
|
||||
}
|
||||
public void Dispose() { }
|
||||
}
|
||||
|
||||
sealed class AddActorAction : IEditorAction
|
||||
|
||||
@@ -14,6 +14,7 @@ using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using OpenRA.Graphics;
|
||||
using OpenRA.Mods.Common.EditorBrushes;
|
||||
using OpenRA.Mods.Common.Graphics;
|
||||
using OpenRA.Mods.Common.Traits;
|
||||
|
||||
namespace OpenRA.Mods.Common.Widgets
|
||||
@@ -94,14 +95,23 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return false;
|
||||
}
|
||||
|
||||
void IEditorBrush.TickRender(WorldRenderer wr, Actor self) { }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAboveShroud(Actor self, WorldRenderer wr) { yield break; }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAnnotations(Actor self, WorldRenderer wr)
|
||||
{
|
||||
if (PastePreviewPosition != null)
|
||||
{
|
||||
yield return new EditorSelectionAnnotationRenderable(Region, editorWidget.SelectionAltColor, editorWidget.SelectionAltOffset, PastePreviewPosition);
|
||||
yield return new EditorSelectionAnnotationRenderable(Region, editorWidget.PasteColor, int2.Zero, PastePreviewPosition);
|
||||
}
|
||||
}
|
||||
|
||||
public void Tick()
|
||||
{
|
||||
PastePreviewPosition = worldRenderer.Viewport.ViewToWorld(Viewport.LastMousePos);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
}
|
||||
public void Dispose() { }
|
||||
}
|
||||
|
||||
sealed class CopyPasteEditorAction : IEditorAction
|
||||
|
||||
@@ -10,7 +10,9 @@
|
||||
#endregion
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using OpenRA.Graphics;
|
||||
using OpenRA.Mods.Common.Graphics;
|
||||
using OpenRA.Mods.Common.Traits;
|
||||
using OpenRA.Widgets;
|
||||
|
||||
@@ -20,6 +22,10 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
{
|
||||
bool HandleMouseInput(MouseInput mi);
|
||||
void Tick();
|
||||
|
||||
void TickRender(WorldRenderer wr, Actor self);
|
||||
IEnumerable<IRenderable> RenderAboveShroud(Actor self, WorldRenderer wr);
|
||||
IEnumerable<IRenderable> RenderAnnotations(Actor self, WorldRenderer wr);
|
||||
}
|
||||
|
||||
public class EditorSelection
|
||||
@@ -255,6 +261,17 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return true;
|
||||
}
|
||||
|
||||
void IEditorBrush.TickRender(WorldRenderer wr, Actor self) { }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAboveShroud(Actor self, WorldRenderer wr) { yield break; }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAnnotations(Actor self, WorldRenderer wr)
|
||||
{
|
||||
if (CurrentDragBounds != null)
|
||||
{
|
||||
yield return new EditorSelectionAnnotationRenderable(CurrentDragBounds, editorWidget.SelectionAltColor, editorWidget.SelectionAltOffset, null);
|
||||
yield return new EditorSelectionAnnotationRenderable(CurrentDragBounds, editorWidget.SelectionMainColor, int2.Zero, null);
|
||||
}
|
||||
}
|
||||
|
||||
public void Tick() { }
|
||||
|
||||
public void Dispose() { }
|
||||
|
||||
@@ -75,6 +75,10 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return true;
|
||||
}
|
||||
|
||||
void IEditorBrush.TickRender(WorldRenderer wr, Actor self) { }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAboveShroud(Actor self, WorldRenderer wr) { yield break; }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAnnotations(Actor self, WorldRenderer wr) { yield break; }
|
||||
|
||||
public void Tick() { }
|
||||
|
||||
public void Dispose() { }
|
||||
|
||||
@@ -10,6 +10,7 @@
|
||||
#endregion
|
||||
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using OpenRA.Graphics;
|
||||
using OpenRA.Mods.Common.Traits;
|
||||
|
||||
@@ -23,13 +24,15 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
readonly World world;
|
||||
readonly EditorViewportControllerWidget editorWidget;
|
||||
readonly EditorActionManager editorActionManager;
|
||||
readonly EditorCursorLayer editorCursor;
|
||||
readonly IResourceLayer resourceLayer;
|
||||
readonly int cursorToken;
|
||||
|
||||
AddResourcesEditorAction action;
|
||||
bool resourceAdded;
|
||||
|
||||
CPos cell;
|
||||
readonly List<IRenderable> preview = new();
|
||||
readonly IResourceRenderer[] resourceRenderers;
|
||||
|
||||
public EditorResourceBrush(EditorViewportControllerWidget editorWidget, string resourceType, WorldRenderer wr)
|
||||
{
|
||||
this.editorWidget = editorWidget;
|
||||
@@ -37,11 +40,13 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
worldRenderer = wr;
|
||||
world = wr.World;
|
||||
editorActionManager = world.WorldActor.Trait<EditorActionManager>();
|
||||
editorCursor = world.WorldActor.Trait<EditorCursorLayer>();
|
||||
resourceLayer = world.WorldActor.Trait<IResourceLayer>();
|
||||
action = new AddResourcesEditorAction(resourceType, resourceLayer);
|
||||
|
||||
cursorToken = editorCursor.SetResource(wr, resourceType);
|
||||
resourceRenderers = world.WorldActor.TraitsImplementing<IResourceRenderer>().ToArray();
|
||||
cell = wr.Viewport.ViewToWorld(wr.Viewport.WorldToViewPx(Viewport.LastMousePos));
|
||||
UpdatePreview();
|
||||
|
||||
action = new AddResourcesEditorAction(resourceType, resourceLayer);
|
||||
}
|
||||
|
||||
public bool HandleMouseInput(MouseInput mi)
|
||||
@@ -61,9 +66,6 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return false;
|
||||
}
|
||||
|
||||
if (editorCursor.CurrentToken != cursorToken)
|
||||
return false;
|
||||
|
||||
var cell = worldRenderer.Viewport.ViewToWorld(mi.Location);
|
||||
|
||||
if (mi.Button == MouseButton.Left && mi.Event != MouseInputEvent.Up && resourceLayer.CanAddResource(ResourceType, cell))
|
||||
@@ -81,12 +83,30 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return true;
|
||||
}
|
||||
|
||||
void UpdatePreview()
|
||||
{
|
||||
var pos = world.Map.CenterOfCell(cell);
|
||||
|
||||
preview.Clear();
|
||||
preview.AddRange(resourceRenderers.SelectMany(r => r.RenderPreview(worldRenderer, ResourceType, pos)));
|
||||
}
|
||||
|
||||
void IEditorBrush.TickRender(WorldRenderer wr, Actor self)
|
||||
{
|
||||
var currentCell = wr.Viewport.ViewToWorld(Viewport.LastMousePos);
|
||||
if (cell != currentCell)
|
||||
{
|
||||
cell = currentCell;
|
||||
UpdatePreview();
|
||||
}
|
||||
}
|
||||
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAboveShroud(Actor self, WorldRenderer wr) { return preview; }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAnnotations(Actor self, WorldRenderer wr) { yield break; }
|
||||
|
||||
public void Tick() { }
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
editorCursor.Clear(cursorToken);
|
||||
}
|
||||
public void Dispose() { }
|
||||
}
|
||||
|
||||
readonly struct CellResource
|
||||
|
||||
@@ -20,6 +20,7 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
{
|
||||
public sealed class EditorTileBrush : IEditorBrush
|
||||
{
|
||||
public readonly TerrainTemplateInfo TerrainTemplate;
|
||||
public readonly ushort Template;
|
||||
|
||||
readonly WorldRenderer worldRenderer;
|
||||
@@ -27,11 +28,14 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
readonly ITemplatedTerrainInfo terrainInfo;
|
||||
readonly EditorViewportControllerWidget editorWidget;
|
||||
readonly EditorActionManager editorActionManager;
|
||||
readonly EditorCursorLayer editorCursor;
|
||||
readonly int cursorToken;
|
||||
|
||||
bool painting;
|
||||
|
||||
readonly ITiledTerrainRenderer terrainRenderer;
|
||||
|
||||
CPos cell;
|
||||
readonly List<IRenderable> preview = new();
|
||||
|
||||
public EditorTileBrush(EditorViewportControllerWidget editorWidget, ushort id, WorldRenderer wr)
|
||||
{
|
||||
this.editorWidget = editorWidget;
|
||||
@@ -42,12 +46,12 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
throw new InvalidDataException("EditorTileBrush can only be used with template-based tilesets");
|
||||
|
||||
editorActionManager = world.WorldActor.Trait<EditorActionManager>();
|
||||
editorCursor = world.WorldActor.Trait<EditorCursorLayer>();
|
||||
terrainRenderer = world.WorldActor.Trait<ITiledTerrainRenderer>();
|
||||
|
||||
Template = id;
|
||||
|
||||
var template = terrainInfo.Templates.First(t => t.Value.Id == id).Value;
|
||||
cursorToken = editorCursor.SetTerrainTemplate(wr, template);
|
||||
TerrainTemplate = terrainInfo.Templates.First(t => t.Value.Id == id).Value;
|
||||
cell = wr.Viewport.ViewToWorld(wr.Viewport.WorldToViewPx(Viewport.LastMousePos));
|
||||
UpdatePreview();
|
||||
}
|
||||
|
||||
public bool HandleMouseInput(MouseInput mi)
|
||||
@@ -81,9 +85,6 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
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;
|
||||
|
||||
@@ -143,12 +144,30 @@ namespace OpenRA.Mods.Common.Widgets
|
||||
return false;
|
||||
}
|
||||
|
||||
void UpdatePreview()
|
||||
{
|
||||
var pos = world.Map.CenterOfCell(cell);
|
||||
|
||||
preview.Clear();
|
||||
preview.AddRange(terrainRenderer.RenderPreview(worldRenderer, TerrainTemplate, pos));
|
||||
}
|
||||
|
||||
void IEditorBrush.TickRender(WorldRenderer wr, Actor self)
|
||||
{
|
||||
var currentCell = wr.Viewport.ViewToWorld(Viewport.LastMousePos);
|
||||
if (cell != currentCell)
|
||||
{
|
||||
cell = currentCell;
|
||||
UpdatePreview();
|
||||
}
|
||||
}
|
||||
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAboveShroud(Actor self, WorldRenderer wr) { return preview; }
|
||||
IEnumerable<IRenderable> IEditorBrush.RenderAnnotations(Actor self, WorldRenderer wr) { yield break; }
|
||||
|
||||
public void Tick() { }
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
editorCursor.Clear(cursorToken);
|
||||
}
|
||||
public void Dispose() { }
|
||||
}
|
||||
|
||||
sealed class PaintTileEditorAction : IEditorAction
|
||||
|
||||
Reference in New Issue
Block a user