SliceParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
using MyCaffe.common;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class SliceParameter : LayerParameterBase
    {
        int m_nAxis = 1;
        List<uint> m_rgSlicePoint = new List<uint>();
        uint m_nSliceDim = 0;
 
        public SliceParameter()
        {
        }
 
        [Description("Specifies the axis along which to slice - may be negative to index from the end (e.g., -1 for the last axis).")]
        public int axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        [Description("Specifies the optional slice points which indicate the indexes in the selected dimensions (the number of indices must be equal to the number of top blobs minus one).")]
        public List<uint> slice_point
        {
            get { return m_rgSlicePoint; }
            set { m_rgSlicePoint = value; }
        }
 
        [Description("DEPRECIATED - use 'axis' instead.")]
        [Browsable(false)]
        public uint slice_dim
        {
            get { return m_nSliceDim; }
            set { m_nSliceDim = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            SliceParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            SliceParameter p = (SliceParameter)src;
 
            m_nAxis = p.m_nAxis;
            m_rgSlicePoint = Utility.Clone<uint>(p.m_rgSlicePoint);
            m_nSliceDim = p.m_nSliceDim;
        }
 
        public override LayerParameterBase Clone()
        {
            SliceParameter p = new SliceParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("axis", axis.ToString());
            rgChildren.Add<uint>("slice_point", slice_point);
 
            if (slice_dim != 0)
                rgChildren.Add("slice_dim", slice_dim.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static SliceParameter FromProto(RawProto rp)
        {
            string strVal;
            SliceParameter p = new SliceParameter();
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            p.slice_point = rp.FindArray<uint>("slice_point");
 
            if ((strVal = rp.FindValue("slice_dim")) != null)
                p.slice_dim = uint.Parse(strVal);
 
            return p;
        }
    }
}