From a620f07468780778bd55dcffc30245def37ece69 Mon Sep 17 00:00:00 2001 From: Kristóf Marussy Date: Thu, 6 Aug 2020 16:07:16 +0200 Subject: MoDeS3 unit propagation WIP --- .../src/modes3/run/Modes3ModelGenerator.xtend | 222 +++++++++++++++++++++ 1 file changed, 222 insertions(+) create mode 100644 Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/Modes3ModelGenerator.xtend (limited to 'Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/Modes3ModelGenerator.xtend') diff --git a/Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/Modes3ModelGenerator.xtend b/Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/Modes3ModelGenerator.xtend new file mode 100644 index 00000000..71d1798f --- /dev/null +++ b/Domains/ca.mcgill.rtgmrt.example.modes3/src/modes3/run/Modes3ModelGenerator.xtend @@ -0,0 +1,222 @@ +package modes3.run + +import com.google.common.collect.ImmutableList +import com.google.common.collect.ImmutableSet +import hu.bme.mit.inf.dslreasoner.ecore2logic.EReferenceMapper_RelationsOverTypes_Trace +import hu.bme.mit.inf.dslreasoner.ecore2logic.Ecore2Logic +import hu.bme.mit.inf.dslreasoner.ecore2logic.Ecore2LogicConfiguration +import hu.bme.mit.inf.dslreasoner.ecore2logic.EcoreMetamodelDescriptor +import hu.bme.mit.inf.dslreasoner.ecore2logic.ecore2logicannotations.Ecore2logicannotationsFactory +import hu.bme.mit.inf.dslreasoner.ecore2logic.ecore2logicannotations.Ecore2logicannotationsPackage +import hu.bme.mit.inf.dslreasoner.logic.model.builder.LogicProblemBuilder +import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.LogiclanguagePackage +import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.TypeDefinition +import hu.bme.mit.inf.dslreasoner.logic.model.logicproblem.LogicproblemPackage +import hu.bme.mit.inf.dslreasoner.logic.model.logicresult.ModelResult +import hu.bme.mit.inf.dslreasoner.viatra2logic.Viatra2Logic +import hu.bme.mit.inf.dslreasoner.viatra2logic.Viatra2LogicConfiguration +import hu.bme.mit.inf.dslreasoner.viatra2logic.ViatraQuerySetDescriptor +import hu.bme.mit.inf.dslreasoner.viatra2logic.viatra2logicannotations.Viatra2LogicAnnotationsPackage +import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.TypeInferenceMethod +import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.cardinality.PolyhedralScopePropagatorConstraints +import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.cardinality.PolyhedralScopePropagatorSolver +import hu.bme.mit.inf.dslreasoner.viatrasolver.logic2viatra.cardinality.ScopePropagatorStrategy +import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretation2logic.InstanceModel2Logic +import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialComplexTypeInterpretation +import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialInterpretation +import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialinterpretationPackage +import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.visualisation.PartialInterpretation2Gml +import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.StateCoderStrategy +import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.ViatraReasoner +import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.ViatraReasonerConfiguration +import hu.bme.mit.inf.dslreasoner.visualisation.pi2graphviz.GraphvizVisualiser +import hu.bme.mit.inf.dslreasoner.workspace.FileSystemWorkspace +import java.util.List +import modes3.Modes3Factory +import modes3.Modes3Package +import modes3.queries.Modes3Queries +import org.eclipse.emf.ecore.EClass +import org.eclipse.emf.ecore.EObject +import org.eclipse.emf.ecore.resource.Resource +import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl +import org.eclipse.viatra.query.patternlanguage.emf.EMFPatternLanguageStandaloneSetup +import org.eclipse.viatra.query.runtime.api.ViatraQueryEngineOptions +import org.eclipse.viatra.query.runtime.localsearch.matcher.integration.LocalSearchEMFBackendFactory +import org.eclipse.viatra.query.runtime.rete.matcher.ReteBackendFactory +import org.eclipse.xtend.lib.annotations.FinalFieldsConstructor +import hu.bme.mit.inf.dslreasoner.logic.model.logiclanguage.RelationDefinition +import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.BinaryElementRelationLink + +@FinalFieldsConstructor +class Modes3ModelGenerator { + val MonitoringQuery monitoringQuery + val int modelSize + + val ecore2Logic = new Ecore2Logic + val instanceModel2Logic = new InstanceModel2Logic + val viatra2Logic = new Viatra2Logic(ecore2Logic) + val solver = new ViatraReasoner + extension val LogicProblemBuilder = new LogicProblemBuilder + + def generate() { + val metamodel = createMetamodelDescriptor() + val metamodelLogic = ecore2Logic.transformMetamodel(metamodel, new Ecore2LogicConfiguration) + val segment = ecore2Logic.TypeofEClass(metamodelLogic.trace, Modes3Package.eINSTANCE.segment) + val connectedTo = ecore2Logic.relationOfReference(metamodelLogic.trace, + Modes3Package.eINSTANCE.segment_ConnectedTo) + val connectedToIndicator = (metamodelLogic.trace. + referenceMapperTrace as EReferenceMapper_RelationsOverTypes_Trace).indicators.get( + Modes3Package.eINSTANCE.segment_ConnectedTo) + val inverseAssertion = Assertion( + '''oppositeReference «connectedTo.name» «connectedTo.name»''', + Forall[ + val src = addVar('''src''', segment) + val trg = addVar('''trg''', segment) + connectedToIndicator.call(src, trg) <=> connectedToIndicator.call(trg, src) + ] + ) + metamodelLogic.output.assertions += inverseAssertion + val inverseAnnotation = Ecore2logicannotationsFactory.eINSTANCE.createInverseRelationAssertion => [ + target = inverseAssertion + inverseA = connectedTo + inverseB = connectedTo + ] + metamodelLogic.output.annotations += inverseAnnotation + val initialModel = loadInitialModel() + val initialModelLogic = instanceModel2Logic.transform(metamodelLogic, initialModel) + val queries = loadQueries + val logic = viatra2Logic.transformQueries(queries, initialModelLogic, new Viatra2LogicConfiguration) + val config = new ViatraReasonerConfiguration => [ + runtimeLimit = 3600 + typeScopes => [ + minNewElements = modelSize + maxNewElements = modelSize + minNewElementsByType => [ + put(ecore2Logic.TypeofEClass(metamodelLogic.trace, Modes3Package.eINSTANCE.turnout), 1) + ] + maxNewElementsByType => [ + put(ecore2Logic.TypeofEClass(metamodelLogic.trace, Modes3Package.eINSTANCE.train), 5) + ] + ] + solutionScope.numberOfRequiredSolutions = 1 + nameNewElements = false + typeInferenceMethod = TypeInferenceMethod.PreliminaryAnalysis + stateCoderStrategy = StateCoderStrategy::Neighbourhood + scopePropagatorStrategy = new ScopePropagatorStrategy.Polyhedral( + PolyhedralScopePropagatorConstraints.Relational, PolyhedralScopePropagatorSolver.Clp) +// unitPropagationPatternGenerators += new Modes3UnitPropagationGenerator(ecore2Logic, metamodelLogic.trace) + debugConfiguration.partialInterpretatioVisualiser = null + ] + val workspace = new FileSystemWorkspace("output/", "") + val solution = solver.solve(logic.output, config, workspace) + if (solution instanceof ModelResult) { + println("Saving generated solutions") + val representations = solution.representation + for (representationIndex : 0 ..< representations.size) { + val representation = representations.get(representationIndex) + val representationNumber = representationIndex + 1 + if (representation instanceof PartialInterpretation) { + workspace.writeModel(representation, '''solution«representationNumber».partialinterpretation''') + val partialInterpretation2GML = new PartialInterpretation2Gml + val gml = partialInterpretation2GML.transform(representation) + workspace.writeText('''solution«representationNumber».gml''', gml) + if (representation.newElements.size < 160) { + if (representation instanceof PartialInterpretation) { + representation.problem.types.forEach[println(name)] + val rootType = (representation.problem.types.findFirst [ + name == "Modes3ModelRoot class DefinedPart" + ] as TypeDefinition) + val rootIntepretation = representation.partialtypeinterpratation.filter( + PartialComplexTypeInterpretation).findFirst [ + interpretationOf.name == "Modes3ModelRoot class" + ] + rootIntepretation.elements.removeAll(rootType.elements) + representation.problem.elements.removeAll(rootType.elements) + for (relationInterpretation : representation.partialrelationinterpretation) { + relationInterpretation.relationlinks.removeIf [ link | + if (link instanceof BinaryElementRelationLink) { + rootType.elements.contains(link.param1) || rootType.elements.contains(link.param2) + } else { + false + } + ] + } + rootType.elements.clear + } + val visualiser = new GraphvizVisualiser + val visualisation = visualiser.visualiseConcretization(representation) + visualisation.writeToFile(workspace, '''solution«representationNumber».png''') + } + } else { + workspace.writeText('''solution«representationNumber».txt''', representation.toString) + } + } + } else { + println("Failed to solver problem") + val partial = logic.output + workspace.writeModel(partial, "solution.partialinterpretation") + } + } + + static def createMetamodelDescriptor() { + val eClasses = ImmutableList.copyOf(Modes3Package.eINSTANCE.EClassifiers.filter(EClass)) + new EcoreMetamodelDescriptor( + eClasses, + emptySet, + false, + emptyList, + emptyList, + ImmutableList.copyOf(eClasses.flatMap[EReferences]), + emptyList + ) + } + + static def List loadInitialModel() { + #[Modes3Factory.eINSTANCE.createModes3ModelRoot] + } + + def loadQueries() { + val patternsBuilder = ImmutableList.builder + patternsBuilder.addAll(Modes3Queries.instance.specifications) + val patterns = patternsBuilder.build + val validationPatterns = ImmutableSet.copyOf(patterns.filter [ pattern | + pattern.allAnnotations.exists[name == "Constraint"] + ]) + new ViatraQuerySetDescriptor( + patterns, + validationPatterns, + emptyMap + ) + } + + def static init() { + EMFPatternLanguageStandaloneSetup.doSetup + ViatraQueryEngineOptions.setSystemDefaultBackends(ReteBackendFactory.INSTANCE, ReteBackendFactory.INSTANCE, + LocalSearchEMFBackendFactory.INSTANCE) + LogiclanguagePackage.eINSTANCE.class + LogicproblemPackage.eINSTANCE.class + PartialinterpretationPackage.eINSTANCE.class + Ecore2logicannotationsPackage.eINSTANCE.class + Viatra2LogicAnnotationsPackage.eINSTANCE.class + Resource.Factory.Registry.INSTANCE.extensionToFactoryMap.put("ecore", new XMIResourceFactoryImpl) + Resource.Factory.Registry.INSTANCE.extensionToFactoryMap.put("logicproblem", new XMIResourceFactoryImpl) + Resource.Factory.Registry.INSTANCE.extensionToFactoryMap.put("partialinterpretation", + new XMIResourceFactoryImpl) + } + + def static void main(String[] args) { + if (args.length != 2) { + System.err.println("Usage: ") + } + val monitoringQuery = MonitoringQuery.valueOf(args.get(0)) + val modelSize = Integer.parseInt(args.get(1)) + init() + val generator = new Modes3ModelGenerator(monitoringQuery, modelSize) + generator.generate() + } + + private static enum MonitoringQuery { + closeTrains, + misalignedTurnout + } +} -- cgit v1.2.3-70-g09d2