Skip to content

Commit 9c1babb

Browse files
themarpethemarpe
committed
Exposed Bootloader version available when booting a device in bootloader state
1 parent e6f786a commit 9c1babb

8 files changed

Lines changed: 64 additions & 13 deletions

File tree

‎CMakeLists.txt‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ pybind11_add_module(${TARGET_NAME}
100100
src/pipeline/AssetManagerBindings.cpp
101101
src/openvino/OpenVINOBindings.cpp
102102
src/log/LogBindings.cpp
103+
src/VersionBindings.cpp
103104

104105
src/pipeline/node/NodeBindings.cpp
105106

‎depthai-core‎

‎examples/ColorCamera/rgb_preview.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@
2626
print('Connected cameras: ', device.getConnectedCameras())
2727
# Print out usb speed
2828
print('Usb speed: ', device.getUsbSpeed().name)
29+
# Bootloader version
30+
if device.getBootloaderVersion() is not None:
31+
print('Bootloader version: ', device.getBootloaderVersion())
2932

3033
# Output queue will be used to get the rgb frames from the output defined above
3134
qRgb = device.getOutputQueue(name="rgb", maxSize=4, blocking=False)

‎src/DeviceBindings.cpp‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -279,6 +279,16 @@ static void bindConstructors(ARG& arg){
279279
py::gil_scoped_release release;
280280
return std::make_unique<D>(config, deviceInfo);
281281
}), py::arg("config"), py::arg("deviceInfo"), DOC(dai, DeviceBase, DeviceBase, 19))
282+
283+
// DeviceInfo version
284+
.def(py::init([](const DeviceInfo& deviceInfo){
285+
py::gil_scoped_release release;
286+
return std::make_unique<D>(deviceInfo);
287+
}), py::arg("deviceInfo"), DOC(dai, DeviceBase, DeviceBase, 20))
288+
.def(py::init([](const DeviceInfo& deviceInfo, UsbSpeed maxUsbSpeed){
289+
py::gil_scoped_release release;
290+
return std::make_unique<D>(deviceInfo, maxUsbSpeed);
291+
}), py::arg("deviceInfo"), py::arg("maxUsbSpeed"), DOC(dai, DeviceBase, DeviceBase, 21))
282292
;
283293

