Add python binding for DrawTool::drawRGBAImage()

Author: damienmarchal

bindings/Sofa/src/SofaPython3/Sofa/Core/Binding_VisualParams.cpp

@@ -95,6 +95,13 @@ void moduleAddVisualParams(py::module &m)
     dt.def("drawText", [](DrawTool* self, int x, int y, int fontSize, char* text){
         self->writeOverlayText(x,y, fontSize, sofa::type::RGBAColor::white(), text);
     });
+    dt.def("drawRGBAImage", [](DrawTool* self, const std::string& id, const sofa::type::Vec3& pos, const int w, const int h, const int mode, const char* data){
+        self->drawRGBAImage(id, pos, w, h, mode, data);
+    });
+    dt.def("drawRGBAImage", [](DrawTool* self, const std::string& id, const sofa::type::Vec3& pos, const std::string& filename){
+        self->drawRGBAImage(id, pos, filename);
+    });
+
 }
 
 } /// namespace sofapython3

examples/example-drawing-controller.py

@@ -1,13 +1,16 @@
 import Sofa
 import SofaTypes 
 
+from PIL import Image
+
 class DrawingExamples(Sofa.Core.Controller):
     def __init__(self, *args, **kwargs):
         # These are needed (and the normal way to override from a python class)
         Sofa.Core.Controller.__init__(self, *args, **kwargs)
 
         self.target = kwargs.get("target", None)
         self.mo = kwargs.get("mo", None)
+        self.img = Image.open("oip.jpeg").convert("RGBA")
 
     def draw(self, visual_context):
         dt = visual_context.getDrawTool()
@@ -23,10 +26,14 @@ def draw(self, visual_context):
         dt.drawText(10,10, 12, "Overlay text")
 
         dt.drawFrames(self.mo.position, SofaTypes.Vec3d(0.1,0.1,0.1))
+        w,h = self.img.size 
+        dt.drawRGBAImage("memory", SofaTypes.Vec3d(0.0,0.0,0.0), w ,h, 32, self.img.tobytes())
+        #dt.drawRGBAImage("image", SofaTypes.Vec3d(0.0,0.0,0.0), "oip.jpeg")
 
 def createScene(root):
     root.dt = 0.01
     root.bbox = [[-1,-1,-1],[1,1,1]]
+    root.addObject('RequiredPlugin', name="Sofa.GL.Component.Shader")
     root.addObject('DefaultVisualManagerLoop')
     root.addObject('DefaultAnimationLoop')