juni/crystall-punk-14
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added new HTN operations and preconditions (#27486)

Browse Source 
* Added new HTN operations & preconditions

* Ok I forgot about partial

* Namespace pierce the skies

* Some fixes, debug and new operators

* Bruh git eat my files
This commit is contained in:
Tornado Tech
2024-05-08 19:18:03 +10:00
committed by GitHub
parent 29860a0cf7
commit 31491775e5
14 changed files with 401 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Content.Server/NPC/HTN/Preconditions/GunAmmoPrecondition.cs
View File

@@ -35,7 +35,7 @@ public sealed partial class GunAmmoPrecondition : HTNPrecondition
else
percent = ammoEv.Count / (float) ammoEv.Capacity;
percent = Math.Clamp(percent, 0f, 1f);
percent = System.Math.Clamp(percent, 0f, 1f);
if (MaxPercent < percent)
return false;

12
Content.Server/NPC/HTN/Preconditions/KeyNotExistsPrecondition.cs Normal file
View File

@@ -0,0 +1,12 @@
namespace Content.Server.NPC.HTN.Preconditions;
public sealed partial class KeyNotExistsPrecondition : HTNPrecondition
{
[DataField(required: true)]
public string Key = string.Empty;
public override bool IsMet(NPCBlackboard blackboard)
{
return !blackboard.ContainsKey(Key);
}
}

23
Content.Server/NPC/HTN/Preconditions/Math/KeyBoolEqualsPrecondition.cs Normal file
View File

@@ -0,0 +1,23 @@
namespace Content.Server.NPC.HTN.Preconditions.Math;
/// <summary>
/// Checks for the presence of data in the blackboard and makes a comparison with the specified boolean
/// </summary>
public sealed partial class KeyBoolEqualsPrecondition : HTNPrecondition
{
[Dependency] private readonly IEntityManager _entManager = default!;
[DataField(required: true)]
public string Key = string.Empty;
[DataField(required: true)]
public bool Value;
public override bool IsMet(NPCBlackboard blackboard)
{
if (!blackboard.TryGetValue<bool>(Key, out var value, _entManager))
return false;
return Value == value;
}
}

18
Content.Server/NPC/HTN/Preconditions/Math/KeyFloatEqualsPrecondition.cs Normal file
View File

@@ -0,0 +1,18 @@
namespace Content.Server.NPC.HTN.Preconditions.Math;
public sealed partial class KeyFloatEqualsPrecondition : HTNPrecondition
{
[Dependency] private readonly IEntityManager _entManager = default!;
[DataField(required: true)]
public string Key = string.Empty;
[DataField(required: true)]
public float Value;
public override bool IsMet(NPCBlackboard blackboard)
{
return blackboard.TryGetValue<float>(Key, out var value, _entManager) &&
MathHelper.CloseTo(value, value);
}
}

17
Content.Server/NPC/HTN/Preconditions/Math/KeyFloatGreaterPrecondition.cs Normal file
View File

@@ -0,0 +1,17 @@
namespace Content.Server.NPC.HTN.Preconditions.Math;
public sealed partial class KeyFloatGreaterPrecondition : HTNPrecondition
{
[Dependency] private readonly IEntityManager _entManager = default!;
[DataField(required: true)]
public string Key = string.Empty;
[DataField(required: true)]
public float Value;
public override bool IsMet(NPCBlackboard blackboard)
{
return blackboard.TryGetValue<float>(Key, out var value, _entManager) && value > Value;
}
}

17
Content.Server/NPC/HTN/Preconditions/Math/KeyFloatLessPrecondition.cs Normal file
View File

@@ -0,0 +1,17 @@
namespace Content.Server.NPC.HTN.Preconditions.Math;
public sealed partial class KeyFloatLessPrecondition : HTNPrecondition
{
[Dependency] private readonly IEntityManager _entManager = default!;
[DataField(required: true)]
public string Key = string.Empty;
[DataField(required: true)]
public float Value;
public override bool IsMet(NPCBlackboard blackboard)
{
return blackboard.TryGetValue<float>(Key, out var value, _entManager) && value < Value;
}
}

33
Content.Server/NPC/HTN/PrimitiveTasks/Operators/Math/AddFloatOperator.cs Normal file
View File

@@ -0,0 +1,33 @@
using System.Threading;
using System.Threading.Tasks;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Math;
/// <summary>
/// Gets the key, and adds the value to that float
/// </summary>
public sealed partial class AddFloatOperator : HTNOperator
{
[Dependency] private readonly IEntityManager _entManager = default!;
[DataField(required: true)]
public string TargetKey = string.Empty;
[DataField, ViewVariables(VVAccess.ReadWrite)]
public float Amount;
public override async Task<(bool Valid, Dictionary<string, object>? Effects)> Plan(NPCBlackboard blackboard,
CancellationToken cancelToken)
{
if (!blackboard.TryGetValue<float>(TargetKey, out var value, _entManager))
return (false, null);
return (
true,
new Dictionary<string, object>
{
{ TargetKey, value + Amount }
}
);
}
}

28
Content.Server/NPC/HTN/PrimitiveTasks/Operators/Math/SetBoolOperator.cs Normal file
View File

@@ -0,0 +1,28 @@
using System.Threading;
using System.Threading.Tasks;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Math;
/// <summary>
/// Just sets a blackboard key to a bool
/// </summary>
public sealed partial class SetBoolOperator : HTNOperator
{
[DataField(required: true)]
public string TargetKey = string.Empty;
[DataField, ViewVariables(VVAccess.ReadWrite)]
public bool Value;
public override async Task<(bool Valid, Dictionary<string, object>? Effects)> Plan(NPCBlackboard blackboard,
CancellationToken cancelToken)
{
return (
true,
new Dictionary<string, object>
{
{ TargetKey, Value }
}
);
}
}

18
Content.Server/NPC/HTN/PrimitiveTasks/Operators/SetFloatOperator.cs → Content.Server/NPC/HTN/PrimitiveTasks/Operators/Math/SetFloatOperator.cs
View File

@@ -1,24 +1,28 @@
using System.Threading;
using System.Threading.Tasks;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Math;
/// <summary>
/// Just sets a blackboard key to a float
/// </summary>
public sealed partial class SetFloatOperator : HTNOperator
{
[DataField("targetKey", required: true)] public string TargetKey = string.Empty;
[DataField(required: true)]
public string TargetKey = string.Empty;
[ViewVariables(VVAccess.ReadWrite), DataField("amount")]
[DataField, ViewVariables(VVAccess.ReadWrite)]
public float Amount;
public override async Task<(bool Valid, Dictionary<string, object>? Effects)> Plan(NPCBlackboard blackboard,
CancellationToken cancelToken)
{
return (true, new Dictionary<string, object>()
{
{TargetKey, Amount},
});
return (
true,
new Dictionary<string, object>
{
{ TargetKey, Amount }
}
);
}
}

34
Content.Server/NPC/HTN/PrimitiveTasks/Operators/Math/SetRandomFloatOperator.cs Normal file
View File

@@ -0,0 +1,34 @@
using System.Threading;
using System.Threading.Tasks;
using Robust.Shared.Random;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Math;
/// <summary>
/// Sets a random float from MinAmount to MaxAmount to blackboard
/// </summary>
public sealed partial class SetRandomFloatOperator : HTNOperator
{
[Dependency] private readonly IRobustRandom _random = default!;
[DataField(required: true)]
public string TargetKey = string.Empty;
[DataField, ViewVariables(VVAccess.ReadWrite)]
public float MaxAmount = 1f;
[DataField, ViewVariables(VVAccess.ReadWrite)]
public float MinAmount = 0f;
public override async Task<(bool Valid, Dictionary<string, object>? Effects)> Plan(NPCBlackboard blackboard,
CancellationToken cancelToken)
{
return (
true,
new Dictionary<string, object>
{
{ TargetKey, _random.NextFloat(MinAmount, MaxAmount) }
}
);
}
}

28
Content.Server/NPC/HTN/PrimitiveTasks/Operators/PlaySoundOperator.cs Normal file
View File

@@ -0,0 +1,28 @@
using Robust.Server.Audio;
using Robust.Shared.Audio;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators;
public sealed partial class PlaySoundOperator : HTNOperator
{
private AudioSystem _audio = default!;
[DataField(required: true)]
public SoundSpecifier? Sound;
public override void Initialize(IEntitySystemManager sysManager)
{
base.Initialize(sysManager);
_audio = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<AudioSystem>();
}
public override HTNOperatorStatus Update(NPCBlackboard blackboard, float frameTime)
{
var uid = blackboard.GetValue<EntityUid>(NPCBlackboard.Owner);
_audio.PlayPvs(Sound, uid);
return base.Update(blackboard, frameTime);
}
}

36
Content.Server/NPC/HTN/PrimitiveTasks/Operators/SayKeyOperator.cs Normal file
View File

@@ -0,0 +1,36 @@
using Content.Server.Chat.Systems;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators;
public sealed partial class SayKeyOperator : HTNOperator
{
[Dependency] private readonly IEntityManager _entManager = default!;
private ChatSystem _chat = default!;
[DataField(required: true)]
public string Key = string.Empty;
/// <summary>
/// Whether to hide message from chat window and logs.
/// </summary>
[DataField]
public bool Hidden;
public override void Initialize(IEntitySystemManager sysManager)
{
base.Initialize(sysManager);
_chat = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<ChatSystem>();
}
public override HTNOperatorStatus Update(NPCBlackboard blackboard, float frameTime)
{
if (!blackboard.TryGetValue<object>(Key, out var value, _entManager))
return HTNOperatorStatus.Failed;
var speaker = blackboard.GetValue<EntityUid>(NPCBlackboard.Owner);
_chat.TrySendInGameICMessage(speaker, value.ToString() ?? "Oh no...", InGameICChatType.Speak, hideChat: Hidden, hideLog: Hidden);
return base.Update(blackboard, frameTime);
}
}