essentia-codegen 0.1.5

Code generation utilities for Essentia Rust bindings
Documentation 
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

use convert_case::{Case, Casing};
use essentia_core::{Introspection, algorithm::ParameterInfo};
use proc_macro2::TokenStream;
use quote::{format_ident, quote};

use crate::algorithm_generation::common::{
    data_type_to_phantom, sanitize_identifier_string, string_to_doc_comment,
};

fn generate_parameter_function_docs(parameter: &ParameterInfo) -> TokenStream {
    let name = parameter.name();
    let description = parameter.description();
    let doc = format!("Sets the `{}` parameter.\n\n{}", name, description);

    string_to_doc_comment(&doc)
}

pub fn generate_parameter_functions(algorithm_introspection: &Introspection) -> Vec<TokenStream> {
    algorithm_introspection
        .parameters()
        .map(|parameter| {
            let parameter_name = parameter.name();
            let identifier = format_ident!("{}", sanitize_identifier_string(&parameter_name.to_case(Case::Snake)));
            let variant = data_type_to_phantom(&parameter.parameter_type());
            let doc_comment = generate_parameter_function_docs(parameter);

            quote! {
                #doc_comment
                pub fn #identifier(mut self, value: impl crate::data::TryIntoDataContainer<#variant>) -> Result<Self, crate::algorithm::ParameterError> {
                    self.algorithm.set_parameter(#parameter_name, value)?;
                    Ok(self)
                }
            }
        })
        .collect()
}