cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Gracefully fail resource path lookup for Python on Windows

Browse Source 
PiperOrigin-RevId: 520513921
This commit is contained in:
Sebastian Schmidt 2023-03-29 20:12:12 -07:00 committed by Copybara-Service
parent ee1807d8e3
commit ac52859f1d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mediapipe/tasks/cc/core/external_file_handler.cc
View File

@ -100,10 +100,11 @@ absl::StatusOr<std::string> PathToResourceAsFile(std::string path) {
#ifndef _WIN32 #ifndef _WIN32
return path; return path;
#else #else
if (absl::StartsWith(path, "./")) { std::string qualified_path = path;
path = "mediapipe" + path.substr(1); if (absl::StartsWith(qualified_path, "./")) {
qualified_path = "mediapipe" + qualified_path.substr(1);
} else if (path[0] != '/') { } else if (path[0] != '/') {
path = "mediapipe/" + path; qualified_path = "mediapipe/" + qualified_path;
} }
std::string error; std::string error;
@ -112,9 +113,10 @@ absl::StatusOr<std::string> PathToResourceAsFile(std::string path) {
std::unique_ptr<::bazel::tools::cpp::runfiles::Runfiles> runfiles( std::unique_ptr<::bazel::tools::cpp::runfiles::Runfiles> runfiles(
::bazel::tools::cpp::runfiles::Runfiles::Create("", &error)); ::bazel::tools::cpp::runfiles::Runfiles::Create("", &error));
if (!runfiles) { if (!runfiles) {
return absl::InternalError("Unable to initialize runfiles: " + error); // Return the original path when Runfiles is not available (e.g. for Python)
return path;
} }
return runfiles->Rlocation(path); return runfiles->Rlocation(qualified_path);
#endif // _WIN32 #endif // _WIN32
} }