[Warp] Exposed HW engines used and interpolation

themarpe · themarpe · commit eacbc8d691f5 · 2022-09-26T15:46:22.000+02:00
diff --git a/depthai-core b/depthai-core
@@ -1 +1 @@
-Subproject commit b88ca03d9e1e0f20d8c082de9a53da22bc86ec07
+Subproject commit 10ff2b413639f7d746653c3d747107ae990963b8
diff --git a/examples/Warp/warp_mesh.py b/examples/Warp/warp_mesh.py
@@ -24,22 +24,26 @@
 WARP1_OUTPUT_FRAME_SIZE = (992,500)
 warp1.setOutputSize(WARP1_OUTPUT_FRAME_SIZE)
 warp1.setMaxOutputFrameSize(WARP1_OUTPUT_FRAME_SIZE[0] * WARP1_OUTPUT_FRAME_SIZE[1] * 3)
+warp1.setHwIds([1])
+warp1.setInterpolation(dai.node.Warp.Properties.Interpolation.BYPASS)
 
 camRgb.preview.link(warp1.inputImage)
 xout1 = pipeline.create(dai.node.XLinkOut)
 xout1.setStreamName('out1')
 warp1.out.link(xout1.input)
 
 # Warp preview frame 2
-warp2 = pipeline.create(dai.node.Warp);
+warp2 = pipeline.create(dai.node.Warp)
 # Create a custom warp mesh
 mesh2 = [
     (20, 20), (250, 100), (460, 20),
     (100, 250), (250, 250), (400, 250),
     (20, 480), (250, 400), (460,480)
 ]
-warp2.setWarpMesh(mesh2, 3, 3);
-warp2.setMaxOutputFrameSize(maxFrameSize);
+warp2.setWarpMesh(mesh2, 3, 3)
+warp2.setMaxOutputFrameSize(maxFrameSize)
+warp1.setHwIds([2])
+warp2.setInterpolation(dai.node.Warp.Properties.Interpolation.BICUBIC)
 
 camRgb.preview.link(warp2.inputImage);
 xout2 = pipeline.create(dai.node.XLinkOut)
diff --git a/src/pipeline/node/WarpBindings.cpp b/src/pipeline/node/WarpBindings.cpp
@@ -12,6 +12,8 @@ void bind_warp(pybind11::module& m, void* pCallstack){
 
     // Node and Properties declare upfront
     // TODO(themarpe) - properties
+    py::class_<Warp::Properties> warpProperties(m, "WarpProperties", DOC(dai, WarpProperties));
+    py::enum_<Warp::Properties::Interpolation> warpPropertiesInterpolation(warpProperties, "Interpolation", DOC(dai, WarpProperties, Interpolation));
     auto warp = ADD_NODE(Warp);
 
     ///////////////////////////////////////////////////////////////////////
@@ -26,6 +28,13 @@ void bind_warp(pybind11::module& m, void* pCallstack){
     ///////////////////////////////////////////////////////////////////////
     ///////////////////////////////////////////////////////////////////////
 
+    // properties
+    warpPropertiesInterpolation
+        .value("BILINEAR", Warp::Properties::Interpolation::BILINEAR)
+        .value("BICUBIC", Warp::Properties::Interpolation::BICUBIC)
+        .value("BYPASS", Warp::Properties::Interpolation::BYPASS)
+        ;
+
     // ImageManip Node
     warp
         // .def_readonly("inputConfig", &Warp::inputConfig, DOC(dai, node, Warp, inputConfig))
@@ -44,6 +53,13 @@ void bind_warp(pybind11::module& m, void* pCallstack){
 
         .def("setWarpMesh", py::overload_cast<const std::vector<Point2f>&, int, int>(&Warp::setWarpMesh), DOC(dai, node, Warp, setWarpMesh))
         .def("setWarpMesh", py::overload_cast<const std::vector<std::pair<float,float>>&, int, int>(&Warp::setWarpMesh), DOC(dai, node, Warp, setWarpMesh))
+
+        .def("setHwIds", &Warp::setHwIds, DOC(dai, node, Warp, setHwIds))
+        .def("getHwIds", &Warp::getHwIds, DOC(dai, node, Warp, getHwIds))
+        .def("setInterpolation", &Warp::setInterpolation, DOC(dai, node, Warp, setInterpolation))
+        .def("getInterpolation", &Warp::getInterpolation, DOC(dai, node, Warp, getInterpolation))
         ;
 
+    daiNodeModule.attr("Warp").attr("Properties") = warpProperties;
+
 }
