/*******************************************************************************
* Copyright (c) 2007 Spring IDE Developers
* 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:
* Spring IDE Developers - initial API and implementation
*******************************************************************************/
package org.springframework.ide.eclipse.webflow.ui.graph.parts;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import java.util.ArrayList;
import java.util.List;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gef.editparts.AbstractTreeEditPart;
import org.eclipse.jface.viewers.DecoratingLabelProvider;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.swt.graphics.Image;
import org.springframework.ide.eclipse.webflow.core.model.IActionState;
import org.springframework.ide.eclipse.webflow.core.model.IAttributeEnabled;
import org.springframework.ide.eclipse.webflow.core.model.IDecisionState;
import org.springframework.ide.eclipse.webflow.core.model.IInlineFlowState;
import org.springframework.ide.eclipse.webflow.core.model.IState;
import org.springframework.ide.eclipse.webflow.core.model.ISubflowState;
import org.springframework.ide.eclipse.webflow.core.model.IViewState;
import org.springframework.ide.eclipse.webflow.core.model.IWebflowModelElement;
import org.springframework.ide.eclipse.webflow.core.model.IWebflowState;
import org.springframework.ide.eclipse.webflow.ui.graph.model.WebflowModelLabelDecorator;
import org.springframework.ide.eclipse.webflow.ui.graph.model.WebflowModelLabelProvider;
import org.springframework.ide.eclipse.webflow.ui.graph.policies.StateEditPolicy;
import org.springframework.ide.eclipse.webflow.ui.graph.policies.StateTreeContainerEditPolicy;
import org.springframework.ide.eclipse.webflow.ui.graph.policies.StateTreeEditPolicy;
/**
* @author Christian Dupuis
*/
public class StateTreeEditPart extends AbstractTreeEditPart implements
PropertyChangeListener {
/**
*
*/
protected static ILabelProvider labelProvider = new DecoratingLabelProvider(
new WebflowModelLabelProvider(), new WebflowModelLabelDecorator());
/**
*
*/
protected static WebflowModelLabelProvider tLabelProvider = new WebflowModelLabelProvider();
/*
* (non-Javadoc)
* @see org.eclipse.gef.editparts.AbstractEditPart#activate()
*/
public void activate() {
super.activate();
if (getModel() != null) {
((IWebflowModelElement) getModel()).addPropertyChangeListener(this);
}
}
/*
* (non-Javadoc)
* @see org.eclipse.gef.editparts.AbstractTreeEditPart#createEditPolicies()
*/
protected void createEditPolicies() {
installEditPolicy(EditPolicy.COMPONENT_ROLE, new StateEditPolicy());
installEditPolicy(EditPolicy.TREE_CONTAINER_ROLE,
new StateTreeContainerEditPolicy());
installEditPolicy(EditPolicy.PRIMARY_DRAG_ROLE,
new StateTreeEditPolicy());
}
/*
* (non-Javadoc)
* @see org.eclipse.gef.editparts.AbstractEditPart#deactivate()
*/
public void deactivate() {
if (getModel() != null) {
((IWebflowModelElement) getModel()).removePropertyChangeListener(this);
}
super.deactivate();
}
/*
* (non-Javadoc)
* @see org.eclipse.gef.editparts.AbstractEditPart#getModelChildren()
*/
protected List getModelChildren() {
List<IWebflowModelElement> children = new ArrayList<IWebflowModelElement>();
if (getModel() instanceof IAttributeEnabled) {
IAttributeEnabled properties = (IAttributeEnabled) getModel();
if (properties.getAttributes() != null) {
children.addAll(properties.getAttributes());
}
}
if (getModel() instanceof IState
&& !(getModel() instanceof IWebflowState)) {
IState state = (IState) getModel();
if (state.getEntryActions() != null) {
children.addAll((state.getEntryActions().getEntryActions()));
}
}
if (getModel() instanceof IActionState) {
if (((IActionState) getState()).getActions() != null) {
children.addAll(((IActionState) getState()).getActions());
}
}
if (getModel() instanceof IViewState) {
if (((IViewState) getState()).getRenderActions() != null) {
children.addAll(((IViewState) getState()).getRenderActions()
.getRenderActions());
}
}
else if (getModel() instanceof ISubflowState) {
if (((ISubflowState) getModel()).getAttributeMapper() != null)
children.add(((ISubflowState) getModel()).getAttributeMapper());
}
else if (getModel() instanceof IInlineFlowState) {
children.addAll(((IInlineFlowState) getModel()).getWebFlowState()
.getStates());
children.addAll(((IInlineFlowState) getModel()).getWebFlowState()
.getInlineFlowStates());
}
else if (getModel() instanceof IWebflowState) {
if (((IWebflowState) getState()).getStates() != null
&& ((IWebflowState) getState()).getInlineFlowStates() != null) {
children.addAll(((IWebflowState) getState()).getStates());
children.addAll(((IWebflowState) getState())
.getInlineFlowStates());
}
}
else if (getModel() instanceof IDecisionState) {
if (((IDecisionState) getModel()).getIfs() != null) {
children.addAll(((IDecisionState) getModel()).getIfs());
}
}
if (getModel() instanceof IState
&& !(getModel() instanceof IWebflowState)) {
IState state = (IState) getModel();
if (state.getExitActions() != null) {
children.addAll((state.getExitActions().getExitActions()));
}
if (state.getExceptionHandlers() != null) {
children.addAll((state.getExceptionHandlers()));
}
}
return children;
}
/**
* @return
*/
protected IState getState() {
return (IState) getModel();
}
/*
* (non-Javadoc)
* @see java.beans.PropertyChangeListener#propertyChange(java.beans.PropertyChangeEvent)
*/
public void propertyChange(PropertyChangeEvent change) {
if (change.getPropertyName().equals(IWebflowModelElement.ADD_CHILDREN)) {
addChild(createChild(change.getNewValue()), ((Integer) change
.getOldValue()).intValue());
}
else if (change.getPropertyName().equals(
IWebflowModelElement.REMOVE_CHILDREN)) {
// remove child
removeChild((EditPart) getViewer().getEditPartRegistry().get(
change.getNewValue()));
}
else if (change.getPropertyName().equals(
IWebflowModelElement.MOVE_CHILDREN)) {
refreshChildren();
}
else {
refreshVisuals();
}
}
/*
* (non-Javadoc)
* @see org.eclipse.gef.editparts.AbstractTreeEditPart#refreshVisuals()
*/
protected void refreshVisuals() {
Image image = labelProvider.getImage(getModel());
String text = tLabelProvider.getText(getModel(), true, false, false);
if (image != null) {
setWidgetImage(image);
}
setWidgetText(text);
}
}