Joonas Rikkonen
66 lines
2.8 KiB
C#
66 lines
2.8 KiB
C#
using System.Linq;
|
|
using System.Collections.Generic;
|
|
using Barotrauma.Extensions;
|
|
using Microsoft.Xna.Framework;
|
|
|
|
namespace Barotrauma
|
|
{
|
|
// TODO: use objective loop and sort the targets by severity
|
|
class AIObjectiveExtinguishFires : AIObjective
|
|
{
|
|
public override string DebugTag => "extinguish fires";
|
|
public override bool ForceRun => true;
|
|
public override bool KeepDivingGearOn => true;
|
|
|
|
private Dictionary<Hull, AIObjectiveExtinguishFire> extinguishObjectives = new Dictionary<Hull, AIObjectiveExtinguishFire>();
|
|
|
|
public AIObjectiveExtinguishFires(Character character, float priorityModifier = 1) : base(character, "", priorityModifier) { }
|
|
|
|
public override float GetPriority(AIObjectiveManager objectiveManager)
|
|
{
|
|
if (character.Submarine == null) { return 0; }
|
|
int fireCount = character.Submarine.GetHulls(true).Sum(h => h.FireSources.Count);
|
|
if (objectiveManager.CurrentOrder == this && fireCount > 0)
|
|
{
|
|
return AIObjectiveManager.OrderPriority;
|
|
}
|
|
float basePriority = MathHelper.Clamp(Priority, 0, 10);
|
|
float maxMultiplier = MathHelper.Min(PriorityModifier, 1);
|
|
float max = MathHelper.Min((AIObjectiveManager.OrderPriority + 20) * maxMultiplier, 90);
|
|
return MathHelper.Clamp(basePriority + fireCount * 20, 0, max);
|
|
}
|
|
|
|
public override bool IsCompleted() => false;
|
|
public override bool CanBeCompleted => true;
|
|
|
|
public override bool IsDuplicate(AIObjective otherObjective)
|
|
{
|
|
return otherObjective is AIObjectiveExtinguishFires;
|
|
}
|
|
|
|
protected override void Act(float deltaTime)
|
|
{
|
|
SyncRemovedObjectives(extinguishObjectives, Hull.hullList);
|
|
if (character.Submarine == null) { return; }
|
|
foreach (Hull hull in Hull.hullList)
|
|
{
|
|
if (hull.FireSources.None()) { continue; }
|
|
if (hull.Submarine == null) { continue; }
|
|
if (hull.Submarine.TeamID != character.TeamID) { continue; }
|
|
// If the character is inside, only take connected hulls into account.
|
|
if (character.Submarine != null && !character.Submarine.IsEntityFoundOnThisSub(hull, true)) { continue; }
|
|
if (!extinguishObjectives.TryGetValue(hull, out AIObjectiveExtinguishFire objective))
|
|
{
|
|
objective = new AIObjectiveExtinguishFire(character, hull);
|
|
extinguishObjectives.Add(hull, objective);
|
|
AddSubObjective(objective);
|
|
}
|
|
}
|
|
if (extinguishObjectives.None())
|
|
{
|
|
character?.Speak(TextManager.Get("DialogNoFire"), null, 3.0f, "nofire", 30.0f);
|
|
}
|
|
}
|
|
}
|
|
}
|