EngineParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class EngineParameter : LayerParameterBase 
    {
        Engine m_engine = Engine.DEFAULT;
 
        public enum Engine
        {
            DEFAULT = 0,
            CAFFE = 1,
            CUDNN = 2
        }
 
        public EngineParameter()
        {
        }
 
        [Description("Specifies the engine to use 'CAFFE' or CUDNN.  In most instances CUDNN is the default.")]
        public Engine engine
        {
            get { return m_engine; }
            set { m_engine = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            EngineParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
 
        public override void Copy(LayerParameterBase src)
        {
            EngineParameter p = (EngineParameter)src;
            m_engine = p.m_engine;
        }
 
        public override LayerParameterBase Clone()
        {
            EngineParameter p = new EngineParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            if (engine != Engine.DEFAULT)
                rgChildren.Add("engine", engine.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static EngineParameter FromProto(RawProto rp)
        {
            string strVal;
            EngineParameter p = new EngineParameter();
 
            if ((strVal = rp.FindValue("engine")) != null)
            {
                switch (strVal)
                {
                    case "DEFAULT":
                        p.engine = Engine.DEFAULT;
                        break;
 
                    case "CAFFE":
                        p.engine = Engine.CAFFE;
                        break;
 
                    case "CUDNN":
                        p.engine = Engine.CUDNN;
                        break;
 
                    default:
                        throw new Exception("Unknown 'engine' value: " + strVal);
                }
            }
 
            return p;
        }
    }
}