adding a loader for the vehicle routing problem

Author: ndrwnaguib

libecole/CMakeLists.txt

@@ -21,6 +21,7 @@ add_library(
 	src/instance/independent-set.cpp
 	src/instance/combinatorial-auction.cpp
 	src/instance/capacitated-facility-location.cpp
+	src/instance/capacitated-vehicle-routing.cpp
 
 	src/reward/is-done.cpp
 	src/reward/lp-iterations.cpp

libecole/include/ecole/instance/capacitated-vehicle-routing.hpp

@@ -0,0 +1,38 @@
+#pragma once
+
+#include <cstddef>
+#include <string>
+#include <utility>
+#include <vector>
+
+#include "ecole/export.hpp"
+#include "ecole/instance/abstract.hpp"
+#include "ecole/random.hpp"
+
+namespace ecole::instance {
+
+class ECOLE_EXPORT CapacitatedVehicleRoutingLoader : public InstanceGenerator {
+public:
+	struct ECOLE_EXPORT Parameters {
+		std::string filename;    // NOLINT(readability-magic-numbers)
+		std::size_t n_vehicles;  // NOLINT(readability-magic-numbers)
+	};
+
+	ECOLE_EXPORT static scip::Model generate_instance(Parameters parameters, RandomGenerator& rng);
+
+	ECOLE_EXPORT CapacitatedVehicleRoutingLoader(Parameters parameters, RandomGenerator rng);
+	ECOLE_EXPORT CapacitatedVehicleRoutingLoader(Parameters parameters);
+	ECOLE_EXPORT CapacitatedVehicleRoutingLoader();
+
+	ECOLE_EXPORT scip::Model next() override;
+	ECOLE_EXPORT void seed(Seed seed) override;
+	[[nodiscard]] ECOLE_EXPORT bool done() const override { return false; }
+
+	[[nodiscard]] ECOLE_EXPORT Parameters const& get_parameters() const noexcept { return parameters; }
+
+private:
+	RandomGenerator rng;
+	Parameters parameters;
+};
+
+}  // namespace ecole::instance