居于gmf的图元自动对齐实现

背景：

加载模型文件之后，可以通过 以下方法获取了 资源文件

/**

* 针对单表维护

* @generated NOT

*/

public static Resource createDiagram4singletable(URI diagramURI, URI modelURI,

IProgressMonitor progressMonitor) {

TransactionalEditingDomain editingDomain = GMFEditingDomainFactory.INSTANCE

.createEditingDomain();

progressMonitor.beginTask(

Messages.PraDiagramEditorUtil_CreateDiagramProgressTask, 3);

final Resource diagramResource = editingDomain.getResourceSet()

.createResource(diagramURI);

//

/*final Resource modelResource = editingDomain.getResourceSet()

.createResource(modelURI);*/

//获取资源

final Resource modelResource = editingDomain.getResourceSet()

.getResource(modelURI, true);

final String diagramName = diagramURI.lastSegment();

AbstractTransactionalCommand command = new AbstractTransactionalCommand(

editingDomain,

Messages.PraDiagramEditorUtil_CreateDiagramCommandLabel,

Collections.EMPTY_LIST) {

protected CommandResult doExecuteWithResult(

IProgressMonitor monitor, IAdaptable info)

throws ExecutionException {

//---------------------获取资源-------------------------

CjsAutomation model = null;

try{

model = (com.cellcom.studio.pra.impl.CjsAutomationImpl)modelResource.getContents().get(0);

}catch(Exception e){

e.printStackTrace();

}

attachModelToResource(model, modelResource);

//------------------------------------------------------

Diagram diagram = ViewService.createDiagram(model,

CjsAutomationEditPart.MODEL_ID,

PraDiagramEditorPlugin.DIAGRAM_PREFERENCES_HINT);

if (diagram != null) {

diagramResource.getContents().add(diagram);

diagram.setName(diagramName);

diagram.setElement(model);

}

try {

modelResource

.save(com.cellcom.studio.pra.diagram.part.PraDiagramEditorUtil

.getSaveOptions());

diagramResource

.save(com.cellcom.studio.pra.diagram.part.PraDiagramEditorUtil

.getSaveOptions());

} catch (IOException e) {

PraDiagramEditorPlugin.getInstance().logError(

"Unable to store model and diagram resources", e); //$NON-NLS-1$

}

return CommandResult.newOKCommandResult();

}

};

try {

OperationHistoryFactory.getOperationHistory().execute(command,

new SubProgressMonitor(progressMonitor, 1), null);

} catch (ExecutionException e) {

PraDiagramEditorPlugin.getInstance().logError(

"Unable to create model and diagram", e); //$NON-NLS-1$

}

setCharset(WorkspaceSynchronizer.getFile(modelResource));

setCharset(WorkspaceSynchronizer.getFile(diagramResource));

return diagramResource;

}

将该文件打开，因为只有在打开的状态下，才可以获取父类的editpart （其实是CjsAutomationEditPart）

获取要点是：((DiagramEditPart)rootEditPart[0].getContents());获取父editpart – CjsAutomationEditPart

if (diagram != null) {

try {

BzaDiagramEditorUtil.openDiagram(diagram);

} catch (PartInitException e) {

ErrorDialog.openError(getContainer().getShell(),

Messages.BzaCreationWizardOpenEditorError,

null, e.getStatus());

}

}

//Diagram d = (Diagram)diagram.getContents().get(0);

//____________

final RootEditPart[] rootEditPart = new RootEditPart[1];

PlatformUI.getWorkbench().getDisplay().syncExec(new Runnable() {

public void run() {

IWorkbenchPage page = PlatformUI.getWorkbench()

.getActiveWorkbenchWindow().getActivePage();

if (page != null) {

IEditorPart editor = page.getActiveEditor();

if (editor instanceof GraphicalEditor) {

GraphicalEditor graphicalEditor = (GraphicalEditor) editor;

rootEditPart[0] = (RootEditPart)graphicalEditor.getAdapter(EditPart.class);

}

}

}

}); //这段方法是获取当前editpart --很重要

//(CjsAutomationEditPart)de.ge

ArrangeRequest ar = new ArrangeRequest(ActionIds.ACTION_ARRANGE_ALL);

//ar.setPartsToArrange(((DiagramEditPart)rootEditPart[0].getContents()).getChildren());

//((DiagramEditPart)rootEditPart[0].getContents()).getChildren();//获取当前流程图里面有哪些editpart

//((DiagramEditPart)rootEditPart[0].getContents());获取父editpart - CjsAutomationEditPart

//图元对齐

EditPart4ArrangeCommand.ShapeContainerEditPolicy es = new EditPart4ArrangeCommand.ShapeContainerEditPolicy(((DiagramEditPart)rootEditPart[0].getContents()));

try{

es.getArrangeCommand(ar).execute();

}catch(Exception e){

e.printStackTrace();

}

closeAndSaveDiagram(diagram);

发送命令后，因为文件被修改了，所以还得通过IDE来保存这些修改

/**

* 关闭并且保存

* @param diagram

* @return

* @throws PartInitException

*/

public static boolean closeAndSaveDiagram(Resource diagram)

throws PartInitException {

String path = diagram.getURI().toPlatformString(true);

IResource workspaceResource = ResourcesPlugin.getWorkspace().getRoot()

.findMember(new Path(path));

if (workspaceResource instanceof IFile) {

IWorkbenchPage page = PlatformUI.getWorkbench()

.getActiveWorkbenchWindow().getActivePage();

//保存

//page.savePerspective();

page.saveAllEditors(false);

//page.saveEditor(new BzaDiagramEditor(), true);

//关闭

//page.close();

/*return null != page.openEditor(new FileEditorInput(

(IFile) workspaceResource), BzaDiagramEditor.ID);*/

}

return false;

}

以下是发送对齐命令的实现

import org.eclipse.gef.EditPart;

import org.eclipse.gef.commands.Command;

import org.eclipse.gmf.runtime.diagram.ui.editparts.ShapeEditPart;

import org.eclipse.gmf.runtime.diagram.ui.editpolicies.ContainerEditPolicy;

import org.eclipse.gmf.runtime.diagram.ui.requests.ArrangeRequest;

import org.eclipse.gmf.runtime.notation.View;

/**

* 为单表维护提供图元对齐的命令

* @author xgchen

* TODO

* 2009-7-22 下午03:12:07

*/

public class EditPart4ArrangeCommand extends ShapeEditPart {

public EditPart4ArrangeCommand(View view) {

super(view);

// TODO Auto-generated constructor stub

}

public static class ShapeContainerEditPolicy

extends ContainerEditPolicy {

public ShapeContainerEditPolicy(EditPart host){

super.setHost(host);

}

public Command getArrangeCommand(ArrangeRequest request) {

return super.getArrangeCommand(request);

}

}

}