1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
\r
3 * in Industry THTH ry.
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.modeling.ui.actions;
\r
14 import java.util.ArrayList;
\r
15 import java.util.Collections;
\r
16 import java.util.List;
\r
17 import java.util.concurrent.atomic.AtomicReference;
\r
19 import org.eclipse.jface.action.Action;
\r
20 import org.eclipse.jface.action.ContributionItem;
\r
21 import org.eclipse.jface.action.IContributionItem;
\r
22 import org.eclipse.jface.resource.ImageDescriptor;
\r
23 import org.eclipse.jface.resource.JFaceResources;
\r
24 import org.eclipse.jface.resource.LocalResourceManager;
\r
25 import org.eclipse.jface.resource.ResourceManager;
\r
26 import org.eclipse.swt.SWT;
\r
27 import org.eclipse.swt.events.SelectionEvent;
\r
28 import org.eclipse.swt.events.SelectionListener;
\r
29 import org.eclipse.swt.widgets.Menu;
\r
30 import org.eclipse.swt.widgets.MenuItem;
\r
31 import org.simantics.Simantics;
\r
32 import org.simantics.db.ReadGraph;
\r
33 import org.simantics.db.Resource;
\r
34 import org.simantics.db.WriteGraph;
\r
35 import org.simantics.db.common.NamedResource;
\r
36 import org.simantics.db.common.request.ObjectsWithType;
\r
37 import org.simantics.db.common.request.WriteRequest;
\r
38 import org.simantics.db.exception.DatabaseException;
\r
39 import org.simantics.layer0.Layer0;
\r
40 import org.simantics.project.IProject;
\r
41 import org.simantics.simulation.experiment.IExperiment;
\r
42 import org.simantics.simulation.ontology.SimulationResource;
\r
43 import org.simantics.simulation.project.IExperimentManager;
\r
44 import org.simantics.ui.contribution.DynamicMenuContribution;
\r
45 import org.simantics.ui.utils.ResourceAdaptionUtils;
\r
48 * @author Tuukka Lehtonen
\r
50 public class SetInitialState extends DynamicMenuContribution {
\r
52 private ResourceManager resourceManager;
\r
54 public SetInitialState() {
\r
55 resourceManager = new LocalResourceManager(JFaceResources.getResources());
\r
59 public void dispose() {
\r
60 if (resourceManager != null) {
\r
61 resourceManager.dispose();
\r
62 resourceManager = null;
\r
68 protected boolean preAcceptSelection(Object[] selection) {
\r
69 return selection != null && selection.length == 1;
\r
73 protected IContributionItem[] getContributionItems(ReadGraph graph, Object[] selection) throws DatabaseException {
\r
74 if (selection.length != 1)
\r
77 final Resource experiment = ResourceAdaptionUtils.adaptToResource(selection[0]);
\r
78 if (experiment == null)
\r
81 Layer0 L0 = Layer0.getInstance(graph);
\r
82 SimulationResource SIMU = SimulationResource.getInstance(graph);
\r
83 if (!graph.isInstanceOf(experiment, SIMU.Experiment))
\r
86 // Check that the experiment is not currently active.
\r
87 IProject project = Simantics.peekProject();
\r
88 if (project != null) {
\r
89 IExperimentManager expMan = project.getHint(IExperimentManager.KEY_EXPERIMENT_MANAGER);
\r
90 if (expMan != null) {
\r
91 for (IExperiment exp : new IExperiment[] { expMan.getActiveExperiment() }) {
\r
92 if (exp != null && experiment.equals(exp.getResource()))
\r
98 final List<NamedResource> states = new ArrayList<NamedResource>();
\r
99 Resource model = graph.getSingleObject(experiment, L0.PartOf);
\r
100 for (Resource obj : graph.syncRequest(new ObjectsWithType(model, L0.ConsistsOf, SIMU.State))) {
\r
101 String name = graph.adapt(obj, String.class);
\r
102 states.add(new NamedResource(name, obj));
\r
105 final AtomicReference<NamedResource> currentInitialState = new AtomicReference<NamedResource>();
\r
106 Resource currentState = graph.getPossibleObject(experiment, SIMU.HasInitialState);
\r
107 if (currentState != null) {
\r
108 for (NamedResource state : states) {
\r
109 if (state.getResource().equals(currentState))
\r
110 currentInitialState.set(state);
\r
114 // Sort the open with actions in ascending lexicographical order.
\r
115 Collections.sort(states);
\r
116 if (states.isEmpty())
\r
119 return new IContributionItem[] {
\r
120 new ContributionItem() {
\r
122 public void fill(Menu menu, int index) {
\r
123 MenuItem setInitialState = new MenuItem(menu, SWT.CASCADE, index);
\r
124 setInitialState.setText("Set Initial State");
\r
125 Menu subMenu = new Menu(menu);
\r
126 setInitialState.setMenu(subMenu);
\r
128 for (NamedResource state : states) {
\r
129 addMenuItem(subMenu, new Adapter(experiment, state), state.equals(currentInitialState.get()));
\r
136 private void addMenuItem(Menu subMenu, Adapter adapter, boolean selected) {
\r
137 MenuItem item = new MenuItem(subMenu, SWT.CHECK);
\r
138 item.setText(adapter.getText());
\r
139 item.setSelection(selected);
\r
140 ImageDescriptor imgDesc = adapter.getImageDescriptor();
\r
141 if (imgDesc != null)
\r
142 item.setImage(resourceManager.createImage(imgDesc));
\r
143 item.addSelectionListener(adapter);
\r
146 static class Adapter extends Action implements SelectionListener {
\r
147 private final Resource experiment;
\r
148 private final NamedResource state;
\r
150 public Adapter(Resource experiment, NamedResource state) {
\r
151 super(state.getName());
\r
152 this.experiment = experiment;
\r
153 this.state = state;
\r
157 public void widgetDefaultSelected(SelectionEvent e) {
\r
162 public void widgetSelected(SelectionEvent e) {
\r
167 public void run() {
\r
168 Simantics.getSession().asyncRequest(new WriteRequest() {
\r
170 public void perform(WriteGraph wg) throws DatabaseException {
\r
171 SimulationResource SIMU = SimulationResource.getInstance(wg);
\r
172 wg.deny(experiment, SIMU.HasInitialState);
\r
173 wg.claim(experiment, SIMU.HasInitialState, state.getResource());
\r