DatasetCollection.cs

using MyCaffe.basecode;
using MyCaffe.basecode.descriptors;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.db.temporal
{
    public class DatasetCollection : IDisposable
    {
        Dictionary<int, DataSet> m_rgDatasets = new Dictionary<int, DataSet>();
        Dictionary<string, int> m_rgDatasetIDs = new Dictionary<string, int>();
        Dictionary<int, int> m_rgSourceIDtoDatasetID = new Dictionary<int, int>();
 
        public DatasetCollection()
        {
        }
 
        public void Dispose()
        {
            CleanUp(0);
        }
 
        public void CleanUp(int nDsID)
        {
            if (nDsID == 0)
            {
                foreach (KeyValuePair<int, DataSet> kv in m_rgDatasets)
                {
                    kv.Value.CleanUp();
                }
            }
            else
            {
                if (m_rgDatasets.ContainsKey(nDsID))
                {
                    m_rgDatasets[nDsID].CleanUp();
                    m_rgDatasets.Remove(nDsID);
                }
            }
        }
 
        public DataSet Add(DatasetDescriptor dsd, Log log)
        {
            DataSet ds = Find(dsd.ID);
            if (ds != null)
                return ds;
 
            ds = new DataSet(dsd, log);
            Add(ds);
 
            if (!m_rgSourceIDtoDatasetID.ContainsKey(ds.Dataset.TrainingSource.ID))
                m_rgSourceIDtoDatasetID.Add(ds.Dataset.TrainingSource.ID, ds.Dataset.ID);
            if (!m_rgSourceIDtoDatasetID.ContainsKey(ds.Dataset.TestingSource.ID))
                m_rgSourceIDtoDatasetID.Add(ds.Dataset.TestingSource.ID, ds.Dataset.ID);
 
            return ds;
        }
 
        public void Add(DataSet ds)
        {
            m_rgDatasets.Add(ds.Dataset.ID, ds);
 
            if (!m_rgDatasetIDs.ContainsKey(ds.Dataset.Name))
                m_rgDatasetIDs.Add(ds.Dataset.Name, ds.Dataset.ID);
 
            if (!m_rgSourceIDtoDatasetID.ContainsKey(ds.Dataset.TrainingSource.ID))
                m_rgSourceIDtoDatasetID.Add(ds.Dataset.TrainingSource.ID, ds.Dataset.ID);
            if (!m_rgSourceIDtoDatasetID.ContainsKey(ds.Dataset.TestingSource.ID))
                m_rgSourceIDtoDatasetID.Add(ds.Dataset.TestingSource.ID, ds.Dataset.ID);
        }
 
        public DataSet Find(int nDatasetID)
        {
            if (!m_rgDatasets.ContainsKey(nDatasetID))
                return null;
 
            return m_rgDatasets[nDatasetID];
        }
 
        public DataSet Find(string strDs)
        {
            int nDsID = GetDatasetID(strDs);
            if (nDsID == 0)
                return null;
 
            return Find(nDsID);
        }
 
        public SourceDescriptor FindSourceByID(int nSrcID)
        {
            foreach (KeyValuePair<int, DataSet> kvp in m_rgDatasets)
            {
                SourceDescriptor src = kvp.Value.Dataset.TrainingSource;
                if (src.ID == nSrcID)
                    return src;
 
                src = kvp.Value.Dataset.TestingSource;
                if (src.ID == nSrcID)
                    return src;
            }
 
            return null;
        }
 
        public SourceDescriptor FindSourceByName(string strSrc)
        {
            foreach (KeyValuePair<int, DataSet> kvp in m_rgDatasets)
            {
                SourceDescriptor src = kvp.Value.Dataset.TrainingSource;
                if (src.Name == strSrc)
                    return src;
 
                src = kvp.Value.Dataset.TestingSource;
                if (src.Name == strSrc)
                    return src;
            }
 
            return null;
        }
 
        public TemporalSet FindTemporalSetBySourceID(int nSrcID)
        {
            if (!m_rgSourceIDtoDatasetID.ContainsKey(nSrcID))
                return null;
 
            int nDsID = m_rgSourceIDtoDatasetID[nSrcID];
 
            DataSet ds = Find(nDsID);
            if (ds == null)
                return null;
 
            return ds.GetTemporalSetBySourceID(nSrcID);
        }
 
        public int GetDatasetID(string strDs)
        {
            if (!m_rgDatasetIDs.ContainsKey(strDs))
                return 0;
 
            return m_rgDatasetIDs[strDs];
        }
    }
}