284294
}
@@ -459,6 +469,7 @@ void DeviceBindings::bind(pybind11::module& m, void* pCallstack){
459469
.def_static("getDeviceByMxId", &DeviceBase::getDeviceByMxId, py::arg("mxId"), DOC(dai, DeviceBase, getDeviceByMxId))
460470

461471
// methods
472+
.def("getBootloaderVersion", &DeviceBase::getBootloaderVersion, DOC(dai, DeviceBase, getBootloaderVersion))
462473
.def("isPipelineRunning", [](DeviceBase& d) { py::gil_scoped_release release; return d.isPipelineRunning(); }, DOC(dai, DeviceBase, isPipelineRunning))
463474
.def("startPipeline", [](DeviceBase& d){
464475
// Issue an deprecation warning

‎src/DeviceBootloaderBindings.cpp‎

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@ void DeviceBootloaderBindings::bind(pybind11::module& m, void* pCallstack){
99

1010
// Type definitions
1111
py::class_<DeviceBootloader> deviceBootloader(m, "DeviceBootloader", DOC(dai, DeviceBootloader));
12-
py::class_<DeviceBootloader::Version> deviceBootloaderVersion(deviceBootloader, "Version", DOC(dai, DeviceBootloader, Version));
1312
py::enum_<DeviceBootloader::Type> deviceBootloaderType(deviceBootloader, "Type");
1413
py::enum_<DeviceBootloader::Memory> deviceBootloaderMemory(deviceBootloader, "Memory");
1514
py::enum_<DeviceBootloader::Section> deviceBootloaderSection(deviceBootloader, "Section");
@@ -34,17 +33,6 @@ void DeviceBootloaderBindings::bind(pybind11::module& m, void* pCallstack){
3433

3534

3635
// Bind DeviceBootloader
37-
deviceBootloaderVersion
38-
.def(py::init<const std::string&>(), py::arg("v"), DOC(dai, DeviceBootloader, Version, Version))
39-
.def(py::init<unsigned, unsigned, unsigned>(), py::arg("major"), py::arg("minor"), py::arg("patch"), DOC(dai, DeviceBootloader, Version, Version, 2))
40-
.def("__str__", &DeviceBootloader::Version::toString)
41-
.def("__eq__", &DeviceBootloader::Version::operator==)
42-
.def("__lt__", &DeviceBootloader::Version::operator<)
43-
.def("__gt__", &DeviceBootloader::Version::operator>)
44-
.def("toStringSemver", &DeviceBootloader::Version::toStringSemver)
45-
.def("getBuildInfo", &DeviceBootloader::Version::getBuildInfo)
46-
;
47-
4836
deviceBootloaderType
4937
.value("AUTO", DeviceBootloader::Type::AUTO)
5038
.value("USB", DeviceBootloader::Type::USB)

‎src/VersionBindings.cpp‎

Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
#include "VersionBindings.hpp"
2+
3+
// depthai
4+
#include "depthai/device/Version.hpp"
5+
6+
void VersionBindings::bind(pybind11::module& m, void* pCallstack){
7+
8+
using namespace dai;
9+
10+
// Type definitions
11+
py::class_<Version> version(m, "Version", DOC(dai, Version));
12+
13+
///////////////////////////////////////////////////////////////////////
14+
///////////////////////////////////////////////////////////////////////
15+
///////////////////////////////////////////////////////////////////////
16+
// Call the rest of the type defines, then perform the actual bindings
17+
Callstack* callstack = (Callstack*) pCallstack;
18+
auto cb = callstack->top();
19+
callstack->pop();
20+
cb(m, pCallstack);
21+
///////////////////////////////////////////////////////////////////////
22+
///////////////////////////////////////////////////////////////////////
23+
///////////////////////////////////////////////////////////////////////
24+
25+
// Bind DeviceBootloader
26+
version
27+
.def(py::init<const std::string&>(), py::arg("v"), DOC(dai, Version, Version))
28+
.def(py::init<unsigned, unsigned, unsigned>(), py::arg("major"), py::arg("minor"), py::arg("patch"), DOC(dai, Version, Version, 2))
29+
.def("__str__", &Version::toString, DOC(dai, Version, toString))
30+
.def("__eq__", &Version::operator==)
31+
.def("__lt__", &Version::operator<)
32+
.def("__gt__", &Version::operator>)
33+
.def("toStringSemver", &Version::toStringSemver, DOC(dai, Version, toStringSemver))
34+
.def("getBuildInfo", &Version::getBuildInfo, DOC(dai, Version, getBuildInfo))
35+
.def("getSemver", &Version::getSemver, DOC(dai, Version, getSemver))
36+
;
37+
38+
}

‎src/VersionBindings.hpp‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#pragma once
2+
3+
// pybind
4+
#include "pybind11_common.hpp"
5+
6+
struct VersionBindings {
7+
static void bind(pybind11::module& m, void* pCallstack);
8+
};

‎src/py_bindings.cpp‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
#include "DataQueueBindings.hpp"
2828
#include "openvino/OpenVINOBindings.hpp"
2929
#include "log/LogBindings.hpp"
30+
#include "VersionBindings.hpp"
3031

3132
PYBIND11_MODULE(depthai, m)
3233
{
@@ -43,6 +44,7 @@ PYBIND11_MODULE(depthai, m)
4344
std::deque<StackFunction> callstack;
4445
DatatypeBindings::addToCallstack(callstack);
4546
callstack.push_front(&LogBindings::bind);
47+
callstack.push_front(&VersionBindings::bind);
4648
callstack.push_front(&DataQueueBindings::bind);
4749
callstack.push_front(&OpenVINOBindings::bind);
4850
NodeBindings::addToCallstack(callstack);

0 commit comments

Comments
 (0)