1 files changed, 11 insertions, 1 deletions

diff --git a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretationlanguage/visualisation/PartialInterpretation2Gml.xtend b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretationlanguage/visualisation/PartialInterpretation2Gml.xtend
index 819cae00..6e948827 100644
--- a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretationlanguage/visualisation/PartialInterpretation2Gml.xtend
+++ b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage/src/hu/bme/mit/inf/dslreasoner/viatrasolver/partialinterpretationlanguage/visualisation/PartialInterpretation2Gml.xtend
@@ -12,6 +12,16 @@ import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.Relation
 import java.util.Set

 

 class PartialInterpretation2Gml {

+        def private getElements(PartialInterpretation model) {

+                return

+                        model.problem.elements +

+                        model.newElements +

+                        model.booleanelements+

+                        model.integerelements+

+                        model.stringelement+

+                        model.realelements

+        }

+        

         def public transform(PartialInterpretation i) {

                 val p = i.problem

                 val Map<DefinedElement, Integer> objectToID = new HashMap

@@ -19,7 +29,7 @@ class PartialInterpretation2Gml {
                 '''

                 graph

                 [

-                        «FOR object:p.elements + i.newElements SEPARATOR '\n'»

+                        «FOR object:i.elements SEPARATOR '\n'»

                                 «this.transformObject(object,object.typesOfElement(i),objectToID)»

                         «ENDFOR»

                         «FOR relation:i.partialrelationinterpretation»