package ProjectElectricLocomotive;

import java.util.ArrayList;
import java.util.Dictionary;
import java.util.HashMap;
import java.util.List;
import java.util.stream.Collectors;

public class LocomotivesGenericStorage {
    final HashMap<String, LocomotiveGenericCollection<DrawingLocomotive, DrawingObjectLocomotive>> _locomotiveStorage;
    public List<String> Keys()
    {
        return _locomotiveStorage.keySet().stream().toList();
    }
    private final int _pictureWidth;
    private final int _pictureHeight;
    public LocomotivesGenericStorage(int pictureWidth, int pictureHeight) {
        _locomotiveStorage = new HashMap<>();
        _pictureWidth = pictureWidth;
        _pictureHeight = pictureHeight;
    }

    /// <summary>
    /// Добавление набора
    /// </summary>
    /// <param name="name">Название набора</param>
    public void AddSet(String name)
    {
        if (!(_locomotiveStorage.containsKey(name)))
        {
            _locomotiveStorage.put(name, new LocomotiveGenericCollection<DrawingLocomotive, DrawingObjectLocomotive>(_pictureWidth, _pictureHeight));
        }
    }

    /// <summary>
    /// Удаление набора
    /// </summary>
    /// <param name="name">Название набора</param>
    public void DelSet(String name)
    {
        if (_locomotiveStorage.keySet().contains(name))
        {
            _locomotiveStorage.remove(name);
        }
    }


    /// <summary>
    /// Доступ к набору
    /// </summary>
    /// <param name="ind"></param>
    /// <returns></returns>
    public LocomotiveGenericCollection<DrawingLocomotive, DrawingObjectLocomotive> get(String ind)
    {
            if (_locomotiveStorage.keySet().contains(ind))
            {
                return _locomotiveStorage.get(ind);
            }
            return null;
    }

    public DrawingObjectLocomotive get(String name, int ind)
    {
        if (_locomotiveStorage.keySet().contains(ind))
        {
            return _locomotiveStorage.get(name).GetU(ind);
        }
        return null;
    }
}