Call draw every frame

Author: catilac

crates/processing_pyo3/examples/rectangle.py

@@ -1,7 +1,7 @@
 from processing import *
 
-# TODO: this should be in a setup function
-size(800, 600)
+def setup():
+    size(800, 600)
 
 def draw():
     background(220)
@@ -12,4 +12,4 @@ def draw():
     rect(100, 100, 200, 150)
 
 # TODO: this should happen implicitly on module load somehow
-run(draw)
+run()

crates/processing_pyo3/src/lib.rs

@@ -43,9 +43,30 @@ fn run(module: &Bound<'_, PyModule>) -> PyResult<()> {
     Python::attach(|py| {
         let builtins = PyModule::import(py, "builtins")?;
         let locals = builtins.getattr("locals")?.call0()?;
+
         let setup_fn = locals.get_item("setup")?;
+        let draw_fn = locals.get_item("draw")?;
+
+        // call setup
         setup_fn.call0()?;
 
+        // start draw loop
+        loop {
+            {
+                let mut graphics = get_graphics_mut(module)?;
+                if !graphics.surface.poll_events() {
+                    break;
+                }
+                graphics.begin_draw()?;
+            }
+
+            draw_fn
+                .call0()
+                .map_err(|e| PyRuntimeError::new_err(format!("{e}")))?;
+
+            get_graphics(module)?.end_draw()?;
+        }
+
         Ok(())
     })
 }