feat: soa_attr support for SliceMut/PtrMut/RefMut

Author: curryyzheng

soa-derive-internal/src/input.rs

@@ -19,8 +19,11 @@ pub struct Input {
 
     pub vec_attrs: Vec<Meta>,
     pub slice_attrs: Vec<Meta>,
+    pub slice_mut_attrs: Vec<Meta>,
     pub ref_attrs: Vec<Meta>,
+    pub ref_mut_attrs: Vec<Meta>,
     pub ptr_attrs: Vec<Meta>,
+    pub ptr_mut_attrs: Vec<Meta>,
 }
 
 impl Input {
@@ -33,8 +36,11 @@ impl Input {
         let mut derives: Vec<Ident> = vec![];
         let mut vec_attrs = Vec::new();
         let mut slice_attrs = Vec::new();
+        let mut slice_mut_attrs = Vec::new();
         let mut ref_attrs = Vec::new();
+        let mut ref_mut_attrs = Vec::new();
         let mut ptr_attrs = Vec::new();
+        let mut ptr_mut_attrs = Vec::new();
         for attr in input.attrs {
             if let Ok(meta) = attr.parse_meta() {
                 if meta.path().is_ident("soa_derive") {
@@ -63,23 +69,27 @@ impl Input {
                                     panic!("expected a attribute, got {}", quote!(attr))
                                 }
                             };
-                            match soa_type {
+                            let attrs_mut = match soa_type {
                                 NestedMeta::Meta(Meta::Path(path)) => match path.get_ident() {
                                     Some(ident) => match ident.to_string().as_str() {
-                                        "Vec" => vec_attrs.push(attr),
-                                        "Slice" => slice_attrs.push(attr),
-                                        "Ref" => ref_attrs.push(attr),
-                                        "Ptr" => ptr_attrs.push(attr),
+                                        "Vec" => &mut vec_attrs,
+                                        "Slice" => &mut slice_attrs,
+                                        "SliceMut" => &mut slice_mut_attrs,
+                                        "Ref" => &mut ref_attrs,
+                                        "RefMut" => &mut ref_mut_attrs,
+                                        "Ptr" => &mut ptr_attrs,
+                                        "PtrMut" => &mut ptr_mut_attrs,
                                         _ => panic!("expected a soa type, got {}", ident),
                                     },
                                     None => {
-                                        panic!("expected a soa type, got {}", quote!(#path))
+                                        panic!("expected a soa type, got {}", quote!(#path));
                                     }
                                 },
                                 _ => {
-                                    panic!("expected a soa type, got {}", quote!(#soa_type))
+                                    panic!("expected a soa type, got {}", quote!(#soa_type));
                                 }
-                            }
+                            };
+                            attrs_mut.push(attr);
                         }
                         _ => panic!("expected #[soa_attr(...)], got #[{}]", quote!(#meta)),
                     }
@@ -94,8 +104,11 @@ impl Input {
             visibility: input.vis,
             vec_attrs,
             slice_attrs,
+            slice_mut_attrs,
             ref_attrs,
+            ref_mut_attrs,
             ptr_attrs,
+            ptr_mut_attrs,
         }
     }
 

soa-derive-internal/src/ptr.rs

@@ -8,6 +8,7 @@ pub fn derive(input: &Input) -> TokenStream {
     let visibility = &input.visibility;
     let other_derive = &input.derive_with_exceptions();
     let attrs = &input.ptr_attrs;
+    let mut_attrs = &input.ptr_mut_attrs;
     let vec_name = &input.vec_name();
     let ptr_name = &input.ptr_name();
     let ptr_mut_name = &input.ptr_mut_name();
@@ -56,6 +57,7 @@ pub fn derive(input: &Input) -> TokenStream {
         #[doc = #doc_url]
         /// with struct of array layout.
         #other_derive
+        #(#[#mut_attrs])*
         #[derive(Copy, Clone)]
         #visibility struct #ptr_mut_name {
             #(

soa-derive-internal/src/refs.rs

@@ -8,6 +8,7 @@ pub fn derive(input: &Input) -> TokenStream {
     let visibility = &input.visibility;
     let other_derive = &input.derive_with_exceptions();
     let attrs = &input.ref_attrs;
+    let mut_attrs = &input.ref_mut_attrs;
     let vec_name = &input.vec_name();
     let ref_name = &input.ref_name();
     let ref_mut_name = &input.ref_mut_name();
@@ -52,7 +53,7 @@ pub fn derive(input: &Input) -> TokenStream {
         #[doc = #doc_url]
         /// with struct of array layout.
         #other_derive
-        #(#[#attrs])*
+        #(#[#mut_attrs])*
         #visibility struct #ref_mut_name<'a> {
             #(
                 #[doc = #fields_mut_doc]

soa-derive-internal/src/slice.rs

@@ -241,7 +241,7 @@ pub fn derive_mut(input: &Input) -> TokenStream {
     let slice_name = &input.slice_name();
     let slice_mut_name = &input.slice_mut_name();
     let vec_name = &input.vec_name();
-    let attrs = &input.slice_attrs;
+    let attrs = &input.slice_mut_attrs;
     let ref_mut_name = &input.ref_mut_name();
     let ptr_name = &input.ptr_name();
     let ptr_mut_name = &input.ptr_mut_name();