GMF 经验点滴积累（四）-修改GMF的outline tree，使其按类别展示View

       GMF的outline tree默认显示的层次为：diagram -- notes，如果要展示edges或者按其他层次分类展示views，就需要对其进行定制。

     1、重载DiagramEditor方法getOutlineViewEditPartFactory()，这个方法是创建outline tree节点TreeEditPart的入口函数。

 

      protected EditPartFactory getOutlineViewEditPartFactory() { return new EditPartFactory() { public EditPart createEditPart(EditPart context, Object model) { //顶层 if (model instanceof Diagram )) { return new RootDiagramTreeEditPart((Diagram) model); //需要展示下层元素的 } else if (view instanceof Node && ((Node) view).getElement() instanceof ContainerEle) { return new ContainerTreeEditPart((Node) view); //分类层 } else if (view instanceof EListContainer) { return new CateTreeEditPart((EListContainer) view); //叶子层 } else return new LeafTreeEditPart((View) view); } }; }

 

     2、创建tree相关的TreeEditPart，其中顶层RootDiagramTreeEditPar继承自TreeDiagramEditPart，分类层次节点CateTreeEditpart继承自AbstractTreeEditPart，叶子及其他节点Editpart继承自TreeEditPart。

      代码可参考http://www.cnblogs.com/youngerbaby/archive/2007/03/02/661154.html，本文略。

 

      3、分类的实现，首先创建模型类：

/** * 自定义类，用于记录diagramTreeEditPart下级分类节点所需信息 * * @generated NOT */ public class EListContainer extends EObjectImpl { private EList children; private String type; public EListContainer(EList lst, String type) { this.children = lst; this.type = type; } public EList getChildren() { return children; } public String getType() { return type; } public String getName() { if (type.equals("g1")) return Messages.Group1; else if (type.equals("g2")) return Messages.Group2; else if (type.equals("g3")) return Messages.Group3; else return ""; } }

 

然后在RootDiagramTreeEditPart的getModelChildren方法中，创建分类节点：

 

protected List getModelChildren() { EList children = model.getChildren(); List lst = new ArrayList(); EList g1s= new BasicEList(); EList others = new BasicEList(); // 整理归类 for (Iterator it = children.iterator(); it.hasNext();) { EObject childview = (EObject) it.next(); if (childview instanceof Node && ((Node) childview).getElement() instanceof G1) g1s.add(childview); else others.add(childview); } lst.add(new EListContainer(g1s, "g1")); lst.add(new EListContainer(model.getEdges(), "g2")); lst.add(new EListContainer(others, "g3")); return lst; }

 

4、实现创建删除edge时候，对相关TreeEditPart进行刷新，同样在RootDiagramTreeEditPart类中，覆盖handleNotificationEvent方法

 

@Override protected void handleNotificationEvent(Notification event) { Object feature = event.getFeature(); if (MyPackage.eINSTANCE.getMycontainer_Relation() == feature || || NotationPackage.eINSTANCE.getDiagram_PersistedEdges() == feature) { refreshChildren(); } else { super.handleNotificationEvent(event); } } 

 

5、避免出现选中分类节点后又自动选择根节点的现象，重载DiagramEditor中相关方法

private SelectionSynchronizer synchronizer; protected SelectionSynchronizer getSelectionSynchronizer() { if (synchronizer == null) synchronizer = new SelectionSynchronizer() { public void selectionChanged(SelectionChangedEvent event) { ISelection selection = event.getSelection(); if (selection instanceof IStructuredSelection) { IStructuredSelection s= (IStructuredSelection) selection; if (s.getFirstElement() instanceof CateTreeEditPart) return; } super.selectionChanged(event); } }; return synchronizer; }