alex/ash
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Better version idents

Browse source
This commit is contained in:
Maik Klein 2018-03-09 22:43:45 +01:00
parent 6691e7e79f
commit 2c9d4ed71d
1 changed files with 53 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

78
generator/src/lib.rs
View file

@ -11,6 +11,18 @@ use heck::SnakeCase;
use syn::Ident; use syn::Ident;
pub trait FeatureExt {
fn version_string(&self) -> String;
}
impl FeatureExt for vkxml::Feature {
fn version_string(&self) -> String {
let version = format!("{:.10}", self.version);
let first_0 = version.find("0").expect("should have at least one 0");
// + 1 is correct here because we always have 10 zeroes.
let (version, _) = version.split_at(first_0 + 1);
version.replace(".", "_")
}
}
pub trait CommandExt { pub trait CommandExt {
/// Returns the ident in snake_case and without the 'vk' prefix. /// Returns the ident in snake_case and without the 'vk' prefix.
fn is_device_command(&self) -> bool; fn is_device_command(&self) -> bool;
@ -100,36 +112,52 @@ pub fn gen_load(feature: &vkxml::Feature, commands: &CommandMap) -> quote::Token
}) })
.filter_map(|cmd_ref| commands.get(&cmd_ref.name)) .filter_map(|cmd_ref| commands.get(&cmd_ref.name))
.fold((Vec::new(), Vec::new()), |mut acc, &cmd_ref| { .fold((Vec::new(), Vec::new()), |mut acc, &cmd_ref| {
let basetype = cmd_ref.param[0].basetype.as_str(); if cmd_ref.is_device_command() {
match basetype { acc.0.push(cmd_ref);
"VkDevice" | "VkCommandBuffer" | "VkQueue" => acc.0.push(cmd_ref), } else {
_ => acc.1.push(cmd_ref), acc.1.push(cmd_ref);
}; };
acc acc
}); });
let name = Ident::from("Test"); let name = Ident::from("Test");
let function_pointers: Vec<_> = instance_commands fn generate_function_pointers(ident: Ident, commands: &[&vkxml::Command]) -> quote::Tokens {
.iter() let function_pointers: Vec<_> = commands
.map(|cmd| { .iter()
let fn_name_raw = cmd.name.as_str(); .map(|cmd| {
let fn_name_snake = cmd.command_ident(); let fn_name_raw = cmd.name.as_str();
let params: Vec<_> = cmd.param let fn_name_snake = cmd.command_ident();
.iter() let params: Vec<_> = cmd.param
.map(|field| { .iter()
let name = field.param_ident(); .map(|field| {
let ty = field.type_ident(); let name = field.param_ident();
quote!{#name: #ty} let ty = field.type_ident();
}) quote!{#name: #ty}
.collect(); })
let return_ty = cmd.return_type.type_ident(); .collect();
quote!{ let return_ty = cmd.return_type.type_ident();
#fn_name_snake: extern "system" fn(#(#params,)*) -> #return_ty quote!{
#fn_name_snake: extern "system" fn(#(#params,)*) -> #return_ty
}
})
.collect();
quote!{
pub struct #ident {
#(#function_pointers,)*
} }
})
.collect();
quote!{
pub struct #name {
#(#function_pointers,)*
} }
} }
let version = feature.version_string();
let instance = generate_function_pointers(
Ident::from(format!("InstanceFnV{}", version).as_str()),
&instance_commands,
);
let device = generate_function_pointers(
Ident::from(format!("DeviceFnV{}", version).as_str()),
&instance_commands,
);
quote! {
#instance
#device
}
} }