diff --git a/OpenRA.Mods.RA/Render/RenderBuildingOre.cs b/OpenRA.Mods.RA/Render/RenderBuildingOre.cs index 8759586a4f..1e6f97174e 100755 --- a/OpenRA.Mods.RA/Render/RenderBuildingOre.cs +++ b/OpenRA.Mods.RA/Render/RenderBuildingOre.cs @@ -17,18 +17,25 @@ namespace OpenRA.Mods.RA.Render public override object Create(ActorInitializer init) { return new RenderBuildingOre(init); } } - class RenderBuildingOre : RenderBuilding, INotifyBuildComplete + class RenderBuildingOre : RenderBuilding, INotifyBuildComplete, INotifyCapture { + PlayerResources PlayerResources; + public RenderBuildingOre( ActorInitializer init ) : base(init) { + PlayerResources = init.self.Owner.PlayerActor.Trait(); } public void BuildingComplete( Actor self ) { - var pr = self.Owner.PlayerActor.Trait(); anim.PlayFetchIndex( "idle", - () => (49 * pr.Ore) / (10*pr.OreCapacity)); + () => (49 * PlayerResources.Ore) / (10*PlayerResources.OreCapacity)); + } + + public void OnCapture (Actor self, Actor captor, Player oldOwner, Player newOwner) + { + PlayerResources = newOwner.PlayerActor.Trait(); } } }