ExperimentUtils: Add experimentRootFromString()

Lehdari · Apr 27, 2023 · b380fa5 · b380fa5
1 parent 1f7f094
commit b380fa5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/include/util/ExperimentUtils.hpp b/include/util/ExperimentUtils.hpp
@@ -13,6 +13,11 @@
 #include <nlohmann/json.hpp>
 
 #include <string>
+#include <filesystem>
 
 
 std::string formatExperimentName(const std::string& name, const nlohmann::json& modelConfig);
+
+// Convert a string to an experiment root path.
+// Relative paths are considered to be relative to the experiments directory (doot2::experimentsDirectory)
+std::filesystem::path experimentRootFromString(const std::string& experimentRoot);
diff --git a/src/util/ExperimentUtils.cpp b/src/util/ExperimentUtils.cpp
@@ -9,6 +9,7 @@
 //
 
 #include "util/ExperimentUtils.hpp"
+#include "Constants.hpp"
 
 #include <sstream>
 #include <regex>
@@ -33,3 +34,11 @@ std::string formatExperimentName(const std::string& name, const nlohmann::json&
 
     return experimentName;
 }
+
+std::filesystem::path experimentRootFromString(const std::string& experimentRoot)
+{
+    std::filesystem::path root = experimentRoot;
+    if (!root.is_absolute())
+        root = doot2::experimentsDirectory / root;
+    return root;
+}