VideoDataParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.ssd
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class VideoDataParameter : LayerParameterBase 
    {
        VideoType m_videoType = VideoType.WEBCAM;
        int m_nDeviceID = 0;
        string m_strVideoFile;
        uint m_nSkipFrames = 0;
        uint m_nVideoWidth = 400;
        uint m_nVideoHeight = 300;
 
        public enum VideoType
        {
            WEBCAM = 0,
            VIDEO = 1
        }
 
        public VideoDataParameter()
        {
        }
 
        [Description("Specifies the video type (default = WEBCAM).")]
        public VideoType video_type
        {
            get { return m_videoType; }
            set { m_videoType = value; }
        }
 
        [Description("Specifies the device ID (default = 0).")]
        public int device_id
        {
            get { return m_nDeviceID; }
            set { m_nDeviceID = value; }
        }
 
        [Description("Specifies the video file when using the VIDEO type.")]
        public string video_file
        {
            get { return m_strVideoFile; }
            set { m_strVideoFile = value; }
        }
 
        [Description("Optionally, specifies the number of frames to be skipped before processing a frame (default = 0).")]
        public uint skip_frames
        {
            get { return m_nSkipFrames; }
            set { m_nSkipFrames = value; }
        }
 
        [Description("Optionally, specifies the video width (default = 400).")]
        public uint video_width
        {
            get { return m_nVideoWidth; }
            set { m_nVideoWidth = value; }
        }
 
        [Description("Optionally, specifies the video height (default = 300).")]
        public uint video_height
        {
            get { return m_nVideoHeight; }
            set { m_nVideoHeight = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            VideoDataParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            VideoDataParameter p = (VideoDataParameter)src;
 
            m_videoType = p.m_videoType;
            m_nDeviceID = p.m_nDeviceID;
            m_strVideoFile = p.m_strVideoFile;
            m_nSkipFrames = p.m_nSkipFrames;
            m_nVideoWidth = p.m_nVideoWidth;
            m_nVideoHeight = p.m_nVideoHeight;
        }
 
        public override LayerParameterBase Clone()
        {
            VideoDataParameter p = new VideoDataParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(new RawProto("video_type", m_videoType.ToString()));
            rgChildren.Add(new RawProto("device_id", m_nDeviceID.ToString()));
 
            if (!string.IsNullOrEmpty(m_strVideoFile))
                rgChildren.Add(new RawProto("video_file", m_strVideoFile.ToString()));
 
            rgChildren.Add(new RawProto("skip_frames", m_nSkipFrames.ToString()));
            rgChildren.Add(new RawProto("video_width", m_nVideoWidth.ToString()));
            rgChildren.Add(new RawProto("video_height", m_nVideoHeight.ToString()));
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static VideoDataParameter FromProto(RawProto rp)
        {
            VideoDataParameter p = new VideoDataParameter();
            string strVal;
 
            if ((strVal = rp.FindValue("video_type")) != null)
            {
                if (strVal == VideoType.VIDEO.ToString())
                    p.video_type = VideoType.VIDEO;
                else
                    p.video_type = VideoType.WEBCAM;
            }
 
            if ((strVal = rp.FindValue("device_id")) != null)
                p.device_id = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("video_file")) != null)
                p.video_file = strVal;
 
            if ((strVal = rp.FindValue("skip_frames")) != null)
                p.skip_frames = uint.Parse(strVal);
 
            if ((strVal = rp.FindValue("video_width")) != null)
                p.video_width = uint.Parse(strVal);
 
            if ((strVal = rp.FindValue("video_height")) != null)
                p.video_height = uint.Parse(strVal);
 
            return p;
        }
    }
}