BlobShape.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.IO;
using MyCaffe.basecode;
using MyCaffe.common;
 
namespace MyCaffe.param
{
    public class BlobShape : BaseParameter, ICloneable, IComparable, IBinaryPersist  
    {
        List<int> m_rgDim = new List<int>();
 
        public BlobShape()
        {
        }
 
        public BlobShape(int nNum, int nC, int nH, int nW)
        {
            m_rgDim = new List<int>() { nNum, nC, nH, nW };
        }
 
        public BlobShape(List<int> rgShape)
        {
            m_rgDim = new List<int>();
 
            for (int i = 0; i < rgShape.Count; i++)
            {
                m_rgDim.Add(rgShape[i]);
            }
        }
 
        public void Save(BinaryWriter bw)
        {
            Utility.Save<int>(bw, m_rgDim);
        }
 
        public object Load(BinaryReader br, bool bNewInstance)
        {
            BlobShape b = this;
            
            if (bNewInstance)
                b = new BlobShape();
 
            b.m_rgDim = Utility.Load<int>(br);
 
            return b;
        }
 
        public static BlobShape Load(BinaryReader br)
        {
            BlobShape b = new BlobShape();
            return (BlobShape)b.Load(br, true);
        }
 
        public List<int> dim
        {
            get { return m_rgDim; }
            set { m_rgDim = new List<int>(value); }
        }
 
        public BlobShape Clone()
        {
            BlobShape bs = new BlobShape();
 
            bs.m_rgDim = Utility.Clone<int>(m_rgDim);
 
            return bs;
        }
 
        object ICloneable.Clone()
        {
            return Clone();
        }
 
        public bool Compare(BlobShape bs)
        {
            return Utility.Compare<int>(m_rgDim, bs.m_rgDim);
        }
 
        public int CompareTo(object obj)
        {
            BlobShape bs = obj as BlobShape;
 
            if (bs == null)
                return 1;
 
            if (!Compare(bs))
                return 1;
 
            return 0;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add<int>("dim", m_rgDim);
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static BlobShape FromProto(RawProto rp)
        {
            return new BlobShape(rp.FindArray<int>("dim"));
        }
 
        public override string ToString()
        {
            string strOut = "{";
 
            for (int i = 0; i < m_rgDim.Count; i++)
            {
                strOut += m_rgDim[i].ToString();
                strOut += ",";
            }
 
            strOut = strOut.TrimEnd(',');
            strOut += "}";
 
            return strOut;
        }
    }
}