Switched custom quantities to using a macro for generation.
This commit is contained in:
18
generate-quantity/Cargo.toml
Normal file
18
generate-quantity/Cargo.toml
Normal file
@ -0,0 +1,18 @@
|
||||
[package]
|
||||
name = "generate-quantity"
|
||||
description = "Macros for generating physical quantity type"
|
||||
version.workspace = true
|
||||
edition.workspace = true
|
||||
repository.workspace = true
|
||||
readme.workspace = true
|
||||
license.workspace = true
|
||||
|
||||
[lib]
|
||||
proc-macro = true
|
||||
|
||||
[dependencies.quote]
|
||||
workspace = true
|
||||
[dependencies.syn]
|
||||
workspace = true
|
||||
[dependencies.proc-macro2]
|
||||
workspace = true
|
101
generate-quantity/src/lib.rs
Normal file
101
generate-quantity/src/lib.rs
Normal file
@ -0,0 +1,101 @@
|
||||
extern crate proc_macro;
|
||||
use proc_macro::TokenStream;
|
||||
use proc_macro2::{Span, TokenStream as TokenStream2};
|
||||
use quote::quote;
|
||||
use syn::parse::{Parse, ParseStream};
|
||||
use syn::{parse_macro_input, Ident, LitStr, Token};
|
||||
|
||||
const NUMBER_TYPES: &[&str] = &[
|
||||
"u8", "i8", "u16", "i16", "u32", "i32", "u64", "i64", "u128", "i128", "usize", "isize", "f32",
|
||||
"f64",
|
||||
];
|
||||
|
||||
// Define the input structure for the macro
|
||||
struct QuantityInput {
|
||||
struct_name: Ident,
|
||||
symbol: LitStr,
|
||||
}
|
||||
|
||||
// Implement the parsing for the input structure
|
||||
impl Parse for QuantityInput {
|
||||
fn parse(input: ParseStream) -> syn::Result<Self> {
|
||||
let struct_name: Ident = input.parse()?;
|
||||
input.parse::<Token![,]>()?;
|
||||
let symbol: LitStr = input.parse()?;
|
||||
Ok(QuantityInput {
|
||||
struct_name,
|
||||
symbol,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
/// The following imports must be in scope for this macro to work
|
||||
/// ```
|
||||
/// use physical::quantity::{Quantity, Value};
|
||||
/// use derive_more::{Add, AddAssign, Display, Sub, SubAssign};
|
||||
/// use generate_quantity::quantity_type;
|
||||
/// ```
|
||||
#[proc_macro]
|
||||
pub fn quantity_type(input: TokenStream) -> TokenStream {
|
||||
// Parse the input tokens into the QuantityInput structure
|
||||
let QuantityInput {
|
||||
struct_name,
|
||||
symbol,
|
||||
} = parse_macro_input!(input as QuantityInput);
|
||||
|
||||
//----- Conversion impls ----------------------------------
|
||||
let mut impls: Vec<TokenStream2> = Vec::new();
|
||||
for ¤t_type in NUMBER_TYPES {
|
||||
// Generate conversion methods for all primitive types except the current_type
|
||||
let conversions = NUMBER_TYPES
|
||||
.iter()
|
||||
.filter(|&&t| t != current_type)
|
||||
.map(|&target_type| {
|
||||
let method_name = Ident::new(&format!("as_{}", target_type), Span::call_site());
|
||||
let target_type = Ident::new(target_type, Span::call_site());
|
||||
|
||||
quote! {
|
||||
/// Directly [as] cast this quantities value while maintaining the type of the quantity.
|
||||
#[inline(always)]
|
||||
pub fn #method_name(self) -> #struct_name<#target_type> {
|
||||
#struct_name(self.0 as #target_type)
|
||||
}
|
||||
}
|
||||
})
|
||||
.collect::<Vec<TokenStream2>>();
|
||||
|
||||
let current_type = Ident::new(current_type, Span::call_site());
|
||||
// Generate the impl block for the struct with the current_type
|
||||
let expanded = quote! {
|
||||
impl #struct_name<#current_type> {
|
||||
#(#conversions)*
|
||||
}
|
||||
};
|
||||
|
||||
impls.push(expanded);
|
||||
}
|
||||
|
||||
//----- Output Code ----------------------------------
|
||||
let expanded = quote! {
|
||||
#[derive(Copy, Clone, PartialEq, PartialOrd, Add, AddAssign, Sub, SubAssign, Display)]
|
||||
#[display(fmt = "{} {}", _0, "Self::symbol()")]
|
||||
#[repr(transparent)]
|
||||
pub struct #struct_name<V: Value>(pub V);
|
||||
|
||||
impl<V: Value> Quantity<V> for #struct_name<V> {
|
||||
#[inline(always)]
|
||||
fn value(self) -> V {
|
||||
self.0
|
||||
}
|
||||
|
||||
#[inline(always)]
|
||||
fn symbol() -> &'static str {
|
||||
#symbol
|
||||
}
|
||||
}
|
||||
|
||||
#(#impls)*
|
||||
};
|
||||
|
||||
expanded.into()
|
||||
}
|
Reference in New Issue
Block a user