#nullable enable
// Copyright (c) 2010-2013 AlphaSierraPapa for the SharpDevelop Team
//
// Permission is hereby granted, free of charge, to any person obtaining a copy of this
// software and associated documentation files (the "Software"), to deal in the Software
// without restriction, including without limitation the rights to use, copy, modify, merge,
// publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons
// to whom the Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all copies or
// substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
// INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
// FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
using System.Collections.Generic;
using System.Linq;
namespace ICSharpCode.Decompiler.Util
{
///
/// A dictionary that allows multiple pairs with the same key.
///
public class MultiDictionary : ILookup where TKey : notnull
{
readonly Dictionary> dict;
public MultiDictionary()
{
dict = new Dictionary>();
}
public MultiDictionary(IEqualityComparer? comparer)
{
dict = new Dictionary>(comparer);
}
public void Add(TKey key, TValue value)
{
if (!dict.TryGetValue(key, out List? valueList))
{
valueList = new List();
dict.Add(key, valueList);
}
valueList.Add(value);
}
public bool Remove(TKey key, TValue value)
{
if (dict.TryGetValue(key, out List? valueList))
{
if (valueList.Remove(value))
{
if (valueList.Count == 0)
dict.Remove(key);
return true;
}
}
return false;
}
///
/// Removes all entries with the specified key.
///
/// Returns true if at least one entry was removed.
public bool RemoveAll(TKey key)
{
return dict.Remove(key);
}
public void Clear()
{
dict.Clear();
}
public IReadOnlyList this[TKey key] {
get {
if (dict.TryGetValue(key, out var list))
return list;
else
return EmptyList.Instance;
}
}
///
/// Returns the number of different keys.
///
public int Count {
get { return dict.Count; }
}
public ICollection Keys {
get { return dict.Keys; }
}
public IEnumerable Values {
get { return dict.Values.SelectMany(list => list); }
}
IEnumerable ILookup.this[TKey key] {
get { return this[key]; }
}
public bool Contains(TKey key)
{
return dict.ContainsKey(key);
}
public IEnumerator> GetEnumerator()
{
foreach (var pair in dict)
yield return new Grouping(pair.Key, pair.Value);
}
System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
{
return GetEnumerator();
}
sealed class Grouping : IGrouping
{
readonly TKey key;
readonly List values;
public Grouping(TKey key, List values)
{
this.key = key;
this.values = values;
}
public TKey Key {
get { return key; }
}
public IEnumerator GetEnumerator()
{
return values.GetEnumerator();
}
System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
{
return values.GetEnumerator();
}
}
}
}