3 files changed, 34 insertions, 4 deletions

diff --git a/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentFaultTreeTrace.xtend b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentFaultTreeTrace.xtend
index 7a3e377b..10c91fb4 100644
--- a/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentFaultTreeTrace.xtend
+++ b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentFaultTreeTrace.xtend
@@ -2,6 +2,7 @@ package hu.bme.mit.inf.dslreasoner.faulttree.transformation.ecore2cft
 
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.CftFactory
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.ComponentDefinition
+import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Modality
 import java.util.Map
 import org.eclipse.viatra.query.runtime.api.IPatternMatch
 import org.eclipse.xtend.lib.annotations.Accessors
@@ -13,10 +14,16 @@ class ComponentFaultTreeTrace {
         val Map<IPatternMatch, ComponentInstanceTrace> componentInstancesMap = newHashMap
 
         def instantiateComponent(IPatternMatch patternMatch, ComponentDefinition componenDefinition) {
+                instantiateComponent(patternMatch, componenDefinition, Modality.MUST, false)
+        }
+
+        def instantiateComponent(IPatternMatch patternMatch, ComponentDefinition componenDefinition, Modality exists,
+                boolean allowMultiple) {
                 if (componentInstancesMap.containsKey(patternMatch)) {
                         throw new IllegalArgumentException("Already instantiated component for match: " + patternMatch)
                 }
-                val componentTrace = new ComponentInstanceTrace(componentFaultTree, componenDefinition, nameGenerator)
+                val componentTrace = new ComponentInstanceTrace(componentFaultTree, componenDefinition, nameGenerator,
+                        exists, allowMultiple)
                 componentInstancesMap.put(patternMatch, componentTrace)
                 componentTrace
         }
@@ -30,6 +37,9 @@ class ComponentFaultTreeTrace {
                         throw new IllegalArgumentException("Top level component must have 1 output, got " + outputs.size +
                                 " instead")
                 }
+                if (!trace.appearsExactlyOnce) {
+                        throw new IllegalArgumentException("Top level must appear in the fault tree exactly once")
+                }
                 componentFaultTree.topEvent = outputs.head
         }
 
diff --git a/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentInstanceTrace.xtend b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentInstanceTrace.xtend
index 7353bfe5..158ab2e1 100644
--- a/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentInstanceTrace.xtend
+++ b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/ComponentInstanceTrace.xtend
@@ -6,18 +6,23 @@ import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.ComponentDefinition
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.ComponentFaultTree
 import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.EventDeclaration
 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 ComponentInstanceTrace {
         val componentInstance = CftFactory.eINSTANCE.createComponent
         val Map<InputEvent, InputTrace> inputEventsMap
         val Map<EventDeclaration, Output> outputEventsMap
 
         protected new(ComponentFaultTree faultTree, ComponentDefinition componentDefinition,
-                ComponentNameGenerator nameGenerator) {
+                ComponentNameGenerator nameGenerator, Modality exists, boolean multipleAllowed) {
                 componentInstance.componentDefinition = componentDefinition
                 componentInstance.name = nameGenerator.nextName(componentDefinition)
+                componentInstance.exists = exists
+                componentInstance.multipleAllowed = multipleAllowed
                 inputEventsMap = Maps.newHashMapWithExpectedSize(componentDefinition.inputEvents.size)
                 for (inputEvent : componentDefinition.inputEvents) {
                         val inputTrace = new InputTrace(componentInstance, inputEvent)
@@ -34,6 +39,11 @@ class ComponentInstanceTrace {
         }
 
         def void assign(EventDeclaration inputEvent, ComponentInstanceTrace sourceComponent, EventDeclaration outputEvent) {
+                assign(inputEvent, sourceComponent, outputEvent, Modality.MUST)
+        }
+
+        def void assign(EventDeclaration inputEvent, ComponentInstanceTrace sourceComponent, EventDeclaration outputEvent,
+                Modality exists) {
                 val inputTrace = inputEventsMap.get(inputEvent)
                 if (inputTrace === null) {
                         throw new IllegalArgumentException("Unknown input: " + inputEvent)
@@ -42,10 +52,14 @@ class ComponentInstanceTrace {
                 if (output === null) {
                         throw new IllegalArgumentException("Unknown output: " + outputEvent)
                 }
-                inputTrace.assign(output)
+                inputTrace.assign(output, exists)
         }
 
         protected def getOutputs() {
                 componentInstance.outputs
         }
+
+        protected def appearsExactlyOnce() {
+                componentInstance.appearsExactlyOnce
+        }
 }
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
                 }
         }
 }