NaiveClass.java

import java.util.ArrayList;

/**
 * Die Klasse NaiveClass bildet die Naive Datenstruktur (eine ArrayList) ab.
 * Jedes Element ist eine Station
 */
public class NaiveClass {
    private ArrayList<Station> stations;
    private int[] counterArray;//Index 1: Airportcounter, Index 2: Trainstationcounter

    /**
     * Fügt die aus der Main eingelesenen Stations, in die Objektliste stations.
     *
     * @param stations
     */
    public void addPointsInList(ArrayList<Station> stations) {
        this.stations = stations;
    }

    /**
     * Gibt die Anzahl aller Airports und Trainstations aus, die sich innerhalb des radius,
     * ausgehend vom Punkt (x, y) befinden.
     *
     * @param x      X-Koordinate der Usereingabe
     * @param y      Y-Koordinate der Usereingabe
     * @param radius
     */
    public int[] getStations(double x, double y, double radius) {
        counterArray = new int[3];
        ArrayList<Station> stationsFromCoord = new ArrayList<Station>();

        for (Station s : stations) {
            if (s.getDistance(x, y) <= radius) { //verlgeicht ob distance vom Startpunkt zur nächsten Station im Radius liegt
                stationsFromCoord.add(s);
            }
        }

        for (Station s : stationsFromCoord) {
            if (s.getType() == TypeEnum.AIRPORT) {
                counterArray[1]++; //Airportcounter
            } else {
                counterArray[2]++; //Trainstationcounter
            }
        }

        return counterArray;
    }

    /**
     * Berechnet und gibt, die Anzahl aller Flughäfen,
     * in deren r-Längeneinheiten-Umkreis (radiusFromAirport)
     * sich mindestens n Bahnhöfe (trainstationsFromAirport) befinden, aus.
     *
     * @param trainstationsFromAirport
     * @param radiusFromAirport
     */
    public int getStationsFromAirport(int trainstationsFromAirport, double radiusFromAirport) {
        System.out.println("Calculating........");
        counterArray = new int[3];

        for (Station s : stations) { //Durch Stationsliste durchlaufen
            if (s.getType() == TypeEnum.AIRPORT) { //Nach Airports checken
                for (Station s1 : getStationsFromCoord(s.getX(), s.getY(), radiusFromAirport)) { //Für jeden Airport alle Stationen im Radius abspeichern und durch iterieren
                    if (s1.getType() == TypeEnum.TRAINSTATION) { // falls Station eine Trainstation, dann Trainstation counter eröhen
                        counterArray[2]++; //Trainstation Counter
                        if (counterArray[2] >= trainstationsFromAirport) { //Wenn trainstatiocounter mit der Usereingabe übereinstimmen airportcounter erhöhen
                            counterArray[1]++; //Airport counter
                            break; //wenn mind. anzahl erfüllt ist, brech die schleife ab um unnötige Vergleiche zu vermeiden und den counter unnötigen zu erhöhen
                        }
                    }
                }
                counterArray[2] = 0; //trainstationcounter wieder null setzen für den nächsten airport

            }
        }

       /* System.out.println("Airports with at least " + trainstationsFromAirport + " Trainstations less than " + radiusFromAirport + " away");
        System.out.println("  > " + counterArray[1]);*/
        return counterArray[1];
    }


    /**
     * Returnt alle Stations, die sich innerhalb des übergebenen Radius befinden
     * @param x
     * @param y
     * @param radius
     * @return
     */
    private ArrayList<Station> getStationsFromCoord(double x, double y, double radius){
        ArrayList tempList = new ArrayList();

        for(Station s: stations){
            if(s.getDistance(x,y) <= radius){ //verlgeicht ob distance vom Startpunkt zur nächsten Station im Radius liegt
                tempList.add(s);
            }
        }

        return tempList;
    }

}