swap START/END when ArriveBy is set (#747)

motis-project · Feb 17, 2025 · 1a20267 · 1a20267
1 parent 8fa459b
commit 1a20267
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/journey_to_response.cc b/src/journey_to_response.cc
@@ -45,6 +45,15 @@ api::ModeEnum to_mode(osr::search_profile const m) {
   std::unreachable();
 }
 
+void cleanup_intermodal(api::Itinerary& i) {
+  if (i.legs_.front().from_.name_ == "END") {
+    i.legs_.front().from_.name_ = "START";
+  }
+  if (i.legs_.back().to_.name_ == "START") {
+    i.legs_.back().to_.name_ = "END";
+  }
+}
+
 api::Itinerary journey_to_response(osr::ways const* w,
                                    osr::lookup const* l,
                                    osr::platforms const* pl,
@@ -202,6 +211,8 @@ api::Itinerary journey_to_response(osr::ways const* w,
         j_leg.uses_);
   }
 
+  cleanup_intermodal(itinerary);
+
   return itinerary;
 }