Add missing dictionary

This commit is contained in:
smoogipoo 2020-11-13 17:26:46 +09:00
parent 942ecde0e1
commit 688a442fb3
1 changed files with 4 additions and 0 deletions

View File

@ -154,6 +154,8 @@ public virtual bool Remove(DrawableHitObject h)
return false;
}
private readonly Dictionary<HitObject, HitObjectLifetimeEntry> lifetimeEntryMap = new Dictionary<HitObject, HitObjectLifetimeEntry>();
/// <summary>
/// Adds a <see cref="HitObjectLifetimeEntry"/> for a pooled <see cref="HitObject"/> to this <see cref="Playfield"/>.
/// </summary>
@ -161,6 +163,7 @@ public virtual bool Remove(DrawableHitObject h)
public virtual void Add(HitObjectLifetimeEntry entry)
{
HitObjectContainer.Add(entry);
lifetimeEntryMap[entry.HitObject] = entry;
OnHitObjectAdded(entry.HitObject);
}
@ -173,6 +176,7 @@ public virtual bool Remove(HitObjectLifetimeEntry entry)
{
if (HitObjectContainer.Remove(entry))
{
lifetimeEntryMap.Remove(entry.HitObject);
OnHitObjectRemoved(entry.HitObject);
return true;
}