From 40f78ba587d2844fbf19c35c077ab92c0026174e Mon Sep 17 00:00:00 2001
From: Lukhnos Liu <lukhnos@lukhnos.org>
Date: Mon, 11 Dec 2023 06:40:38 -0800
Subject: [PATCH] Use O(1) lookup for input macros

---
 src/InputMacro.cpp | 58 +++++++++++++++++++++++++---------------------
 src/InputMacro.h   |  4 ++--
 2 files changed, 34 insertions(+), 28 deletions(-)

diff --git a/src/InputMacro.cpp b/src/InputMacro.cpp
index 5e37094..845012b 100644
--- a/src/InputMacro.cpp
+++ b/src/InputMacro.cpp
@@ -4,6 +4,7 @@
 #include <unicode/smpdtfmt.h>
 
 #include <iostream>
+#include <vector>
 
 namespace McBopomofo {
 
@@ -139,38 +140,43 @@ class InputMacroNextYearChineseZodiac : public InputMacro {
   std::string replacement() const;
 };
 
+static void AddMacro(
+    std::unordered_map<std::string, std::unique_ptr<InputMacro>>& m,
+    std::unique_ptr<InputMacro> p) {
+  m.insert({p->name(), std::move(p)});
+}
+
 InputMacroController::InputMacroController() {
-  macros_.push_back(std::make_unique<InputMacroDateTodayShort>());
-  macros_.push_back(std::make_unique<InputMacroDateTodayMedium>());
-  macros_.push_back(std::make_unique<InputMacroDateTodayMediumRoc>());
-  macros_.push_back(std::make_unique<InputMacroDateTodayMediumChinese>());
-  macros_.push_back(std::make_unique<InputMacroDateYesterdayShort>());
-  macros_.push_back(std::make_unique<InputMacroDateYesterdayMedium>());
-  macros_.push_back(std::make_unique<InputMacroDateYesterdayMediumRoc>());
-  macros_.push_back(std::make_unique<InputMacroDateYesterdayMediumChinese>());
-  macros_.push_back(std::make_unique<InputMacroDateTomorrowShort>());
-  macros_.push_back(std::make_unique<InputMacroDateTomorrowMedium>());
-  macros_.push_back(std::make_unique<InputMacroDateTomorrowMediumRoc>());
-  macros_.push_back(std::make_unique<InputMacroDateTomorrowMediumChinese>());
-  macros_.push_back(std::make_unique<InputMacroDateTimeNowShort>());
-  macros_.push_back(std::make_unique<InputMacroDateTimeNowMedium>());
-  macros_.push_back(std::make_unique<InputMacroTimeZoneStandard>());
-  macros_.push_back(std::make_unique<InputMacroTimeZoneShortGeneric>());
-  macros_.push_back(std::make_unique<InputMacroThisYearGanZhi>());
-  macros_.push_back(std::make_unique<InputMacroLastYearGanZhi>());
-  macros_.push_back(std::make_unique<InputMacroNextYearGanZhi>());
-  macros_.push_back(std::make_unique<InputMacroThisYearChineseZodiac>());
-  macros_.push_back(std::make_unique<InputMacroLastYearChineseZodiac>());
-  macros_.push_back(std::make_unique<InputMacroNextYearChineseZodiac>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTodayShort>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTodayMedium>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTodayMediumRoc>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTodayMediumChinese>());
+  AddMacro(macros_, std::make_unique<InputMacroDateYesterdayShort>());
+  AddMacro(macros_, std::make_unique<InputMacroDateYesterdayMedium>());
+  AddMacro(macros_, std::make_unique<InputMacroDateYesterdayMediumRoc>());
+  AddMacro(macros_, std::make_unique<InputMacroDateYesterdayMediumChinese>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTomorrowShort>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTomorrowMedium>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTomorrowMediumRoc>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTomorrowMediumChinese>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTimeNowShort>());
+  AddMacro(macros_, std::make_unique<InputMacroDateTimeNowMedium>());
+  AddMacro(macros_, std::make_unique<InputMacroTimeZoneStandard>());
+  AddMacro(macros_, std::make_unique<InputMacroTimeZoneShortGeneric>());
+  AddMacro(macros_, std::make_unique<InputMacroThisYearGanZhi>());
+  AddMacro(macros_, std::make_unique<InputMacroLastYearGanZhi>());
+  AddMacro(macros_, std::make_unique<InputMacroNextYearGanZhi>());
+  AddMacro(macros_, std::make_unique<InputMacroThisYearChineseZodiac>());
+  AddMacro(macros_, std::make_unique<InputMacroLastYearChineseZodiac>());
+  AddMacro(macros_, std::make_unique<InputMacroNextYearChineseZodiac>());
 }
 
 InputMacroController::~InputMacroController() {}
 
 std::string InputMacroController::handle(std::string input) {
-  for (auto& macro : macros_) {
-    if (input == macro->name()) {
-      return macro->replacement();
-    }
+  const auto& it = macros_.find(input);
+  if (it != macros_.cend()) {
+    return it->second->replacement();
   }
   return input;
 }
diff --git a/src/InputMacro.h b/src/InputMacro.h
index d32c5c1..7215f85 100644
--- a/src/InputMacro.h
+++ b/src/InputMacro.h
@@ -3,7 +3,7 @@
 
 #include <memory>
 #include <string>
-#include <vector>
+#include <unordered_map>
 
 namespace McBopomofo {
 class InputMacro {
@@ -19,7 +19,7 @@ class InputMacroController {
   std::string handle(std::string input);
 
  private:
-  std::vector<std::unique_ptr<InputMacro>> macros_;
+  std::unordered_map<std::string, std::unique_ptr<InputMacro>> macros_;
 };
 
 }  // namespace McBopomofo