Set wpi::Now source on roborio when executing -m cscore

- Fixes #62
robotpy · Jan 21, 2024 · ec7f385 · ec7f385
1 parent a07b9d9
commit ec7f385
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/subprojects/robotpy-cscore/cscore/__main__.py b/subprojects/robotpy-cscore/cscore/__main__.py
@@ -159,4 +159,12 @@ def main():
 
 
 if __name__ == "__main__":
+    # Setup wpi::now on roborio when executed as __main__
+    try:
+        from ._cscore import _setupWpiNow  # type: ignore
+
+        _setupWpiNow()
+    except ImportError:
+        pass
+
     main()
diff --git a/subprojects/robotpy-cscore/cscore/src/main.cpp b/subprojects/robotpy-cscore/cscore/src/main.cpp
@@ -3,6 +3,12 @@
 
 #include "cscore_cpp.h"
 
+#ifdef __FRC_ROBORIO__
+extern "C" {
+    void WPI_Impl_SetupNowUseDefaultOnRio(void);
+}
+#endif
+
 RPYBUILD_PYBIND11_MODULE(m) {
     initWrapper(m);
 
@@ -16,4 +22,8 @@ RPYBUILD_PYBIND11_MODULE(m) {
         CS_Shutdown();
     });
     m.add_object("_cleanup", cleanup);
+
+    #ifdef __FRC_ROBORIO__
+    m.def("_setupWpiNow", WPI_Impl_SetupNowUseDefaultOnRio);
+    #endif
 }