org.simantics.plant3d/src/org/simantics/plant3d/actions/AddComponentAction.java

   1 package org.simantics.plant3d.actions;
   2
   3 import java.awt.event.KeyEvent;
   4 import java.awt.event.MouseEvent;
   5 import java.awt.event.MouseWheelEvent;
   6 import java.util.HashSet;
   7 import java.util.Set;
   8
   9 import javax.vecmath.Vector3d;
  10
  11 import org.eclipse.swt.widgets.Display;
  12 import org.simantics.g3d.scenegraph.NodeMap;
  13 import org.simantics.g3d.scenegraph.base.INode;
  14 import org.simantics.g3d.vtk.swt.InteractiveVtkComposite;
  15 import org.simantics.g3d.vtk.swt.vtkSwtAction;
  16 import org.simantics.plant3d.Activator;
  17 import org.simantics.plant3d.dialog.ComponentSelectionDialog;
  18 import org.simantics.plant3d.gizmo.TerminalSelectionGizmo;
  19 import org.simantics.plant3d.scenegraph.InlineComponent;
  20 import org.simantics.plant3d.scenegraph.Nozzle;
  21 import org.simantics.plant3d.scenegraph.P3DRootNode;
  22 import org.simantics.plant3d.scenegraph.PipeRun;
  23 import org.simantics.plant3d.scenegraph.PipelineComponent;
  24 import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint;
  25 import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint.Direction;
  26 import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint.PositionType;
  27 import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
  28 import org.simantics.plant3d.utils.ComponentUtils;
  29 import org.simantics.plant3d.utils.Item;
  30 import org.simantics.plant3d.utils.ComponentUtils.InsertInstruction;
  31 import org.simantics.plant3d.utils.Item.Type;
  32 import org.simantics.utils.threads.ThreadUtils;
  33 import org.simantics.utils.ui.ExceptionUtils;
  34
  35 import vtk.vtkProp;
  36
  37 public class AddComponentAction extends vtkSwtAction {
  38
  39
  40         private P3DRootNode root;
  41         private PipelineComponent component;
  42         private NodeMap<vtkProp,INode> nodeMap;
  43
  44         private TerminalSelectionGizmo gizmo;
  45
  46         private Set<PositionType> allowed = new HashSet<PositionType>();
  47
  48         private Item toAdd = null;
  49         private PositionType insertPosition;
  50         private boolean insertAdjustable;
  51         private boolean lengthAdjustable;
  52
  53         private String libUri;
  54
  55         public AddComponentAction(InteractiveVtkComposite panel, P3DRootNode root, String libUri) {
  56                 super(panel);
  57                 this.root = root;
  58                 setText("Add Component");
  59                 setImageDescriptor(Activator.imageDescriptorFromPlugin(Activator.PLUGIN_ID, "icons/Component.png"));
  60                 nodeMap = root.getNodeMap();
  61                 gizmo = new TerminalSelectionGizmo(panel);
  62                 this.libUri = libUri;
  63         }
  64
  65         public void setComponent(PipelineComponent component) {
  66                 this.component = component;
  67
  68                 allowed.clear();
  69                 if (component instanceof Nozzle) {
  70                         if (component.getNext() == null && component.getPrevious() == null) {
  71                                 allowed.add(PositionType.NEXT);
  72                         }
  73                 } else {
  74                         if (component.getNext() == null) {
  75                                 allowed.add(PositionType.NEXT);
  76                         }
  77                         if (component.getPrevious() == null) {
  78                                 allowed.add(PositionType.PREVIOUS);
  79                         }
  80                         if (component instanceof InlineComponent && !component.getControlPoint().isFixed()){
  81                                 allowed.add(PositionType.SPLIT);
  82                         }
  83                 }
  84                 setEnabled(allowed.size() > 0);
  85         }
  86
  87         private Double length;
  88         private Double angle;
  89         private Double diameter;
  90         private Double turnRadius;
  91
  92         @Override
  93         public void run() {
  94
  95                 ComponentSelectionDialog dialog = new ComponentSelectionDialog(Display.getCurrent().getActiveShell(), allowed, component, libUri);
  96                 if (dialog.open() == ComponentSelectionDialog.CANCEL)
  97                         return;
  98                 toAdd = dialog.getSelected();
  99                 if (toAdd == null)
 100                         return;
 101                 this.insertPosition = dialog.getInsertPosition();
 102                 this.insertAdjustable = dialog.isInsertAdjustable();
 103                 this.lengthAdjustable = dialog.isLenghtAdjustable();
 104                 this.length = dialog.getLength();
 105                 this.angle = dialog.getAngle();
 106                 this.diameter = dialog.getDiameter();
 107                 this.turnRadius = dialog.getTurnRadius();
 108                 allowed = dialog.filterAllowed();
 109                 gizmo.setComponent(component, allowed);
 110                 super.run();
 111                 panel.refresh();
 112         }
 113
 114         @Override
 115         public boolean keyPressed(KeyEvent e) {
 116                 if (e.getKeyCode() == KeyEvent.VK_ESCAPE)
 117                         panel.useDefaultAction();
 118                 return true;
 119
 120         }
 121
 122         public void attach() {
 123                 if (component == null)
 124                         return;
 125
 126                 super.attach();
 127                 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
 128                         public void run() {
 129                                 attachUI();
 130                         }
 131                 });
 132
 133         }
 134
 135         public void deattach() {
 136 //              deactivate();
 137                 component = null;
 138                 nodeMap.commit("Add component");
 139                 deattachUI();
 140                 super.deattach();
 141                 panel.refresh();
 142         }
 143
 144         private void attachUI() {
 145                 //panel.setCursor(activeCursor);
 146                 gizmo.attach(panel);
 147         }
 148
 149         private void deattachUI() {
 150                 //panel.setCursor(Cursor.getPredefinedCursor(Cursor.DEFAULT_CURSOR));
 151                 gizmo.deattach();
 152         }
 153
 154         @Override
 155         public boolean mouseMoved(MouseEvent e) {
 156                 return getDefaultAction().mouseMoved(e);
 157         }
 158
 159         @Override
 160         public boolean mousePressed(MouseEvent e) {
 161                 return getDefaultAction().mousePressed(e);
 162         }
 163
 164         @Override
 165         public boolean mouseReleased(MouseEvent e) {
 166                 return getDefaultAction().mouseReleased(e);
 167         }
 168
 169         @Override
 170         public boolean mouseDragged(MouseEvent e) {
 171                 return getDefaultAction().mouseDragged(e);
 172         }
 173
 174         public void doInsert(PositionType position) {
 175                 try  {
 176                         InsertInstruction inst = new InsertInstruction();
 177                         inst.typeUri = toAdd.getUri();
 178                         inst.angle = angle;
 179                         inst.diameter = diameter;
 180                         inst.length = length;
 181                         inst.turnRadius = turnRadius;
 182                         inst.insertPosition = insertPosition;
 183                         inst.position = position;
 184                         ComponentUtils.addComponent(root, component, inst);
 185                 } catch (Exception e) {
 186                         ExceptionUtils.logAndShowError("Cannot add component", e);
 187                 }
 188         }
 189
 190         public boolean mouseClicked(MouseEvent e) {
 191                 if (e.getClickCount() == 1 && e.getButton() == MouseEvent.BUTTON1) {
 192                         int type = panel.getPickType();
 193                         //panel.setPickType(0);
 194                         panel.setPickType(5);
 195                         vtkProp[] picked = panel.pick(e.getX(), e.getY());
 196                         panel.setPickType(type);
 197                         PositionType position = gizmo.getPickedPosition(picked);
 198
 199                         if (position != null) {
 200                                 doInsert(position);
 201                                 panel.useDefaultAction();
 202                                 return true;
 203                         }
 204                 }
 205                 return getDefaultAction().mouseClicked(e);
 206         }
 207
 208         @Override
 209         public boolean mouseWheelMoved(MouseWheelEvent e) {
 210                 return getDefaultAction().mouseWheelMoved(e);
 211         }
 212 }