shader imports figured out

Dimchikkk · Nov 2, 2023 · a0307d3 · a0307d3
1 parent 07c5001
commit a0307d3
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/assets/shaders/test.wgsl b/assets/shaders/test.wgsl
@@ -1,5 +1,6 @@
 #import bevy_sprite::mesh2d_view_bindings  globals
 #import bevy_sprite::mesh2d_vertex_output  MeshVertexOutput
+#import test_export
 
 struct CustomMaterial {
     color: vec4<f32>,
@@ -20,7 +21,9 @@ fn fragment(
 let t_1 = sin(globals.time*2.0)*0.5+0.5;
 let t_2 = cos(globals.time*2.0);
 
-let color: vec4<f32> = vec4(t_1, t_2, t_1, 1.0);
+var color: vec4<f32> = vec4(t_1, t_2, t_1, 0.0);
+
+color.a = test_export::i_say_one();
 
     return color * textureSample(base_color_texture, base_color_sampler, mesh.uv);
 }
diff --git a/assets/shaders/test_export.wgsl b/assets/shaders/test_export.wgsl
@@ -0,0 +1,5 @@
+#define_import_path test_export
+
+fn i_say_one() -> f32 {
+    return 1.0;
+  }
diff --git a/examples/shader_2d.rs b/examples/shader_2d.rs
@@ -1,10 +1,7 @@
 use bevy::{
     prelude::*,
     reflect::{TypePath, TypeUuid},
-    render::{
-        render_resource::{AsBindGroup, ShaderRef},
-        texture::DEFAULT_IMAGE_HANDLE,
-    },
+    render::render_resource::{AsBindGroup, ShaderRef},
     sprite::{Material2d, MaterialMesh2dBundle},
     window::PrimaryWindow,
 };
@@ -43,9 +40,13 @@ fn setup(
     mut commands: Commands,
     mut materials: ResMut<Assets<CustomMaterial>>,
     mut meshes: ResMut<Assets<Mesh>>,
+    asset_server: Res<AssetServer>,
 ) {
     commands.spawn(Camera2dBundle::default());
 
+    // Shaders need to be in loaded to be accessed
+    commands.spawn(asset_server.load("shaders/test_export.wgsl") as Handle<Shader>);
+
     // Sprite editor
     let editor = commands
         .spawn((CosmicEditBundle {