1
0
Fork 0
forked from koniifer/ableos

another shrimple macro!

This commit is contained in:
IntoTheNight 2023-07-09 17:18:41 +05:30
parent 7ac33efcbc
commit 4252719a72
2 changed files with 29 additions and 27 deletions

View file

@ -23,36 +23,26 @@ impl DeviceTree {
let mut dt = Self { let mut dt = Self {
devices: HashMap::new(), devices: HashMap::new(),
}; };
device_tree!(dt, [
// Human input devices "Mice",
{ "Keyboards",
dt.devices.insert("Mice".to_string(), Vec::new()); "Controllers",
dt.devices.insert("Keyboards".to_string(), Vec::new()); "Generic HIDs",
dt.devices.insert("Controllers".to_string(), Vec::new()); "Disk Drives",
// Human Input Devices that do not fit into other catagories go in this spot "CD Drives",
dt.devices.insert("Generic HIDs".to_string(), Vec::new()); "Batteries",
} "Monitors",
"GPUs",
{ "CPUs",
dt.devices.insert("Disk Drives".to_string(), Vec::new()); "USB",
dt.devices.insert("CD Drives".to_string(), Vec::new()); "Serial Ports",
} "Cameras",
"Biometric Devices",
dt.devices.insert("Batteries".to_string(), Vec::new()); ]);
dt.devices.insert("Monitors".to_string(), Vec::new());
dt.devices.insert("GPUs".to_string(), Vec::new());
dt.devices.insert("CPUs".to_string(), Vec::new());
dt.devices.insert("USB".to_string(), Vec::new());
dt.devices.insert("Serial Ports".to_string(), Vec::new());
dt.devices.insert("Cameras".to_string(), Vec::new());
dt.devices
.insert("Biometric Devices".to_string(), Vec::new());
dt dt
} }
} }
use crate::utils::TAB; use crate::{utils::TAB, device_tree};
use crate::tab; use crate::tab;
impl fmt::Display for DeviceTree { impl fmt::Display for DeviceTree {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {

View file

@ -14,3 +14,15 @@ macro_rules! tab {
TAB.repeat($num) TAB.repeat($num)
} }
} }
// NOTE: this only reduces the code duplication in source code not in generated code!
// Written by Yours Truly: Munir
/// A simple macro to reduce code duplication when we insert device types into the device tree
#[macro_export]
macro_rules! device_tree {
($devtree:expr, $dev_type_vec:expr) => {
for each_device_type in $dev_type_vec {
$devtree.devices.insert(each_device_type.to_string(), Vec::new());
}
};
}