/*
* Copyright Text ->
*
* Copyright (c) 2012-2016 Marsha Chechik, Alessio Di Sandro, Michalis Famelis,
* Rick Salay.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Alessio Di Sandro - Implementation.
*/
package edu.toronto.cs.se.modelepedia.statemachine.diagram.navigator;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.edit.domain.AdapterFactoryEditingDomain;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.emf.workspace.util.WorkspaceSynchronizer;
import org.eclipse.gmf.runtime.emf.core.GMFEditingDomainFactory;
import org.eclipse.gmf.runtime.notation.Diagram;
import org.eclipse.gmf.runtime.notation.Edge;
import org.eclipse.gmf.runtime.notation.Node;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.ui.IMemento;
import org.eclipse.ui.navigator.ICommonContentExtensionSite;
import org.eclipse.ui.navigator.ICommonContentProvider;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.FinalStateEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.InitialStateEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.StateActionEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.StateEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.StateMachineEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.StateStateInternalActionsCompartmentEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.edit.parts.TransitionEditPart;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.part.Messages;
import edu.toronto.cs.se.modelepedia.statemachine.diagram.part.StateMachineVisualIDRegistry;
/**
* @generated
*/
public class StateMachineNavigatorContentProvider implements
ICommonContentProvider {
/**
* @generated
*/
private static final Object[] EMPTY_ARRAY = new Object[0];
/**
* @generated
*/
private Viewer myViewer;
/**
* @generated
*/
private AdapterFactoryEditingDomain myEditingDomain;
/**
* @generated
*/
private WorkspaceSynchronizer myWorkspaceSynchronizer;
/**
* @generated
*/
private Runnable myViewerRefreshRunnable;
/**
* @generated
*/
@SuppressWarnings({ "unchecked", "serial", "rawtypes" })
public StateMachineNavigatorContentProvider() {
TransactionalEditingDomain editingDomain = GMFEditingDomainFactory.INSTANCE
.createEditingDomain();
myEditingDomain = (AdapterFactoryEditingDomain) editingDomain;
myEditingDomain.setResourceToReadOnlyMap(new HashMap() {
public Object get(Object key) {
if (!containsKey(key)) {
put(key, Boolean.TRUE);
}
return super.get(key);
}
});
myViewerRefreshRunnable = new Runnable() {
public void run() {
if (myViewer != null) {
myViewer.refresh();
}
}
};
myWorkspaceSynchronizer = new WorkspaceSynchronizer(editingDomain,
new WorkspaceSynchronizer.Delegate() {
public void dispose() {
}
public boolean handleResourceChanged(final Resource resource) {
unloadAllResources();
asyncRefresh();
return true;
}
public boolean handleResourceDeleted(Resource resource) {
unloadAllResources();
asyncRefresh();
return true;
}
public boolean handleResourceMoved(Resource resource,
final URI newURI) {
unloadAllResources();
asyncRefresh();
return true;
}
});
}
/**
* @generated
*/
public void dispose() {
myWorkspaceSynchronizer.dispose();
myWorkspaceSynchronizer = null;
myViewerRefreshRunnable = null;
myViewer = null;
unloadAllResources();
((TransactionalEditingDomain) myEditingDomain).dispose();
myEditingDomain = null;
}
/**
* @generated
*/
public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
myViewer = viewer;
}
/**
* @generated
*/
void unloadAllResources() {
for (Resource nextResource : myEditingDomain.getResourceSet()
.getResources()) {
nextResource.unload();
}
}
/**
* @generated
*/
void asyncRefresh() {
if (myViewer != null && !myViewer.getControl().isDisposed()) {
myViewer.getControl().getDisplay()
.asyncExec(myViewerRefreshRunnable);
}
}
/**
* @generated
*/
public Object[] getElements(Object inputElement) {
return getChildren(inputElement);
}
/**
* @generated
*/
public void restoreState(IMemento aMemento) {
}
/**
* @generated
*/
public void saveState(IMemento aMemento) {
}
/**
* @generated
*/
public void init(ICommonContentExtensionSite aConfig) {
}
/**
* @generated
*/
public Object[] getChildren(Object parentElement) {
if (parentElement instanceof IFile) {
IFile file = (IFile) parentElement;
URI fileURI = URI.createPlatformResourceURI(file.getFullPath()
.toString(), true);
Resource resource = myEditingDomain.getResourceSet().getResource(
fileURI, true);
ArrayList<StateMachineNavigatorItem> result = new ArrayList<StateMachineNavigatorItem>();
ArrayList<View> topViews = new ArrayList<View>(resource
.getContents().size());
for (EObject o : resource.getContents()) {
if (o instanceof View) {
topViews.add((View) o);
}
}
result.addAll(createNavigatorItems(
selectViewsByType(topViews, StateMachineEditPart.MODEL_ID),
file, false));
return result.toArray();
}
if (parentElement instanceof StateMachineNavigatorGroup) {
StateMachineNavigatorGroup group = (StateMachineNavigatorGroup) parentElement;
return group.getChildren();
}
if (parentElement instanceof StateMachineNavigatorItem) {
StateMachineNavigatorItem navigatorItem = (StateMachineNavigatorItem) parentElement;
if (navigatorItem.isLeaf() || !isOwnView(navigatorItem.getView())) {
return EMPTY_ARRAY;
}
return getViewChildren(navigatorItem.getView(), parentElement);
}
/*
* Due to plugin.xml restrictions this code will be called only for views representing
* shortcuts to this diagram elements created on other diagrams.
*/
if (parentElement instanceof IAdaptable) {
View view = (View) ((IAdaptable) parentElement)
.getAdapter(View.class);
if (view != null) {
return getViewChildren(view, parentElement);
}
}
return EMPTY_ARRAY;
}
/**
* @generated
*/
private Object[] getViewChildren(View view, Object parentElement) {
switch (StateMachineVisualIDRegistry.getVisualID(view)) {
case StateMachineEditPart.VISUAL_ID: {
LinkedList<StateMachineAbstractNavigatorItem> result = new LinkedList<StateMachineAbstractNavigatorItem>();
result.addAll(getForeignShortcuts((Diagram) view, parentElement));
Diagram sv = (Diagram) view;
StateMachineNavigatorGroup links = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_StateMachine_1000_links,
"icons/linksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
Collection<View> connectedViews;
connectedViews = getChildrenByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(StateEditPart.VISUAL_ID));
result.addAll(createNavigatorItems(connectedViews, parentElement,
false));
connectedViews = getChildrenByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(InitialStateEditPart.VISUAL_ID));
result.addAll(createNavigatorItems(connectedViews, parentElement,
false));
connectedViews = getChildrenByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(FinalStateEditPart.VISUAL_ID));
result.addAll(createNavigatorItems(connectedViews, parentElement,
false));
connectedViews = getDiagramLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
links.addChildren(createNavigatorItems(connectedViews, links, false));
if (!links.isEmpty()) {
result.add(links);
}
return result.toArray();
}
case StateEditPart.VISUAL_ID: {
LinkedList<StateMachineAbstractNavigatorItem> result = new LinkedList<StateMachineAbstractNavigatorItem>();
Node sv = (Node) view;
StateMachineNavigatorGroup incominglinks = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_State_2001_incominglinks,
"icons/incomingLinksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
StateMachineNavigatorGroup outgoinglinks = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_State_2001_outgoinglinks,
"icons/outgoingLinksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
Collection<View> connectedViews;
connectedViews = getChildrenByType(
Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(StateStateInternalActionsCompartmentEditPart.VISUAL_ID));
connectedViews = getChildrenByType(connectedViews,
StateMachineVisualIDRegistry
.getType(StateActionEditPart.VISUAL_ID));
result.addAll(createNavigatorItems(connectedViews, parentElement,
false));
connectedViews = getIncomingLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
incominglinks.addChildren(createNavigatorItems(connectedViews,
incominglinks, true));
connectedViews = getOutgoingLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
outgoinglinks.addChildren(createNavigatorItems(connectedViews,
outgoinglinks, true));
if (!incominglinks.isEmpty()) {
result.add(incominglinks);
}
if (!outgoinglinks.isEmpty()) {
result.add(outgoinglinks);
}
return result.toArray();
}
case InitialStateEditPart.VISUAL_ID: {
LinkedList<StateMachineAbstractNavigatorItem> result = new LinkedList<StateMachineAbstractNavigatorItem>();
Node sv = (Node) view;
StateMachineNavigatorGroup incominglinks = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_InitialState_2002_incominglinks,
"icons/incomingLinksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
StateMachineNavigatorGroup outgoinglinks = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_InitialState_2002_outgoinglinks,
"icons/outgoingLinksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
Collection<View> connectedViews;
connectedViews = getIncomingLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
incominglinks.addChildren(createNavigatorItems(connectedViews,
incominglinks, true));
connectedViews = getOutgoingLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
outgoinglinks.addChildren(createNavigatorItems(connectedViews,
outgoinglinks, true));
if (!incominglinks.isEmpty()) {
result.add(incominglinks);
}
if (!outgoinglinks.isEmpty()) {
result.add(outgoinglinks);
}
return result.toArray();
}
case FinalStateEditPart.VISUAL_ID: {
LinkedList<StateMachineAbstractNavigatorItem> result = new LinkedList<StateMachineAbstractNavigatorItem>();
Node sv = (Node) view;
StateMachineNavigatorGroup incominglinks = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_FinalState_2003_incominglinks,
"icons/incomingLinksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
StateMachineNavigatorGroup outgoinglinks = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_FinalState_2003_outgoinglinks,
"icons/outgoingLinksNavigatorGroup.gif", parentElement); //$NON-NLS-1$
Collection<View> connectedViews;
connectedViews = getIncomingLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
incominglinks.addChildren(createNavigatorItems(connectedViews,
incominglinks, true));
connectedViews = getOutgoingLinksByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(TransitionEditPart.VISUAL_ID));
outgoinglinks.addChildren(createNavigatorItems(connectedViews,
outgoinglinks, true));
if (!incominglinks.isEmpty()) {
result.add(incominglinks);
}
if (!outgoinglinks.isEmpty()) {
result.add(outgoinglinks);
}
return result.toArray();
}
case TransitionEditPart.VISUAL_ID: {
LinkedList<StateMachineAbstractNavigatorItem> result = new LinkedList<StateMachineAbstractNavigatorItem>();
Edge sv = (Edge) view;
StateMachineNavigatorGroup target = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_Transition_4001_target,
"icons/linkTargetNavigatorGroup.gif", parentElement); //$NON-NLS-1$
StateMachineNavigatorGroup source = new StateMachineNavigatorGroup(
Messages.NavigatorGroupName_Transition_4001_source,
"icons/linkSourceNavigatorGroup.gif", parentElement); //$NON-NLS-1$
Collection<View> connectedViews;
connectedViews = getLinksTargetByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(StateEditPart.VISUAL_ID));
target.addChildren(createNavigatorItems(connectedViews, target,
true));
connectedViews = getLinksTargetByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(InitialStateEditPart.VISUAL_ID));
target.addChildren(createNavigatorItems(connectedViews, target,
true));
connectedViews = getLinksTargetByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(FinalStateEditPart.VISUAL_ID));
target.addChildren(createNavigatorItems(connectedViews, target,
true));
connectedViews = getLinksSourceByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(StateEditPart.VISUAL_ID));
source.addChildren(createNavigatorItems(connectedViews, source,
true));
connectedViews = getLinksSourceByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(InitialStateEditPart.VISUAL_ID));
source.addChildren(createNavigatorItems(connectedViews, source,
true));
connectedViews = getLinksSourceByType(Collections.singleton(sv),
StateMachineVisualIDRegistry
.getType(FinalStateEditPart.VISUAL_ID));
source.addChildren(createNavigatorItems(connectedViews, source,
true));
if (!target.isEmpty()) {
result.add(target);
}
if (!source.isEmpty()) {
result.add(source);
}
return result.toArray();
}
}
return EMPTY_ARRAY;
}
/**
* @generated
*/
private Collection<View> getLinksSourceByType(Collection<Edge> edges,
String type) {
LinkedList<View> result = new LinkedList<View>();
for (Edge nextEdge : edges) {
View nextEdgeSource = nextEdge.getSource();
if (type.equals(nextEdgeSource.getType())
&& isOwnView(nextEdgeSource)) {
result.add(nextEdgeSource);
}
}
return result;
}
/**
* @generated
*/
private Collection<View> getLinksTargetByType(Collection<Edge> edges,
String type) {
LinkedList<View> result = new LinkedList<View>();
for (Edge nextEdge : edges) {
View nextEdgeTarget = nextEdge.getTarget();
if (type.equals(nextEdgeTarget.getType())
&& isOwnView(nextEdgeTarget)) {
result.add(nextEdgeTarget);
}
}
return result;
}
/**
* @generated
*/
private Collection<View> getOutgoingLinksByType(
Collection<? extends View> nodes, String type) {
LinkedList<View> result = new LinkedList<View>();
for (View nextNode : nodes) {
result.addAll(selectViewsByType(nextNode.getSourceEdges(), type));
}
return result;
}
/**
* @generated
*/
private Collection<View> getIncomingLinksByType(
Collection<? extends View> nodes, String type) {
LinkedList<View> result = new LinkedList<View>();
for (View nextNode : nodes) {
result.addAll(selectViewsByType(nextNode.getTargetEdges(), type));
}
return result;
}
/**
* @generated
*/
private Collection<View> getChildrenByType(
Collection<? extends View> nodes, String type) {
LinkedList<View> result = new LinkedList<View>();
for (View nextNode : nodes) {
result.addAll(selectViewsByType(nextNode.getChildren(), type));
}
return result;
}
/**
* @generated
*/
private Collection<View> getDiagramLinksByType(
Collection<Diagram> diagrams, String type) {
ArrayList<View> result = new ArrayList<View>();
for (Diagram nextDiagram : diagrams) {
result.addAll(selectViewsByType(nextDiagram.getEdges(), type));
}
return result;
}
// TODO refactor as static method
/**
* @generated
*/
private Collection<View> selectViewsByType(Collection<View> views,
String type) {
ArrayList<View> result = new ArrayList<View>();
for (View nextView : views) {
if (type.equals(nextView.getType()) && isOwnView(nextView)) {
result.add(nextView);
}
}
return result;
}
/**
* @generated
*/
private boolean isOwnView(View view) {
return StateMachineEditPart.MODEL_ID
.equals(StateMachineVisualIDRegistry.getModelID(view));
}
/**
* @generated
*/
private Collection<StateMachineNavigatorItem> createNavigatorItems(
Collection<View> views, Object parent, boolean isLeafs) {
ArrayList<StateMachineNavigatorItem> result = new ArrayList<StateMachineNavigatorItem>(
views.size());
for (View nextView : views) {
result.add(new StateMachineNavigatorItem(nextView, parent, isLeafs));
}
return result;
}
/**
* @generated
*/
private Collection<StateMachineNavigatorItem> getForeignShortcuts(
Diagram diagram, Object parent) {
LinkedList<View> result = new LinkedList<View>();
for (Iterator<View> it = diagram.getChildren().iterator(); it.hasNext();) {
View nextView = it.next();
if (!isOwnView(nextView)
&& nextView.getEAnnotation("Shortcut") != null) { //$NON-NLS-1$
result.add(nextView);
}
}
return createNavigatorItems(result, parent, false);
}
/**
* @generated
*/
public Object getParent(Object element) {
if (element instanceof StateMachineAbstractNavigatorItem) {
StateMachineAbstractNavigatorItem abstractNavigatorItem = (StateMachineAbstractNavigatorItem) element;
return abstractNavigatorItem.getParent();
}
return null;
}
/**
* @generated
*/
public boolean hasChildren(Object element) {
return element instanceof IFile || getChildren(element).length > 0;
}
}