odm: pass booking parameters

motis-project · Feb 20, 2025 · fb604ba · fb604ba
1 parent dc19f80
commit fb604ba
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/include/motis/odm/prima.h b/include/motis/odm/prima.h
@@ -22,10 +22,10 @@ struct direct_ride {
 };
 
 struct capacities {
-  std::uint8_t wheelchairs_;
-  std::uint8_t bikes_;
-  std::uint8_t passengers_;
-  std::uint8_t luggage_;
+  std::int64_t wheelchairs_;
+  std::int64_t bikes_;
+  std::int64_t passengers_;
+  std::int64_t luggage_;
 };
 
 struct prima {

diff --git a/src/odm/prima.cc b/src/odm/prima.cc
@@ -30,11 +30,11 @@ void prima::init(api::Place const& from,
   cap_ = {
       .wheelchairs_ = static_cast<std::uint8_t>(
           query.pedestrianProfile_ == api::PedestrianProfileEnum::WHEELCHAIR
-              ? 1
-              : 0),
+              ? 1U
+              : 0U),
       .bikes_ = static_cast<std::uint8_t>(query.requireBikeTransport_ ? 1 : 0),
-      .passengers_ = 1U,
-      .luggage_ = 0U};
+      .passengers_ = query.passengers_.value_or(1U),
+      .luggage_ = query.luggage_.value_or(0U)};
 }
 
 std::int64_t to_millis(n::unixtime_t const t) {