aboutsummaryrefslogtreecommitdiffstats
path: root/Stochastic/hu.bme.mit.inf.dslreasoner.faulttree.transformation/src/hu/bme/mit/inf/dslreasoner/faulttree/transformation/cft2ft/EventMaterializer.xtend
blob: c9aefe51631f1a29d732fb2ca4746c7ba74cc7de (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
package hu.bme.mit.inf.dslreasoner.faulttree.transformation.cft2ft

import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.AndGateDefinition
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.BasicEventDefinition
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Component
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.EventDeclaration
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.GateDefinition
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.InputEvent
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.KOfMGateDefinition
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.OrGateDefinition
import hu.bme.mit.inf.dslreasoner.faulttree.model.cft.Output
import hu.bme.mit.inf.dslreasoner.faulttree.model.ft.ConstantEvent
import hu.bme.mit.inf.dslreasoner.faulttree.model.ft.Event
import hu.bme.mit.inf.dslreasoner.faulttree.model.ft.FtFactory
import java.util.LinkedHashSet
import java.util.Map
import org.eclipse.emf.ecore.util.EcoreUtil
import org.eclipse.xtend.lib.annotations.Data

class EventMaterializer {
	extension val FtFactory = FtFactory.eINSTANCE

	val Map<EventKey<EventDeclaration>, Event> materializationCache = newHashMap
	val Map<EventKey<InputEvent>, EventCollection> multipleInputCache = newHashMap
	val ConstantEvent falseEvent
	val ConstantEvent trueEvent
	val path = new LinkedHashSet<EventKey<? extends EventDeclaration>>

	new() {
		falseEvent = createConstantEvent => [
			failed = false
		]
		trueEvent = createConstantEvent => [
			failed = true
		]
	}

	def getOrMaterialize(Output output) {
		getOrMaterialize(output.component, output.eventDeclaration)
	}

	def Event getOrMaterialize(Component component, EventDeclaration eventDeclaration) {
		val eventKey = new EventKey(component, eventDeclaration)
		pushEventKey(eventKey)
		try {
			// computeIfAbsent cannot be used recursively, so we must manually cache the event. 
			var event = materializationCache.get(eventKey)
			if (event === null) {
				event = materialize(component, eventDeclaration)
				materializationCache.put(eventKey, event)
			}
			event
		} finally {
			popEventKey(eventKey)
		}
	}

	protected def materialize(Component component, EventDeclaration eventDeclaration) {
		val eventName = component.name + "_" + eventDeclaration.name
		val event = switch (eventDeclaration) {
			InputEvent:
				return materializeConnectedEvent(component, eventDeclaration)
			BasicEventDefinition: {
				val basicEvent = createBasicEvent
				basicEvent.distribution = EcoreUtil.copy(eventDeclaration.distribution)
				basicEvent
			}
			GateDefinition: {
				val inputs = collectInputs(component, eventDeclaration)
				val gate = switch (eventDeclaration) {
					AndGateDefinition:
						if (inputs.containsFalseEvent) {
							return falseEvent
						} else if (inputs.empty) {
							return trueEvent
						} else if (inputs.containsExactlyOneRandomEvent) {
							return inputs.toSingleRandomEvent
						} else {
							createAndGate
						}
					OrGateDefinition:
						if (inputs.containsTrueEvent) {
							return trueEvent
						} else if (inputs.empty) {
							return falseEvent
						} else if (inputs.containsExactlyOneRandomEvent) {
							return inputs.toSingleRandomEvent
						} else {
							createOrGate
						}
					KOfMGateDefinition: {
						val requiredTrueInputs = inputs.count * eventDeclaration.k / eventDeclaration.m
						val k = requiredTrueInputs - inputs.getTrueEventCount
						val m = inputs.variableEventCount
						if (k == 0) {
							return trueEvent
						} else if (k > m) {
							return falseEvent
						} else if (inputs.containsExactlyOneRandomEvent) {
							return inputs.toSingleRandomEvent
						} else if (k == 1) {
							createOrGate
						} else if (k == m) {
							createAndGate
						} else {
							val kOfMGate = createKOfMGate
							kOfMGate.k = k
							kOfMGate
						}
					}
					default:
						throw new IllegalArgumentException("Unknown gate definition: " + eventDeclaration)
				}
				gate.inputEvents.addAll(inputs.getRandomEvents)
				gate
			}
			default:
				throw new IllegalArgumentException("Unknown event declaration: " + eventDeclaration)
		}
		event.name = eventName
		event
	}

	protected def materializeConnectedEvent(Component component, InputEvent inputEvent) {
		if (inputEvent.multiple) {
			throw new IllegalArgumentException('''Cannot materialize multiple nput «component.name»_«inputEvent.name»''')
		}
		val input = findInput(component, inputEvent)
		val incomingConnections = input.incomingConnections
		if (incomingConnections.size != 1) {
			throw new IllegalArgumentException('''Input «component.name»_«inputEvent.name» has «incomingConnections.size» connections instead of 1''')
		}
		val output = incomingConnections.head.output
		getOrMaterialize(output.component, output.eventDeclaration)
	}

	protected def collectInputs(Component component, GateDefinition gateDefinition) {
		val builder = EventCollection.builder
		for (inputEventDeclaration : gateDefinition.inputEvents) {
			switch (inputEventDeclaration) {
				InputEvent case inputEventDeclaration.multiple: {
					val materializedEvents = getOrMaterializeConnectedEvents(component, inputEventDeclaration)
					builder.addAll(materializedEvents)
				}
				default:
					builder.add(getOrMaterialize(component, inputEventDeclaration))
			}
		}
		builder.build
	}

	protected def getOrMaterializeConnectedEvents(Component component, InputEvent inputEvent) {
		val inputKey = new EventKey(component, inputEvent)
		pushEventKey(inputKey)
		try {
			// computeIfAbsent cannot be used recursively, so we must manually cache the event. 
			var eventCollection = multipleInputCache.get(inputKey)
			if (eventCollection === null) {
				eventCollection = materializeConnectedEvents(component, inputEvent)
				multipleInputCache.put(inputKey, eventCollection)
			}
			eventCollection
		} finally {
			popEventKey(inputKey)
		}
	}

	protected def materializeConnectedEvents(Component component, InputEvent inputEvent) {
		val input = findInput(component, inputEvent)
		val builder = EventCollection.builder
		for (connection : input.incomingConnections) {
			val materializedEvent = getOrMaterialize(connection.output)
			builder.add(materializedEvent)
		}
		builder.build
	}

	protected def findInput(Component component, InputEvent inputEvent) {
		val input = component.inputs.findFirst[it.inputEvent == inputEvent]
		if (input === null) {
			throw new IllegalArgumentException('''No input «inputEvent» in component «component»''')
		}
		return input
	}

	private def pushEventKey(EventKey<? extends EventDeclaration> eventKey) {
		if (!path.add(eventKey)) {
			throw new IllegalStateException(
			'''Circular dependency [«FOR ancestor : path»«ancestor», «ENDFOR»«eventKey»] detected''')
		}
	}

	private def popEventKey(EventKey<? extends EventDeclaration> eventKey) {
		path.remove(eventKey)
	}

	@Data
	protected static class EventKey<T extends EventDeclaration> {
		val Component component
		val T event

		override toString() '''«component.name»_«event.name»'''
	}
}