Use bytemuck for the push constants (#668)

* Use bytemuck for the push constants

* Use the released version of bytemuck

Lokathor/bytemuck#69 landed

That version also works 🎉

examples/runners/wgpu/Cargo.toml

@@ -23,6 +23,7 @@ wgpu = "0.7.0"
 winit = { version = "0.24", features = ["web-sys"] }
 clap = "3.0.0-beta.2"
 strum = { version = "0.20", default_features = false, features = ["derive"] }
+bytemuck = "1.6.3"
 
 [target.'cfg(not(any(target_os = "android", target_arch = "wasm32")))'.dependencies]
 spirv-builder = { path = "../../../crates/spirv-builder", default-features = false, features = ["watch"] }

examples/runners/wgpu/src/graphics.rs

@@ -23,10 +23,6 @@ mod shaders {
     include!(concat!(env!("OUT_DIR"), "/entry_points.rs"));
 }
 
-unsafe fn any_as_u8_slice<T: Sized>(p: &T) -> &[u8] {
-    ::std::slice::from_raw_parts((p as *const T) as *const u8, ::std::mem::size_of::<T>())
-}
-
 fn mouse_button_index(button: MouseButton) -> usize {
     match button {
         MouseButton::Left => 0,
@@ -198,9 +194,11 @@ async fn run(
                         };
 
                         rpass.set_pipeline(render_pipeline);
-                        rpass.set_push_constants(wgpu::ShaderStage::all(), 0, unsafe {
-                            any_as_u8_slice(&push_constants)
-                        });
+                        rpass.set_push_constants(
+                            wgpu::ShaderStage::all(),
+                            0,
+                            bytemuck::bytes_of(&push_constants),
+                        );
                         rpass.draw(0..3, 0..1);
                     }
 

examples/shaders/shared/Cargo.toml

@@ -8,3 +8,4 @@ publish = false
 
 [dependencies]
 spirv-std = { path = "../../../crates/spirv-std", features = ["glam"] }
+bytemuck = { version = "1.6.3", features = ["derive"] }

examples/shaders/shared/src/lib.rs

@@ -12,7 +12,9 @@ pub use spirv_std::glam;
 #[cfg(target_arch = "spirv")]
 use spirv_std::num_traits::Float;
 
-#[derive(Copy, Clone)]
+use bytemuck::{Pod, Zeroable};
+
+#[derive(Copy, Clone, Pod, Zeroable)]
 #[repr(C)]
 pub struct ShaderConstants {
     pub width: u32,