Files
ss14-wl/Content.Shared/Preferences/HumanoidCharacterProfile.cs

1073 lines
38 KiB
C#
Raw Permalink Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
using System.Linq;
using System.Text.RegularExpressions;
using Content.Shared._WL.Skills; // WL-Skills
using Content.Shared.CCVar;
using Content.Shared.Corvax.TTS;
using Content.Shared.GameTicking;
using Content.Shared.Humanoid;
using Content.Shared.Humanoid.Prototypes;
using Content.Shared.Preferences.Loadouts;
using Content.Shared.Roles;
using Content.Shared.Traits;
using Robust.Shared.Collections;
using Robust.Shared.Configuration;
using Robust.Shared.Enums;
using Robust.Shared.Player;
using Robust.Shared.Prototypes;
using Robust.Shared.Random;
using Robust.Shared.Serialization;
using Robust.Shared.Utility;
namespace Content.Shared.Preferences
{
/// <summary>
/// Character profile. Looks immutable, but uses non-immutable semantics internally for serialization/code sanity purposes.
/// </summary>
[DataDefinition]
[Serializable, NetSerializable]
public sealed partial class HumanoidCharacterProfile : ICharacterProfile
{
private static readonly Regex RestrictedNameRegex = new("[^А-Яа-яёЁ0-9' \"-]"); // Corvax-Localization + WL-Changes
private static readonly Regex ICNameCaseRegex = new(@"^(?<word>\w)|\b(?<word>\w)(?=\w*$)");
//WL-Changes-start
public const int MaxDescLength = 512 * 2; // WL-CharacterInfo: Increase
public const int MaxRecordLength = 4096; // WL-Records
[DataField]
private Dictionary<string, string> _jobSubnames = new();
[DataField]
private Dictionary<string, bool> _jobUnblockings = new();
//WL-Changes-end
/// <summary>
/// Job preferences for initial spawn.
/// </summary>
[DataField]
private Dictionary<ProtoId<JobPrototype>, JobPriority> _jobPriorities = new()
{
{
SharedGameTicker.FallbackOverflowJob, JobPriority.High
}
};
/// <summary>
/// Antags we have opted in to.
/// </summary>
[DataField]
private HashSet<ProtoId<AntagPrototype>> _antagPreferences = new();
/// <summary>
/// Enabled traits.
/// </summary>
[DataField]
private HashSet<ProtoId<TraitPrototype>> _traitPreferences = new();
/// <summary>
/// <see cref="_loadouts"/>
/// </summary>
public IReadOnlyDictionary<string, RoleLoadout> Loadouts => _loadouts;
[DataField]
private Dictionary<string, RoleLoadout> _loadouts = new();
[DataField]
public string Name { get; set; } = "John Doe";
/// <summary>
/// Detailed text that can appear for the character if <see cref="CCVars.FlavorText"/> is enabled.
/// </summary>
[DataField]
public string FlavorText { get; set; } = string.Empty;
/// <summary>
/// Associated <see cref="SpeciesPrototype"/> for this profile.
/// </summary>
[DataField]
public ProtoId<SpeciesPrototype> Species { get; set; } = SharedHumanoidAppearanceSystem.DefaultSpecies;
[DataField]
public string Voice { get; set; } = SharedHumanoidAppearanceSystem.DefaultVoice;
[DataField]
public int Age { get; set; } = 18;
[DataField]
public Sex Sex { get; private set; } = Sex.Male;
[DataField]
public Gender Gender { get; private set; } = Gender.Male;
/// <summary>
/// <see cref="Appearance"/>
/// </summary>
public ICharacterAppearance CharacterAppearance => Appearance;
/// <summary>
/// Stores markings, eye colors, etc for the profile.
/// </summary>
[DataField]
public HumanoidCharacterAppearance Appearance { get; set; } = new();
/// <summary>
/// When spawning into a round what's the preferred spot to spawn.
/// </summary>
[DataField]
public SpawnPriorityPreference SpawnPriority { get; private set; } = SpawnPriorityPreference.None;
/// <summary>
/// <see cref="_jobPriorities"/>
/// </summary>
public IReadOnlyDictionary<ProtoId<JobPrototype>, JobPriority> JobPriorities => _jobPriorities;
/// <summary>
/// <see cref="_antagPreferences"/>
/// </summary>
public IReadOnlySet<ProtoId<AntagPrototype>> AntagPreferences => _antagPreferences;
/// <summary>
/// <see cref="_traitPreferences"/>
/// </summary>
public IReadOnlySet<ProtoId<TraitPrototype>> TraitPreferences => _traitPreferences;
/// <summary>
/// If we're unable to get one of our preferred jobs do we spawn as a fallback job or do we stay in lobby.
/// </summary>
[DataField]
public PreferenceUnavailableMode PreferenceUnavailable { get; private set; } =
PreferenceUnavailableMode.SpawnAsOverflow;
public HumanoidCharacterProfile(
string name,
string flavortext,
string ooctext, // WL-OOCText
string species,
string voice, // Corvax-TTS
int age,
int height,
Sex sex,
Gender gender,
HumanoidCharacterAppearance appearance,
SpawnPriorityPreference spawnPriority,
Dictionary<ProtoId<JobPrototype>, JobPriority> jobPriorities,
PreferenceUnavailableMode preferenceUnavailable,
//WL-Changes-start
Dictionary<string, string> jobSubnames,
//WL-Changes-end
HashSet<ProtoId<AntagPrototype>> antagPreferences,
HashSet<ProtoId<TraitPrototype>> traitPreferences,
Dictionary<string, RoleLoadout> loadouts,
//WL-Changes-start
Dictionary<string, bool> jobUnblockings,
string medicalRecord, // WL-Records
string securityRecord, // WL-Records
string employmentRecord, // WL-Records
Dictionary<string, Dictionary<byte, int>> skills // WL-Skills
//WL-Changes-end
)
{
Name = name;
FlavorText = flavortext;
OocText = ooctext; // WL-OOCText
Species = species;
Voice = voice; // Corvax-TTS
Age = age; // WL-Height
Height = height; // WL-Heigh
Sex = sex;
Gender = gender;
Appearance = appearance;
SpawnPriority = spawnPriority;
_jobPriorities = jobPriorities;
PreferenceUnavailable = preferenceUnavailable;
_antagPreferences = antagPreferences;
_traitPreferences = traitPreferences;
_loadouts = loadouts;
//WL-Changes-start
_jobSubnames = jobSubnames;
_jobUnblockings = jobUnblockings;
MedicalRecord = medicalRecord; // WL-Records
SecurityRecord = securityRecord; // WL-Records
EmploymentRecord = employmentRecord; // WL-Records
Skills = skills; // WL-Skills
//WL-Changes-end
var hasHighPrority = false;
foreach (var (key, value) in _jobPriorities)
{
if (value == JobPriority.Never)
_jobPriorities.Remove(key);
else if (value != JobPriority.High)
continue;
if (hasHighPrority)
_jobPriorities[key] = JobPriority.Medium;
hasHighPrority = true;
}
}
/// <summary>Copy constructor</summary>
public HumanoidCharacterProfile(HumanoidCharacterProfile other)
: this(other.Name,
other.FlavorText,
other.OocText, // WL-Heigh
other.Species,
other.Voice,
other.Age,
other.Height, // WL-Heigh
other.Sex,
other.Gender,
other.Appearance.Clone(),
other.SpawnPriority,
new Dictionary<ProtoId<JobPrototype>, JobPriority>(other.JobPriorities),
other.PreferenceUnavailable,
new Dictionary<string, string>(other.JobSubnames), //WL-Changes
new HashSet<ProtoId<AntagPrototype>>(other.AntagPreferences),
new HashSet<ProtoId<TraitPrototype>>(other.TraitPreferences),
new Dictionary<string, RoleLoadout>(other.Loadouts),
new(other.JobUnblockings), // WL-Heigh
other.MedicalRecord, // WL-Records
other.SecurityRecord, // WL-Records
other.EmploymentRecord, // WL-Records
other.Skills) // WL-Skills
{
}
/// <summary>
/// Get the default humanoid character profile, using internal constant values.
/// Defaults to <see cref="SharedHumanoidAppearanceSystem.DefaultSpecies"/> for the species.
/// </summary>
/// <returns></returns>
public HumanoidCharacterProfile()
{
}
/// <summary>
/// Return a default character profile, based on species.
/// </summary>
/// <param name="species">The species to use in this default profile. The default species is <see cref="SharedHumanoidAppearanceSystem.DefaultSpecies"/>.</param>
/// <returns>Humanoid character profile with default settings.</returns>
public static HumanoidCharacterProfile DefaultWithSpecies(string? species = null)
{
species ??= SharedHumanoidAppearanceSystem.DefaultSpecies;
return new()
{
Species = species,
Appearance = HumanoidCharacterAppearance.DefaultWithSpecies(species),
};
}
// TODO: This should eventually not be a visual change only.
public static HumanoidCharacterProfile Random(HashSet<string>? ignoredSpecies = null)
{
var prototypeManager = IoCManager.Resolve<IPrototypeManager>();
var random = IoCManager.Resolve<IRobustRandom>();
var species = random.Pick(prototypeManager
.EnumeratePrototypes<SpeciesPrototype>()
.Where(x => ignoredSpecies == null ? x.RoundStart : x.RoundStart && !ignoredSpecies.Contains(x.ID))
.ToArray()
).ID;
return RandomWithSpecies(species);
}
public static HumanoidCharacterProfile RandomWithSpecies(string? species = null)
{
species ??= SharedHumanoidAppearanceSystem.DefaultSpecies;
var prototypeManager = IoCManager.Resolve<IPrototypeManager>();
var random = IoCManager.Resolve<IRobustRandom>();
var sex = Sex.Unsexed;
var age = 18;
var height = 165; // WL-Height
if (prototypeManager.TryIndex<SpeciesPrototype>(species, out var speciesPrototype))
{
sex = random.Pick(speciesPrototype.Sexes);
age = random.Next(speciesPrototype.MinAge, speciesPrototype.OldAge); // people don't look and keep making 119 year old characters with zero rp, cap it at middle aged
height = random.Next(speciesPrototype.MinHeight, speciesPrototype.MaxHeight); // WL-Height
}
// Corvax-TTS-Start
var voiceId = random.Pick(prototypeManager
.EnumeratePrototypes<TTSVoicePrototype>()
.Where(o => CanHaveVoice(o, sex)).ToArray()
).ID;
// Corvax-TTS-End
var gender = Gender.Epicene;
switch (sex)
{
case Sex.Male:
gender = Gender.Male;
break;
case Sex.Female:
gender = Gender.Female;
break;
}
var name = GetName(species, gender);
return new HumanoidCharacterProfile()
{
Name = name,
Sex = sex,
Age = age,
Gender = gender,
Species = species,
Voice = voiceId, // Corvax-TTS
Appearance = HumanoidCharacterAppearance.Random(species, sex),
};
}
//WL-Changes-start
[DataField] public string OocText { get; private set; } = ""; // WL-OOCText
[DataField]
public string MedicalRecord { get; set; } = string.Empty;
[DataField]
public string SecurityRecord { get; set; } = string.Empty;
[DataField]
public string EmploymentRecord { get; set; } = string.Empty;
[DataField("height")] public int Height { get; private set; } = 165; // WL-Height
public IReadOnlyDictionary<string, string> JobSubnames => _jobSubnames; //WL-changes
public IReadOnlyDictionary<string, bool> JobUnblockings => _jobUnblockings;
[DataField] public Dictionary<string, Dictionary<byte, int>> Skills { get; set; } = new(); // WL-Skills
//WL-Changes-end
public HumanoidCharacterProfile WithName(string name)
{
return new(this) { Name = name };
}
public HumanoidCharacterProfile WithFlavorText(string flavorText)
{
return new(this) { FlavorText = flavorText };
}
// WL-OOCText-Start
public HumanoidCharacterProfile WithOocText(string oocText)
{
return new(this) { OocText = oocText };
}
// WL-OOCText-End
// WL-Records-Start
public HumanoidCharacterProfile WithMedicalRecord(string record)
{
return new(this) { MedicalRecord = record };
}
public HumanoidCharacterProfile WithSecurityRecord(string record)
{
return new(this) { SecurityRecord = record };
}
public HumanoidCharacterProfile WithEmploymentRecord(string record)
{
return new(this) { EmploymentRecord = record };
}
// WL-Records-End
public HumanoidCharacterProfile WithAge(int age)
{
return new(this) { Age = age };
}
// WL-Height-Start
public HumanoidCharacterProfile WithHeight(int height)
{
return new(this) { Height = height };
}
// WL-Height-End
public HumanoidCharacterProfile WithSex(Sex sex)
{
return new(this) { Sex = sex };
}
public HumanoidCharacterProfile WithGender(Gender gender)
{
return new(this) { Gender = gender };
}
public HumanoidCharacterProfile WithSpecies(string species)
{
return new(this) { Species = species };
}
// Corvax-TTS-Start
public HumanoidCharacterProfile WithVoice(string voice)
{
return new(this) { Voice = voice };
}
// Corvax-TTS-End
public HumanoidCharacterProfile WithCharacterAppearance(HumanoidCharacterAppearance appearance)
{
return new(this) { Appearance = appearance };
}
public HumanoidCharacterProfile WithSpawnPriorityPreference(SpawnPriorityPreference spawnPriority)
{
return new(this) { SpawnPriority = spawnPriority };
}
public HumanoidCharacterProfile WithJobPriorities(IEnumerable<KeyValuePair<ProtoId<JobPrototype>, JobPriority>> jobPriorities)
{
var dictionary = new Dictionary<ProtoId<JobPrototype>, JobPriority>(jobPriorities);
var hasHighPrority = false;
foreach (var (key, value) in dictionary)
{
if (value == JobPriority.Never)
dictionary.Remove(key);
else if (value != JobPriority.High)
continue;
if (hasHighPrority)
dictionary[key] = JobPriority.Medium;
hasHighPrority = true;
}
return new(this)
{
_jobPriorities = dictionary
};
}
//WL-Changes-start
public HumanoidCharacterProfile WithJobSubname(string jobId, string subname)
{
var dict = new Dictionary<string, string>(_jobSubnames);
dict[jobId] = subname;
return new(this)
{
_jobSubnames = dict
};
}
public HumanoidCharacterProfile WithJobUnblocking(string jobId, bool value)
{
var dict = new Dictionary<string, bool>(_jobUnblockings);
dict[jobId] = value;
return new(this)
{
_jobUnblockings = dict
};
}
public HumanoidCharacterProfile WithSkill(string jobName, byte skillKey, int level)
{
var newSkills = new Dictionary<string, Dictionary<byte, int>>(Skills);
if (!newSkills.TryGetValue(jobName, out var jobSkills))
{
jobSkills = new Dictionary<byte, int>();
newSkills[jobName] = jobSkills;
}
var newJobSkills = new Dictionary<byte, int>(jobSkills);
newJobSkills[skillKey] = level;
newSkills[jobName] = newJobSkills;
return new(this) { Skills = newSkills };
}
//WL-Changes-end
public HumanoidCharacterProfile WithJobPriority(ProtoId<JobPrototype> jobId, JobPriority priority)
{
var dictionary = new Dictionary<ProtoId<JobPrototype>, JobPriority>(_jobPriorities);
if (priority == JobPriority.Never)
{
dictionary.Remove(jobId);
}
else if (priority == JobPriority.High)
{
// There can only ever be one high priority job.
foreach (var (job, value) in dictionary)
{
if (value == JobPriority.High)
dictionary[job] = JobPriority.Medium;
}
dictionary[jobId] = priority;
}
else
{
dictionary[jobId] = priority;
}
return new(this)
{
_jobPriorities = dictionary,
};
}
public HumanoidCharacterProfile WithPreferenceUnavailable(PreferenceUnavailableMode mode)
{
return new(this) { PreferenceUnavailable = mode };
}
public HumanoidCharacterProfile WithAntagPreferences(IEnumerable<ProtoId<AntagPrototype>> antagPreferences)
{
return new(this)
{
_antagPreferences = new (antagPreferences),
};
}
public HumanoidCharacterProfile WithAntagPreference(ProtoId<AntagPrototype> antagId, bool pref)
{
var list = new HashSet<ProtoId<AntagPrototype>>(_antagPreferences);
if (pref)
{
list.Add(antagId);
}
else
{
list.Remove(antagId);
}
return new(this)
{
_antagPreferences = list,
};
}
public HumanoidCharacterProfile WithTraitPreference(ProtoId<TraitPrototype> traitId, IPrototypeManager protoManager)
{
// null category is assumed to be default.
if (!protoManager.TryIndex(traitId, out var traitProto))
return new(this);
var category = traitProto.Category;
// Category not found so dump it.
TraitCategoryPrototype? traitCategory = null;
if (category != null && !protoManager.Resolve(category, out traitCategory))
return new(this);
var list = new HashSet<ProtoId<TraitPrototype>>(_traitPreferences) { traitId };
if (traitCategory == null || traitCategory.MaxTraitPoints < 0)
{
return new(this)
{
_traitPreferences = list,
};
}
var count = 0;
foreach (var trait in list)
{
// If trait not found or another category don't count its points.
if (!protoManager.TryIndex<TraitPrototype>(trait, out var otherProto) ||
otherProto.Category != traitCategory)
{
continue;
}
count += otherProto.Cost;
}
if (count > traitCategory.MaxTraitPoints && traitProto.Cost != 0)
{
return new(this);
}
return new(this)
{
_traitPreferences = list,
};
}
public HumanoidCharacterProfile WithoutTraitPreference(ProtoId<TraitPrototype> traitId, IPrototypeManager protoManager)
{
var list = new HashSet<ProtoId<TraitPrototype>>(_traitPreferences);
list.Remove(traitId);
return new(this)
{
_traitPreferences = list,
};
}
public string Summary =>
Loc.GetString(
"humanoid-character-profile-summary",
("name", Name),
("gender", Gender.ToString().ToLowerInvariant()),
("age", Age)
);
public bool MemberwiseEquals(ICharacterProfile maybeOther)
{
if (maybeOther is not HumanoidCharacterProfile other) return false;
if (Name != other.Name) return false;
if (Age != other.Age) return false;
if (Height != other.Height) return false; // WL-Height
if (OocText != other.OocText) return false; // WL-OocText
if (Sex != other.Sex) return false;
if (FlavorText != other.FlavorText) return false; // WL-Changes
if (MedicalRecord != other.MedicalRecord) return false; // WL-Records
if (SecurityRecord != other.SecurityRecord) return false; // WL-Records
if (EmploymentRecord != other.EmploymentRecord) return false; // WL-Records
if (Gender != other.Gender) return false;
if (Species != other.Species) return false;
if (PreferenceUnavailable != other.PreferenceUnavailable) return false;
if (SpawnPriority != other.SpawnPriority) return false;
if (!_jobPriorities.SequenceEqual(other._jobPriorities)) return false;
if (!_antagPreferences.SequenceEqual(other._antagPreferences)) return false;
if (!_traitPreferences.SequenceEqual(other._traitPreferences)) return false;
if (!Loadouts.SequenceEqual(other.Loadouts)) return false;
if (!_jobSubnames.SequenceEqual(other._jobSubnames)) return false; // WL-JobSubnames
if (!_jobUnblockings.SequenceEqual(other._jobUnblockings)) return false; // WL-Changes
// WL-Skills-start
if (Skills.Count != other.Skills.Count) return false;
foreach (var kv in Skills)
{
if (!other.Skills.TryGetValue(kv.Key, out var otherJobSkills))
return false;
var jobSkills = kv.Value;
if (jobSkills.Count != otherJobSkills.Count)
return false;
foreach (var inner in jobSkills)
{
if (!otherJobSkills.TryGetValue(inner.Key, out var otherLevel) || otherLevel != inner.Value)
return false;
}
}
// WL-Skills-end
return Appearance.MemberwiseEquals(other.Appearance);
}
public void EnsureValid(ICommonSession session, IDependencyCollection collection, string[] sponsorPrototypes)
{
var configManager = collection.Resolve<IConfigurationManager>();
var prototypeManager = collection.Resolve<IPrototypeManager>();
if (!prototypeManager.TryIndex(Species, out var speciesPrototype) || speciesPrototype.RoundStart == false)
{
Species = SharedHumanoidAppearanceSystem.DefaultSpecies;
speciesPrototype = prototypeManager.Index(Species);
}
// Corvax-Sponsors-Start: Reset to human if player not sponsor
if (speciesPrototype.SponsorOnly && !sponsorPrototypes.Contains(Species.Id))
{
Species = SharedHumanoidAppearanceSystem.DefaultSpecies;
speciesPrototype = prototypeManager.Index(Species);
}
// Corvax-Sponsors-End
var sex = Sex switch
{
Sex.Male => Sex.Male,
Sex.Female => Sex.Female,
Sex.Unsexed => Sex.Unsexed,
_ => Sex.Male // Invalid enum values.
};
// ensure the species can be that sex and their age fits the founds
var height = Math.Clamp(Height, speciesPrototype.MinHeight, speciesPrototype.MaxHeight); // WL-Height
if (!speciesPrototype.Sexes.Contains(sex))
sex = speciesPrototype.Sexes[0];
var age = Math.Clamp(Age, speciesPrototype.MinAge, speciesPrototype.MaxAge);
var gender = Gender switch
{
Gender.Epicene => Gender.Epicene,
Gender.Female => Gender.Female,
Gender.Male => Gender.Male,
Gender.Neuter => Gender.Neuter,
_ => Gender.Epicene // Invalid enum values.
};
string name;
var maxNameLength = configManager.GetCVar(CCVars.MaxNameLength);
if (string.IsNullOrEmpty(Name))
{
name = GetName(Species, gender);
}
else if (Name.Length > maxNameLength)
{
name = Name[..maxNameLength];
}
else
{
name = Name;
}
name = name.Trim();
if (configManager.GetCVar(CCVars.RestrictedNames))
{
name = RestrictedNameRegex.Replace(name, string.Empty);
}
if (configManager.GetCVar(CCVars.ICNameCase))
{
// This regex replaces the first character of the first and last words of the name with their uppercase version
name = ICNameCaseRegex.Replace(name, m => m.Groups["word"].Value.ToUpper());
}
if (string.IsNullOrEmpty(name))
{
name = GetName(Species, gender);
}
string flavortext;
var maxFlavorTextLength = configManager.GetCVar(CCVars.MaxFlavorTextLength);
if (FlavorText.Length > maxFlavorTextLength)
{
flavortext = FormattedMessage.RemoveMarkupOrThrow(FlavorText)[..maxFlavorTextLength];
}
else
{
flavortext = FormattedMessage.RemoveMarkupOrThrow(FlavorText);
}
var appearance = HumanoidCharacterAppearance.EnsureValid(Appearance, Species, Sex, sponsorPrototypes);
var oocText = OocText.Length > MaxDescLength ? FormattedMessage.RemoveMarkup(OocText)[..MaxDescLength] : FormattedMessage.RemoveMarkup(OocText); // WL-OOCText
// WL-Records-Start
var medicalRecord = MedicalRecord.Length > MaxRecordLength
? FormattedMessage.RemoveMarkupOrThrow(MedicalRecord)[..MaxRecordLength]
: FormattedMessage.RemoveMarkupOrThrow(MedicalRecord);
var securityRecord = SecurityRecord.Length > MaxRecordLength
? FormattedMessage.RemoveMarkupOrThrow(SecurityRecord)[..MaxRecordLength]
: FormattedMessage.RemoveMarkupOrThrow(SecurityRecord);
var employmentRecord = EmploymentRecord.Length > MaxRecordLength
? FormattedMessage.RemoveMarkupOrThrow(EmploymentRecord)[..MaxRecordLength]
: FormattedMessage.RemoveMarkupOrThrow(EmploymentRecord);
// WL-Records-End
var prefsUnavailableMode = PreferenceUnavailable switch
{
PreferenceUnavailableMode.StayInLobby => PreferenceUnavailableMode.StayInLobby,
PreferenceUnavailableMode.SpawnAsOverflow => PreferenceUnavailableMode.SpawnAsOverflow,
_ => PreferenceUnavailableMode.StayInLobby // Invalid enum values.
};
var spawnPriority = SpawnPriority switch
{
SpawnPriorityPreference.None => SpawnPriorityPreference.None,
SpawnPriorityPreference.Arrivals => SpawnPriorityPreference.Arrivals,
SpawnPriorityPreference.Cryosleep => SpawnPriorityPreference.Cryosleep,
_ => SpawnPriorityPreference.None // Invalid enum values.
};
var priorities = new Dictionary<ProtoId<JobPrototype>, JobPriority>(JobPriorities
.Where(p => prototypeManager.TryIndex<JobPrototype>(p.Key, out var job) && job.SetPreference && p.Value switch
{
JobPriority.Never => false, // Drop never since that's assumed default.
JobPriority.Low => true,
JobPriority.Medium => true,
JobPriority.High => true,
_ => false
}));
var hasHighPrio = false;
foreach (var (key, value) in priorities)
{
if (value != JobPriority.High)
continue;
if (hasHighPrio)
priorities[key] = JobPriority.Medium;
hasHighPrio = true;
}
var antags = AntagPreferences
.Where(id => prototypeManager.TryIndex(id, out var antag) && antag.SetPreference)
.ToList();
var traits = TraitPreferences
.Where(prototypeManager.HasIndex)
.ToList();
// WL-Skills-Start
var validSkills = new Dictionary<string, Dictionary<byte, int>>();
foreach (var (jobName, jobSkillDict) in Skills)
{
var validJobSkills = new Dictionary<byte, int>();
foreach (var (skillByte, level) in jobSkillDict)
{
if (Enum.IsDefined(typeof(SkillType), skillByte))
{
var validLevel = Math.Clamp(level, 1, 4);
validJobSkills[skillByte] = validLevel;
}
}
foreach (SkillType skill in Enum.GetValues(typeof(SkillType)))
{
var skillByte = (byte)skill;
if (!validJobSkills.ContainsKey(skillByte))
{
validJobSkills[skillByte] = 1;
}
}
validSkills[jobName] = validJobSkills;
}
if (validSkills.Count == 0)
{
var defaultSkills = new Dictionary<byte, int>();
foreach (SkillType skill in Enum.GetValues(typeof(SkillType)))
{
defaultSkills[(byte)skill] = 1;
}
foreach (var job in _jobPriorities.Keys)
{
validSkills[job.Id] = new Dictionary<byte, int>(defaultSkills);
}
}
// WL-Skills-End
Name = name;
FlavorText = flavortext;
OocText = oocText; // WL-OOCText
MedicalRecord = medicalRecord; // WL-Records
SecurityRecord = securityRecord; // WL-Records
EmploymentRecord = employmentRecord; // WL-Records
Age = age;
Height = height; // WL-Height
Sex = sex;
Gender = gender;
Appearance = appearance;
SpawnPriority = spawnPriority;
Skills = validSkills; // WL-Skills
_jobPriorities.Clear();
foreach (var (job, priority) in priorities)
{
_jobPriorities.Add(job, priority);
}
PreferenceUnavailable = prefsUnavailableMode;
_antagPreferences.Clear();
_antagPreferences.UnionWith(antags);
_traitPreferences.Clear();
_traitPreferences.UnionWith(GetValidTraits(traits, prototypeManager));
// Corvax-TTS-Start
prototypeManager.TryIndex<TTSVoicePrototype>(Voice, out var voice);
if (voice is null || !CanHaveVoice(voice, Sex))
Voice = SharedHumanoidAppearanceSystem.DefaultSexVoice[sex];
// Corvax-TTS-End
// Checks prototypes exist for all loadouts and dump / set to default if not.
var toRemove = new ValueList<string>();
foreach (var (roleName, loadouts) in _loadouts)
{
if (!prototypeManager.HasIndex<RoleLoadoutPrototype>(roleName))
{
toRemove.Add(roleName);
continue;
}
// This happens after we verify the prototype exists
// These values are set equal in the database and we need to make sure they're equal here too!
loadouts.Role = roleName;
loadouts.EnsureValid(this, session, collection);
}
foreach (var value in toRemove)
{
_loadouts.Remove(value);
}
}
/// <summary>
/// Takes in an IEnumerable of traits and returns a List of the valid traits.
/// </summary>
public List<ProtoId<TraitPrototype>> GetValidTraits(IEnumerable<ProtoId<TraitPrototype>> traits, IPrototypeManager protoManager)
{
// Track points count for each group.
var groups = new Dictionary<string, int>();
var result = new List<ProtoId<TraitPrototype>>();
foreach (var trait in traits)
{
if (!protoManager.TryIndex(trait, out var traitProto))
continue;
// Always valid.
if (traitProto.Category == null)
{
result.Add(trait);
continue;
}
// No category so dump it.
if (!protoManager.Resolve(traitProto.Category, out var category))
continue;
var existing = groups.GetOrNew(category.ID);
existing += traitProto.Cost;
// Too expensive.
if (existing > category.MaxTraitPoints)
continue;
groups[category.ID] = existing;
result.Add(trait);
}
return result;
}
// Corvax-TTS-Start
// SHOULD BE NOT PUBLIC, BUT....
public static bool CanHaveVoice(TTSVoicePrototype voice, Sex sex)
{
return voice.RoundStart && sex == Sex.Unsexed || (voice.Sex == sex || voice.Sex == Sex.Unsexed);
}
// Corvax-TTS-End
public ICharacterProfile Validated(ICommonSession session, IDependencyCollection collection, string[] sponsorPrototypes)
{
var profile = new HumanoidCharacterProfile(this);
profile.EnsureValid(session, collection, sponsorPrototypes);
return profile;
}
// sorry this is kind of weird and duplicated,
/// working inside these non entity systems is a bit wack
public static string GetName(string species, Gender gender)
{
var namingSystem = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<NamingSystem>();
return namingSystem.GetName(species, gender);
}
public bool Equals(HumanoidCharacterProfile? other)
{
if (other is null)
return false;
return ReferenceEquals(this, other) || MemberwiseEquals(other);
}
public override bool Equals(object? obj)
{
return obj is HumanoidCharacterProfile other && Equals(other);
}
public override int GetHashCode()
{
var hashCode = new HashCode();
hashCode.Add(_jobPriorities);
hashCode.Add(_antagPreferences);
hashCode.Add(_traitPreferences);
hashCode.Add(_loadouts);
hashCode.Add(Name);
hashCode.Add(FlavorText);
hashCode.Add(Species);
hashCode.Add(Age);
hashCode.Add((int)Sex);
hashCode.Add((int)Gender);
hashCode.Add(Appearance);
hashCode.Add((int)SpawnPriority);
hashCode.Add((int)PreferenceUnavailable);
//WL-Changes-start
hashCode.Add(_jobSubnames);
hashCode.Add(_jobUnblockings);
hashCode.Add(Height);
hashCode.Add(OocText);
hashCode.Add(MedicalRecord);
hashCode.Add(SecurityRecord);
hashCode.Add(EmploymentRecord);
unchecked
{
var skillsHash = 0;
foreach (var jobKv in Skills.OrderBy(k => k.Key))
{
var innerHash = 0;
foreach (var sk in jobKv.Value.OrderBy(k => k.Key))
{
innerHash = HashCode.Combine(innerHash, sk.Key.GetHashCode(), sk.Value.GetHashCode());
}
skillsHash = HashCode.Combine(skillsHash, jobKv.Key.GetHashCode(), innerHash);
}
hashCode.Add(skillsHash);
}
//WL-Changes-end
return hashCode.ToHashCode();
}
public void SetLoadout(RoleLoadout loadout)
{
_loadouts[loadout.Role.Id] = loadout;
}
public HumanoidCharacterProfile WithLoadout(RoleLoadout loadout)
{
// Deep copies so we don't modify the DB profile.
var copied = new Dictionary<string, RoleLoadout>();
foreach (var proto in _loadouts)
{
if (proto.Key == loadout.Role)
continue;
copied[proto.Key] = proto.Value.Clone();
}
copied[loadout.Role] = loadout.Clone();
var profile = Clone();
profile._loadouts = copied;
return profile;
}
public RoleLoadout GetLoadoutOrDefault(string id, ICommonSession? session, ProtoId<SpeciesPrototype>? species, IEntityManager entManager, IPrototypeManager protoManager)
{
if (!_loadouts.TryGetValue(id, out var loadout))
{
loadout = new RoleLoadout(id);
loadout.SetDefault(this, session, protoManager, force: true);
}
loadout.SetDefault(this, session, protoManager);
return loadout;
}
public HumanoidCharacterProfile Clone()
{
return new HumanoidCharacterProfile(this);
}
}
}