diff options
Diffstat (limited to 'Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/Yakindu2CommonModel.xtend')
-rw-r--r-- | Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/Yakindu2CommonModel.xtend | 82 |
1 files changed, 82 insertions, 0 deletions
diff --git a/Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/Yakindu2CommonModel.xtend b/Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/Yakindu2CommonModel.xtend new file mode 100644 index 00000000..d9219d5a --- /dev/null +++ b/Domains/hu.bme.mit.inf.yakinduModelExtractor/src/hu/bme/mit/inf/yakinduModelExtractor/Yakindu2CommonModel.xtend | |||
@@ -0,0 +1,82 @@ | |||
1 | package hu.bme.mit.inf.yakinduModelExtractor | ||
2 | |||
3 | import org.yakindu.sct.model.sgraph.Statechart | ||
4 | import org.yakindu.sct.model.sgraph.SGraphPackage | ||
5 | import hu.bme.mit.inf.dslreasoner.domains.yakindu.sgraph.yakindumm.YakindummPackage | ||
6 | import hu.bme.mit.inf.dslreasoner.domains.yakindu.sgraph.yakindumm.YakindummFactory | ||
7 | import org.eclipse.emf.ecore.EObject | ||
8 | import org.eclipse.emf.ecore.EClass | ||
9 | import java.util.HashMap | ||
10 | import org.eclipse.emf.ecore.EReference | ||
11 | import java.util.List | ||
12 | |||
13 | class Yakindu2CommonModel { | ||
14 | val yakinduSGraphPackage = org.yakindu.sct.model.sgraph.SGraphPackage.eINSTANCE | ||
15 | val commonSGraphPackage = YakindummPackage.eINSTANCE | ||
16 | val extension YakindummFactory factory = YakindummFactory.eINSTANCE | ||
17 | |||
18 | def transform(Statechart s) { | ||
19 | val o2o = new HashMap | ||
20 | o2o.put(s,s.copyObject) | ||
21 | for(content: s.eAllContents.toIterable) { | ||
22 | val copied = content.copyObject | ||
23 | if(copied !== null) { | ||
24 | o2o.put(content,copied) | ||
25 | } | ||
26 | } | ||
27 | |||
28 | for(sourceObjectEntry : o2o.entrySet) { | ||
29 | val originalSource = sourceObjectEntry.key | ||
30 | val copiedSource = sourceObjectEntry.value | ||
31 | for(originalReference : originalSource.eClass.EAllReferences) { | ||
32 | if(originalReference.isMany) { | ||
33 | val originalTargets = originalSource.eGet(originalReference) as List<? extends EObject> | ||
34 | for(originalTarget : originalTargets) { | ||
35 | if(o2o.containsKey(originalTarget)) { | ||
36 | copyReference(originalReference,copiedSource,o2o.get(originalTarget)) | ||
37 | } | ||
38 | } | ||
39 | } else { | ||
40 | val originalTarget = originalSource.eGet(originalReference) as EObject | ||
41 | if(o2o.containsKey(originalTarget)) { | ||
42 | copyReference(originalReference,copiedSource,o2o.get(originalTarget)) | ||
43 | } | ||
44 | } | ||
45 | } | ||
46 | } | ||
47 | return o2o.get(s) | ||
48 | } | ||
49 | |||
50 | def private copyObject(EObject o) { | ||
51 | if(o.createCopy) { | ||
52 | val className = o.eClass.name | ||
53 | //if(className =="Exit") println("Exit") | ||
54 | val correspondingClass = commonSGraphPackage.EClassifiers.filter(EClass).filter[it.name == className].head | ||
55 | if(correspondingClass != null) { | ||
56 | return factory.create(correspondingClass) | ||
57 | } else { | ||
58 | println(className) | ||
59 | return null | ||
60 | } | ||
61 | } | ||
62 | else return null | ||
63 | } | ||
64 | |||
65 | def private copyReference(EReference reference, EObject source, EObject target) { | ||
66 | val correspondingSourceClass = commonSGraphPackage.EClassifiers.filter(EClass).filter[it.name == reference.EContainingClass.name].head | ||
67 | if(correspondingSourceClass !== null) { | ||
68 | val correspondingReference = correspondingSourceClass.EReferences.filter[it.name === reference.name].head | ||
69 | if(correspondingReference !== null) { | ||
70 | if(correspondingReference.isMany) { | ||
71 | (source.eGet(correspondingReference) as List<? super EObject>).add(target) | ||
72 | } else { | ||
73 | source.eSet(correspondingReference,target); | ||
74 | } | ||
75 | } | ||
76 | } | ||
77 | } | ||
78 | |||
79 | def private createCopy(EObject o) { | ||
80 | o.eClass.EPackage === yakinduSGraphPackage | ||
81 | } | ||
82 | } \ No newline at end of file | ||