getSinks and getSources will now return an empty set if the given typ…

…e does not exist in the graph
uber · Dec 2, 2019 · 56b1729 · 56b1729
1 parent 173c586
commit 56b1729
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/src/main/kotlin/motif/core/ResolvedGraph.kt b/core/src/main/kotlin/motif/core/ResolvedGraph.kt
@@ -166,11 +166,11 @@ private class ValidResolvedGraph(
 
     override fun getSources(scope: Scope) = scopeStates.getValue(scope).sourceToSinks.keys.filter { it.scope == scope }
 
-    override fun getSinks(type: Type) = graphState.sinks.getValue(type)
+    override fun getSinks(type: Type) = graphState.sinks[type] ?: emptySet<Sink>()
 
-    override fun getSinks(irType: IrType) = graphState.irTypeToSinks.getValue(irType)
+    override fun getSinks(irType: IrType) = graphState.irTypeToSinks[irType] ?: emptySet<Sink>()
 
-    override fun getSources(irType: IrType) = graphState.irTypeToSources.getValue(irType)
+    override fun getSources(irType: IrType) = graphState.irTypeToSources[irType] ?: emptySet<Source>()
 
     override fun getSinks(scope: Scope) = scopeSinks.computeIfAbsent(scope) {
         scopeStates.getValue(scope).sinks.values.flatMap { sinks ->