109 lines
3.1 KiB
C#
109 lines
3.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using HG;
|
|
using UnityEngine;
|
|
|
|
namespace RoR2.UI;
|
|
|
|
[RequireComponent(typeof(PointViewer))]
|
|
public class SniperTargetViewer : MonoBehaviour
|
|
{
|
|
public GameObject visualizerPrefab;
|
|
|
|
private PointViewer pointViewer;
|
|
|
|
private HUD hud;
|
|
|
|
private Dictionary<UnityObjectWrapperKey<HurtBox>, GameObject> hurtBoxToVisualizer = new Dictionary<UnityObjectWrapperKey<HurtBox>, GameObject>();
|
|
|
|
private List<HurtBox> displayedTargets = new List<HurtBox>();
|
|
|
|
private List<HurtBox> previousDisplayedTargets = new List<HurtBox>();
|
|
|
|
private void Awake()
|
|
{
|
|
pointViewer = GetComponent<PointViewer>();
|
|
OnTransformParentChanged();
|
|
}
|
|
|
|
private void OnTransformParentChanged()
|
|
{
|
|
hud = GetComponentInParent<HUD>();
|
|
}
|
|
|
|
private void OnDisable()
|
|
{
|
|
SetDisplayedTargets(Array.Empty<HurtBox>());
|
|
hurtBoxToVisualizer.Clear();
|
|
}
|
|
|
|
private void Update()
|
|
{
|
|
List<HurtBox> list = CollectionPool<HurtBox, List<HurtBox>>.RentCollection();
|
|
if ((bool)hud && (bool)hud.targetMaster)
|
|
{
|
|
TeamIndex teamIndex = hud.targetMaster.teamIndex;
|
|
IReadOnlyList<HurtBox> readOnlySniperTargetsList = HurtBox.readOnlySniperTargetsList;
|
|
int i = 0;
|
|
for (int count = readOnlySniperTargetsList.Count; i < count; i++)
|
|
{
|
|
HurtBox hurtBox = readOnlySniperTargetsList[i];
|
|
if ((bool)hurtBox.healthComponent && hurtBox.healthComponent.alive && FriendlyFireManager.ShouldDirectHitProceed(hurtBox.healthComponent, teamIndex) && (object)hurtBox.healthComponent.body != hud.targetMaster.GetBody())
|
|
{
|
|
list.Add(hurtBox);
|
|
}
|
|
}
|
|
}
|
|
SetDisplayedTargets(list);
|
|
list = CollectionPool<HurtBox, List<HurtBox>>.ReturnCollection(list);
|
|
}
|
|
|
|
private void OnTargetDiscovered(HurtBox hurtBox)
|
|
{
|
|
if (!hurtBoxToVisualizer.ContainsKey(hurtBox))
|
|
{
|
|
GameObject value = pointViewer.AddElement(new PointViewer.AddElementRequest
|
|
{
|
|
elementPrefab = visualizerPrefab,
|
|
target = hurtBox.transform,
|
|
targetWorldVerticalOffset = 0f,
|
|
targetWorldRadius = HurtBox.sniperTargetRadius,
|
|
scaleWithDistance = true
|
|
});
|
|
hurtBoxToVisualizer.Add(hurtBox, value);
|
|
}
|
|
else
|
|
{
|
|
Debug.LogWarning($"Already discovered hurtbox: {hurtBox}");
|
|
}
|
|
}
|
|
|
|
private void OnTargetLost(HurtBox hurtBox)
|
|
{
|
|
if (hurtBoxToVisualizer.TryGetValue(hurtBox, out var value))
|
|
{
|
|
pointViewer.RemoveElement(value);
|
|
}
|
|
}
|
|
|
|
private void SetDisplayedTargets(IReadOnlyList<HurtBox> newDisplayedTargets)
|
|
{
|
|
Util.Swap(ref displayedTargets, ref previousDisplayedTargets);
|
|
displayedTargets.Clear();
|
|
ListUtils.AddRange(displayedTargets, newDisplayedTargets);
|
|
List<HurtBox> list = CollectionPool<HurtBox, List<HurtBox>>.RentCollection();
|
|
List<HurtBox> list2 = CollectionPool<HurtBox, List<HurtBox>>.RentCollection();
|
|
ListUtils.FindExclusiveEntriesByReference(displayedTargets, previousDisplayedTargets, list, list2);
|
|
foreach (HurtBox item in list2)
|
|
{
|
|
OnTargetLost(item);
|
|
}
|
|
foreach (HurtBox item2 in list)
|
|
{
|
|
OnTargetDiscovered(item2);
|
|
}
|
|
list2 = CollectionPool<HurtBox, List<HurtBox>>.ReturnCollection(list2);
|
|
list = CollectionPool<HurtBox, List<HurtBox>>.ReturnCollection(list);
|
|
}
|
|
}
|