StateBase.cs

using MyCaffe.basecode;
using MyCaffe.gym;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.trainers
{
    public class StateBase
    {
        bool m_bDone = false;
        bool m_bValid = true;
        double m_dfReward = 0;
        int m_nActionCount = 0;
        SimpleDatum m_data = null;
        SimpleDatum m_clip = null;
        SimpleDatum m_label = null;
        double[] m_rgState = null;
        double[] m_rgClip = null;
        Image m_img = null;
        double m_dfTestingPercent = 0.2; // default to 20% of the tail end of the data for testing.
        List<DataPoint> m_rgHistory = null;
 
        public StateBase(int nActionCount)
        {
            m_nActionCount = nActionCount;
        }
 
        public double TestingPercent
        {
            get { return m_dfTestingPercent; }
            set { m_dfTestingPercent = value; }
        }
 
        public bool IsValid
        {
            get { return m_bValid; }
            set { m_bValid = value; }
        }
 
        public double Reward
        {
            get { return m_dfReward; }
            set { m_dfReward = value; }
        }
 
        public bool Done
        {
            get { return m_bDone; }
            set { m_bDone = value; }
        }
 
        public List<DataPoint> History
        {
            get { return m_rgHistory; }
            set { m_rgHistory = value; }
        }
 
        public int ActionCount
        {
            get { return m_nActionCount; }           
        }
 
        public SimpleDatum Data
        {
            get { return m_data; }
            set { m_data = value; }
        }
 
        public double[] RawState
        {
            get { return m_rgState; }
            set { m_rgState = value; }
        }
 
        public SimpleDatum Clip
        {
            get { return m_clip; }
            set { m_clip = value; }
        }
 
        public double[] RawClip
        {
            get { return m_rgClip; }
            set { m_rgClip = value; }
        }
 
        public SimpleDatum Label
        {
            get { return m_label; }
            set { m_label = value; }
        }
 
        public Image RawImage
        {
            get { return m_img; }
            set { m_img = value; }
        }
 
        public virtual StateBase Clone()
        {
            StateBase s = new StateBase(m_nActionCount);
 
            s.m_bDone = m_bDone;
            s.m_bValid = m_bValid;
            s.m_dfReward = m_dfReward;
            s.m_data = new SimpleDatum(m_data, true);
            s.m_rgState = Utility.Clone<double>(m_rgState);
 
            if (m_clip != null)
                s.m_clip = new SimpleDatum(m_clip, true);
 
            s.m_img = (m_img == null) ? null : new Bitmap(m_img);
 
            return s;
        }
 
        public override string ToString()
        {
            double[] rgData = m_data.GetData<double>();
 
            string str = "{";
            for (int i = 0; i < rgData.Length && i < 5; i++)
            {
                str += rgData[i].ToString("N4") + ",";
            }
            str = str.TrimEnd(',');
            str += "}";
 
            return "State = " + str + " Done = " + m_bDone.ToString();
        }
    }
}