Simplify if statement with unordered_map

neuronsimulator · Aug 14, 2024 · 266eacd · 266eacd
1 parent 09d0490
commit 266eacd
Showing 1 changed file with 7 additions and 14 deletions.
diff --git a/src/nmodl/solve.cpp b/src/nmodl/solve.cpp
@@ -126,20 +126,13 @@ void solvhandler() {
         lq = lq->next;
         method = SYM(lq);
         cvodemethod_ = 0;
-        if (method && strcmp(method->name, "after_cvode") == 0) {
-            method = (Symbol*) 0;
-            lq->element.sym = (Symbol*) 0;
-            cvodemethod_ = 1;
-        }
-        if (method && strcmp(method->name, "cvode_t") == 0) {
-            method = (Symbol*) 0;
-            lq->element.sym = (Symbol*) 0;
-            cvodemethod_ = 2;
-        }
-        if (method && strcmp(method->name, "cvode_v") == 0) {
-            method = (Symbol*) 0;
-            lq->element.sym = (Symbol*) 0;
-            cvodemethod_ = 3;
+        const std::unordered_map<std::string, int> methods = {{"after_cvode", 1},
+                                                              {"cvode_t", 2},
+                                                              {"cvode_v", 3}};
+        if (method && methods.count(method->name)) {
+            cvodemethod_ = methods.at(method->name);
+            method = nullptr;
+            lq->element.sym = nullptr;
         }
         lq = lq->next;
         errstmt = LST(lq);