LabelBBox.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.basecode
{
    public class LabelBBox
    {
        DictionaryMap<List<NormalizedBBox>> m_rgItems = new DictionaryMap<List<NormalizedBBox>>(null);
 
        public LabelBBox()
        {
        }
 
        public List<KeyValuePair<int, List<NormalizedBBox>>> ToList()
        {
            return m_rgItems.Map.ToList();
        }
 
        public void Add(int nLabel, NormalizedBBox bbox)
        {
            if (m_rgItems[nLabel] == null)
                m_rgItems[nLabel] = new List<NormalizedBBox>();
 
            m_rgItems[nLabel].Add(bbox);
        }
 
        public int Count
        {
            get { return m_rgItems.Count; }
        }
 
        public bool Contains(int nLabel)
        {
            return m_rgItems.Map.ContainsKey(nLabel);
        }
 
        public List<NormalizedBBox> this[int nLabel]
        {
            get
            {
                if (m_rgItems[nLabel] == null)
                    m_rgItems[nLabel] = new List<NormalizedBBox>();
 
                return m_rgItems[nLabel];
            }
            set
            {
                m_rgItems[nLabel] = value;
            }
        }
 
        public void Clear()
        {
            m_rgItems.Clear();
        }
    }
}