From 23903a969dd6a8fc10d8085f90197c8cbedd5171 Mon Sep 17 00:00:00 2001
From: joevt <joevt@shaw.ca>
Date: Mon, 18 Dec 2023 01:17:41 -0800
Subject: [PATCH] Add get_comp_by_name_optional.

---
 machines/machinebase.cpp | 8 ++++++++
 machines/machinebase.h   | 1 +
 2 files changed, 9 insertions(+)

diff --git a/machines/machinebase.cpp b/machines/machinebase.cpp
index a420f56..6c53204 100644
--- a/machines/machinebase.cpp
+++ b/machines/machinebase.cpp
@@ -58,6 +58,14 @@ HWComponent* MachineBase::get_comp_by_name(std::string name) {
     }
 }
 
+HWComponent* MachineBase::get_comp_by_name_optional(std::string name) {
+    if (this->device_map.count(name)) {
+        return this->device_map[name].get();
+    } else {
+        return nullptr;
+    }
+}
+
 HWComponent* MachineBase::get_comp_by_type(HWCompType type) {
     std::string comp_name;
     bool found = false;
diff --git a/machines/machinebase.h b/machines/machinebase.h
index a2df568..da8bb50 100644
--- a/machines/machinebase.h
+++ b/machines/machinebase.h
@@ -41,6 +41,7 @@ public:
 
     void add_device(std::string name, std::unique_ptr<HWComponent> dev_obj);
     HWComponent* get_comp_by_name(std::string name);
+    HWComponent* get_comp_by_name_optional(std::string name);
     HWComponent* get_comp_by_type(HWCompType type);
     int postinit_devices();