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         public AddComponentAction(InteractiveVtkComposite panel, P3DRootNode root) {
  54                 super(panel);
  55                 this.root = root;
  56                 setText("Add Component");
  57                 setImageDescriptor(Activator.imageDescriptorFromPlugin(Activator.PLUGIN_ID, "icons/Component.png"));
  58                 nodeMap = root.getNodeMap();
  59                 gizmo = new TerminalSelectionGizmo(panel);
  60         }
  61
  62         public void setComponent(PipelineComponent component) {
  63                 this.component = component;
  64
  65                 allowed.clear();
  66                 if (component instanceof Nozzle) {
  67                     if (component.getNext() == null && component.getPrevious() == null) {
  68                     allowed.add(PositionType.NEXT);
  69                 }
  70                 } else {
  71                 if (component.getNext() == null) {
  72                         allowed.add(PositionType.NEXT);
  73                 }
  74                 if (component.getPrevious() == null) {
  75                         allowed.add(PositionType.PREVIOUS);
  76                 }
  77                 if (component instanceof InlineComponent && !component.getControlPoint().isFixed()){
  78                         allowed.add(PositionType.SPLIT);
  79                 }
  80                 }
  81                 setEnabled(allowed.size() > 0);
  82         }
  83
  84         private Double length;
  85         private Double angle;
  86         private Double diameter;
  87         private Double turnRadius;
  88
  89         @Override
  90         public void run() {
  91
  92                 ComponentSelectionDialog dialog = new ComponentSelectionDialog(Display.getCurrent().getActiveShell(), allowed, component);
  93                 if (dialog.open() == ComponentSelectionDialog.CANCEL)
  94                         return;
  95                 toAdd = dialog.getSelected();
  96                 if (toAdd == null)
  97                         return;
  98                 this.insertPosition = dialog.getInsertPosition();
  99                 this.insertAdjustable = dialog.isInsertAdjustable();
 100                 this.lengthAdjustable = dialog.isLenghtAdjustable();
 101                 this.length = dialog.getLength();
 102                 this.angle = dialog.getAngle();
 103                 this.diameter = dialog.getDiameter();
 104                 this.turnRadius = dialog.getTurnRadius();
 105                 allowed = dialog.filterAllowed();
 106                 gizmo.setComponent(component, allowed);
 107                 super.run();
 108                 panel.refresh();
 109         }
 110
 111         @Override
 112         public boolean keyPressed(KeyEvent e) {
 113                 if (e.getKeyCode() == KeyEvent.VK_ESCAPE)
 114                         panel.useDefaultAction();
 115                 return true;
 116
 117         }
 118
 119         public void attach() {
 120                 if (component == null)
 121                         return;
 122
 123                 super.attach();
 124                 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
 125                         public void run() {
 126                                 attachUI();
 127                         }
 128                 });
 129
 130         }
 131
 132         public void deattach() {
 133 //              deactivate();
 134                 component = null;
 135                 nodeMap.commit("Add component");
 136                 deattachUI();
 137                 super.deattach();
 138                 panel.refresh();
 139         }
 140
 141         private void attachUI() {
 142                 //panel.setCursor(activeCursor);
 143                 gizmo.attach(panel);
 144         }
 145
 146         private void deattachUI() {
 147                 //panel.setCursor(Cursor.getPredefinedCursor(Cursor.DEFAULT_CURSOR));
 148                 gizmo.deattach();
 149         }
 150
 151         @Override
 152         public boolean mouseMoved(MouseEvent e) {
 153                 return getDefaultAction().mouseMoved(e);
 154         }
 155
 156         @Override
 157         public boolean mousePressed(MouseEvent e) {
 158                 return getDefaultAction().mousePressed(e);
 159         }
 160
 161         @Override
 162         public boolean mouseReleased(MouseEvent e) {
 163                 return getDefaultAction().mouseReleased(e);
 164         }
 165
 166         @Override
 167         public boolean mouseDragged(MouseEvent e) {
 168                 return getDefaultAction().mouseDragged(e);
 169         }
 170
 171         public void doInsert(PositionType position) {
 172             try  {
 173                 InsertInstruction inst = new InsertInstruction();
 174                 inst.typeUri = toAdd.getUri();
 175                 inst.angle = angle;
 176                 inst.diameter = diameter;
 177                 inst.length = length;
 178                 inst.turnRadius = turnRadius;
 179                 inst.insertPosition = insertPosition;
 180                 inst.position = position;
 181                 ComponentUtils.addComponent(root, component, inst);
 182             } catch (Exception e) {
 183            ExceptionUtils.logAndShowError("Cannot add component", e);
 184         }
 185 //              try  {
 186 //                      PipelineComponent newComponent = ComponentUtils.createComponent(root,toAdd.getUri());
 187 //                      PipeControlPoint newPcp = newComponent.getControlPoint();
 188 //
 189 //                      PipeControlPoint toPcp = component.getControlPoint();
 190 //                      PipeRun pipeRun = toPcp.getPipeRun();
 191 //
 192 //                      Vector3d dir = null;
 193 //                      Vector3d pos = null;
 194 //
 195 //
 196 //                      if (toPcp.isInline()) {
 197 //                          switch (position) {
 198 //                  case NEXT:
 199 //                      if (toPcp.isDualInline())
 200 //                          toPcp = toPcp.getSubPoint().get(0);
 201 //
 202 //                      break;
 203 //                  case PREVIOUS:
 204 //                      if (toPcp.isDualSub())
 205 //                          toPcp = toPcp.parent;
 206 //                  }
 207 //                          Vector3d start = new Vector3d();
 208 //                          Vector3d end = new Vector3d();
 209 //                  dir = new Vector3d();
 210 //                  toPcp.getInlineControlPointEnds(start, end, dir);
 211 //                  dir.normalize();
 212 //                 switch (position) {
 213 //                case NEXT:
 214 //                    pos = new Vector3d(end);
 215 //                    break;
 216 //                case PREVIOUS:
 217 //                    pos = new Vector3d(start);
 218 //                    break;
 219 //                case SPLIT:
 220 //                    pos = new Vector3d(toPcp.getWorldPosition());
 221 //                    break;
 222 //                }
 223 //
 224 //                      } else if (toPcp.isDirected()) {
 225 //                          dir = new Vector3d(toPcp.getDirection(Direction.NEXT));
 226 //                          pos = new Vector3d(toPcp.getWorldPosition());
 227 //                      } else if (toPcp.isTurn() && toPcp.isFixed()) {
 228 //                          dir = new Vector3d(toPcp.getDirection(position == PositionType.NEXT ? Direction.NEXT : Direction.PREVIOUS));
 229 //                pos = new Vector3d(toPcp.getWorldPosition());
 230 //                if (!lengthAdjustable) {
 231 //                    Vector3d v = new Vector3d(dir);
 232 //                    v.scale(toPcp.getInlineLength());
 233 //                    pos.add(v);
 234 //                } else {
 235 //                    if (insertPosition == PositionType.NEXT) {
 236 //                        Vector3d v = new Vector3d(dir);
 237 //                        v.scale(toPcp.getInlineLength());
 238 //                        pos.add(v);
 239 //                    } else if (insertPosition == PositionType.SPLIT) {
 240 //                        // scale 0.5*length so that we don't remove the length twice from the new component
 241 //                        Vector3d v = new Vector3d(dir);
 242 //                        v.scale(toPcp.getInlineLength()*0.5);
 243 //                        pos.add(v);
 244 //                    }
 245 //                }
 246 //                      }
 247 //
 248 //
 249 //                      if (!toAdd.isSizeChange()) {
 250 //                              String name = component.getPipeRun().getUniqueName(toAdd.getName());
 251 //                              newComponent.setName(name);
 252 //
 253 //                              pipeRun.addChild(newComponent);
 254 //                              if (toAdd.isVariable()) {
 255 //                                      // TODO: these options are not stored into DB. Should they?!
 256 //                                      if (toAdd.getType() == Type.INLINE) {
 257 //                                              newPcp.setLength(length);
 258 ////                                            newPcp.setFixed(true);
 259 //                                      } else if (toAdd.getType() == Type.TURN) {
 260 //                                              newPcp.setTurnAngle(angle);
 261 ////                                            newPcp.setFixed(true);
 262 //                                      }
 263 //                              }
 264 //                              newComponent.updateParameters();
 265 //
 266 //                          Vector3d v = new Vector3d(dir);
 267 //                          if (insertAdjustable) {
 268 //                          if (insertPosition == PositionType.NEXT)
 269 //                              v.scale(newComponent.getControlPoint().getInlineLength());
 270 //                          else if (insertPosition == PositionType.SPLIT)
 271 //                              v.set(0, 0, 0);
 272 //                          else if (insertPosition == PositionType.PREVIOUS)
 273 //                              v.scale(-newComponent.getControlPoint().getInlineLength());
 274 //                          } else {
 275 //                              v.scale(newComponent.getControlPoint().getInlineLength());
 276 //                          }
 277 //                switch (position) {
 278 //                case NEXT:
 279 //                    pos.add(v);
 280 //                    break;
 281 //                case PREVIOUS:
 282 //                    pos.sub(v);
 283 //                    break;
 284 //                case SPLIT:
 285 //                    break;
 286 //                }
 287 //
 288 //                              switch (position) {
 289 //                              case NEXT:
 290 //                                      if (toPcp.isDualInline())
 291 //                                              toPcp = toPcp.getSubPoint().get(0);
 292 //                                      newPcp.insert(toPcp, Direction.NEXT);
 293 //                                      newPcp.setWorldPosition(pos);
 294 //                                      break;
 295 //                              case PREVIOUS:
 296 //                                      if (toPcp.isDualSub())
 297 //                                              toPcp = toPcp.parent;
 298 //                                      newPcp.insert(toPcp, Direction.PREVIOUS);
 299 //                                      newPcp.setWorldPosition(pos);
 300 //                                      break;
 301 //                              case SPLIT:
 302 //                                      PipingRules.splitVariableLengthComponent(newComponent, (InlineComponent)component, true);
 303 //                              }
 304 //                      } else {
 305 //                              PipeRun other = new PipeRun();
 306 //                              String n = root.getUniqueName("PipeRun");
 307 //                              other.setName(n);
 308 //                              other.setPipeDiameter(diameter);
 309 //                              other.setTurnRadius(turnRadius);
 310 //                              root.addChild(other);
 311 //
 312 //
 313 //                              if (position == PositionType.NEXT) {
 314 //                                      PipingRules.addSizeChange(false, pipeRun, other, (InlineComponent)newComponent, toPcp, null);
 315 //                      } else if (position == PositionType.PREVIOUS){
 316 //                              PipingRules.addSizeChange(true, pipeRun, other, (InlineComponent)newComponent, toPcp, null);
 317 //                      }
 318 //                              newPcp.setWorldPosition(pos);
 319 //                              // TODO : chicken-egg problem
 320 //                              newComponent.updateParameters();
 321 //                          Vector3d v = new Vector3d(dir);
 322 //                v.scale(newComponent.getControlPoint().getLength()*0.5);
 323 //                switch (position) {
 324 //                case NEXT:
 325 //                    pos.add(v);
 326 //                    break;
 327 //                case PREVIOUS:
 328 //                    pos.sub(v);
 329 //                    break;
 330 //                case SPLIT:
 331 //                    break;
 332 //                }
 333 //                newPcp.setWorldPosition(pos);
 334 //
 335 //                      }
 336 //
 337 //
 338 //              } catch (Exception e) {
 339 //                      ExceptionUtils.logAndShowError("Cannot add component", e);
 340 //              }
 341         }
 342
 343         public boolean mouseClicked(MouseEvent e) {
 344                 if (e.getClickCount() == 1 && e.getButton() == MouseEvent.BUTTON1) {
 345                         int type = panel.getPickType();
 346                         //panel.setPickType(0);
 347                         panel.setPickType(5);
 348                         vtkProp[] picked = panel.pick(e.getX(), e.getY());
 349                         panel.setPickType(type);
 350                         PositionType position = gizmo.getPickedPosition(picked);
 351
 352                         if (position != null) {
 353                                 doInsert(position);
 354                                 panel.useDefaultAction();
 355                                 return true;
 356                         }
 357                 }
 358                 return getDefaultAction().mouseClicked(e);
 359         }
 360
 361         @Override
 362         public boolean mouseWheelMoved(MouseWheelEvent e) {
 363             return getDefaultAction().mouseWheelMoved(e);
 364         }
 365 }