Add custom asset path for python

Author: catilac

Cargo.toml

@@ -11,6 +11,7 @@ workspace = true
 default = ["wayland"]
 wayland = ["processing_render/wayland"]
 x11 = ["processing_render/x11"]
+python = ["processing_render/python"]
 
 [workspace]
 resolver = "3"

crates/processing_pyo3/Cargo.toml

@@ -17,7 +17,7 @@ x11 = ["processing/x11"]
 
 [dependencies]
 pyo3 = "0.27.0"
-processing = { workspace = true }
+processing = { workspace = true, features = ["python"] }
 bevy = { workspace = true }
 glfw = { version = "0.60.0"}
 

crates/processing_pyo3/src/graphics.rs

@@ -42,6 +42,7 @@ impl Graphics {
         let glfw_ctx =
             GlfwContext::new(width, height).map_err(|e| PyRuntimeError::new_err(format!("{e}")))?;
 
+        // TODO: pass in something to set the directory?
         init().map_err(|e| PyRuntimeError::new_err(format!("{e}")))?;
 
         let surface = glfw_ctx

crates/processing_render/Cargo.toml

@@ -10,6 +10,7 @@ workspace = true
 default = []
 wayland = ["bevy/wayland"]
 x11 = ["bevy/x11"]
+python = []
 
 [dependencies]
 bevy = { workspace = true }
@@ -32,4 +33,4 @@ windows = { version = "0.58", features = ["Win32_Foundation", "Win32_System_Libr
 wasm-bindgen = "0.2"
 wasm-bindgen-futures = "0.4"
 js-sys = "0.3"
-web-sys = { version = "0.3", features = ["Window", "Document", "HtmlCanvasElement"] }
+web-sys = { version = "0.3", features = ["Window", "Document", "HtmlCanvasElement"] }

crates/processing_render/src/image.rs

@@ -3,6 +3,8 @@
 //! It can be created from raw pixel data, loaded from disk, resized, and read back to CPU memory.
 use std::path::PathBuf;
 
+#[cfg(feature = "python")]
+use bevy::asset::{AssetPath, io::AssetSourceId};
 use bevy::{
     asset::{
         LoadState, RenderAssetUsages, handle_internal_asset_events, io::embedded::GetAssetServer,
@@ -138,6 +140,9 @@ pub fn from_handle(
 }
 
 pub fn load(In(path): In<PathBuf>, world: &mut World) -> Result<Entity> {
+    #[cfg(feature = "python")]
+    let path = AssetPath::from_path_buf(path).with_source(AssetSourceId::from("assets_directory"));
+
     let handle: Handle<bevy::image::Image> = world.get_asset_server().load(path);
     while let LoadState::Loading = world.get_asset_server().load_state(&handle) {
         world.run_system_once(handle_internal_asset_events).unwrap();

crates/processing_render/src/lib.rs

@@ -6,6 +6,8 @@ mod surface;
 
 use std::{cell::RefCell, num::NonZero, path::PathBuf, sync::OnceLock};
 
+#[cfg(feature = "python")]
+use bevy::asset::io::AssetSourceBuilder;
 #[cfg(not(target_arch = "wasm32"))]
 use bevy::log::tracing_subscriber;
 use bevy::{
@@ -226,6 +228,13 @@ fn create_app() -> App {
             ..default()
         });
 
+    #[cfg(feature = "python")]
+    app.register_asset_source(
+        "assets_directory",
+        // TODO: set this path to the directory containing the main sketch file
+        AssetSourceBuilder::platform_default("TODO/TODO/TODO/libprocessing/assets", None),
+    );
+
     app.add_plugins(plugins);
     app.add_plugins((ImagePlugin, GraphicsPlugin, SurfacePlugin));
     app.add_systems(First, (clear_transient_meshes, activate_cameras))