Merge pull request #890 from luxonis/getConnectivity_bindings

Get connectivity bindings
luxonis · Nov 20, 2023 · 9c50aa6 · 9c50aa6
2 parents b591040 + d32e049
commit 9c50aa6
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 9 deletions.
diff --git a/depthai-core b/depthai-core
diff --git a/src/DeviceBindings.cpp b/src/DeviceBindings.cpp
@@ -611,6 +611,7 @@ void DeviceBindings::bind(pybind11::module& m, void* pCallstack){
         .def("getCrashDump", [](DeviceBase& d, bool clearCrashDump) { py::gil_scoped_release release; return d.getCrashDump(clearCrashDump); }, py::arg("clearCrashDump") = true, DOC(dai, DeviceBase, getCrashDump))
         .def("hasCrashDump", [](DeviceBase& d) { py::gil_scoped_release release; return d.hasCrashDump(); }, DOC(dai, DeviceBase, hasCrashDump))
         .def("getConnectedCameras", [](DeviceBase& d) { py::gil_scoped_release release; return d.getConnectedCameras(); }, DOC(dai, DeviceBase, getConnectedCameras))
+        .def("getConnectionInterfaces", [](DeviceBase& d) { py::gil_scoped_release release; return d.getConnectionInterfaces(); }, DOC(dai, DeviceBase, getConnectionInterfaces))
         .def("getConnectedCameraFeatures", [](DeviceBase& d) { py::gil_scoped_release release; return d.getConnectedCameraFeatures(); }, DOC(dai, DeviceBase, getConnectedCameraFeatures))
         .def("getCameraSensorNames", [](DeviceBase& d) { py::gil_scoped_release release; return d.getCameraSensorNames(); }, DOC(dai, DeviceBase, getCameraSensorNames))
         .def("getConnectedIMU", [](DeviceBase& d) { py::gil_scoped_release release; return d.getConnectedIMU(); }, DOC(dai, DeviceBase, getConnectedIMU))

diff --git a/src/pipeline/CommonBindings.cpp b/src/pipeline/CommonBindings.cpp
@@ -5,6 +5,7 @@
 
 // depthai-shared
 #include "depthai-shared/common/CameraBoardSocket.hpp"
+#include "depthai-shared/common/Connectivity.hpp"
 #include "depthai-shared/common/EepromData.hpp"
 #include "depthai-shared/common/CameraImageOrientation.hpp"
 #include "depthai-shared/common/CameraSensorType.hpp"
@@ -40,6 +41,7 @@ void CommonBindings::bind(pybind11::module& m, void* pCallstack){
     py::class_<Point3f> point3f(m, "Point3f", DOC(dai, Point3f));
     py::class_<Size2f> size2f(m, "Size2f", DOC(dai, Size2f));
     py::enum_<CameraBoardSocket> cameraBoardSocket(m, "CameraBoardSocket", DOC(dai, CameraBoardSocket));
+    py::enum_<Connectivity> connectivity(m, "Connectivity", DOC(dai, Connectivity));
     py::enum_<CameraSensorType> cameraSensorType(m, "CameraSensorType", DOC(dai, CameraSensorType));
     py::enum_<CameraImageOrientation> cameraImageOrientation(m, "CameraImageOrientation", DOC(dai, CameraImageOrientation));
     py::class_<CameraSensorConfig> cameraSensorConfig(m, "CameraSensorConfig", DOC(dai, CameraSensorConfig));
@@ -183,7 +185,12 @@ void CommonBindings::bind(pybind11::module& m, void* pCallstack){
         .value("TOF", CameraSensorType::TOF)
         .value("THERMAL", CameraSensorType::THERMAL)
     ;
-
+    // Connectivity enum bindings
+    connectivity
+        .value("USB", Connectivity::USB)
+        .value("ETHERNET", Connectivity::ETHERNET)
+        .value("WIFI", Connectivity::WIFI)
+    ;
     // CameraImageOrientation enum bindings
     cameraImageOrientation
         .value("AUTO", CameraImageOrientation::AUTO)

diff --git a/src/pipeline/PipelineBindings.cpp b/src/pipeline/PipelineBindings.cpp
@@ -29,6 +29,7 @@
 #include "depthai/pipeline/node/AprilTag.hpp"
 #include "depthai/pipeline/node/DetectionParser.hpp"
 #include "depthai/pipeline/node/UVC.hpp"
+#include "depthai/pipeline/node/Warp.hpp"
 
 // depthai-shared
 #include "depthai-shared/properties/GlobalProperties.hpp"

diff --git a/src/pipeline/datatype/EncodedFrameBindings.cpp b/src/pipeline/datatype/EncodedFrameBindings.cpp
@@ -82,15 +82,15 @@ void bind_encodedframe(pybind11::module &m, void *pCallstack) {
       .def(py::init<>())
       // getters
       .def("getTimestamp",
-           py::overload_cast<>(&EncodedFrame::getTimestamp, py::const_),
-           DOC(dai, EncodedFrame, getTimestamp))
+           py::overload_cast<>(&EncodedFrame::Buffer::getTimestamp, py::const_),
+           DOC(dai, Buffer, getTimestamp))
       .def("getTimestampDevice",
-           py::overload_cast<>(&EncodedFrame::getTimestampDevice, py::const_),
-           DOC(dai, EncodedFrame, getTimestampDevice))
+           py::overload_cast<>(&EncodedFrame::Buffer::getTimestampDevice, py::const_),
+           DOC(dai, Buffer, getTimestampDevice))
       .def("getInstanceNum", &EncodedFrame::getInstanceNum,
            DOC(dai, EncodedFrame, getInstanceNum))
-      .def("getSequenceNum", &EncodedFrame::getSequenceNum,
-           DOC(dai, EncodedFrame, getSequenceNum))
+      .def("getSequenceNum", &EncodedFrame::Buffer::getSequenceNum,
+           DOC(dai, Buffer, getSequenceNum))
       .def("getExposureTime", &EncodedFrame::getExposureTime,
            DOC(dai, EncodedFrame, getExposureTime))
       .def("getSensitivity", &EncodedFrame::getSensitivity,
@@ -116,7 +116,7 @@ void bind_encodedframe(pybind11::module &m, void *pCallstack) {
       .def("setTimestampDevice", &EncodedFrame::setTimestampDevice,
            DOC(dai, EncodedFrame, setTimestampDevice))
       .def("setSequenceNum", &EncodedFrame::setSequenceNum,
-           DOC(dai, EncodedFrame, getSequenceNum))
+           DOC(dai, EncodedFrame, setSequenceNum))
       .def("setQuality", &EncodedFrame::setQuality,
            DOC(dai, EncodedFrame, getQuality))
       .def("setBitrate", &EncodedFrame::setBitrate,