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

mod imports;
use imports::*;
mod altrios_api;
mod history_vec_derive;
mod hm_derive;
mod serde_api_derive;
mod utilities;

#[proc_macro_error]
#[proc_macro_attribute]
/// macro for creating appropriate setters and getters for pyo3 struct attributes
/// and other, non-python API functionality
pub fn altrios_api(attr: TokenStream, item: TokenStream) -> TokenStream {
    altrios_api::altrios_api(attr, item)
}

#[proc_macro_derive(HistoryVec)]
/// generate HistoryVec that acts like a vec of States but
/// stores each field of state as a vec field.
pub fn history_vec_derive(input: TokenStream) -> TokenStream {
    history_vec_derive::history_vec_derive(input)
}

#[proc_macro_derive(HistoryMethods, attributes(has_state))]
/// Generate `step` and `save_state` methods that work for struct and any
/// nested fields with the `#[has_state]` attribute.  
pub fn history_methods_derive(input: TokenStream) -> TokenStream {
    hm_derive::history_methods_derive(input)
}

#[proc_macro_error]
#[proc_macro_derive(SerdeAPI)]
/// macro for deriving default implementation of SerdeAPI trait
pub fn serde_api_derive(item: TokenStream) -> TokenStream {
    serde_api_derive::serde_api_derive(item)
}