Solvers/VIATRA-Solver/org.eclipse.viatra.dse/src/org/eclipse/viatra/dse/solutionstore/ModelSaverSolutionFoundHandler.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

/*******************************************************************************
 * Copyright (c) 2010-2017, Andras Szabolcs Nagy and Daniel Varro
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-v20.html.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *******************************************************************************/
package org.eclipse.viatra.dse.solutionstore;

import java.util.HashSet;

import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.viatra.dse.api.SolutionTrajectory;
import org.eclipse.viatra.dse.base.ThreadContext;
import org.eclipse.viatra.dse.util.EMFHelper;

public class ModelSaverSolutionFoundHandler implements ISolutionFoundHandler {

    private HashSet<Object> savedSolutions = new HashSet<Object>();
    private ISolutionNameProvider solutionNameProvider;

    public ModelSaverSolutionFoundHandler() {
        solutionNameProvider = new IdBasedSolutionNameProvider("solution", "xmi");
    }

    public ModelSaverSolutionFoundHandler(String extension) {
        solutionNameProvider = new IdBasedSolutionNameProvider("solution", extension);
    }

    public ModelSaverSolutionFoundHandler(String prefix, String extension) {
        solutionNameProvider = new IdBasedSolutionNameProvider(prefix, extension);
    }

    public ModelSaverSolutionFoundHandler(ISolutionNameProvider solutionNameProvider) {
        this.solutionNameProvider = solutionNameProvider;
    }

    @Override
    public void solutionTriedToSave(ThreadContext context, SolutionTrajectory trajectory) {
    }

    @Override
    public void solutionFound(ThreadContext context, SolutionTrajectory trajectory) {
        Object stateCode = trajectory.getSolution().getStateCode();

        if (savedSolutions.contains(stateCode)) {
            return;
        }

        savedSolutions.add(stateCode);
        Notifier clonedModel = EMFHelper.clone(context.getModel());
        EMFHelper.saveModel(clonedModel, solutionNameProvider.getName());
    }
}