Add: check for RegisterCallback

yonghenghuanmie · Jul 19, 2022 · c254021 · c254021
1 parent 513c332
commit c254021
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/StateMachine/StateMachine.h b/StateMachine/StateMachine.h
@@ -18,7 +18,7 @@ class StateMachine :public Graph<StateType, Operation>
 		for (auto&& [new_state, op] : paths)
 			if (op == operation)
 			{
-				if (callback_map[new_state](derive, state, new_state))
+				if (!callback_map.contains(new_state) || callback_map[new_state](derive, state, new_state))
 				{
 					state = new_state;
 					return true;