using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using GeometryTasks;
using System.Drawing;

namespace GeometryPainting {
    public static class SegmentExtension {
        //Словарь цветов с ключом по отрезку
        public static Dictionary<Segment, Color> Colors = new Dictionary<Segment, Color>();
        //Если для отрезка есть цвет, то возвращает его, иначе возвращает черный
        public static Color GetColor(this Segment segment) {
            if (Colors.ContainsKey(segment)) {
                return Colors[segment];
            }
            else {
                return Color.Black;
            }
        }
        //Если у отрезка был цвет, то изменяет значение по ключу, иначе добавляет новый элемент словаря
        public static void SetColor(this Segment segment, Color color) {
            if (Colors.ContainsKey(segment)) {
                Colors[segment] = color;
            }
            else {
                Colors.Add(segment, color);
            }
        }
    }
}