127 lines
3.9 KiB
C#
127 lines
3.9 KiB
C#
#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 System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using OpenRA.Graphics;
|
|
using OpenRA.Mods.Common.Traits;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Cnc.Traits
|
|
{
|
|
[TraitLocation(SystemActors.World)]
|
|
sealed class TSResourceLayerInfo : ResourceLayerInfo
|
|
{
|
|
public readonly string VeinType = "Veins";
|
|
|
|
[ActorReference]
|
|
[Desc("Actor types that should be treated as veins for adjacency.")]
|
|
public readonly HashSet<string> VeinholeActors = new() { };
|
|
|
|
public override object Create(ActorInitializer init) { return new TSResourceLayer(init.Self, this); }
|
|
}
|
|
|
|
sealed class TSResourceLayer : ResourceLayer, INotifyActorDisposing
|
|
{
|
|
readonly TSResourceLayerInfo info;
|
|
readonly HashSet<CPos> veinholeCells = new();
|
|
|
|
public TSResourceLayer(Actor self, TSResourceLayerInfo info)
|
|
: base(self, info)
|
|
{
|
|
this.info = info;
|
|
}
|
|
|
|
protected override void WorldLoaded(World w, WorldRenderer wr)
|
|
{
|
|
// Cache locations of veinhole actors
|
|
w.ActorAdded += ActorAddedToWorld;
|
|
w.ActorRemoved += ActorRemovedFromWorld;
|
|
foreach (var a in w.Actors)
|
|
ActorAddedToWorld(a);
|
|
|
|
base.WorldLoaded(w, wr);
|
|
}
|
|
|
|
void ActorAddedToWorld(Actor a)
|
|
{
|
|
if (info.VeinholeActors.Contains(a.Info.Name))
|
|
foreach (var cell in a.OccupiesSpace.OccupiedCells())
|
|
veinholeCells.Add(cell.Cell);
|
|
}
|
|
|
|
void ActorRemovedFromWorld(Actor a)
|
|
{
|
|
if (info.VeinholeActors.Contains(a.Info.Name))
|
|
foreach (var cell in a.OccupiesSpace.OccupiedCells())
|
|
veinholeCells.Remove(cell.Cell);
|
|
}
|
|
|
|
void INotifyActorDisposing.Disposing(Actor self)
|
|
{
|
|
self.World.ActorAdded -= ActorAddedToWorld;
|
|
self.World.ActorRemoved -= ActorRemovedFromWorld;
|
|
}
|
|
|
|
bool IsValidResourceNeighbour(CPos cell, CPos neighbour)
|
|
{
|
|
if (!Map.Contains(neighbour))
|
|
return false;
|
|
|
|
// Non-vein resources are not allowed in the cardinal neighbours to
|
|
// an already existing vein cell
|
|
return Content[neighbour].Type != info.VeinType;
|
|
}
|
|
|
|
bool IsValidVeinNeighbour(CPos cell, CPos neighbour)
|
|
{
|
|
if (!Map.Contains(neighbour))
|
|
return false;
|
|
|
|
// Cell is automatically valid if it contains a veinhole actor
|
|
if (veinholeCells.Contains(neighbour))
|
|
return true;
|
|
|
|
// Neighbour must be flat or a cardinal slope, unless the resource cell itself is a slope
|
|
if (Map.Ramp[cell] == 0 && Map.Ramp[neighbour] > 4)
|
|
return false;
|
|
|
|
// Neighbour must be have a compatible terrain type (which also implies no other resources)
|
|
var neighbourTerrain = Map.GetTerrainInfo(neighbour).Type;
|
|
var veinInfo = info.ResourceTypes[info.VeinType];
|
|
return neighbourTerrain == veinInfo.TerrainType || veinInfo.AllowedTerrainTypes.Contains(neighbourTerrain);
|
|
}
|
|
|
|
protected override bool AllowResourceAt(string resourceType, CPos cell)
|
|
{
|
|
if (!Map.Contains(cell))
|
|
return false;
|
|
|
|
// Resources are allowed on flat terrain and cardinal slopes
|
|
if (Map.Ramp[cell] > 4)
|
|
return false;
|
|
|
|
if (!info.ResourceTypes.TryGetValue(resourceType, out var resourceInfo))
|
|
return false;
|
|
|
|
if (!resourceInfo.AllowedTerrainTypes.Contains(Map.GetTerrainInfo(cell).Type))
|
|
return false;
|
|
|
|
// Ensure there is space for the vein border tiles (not needed on ramps)
|
|
var check = resourceType == info.VeinType ? (Func<CPos, CPos, bool>)IsValidVeinNeighbour : IsValidResourceNeighbour;
|
|
var blockedByNeighbours = Map.Ramp[cell] == 0 && !Common.Util.ExpandFootprint(cell, false).All(c => check(cell, c));
|
|
|
|
return !blockedByNeighbours && (resourceType == info.VeinType || !BuildingInfluence.AnyBuildingAt(cell));
|
|
}
|
|
}
|
|
}
|