DB.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.db.image;
using MyCaffe.basecode;
using MyCaffe.basecode.descriptors;
 
namespace MyCaffe.data
{
    public class DB<T>
    {
        IXImageDatabaseBase m_db = null;
        int m_nSrcID = 0;
        string m_strSrc = null;
 
        public DB(IXImageDatabaseBase imgDb)
        {
            m_db = imgDb;
        }
 
        public void Open(string strSrc)
        {
            m_strSrc = strSrc;
            m_nSrcID = m_db.GetSourceID(strSrc);
        }
 
        public void Close()
        {
            m_nSrcID = 0;
        }
 
        public Cursor<T> NewCursor(DataTransformer<T> transformer, Log log = null)
        {
            return new Cursor<T>(m_db, transformer, m_strSrc, log);
        }
    }
 
    public class Cursor<T> 
    {
        Log m_log = null;
        string m_strSrc;
        IXImageDatabaseBase m_db;
        DataTransformer<T> m_transformer;
        int m_nSrcID = 0;
        int m_nCount = 0;
        int m_nIdx = 0;
 
        public Cursor(IXImageDatabaseBase db, DataTransformer<T> transformer, string strSrc, Log log = null)
        {
            m_log = log;
            m_db = db;
            m_transformer = transformer;
            SourceDescriptor src = m_db.GetSourceByName(strSrc);
            m_strSrc = src.Name;
            m_nSrcID = src.ID;
            m_nCount = src.ImageCount;
        }
 
        public void Reset()
        {
            m_nIdx = 0;
        }
 
        public void Next()
        {
            m_nIdx++;
        }
 
        public int Count
        {
            get { return m_nCount; }
        }
 
        public bool IsValid
        {
            get
            {
                if (m_nIdx >= m_nCount)
                    return false;
 
                return true;
            }
        }
 
        public void SeekToFirst()
        {
            m_nIdx = 0;
        }
 
        public SimpleDatum GetValue(int? nLabel = null, bool bLoadDataCriteria = false, DB_ITEM_SELECTION_METHOD? imgSel = null)
        {
            SimpleDatum sd = m_db.QueryItem(m_nSrcID, m_nIdx, null, imgSel, nLabel, bLoadDataCriteria, false);
 
            if (m_log != null)
                m_log.WriteLine(m_strSrc + ": Idx = " + sd.Index.ToString() + " Label = " + sd.Label.ToString());
 
            m_transformer.TransformLabel(sd);
 
            return sd;
        }
    }
}