ATITD-Tools/Desert-Paint-Codex Files · Models/ReactionSet.cs

Files @ d63ade4d8489

Branch filter:

Location: ATITD-Tools/Desert-Paint-Codex/Models/ReactionSet.cs

d63ade4d8489 1.5 KiB text/x-csharp Show Annotation Show as Raw Download as Raw

Jason Maltzen

Recipe generation no longer incorrectly skips some recipes when a catalyst is included. This should result in some less expensive receips, and more total recipes tested.

using System.Collections.Generic;

namespace DesertPaintCodex.Models
{
    public class ReactionSet
    {
        // ingredient -> [ingredient, reaction]
        private readonly Dictionary<string, Dictionary<string, Reaction?>> _reactions = new();

        public Reaction? Find(Reagent reagent1, Reagent reagent2)
        {
            Reaction? reaction = null;
            _reactions.TryGetValue(reagent1.PracticalPaintName, out Dictionary<string, Reaction?>? secondReagentDict);
            secondReagentDict?.TryGetValue(reagent2.PracticalPaintName, out reaction);
            return reaction;
        }

        public void Set(Reagent reagent1, Reagent reagent2, Reaction? reaction)
        {
            _reactions.TryGetValue(reagent1.PracticalPaintName, out Dictionary<string, Reaction?>? secondReagentDict);
            if (secondReagentDict == null)
            {
                secondReagentDict = new Dictionary<string, Reaction?>();
                _reactions.Add(reagent1.PracticalPaintName, secondReagentDict);
            }
            secondReagentDict[reagent2.PracticalPaintName] = reaction;
        }

        public void Remove(Reagent reagent1, Reagent reagent2)
        {
            _reactions.TryGetValue(reagent1.PracticalPaintName, out Dictionary<string, Reaction?>? secondReagentDict);
            secondReagentDict?.Remove(reagent2.PracticalPaintName);
        }

        public void Clear()
        {
            _reactions.Clear();
        }
    }
}