From 58e50f31d949bce12779b9f36d33d9eedd7226a7 Mon Sep 17 00:00:00 2001
From: Nelson Tavares de Sousa <stu103017@mail.uni-kiel.de>
Date: Fri, 21 Aug 2015 14:56:12 +0200
Subject: [PATCH] added containsKey check
---
src/main/java/teetime/framework/AbstractStage.java | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/main/java/teetime/framework/AbstractStage.java b/src/main/java/teetime/framework/AbstractStage.java
index ad20973a..21ea9505 100644
--- a/src/main/java/teetime/framework/AbstractStage.java
+++ b/src/main/java/teetime/framework/AbstractStage.java
@@ -60,8 +60,9 @@ public abstract class AbstractStage extends Stage {
Class<? extends ISignal> signalClass = signal.getClass();
Set<InputPort<?>> signalReceivedInputPorts;
- signalReceivedInputPorts = signalMap.get(signalClass);
- if (signalReceivedInputPorts == null) {
+ if (signalMap.containsKey(signalClass)) {
+ signalReceivedInputPorts = signalMap.get(signalClass);
+ } else {
signalReceivedInputPorts = new HashSet<InputPort<?>>();
signalMap.put(signalClass, signalReceivedInputPorts);
}
--
GitLab