initial commit

classes.py

@@ -0,0 +1,85 @@
+from enum import Enum
+import math
+
+
+class ParsingMode(Enum):
+    LOCATIONS = 1
+    FLIGHTSCHEDULE = 2
+    FINDCONNECTION = 3
+
+
+class LocationType(Enum):
+    LOCATION = 1
+    HALTESTELLE = 2
+    FLUGHAFEN = 3
+
+
+class TransportData():
+    def __init__(self, co2: float, speed: float, distance_penalty: float):
+        self.co2 = co2
+        self.speed = speed
+        self.distance_penalty = distance_penalty
+
+
+class TransportMethod(Enum):
+    """
+    (co2 emission, speed, distance_penalty)
+    """
+    WALK = TransportData(0, 4, 0.2)
+    CITY = TransportData(0.189, 30, 0.2)
+    AUTOBAHN = TransportData(0.189, 100, 0.2)
+    OEPNV = TransportData(0.055, 30, 0.1)
+    ICE = TransportData(0.055, 100, 0.1)
+    AIRPLANE = TransportData(0.2113, 900, 0.02)
+
+
+class Coordinate:
+    def __init__(self, lat: float, long: float):
+        self.lat = lat
+        self.long = long
+
+    def __str__(self):
+        return f"Lat: {self.lat}, Long: {self.long}"
+
+    def __repr__(self):
+        return self.__str__()
+
+
+class Location:
+    def __init__(self, coord: Coordinate, continent: int, name: str, type: LocationType):
+        self.coord = coord
+        self.continent = continent
+        self.name = name
+        self.type = type
+
+    def __str__(self):
+        return f"{{Location: {self.coord}, continent: {self.continent}, name: {self.name}, type: {self.type}}}"
+
+    def __repr__(self):
+        return self.__str__()
+
+    def distance(self, loc2) -> float:
+        """
+        Calculate the distance im km between two locations with the help of the
+        Seitenkosinussatz
+        """
+        rErde = 6378.388
+        lat1 = float(self.coord.lat)
+        long1 = float(self.coord.long)
+        lat2 = float(loc2.coord.lat)
+        long2 = float(loc2.coord.long)
+        inner = math.sin(lat1) * math.sin(lat2) + math.cos(lat1) * math.cos(lat2) * math.cos(long2 - long1)
+        return rErde * math.acos(inner)
+
+
+class DataSet:
+    def __init__(self, locations: dict, flights: dict, connection: tuple):
+        self.locations = locations
+        self.flights = flights
+        self.connection = connection
+
+    def __str__(self):
+        return f"locations: {self.locations}, flights: {self.flights}, connection: {self.connection}"
+
+    def __repr__(self):
+        return self.__str__()