namespace Yw.WinFrmUI
{
    /// <summary>
    /// 水力工况信息ViewModel
    /// </summary>
    public class HydroWorkingInfoViewModel
    {
        /// <summary>
        /// 
        /// </summary>
        public HydroWorkingInfoViewModel()
        {
            this.Reservoirs = new List<HydroWorkingReservoirViewModel>();
            this.Tanks = new List<HydroWorkingTankViewModel>();
            this.Pumps = new List<HydroWorkingPumpViewModel>();
            this.Valves = new List<HydroWorkingValveViewModel>();
            this.Resistances = new List<HydroWorkingResistanceViewModel>();
        }

        /// <summary>
        /// 水库
        /// </summary>
        public List<HydroWorkingReservoirViewModel> Reservoirs { get; set; }

        /// <summary>
        /// 水池
        /// </summary>
        public List<HydroWorkingTankViewModel> Tanks { get; set; }

        /// <summary>
        /// 水泵
        /// </summary>
        public List<HydroWorkingPumpViewModel> Pumps { get; set; }

        /// <summary>
        /// 阀门
        /// </summary>
        public List<HydroWorkingValveViewModel> Valves { get; set; }

        /// <summary>
        /// 阻件
        /// </summary>
        public List<HydroWorkingResistanceViewModel> Resistances { get; set; }


        /// <summary>
        /// 获取所有工况列表
        /// </summary>
        public List<HydroWorkingVisualViewModel> GetAllWorkingList()
        {
            var list = new List<HydroWorkingVisualViewModel>();
            this.Reservoirs?.ForEach(x => list.Add(x));
            this.Tanks?.ForEach(x => list.Add(x));
            this.Pumps?.ForEach(x => list.Add(x));
            this.Valves?.ForEach(x => list.Add(x));
            return list;
        }

        /// <summary>
        /// 添加工况
        /// </summary>
        public bool AppendWorking(HydroWorkingVisualViewModel working)
        {
            if (working == null)
            {
                return false;
            }
            var type = working.GetType();
            if (type.FullName == typeof(HydroWorkingReservoirViewModel).FullName)
            {
                if (this.Reservoirs.Exists(x => x.Code == working.Code))
                {
                    return false;
                }
                this.Reservoirs.Add(working as HydroWorkingReservoirViewModel);
                return true;
            }
            else if (type.FullName == typeof(HydroWorkingTankViewModel).FullName)
            {
                if (this.Tanks.Exists(x => x.Code == working.Code))
                {
                    return false;
                }
                this.Tanks.Add(working as HydroWorkingTankViewModel);
                return true;
            }
            else if (type.FullName == typeof(HydroWorkingPumpViewModel).FullName)
            {
                if (this.Pumps.Exists(x => x.Code == working.Code))
                {
                    return false;
                }
                this.Pumps.Add(working as HydroWorkingPumpViewModel);
                return true;
            }
            else if (type.FullName == typeof(HydroWorkingValveViewModel).FullName)
            {
                if (this.Valves.Exists(x => x.Code == working.Code))
                {
                    return false;
                }
                this.Valves.Add(working as HydroWorkingValveViewModel);
                return true;
            }
            return false;
        }



    }
}