using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Windows.Media;

namespace FingersDance.Data
{
    public class DataDictionary
    {
        Dictionary<int, List<Color>> _Data = new Dictionary<int, List<Color>>();

        public DataDictionary()
        {
            if (_Data == null)
                _Data = new Dictionary<int, List<Color>>();
        }

        public void AddAnnotation(Annotation annotation)
        {
            try
            {
                //  Recherche de l'élément qui précède le TC de début
                List<Color> temp = new List<Color>();
                int value = 0;
                for (int i = (int)annotation.TcBegin; i >= 0; i--)
                    if (_Data.ContainsKey(i))
                    {
                        value = i;
                        i = -1;
                    }

                //  Ajout des couleurs de l'élément qui précède
                if (!_Data.ContainsKey((int)annotation.TcBegin))
                {
                    if (_Data.ContainsKey(value))
                        foreach (Color elt in _Data[value])
                            temp.Add(elt);
                    _Data.Add((int)annotation.TcBegin, temp);
                }

                //  Ajout de la couleur à tous les éléments entre le début et la fin de l'annotation
                foreach (KeyValuePair<int, List<Color>> elt in _Data)
                    if (elt.Key >= (int)annotation.TcBegin && elt.Key < (int)(annotation.TcBegin + annotation.Dur) && !elt.Value.Contains(annotation.Color))
                        elt.Value.Add(annotation.Color);

                //  Vérification que début et fin correspondent à deux éléments distincts et ajout de la fin de l'annotation si nécessaire (tc de fin inexistant)
                if (((int)annotation.TcBegin) != ((int)(annotation.TcBegin + annotation.Dur)))
                {
                    temp = new List<Color>();
                    if (!_Data.ContainsKey((int)(annotation.TcBegin + annotation.Dur)))
                    {
                        for (int i = (int)(annotation.TcBegin + annotation.Dur); i >= 0; i--)
                            if (_Data.ContainsKey(i))
                            {
                                value = i;
                                i = -1;
                            }
                        if (_Data.ContainsKey(value))
                            foreach (Color elt in _Data[value])
                                temp.Add(elt);
                        temp.Remove(annotation.Color);
                        _Data.Add((int)(annotation.TcBegin + annotation.Dur), temp);
                    }
                }
                else if (!_Data.ContainsKey((int)(annotation.TcBegin + 1)))
                {
                    temp = new List<Color>();
                    foreach (Color elt in _Data[(int)annotation.TcBegin])
                        temp.Add(elt);
                    _Data.Add((int)(annotation.TcBegin + 1), temp);
                }
            }
            catch { }
        }

        public void RemoveAnnotation(Annotation annotation)
        {
            try
            {
                //  Suppression de la couleur à tous les éléments entre le début et la fin de l'annotation
                foreach (KeyValuePair<int, List<Color>> elt in _Data)
                    if (elt.Key >= (int)annotation.TcBegin && elt.Key < (int)(annotation.TcBegin + annotation.Dur) && elt.Value.Contains(annotation.Color))
                        elt.Value.Remove(annotation.Color);
            }
            catch { }
        }

        public List<Color> GetColors(int TC)
        {
            try
            {
                for (int i = TC; i >= 0; i--)
                    if (_Data.ContainsKey(i))
                        return _Data[i];
                return new List<Color>();
            }
            catch (Exception)
            {
                return new List<Color>();
            }
        }
    }
}