1 files changed, 29 insertions, 0 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
new file mode 100644
index 00000000..c529a09b
--- /dev/null
+++ b/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/ecore2cft/InputTrace.xtend
@@ -0,0 +1,29 @@
+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.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.Output
+import java.util.Map
+
+class InputTrace {
+        val Input input = CftFactory.eINSTANCE.createInput
+        val Map<Output, Connection> connectionsMap = newHashMap
+
+        protected new(Component component, InputEvent inputEvent) {
+                input.inputEvent = inputEvent
+                component.inputs += input
+        }
+
+        def void assign(Output output) {
+                val connection = connectionsMap.get(output)
+                if (connection === null) {
+                        val newConnection = CftFactory.eINSTANCE.createConnection
+                        newConnection.output = output
+                        input.incomingConnections += newConnection
+                        connectionsMap.put(output, newConnection)
+                }
+        }
+}