Labelset.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Drawing;
using MyCaffe.basecode;
using MyCaffe.basecode.descriptors;
 
namespace MyCaffe.db.image
{
    public class LabelSet : IDisposable
    {
        CryptoRandom m_random = null;
        LabelDescriptor m_label;
        SimpleDatum[] m_rgImages;
        List<int> m_rgIdx = new List<int>();
        int m_nCurrentIdx = 0;
 
        public LabelSet(LabelDescriptor lbl, CryptoRandom random)
        {
            m_random = random;
            m_label = lbl;
            m_rgImages = new SimpleDatum[lbl.ImageCount];
        }
 
        public LabelDescriptor Label
        {
            get { return m_label; }
            set { m_label = value; }
        }
 
        public int Count
        {
            get { return m_rgImages.Length; }
        }
 
        public void Clear()
        {
            m_rgImages = new SimpleDatum[m_rgImages.Length];
            m_nCurrentIdx = 0;
        }
 
        public SimpleDatum this[int nIdx]
        {
            get { return m_rgImages[nIdx]; }
            set { m_rgImages[nIdx] = value; }
        }
 
        public bool IsLoaded
        {
            get
            {
                if (m_nCurrentIdx == m_rgImages.Length)
                    return true;
                return false;
            }
        }
 
        public void Add(SimpleDatum s)
        {
            if (m_nCurrentIdx < m_rgImages.Length)
            {
                m_rgImages[m_nCurrentIdx] = s;
                m_nCurrentIdx++;
            }
        }
 
        public SimpleDatum GetImage(int nIdx, DB_ITEM_SELECTION_METHOD selectionMethod)
        {
            if ((selectionMethod & DB_ITEM_SELECTION_METHOD.RANDOM) != DB_ITEM_SELECTION_METHOD.RANDOM)
                throw new ArgumentException("Label balancing image selection only supports the RANDOM and RANDOM+BOOST selection methods");
 
            if (m_nCurrentIdx == 0)
                return null;
 
            int nLastIdx = 0;
            int nFixedIdx = -1;
            int nImageIdx = 0;
 
            return GetImage(m_rgImages, m_rgIdx, m_nCurrentIdx, nIdx, m_random, selectionMethod, ref nLastIdx, ref nFixedIdx, out nImageIdx);
        }
 
        public static SimpleDatum GetImage(SimpleDatum[] rgImages, List<int> rgIdx, int nCount, int nIdx, CryptoRandom random, DB_ITEM_SELECTION_METHOD selectionMethod, ref int nLastIndex, ref int nFixedIndex, out int nImageIdx)
        {
            if ((selectionMethod & DB_ITEM_SELECTION_METHOD.BOOST) == DB_ITEM_SELECTION_METHOD.BOOST)
            {
                IEnumerable<SimpleDatum> iQuery = rgImages.Where(p => p != null && p.Boost > 0);
                List<SimpleDatum> rgItems = new List<SimpleDatum>(iQuery);               
 
                if (rgItems.Count > 0)
                {
                    if (rgIdx.Count > rgItems.Count)
                        rgIdx.Clear();
 
                    if (rgIdx.Count == 0)
                    {
                        for (int i = 0; i < rgItems.Count; i++)
                        {
                            rgIdx.Add(i);
                        }
                    }
 
                    if ((selectionMethod & DB_ITEM_SELECTION_METHOD.RANDOM) == DB_ITEM_SELECTION_METHOD.RANDOM)
                    {
                        nIdx = rgIdx[random.Next(rgIdx.Count)];
                        rgIdx.Remove(nIdx);
                    }
 
                    SimpleDatum sd = rgItems[nIdx];
                    nImageIdx = nIdx;
 
                    return sd;
                }
            }
 
            int nMin = ((selectionMethod & DB_ITEM_SELECTION_METHOD.PAIR) == DB_ITEM_SELECTION_METHOD.PAIR) ? 2 : 1;
            if (rgIdx.Count < nMin)
            {
                rgIdx.Clear();
 
                for (int i = 0; i < rgImages.Length; i++)
                {
                    rgIdx.Add(i);
                }
            }
 
            if ((selectionMethod & DB_ITEM_SELECTION_METHOD.PAIR) == DB_ITEM_SELECTION_METHOD.PAIR)
            {
                nIdx = nLastIndex + 1;
 
                if (nIdx == rgIdx.Count)
                    nIdx = 0;
 
                rgIdx.Remove(nIdx);
            }
            else if ((selectionMethod & DB_ITEM_SELECTION_METHOD.RANDOM) == DB_ITEM_SELECTION_METHOD.RANDOM)
            {
                nIdx = rgIdx[random.Next(rgIdx.Count)];
                rgIdx.Remove(nIdx);
            }
            else if (selectionMethod == DB_ITEM_SELECTION_METHOD.FIXEDINDEX)
            {
                nFixedIndex = nIdx;
            }
            else if ((selectionMethod & DB_ITEM_SELECTION_METHOD.CLEARFIXEDINDEX) == DB_ITEM_SELECTION_METHOD.CLEARFIXEDINDEX)
            {
                nFixedIndex = -1;
            }
 
            if (nFixedIndex >= 0)
                nIdx = nFixedIndex;
 
            if (nIdx >= rgImages.Length)
                nIdx = nIdx % rgImages.Length;
 
            nLastIndex = nIdx;
            nImageIdx = nIdx;
 
            return rgImages[nIdx];
        }
 
        public void Unload()
        {
            for (int i = 0; i < m_rgImages.Length; i++)
            {
                m_rgImages[i] = null;
            }
        }
 
        protected virtual void Dispose(bool bDisposing)
        {
            if (m_random != null)
            {
                m_random.Dispose();
                m_random = null;
            }
        }
 
        public void Dispose()
        {
            Dispose(true);
        }
    }
}