Skip to main content
summaryrefslogtreecommitdiffstats
blob: c51b2fed5703ae9406941bf49871e0547aa12823 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/*******************************************************************************
 * Copyright (c) 2004, 2007 Boeing.
 * 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:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.framework.ui.data.model.editor.part;

import java.util.List;
import org.eclipse.draw2d.ConnectionLayer;
import org.eclipse.draw2d.FanRouter;
import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.FreeformLayer;
import org.eclipse.draw2d.FreeformLayout;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.MarginBorder;
import org.eclipse.draw2d.ShortestPathConnectionRouter;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gef.LayerConstants;
import org.eclipse.gef.SnapToGeometry;
import org.eclipse.gef.SnapToHelper;
import org.eclipse.gef.editparts.AbstractGraphicalEditPart;
import org.eclipse.gef.editpolicies.RootComponentEditPolicy;
import org.eclipse.gef.editpolicies.SnapFeedbackPolicy;
import org.eclipse.osee.framework.ui.data.model.editor.model.IModelListener;
import org.eclipse.osee.framework.ui.data.model.editor.model.ODMDiagram;
import org.eclipse.osee.framework.ui.data.model.editor.policy.DiagramLayoutEditPolicy;
import org.eclipse.swt.SWT;

/**
 * @author Roberto E. Escobar
 */
public class DiagramEditPart extends AbstractGraphicalEditPart implements LayerConstants {

   protected IModelListener modelListener = new IModelListener() {

      @Override
      public void onModelEvent(Object object) {
         handleModelEvent(object);
      }
   };

   public DiagramEditPart(Object model) {
      super();
      setModel(model);
   }

   @Override
   public void activate() {
      super.activate();
      ((ODMDiagram) getModel()).addListener(modelListener);
   }

   @Override
   protected IFigure createFigure() {
      Figure f = new FreeformLayer();
      f.setBorder(new MarginBorder(5));
      f.setLayoutManager(new FreeformLayout());

      ConnectionLayer connLayer = (ConnectionLayer) getLayer(LayerConstants.CONNECTION_LAYER);
      FanRouter router = new FanRouter();
      router.setSeparation(20);
      router.setNextRouter(new ShortestPathConnectionRouter(f));
      connLayer.setConnectionRouter(router);
      connLayer.setAntialias(SWT.ON);

      return f;
   }

   @Override
   protected void createEditPolicies() {
      installEditPolicy(EditPolicy.COMPONENT_ROLE, new RootComponentEditPolicy());
      installEditPolicy(EditPolicy.LAYOUT_ROLE, new DiagramLayoutEditPolicy());
      installEditPolicy("Snap Feedback", new SnapFeedbackPolicy());
   }

   @Override
   public void deactivate() {
      ((ODMDiagram) getModel()).removeListener(modelListener);
      super.deactivate();
   }

   protected void handleModelEvent(Object object) {
      refreshVisuals();
      refreshChildren();
   }

   @Override
   public Object getAdapter(Class adapter) {
      if (SnapToHelper.class == adapter) {
         Object snapPropertObject = getViewer().getProperty(SnapToGeometry.PROPERTY_SNAP_ENABLED);
         return Boolean.TRUE.equals(snapPropertObject) ? new SnapToGeometry(this) : null;
      }
      return super.getAdapter(adapter);
   }

   @Override
   protected List getModelChildren() {
      return ((ODMDiagram) getModel()).getContent();
   }
}

Back to the top