ConstantParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using System.ComponentModel;
using System.Configuration;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class ConstantParameter : LayerParameterBase 
    {
        BlobShape m_outputShape = new BlobShape();
        List<float> m_rgF = new List<float>();
        string m_strBinaryDataFile = null;
 
        public ConstantParameter()
        {
        }
 
        [Description("Specifies the output shape.")]
        public BlobShape output_shape
        {
            get { return m_outputShape; }
            set { m_outputShape = value; }
        }
 
        [Description("Specifies a binary data file containing the values to load.")]
        public string binary_data_file
        {
            get { return m_strBinaryDataFile; }
            set { m_strBinaryDataFile = value; }
        }
 
        [Description("Specifies a set of float values used to fill the output.  When only one item is specified, all outputs are set to that value.")]
        public List<float> values_f
        {
            get { return m_rgF; }
            set { m_rgF = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            ConstantParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            ConstantParameter p = (ConstantParameter)src;
            m_outputShape = p.output_shape.Clone();
            m_rgF = new List<float>(p.m_rgF);
            m_strBinaryDataFile = p.binary_data_file;
        }
 
        public override LayerParameterBase Clone()
        {
            ConstantParameter p = new ConstantParameter();
            p.Copy(this);
            return p;
        }
 
        private static string replace(string str, char chTarget, char chReplace)
        {
            string strOut = "";
 
            foreach (char ch in str)
            {
                if (ch == chTarget)
                    strOut += chReplace;
                else
                    strOut += ch;
            }
 
            return strOut;
        }
 
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(m_outputShape.ToProto("output_shape"));
            rgChildren.Add<float>("valuef", m_rgF);
 
            if (!string.IsNullOrEmpty(m_strBinaryDataFile))
                rgChildren.Add("binary_data_file", replace(m_strBinaryDataFile, ' ', '~'));
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static ConstantParameter FromProto(RawProto rp)
        {
            string strVal;
            ConstantParameter p = new ConstantParameter();
 
            RawProto shape = rp.FindChild("output_shape");
            if (shape != null)
                p.m_outputShape = BlobShape.FromProto(shape);
 
            p.m_rgF = rp.FindArray<float>("valuef");
 
            strVal = rp.FindValue("binary_data_file");
            if (strVal != null)
                p.m_strBinaryDataFile = replace(strVal, '~', ' ');
 
            return p;
        }
    }
}