#[path = "./common/routing.rs"]
mod common;
use crate::common::define_routing_data;
use std::iter::once;
use std::sync::Arc;
use vrp_core::models::common::TimeWindow;
use vrp_core::prelude::*;
fn define_problem(goal: GoalContext, transport: Arc<dyn TransportCost>) -> GenericResult<Problem> {
let pudos = (1..=2)
.map(|idx| {
let location_idx = if idx == 1 { 1 } else { 3 };
MultiBuilder::default()
.id(format!("pudo{idx}").as_str())
.add_job(
SingleBuilder::default()
.demand(Demand::pudo_pickup(1))
.times(vec![TimeWindow::new(0., 1000.)])?
.duration(10.)?
.location(location_idx)?
.build()?,
)
.add_job(
SingleBuilder::default()
.demand(Demand::pudo_delivery(1))
.times(vec![TimeWindow::new(0., 1000.)])?
.duration(10.)?
.location(location_idx + 1)?
.build()?,
)
.build_as_job()
})
.collect::<Result<Vec<_>, _>>()?;
let vehicle = VehicleBuilder::default()
.id("v1".to_string().as_str())
.add_detail(
VehicleDetailBuilder::default()
.set_start_location(0)
.set_start_time(0.)
.set_end_location(0)
.set_end_time(10000.)
.build()?,
)
.capacity(SingleDimLoad::new(1))
.build()?;
ProblemBuilder::default()
.add_jobs(pudos.into_iter())
.add_vehicles(once(vehicle))
.with_goal(goal)
.with_transport_cost(transport)
.build()
}
fn define_goal(transport: Arc<dyn TransportCost>) -> GenericResult<GoalContext> {
let minimize_unassigned = MinimizeUnassignedBuilder::new("min-unassigned").build()?;
let capacity_feature = CapacityFeatureBuilder::<SingleDimLoad>::new("capacity").build()?;
let transport_feature = TransportFeatureBuilder::new("min-distance")
.set_transport_cost(transport)
.set_time_constrained(true)
.build_minimize_distance()?;
GoalContextBuilder::with_features(&[minimize_unassigned, transport_feature, capacity_feature])?.build()
}
fn main() -> GenericResult<()> {
let transport = Arc::new(define_routing_data()?);
let goal = define_goal(transport.clone())?;
let problem = Arc::new(define_problem(goal, transport)?);
let config = VrpConfigBuilder::new(problem.clone())
.prebuild()?
.with_max_time(Some(5))
.with_max_generations(Some(10))
.build()?;
let solution = Solver::new(problem, config).solve()?;
assert!(solution.unassigned.is_empty(), "has unassigned jobs, but all jobs must be assigned");
assert_eq!(solution.routes.len(), 1, "one tour should be there");
assert_eq!(solution.cost, 1105., "unexpected cost (total distance traveled)");
println!(
"\nIn solution, locations are visited in the following order:\n{:?}\n",
solution.get_locations().map(Iterator::collect::<Vec<_>>).collect::<Vec<_>>()
);
Ok(())
}