eflips-depot 4.3.6__py3-none-any.whl → 4.3.7__py3-none-any.whl

eflips/depot/api/__init__.py

@@ -35,8 +35,7 @@ from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
 from math import ceil
-from eflips.model import Process
-from typing import Any, Dict, Optional, Tuple, Union, List
+from typing import Any, Dict, Optional, Union, List
 
 import numpy as np
 import sqlalchemy.orm
@@ -55,6 +54,7 @@ from eflips.model import (
     Vehicle,
     VehicleType,
 )
+from eflips.model import Process
 from sqlalchemy.orm import Session
 from sqlalchemy.sql import select
 
@@ -803,6 +803,9 @@ def generate_realistic_depot_layout(
 
         # STEP 3: Put "all direct" depots at these spots and find the vehicle counts
         depot_stations = []
+        vehicle_type_rotation_dict_by_station: Dict[
+            Station, Dict[VehicleType, List[Rotation]]
+        ] = dict()
         for (
             first_last_stop_tup,
             vehicle_type_rotation_dict,
@@ -811,6 +814,11 @@ def generate_realistic_depot_layout(
             if first_stop != last_stop:
                 raise ValueError("First and last stop of a rotation are not the same.")
             depot_stations.append(first_stop)
+            vehicle_type_rotation_dict_by_station[
+                first_stop
+            ] = vehicle_type_rotation_dict
+
+        del first_last_stop_tup, vehicle_type_rotation_dict
 
         all_direct_counts: Dict[
             Station, Dict[VehicleType, int]
@@ -820,7 +828,7 @@ def generate_realistic_depot_layout(
             stations=depot_stations,
             scenario=scenario,
             session=session,
-            vehicle_type_dict=vehicle_type_rotation_dict,
+            vehicle_type_dict_by_station=vehicle_type_rotation_dict_by_station,
             shunting_duration=shunting_duration,
         )
 
@@ -847,7 +855,9 @@ def generate_realistic_depot_layout(
                 direct_counts=direct_counts,
                 line_counts=line_counts,
                 line_length=line_length,
-                vehicle_type_rotation_dict=vehicle_type_rotation_dict,
+                vehicle_type_rotation_dict=vehicle_type_rotation_dict_by_station[
+                    station
+                ],
                 shunting_duration=shunting_duration,
             )
 
@@ -926,7 +936,9 @@ def generate_realistic_depot_layout(
                 direct_counts=direct_counts,
                 line_counts=line_counts,
                 line_length=line_length,
-                vehicle_type_rotation_dict=vehicle_type_rotation_dict,
+                vehicle_type_rotation_dict=vehicle_type_rotation_dict_by_station[
+                    station
+                ],
                 shunting_duration=shunting_duration,
             )
 
@@ -937,7 +949,7 @@ def vehicle_counts_for_direct_layout(
     stations: List[Station],
     scenario: Scenario,
     session: sqlalchemy.orm.session.Session,
-    vehicle_type_dict: Dict[VehicleType, List[Rotation]],
+    vehicle_type_dict_by_station: Dict[Station, Dict[VehicleType, List[Rotation]]],
     shunting_duration: timedelta = timedelta(minutes=5),
 ) -> Dict[Station, Dict[VehicleType, int]]:
     """
@@ -959,7 +971,7 @@ def vehicle_counts_for_direct_layout(
         # Generate the depot
         direct_counts = {}
         line_counts = {}
-        for vehicle_type, rotations in vehicle_type_dict.items():
+        for vehicle_type, rotations in vehicle_type_dict_by_station[station].items():
             direct_counts[vehicle_type] = len(rotations)
             line_counts[vehicle_type] = 1
 
@@ -972,7 +984,7 @@ def vehicle_counts_for_direct_layout(
             direct_counts=direct_counts,
             line_counts=line_counts,
             line_length=8,  # We don't care about the line length here
-            vehicle_type_rotation_dict=vehicle_type_dict,
+            vehicle_type_rotation_dict=vehicle_type_dict_by_station[station],
             shunting_duration=shunting_duration,
         )
 

{eflips_depot-4.3.6.dist-info → eflips_depot-4.3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: eflips-depot
-Version: 4.3.6
+Version: 4.3.7
 Summary: Depot Simulation for eFLIPS
 Home-page: https://github.com/mpm-tu-berlin/eflips-depot
 License: AGPL-3.0-or-later

{eflips_depot-4.3.6.dist-info → eflips_depot-4.3.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 eflips/depot/__init__.py,sha256=n7jte8R6j_Ad4Mp4hkklKwil5r8u8Q_SbXrCC-nf5jM,1556
-eflips/depot/api/__init__.py,sha256=vP1OGzKcRXyoL-TsqbzDRl1OalqmCfib4suFons1aMg,93972
+eflips/depot/api/__init__.py,sha256=6voBU0iXtWsIRecVNwI6qF6VVu1w6X-sGrNYZJqWEoA,94478
 eflips/depot/api/defaults/default_settings.json,sha256=0eUDTw_rtLQFvthP8oJL93iRXlmAOravAg-4qqGMQAY,5375
 eflips/depot/api/private/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 eflips/depot/api/private/depot.py,sha256=IpXQR2GuEUuilP02njKpvExgSkktkyHPmcn2j_Q3YPk,21276
@@ -35,7 +35,7 @@ eflips/depot/simulation.py,sha256=ee0qTzOzG-8ybN36ie_NJallXfC7jUaS9JZvaYFziLs,10
 eflips/depot/smart_charging.py,sha256=C3BYqzn2-OYY4ipXm0ETtavbAM9QXZMYULBpVoChf0E,54311
 eflips/depot/standalone.py,sha256=VxcTzBaB67fNJUMmjPRwKXjhqTy6oQ41Coote2LvAmk,22338
 eflips/depot/validation.py,sha256=TIuY7cQtEJI4H2VVMSuY5IIVkacEEZ67weeMuY3NSAM,7097
-eflips_depot-4.3.6.dist-info/LICENSE.md,sha256=KB4XTk1fPHjtZCYDyPyreu6h1LVJVZXYg-5vePcWZAc,34143
-eflips_depot-4.3.6.dist-info/METADATA,sha256=ly5A27qHm7fI0i6PWaEmVYb9KjvGAxwk8vHM5a6ZFC8,5839
-eflips_depot-4.3.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-eflips_depot-4.3.6.dist-info/RECORD,,
+eflips_depot-4.3.7.dist-info/LICENSE.md,sha256=KB4XTk1fPHjtZCYDyPyreu6h1LVJVZXYg-5vePcWZAc,34143
+eflips_depot-4.3.7.dist-info/METADATA,sha256=8OdRATPcaCLTdBCag46dA-vZJQJVScvoL9hSnTOXU2E,5839
+eflips_depot-4.3.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+eflips_depot-4.3.7.dist-info/RECORD,,