101 lines
2.7 KiB
C#
Executable File
101 lines
2.7 KiB
C#
Executable File
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using OpenRa.Game.GameRules;
|
|
|
|
namespace OpenRa.Game.Traits
|
|
{
|
|
class ProductionInfo : ITraitInfo
|
|
{
|
|
public object Create(Actor self) { return new Production(self); }
|
|
}
|
|
|
|
class Production : IIssueOrder, IResolveOrder, IProducer, ITags
|
|
{
|
|
bool isPrimary = false;
|
|
public bool IsPrimary { get { return isPrimary; } }
|
|
|
|
public Production( Actor self ) { }
|
|
|
|
public virtual int2? CreationLocation( Actor self, LegacyUnitInfo producee )
|
|
{
|
|
return ( 1 / 24f * self.CenterLocation ).ToInt2();
|
|
}
|
|
|
|
public virtual int CreationFacing( Actor self, Actor newUnit )
|
|
{
|
|
return newUnit.Info.InitialFacing;
|
|
}
|
|
|
|
public bool Produce( Actor self, LegacyUnitInfo producee )
|
|
{
|
|
var location = CreationLocation( self, producee );
|
|
if( location == null || Game.UnitInfluence.GetUnitsAt( location.Value ).Any() )
|
|
return false;
|
|
|
|
var newUnit = new Actor( producee, location.Value, self.Owner );
|
|
newUnit.traits.Get<Unit>().Facing = CreationFacing( self, newUnit ); ;
|
|
|
|
var rp = self.traits.GetOrDefault<RallyPoint>();
|
|
if( rp != null )
|
|
{
|
|
var mobile = newUnit.traits.GetOrDefault<Mobile>();
|
|
if( mobile != null )
|
|
newUnit.QueueActivity( new Activities.Move( rp.rallyPoint, 1 ) );
|
|
}
|
|
|
|
var bi = self.Info as LegacyBuildingInfo;
|
|
if (bi != null && bi.SpawnOffset != null)
|
|
newUnit.CenterLocation = self.CenterLocation
|
|
+ new float2(bi.SpawnOffset[0], bi.SpawnOffset[1]);
|
|
|
|
Game.world.Add( newUnit );
|
|
|
|
foreach (var t in self.traits.WithInterface<INotifyProduction>())
|
|
t.UnitProduced(self, newUnit);
|
|
|
|
return true;
|
|
}
|
|
|
|
public IEnumerable<TagType> GetTags()
|
|
{
|
|
yield return (isPrimary) ? TagType.Primary : TagType.None;
|
|
}
|
|
|
|
public Order IssueOrder(Actor self, int2 xy, MouseInput mi, Actor underCursor)
|
|
{
|
|
if (mi.Button == MouseButton.Right && underCursor == self)
|
|
return new Order("Deploy", self, null, int2.Zero, null);
|
|
return null;
|
|
}
|
|
|
|
public void ResolveOrder(Actor self, Order order)
|
|
{
|
|
if (order.OrderString == "Deploy")
|
|
SetPrimaryProducer(self, !isPrimary);
|
|
}
|
|
|
|
public void SetPrimaryProducer(Actor self, bool state)
|
|
{
|
|
if (state == false)
|
|
{
|
|
isPrimary = false;
|
|
return;
|
|
}
|
|
|
|
// Cancel existing primaries
|
|
foreach (var p in (self.Info as LegacyBuildingInfo).Produces)
|
|
{
|
|
foreach (var b in Game.world.Actors.Where(x => x.traits.Contains<Production>()
|
|
&& x.Owner == self.Owner
|
|
&& x.traits.Get<Production>().IsPrimary == true
|
|
&& (x.Info as LegacyBuildingInfo).Produces.Contains(p)))
|
|
{
|
|
b.traits.Get<Production>().SetPrimaryProducer(b, false);
|
|
}
|
|
}
|
|
isPrimary = true;
|
|
Sound.Play("pribldg1.aud");
|
|
}
|
|
}
|
|
}
|