ImageSetBase.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Drawing;
using MyCaffe.basecode;
using System.Threading;
using System.Diagnostics;
using MyCaffe.basecode.descriptors;
 
namespace MyCaffe.db.image
{
    public class ImageSetBase : IDisposable
    {
        protected DatasetFactory m_factory;
        protected SourceDescriptor m_src;
 
        public ImageSetBase(DatasetFactory factory, SourceDescriptor src)
        {
            m_src = new SourceDescriptor(src);
            m_factory = new DatasetFactory(factory);
            m_factory.Open(src.ID);
        }
 
        protected virtual void Dispose(bool bDisposing)
        {
            if (m_factory != null)
            {
                m_factory.Close();
                m_factory.Dispose();
                m_factory = null;
            }
        }
 
        public void Dispose()
        {
            Dispose(true);
        }
 
        public SourceDescriptor Source
        {
            get { return m_src; }
        }
 
        public void SetLabelMapping(LabelMapping map)
        {
            m_factory.SetLabelMapping(map, m_src.ID);
        }
 
        public void UpdateLabelMapping(int nNewLabel, List<int> rgOriginalLabels)
        {
            m_factory.UpdateLabelMapping(nNewLabel, rgOriginalLabels, m_src.ID);
        }
 
        public void ResetLabels(int nProjectId)
        {
            m_factory.ResetLabels(nProjectId, m_src.ID);
        }
 
        public void DeleteLabelBoosts(int nProjectId)
        {
            m_factory.DeleteLabelBoosts(nProjectId, m_src.ID);
        }
 
        public void AddLabelBoost(int nProjectId, int nLabel, double dfBoost)
        {
            m_factory.AddLabelBoost(nProjectId, nLabel, dfBoost, m_src.ID);
        }
 
        public string GetLabelBoostsAsText(int nProjectId)
        {
            return m_factory.GetLabelBoostsAsText(nProjectId, m_src.ID);
        }
 
        public Dictionary<int, int> LoadLabelCounts()
        {
            return m_factory.LoadLabelCounts(m_src.ID);
        }
 
        public void UpdateLabelCounts(int nProjectId)
        {
            m_factory.UpdateLabelCounts(m_src.ID, nProjectId);
        }
 
        public string GetLabelCountsAsText()
        {
            return m_factory.GetLabelCountsAsText(m_src.ID);
        }
    }
}