Avoid double lookups on the hot path

No effect on my naive benchmark.
DOMjudge · Oct 20, 2019 · 4bd35d9 · 4bd35d9
1 parent a49b48e
commit 4bd35d9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/libchecktestdata.cc b/libchecktestdata.cc
@@ -216,8 +216,14 @@ value_t getvar(const expr& var, int use_preset = 0)
 		if ( gendata && preset.count(var.val) && preset[var.val].count(ind) ) {
 			return preset[var.val][ind];
 		}
-		if ( variable.count(var.val) && variable[var.val].count(ind) ) {
-			return variable[var.val][ind];
+		// Avoid double lookups on the hot path
+		auto it = variable.find(var.val);
+		if (it != variable.end()) {
+			auto& map = it->second;
+			auto it2 = map.find(ind);
+			if (it2 != map.end()) {
+				return it2->second;
+			}
 		}
 	}
 	cerr << "variable " << var << " undefined in " << program[prognr] << endl;