using SWRIS.Models;
using System.Collections.ObjectModel;
using System.Windows;
using System.Windows.Controls;

namespace SWRIS.Selectors
{
    public class EquipmentsLayoutTemplateSelector : DataTemplateSelector
    {
        public DataTemplate FourTemplate { get; set; }
        public DataTemplate ThreeTemplate { get; set; }
        public DataTemplate TwoTemplate { get; set; }
        public DataTemplate OneTemplate { get; set; }

        public override DataTemplate SelectTemplate(object item, DependencyObject container)
        {
            if (item is ObservableCollection<EquipmentDataModel> equipments)
            {
                switch (equipments.Count)
                {
                    case 1:
                        return OneTemplate;
                    case 2:
                        return TwoTemplate;
                    case 3:
                        return ThreeTemplate;
                    case 4:
                        return FourTemplate;
                }
            }
            return base.SelectTemplate(item, container);
        }
    }
}