sprite: use bevy_transform types in sprite entities

crates/bevy_derive/src/render_resource.rs

@@ -9,6 +9,7 @@ pub fn derive_render_resource(input: TokenStream) -> TokenStream {
 
     let bevy_render_path: Path = get_path(&modules.bevy_render);
     let bevy_asset_path: Path = get_path(&modules.bevy_asset);
+    let bevy_core_path: Path = get_path(&modules.bevy_core);
     let struct_name = &ast.ident;
 
     TokenStream::from(quote! {
@@ -17,9 +18,11 @@ pub fn derive_render_resource(input: TokenStream) -> TokenStream {
                 Some(#bevy_render_path::render_resource::RenderResourceType::Buffer)
             }
             fn write_buffer_bytes(&self, buffer: &mut [u8]) {
+                use #bevy_core_path::bytes::Bytes;
                 self.write_bytes(buffer);
             }
             fn buffer_byte_len(&self) -> Option<usize> {
+                use #bevy_core_path::bytes::Bytes;
                 Some(self.byte_len())
             }
             fn texture(&self) -> Option<#bevy_asset_path::Handle<#bevy_render_path::texture::Texture>> {

crates/bevy_sprite/src/entity.rs

@@ -1,30 +1,58 @@
 use crate::{
-    render::SPRITE_PIPELINE_HANDLE, sprite::Sprite, ColorMaterial, Quad, TextureAtlas,
+    render::SPRITE_PIPELINE_HANDLE, sprite::Sprite, ColorMaterial, TextureAtlas,
     TextureAtlasSprite, QUAD_HANDLE, SPRITE_SHEET_PIPELINE_HANDLE,
 };
 use bevy_app::EntityArchetype;
 use bevy_asset::Handle;
-use bevy_render::{draw::Draw, mesh::Mesh, pipeline::{PipelineSpecialization, RenderPipelines, RenderPipeline, DynamicBinding}};
+use bevy_render::{
+    draw::Draw,
+    mesh::Mesh,
+    pipeline::{DynamicBinding, PipelineSpecialization, RenderPipeline, RenderPipelines},
+};
+use bevy_transform::prelude::{Rotation, Scale, Transform, Translation};
 
 #[derive(EntityArchetype)]
 pub struct SpriteEntity {
     pub sprite: Sprite,
-    pub quad: Quad,
     pub mesh: Handle<Mesh>, // TODO: maybe abstract this out
     pub material: Handle<ColorMaterial>,
     pub draw: Draw,
     pub render_pipelines: RenderPipelines,
+    pub transform: Transform,
+    pub translation: Translation,
+    pub rotation: Rotation,
+    pub scale: Scale,
 }
 
 impl Default for SpriteEntity {
     fn default() -> Self {
         Self {
             mesh: QUAD_HANDLE,
-            render_pipelines: RenderPipelines::from_handles(&[SPRITE_PIPELINE_HANDLE]),
+            render_pipelines: RenderPipelines::from_pipelines(vec![RenderPipeline::specialized(
+                SPRITE_PIPELINE_HANDLE,
+                PipelineSpecialization {
+                    dynamic_bindings: vec![
+                        // Transform
+                        DynamicBinding {
+                            bind_group: 2,
+                            binding: 0,
+                        },
+                        // Sprite
+                        DynamicBinding {
+                            bind_group: 2,
+                            binding: 1,
+                        },
+                    ],
+                    ..Default::default()
+                },
+            )]),
             sprite: Default::default(),
-            quad: Default::default(),
             material: Default::default(),
             draw: Default::default(),
+            transform: Default::default(),
+            translation: Default::default(),
+            rotation: Default::default(),
+            scale: Default::default(),
         }
     }
 }

crates/bevy_sprite/src/render/mod.rs

@@ -1,4 +1,4 @@
-use crate::{ColorMaterial, Quad, TextureAtlas, TextureAtlasSprite};
+use crate::{ColorMaterial, Quad, TextureAtlas, TextureAtlasSprite, Sprite};
 use bevy_asset::{Assets, Handle};
 use bevy_render::{
     base_render_graph,
@@ -111,6 +111,7 @@ pub fn build_sprite_pipeline(shaders: &mut Assets<Shader>) -> PipelineDescriptor
 pub mod node {
     pub const COLOR_MATERIAL: &'static str = "color_material";
     pub const QUAD: &'static str = "quad";
+    pub const SPRITE: &'static str = "sprite";
     pub const SPRITE_SHEET: &'static str = "sprite_sheet";
     pub const SPRITE_SHEET_SPRITE: &'static str = "sprite_sheet_sprite";
 }
@@ -131,6 +132,9 @@ impl SpriteRenderGraphBuilder for RenderGraph {
         self.add_system_node(node::QUAD, RenderResourcesNode::<Quad>::new(false));
         self.add_node_edge(node::QUAD, base_render_graph::node::MAIN_PASS)
             .unwrap();
+        self.add_system_node(node::SPRITE, RenderResourcesNode::<Sprite>::new(true));
+        self.add_node_edge(node::SPRITE, base_render_graph::node::MAIN_PASS)
+            .unwrap();
 
         self.add_system_node(
             node::SPRITE_SHEET,

crates/bevy_sprite/src/render/sprite.frag

@@ -4,13 +4,13 @@ layout(location = 0) in vec2 v_Uv;
 
 layout(location = 0) out vec4 o_Target;
 
-layout(set = 2, binding = 0) uniform ColorMaterial_color {
+layout(set = 1, binding = 0) uniform ColorMaterial_color {
     vec4 Color;
 };
 
 # ifdef COLORMATERIAL_TEXTURE 
-layout(set = 3, binding = 0) uniform texture2D ColorMaterial_texture;
+layout(set = 1, binding = 1) uniform texture2D ColorMaterial_texture;
-layout(set = 3, binding = 1) uniform sampler ColorMaterial_texture_sampler;
+layout(set = 1, binding = 2) uniform sampler ColorMaterial_texture_sampler;
 # endif
 
 void main() {

crates/bevy_sprite/src/render/sprite.vert

@@ -10,15 +10,15 @@ layout(set = 0, binding = 0) uniform Camera2d {
     mat4 ViewProj;
 };
 
-layout(set = 1, binding = 0) uniform Quad {
+layout(set = 2, binding = 0) uniform Transform {
-    vec2 Quad_Position;
+    mat4 Model;
-    vec2 Quad_Size;
+};
-    float Quad_ZIndex;
+layout(set = 2, binding = 1) uniform Sprite {
+    vec2 Sprite_size;
 };
 
 void main() {
     v_Uv = Vertex_Uv;
-    vec3 position = Vertex_Position * vec3(Quad_Size, 0.0);
+    vec3 position = Vertex_Position * vec3(Sprite_size, 1.0);
-    position = position + vec3(Quad_Position, Quad_ZIndex);
+    gl_Position = ViewProj * Model * vec4(position, 1.0);
-    gl_Position = ViewProj * vec4(position, 1.0);
 }

crates/bevy_sprite/src/sprite.rs

@@ -1,34 +1,35 @@
-use crate::{ColorMaterial, Quad};
+use crate::ColorMaterial;
 use bevy_asset::{Assets, Handle};
-use bevy_render::texture::Texture;
+use bevy_core::bytes::Byteable;
+use bevy_render::{
+    render_resource::{RenderResource, RenderResources},
+    texture::Texture,
+};
 pub use legion::prelude::*;
+use glam::Vec2;
+
+#[repr(C)]
+#[derive(Default, RenderResources, RenderResource)]
+#[render_resources(from_self)]
 pub struct Sprite {
-    pub scale: f32,
+    size: Vec2,
 }
 
-impl Default for Sprite {
+// SAFE: sprite is repr(C) and only consists of byteables
-    fn default() -> Self {
+unsafe impl Byteable for Sprite {}
-        Sprite { scale: 1.0 }
-    }
-}
 
+// TODO: port to system fn
 pub fn sprite_system() -> Box<dyn Schedulable> {
     SystemBuilder::new("sprite_system")
         .read_resource::<Assets<ColorMaterial>>()
         .read_resource::<Assets<Texture>>()
-        .with_query(
+        .with_query(<(Write<Sprite>, Read<Handle<ColorMaterial>>)>::query())
-            <(Read<Sprite>, Read<Handle<ColorMaterial>>, Write<Quad>)>::query().filter(
-                changed::<Sprite>() | changed::<Quad>() | changed::<Handle<ColorMaterial>>(),
-            ),
-        )
         .build(|_, world, (materials, textures), query| {
-            for (sprite, handle, mut rect) in query.iter_mut(world) {
+            for (mut sprite, handle) in query.iter_mut(world) {
                 let material = materials.get(&handle).unwrap();
                 if let Some(texture_handle) = material.texture {
                     if let Some(texture) = textures.get(&texture_handle) {
-                        let aspect = texture.aspect();
+                        sprite.size = texture.size;
-                        *rect.size.x_mut() = texture.size.x() * sprite.scale;
-                        *rect.size.y_mut() = rect.size.x() * aspect;
                     }
                 }
             }

examples/2d/texture_atlas.rs

@@ -81,11 +81,7 @@ fn load_atlas(
             // draw the atlas itself
             .add_entity(SpriteEntity {
                 material: materials.add(texture_atlas_texture.into()),
-                quad: Quad {
+                translation: Vec3::new(-300.0, 0., 0.0).into(),
-                    position: Vec2::new(-300.0, 0.),
-                    ..Default::default()
-                },
-                sprite: Sprite { scale: 1.0 },
                 ..Default::default()
             });
 

examples/ui/font_atlas_debug.rs

@@ -42,11 +42,7 @@ fn atlas_render_system(
             let texture_atlas = texture_atlases.get(&font_atlas.texture_atlas).unwrap();
             command_buffer.build().add_entity(SpriteEntity {
                 material: materials.add(texture_atlas.texture.into()),
-                quad: Quad {
+                translation: Vec3::new(-300.0, 0., 0.0).into(),
-                    position: Vec2::new(-300.0, 0.),
-                    ..Default::default()
-                },
-                sprite: Sprite { scale: 1.0 },
                 ..Default::default()
             });
             break;