60 lines
2.1 KiB
C#
60 lines
2.1 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Content.Server.GameObjects.Components.Chemistry;
|
|
using Content.Server.GameObjects.Components.Weapon.Ranged.Barrels;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Serialization.Manager.Attributes;
|
|
|
|
namespace Content.Server.GameObjects.Components.Weapon.Ranged.Ammunition
|
|
{
|
|
[RegisterComponent]
|
|
public class ChemicalAmmoComponent : Component
|
|
{
|
|
public override string Name => "ChemicalAmmo";
|
|
|
|
[DataField("fractionTransfered")]
|
|
private float _fractionTransfered = 1;
|
|
|
|
public override void HandleMessage(ComponentMessage message, IComponent? component)
|
|
{
|
|
base.HandleMessage(message, component);
|
|
switch (message)
|
|
{
|
|
case BarrelFiredMessage barrelFired:
|
|
TransferSolution(barrelFired);
|
|
break;
|
|
}
|
|
}
|
|
|
|
private void TransferSolution(BarrelFiredMessage barrelFired)
|
|
{
|
|
if (!Owner.TryGetComponent<SolutionContainerComponent>(out var ammoSolutionContainer))
|
|
return;
|
|
|
|
var projectiles = barrelFired.FiredProjectiles;
|
|
|
|
var projectileSolutionContainers = new List<SolutionContainerComponent>();
|
|
foreach (var projectile in projectiles)
|
|
{
|
|
if (projectile.TryGetComponent<SolutionContainerComponent>(out var projectileSolutionContainer))
|
|
{
|
|
projectileSolutionContainers.Add(projectileSolutionContainer);
|
|
}
|
|
}
|
|
|
|
if (!projectileSolutionContainers.Any())
|
|
return;
|
|
|
|
var solutionPerProjectile = ammoSolutionContainer.CurrentVolume * (1 / projectileSolutionContainers.Count);
|
|
|
|
foreach (var projectileSolutionContainer in projectileSolutionContainers)
|
|
{
|
|
var solutionToTransfer = ammoSolutionContainer.SplitSolution(solutionPerProjectile);
|
|
projectileSolutionContainer.TryAddSolution(solutionToTransfer);
|
|
}
|
|
|
|
ammoSolutionContainer.RemoveAllSolution();
|
|
}
|
|
}
|
|
}
|