/*******************************************************************************
* Copyright (c) 2001, 2009 IBM Corporation and others.
* 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:
* IBM Corporation - initial API and implementation
*******************************************************************************/
package org.eclipse.wst.xsd.ui.internal.adt.design.editparts;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.Label;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gef.Request;
import org.eclipse.gef.RequestConstants;
import org.eclipse.gef.requests.LocationRequest;
import org.eclipse.gef.ui.actions.ActionRegistry;
import org.eclipse.jface.action.IAction;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDComplexTypeDefinitionAdapter;
import org.eclipse.wst.xsd.ui.internal.adt.actions.SetInputToGraphView;
import org.eclipse.wst.xsd.ui.internal.adt.design.editparts.model.Compartment;
import org.eclipse.wst.xsd.ui.internal.adt.design.editpolicies.ADTDirectEditPolicy;
import org.eclipse.wst.xsd.ui.internal.adt.design.editpolicies.ADTSelectionFeedbackEditPolicy;
import org.eclipse.wst.xsd.ui.internal.adt.design.editpolicies.KeyBoardAccessibilityEditPolicy;
import org.eclipse.wst.xsd.ui.internal.adt.design.figures.IStructureFigure;
import org.eclipse.wst.xsd.ui.internal.adt.facade.IStructure;
import org.eclipse.wst.xsd.ui.internal.common.actions.OpenInNewEditor;
import org.eclipse.wst.xsd.ui.internal.design.editparts.SpaceFillerForFieldEditPart;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDSchema;
public class StructureEditPart extends BaseTypeConnectingEditPart implements INamedEditPart
{
protected List compartmentList = null;
protected ADTDirectEditPolicy adtDirectEditPolicy = new ADTDirectEditPolicy();
protected TypeReferenceConnection connectionFigure;
protected TypeReferenceConnection connectionFeedbackFigure;
/**
* TODO cs... I'm sure this has something to do with the way we wanted to rework compartment creation
* I suppose we could have subclasses override this method instead of getModelChildren()
*
* @deprecated
*/
protected Compartment[] getCompartments()
{
return null;
}
protected IFigure createFigure()
{
IStructureFigure figure = getFigureFactory().createStructureFigure(getModel());
return figure;
}
public IStructureFigure getStructureFigure()
{
return (IStructureFigure)getFigure();
}
public IFigure getContentPane()
{
return getStructureFigure().getContentPane();
}
public EditPart doGetRelativeEditPart(EditPart editPart, int direction)
{
EditPart result = null;
if (direction == KeyBoardAccessibilityEditPolicy.IN_TO_FIRST_CHILD)
{
for (Iterator i = getChildren().iterator(); i.hasNext();)
{
CompartmentEditPart compartment = (CompartmentEditPart)i.next();
if (compartment.hasContent())
{
for (Iterator contentChildren = compartment.getChildren().iterator(); contentChildren.hasNext(); )
{
Object child = contentChildren.next();
if (child instanceof SectionEditPart)
{
SectionEditPart part = (SectionEditPart) child;
result = (EditPart)part.getChildren().get(0);
return result;
}
else if (child instanceof BaseFieldEditPart && (!(child instanceof SpaceFillerForFieldEditPart)))
{
result = (EditPart)child;
return result;
}
}
}
}
}
else
{
return super.doGetRelativeEditPart(editPart, direction);
}
return result;
}
protected void createEditPolicies()
{
super.createEditPolicies();
installEditPolicy(EditPolicy.SELECTION_FEEDBACK_ROLE, new ADTSelectionFeedbackEditPolicy());
installEditPolicy(EditPolicy.DIRECT_EDIT_ROLE, adtDirectEditPolicy);
}
protected IStructure getStructure()
{
return (IStructure)getModel();
}
protected List getModelChildren()
{
if (compartmentList == null)
{
compartmentList = new ArrayList();
// TODO.. this needs to be moved to the xsd specific version of this class
compartmentList.add(new Compartment(getStructure(), "attribute")); //$NON-NLS-1$
compartmentList.add(new Compartment(getStructure(), "element")); //$NON-NLS-1$
}
return compartmentList;
}
protected void refreshChildren()
{
super.refreshChildren();
//getFigure().invalidateTree();
}
protected void refreshVisuals()
{
super.refreshVisuals();
getStructureFigure().refreshVisuals(getModel());
}
public void addFeedback()
{
getStructureFigure().addSelectionFeedback();
super.addFeedback();
}
public void removeFeedback()
{
getStructureFigure().removeSelectionFeedback();
super.removeFeedback();
}
public Label getNameLabelFigure()
{
return getStructureFigure().getNameLabel();
}
public void performDirectEdit(Point cursorLocation)
{
}
public void performRequest(Request request)
{
if (request.getType() == RequestConstants.REQ_DIRECT_EDIT)
{
if (request instanceof LocationRequest)
{
// uncomment for direct edit of name (add else)
// LocationRequest locationRequest = (LocationRequest)request;
// Point p = locationRequest.getLocation();
// if (hitTest(getNameLabelFigure(), p))
// {
// LabelEditManager manager = new LabelEditManager(this, new LabelCellEditorLocator(this, p));
// NameUpdateCommandWrapper wrapper = new NameUpdateCommandWrapper();
// adtDirectEditPolicy.setUpdateCommand(wrapper);
// manager.show();
// }
}
}
else if (request.getType() == RequestConstants.REQ_OPEN)
{
Object model = getModel();
if (request instanceof LocationRequest)
{
LocationRequest locationRequest = (LocationRequest)request;
Point p = locationRequest.getLocation();
if (getStructureFigure().hitTestHeader(p))
{
// TODO: !!! This should be moved to the adt-xsd package
//
if (model instanceof XSDComplexTypeDefinitionAdapter)
{
XSDComplexTypeDefinitionAdapter adapter = (XSDComplexTypeDefinitionAdapter)model;
XSDComplexTypeDefinition ct = (XSDComplexTypeDefinition)adapter.getTarget();
IWorkbench workbench = PlatformUI.getWorkbench();
IWorkbenchWindow workbenchWindow = workbench.getActiveWorkbenchWindow();
IEditorPart editorPart = workbenchWindow.getActivePage().getActiveEditor();
Object schema = editorPart.getAdapter(XSDSchema.class);
ActionRegistry registry = getEditorActionRegistry(editorPart);
if (registry != null)
{
if (schema == ct.getSchema())
{
IAction action = registry.getAction(SetInputToGraphView.ID);
action.run();
}
else
{
IAction action = registry.getAction(OpenInNewEditor.ID);
action.run();
}
}
}
}
}
}
}
protected ActionRegistry getEditorActionRegistry(IEditorPart editor)
{
return (ActionRegistry) editor.getAdapter(ActionRegistry.class);
}
protected boolean shouldDrawConnection()
{
return false;
}
public TypeReferenceConnection createConnectionFigure()
{
return null;
}
public String getReaderText()
{
// return getStructureFigure().getNameLabel().getText();
return getNameLabelFigure().getText();
}
}