using ProjectLainer.MovementStrategy;
using ProjectLainer.DrawningObjects;
using System.Runtime.CompilerServices;

namespace ProjectLainer.Generics
{
    internal class LainersGenericCollection<T, U>
    where T : DrawingLainer
    where U : IMoveableObject
    {
        
        private readonly int _pictureWidth;
        
        private readonly int _pictureHeight;
        
        private readonly int _placeSizeWidth = 210;
       
        private readonly int _placeSizeHeight = 90;
        
        public readonly SetGeneric<T> _collection;
        
        public LainersGenericCollection(int picWidth, int picHeight)
        {
            int width = picWidth / _placeSizeWidth;
            int height = picHeight / _placeSizeHeight;
            _pictureWidth = picWidth;
            _pictureHeight = picHeight;
            _collection = new SetGeneric<T>(width * height);
        }
        
        public static bool operator +(LainersGenericCollection<T, U> collect, T? obj)
        {
            if (obj != null && collect != null)
            {
                var value = collect._collection.Insert(obj);
                if(value <= collect._collection.Count && value >= 0)
                {
                    return true;
                }
                else { return false; }
            }
            return false;
            
        }
        public static bool? operator -(LainersGenericCollection<T, U> collect, int
        pos)
        {
            T? obj = collect?._collection.Get(pos);
            if (obj != null && collect != null)
            {
                return collect._collection.Remove(pos);
            }
            else
            {
                return null;
            }
        } 
        
        public U? GetU(int pos)
        {
            return (U?)_collection.Get(pos)?.GetMoveableObject;
        }
        
        public Bitmap ShowLainers()
        {
            Bitmap bmp = new(_pictureWidth, _pictureHeight);
            Graphics gr = Graphics.FromImage(bmp);
            DrawBackground(gr);
            DrawObjects(gr);
            return bmp;
        }
       
        private void DrawBackground(Graphics g)
        {
            Pen pen = new(Color.Black, 3);
            for (int i = 0; i < _pictureWidth / _placeSizeWidth; i++)
            {
                for (int j = 0; j < _pictureHeight / _placeSizeHeight +
                1; ++j)
                {
                    g.DrawLine(pen, i * _placeSizeWidth, j *
                    _placeSizeHeight, i * _placeSizeWidth + _placeSizeWidth / 2, j *
                    _placeSizeHeight);
                }
                g.DrawLine(pen, i * _placeSizeWidth, 0, i *
                _placeSizeWidth, _pictureHeight / _placeSizeHeight * _placeSizeHeight);
            }
        }
        private void DrawObjects(Graphics g)
        {
            
            for (int i = 0; i < _collection.Count; i++)
            {
                DrawingLainer? lainer = _collection.Get(i);
                if (lainer != null)
                {
                    int inRow = _pictureWidth / _placeSizeWidth;
                    
                    lainer.SetPosition((inRow - 1 - (i % inRow)) * _placeSizeWidth, i / inRow * _placeSizeHeight);
                    lainer.DrawTransport(g);
                }
            }
        }
    }
}