Scope support for attributes

2 files changed, 94 insertions, 9 deletions

diff --git a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/InstanceModel2PartialInterpretation.xtend b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/InstanceModel2PartialInterpretation.xtend
index f4fc1fce..bd0f4ca4 100644
--- a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/InstanceModel2PartialInterpretation.xtend
+++ b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/InstanceModel2PartialInterpretation.xtend
@@ -20,6 +20,10 @@ import org.eclipse.emf.ecore.EObject
 import org.eclipse.emf.ecore.resource.Resource

 

 import static extension hu.bme.mit.inf.dslreasoner.util.CollectionsUtil.*

+import java.util.HashSet

+import java.util.Set

+import java.math.BigDecimal

+import org.eclipse.emf.ecore.EAttribute

 

 class InstanceModel2PartialInterpretation {

         val extension LogiclanguageFactory factory = LogiclanguageFactory.eINSTANCE

@@ -43,7 +47,12 @@ class InstanceModel2PartialInterpretation {
         {

                 val problem = metamodelTranslationResult.output

                 val ecore2LogicTrace = metamodelTranslationResult.trace

-                val tracedOutput = partialInterpretationInitialiser.initialisePartialInterpretation(problem, new TypeScopes)

+                

+                val referencesUsed = ecore2Logic.allReferencesInScope(ecore2LogicTrace).toSet

+                val attributesUsed = ecore2Logic.allAttributesInScope(ecore2LogicTrace).toSet

+                

+                val typeScope = createTypeScopesFromKnownAttributeValues(objects,attributesUsed)

+                val tracedOutput = partialInterpretationInitialiser.initialisePartialInterpretation(problem, typeScope)

                 val partialInterpretation = tracedOutput.output

                 val partialInterpretationTrace = tracedOutput.trace

                 

@@ -54,7 +63,7 @@ class InstanceModel2PartialInterpretation {
                         val object = objects.get(objectIndex)

                         val element = createDefinedElement => [

                                 it.name = if(withID) 

-                                        { '''o «objectIndex»''' } else 

+                                        { '''o «objectIndex+1»''' } else 

                                         { null }

                         ]

                         

@@ -71,8 +80,6 @@ class InstanceModel2PartialInterpretation {
                         object2DefinedElement.put(object, element)

                 }

                 

-                val referencesUsed = ecore2Logic.allReferencesInScope(ecore2LogicTrace).toSet

-                val attributesUsed = ecore2Logic.allAttributesInScope(ecore2LogicTrace).toSet

                 for(source : objects) {

                         // Translating the references

                         for(reference : source.eClass.EAllReferences.filter[

@@ -126,6 +133,52 @@ class InstanceModel2PartialInterpretation {
                 return partialInterpretation

         }

         

+        private def createTypeScopesFromKnownAttributeValues(List<EObject> objects, Set<EAttribute> attributesUsed) {

+                val Set<Integer> integers = new HashSet

+                val Set<BigDecimal> reals = new HashSet

+                val Set<String> strings = new HashSet

+                for(object: objects) {

+                        for(attribute : object.eClass.EAllAttributes.filter[attributesUsed.contains(it)]) {

+                                val value = object.eGet(attribute)

+                                if(value !== null) {

+                                        if(value instanceof List<?>) {

+                                                for(v : value) {

+                                                        shortValue(v,integers,reals,strings)

+                                                }

+                                        } else {

+                                                shortValue(value,integers,reals,strings)

+                                        }

+                                }

+                        }

+                }

+                return new TypeScopes => [

+                        it.knownIntegers += integers

+                        it.knownReals += reals

+                        it.knownStrings += strings

+                ]

+        }

+        private def dispatch shortValue(Boolean value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                // Do nothing

+        }

+        private def dispatch shortValue(Integer value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                integers += value

+        }

+        private def dispatch shortValue(Float value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                reals += BigDecimal.valueOf(value)

+        }

+        private def dispatch shortValue(Double value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                reals += BigDecimal.valueOf(value)

+        }

+        private def dispatch shortValue(String value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                strings += value

+        }

+        private def dispatch shortValue(Void value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                // Do nothing

+        }

+        private def dispatch shortValue(Object value, Set<Integer> integers, Set<BigDecimal> reals, Set<String> strings) {

+                // Do nothing

+        }

+        

         protected def translateLink(PartialRelationInterpretation interpretation, DefinedElement source, DefinedElement target) {

                 interpretation.relationlinks += createBinaryElementRelationLink => [it.param1 = source it.param2 = target]

         }

@@ -138,7 +191,23 @@ class InstanceModel2PartialInterpretation {
                 else throw new AssertionError('''term should be a defined element?''')

         }

         

-        dispatch protected def translateValue(Object value, Ecore2Logic_Trace ecore2LogicTrace, Problem2PartialInterpretationTrace partialInterpretationTrace) {

-                //throw new UnsupportedOperationException('''Mapping of «value.class.simpleName» in partial models is currently not supported!''')

+        dispatch protected def translateValue(Boolean value, Ecore2Logic_Trace ecore2LogicTrace, Problem2PartialInterpretationTrace partialInterpretationTrace) {

+                value.lookup(partialInterpretationTrace.primitiveValues.booleanMap)

+        }

+        

+        dispatch protected def translateValue(Integer value, Ecore2Logic_Trace ecore2LogicTrace, Problem2PartialInterpretationTrace partialInterpretationTrace) {

+                value.lookup(partialInterpretationTrace.primitiveValues.integerMap)

+        }

+        

+        dispatch protected def translateValue(Double value, Ecore2Logic_Trace ecore2LogicTrace, Problem2PartialInterpretationTrace partialInterpretationTrace) {

+                BigDecimal.valueOf(value).lookup(partialInterpretationTrace.primitiveValues.realMap)

+        }

+        

+        dispatch protected def translateValue(Float value, Ecore2Logic_Trace ecore2LogicTrace, Problem2PartialInterpretationTrace partialInterpretationTrace) {

+                BigDecimal.valueOf(value).lookup(partialInterpretationTrace.primitiveValues.realMap)

+        }

+        

+        dispatch protected def translateValue(String value, Ecore2Logic_Trace ecore2LogicTrace, Problem2PartialInterpretationTrace partialInterpretationTrace) {

+                value.lookup(partialInterpretationTrace.primitiveValues.stringMap)

         }

 }
\ No newline at end of file
diff --git a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/PartialInterpretation2Logic.xtend b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/PartialInterpretation2Logic.xtend
index 5d8d9313..bba4ae92 100644
--- a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/PartialInterpretation2Logic.xtend
+++ b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretation2logic/PartialInterpretation2Logic.xtend
@@ -24,6 +24,9 @@ import org.eclipse.xtend.lib.annotations.Data
 

 import static extension hu.bme.mit.inf.dslreasoner.util.CollectionsUtil.*

 import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PrimitiveElement

+import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.BooleanElement

+import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.IntegerElement

+import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.RealElement

 

 @Data class PartialInterpretation2Logic_Trace {

         Map<DefinedElement,DefinedElement> new2Old = new HashMap

@@ -148,7 +151,7 @@ class PartialInterpretation2Logic {
                         val ^annotation= Partial2logicannotationsFactory.eINSTANCE.createPartialModelRelation2Assertion

                         ^annotation.target = assertion

                         ^annotation.targetRelation = relation

-                        ^annotation.links += links.map[EcoreUtil.copy(it)]

+                        //^annotation.links += links.map[EcoreUtil.copy(it)]

                         //val error= assertion.eAllContents.toIterable.filter(SymbolicValue).filter[it.symbolicReference === null] 

                         //error.forEach[println("error")]

                         p.add(assertion)

@@ -162,12 +165,25 @@ class PartialInterpretation2Logic {
                         if((link.param1 !== null) && (link.param2 !== null)) {

                                 return createSymbolicValue=>[

                                         it.symbolicReference=relationDeclaration

-                                        it.parameterSubstitutions += createSymbolicValue => [it.symbolicReference = link.param1]

-                                        it.parameterSubstitutions += createSymbolicValue => [it.symbolicReference = link.param2]

+                                        it.parameterSubstitutions += createValueInLink(link.param1)

+                                        it.parameterSubstitutions += createValueInLink(link.param2)

                                 ]

                         } else {

                                 throw new IllegalArgumentException

                         }

                 } else throw new UnsupportedOperationException

         }

+        

+        def private dispatch createValueInLink(BooleanElement element) {

+                return element.value.asTerm

+        }

+        def private dispatch createValueInLink(IntegerElement element) {

+                return element.value.asTerm

+        }

+        def private dispatch createValueInLink(RealElement element) {

+                return element.value.asTerm

+        }

+        def private dispatch createValueInLink(DefinedElement element) {

+                return createSymbolicValue => [it.symbolicReference = element]

+        }

 }
\ No newline at end of file