1 files changed, 66 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/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 @@
+package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.validation
+
+import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app.Domain
+import com.google.common.reflect.ClassPath
+import ecore.Ecore
+import hu.bme.mit.inf.dslreasoner.partialsnapshot_mavo.yakindu.Patterns
+import java.util.ArrayList
+import org.eclipse.emf.ecore.EObject
+import org.eclipse.viatra.addon.validation.core.api.IConstraintSpecification
+
+class ViolationCheck {
+        /**
+         * Return the total number of violations
+         */
+        def static int calculateViolationCounts(EObject root, Domain d){
+                var packageName = '';
+                if(d == Domain.Yakindumm){
+                        packageName = 'constraints.yakindumm';
+                }else if (d == Domain.Ecore){
+                        packageName = 'constraints.ecore';
+                }else if (d == Domain.Github){
+                        return -1;
+                }
+                
+                var constriants = loadConstraints(packageName);
+                var collections = new ConstraintCollection(constriants, Ecore.instance);
+                collections.addModel(root);
+                var results = collections.calculateViolations();
+                if(results.size > 0){
+                        return results.get(0);
+                }else{
+                        throw new IllegalArgumentException("Calculate Violation Failed");
+                }
+        }
+        
+        /**
+         * return a map contain the count for each type of violation
+         */
+        def static violationMaps(EObject root){
+                var constriants = loadConstraints('hu.bme.mit.inf.dslreasoner.partialsnapshot_mavo.yakindu');
+                var collections = new ConstraintCollection(constriants, Patterns.instance);
+                collections.addModel(root);
+                var results = collections.calculateViolationMaps();
+                if(results.size > 0){
+                        return results.get(0);
+                }else{
+                        throw new IllegalArgumentException("Calculate Violation Failed");
+                }
+        }
+        
+        def static loadConstraints(String packageName){
+                val constraints = new ArrayList<IConstraintSpecification>();
+                
+                val classPath = ClassPath.from(ClassLoader.systemClassLoader);
+                val classInfos = classPath.getTopLevelClasses(packageName);
+                for(info : classInfos){
+                        if(info.load.interfaces.contains(IConstraintSpecification)){
+                                //IConstraintSpecification only has one constructor with empty argument list
+                                var constructor = info.load.constructors.get(0);
+                                var instance = constructor.newInstance();
+                                constraints.add(instance as IConstraintSpecification);
+                        }
+                }
+                return constraints
+        }
+}