namespace GeometryTD.Definition
{
    public class RemoveRandomCompsEffect : EventEffectBase
    {
        public override EventEffectType EffectType => EventEffectType.RemoveRandomComps;
        public override EventEffectParam Param => _param;
        private RemoveRandomCompsParam _param;

        public RemoveRandomCompsEffect(RemoveRandomCompsParam param, float? probability = null)
        {
            _param = param;
            Probability = probability;
        }
    }

    public class RemoveRandomCompsParam : EventEffectParam
    {
        public int Count;
        public RarityType Rarity;

        public RemoveRandomCompsParam(int Count, RarityType Rarity)
        {
            this.Count = Count;
            this.Rarity = Rarity;
        }
    }
}