LabelTracker.cs

using System.Collections.Generic;
using Diz.Core.Interfaces;
using Diz.LogWriter.util;

namespace Diz.LogWriter;

public class LabelTracker
{

    private readonly ILogCreatorForGenerator logCreator;
    public ILogCreatorDataSource<IData> Data => logCreator?.Data;
    public List<int> VisitedLabelSnesAddresses { get; } = new();
        
    private Dictionary<int, IReadOnlyLabel> cachedUnvisitedLabels;
        
    public LabelTracker(ILogCreatorForGenerator logCreator)
    {
        this.logCreator = logCreator;
    }

    public Dictionary<int, IReadOnlyLabel> UnvisitedLabels
    {
        get
        {
            CacheUnvisitedLabels();
            return cachedUnvisitedLabels;
        }
    }

    public void OnLabelVisited(int snesAddress)
    {
        VisitedLabelSnesAddresses.Add(snesAddress);
        cachedUnvisitedLabels = null;
    }

    private void CacheUnvisitedLabels()
    {
        if (cachedUnvisitedLabels != null) // pretty sure this is the right thing to do?
            return;
            
        cachedUnvisitedLabels = new Dictionary<int, IReadOnlyLabel>();
        if (VisitedLabelSnesAddresses == null)
            return;

        foreach (var (snesAddress, label) in Data.Labels.Labels)
        {
            if (VisitedLabelSnesAddresses.Contains(snesAddress))
                continue;

            cachedUnvisitedLabels.Add(snesAddress, label);
        }
    }
}