1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
use proc_macro::TokenStream;
use quote::quote;
#[proc_macro_derive(Modulation)]
pub fn modulation_derive(input: TokenStream) -> TokenStream {
let ast = syn::parse(input).unwrap();
impl_modulation_macro(&ast)
}
fn impl_modulation_macro(ast: &syn::DeriveInput) -> TokenStream {
let name = &ast.ident;
let generics = &ast.generics;
let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
let gen = quote! {
impl #impl_generics Modulation for #name #ty_generics #where_clause {
fn build(&mut self) -> Result<()>{
self.calc()
}
fn buffer(&self) -> &[u8] {
&self.buffer
}
fn remaining(&self) -> usize {
self.buffer().len() - self.sent()
}
fn head(&self) -> *const u8 {
unsafe { self.buffer().as_ptr().add(self.sent()) }
}
fn sent(&self) -> usize {
self.sent
}
fn send(&mut self, sent: usize){
self.sent += sent;
}
fn sampling_frequency_division(&self) -> u16 {
self.sampling_freq_div
}
fn sampling_freq(&self) -> f64 {
autd3_core::hardware_defined::MOD_SAMPLING_FREQ_BASE as f64 / self.sampling_freq_div as f64
}
}
};
gen.into()
}
#[proc_macro_derive(Gain)]
pub fn gain_derive(input: TokenStream) -> TokenStream {
let ast = syn::parse(input).unwrap();
impl_gain_macro(&ast)
}
fn impl_gain_macro(ast: &syn::DeriveInput) -> TokenStream {
let name = &ast.ident;
let generics = &ast.generics;
let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
let gen = quote! {
impl #impl_generics Gain for #name #ty_generics #where_clause {
fn build(&mut self, geometry: &Geometry) -> Result<()>{
if self.built() {return Ok(())}
let buf: DataArray = unsafe { std::mem::zeroed() };
self.data = vec![buf; geometry.num_devices()];
self.calc(geometry)
}
fn data(&self) -> &[DataArray]{
&self.data
}
fn built(&self) -> bool {
self.built
}
}
};
gen.into()
}