diff --git a/jsprit-core/src/main/java/algorithms/InfiniteVehicles.java b/jsprit-core/src/main/java/algorithms/InfiniteVehicles.java index e41d6de8..84da957b 100644 --- a/jsprit-core/src/main/java/algorithms/InfiniteVehicles.java +++ b/jsprit-core/src/main/java/algorithms/InfiniteVehicles.java @@ -93,15 +93,19 @@ class InfiniteVehicles implements VehicleFleetManager{ @Override public Collection getAvailableVehicles() { - // TODO Auto-generated method stub - return null; + return types.values(); } @Override - public Collection getAvailableVehicles( - String withoutThisType, String locationId) { - // TODO Auto-generated method stub - return null; + public Collection getAvailableVehicles(String withoutThisType, String locationId) { + Collection vehicles = new ArrayList(); + TypeKey thisKey = new TypeKey(withoutThisType,locationId); + for(TypeKey key : types.keySet()){ + if(!key.equals(thisKey)){ + vehicles.add(types.get(key)); + } + } + return vehicles; } }