add select_buffers to avoid the need to directly reference generate buffer struct

Signed-off-by: Teo Koon Peng <[email protected]>

Author: koonpeng

macros/src/buffer.rs

@@ -31,6 +31,20 @@ pub(crate) fn impl_joined_value(input_struct: &ItemStruct) -> Result<TokenStream
 
         #buffer_struct
 
+        impl #impl_generics #struct_ident #ty_generics #where_clause {
+            fn select_buffers(
+                #(
+                    #field_ident: ::bevy_impulse::Buffer<#field_type>,
+                )*
+            ) -> #struct_buffer_ident #ty_generics {
+                #struct_buffer_ident {
+                    #(
+                        #field_ident,
+                    )*
+                }
+            }
+        }
+
         #impl_buffer_map_layout
 
         #impl_joined

src/buffer/buffer_map.rs

@@ -369,12 +369,12 @@ mod tests {
         let mut context = TestingContext::minimal_plugins();
 
         let workflow = context.spawn_io_workflow(|scope, builder| {
-            let buffers = __bevy_impulse_TestJoinedValue_Buffers {
-                integer: builder.create_buffer(BufferSettings::default()),
-                float: builder.create_buffer(BufferSettings::default()),
-                string: builder.create_buffer(BufferSettings::default()),
-                generic: builder.create_buffer(BufferSettings::default()),
-            };
+            let buffers = TestJoinedValue::select_buffers(
+                builder.create_buffer(BufferSettings::default()),
+                builder.create_buffer(BufferSettings::default()),
+                builder.create_buffer(BufferSettings::default()),
+                builder.create_buffer(BufferSettings::default()),
+            );
 
             scope.input.chain(builder).fork_unzip((
                 |chain: Chain<_>| chain.connect(buffers.integer.input_slot()),