derive: break up derive crate into modules
This commit is contained in:
Carter Anderson
parent
7019ff015b
commit
8c196139d4
20
crates/bevy_derive/src/app_plugin.rs
Normal file
20
crates/bevy_derive/src/app_plugin.rs
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
use quote::quote;
|
||||||
|
use syn::{parse_macro_input, DeriveInput};
|
||||||
|
use proc_macro::TokenStream;
|
||||||
|
|
||||||
|
pub fn derive_app_plugin(input: TokenStream) -> TokenStream {
|
||||||
|
let ast = parse_macro_input!(input as DeriveInput);
|
||||||
|
let struct_name = &ast.ident;
|
||||||
|
|
||||||
|
TokenStream::from(quote! {
|
||||||
|
#[no_mangle]
|
||||||
|
pub extern "C" fn _create_plugin() -> *mut bevy::app::AppPlugin {
|
||||||
|
// TODO: without this the assembly does nothing. why is that the case?
|
||||||
|
print!("");
|
||||||
|
// make sure the constructor is the correct type.
|
||||||
|
let object = #struct_name {};
|
||||||
|
let boxed = Box::new(object);
|
||||||
|
Box::into_raw(boxed)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
44
crates/bevy_derive/src/bytes.rs
Normal file
44
crates/bevy_derive/src/bytes.rs
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
use crate::modules::{get_modules, get_path};
|
||||||
|
use proc_macro::TokenStream;
|
||||||
|
use quote::quote;
|
||||||
|
use syn::{parse_macro_input, Data, DataStruct, DeriveInput, Fields};
|
||||||
|
|
||||||
|
pub fn derive_bytes(input: TokenStream) -> TokenStream {
|
||||||
|
let ast = parse_macro_input!(input as DeriveInput);
|
||||||
|
let fields = match &ast.data {
|
||||||
|
Data::Struct(DataStruct {
|
||||||
|
fields: Fields::Named(fields),
|
||||||
|
..
|
||||||
|
}) => &fields.named,
|
||||||
|
_ => panic!("expected a struct with named fields"),
|
||||||
|
};
|
||||||
|
|
||||||
|
let modules = get_modules(&ast);
|
||||||
|
let bevy_core_path = get_path(&modules.bevy_core);
|
||||||
|
|
||||||
|
let fields = fields
|
||||||
|
.iter()
|
||||||
|
.map(|field| field.ident.as_ref().unwrap())
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
let generics = ast.generics;
|
||||||
|
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
||||||
|
|
||||||
|
let struct_name = &ast.ident;
|
||||||
|
|
||||||
|
TokenStream::from(quote! {
|
||||||
|
impl #impl_generics #bevy_core_path::bytes::Bytes for #struct_name#ty_generics {
|
||||||
|
fn write_bytes(&self, buffer: &mut [u8]) {
|
||||||
|
let mut offset: usize = 0;
|
||||||
|
#(let byte_len = self.#fields.byte_len();
|
||||||
|
self.#fields.write_bytes(&mut buffer[offset..(offset + byte_len)]);
|
||||||
|
offset += byte_len;)*
|
||||||
|
}
|
||||||
|
fn byte_len(&self) -> usize {
|
||||||
|
let mut byte_len: usize = 0;
|
||||||
|
#(byte_len += self.#fields.byte_len();)*
|
||||||
|
byte_len
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
73
crates/bevy_derive/src/entity_archetype.rs
Normal file
73
crates/bevy_derive/src/entity_archetype.rs
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
use crate::modules::{get_modules, get_path};
|
||||||
|
use darling::FromMeta;
|
||||||
|
use proc_macro::TokenStream;
|
||||||
|
use quote::quote;
|
||||||
|
use syn::{parse_macro_input, Data, DataStruct, DeriveInput, Fields, Ident};
|
||||||
|
|
||||||
|
#[derive(FromMeta, Debug, Default)]
|
||||||
|
struct EntityArchetypeAttributeArgs {
|
||||||
|
#[darling(default)]
|
||||||
|
pub tag: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn derive_entity_archetype(input: TokenStream) -> TokenStream {
|
||||||
|
let ast = parse_macro_input!(input as DeriveInput);
|
||||||
|
let fields = match &ast.data {
|
||||||
|
Data::Struct(DataStruct {
|
||||||
|
fields: Fields::Named(fields),
|
||||||
|
..
|
||||||
|
}) => &fields.named,
|
||||||
|
_ => panic!("expected a struct with named fields"),
|
||||||
|
};
|
||||||
|
|
||||||
|
let modules = get_modules(&ast);
|
||||||
|
let bevy_app_path = get_path(&modules.bevy_app);
|
||||||
|
let legion_path = get_path(&modules.legion);
|
||||||
|
|
||||||
|
let tag_fields = fields
|
||||||
|
.iter()
|
||||||
|
.filter(|f| {
|
||||||
|
f.attrs
|
||||||
|
.iter()
|
||||||
|
.find(|a| a.path.get_ident().as_ref().unwrap().to_string() == "tag")
|
||||||
|
.is_some()
|
||||||
|
})
|
||||||
|
.map(|field| field.ident.as_ref().unwrap())
|
||||||
|
.collect::<Vec<&Ident>>();
|
||||||
|
|
||||||
|
let component_fields = fields
|
||||||
|
.iter()
|
||||||
|
.filter(|f| {
|
||||||
|
f.attrs
|
||||||
|
.iter()
|
||||||
|
.find(|a| a.path.get_ident().as_ref().unwrap().to_string() == "tag")
|
||||||
|
.is_none()
|
||||||
|
})
|
||||||
|
.map(|field| field.ident.as_ref().unwrap())
|
||||||
|
.collect::<Vec<&Ident>>();
|
||||||
|
|
||||||
|
let generics = ast.generics;
|
||||||
|
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
||||||
|
|
||||||
|
let struct_name = &ast.ident;
|
||||||
|
|
||||||
|
TokenStream::from(quote! {
|
||||||
|
impl #impl_generics #bevy_app_path::EntityArchetype for #struct_name#ty_generics {
|
||||||
|
fn insert(self, world: &mut #legion_path::prelude::World) -> #legion_path::prelude::Entity {
|
||||||
|
*world.insert((#(self.#tag_fields,)*),
|
||||||
|
vec![(
|
||||||
|
#(self.#component_fields,)*
|
||||||
|
)
|
||||||
|
]).first().unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn insert_command_buffer(self, command_buffer: &mut #legion_path::prelude::CommandBuffer) -> #legion_path::prelude::Entity {
|
||||||
|
*command_buffer.insert((#(self.#tag_fields,)*),
|
||||||
|
vec![(
|
||||||
|
#(self.#component_fields,)*
|
||||||
|
)
|
||||||
|
]).first().unwrap()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
@ -1,485 +1,40 @@
|
|||||||
extern crate proc_macro;
|
extern crate proc_macro;
|
||||||
|
|
||||||
|
mod entity_archetype;
|
||||||
mod modules;
|
mod modules;
|
||||||
|
mod uniforms;
|
||||||
|
mod bytes;
|
||||||
|
mod resource;
|
||||||
|
mod app_plugin;
|
||||||
|
|
||||||
use darling::FromMeta;
|
|
||||||
use inflector::Inflector;
|
|
||||||
use modules::{get_modules, get_path};
|
|
||||||
use proc_macro::TokenStream;
|
use proc_macro::TokenStream;
|
||||||
use quote::{format_ident, quote};
|
|
||||||
use syn::{parse_macro_input, Data, DataStruct, DeriveInput, Field, Fields, Ident, Path};
|
|
||||||
|
|
||||||
#[derive(FromMeta, Debug, Default)]
|
|
||||||
struct EntityArchetypeAttributeArgs {
|
|
||||||
#[darling(default)]
|
|
||||||
pub tag: Option<bool>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[proc_macro_derive(FromResources, attributes(module))]
|
#[proc_macro_derive(FromResources, attributes(module))]
|
||||||
pub fn derive_resource(input: TokenStream) -> TokenStream {
|
pub fn derive_resource(input: TokenStream) -> TokenStream {
|
||||||
let ast = parse_macro_input!(input as DeriveInput);
|
resource::derive_resource(input)
|
||||||
let fields = match &ast.data {
|
|
||||||
Data::Struct(DataStruct {
|
|
||||||
fields: Fields::Named(fields),
|
|
||||||
..
|
|
||||||
}) => &fields.named,
|
|
||||||
_ => panic!("expected a struct with named fields"),
|
|
||||||
};
|
|
||||||
|
|
||||||
let modules = get_modules(&ast);
|
|
||||||
let bevy_app_path = get_path(&modules.bevy_app);
|
|
||||||
|
|
||||||
let field_types = fields.iter().map(|field| &field.ty);
|
|
||||||
|
|
||||||
let fields = fields.iter().map(|field| field.ident.as_ref().unwrap());
|
|
||||||
|
|
||||||
let generics = ast.generics;
|
|
||||||
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
|
||||||
|
|
||||||
let struct_name = &ast.ident;
|
|
||||||
|
|
||||||
TokenStream::from(quote! {
|
|
||||||
impl #impl_generics #bevy_app_path::FromResources for #struct_name#ty_generics {
|
|
||||||
fn from_resources(resources: &Resources) -> Self {
|
|
||||||
use #bevy_app_path::FromResources;
|
|
||||||
#struct_name {
|
|
||||||
#(#fields: <#field_types>::from_resources(resources),)*
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[proc_macro_derive(Bytes, attributes(module))]
|
#[proc_macro_derive(Bytes, attributes(module))]
|
||||||
pub fn derive_bytes(input: TokenStream) -> TokenStream {
|
pub fn derive_bytes(input: TokenStream) -> TokenStream {
|
||||||
let ast = parse_macro_input!(input as DeriveInput);
|
bytes::derive_bytes(input)
|
||||||
let fields = match &ast.data {
|
|
||||||
Data::Struct(DataStruct {
|
|
||||||
fields: Fields::Named(fields),
|
|
||||||
..
|
|
||||||
}) => &fields.named,
|
|
||||||
_ => panic!("expected a struct with named fields"),
|
|
||||||
};
|
|
||||||
|
|
||||||
let modules = get_modules(&ast);
|
|
||||||
let bevy_core_path = get_path(&modules.bevy_core);
|
|
||||||
|
|
||||||
let fields = fields
|
|
||||||
.iter()
|
|
||||||
.map(|field| field.ident.as_ref().unwrap())
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
let generics = ast.generics;
|
|
||||||
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
|
||||||
|
|
||||||
let struct_name = &ast.ident;
|
|
||||||
|
|
||||||
TokenStream::from(quote! {
|
|
||||||
impl #impl_generics #bevy_core_path::bytes::Bytes for #struct_name#ty_generics {
|
|
||||||
fn write_bytes(&self, buffer: &mut [u8]) {
|
|
||||||
let mut offset: usize = 0;
|
|
||||||
#(let byte_len = self.#fields.byte_len();
|
|
||||||
self.#fields.write_bytes(&mut buffer[offset..(offset + byte_len)]);
|
|
||||||
offset += byte_len;)*
|
|
||||||
}
|
|
||||||
fn byte_len(&self) -> usize {
|
|
||||||
let mut byte_len: usize = 0;
|
|
||||||
#(byte_len += self.#fields.byte_len();)*
|
|
||||||
byte_len
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[proc_macro_derive(Uniform, attributes(uniform, module))]
|
#[proc_macro_derive(Uniform, attributes(uniform, module))]
|
||||||
pub fn derive_uniform(input: TokenStream) -> TokenStream {
|
pub fn derive_uniform(input: TokenStream) -> TokenStream {
|
||||||
let ast = parse_macro_input!(input as DeriveInput);
|
uniforms::derive_uniform(input)
|
||||||
|
|
||||||
let modules = get_modules(&ast);
|
|
||||||
let bevy_asset_path = get_path(&modules.bevy_asset);
|
|
||||||
let bevy_core_path = get_path(&modules.bevy_core);
|
|
||||||
let bevy_render_path = get_path(&modules.bevy_render);
|
|
||||||
|
|
||||||
let generics = ast.generics;
|
|
||||||
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
|
||||||
|
|
||||||
let struct_name = &ast.ident;
|
|
||||||
let struct_name_string = struct_name.to_string();
|
|
||||||
|
|
||||||
TokenStream::from(quote! {
|
|
||||||
impl #impl_generics #bevy_render_path::shader::AsUniforms for #struct_name#ty_generics {
|
|
||||||
fn get_field_infos() -> &'static [#bevy_render_path::shader::FieldInfo] {
|
|
||||||
static FIELD_INFOS: &[#bevy_render_path::shader::FieldInfo] = &[
|
|
||||||
#bevy_render_path::shader::FieldInfo {
|
|
||||||
name: #struct_name_string,
|
|
||||||
uniform_name: #struct_name_string,
|
|
||||||
texture_name: #struct_name_string,
|
|
||||||
sampler_name: #struct_name_string,
|
|
||||||
is_instanceable: false,
|
|
||||||
}
|
|
||||||
];
|
|
||||||
&FIELD_INFOS
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_field_bind_type(&self, name: &str) -> Option<#bevy_render_path::shader::FieldBindType> {
|
|
||||||
use #bevy_render_path::shader::GetFieldBindType;
|
|
||||||
match name {
|
|
||||||
#struct_name_string => self.get_bind_type(),
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn write_uniform_bytes(&self, name: &str, buffer: &mut [u8]) {
|
|
||||||
use #bevy_core_path::bytes::Bytes;
|
|
||||||
match name {
|
|
||||||
#struct_name_string => self.write_bytes(buffer),
|
|
||||||
_ => {},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn uniform_byte_len(&self, name: &str) -> usize {
|
|
||||||
use #bevy_core_path::bytes::Bytes;
|
|
||||||
match name {
|
|
||||||
#struct_name_string => self.byte_len(),
|
|
||||||
_ => 0,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_uniform_texture(&self, name: &str) -> Option<#bevy_asset_path::Handle<#bevy_render_path::texture::Texture>> {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: move this to field_info and add has_shader_def(&self, &str) -> bool
|
|
||||||
// TODO: this will be very allocation heavy. find a way to either make this allocation free
|
|
||||||
// or alternatively only run it when the shader_defs have changed
|
|
||||||
fn get_shader_defs(&self) -> Option<Vec<String>> {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_vertex_buffer_descriptor() -> Option<&'static #bevy_render_path::pipeline::VertexBufferDescriptor> {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[proc_macro_derive(EntityArchetype, attributes(tag, module))]
|
#[proc_macro_derive(EntityArchetype, attributes(tag, module))]
|
||||||
pub fn derive_entity_archetype(input: TokenStream) -> TokenStream {
|
pub fn derive_entity_archetype(input: TokenStream) -> TokenStream {
|
||||||
let ast = parse_macro_input!(input as DeriveInput);
|
entity_archetype::derive_entity_archetype(input)
|
||||||
let fields = match &ast.data {
|
|
||||||
Data::Struct(DataStruct {
|
|
||||||
fields: Fields::Named(fields),
|
|
||||||
..
|
|
||||||
}) => &fields.named,
|
|
||||||
_ => panic!("expected a struct with named fields"),
|
|
||||||
};
|
|
||||||
|
|
||||||
let modules = get_modules(&ast);
|
|
||||||
let bevy_app_path = get_path(&modules.bevy_app);
|
|
||||||
let legion_path = get_path(&modules.legion);
|
|
||||||
|
|
||||||
let tag_fields = fields
|
|
||||||
.iter()
|
|
||||||
.filter(|f| {
|
|
||||||
f.attrs
|
|
||||||
.iter()
|
|
||||||
.find(|a| a.path.get_ident().as_ref().unwrap().to_string() == "tag")
|
|
||||||
.is_some()
|
|
||||||
})
|
|
||||||
.map(|field| field.ident.as_ref().unwrap())
|
|
||||||
.collect::<Vec<&Ident>>();
|
|
||||||
|
|
||||||
let component_fields = fields
|
|
||||||
.iter()
|
|
||||||
.filter(|f| {
|
|
||||||
f.attrs
|
|
||||||
.iter()
|
|
||||||
.find(|a| a.path.get_ident().as_ref().unwrap().to_string() == "tag")
|
|
||||||
.is_none()
|
|
||||||
})
|
|
||||||
.map(|field| field.ident.as_ref().unwrap())
|
|
||||||
.collect::<Vec<&Ident>>();
|
|
||||||
|
|
||||||
let generics = ast.generics;
|
|
||||||
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
|
||||||
|
|
||||||
let struct_name = &ast.ident;
|
|
||||||
|
|
||||||
TokenStream::from(quote! {
|
|
||||||
impl #impl_generics #bevy_app_path::EntityArchetype for #struct_name#ty_generics {
|
|
||||||
fn insert(self, world: &mut #legion_path::prelude::World) -> #legion_path::prelude::Entity {
|
|
||||||
*world.insert((#(self.#tag_fields,)*),
|
|
||||||
vec![(
|
|
||||||
#(self.#component_fields,)*
|
|
||||||
)
|
|
||||||
]).first().unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn insert_command_buffer(self, command_buffer: &mut #legion_path::prelude::CommandBuffer) -> #legion_path::prelude::Entity {
|
|
||||||
*command_buffer.insert((#(self.#tag_fields,)*),
|
|
||||||
vec![(
|
|
||||||
#(self.#component_fields,)*
|
|
||||||
)
|
|
||||||
]).first().unwrap()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: ensure shader_def and instance/vertex are mutually exclusive
|
|
||||||
#[derive(FromMeta, Debug, Default)]
|
|
||||||
struct UniformAttributeArgs {
|
|
||||||
#[darling(default)]
|
|
||||||
pub ignore: Option<bool>,
|
|
||||||
#[darling(default)]
|
|
||||||
pub shader_def: Option<bool>,
|
|
||||||
#[darling(default)]
|
|
||||||
pub instance: Option<bool>,
|
|
||||||
#[darling(default)]
|
|
||||||
pub vertex: Option<bool>,
|
|
||||||
#[darling(default)]
|
|
||||||
pub buffer: Option<bool>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Default)]
|
|
||||||
struct UniformAttributes {
|
|
||||||
pub ignore: bool,
|
|
||||||
pub shader_def: bool,
|
|
||||||
pub instance: bool,
|
|
||||||
pub vertex: bool,
|
|
||||||
pub buffer: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<UniformAttributeArgs> for UniformAttributes {
|
|
||||||
fn from(args: UniformAttributeArgs) -> Self {
|
|
||||||
UniformAttributes {
|
|
||||||
ignore: args.ignore.unwrap_or(false),
|
|
||||||
shader_def: args.shader_def.unwrap_or(false),
|
|
||||||
instance: args.instance.unwrap_or(false),
|
|
||||||
vertex: args.vertex.unwrap_or(false),
|
|
||||||
buffer: args.buffer.unwrap_or(false),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static UNIFORM_ATTRIBUTE_NAME: &'static str = "uniform";
|
|
||||||
#[proc_macro_derive(Uniforms, attributes(uniform, module))]
|
#[proc_macro_derive(Uniforms, attributes(uniform, module))]
|
||||||
pub fn derive_uniforms(input: TokenStream) -> TokenStream {
|
pub fn derive_uniforms(input: TokenStream) -> TokenStream {
|
||||||
let ast = parse_macro_input!(input as DeriveInput);
|
uniforms::derive_uniforms(input)
|
||||||
let modules = get_modules(&ast);
|
|
||||||
|
|
||||||
let bevy_render_path: Path = get_path(&modules.bevy_render);
|
|
||||||
let bevy_core_path: Path = get_path(&modules.bevy_core);
|
|
||||||
let bevy_asset_path: Path = get_path(&modules.bevy_asset);
|
|
||||||
|
|
||||||
let fields = match &ast.data {
|
|
||||||
Data::Struct(DataStruct {
|
|
||||||
fields: Fields::Named(fields),
|
|
||||||
..
|
|
||||||
}) => &fields.named,
|
|
||||||
_ => panic!("expected a struct with named fields"),
|
|
||||||
};
|
|
||||||
|
|
||||||
let field_attributes = fields
|
|
||||||
.iter()
|
|
||||||
.map(|f| {
|
|
||||||
(
|
|
||||||
f,
|
|
||||||
f.attrs
|
|
||||||
.iter()
|
|
||||||
.find(|a| {
|
|
||||||
a.path.get_ident().as_ref().unwrap().to_string() == UNIFORM_ATTRIBUTE_NAME
|
|
||||||
})
|
|
||||||
.map(|a| {
|
|
||||||
UniformAttributes::from(
|
|
||||||
UniformAttributeArgs::from_meta(&a.parse_meta().unwrap())
|
|
||||||
.unwrap_or_else(|_err| UniformAttributeArgs::default()),
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.unwrap_or_else(|| UniformAttributes::default()),
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.collect::<Vec<(&Field, UniformAttributes)>>();
|
|
||||||
|
|
||||||
let struct_name = &ast.ident;
|
|
||||||
|
|
||||||
let mut active_uniform_field_names = Vec::new();
|
|
||||||
let mut active_uniform_field_name_strings = Vec::new();
|
|
||||||
let mut uniform_name_strings = Vec::new();
|
|
||||||
let mut texture_and_sampler_name_strings = Vec::new();
|
|
||||||
let mut texture_and_sampler_name_idents = Vec::new();
|
|
||||||
let mut field_infos = Vec::new();
|
|
||||||
|
|
||||||
let mut vertex_buffer_field_names_pascal = Vec::new();
|
|
||||||
let mut vertex_buffer_field_types = Vec::new();
|
|
||||||
|
|
||||||
let mut shader_def_field_names = Vec::new();
|
|
||||||
let mut shader_def_field_names_screaming_snake = Vec::new();
|
|
||||||
|
|
||||||
for (f, attrs) in field_attributes.iter() {
|
|
||||||
let field_name = f.ident.as_ref().unwrap().to_string();
|
|
||||||
if !attrs.ignore {
|
|
||||||
active_uniform_field_names.push(&f.ident);
|
|
||||||
active_uniform_field_name_strings.push(field_name.clone());
|
|
||||||
let uniform = format!("{}_{}", struct_name, field_name);
|
|
||||||
let texture = format!("{}", uniform);
|
|
||||||
let sampler = format!("{}_sampler", uniform);
|
|
||||||
uniform_name_strings.push(uniform.clone());
|
|
||||||
texture_and_sampler_name_strings.push(texture.clone());
|
|
||||||
texture_and_sampler_name_strings.push(sampler.clone());
|
|
||||||
texture_and_sampler_name_idents.push(f.ident.clone());
|
|
||||||
texture_and_sampler_name_idents.push(f.ident.clone());
|
|
||||||
let is_instanceable = attrs.instance;
|
|
||||||
field_infos.push(quote!(#bevy_render_path::shader::FieldInfo {
|
|
||||||
name: #field_name,
|
|
||||||
uniform_name: #uniform,
|
|
||||||
texture_name: #texture,
|
|
||||||
sampler_name: #sampler,
|
|
||||||
is_instanceable: #is_instanceable,
|
|
||||||
}));
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
if attrs.shader_def {
|
|
||||||
shader_def_field_names.push(&f.ident);
|
|
||||||
shader_def_field_names_screaming_snake.push(field_name.to_screaming_snake_case())
|
|
||||||
}
|
|
||||||
|
|
||||||
if attrs.instance || attrs.vertex {
|
|
||||||
vertex_buffer_field_types.push(&f.ty);
|
|
||||||
let pascal_field = f.ident.as_ref().unwrap().to_string().to_pascal_case();
|
|
||||||
vertex_buffer_field_names_pascal.push(if attrs.instance {
|
|
||||||
format!("I_{}_{}", struct_name, pascal_field)
|
|
||||||
} else {
|
|
||||||
format!("{}_{}", struct_name, pascal_field)
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let struct_name_string = struct_name.to_string();
|
|
||||||
let struct_name_uppercase = struct_name_string.to_uppercase();
|
|
||||||
let field_infos_ident = format_ident!("{}_FIELD_INFO", struct_name_uppercase);
|
|
||||||
let vertex_buffer_descriptor_ident =
|
|
||||||
format_ident!("{}_VERTEX_BUFFER_DESCRIPTOR", struct_name_uppercase);
|
|
||||||
|
|
||||||
TokenStream::from(quote! {
|
|
||||||
static #field_infos_ident: &[#bevy_render_path::shader::FieldInfo] = &[
|
|
||||||
#(#field_infos,)*
|
|
||||||
];
|
|
||||||
|
|
||||||
static #vertex_buffer_descriptor_ident: #bevy_render_path::once_cell::sync::Lazy<#bevy_render_path::pipeline::VertexBufferDescriptor> =
|
|
||||||
#bevy_render_path::once_cell::sync::Lazy::new(|| {
|
|
||||||
use #bevy_render_path::pipeline::{VertexFormat, AsVertexFormats, VertexAttributeDescriptor};
|
|
||||||
|
|
||||||
let mut vertex_formats: Vec<(&str,&[VertexFormat])> = vec![
|
|
||||||
#((#vertex_buffer_field_names_pascal, <#vertex_buffer_field_types>::as_vertex_formats()),)*
|
|
||||||
];
|
|
||||||
|
|
||||||
let mut shader_location = 0;
|
|
||||||
let mut offset = 0;
|
|
||||||
let vertex_attribute_descriptors = vertex_formats.drain(..).map(|(name, formats)| {
|
|
||||||
formats.iter().enumerate().map(|(i, format)| {
|
|
||||||
let size = format.get_size();
|
|
||||||
let formatted_name = if formats.len() > 1 {
|
|
||||||
format!("{}_{}", name, i)
|
|
||||||
} else {
|
|
||||||
format!("{}", name)
|
|
||||||
};
|
|
||||||
let descriptor = VertexAttributeDescriptor {
|
|
||||||
name: formatted_name.into(),
|
|
||||||
offset,
|
|
||||||
format: *format,
|
|
||||||
shader_location,
|
|
||||||
};
|
|
||||||
offset += size;
|
|
||||||
shader_location += 1;
|
|
||||||
descriptor
|
|
||||||
}).collect::<Vec<VertexAttributeDescriptor>>()
|
|
||||||
}).flatten().collect::<Vec<VertexAttributeDescriptor>>();
|
|
||||||
|
|
||||||
#bevy_render_path::pipeline::VertexBufferDescriptor {
|
|
||||||
attributes: vertex_attribute_descriptors,
|
|
||||||
name: #struct_name_string.into(),
|
|
||||||
step_mode: #bevy_render_path::pipeline::InputStepMode::Instance,
|
|
||||||
stride: offset,
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
impl #bevy_render_path::shader::AsUniforms for #struct_name {
|
|
||||||
fn get_field_infos() -> &'static [#bevy_render_path::shader::FieldInfo] {
|
|
||||||
#field_infos_ident
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_field_bind_type(&self, name: &str) -> Option<#bevy_render_path::shader::FieldBindType> {
|
|
||||||
use #bevy_render_path::shader::GetFieldBindType;
|
|
||||||
match name {
|
|
||||||
#(#active_uniform_field_name_strings => self.#active_uniform_field_names.get_bind_type(),)*
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_uniform_texture(&self, name: &str) -> Option<#bevy_asset_path::Handle<#bevy_render_path::texture::Texture>> {
|
|
||||||
use #bevy_render_path::shader::GetTexture;
|
|
||||||
match name {
|
|
||||||
#(#texture_and_sampler_name_strings => self.#texture_and_sampler_name_idents.get_texture(),)*
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn write_uniform_bytes(&self, name: &str, buffer: &mut [u8]) {
|
|
||||||
use #bevy_core_path::bytes::Bytes;
|
|
||||||
match name {
|
|
||||||
#(#uniform_name_strings => self.#active_uniform_field_names.write_bytes(buffer),)*
|
|
||||||
_ => {},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn uniform_byte_len(&self, name: &str) -> usize {
|
|
||||||
use #bevy_core_path::bytes::Bytes;
|
|
||||||
match name {
|
|
||||||
#(#uniform_name_strings => self.#active_uniform_field_names.byte_len(),)*
|
|
||||||
_ => 0,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: move this to field_info and add has_shader_def(&self, &str) -> bool
|
|
||||||
// TODO: this will be very allocation heavy. find a way to either make this allocation free
|
|
||||||
// or alternatively only run it when the shader_defs have changed
|
|
||||||
fn get_shader_defs(&self) -> Option<Vec<String>> {
|
|
||||||
use #bevy_render_path::shader::ShaderDefSuffixProvider;
|
|
||||||
let mut potential_shader_defs: Vec<(&'static str, Option<&'static str>)> = vec![
|
|
||||||
#((#shader_def_field_names_screaming_snake, self.#shader_def_field_names.get_shader_def()),)*
|
|
||||||
];
|
|
||||||
|
|
||||||
Some(potential_shader_defs.drain(..)
|
|
||||||
.filter(|(f, shader_def)| shader_def.is_some())
|
|
||||||
.map(|(f, shader_def)| format!("{}_{}{}", #struct_name_uppercase, f, shader_def.unwrap()))
|
|
||||||
.collect::<Vec<String>>())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_vertex_buffer_descriptor() -> Option<&'static #bevy_render_path::pipeline::VertexBufferDescriptor> {
|
|
||||||
if #vertex_buffer_descriptor_ident.attributes.len() == 0 {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(&#vertex_buffer_descriptor_ident)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[proc_macro_derive(DynamicAppPlugin)]
|
#[proc_macro_derive(DynamicAppPlugin)]
|
||||||
pub fn derive_app_plugin(input: TokenStream) -> TokenStream {
|
pub fn derive_app_plugin(input: TokenStream) -> TokenStream {
|
||||||
let ast = parse_macro_input!(input as DeriveInput);
|
app_plugin::derive_app_plugin(input)
|
||||||
let struct_name = &ast.ident;
|
|
||||||
|
|
||||||
TokenStream::from(quote! {
|
|
||||||
#[no_mangle]
|
|
||||||
pub extern "C" fn _create_plugin() -> *mut bevy::app::AppPlugin {
|
|
||||||
// TODO: without this the assembly does nothing. why is that the case?
|
|
||||||
print!("");
|
|
||||||
// make sure the constructor is the correct type.
|
|
||||||
let object = #struct_name {};
|
|
||||||
let boxed = Box::new(object);
|
|
||||||
Box::into_raw(boxed)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|||||||
38
crates/bevy_derive/src/resource.rs
Normal file
38
crates/bevy_derive/src/resource.rs
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
use crate::modules::{get_modules, get_path};
|
||||||
|
use proc_macro::TokenStream;
|
||||||
|
use quote::quote;
|
||||||
|
use syn::{parse_macro_input, Data, DataStruct, DeriveInput, Fields};
|
||||||
|
|
||||||
|
pub fn derive_resource(input: TokenStream) -> TokenStream {
|
||||||
|
let ast = parse_macro_input!(input as DeriveInput);
|
||||||
|
let fields = match &ast.data {
|
||||||
|
Data::Struct(DataStruct {
|
||||||
|
fields: Fields::Named(fields),
|
||||||
|
..
|
||||||
|
}) => &fields.named,
|
||||||
|
_ => panic!("expected a struct with named fields"),
|
||||||
|
};
|
||||||
|
|
||||||
|
let modules = get_modules(&ast);
|
||||||
|
let bevy_app_path = get_path(&modules.bevy_app);
|
||||||
|
|
||||||
|
let field_types = fields.iter().map(|field| &field.ty);
|
||||||
|
|
||||||
|
let fields = fields.iter().map(|field| field.ident.as_ref().unwrap());
|
||||||
|
|
||||||
|
let generics = ast.generics;
|
||||||
|
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
||||||
|
|
||||||
|
let struct_name = &ast.ident;
|
||||||
|
|
||||||
|
TokenStream::from(quote! {
|
||||||
|
impl #impl_generics #bevy_app_path::FromResources for #struct_name#ty_generics {
|
||||||
|
fn from_resources(resources: &Resources) -> Self {
|
||||||
|
use #bevy_app_path::FromResources;
|
||||||
|
#struct_name {
|
||||||
|
#(#fields: <#field_types>::from_resources(resources),)*
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
317
crates/bevy_derive/src/uniforms.rs
Normal file
317
crates/bevy_derive/src/uniforms.rs
Normal file
@ -0,0 +1,317 @@
|
|||||||
|
use darling::FromMeta;
|
||||||
|
use inflector::Inflector;
|
||||||
|
use crate::modules::{get_modules, get_path};
|
||||||
|
use proc_macro::TokenStream;
|
||||||
|
use quote::{format_ident, quote};
|
||||||
|
use syn::{parse_macro_input, Data, DataStruct, DeriveInput, Field, Fields, Path};
|
||||||
|
|
||||||
|
// TODO: ensure shader_def and instance/vertex are mutually exclusive
|
||||||
|
#[derive(FromMeta, Debug, Default)]
|
||||||
|
struct UniformAttributeArgs {
|
||||||
|
#[darling(default)]
|
||||||
|
pub ignore: Option<bool>,
|
||||||
|
#[darling(default)]
|
||||||
|
pub shader_def: Option<bool>,
|
||||||
|
#[darling(default)]
|
||||||
|
pub instance: Option<bool>,
|
||||||
|
#[darling(default)]
|
||||||
|
pub vertex: Option<bool>,
|
||||||
|
#[darling(default)]
|
||||||
|
pub buffer: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
struct UniformAttributes {
|
||||||
|
pub ignore: bool,
|
||||||
|
pub shader_def: bool,
|
||||||
|
pub instance: bool,
|
||||||
|
pub vertex: bool,
|
||||||
|
pub buffer: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<UniformAttributeArgs> for UniformAttributes {
|
||||||
|
fn from(args: UniformAttributeArgs) -> Self {
|
||||||
|
UniformAttributes {
|
||||||
|
ignore: args.ignore.unwrap_or(false),
|
||||||
|
shader_def: args.shader_def.unwrap_or(false),
|
||||||
|
instance: args.instance.unwrap_or(false),
|
||||||
|
vertex: args.vertex.unwrap_or(false),
|
||||||
|
buffer: args.buffer.unwrap_or(false),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static UNIFORM_ATTRIBUTE_NAME: &'static str = "uniform";
|
||||||
|
|
||||||
|
pub fn derive_uniforms(input: TokenStream) -> TokenStream {
|
||||||
|
let ast = parse_macro_input!(input as DeriveInput);
|
||||||
|
let modules = get_modules(&ast);
|
||||||
|
|
||||||
|
let bevy_render_path: Path = get_path(&modules.bevy_render);
|
||||||
|
let bevy_core_path: Path = get_path(&modules.bevy_core);
|
||||||
|
let bevy_asset_path: Path = get_path(&modules.bevy_asset);
|
||||||
|
|
||||||
|
let fields = match &ast.data {
|
||||||
|
Data::Struct(DataStruct {
|
||||||
|
fields: Fields::Named(fields),
|
||||||
|
..
|
||||||
|
}) => &fields.named,
|
||||||
|
_ => panic!("expected a struct with named fields"),
|
||||||
|
};
|
||||||
|
|
||||||
|
let field_attributes = fields
|
||||||
|
.iter()
|
||||||
|
.map(|f| {
|
||||||
|
(
|
||||||
|
f,
|
||||||
|
f.attrs
|
||||||
|
.iter()
|
||||||
|
.find(|a| {
|
||||||
|
a.path.get_ident().as_ref().unwrap().to_string() == UNIFORM_ATTRIBUTE_NAME
|
||||||
|
})
|
||||||
|
.map(|a| {
|
||||||
|
UniformAttributes::from(
|
||||||
|
UniformAttributeArgs::from_meta(&a.parse_meta().unwrap())
|
||||||
|
.unwrap_or_else(|_err| UniformAttributeArgs::default()),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.unwrap_or_else(|| UniformAttributes::default()),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.collect::<Vec<(&Field, UniformAttributes)>>();
|
||||||
|
|
||||||
|
let struct_name = &ast.ident;
|
||||||
|
|
||||||
|
let mut active_uniform_field_names = Vec::new();
|
||||||
|
let mut active_uniform_field_name_strings = Vec::new();
|
||||||
|
let mut uniform_name_strings = Vec::new();
|
||||||
|
let mut texture_and_sampler_name_strings = Vec::new();
|
||||||
|
let mut texture_and_sampler_name_idents = Vec::new();
|
||||||
|
let mut field_infos = Vec::new();
|
||||||
|
|
||||||
|
let mut vertex_buffer_field_names_pascal = Vec::new();
|
||||||
|
let mut vertex_buffer_field_types = Vec::new();
|
||||||
|
|
||||||
|
let mut shader_def_field_names = Vec::new();
|
||||||
|
let mut shader_def_field_names_screaming_snake = Vec::new();
|
||||||
|
|
||||||
|
for (f, attrs) in field_attributes.iter() {
|
||||||
|
let field_name = f.ident.as_ref().unwrap().to_string();
|
||||||
|
if !attrs.ignore {
|
||||||
|
active_uniform_field_names.push(&f.ident);
|
||||||
|
active_uniform_field_name_strings.push(field_name.clone());
|
||||||
|
let uniform = format!("{}_{}", struct_name, field_name);
|
||||||
|
let texture = format!("{}", uniform);
|
||||||
|
let sampler = format!("{}_sampler", uniform);
|
||||||
|
uniform_name_strings.push(uniform.clone());
|
||||||
|
texture_and_sampler_name_strings.push(texture.clone());
|
||||||
|
texture_and_sampler_name_strings.push(sampler.clone());
|
||||||
|
texture_and_sampler_name_idents.push(f.ident.clone());
|
||||||
|
texture_and_sampler_name_idents.push(f.ident.clone());
|
||||||
|
let is_instanceable = attrs.instance;
|
||||||
|
field_infos.push(quote!(#bevy_render_path::shader::FieldInfo {
|
||||||
|
name: #field_name,
|
||||||
|
uniform_name: #uniform,
|
||||||
|
texture_name: #texture,
|
||||||
|
sampler_name: #sampler,
|
||||||
|
is_instanceable: #is_instanceable,
|
||||||
|
}));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if attrs.shader_def {
|
||||||
|
shader_def_field_names.push(&f.ident);
|
||||||
|
shader_def_field_names_screaming_snake.push(field_name.to_screaming_snake_case())
|
||||||
|
}
|
||||||
|
|
||||||
|
if attrs.instance || attrs.vertex {
|
||||||
|
vertex_buffer_field_types.push(&f.ty);
|
||||||
|
let pascal_field = f.ident.as_ref().unwrap().to_string().to_pascal_case();
|
||||||
|
vertex_buffer_field_names_pascal.push(if attrs.instance {
|
||||||
|
format!("I_{}_{}", struct_name, pascal_field)
|
||||||
|
} else {
|
||||||
|
format!("{}_{}", struct_name, pascal_field)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let struct_name_string = struct_name.to_string();
|
||||||
|
let struct_name_uppercase = struct_name_string.to_uppercase();
|
||||||
|
let field_infos_ident = format_ident!("{}_FIELD_INFO", struct_name_uppercase);
|
||||||
|
let vertex_buffer_descriptor_ident =
|
||||||
|
format_ident!("{}_VERTEX_BUFFER_DESCRIPTOR", struct_name_uppercase);
|
||||||
|
|
||||||
|
TokenStream::from(quote! {
|
||||||
|
static #field_infos_ident: &[#bevy_render_path::shader::FieldInfo] = &[
|
||||||
|
#(#field_infos,)*
|
||||||
|
];
|
||||||
|
|
||||||
|
static #vertex_buffer_descriptor_ident: #bevy_render_path::once_cell::sync::Lazy<#bevy_render_path::pipeline::VertexBufferDescriptor> =
|
||||||
|
#bevy_render_path::once_cell::sync::Lazy::new(|| {
|
||||||
|
use #bevy_render_path::pipeline::{VertexFormat, AsVertexFormats, VertexAttributeDescriptor};
|
||||||
|
|
||||||
|
let mut vertex_formats: Vec<(&str,&[VertexFormat])> = vec![
|
||||||
|
#((#vertex_buffer_field_names_pascal, <#vertex_buffer_field_types>::as_vertex_formats()),)*
|
||||||
|
];
|
||||||
|
|
||||||
|
let mut shader_location = 0;
|
||||||
|
let mut offset = 0;
|
||||||
|
let vertex_attribute_descriptors = vertex_formats.drain(..).map(|(name, formats)| {
|
||||||
|
formats.iter().enumerate().map(|(i, format)| {
|
||||||
|
let size = format.get_size();
|
||||||
|
let formatted_name = if formats.len() > 1 {
|
||||||
|
format!("{}_{}", name, i)
|
||||||
|
} else {
|
||||||
|
format!("{}", name)
|
||||||
|
};
|
||||||
|
let descriptor = VertexAttributeDescriptor {
|
||||||
|
name: formatted_name.into(),
|
||||||
|
offset,
|
||||||
|
format: *format,
|
||||||
|
shader_location,
|
||||||
|
};
|
||||||
|
offset += size;
|
||||||
|
shader_location += 1;
|
||||||
|
descriptor
|
||||||
|
}).collect::<Vec<VertexAttributeDescriptor>>()
|
||||||
|
}).flatten().collect::<Vec<VertexAttributeDescriptor>>();
|
||||||
|
|
||||||
|
#bevy_render_path::pipeline::VertexBufferDescriptor {
|
||||||
|
attributes: vertex_attribute_descriptors,
|
||||||
|
name: #struct_name_string.into(),
|
||||||
|
step_mode: #bevy_render_path::pipeline::InputStepMode::Instance,
|
||||||
|
stride: offset,
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
impl #bevy_render_path::shader::AsUniforms for #struct_name {
|
||||||
|
fn get_field_infos() -> &'static [#bevy_render_path::shader::FieldInfo] {
|
||||||
|
#field_infos_ident
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_field_bind_type(&self, name: &str) -> Option<#bevy_render_path::shader::FieldBindType> {
|
||||||
|
use #bevy_render_path::shader::GetFieldBindType;
|
||||||
|
match name {
|
||||||
|
#(#active_uniform_field_name_strings => self.#active_uniform_field_names.get_bind_type(),)*
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_uniform_texture(&self, name: &str) -> Option<#bevy_asset_path::Handle<#bevy_render_path::texture::Texture>> {
|
||||||
|
use #bevy_render_path::shader::GetTexture;
|
||||||
|
match name {
|
||||||
|
#(#texture_and_sampler_name_strings => self.#texture_and_sampler_name_idents.get_texture(),)*
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn write_uniform_bytes(&self, name: &str, buffer: &mut [u8]) {
|
||||||
|
use #bevy_core_path::bytes::Bytes;
|
||||||
|
match name {
|
||||||
|
#(#uniform_name_strings => self.#active_uniform_field_names.write_bytes(buffer),)*
|
||||||
|
_ => {},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn uniform_byte_len(&self, name: &str) -> usize {
|
||||||
|
use #bevy_core_path::bytes::Bytes;
|
||||||
|
match name {
|
||||||
|
#(#uniform_name_strings => self.#active_uniform_field_names.byte_len(),)*
|
||||||
|
_ => 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: move this to field_info and add has_shader_def(&self, &str) -> bool
|
||||||
|
// TODO: this will be very allocation heavy. find a way to either make this allocation free
|
||||||
|
// or alternatively only run it when the shader_defs have changed
|
||||||
|
fn get_shader_defs(&self) -> Option<Vec<String>> {
|
||||||
|
use #bevy_render_path::shader::ShaderDefSuffixProvider;
|
||||||
|
let mut potential_shader_defs: Vec<(&'static str, Option<&'static str>)> = vec![
|
||||||
|
#((#shader_def_field_names_screaming_snake, self.#shader_def_field_names.get_shader_def()),)*
|
||||||
|
];
|
||||||
|
|
||||||
|
Some(potential_shader_defs.drain(..)
|
||||||
|
.filter(|(f, shader_def)| shader_def.is_some())
|
||||||
|
.map(|(f, shader_def)| format!("{}_{}{}", #struct_name_uppercase, f, shader_def.unwrap()))
|
||||||
|
.collect::<Vec<String>>())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_vertex_buffer_descriptor() -> Option<&'static #bevy_render_path::pipeline::VertexBufferDescriptor> {
|
||||||
|
if #vertex_buffer_descriptor_ident.attributes.len() == 0 {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(&#vertex_buffer_descriptor_ident)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn derive_uniform(input: TokenStream) -> TokenStream {
|
||||||
|
let ast = parse_macro_input!(input as DeriveInput);
|
||||||
|
|
||||||
|
let modules = get_modules(&ast);
|
||||||
|
let bevy_asset_path = get_path(&modules.bevy_asset);
|
||||||
|
let bevy_core_path = get_path(&modules.bevy_core);
|
||||||
|
let bevy_render_path = get_path(&modules.bevy_render);
|
||||||
|
|
||||||
|
let generics = ast.generics;
|
||||||
|
let (impl_generics, ty_generics, _where_clause) = generics.split_for_impl();
|
||||||
|
|
||||||
|
let struct_name = &ast.ident;
|
||||||
|
let struct_name_string = struct_name.to_string();
|
||||||
|
|
||||||
|
TokenStream::from(quote! {
|
||||||
|
impl #impl_generics #bevy_render_path::shader::AsUniforms for #struct_name#ty_generics {
|
||||||
|
fn get_field_infos() -> &'static [#bevy_render_path::shader::FieldInfo] {
|
||||||
|
static FIELD_INFOS: &[#bevy_render_path::shader::FieldInfo] = &[
|
||||||
|
#bevy_render_path::shader::FieldInfo {
|
||||||
|
name: #struct_name_string,
|
||||||
|
uniform_name: #struct_name_string,
|
||||||
|
texture_name: #struct_name_string,
|
||||||
|
sampler_name: #struct_name_string,
|
||||||
|
is_instanceable: false,
|
||||||
|
}
|
||||||
|
];
|
||||||
|
&FIELD_INFOS
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_field_bind_type(&self, name: &str) -> Option<#bevy_render_path::shader::FieldBindType> {
|
||||||
|
use #bevy_render_path::shader::GetFieldBindType;
|
||||||
|
match name {
|
||||||
|
#struct_name_string => self.get_bind_type(),
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn write_uniform_bytes(&self, name: &str, buffer: &mut [u8]) {
|
||||||
|
use #bevy_core_path::bytes::Bytes;
|
||||||
|
match name {
|
||||||
|
#struct_name_string => self.write_bytes(buffer),
|
||||||
|
_ => {},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn uniform_byte_len(&self, name: &str) -> usize {
|
||||||
|
use #bevy_core_path::bytes::Bytes;
|
||||||
|
match name {
|
||||||
|
#struct_name_string => self.byte_len(),
|
||||||
|
_ => 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_uniform_texture(&self, name: &str) -> Option<#bevy_asset_path::Handle<#bevy_render_path::texture::Texture>> {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: move this to field_info and add has_shader_def(&self, &str) -> bool
|
||||||
|
// TODO: this will be very allocation heavy. find a way to either make this allocation free
|
||||||
|
// or alternatively only run it when the shader_defs have changed
|
||||||
|
fn get_shader_defs(&self) -> Option<Vec<String>> {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_vertex_buffer_descriptor() -> Option<&'static #bevy_render_path::pipeline::VertexBufferDescriptor> {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user