Multidict.cs

   1 using System;
   2 using System.Collections.Generic;
   3 using System.Text;
   4
   5 namespace Gitrdone
   6 {
   7     public class Multidict<KeyType, ValueType> : System.Collections.IEnumerable
   8     {
   9         protected Dictionary<KeyType, List<ValueType>> m_dict = new Dictionary<KeyType, List<ValueType>>();
  10
  11         public Multidict()
  12         {
  13         }
  14
  15         public List<ValueType> Get(KeyType key)
  16         {
  17             List<ValueType> items;
  18             if (m_dict.TryGetValue(key, out items))
  19             {
  20                 return items;
  21             }
  22             return null;
  23         }
  24
  25         public bool Add(KeyType key, ValueType value)
  26         {
  27             return Add(key, value, false);
  28         }
  29
  30         public bool Add(KeyType key, ValueType value, bool bAllowDups)
  31         {
  32             List<ValueType> items;
  33             if (!m_dict.TryGetValue(key, out items))
  34             {
  35                 items = new List<ValueType>();
  36                 m_dict[key] = items;
  37             }
  38             if (!items.Contains(value) || bAllowDups)
  39             {
  40                 items.Add(value);
  41                 return true;
  42             }
  43             return false;
  44         }
  45
  46         public System.Collections.IEnumerator GetEnumerator()
  47         {
  48             return m_dict.GetEnumerator();
  49         }
  50     }
  51 }