diff options
Diffstat (limited to 'Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation')
2 files changed, 146 insertions, 0 deletions
diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation/ConstraintCollection.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation/ConstraintCollection.xtend new file mode 100644 index 00000000..685c5836 --- /dev/null +++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation/ConstraintCollection.xtend | |||
@@ -0,0 +1,80 @@ | |||
1 | package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.validation | ||
2 | |||
3 | import java.util.ArrayList | ||
4 | import java.util.HashMap | ||
5 | import java.util.List | ||
6 | import java.util.Map | ||
7 | import org.eclipse.emf.common.notify.Notifier | ||
8 | import org.eclipse.emf.common.util.URI | ||
9 | import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl | ||
10 | import org.eclipse.viatra.addon.validation.core.api.IConstraintSpecification | ||
11 | import org.eclipse.viatra.query.runtime.api.ViatraQueryEngine | ||
12 | import org.eclipse.viatra.query.runtime.api.impl.BaseGeneratedPatternGroup | ||
13 | import org.eclipse.viatra.query.runtime.emf.EMFScope | ||
14 | |||
15 | class ConstraintCollection{ | ||
16 | val constraints = new ArrayList<IConstraintSpecification>(); | ||
17 | var BaseGeneratedPatternGroup patterns; | ||
18 | var List<Notifier> resources = new ArrayList<Notifier>(); | ||
19 | |||
20 | |||
21 | new(List<IConstraintSpecification> constraints, List<String> uris, BaseGeneratedPatternGroup patterns){ | ||
22 | this.constraints.addAll(constraints); | ||
23 | this.patterns = patterns; | ||
24 | setURIs(uris); | ||
25 | } | ||
26 | |||
27 | new(List<IConstraintSpecification> constraints, BaseGeneratedPatternGroup patterns){ | ||
28 | this.constraints.addAll(constraints); | ||
29 | this.patterns = patterns; | ||
30 | } | ||
31 | |||
32 | def addModel(Notifier n ){ | ||
33 | resources.add(n); | ||
34 | } | ||
35 | |||
36 | def setURIs(List<String> uris){ | ||
37 | val resSet = new ResourceSetImpl(); | ||
38 | |||
39 | for(uri : uris){ | ||
40 | var resource = resSet.getResource(URI.createURI(uri), true); | ||
41 | resources.add(resource); | ||
42 | } | ||
43 | |||
44 | println('reading model finished') | ||
45 | } | ||
46 | |||
47 | def List<Integer> calculateViolations(){ | ||
48 | var results = new ArrayList<Integer>(); | ||
49 | |||
50 | for(resource : resources){ | ||
51 | val engine = initEngine(resource); | ||
52 | var matches = constraints.stream.mapToInt([ele| ele.querySpecification.getMatcher(engine).countMatches]).sum(); | ||
53 | results.add(matches); | ||
54 | } | ||
55 | |||
56 | return results; | ||
57 | } | ||
58 | |||
59 | def ArrayList<Map<String, Integer>> calculateViolationMaps(){ | ||
60 | val result = new ArrayList<Map<String, Integer>>() | ||
61 | |||
62 | for(resource : resources){ | ||
63 | val map = new HashMap<String, Integer>(); | ||
64 | val engine = initEngine(resource); | ||
65 | constraints.forEach[ | ||
66 | var count = it.querySpecification.getMatcher(engine).countMatches; | ||
67 | map.put(it.querySpecification.simpleName, count); | ||
68 | ]; | ||
69 | result.add(map); | ||
70 | } | ||
71 | return result; | ||
72 | } | ||
73 | |||
74 | private def initEngine(Notifier r){ | ||
75 | var engine = ViatraQueryEngine.on(new EMFScope(r)); | ||
76 | //init patterns with the new engine | ||
77 | patterns.prepare(engine); | ||
78 | return engine; | ||
79 | } | ||
80 | } | ||
diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation/ViolationCheck.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation/ViolationCheck.xtend new file mode 100644 index 00000000..4db5f940 --- /dev/null +++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/validation/ViolationCheck.xtend | |||
@@ -0,0 +1,66 @@ | |||
1 | package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.validation | ||
2 | |||
3 | import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app.Domain | ||
4 | import com.google.common.reflect.ClassPath | ||
5 | import ecore.Ecore | ||
6 | import hu.bme.mit.inf.dslreasoner.partialsnapshot_mavo.yakindu.Patterns | ||
7 | import java.util.ArrayList | ||
8 | import org.eclipse.emf.ecore.EObject | ||
9 | import org.eclipse.viatra.addon.validation.core.api.IConstraintSpecification | ||
10 | |||
11 | class ViolationCheck { | ||
12 | /** | ||
13 | * Return the total number of violations | ||
14 | */ | ||
15 | def static int calculateViolationCounts(EObject root, Domain d){ | ||
16 | var packageName = ''; | ||
17 | if(d == Domain.Yakindumm){ | ||
18 | packageName = 'constraints.yakindumm'; | ||
19 | }else if (d == Domain.Ecore){ | ||
20 | packageName = 'constraints.ecore'; | ||
21 | }else if (d == Domain.Github){ | ||
22 | return -1; | ||
23 | } | ||
24 | |||
25 | var constriants = loadConstraints(packageName); | ||
26 | var collections = new ConstraintCollection(constriants, Ecore.instance); | ||
27 | collections.addModel(root); | ||
28 | var results = collections.calculateViolations(); | ||
29 | if(results.size > 0){ | ||
30 | return results.get(0); | ||
31 | }else{ | ||
32 | throw new IllegalArgumentException("Calculate Violation Failed"); | ||
33 | } | ||
34 | } | ||
35 | |||
36 | /** | ||
37 | * return a map contain the count for each type of violation | ||
38 | */ | ||
39 | def static violationMaps(EObject root){ | ||
40 | var constriants = loadConstraints('hu.bme.mit.inf.dslreasoner.partialsnapshot_mavo.yakindu'); | ||
41 | var collections = new ConstraintCollection(constriants, Patterns.instance); | ||
42 | collections.addModel(root); | ||
43 | var results = collections.calculateViolationMaps(); | ||
44 | if(results.size > 0){ | ||
45 | return results.get(0); | ||
46 | }else{ | ||
47 | throw new IllegalArgumentException("Calculate Violation Failed"); | ||
48 | } | ||
49 | } | ||
50 | |||
51 | def static loadConstraints(String packageName){ | ||
52 | val constraints = new ArrayList<IConstraintSpecification>(); | ||
53 | |||
54 | val classPath = ClassPath.from(ClassLoader.systemClassLoader); | ||
55 | val classInfos = classPath.getTopLevelClasses(packageName); | ||
56 | for(info : classInfos){ | ||
57 | if(info.load.interfaces.contains(IConstraintSpecification)){ | ||
58 | //IConstraintSpecification only has one constructor with empty argument list | ||
59 | var constructor = info.load.constructors.get(0); | ||
60 | var instance = constructor.newInstance(); | ||
61 | constraints.add(instance as IConstraintSpecification); | ||
62 | } | ||
63 | } | ||
64 | return constraints | ||
65 | } | ||
66 | } | ||