aboutsummaryrefslogtreecommitdiffstats
path: root/Framework/hu.bme.mit.inf.dslreasoner.ecore2logic/src/hu/bme/mit/inf/dslreasoner/ecore2logic/EClassMapper.xtend
blob: 82abd252ae68d8656204c879533988029b0fa5a0 (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
package hu.bme.mit.inf.dslreasoner.ecore2logic

import hu.bme.mit.inf.dslreasoner.logic.model.builder.LogicProblemBuilder
import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.Type
import hu.bme.mit.inf.dslreasoner.logic.model.logicproblem.LogicProblem
import java.util.Map
import org.eclipse.emf.ecore.EClass
import java.util.HashMap

interface EClassMapper {
	def void transformEClasses(Ecore2Logic_Trace trace, LogicProblem problem, Iterable<EClass> classes, int numberOfObjects, EcoreMetamodelDescriptor metamodelDescriptor)
	def Iterable<EClass> allClassesInScope(Ecore2Logic_Trace trace) 
	def Type TypeofEClass(Ecore2Logic_Trace trace, EClass type)
}

class EClassMapper_AllElementAsObject_Trace implements Trace<EClassMapper_AllElementAsObject>{
	public var Map<EClass, Type> typeMap = new HashMap;
}

class EClassMapper_AllElementAsObject implements EClassMapper{
	val extension LogicProblemBuilder builder = new LogicProblemBuilder
	
	override transformEClasses(Ecore2Logic_Trace trace, LogicProblem problem,
		Iterable<EClass> classes, int numberOfObjects, EcoreMetamodelDescriptor metamodelDescriptor
	) {
		val classMapperTrace = new EClassMapper_AllElementAsObject_Trace
		trace.classMapperTrace = classMapperTrace
		for(c:classes) {
			val logicType = problem.add(TypeDeclaration('''«c.name» class''',c.isAbstract || c.isInterface))
			classMapperTrace.typeMap.put(c,logicType)
		}
		for(c:classes) {
			for(s : c.ESuperTypes) {
				Supertype(classMapperTrace.typeMap.get(c),classMapperTrace.typeMap.get(s))
			}
		}
	}
	
	def asTrace(Trace<? extends EClassMapper> o) { o as EClassMapper_AllElementAsObject_Trace }
	
	override TypeofEClass(Ecore2Logic_Trace trace, EClass type) {
		val typeMap = trace.classMapperTrace.asTrace.typeMap
		if(typeMap.containsKey(type)) {
			typeMap.get(type)
		} else {
			throw new IllegalArgumentException('''Class «type.name» is not translated to logic!''')
		}
	}
	
	override allClassesInScope(Ecore2Logic_Trace trace) {
		trace.classMapperTrace.asTrace.typeMap.keySet
	}
}