1 files changed, 7 insertions, 1 deletions

diff --git a/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/InputTrace.xtend b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/InputTrace.xtend
index c529a09b..b892eff1 100644
--- a/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/InputTrace.xtend
+++ b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/InputTrace.xtend
@@ -5,9 +5,12 @@ import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Component
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Connection
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Input
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.InputEvent
+import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Modality
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Output
 import java.util.Map
 
+import static extension hu.bme.mit.inf.dslreasoner.faulttree.model.util.CftExtensions.*
+
 class InputTrace {
         val Input input = CftFactory.eINSTANCE.createInput
         val Map<Output, Connection> connectionsMap = newHashMap
@@ -17,13 +20,16 @@ class InputTrace {
                 component.inputs += input
         }
 
-        def void assign(Output output) {
+        def void assign(Output output, Modality exists) {
                 val connection = connectionsMap.get(output)
                 if (connection === null) {
                         val newConnection = CftFactory.eINSTANCE.createConnection
                         newConnection.output = output
+                        newConnection.exists = exists
                         input.incomingConnections += newConnection
                         connectionsMap.put(output, newConnection)
+                } else if (exists.isMoreConcreteThan(connection.exists)) {
+                        connection.exists = exists
                 }
         }
 }