package ProjectStormtrooper;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

public class PlanesGenericStorage {
    final HashMap<String, PlanesGenericCollection<DrawingPlane, DrawingObjectPlane>> _planeStorages;

    public ArrayList<String> Keys() {
        return (ArrayList<String>) _planeStorages.keySet().stream().toList();
    }

    private final int _pictureWidth;
    private final int _pictureHeight;

    public PlanesGenericStorage(int pictureWidth, int pictureHeight) {
        _planeStorages = new HashMap<>();
        _pictureWidth = pictureWidth;
        _pictureHeight = pictureHeight;
    }

    public void AddSet(String name) {
        _planeStorages.put(name, new PlanesGenericCollection<>(_pictureWidth, _pictureHeight));
    }

    public void DelSet(String name) {
        _planeStorages.remove(name);
    }

    public PlanesGenericCollection<DrawingPlane, DrawingObjectPlane> Get(String ind) {
        if (_planeStorages.containsKey(ind))
            return _planeStorages.get(ind);
        return null;
    }
}