Rename Raw-rs to Rawkit (#2088)

* Rename within files * Rename in CI * Rename the folder and file names * Rename raw_rs to rawkit * Add example to README * Add initial documentation * Small API changes and extra documentation * Bump versions and stuff * Readme improvements * Merge proc-macro crates into one * Add README to rawkit-proc-macros * Remove keywords and categories * Add licenses to rawkit-proc-macros --------- Co-authored-by: Keavon Chambers <keavon@keavon.com>
2026-09-22 01:48:11 +08:00 · 2024-11-03 08:10:39 +00:00
parent 8d3da83606
commit 8fdecaa487
89 changed files with 664 additions and 356 deletions
@@ -0,0 +1,98 @@
+use proc_macro::TokenStream;
+use quote::{quote, ToTokens};
+use toml::{Table, Value};
+
+use std::fs;
+use std::path::Path;
+
+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()
+}