Initial commit

src/prediction/PredictionManager.cpp

@@ -0,0 +1,71 @@
+
+#include <boost/program_options.hpp>
+#include <boost/format.hpp>
+#include <wfr/prediction/services/oracle/OraclePredictionService.h>
+#include <wfr/prediction/PredictionManager.h>
+#include <wfr/prediction/PredictionService.h>
+#include <iostream>
+
+
+
+namespace wfr {
+namespace prediction {
+
+
+PredictionManager PredictionManager::singleton;
+
+void PredictionManager::init(int argc, char** argv) {
+	std::string predictionConfig;
+    boost::program_options::options_description opts("Prediction Options");
+    opts.add_options()
+    	    ("predictionconfig", boost::program_options::value<std::string>(), "Set prediction service config.")
+    	    ("help", "Print help message.");
+    boost::program_options::variables_map vm;
+    auto parsed = boost::program_options::command_line_parser(argc, argv).options(opts).allow_unregistered().run();
+    boost::program_options::store(parsed, vm);
+    boost::program_options::notify(vm);
+    if (vm.count("help")) {
+    	std::cerr << opts << std::endl;
+    	return;
+    }
+    if (vm.count("predictionconfig")) {
+    	predictionConfig = vm["predictionconfig"].as<std::string>();
+    	std::cerr << "Prediction Service Config is set to \""<< predictionConfig << "\"." << std::endl;
+    }
+    singleton.service = new OraclePredictionService(predictionConfig);
+}
+
+
+void PredictionManager::finish() noexcept {
+}
+
+
+PredictionManager::~PredictionManager() noexcept {
+	if (service) {
+		delete service;
+		service = nullptr;
+	}
+}
+
+
+ExecutionTimeResult PredictionManager::predictExecutionTime(wfs::ActivityInstance& instance, wfs::AtomicActivityImplementation& implementation, resource::ComputingResourceType& resourceType, const wfr_size_t& slots) noexcept {
+	if (service) {
+		return service->predictExecutionTime(instance, implementation, resourceType, slots);
+	} else {
+		std::cerr << "No Prediction Service configured." << std::endl;
+		return ExecutionTimeResult();
+	}
+}
+
+CommunicationTimeResult PredictionManager::predictCommunicationTime(wfs::ActivityInstance& srcInstance, resource::ComputingResource& srcResource,
+		wfs::ActivityInstance& destInstance, resource::ComputingResource& destResource) noexcept {
+	if (service) {
+		return service->predictCommunicationTime(srcInstance, srcResource, destInstance, destResource);
+	} else {
+		std::cerr << "No Prediction Service configured." << std::endl;
+		return CommunicationTimeResult();
+	}
+}
+
+} /* namespace prediction */
+} /* namespace wfr */