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

public class Airport {
    private String location;
    private ArrayList<Runway> runways;
    private ArrayList<Gate> gates;
    private HashMap<Airplane, Runway> airplanes;

    public Airport(String location) {
        this.location = location;
        this.runways = new ArrayList<Runway>();
        this.gates = new ArrayList<Gate>();
        this.airplanes = new HashMap<>();
    }

    public void addRunway(Runway runway) {
        runways.add(runway);
    }

    public void removeRunway(Runway runway) {
        runways.remove(runway);
    }

    public ArrayList<Runway> getRunways() {
        return runways;
    }

    public HashMap<Airplane, Runway> getAirplanes() {
        return airplanes;
    }

    public void addAirplanes(Airplane airplane, Runway runway) {
        this.airplanes.put(airplane, runway);
    }

    public void removeAirplanes(Airplane airplane, Runway runway) {
        this.airplanes.remove(airplane, runway);
    }

    public Runway getRunwayForAirplane(Airplane airplane) {
        return airplanes.get(airplane);
    }
}