76 lines
2.2 KiB
C#
76 lines
2.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
using Robust.Shared.Serialization;
|
|
|
|
namespace Content.Shared.Damage
|
|
{
|
|
[Serializable, NetSerializable]
|
|
public enum DamageType
|
|
{
|
|
Blunt,
|
|
Slash,
|
|
Piercing,
|
|
Heat,
|
|
Shock,
|
|
Cold,
|
|
Poison,
|
|
Radiation,
|
|
Asphyxiation,
|
|
Bloodloss,
|
|
Cellular
|
|
}
|
|
|
|
public static class DamageTypeExtensions
|
|
{
|
|
// TODO: Automatically generate this
|
|
private static readonly ImmutableDictionary<DamageType, DamageClass> TypeToClass =
|
|
new Dictionary<DamageType, DamageClass>
|
|
{
|
|
{DamageType.Blunt, DamageClass.Brute},
|
|
{DamageType.Slash, DamageClass.Brute},
|
|
{DamageType.Piercing, DamageClass.Brute},
|
|
{DamageType.Heat, DamageClass.Burn},
|
|
{DamageType.Shock, DamageClass.Burn},
|
|
{DamageType.Cold, DamageClass.Burn},
|
|
{DamageType.Poison, DamageClass.Toxin},
|
|
{DamageType.Radiation, DamageClass.Toxin},
|
|
{DamageType.Asphyxiation, DamageClass.Airloss},
|
|
{DamageType.Bloodloss, DamageClass.Airloss},
|
|
{DamageType.Cellular, DamageClass.Genetic }
|
|
|
|
}.ToImmutableDictionary();
|
|
|
|
public static DamageClass ToClass(this DamageType type)
|
|
{
|
|
return TypeToClass[type];
|
|
}
|
|
|
|
public static Dictionary<DamageType, int> ToDictionary()
|
|
{
|
|
return Enum.GetValues(typeof(DamageType))
|
|
.Cast<DamageType>()
|
|
.ToDictionary(type => type, type => 0);
|
|
}
|
|
|
|
public static Dictionary<DamageClass, int> ToClassDictionary(IReadOnlyDictionary<DamageType, int> types)
|
|
{
|
|
var classes = DamageClassExtensions.ToDictionary();
|
|
|
|
foreach (var @class in classes.Keys.ToList())
|
|
foreach (var type in @class.ToTypes())
|
|
{
|
|
if (!types.TryGetValue(type, out var damage))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
classes[@class] += damage;
|
|
}
|
|
|
|
return classes;
|
|
}
|
|
}
|
|
}
|