Pass sketch file based on python file being executed

Author: catilac

crates/processing_pyo3/examples/rectangle.py

@@ -7,8 +7,8 @@ def draw():
     background(220)
 
     fill(255, 0, 100)
-    stroke(0)
-    stroke_weight(2)
+    stroke(1)
+    stroke_weight(25)
     rect(100, 100, 200, 150)
 
 # TODO: this should happen implicitly on module load somehow

crates/processing_pyo3/src/graphics.rs

@@ -120,17 +120,18 @@ impl Drop for Graphics {
 #[pymethods]
 impl Graphics {
     #[new]
-    pub fn new(width: u32, height: u32, asset_path: &str) -> PyResult<Self> {
+    pub fn new(
+        width: u32,
+        height: u32,
+        asset_path: &str,
+        sketch_root_path: &str,
+    ) -> PyResult<Self> {
         let glfw_ctx =
             GlfwContext::new(width, height).map_err(|e| PyRuntimeError::new_err(format!("{e}")))?;
 
         let mut config = Config::new();
         config.set(ConfigKey::AssetRootPath, asset_path.to_string());
-        config.set(
-            // TODO: this needs to be handed to us by python
-            ConfigKey::SketchRootPath,
-            "/home/moon/Code/libprocessing/crates/processing_pyo3/examples".to_string(),
-        );
+        config.set(ConfigKey::SketchRootPath, sketch_root_path.to_string());
         init(config).map_err(|e| PyRuntimeError::new_err(format!("{e}")))?;
 
         let surface = glfw_ctx
@@ -160,10 +161,6 @@ impl Graphics {
                 source: "".to_string(),
             }),
         }
-
-        // Ok(Sketch {
-        //     source: sketch.unwrap().source,
-        // })
     }
 
     pub fn background(&self, args: Vec<f32>) -> PyResult<()> {

crates/processing_pyo3/src/lib.rs

@@ -71,11 +71,30 @@ fn get_asset_root() -> PyResult<String> {
     })
 }
 
+fn get_sketch_root() -> PyResult<String> {
+    Python::attach(|py| {
+        let sys = PyModule::import(py, "sys")?;
+        let argv: Vec<String> = sys.getattr("argv")?.extract()?;
+        let filename: &str = argv[0].as_str();
+        let os = PyModule::import(py, "os")?;
+        let path = os.getattr("path")?;
+        let dirname = path.getattr("dirname")?.call1((filename,))?;
+        let abspath = path.getattr("abspath")?.call1((dirname,))?;
+        Ok(abspath.to_string())
+    })
+}
+
 #[pyfunction]
 #[pyo3(pass_module)]
 fn size(module: &Bound<'_, PyModule>, width: u32, height: u32) -> PyResult<()> {
     let asset_path: String = get_asset_root()?;
-    let graphics = Graphics::new(width, height, asset_path.as_str())?;
+    let sketch_root_path: String = get_sketch_root()?;
+    let graphics = Graphics::new(
+        width,
+        height,
+        asset_path.as_str(),
+        sketch_root_path.as_str(),
+    )?;
     module.setattr("_graphics", graphics)?;
     Ok(())
 }