Save uncongested travel time matrices to Results when using congested transit assignment every iteration

Author: jalovaj2

Scripts/assignment/assignment_period.py

@@ -161,7 +161,7 @@ def assign(self, matrices: dict, iteration: Union[int,str]) -> Dict:
             raise ValueError("Iteration number not valid")
 
         mtxs = {imp_type: self._get_matrices(imp_type, iteration=="last")
-            for imp_type in ("time", "cost", "dist")}
+            for imp_type in ("time", "cost", "dist", "congest_time")}
         # fix the emme path analysis results
         # (dist and cost are zero if path not found but we want it to
         # be the default value 999999)
@@ -831,6 +831,11 @@ def _assign_congested_transit(self):
             stopping_criteria=param.trass_stop,
             log_worksheets=False, scenario=self.emme_scenario,
             save_strategies=True)
+        # save uncongested results for both both classes
+        for tc in specs:
+            self.emme_project.strategy_analysis(
+                specs[tc].strategy_analysis_spec, scenario=self.emme_scenario,
+                class_name=tc)
         # save results for both classes
         for tc in specs:
             self.emme_project.matrix_results(

Scripts/assignment/datatypes/transit.py

@@ -90,6 +90,34 @@ def __init__(self,
                 "distance": emme_matrices["dist"],
             },
         }
+        self.strategy_analysis_spec = {
+            "type": "EXTENDED_TRANSIT_STRATEGY_ANALYSIS",
+            "trip_components": {
+                "boarding": None,
+                "in_vehicle": "@ccost",
+                "alighting": None,
+                "aux_transit": None,
+            },
+            "sub_path_combination_operator": "+",
+            "sub_strategy_combination_operator": "average",
+            "selected_demand_and_transit_volumes": {
+                "sub_strategies_to_retain": "ALL",
+                "selection_threshold": {
+                    "lower": -999999,
+                    "upper": 999999
+                }
+            },
+            "analyzed_demand": None,
+            "constraint": None,
+            "results": {
+                "strategy_values": emme_matrices["congest_time"],
+                "selected_demand": None,
+                "transit_volumes": None,
+                "aux_transit_volumes": None,
+                "total_boardings": None,
+                "total_alightings": None
+            }
+        }
         if count_zone_boardings:
             jlevel1 = JourneyLevel(
                 headway_attribute, boarded=False, count_zone_boardings=True)

Scripts/assignment/emme_bindings/emme_project.py

@@ -66,6 +66,8 @@ def __init__(self,
             "inro.emme.transit_assignment.extended.network_results")
         self.create_extra_attribute = self.modeller.tool(
             "inro.emme.data.extra_attribute.create_extra_attribute")
+        self.strategy_analysis = self.modeller.tool(
+            "inro.emme.transit_assignment.extended.strategy_based_analysis")
 
     def write(self, message: str):
         """Write to logbook."""

Scripts/modelsystem.py

@@ -322,8 +322,11 @@ def run_iteration(self, previous_iter_impedance, iteration=None):
             impedance[tp] = ap.assign(self.dtm.demand[tp], iteration)
             if tp == "aht":
                 self._update_ratios(impedance[tp], tp)
-            if iteration=="last": # TODO: Get uncongested time from assignment results
-                log.warn('TODO: Get uncongested transit time from the assignment results')
+            if iteration=="last" and param.always_congested:
+                impedance[tp]["time"]["transit_uncongested_work"] = impedance[tp]["time"]["transit_work"] - impedance[tp]["congest_time"]["transit_work"]
+                impedance[tp]["time"]["transit_uncongested_leisure"] = impedance[tp]["time"]["transit_leisure"] - impedance[tp]["congest_time"]["transit_leisure"]
+                self._save_to_omx(impedance[tp], tp)
+            elif iteration=="last":
                 impedance[tp]["time"]["transit_uncongested"] = previous_iter_impedance[tp]["time"]["transit_work"]
                 self._save_to_omx(impedance[tp], tp)
         if iteration=="last":

Scripts/parameters/assignment.py

@@ -464,8 +464,8 @@ def calc_segment_cost(transit_volume, line_capacity, segment):
 emme_matrices = {
     "car_work": ("demand", "time", "dist", "cost", "gen_cost"),
     "car_leisure": ("demand", "time", "dist", "cost", "gen_cost"),
-    "transit_work": ("demand", "time", "dist", "cost"),
-    "transit_leisure": ("demand", "time", "dist", "cost"),
+    "transit_work": ("demand", "time", "dist", "cost", "congest_time"),
+    "transit_leisure": ("demand", "time", "dist", "cost", "congest_time"),
     "bike": ("demand", "time", "dist"),
     "walk": ("time", "dist"),
     "trailer_truck": ("demand", "time", "dist", "cost", "gen_cost"),