EmitConstraint.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using MyCaffe.basecode;
 
namespace MyCaffe.param.ssd
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class EmitConstraint : OptionalParameter
    {
        EmitType m_emitType = EmitType.CENTER;
        float m_fEmitOverlap = 0;
 
        public enum EmitType
        {
            CENTER = 0,
            MIN_OVERLAP = 1
        }
 
        public EmitConstraint(bool bActive) : base(bActive)
        {
        }
 
        [Description("Get/set the emit type.")]
        public EmitType emit_type
        {
            get { return m_emitType; }
            set { m_emitType = value; }
        }
 
        [Description("Get/set the emit overlap used with MIN_OVERLAP.")]
        public float emit_overlap
        {
            get { return m_fEmitOverlap; }
            set { m_fEmitOverlap = value; }
        }
 
        public override void Copy(OptionalParameter src)
        {
            base.Copy(src);
 
            if (src is EmitConstraint)
            {
                EmitConstraint p = (EmitConstraint)src;
                m_emitType = p.emit_type;
                m_fEmitOverlap = p.emit_overlap;
            }
        }
 
        public EmitConstraint Clone()
        {
            EmitConstraint p = new param.ssd.EmitConstraint(Active);
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProto rpBase = base.ToProto("option");
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(rpBase);
            rgChildren.Add(new RawProto("emit_type", m_emitType.ToString()));
            rgChildren.Add(new RawProto("emit_overlap", m_fEmitOverlap.ToString()));
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static new EmitConstraint FromProto(RawProto rp)
        {
            EmitConstraint p = new EmitConstraint(false);
            string strVal;
 
            RawProto rpOption = rp.FindChild("option");
            if (rpOption != null)
                ((OptionalParameter)p).Copy(OptionalParameter.FromProto(rpOption));
 
            if ((strVal = rp.FindValue("emit_type")) != null)
            {
                switch (strVal)
                {
                    case "CENTER":
                        p.emit_type = EmitType.CENTER;
                        break;
 
                    case "MIN_OVERLAP":
                        p.emit_type = EmitType.MIN_OVERLAP;
                        break;
 
                    default:
                        throw new Exception("Unknown emit_type '" + strVal + "'!");
                }
            }
 
            if ((strVal = rp.FindValue("emit_overlap")) != null)
                p.emit_overlap = BaseParameter.ParseFloat(strVal);
 
            return p;
        }
    }
}