| use proc_macro::TokenStream; |
| use quote::{ToTokens, quote}; |
| use std::fs; |
| use std::path::Path; |
| use toml::{Table, Value}; |
|
|
| enum CustomValue { |
| String(String), |
| Integer(i64), |
| Float(f64), |
| Boolean(bool), |
| Array(Vec<CustomValue>), |
| } |
|
|
| impl ToTokens for CustomValue { |
| fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) { |
| match self { |
| CustomValue::String(x) => x.to_tokens(tokens), |
| CustomValue::Integer(x) => { |
| let x: proc_macro2::TokenStream = format!("{:?}", x).parse().unwrap(); |
| x.to_tokens(tokens) |
| } |
| CustomValue::Float(x) => { |
| let x: proc_macro2::TokenStream = format!("{:?}", x).parse().unwrap(); |
| x.to_tokens(tokens) |
| } |
| CustomValue::Boolean(x) => x.to_tokens(tokens), |
| CustomValue::Array(x) => quote! { [ #( #x ),* ] }.to_tokens(tokens), |
| } |
| } |
| } |
|
|
| impl From<Value> for CustomValue { |
| fn from(value: Value) -> Self { |
| match value { |
| Value::String(x) => CustomValue::String(x), |
| Value::Integer(x) => CustomValue::Integer(x), |
| Value::Float(x) => CustomValue::Float(x), |
| Value::Boolean(x) => CustomValue::Boolean(x), |
| Value::Array(x) => CustomValue::Array(x.into_iter().map(|x| x.into()).collect()), |
| _ => panic!("Unsupported data type"), |
| } |
| } |
| } |
|
|
| pub fn build_camera_data() -> TokenStream { |
| let mut camera_data: Vec<(String, Table)> = Vec::new(); |
|
|
| let mut path = Path::new(&std::env::var("CARGO_MANIFEST_DIR").unwrap()).to_path_buf(); |
| path.push("camera_data"); |
|
|
| fs::read_dir(path).unwrap().for_each(|entry| { |
| let company_name_path = entry.unwrap().path(); |
| if !company_name_path.is_dir() { |
| panic!("camera_data should only contain folders of company names") |
| } |
|
|
| let company_name = company_name_path.file_name().unwrap().to_str().unwrap().to_string(); |
|
|
| fs::read_dir(company_name_path).unwrap().for_each(|entry| { |
| let model_path = entry.unwrap().path(); |
| if !model_path.is_file() || model_path.extension().unwrap() != "toml" { |
| panic!("The folders within camera_data should only contain toml files") |
| } |
|
|
| let name = company_name.clone() + " " + model_path.file_stem().unwrap().to_str().unwrap(); |
|
|
| let mut values: Table = toml::from_str(&fs::read_to_string(model_path).unwrap()).unwrap(); |
|
|
| if let Some(val) = values.get_mut("xyz_to_camera") { |
| *val = Value::Array(val.as_array().unwrap().iter().map(|x| Value::Integer((x.as_float().unwrap() * 10_000.) as i64)).collect()); |
| } |
|
|
| camera_data.push((name, values)) |
| }); |
| }); |
|
|
| let x: Vec<_> = camera_data |
| .iter() |
| .map(|(name, camera_data)| { |
| let keys: Vec<_> = camera_data.keys().map(|key| syn::Ident::new(key, proc_macro2::Span::call_site())).collect(); |
| let values: Vec<CustomValue> = camera_data.values().cloned().map(|x| x.into()).collect(); |
|
|
| quote! { |
| ( |
| #name, |
| CameraData { |
| #( #keys: #values, )* |
| ..CameraData::DEFAULT |
| } |
| ) |
| } |
| }) |
| .collect(); |
|
|
| quote!([ #(#x),* ]).into() |
| } |
|
|
