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

   1 package org.simantics.plant3d.actions;
   2
   3 import java.awt.event.KeyEvent;
   4 import java.awt.event.MouseEvent;
   5
   6 import javax.vecmath.Point3d;
   7 import javax.vecmath.Vector3d;
   8
   9 import org.simantics.g3d.math.MathTools;
  10 import org.simantics.g3d.math.Ray;
  11 import org.simantics.g3d.scenegraph.IG3DNode;
  12 import org.simantics.g3d.scenegraph.base.INode;
  13 import org.simantics.g3d.toolbar.ToolComposite;
  14 import org.simantics.g3d.vtk.common.VTKNodeMap;
  15 import org.simantics.g3d.vtk.swt.InteractiveVtkComposite;
  16 import org.simantics.g3d.vtk.swt.TranslateAction;
  17 import org.simantics.g3d.vtk.utils.vtkUtil;
  18 import org.simantics.plant3d.Activator;
  19 import org.simantics.plant3d.scenegraph.InlineComponent;
  20 import org.simantics.plant3d.scenegraph.PipelineComponent;
  21 import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint.Direction;
  22
  23 public class TranslateInlineAction extends TranslateAction{
  24
  25     private boolean inline;
  26
  27         private Vector3d s;
  28         private Vector3d e;
  29         private Vector3d dir;
  30
  31         public TranslateInlineAction(InteractiveVtkComposite panel, VTKNodeMap<?, ? extends INode> nodeMap, ToolComposite toolComposite) {
  32                 super(panel, nodeMap, toolComposite);
  33                 setImageDescriptor(Activator.imageDescriptorFromPlugin("com.famfamfam.silk", "icons/arrow_refresh.png"));
  34         }
  35
  36         @Override
  37         public void setNode(IG3DNode node) {
  38                 super.setNode(node);
  39                 if (node instanceof InlineComponent) {
  40                         InlineComponent comp = (InlineComponent)node;
  41                         if (comp.isVariableLength()) {
  42                 setEnabled(false);
  43                         } else if (comp.getNext() == null || comp.getPrevious() == null) {
  44                 setEnabled(true);
  45                 inline = false;
  46                 dir = comp.getControlPoint().getPathLegDirection(Direction.NEXT);
  47                 dir.normalize();
  48             } else {
  49                                 setEnabled(true);
  50                                 inline = true;
  51                                 PipelineComponent prev = comp.getPrevious();
  52                                 PipelineComponent next = comp.getNext();
  53                                 if (prev == null || next == null) {
  54                                         // TODO : we should support open ended translation (when translated component is the first or last of the run).
  55                                         setEnabled(false);
  56                                         return;
  57                                 }
  58                                 if (prev.getControlPoint().isInline() && !prev.getControlPoint().isFixedLength() && prev.getPrevious() != null)
  59                                         prev = prev.getPrevious();
  60                                 if (next.getControlPoint().isInline() && !next.getControlPoint().isFixedLength() && next.getNext() != null) {
  61                                         next = next.getNext();
  62                                 }
  63                                 Point3d ns = new Point3d();
  64                                 Point3d ne = new Point3d();
  65                                 Point3d ps = new Point3d();
  66                                 Point3d pe = new Point3d();
  67                                 next.getEnds(ns, ne);
  68                                 prev.getEnds(ps, pe);
  69                                 dir = comp.getControlPoint().getPathLegDirection(Direction.NEXT);
  70                                 dir.normalize();
  71                                 // We may have offsets in the path leg, hence we have to project the coordinates.
  72                                 Vector3d wp = node.getWorldPosition();
  73                                 if (prev.getControlPoint().isVariableLength())
  74                                         s = MathTools.closestPointOnStraight(ps, wp, dir);
  75                                 else
  76                                         s = MathTools.closestPointOnStraight(pe, wp, dir);
  77                                 if (next.getControlPoint().isVariableLength())
  78                                         e = MathTools.closestPointOnStraight(ne, wp, dir);
  79                                 else
  80                                         e = MathTools.closestPointOnStraight(ns, wp, dir);
  81                                 // Remove component's own space from end points to get actual movement range
  82                                 double l = comp.getControlPoint().getInlineLength();
  83                                 Vector3d ld = new Vector3d(dir);
  84                                 ld.scale(l);
  85                                 s.add(ld);
  86                                 e.sub(ld);
  87                         }
  88                 } else {
  89                         setEnabled(false);
  90                 }
  91         }
  92
  93         @Override
  94         protected void createTools(ToolComposite toolComposite) {
  95
  96         }
  97
  98         @Override
  99         public boolean keyPressed(KeyEvent e) {
 100                 if (e.getKeyCode() == KeyEvent.VK_ESCAPE)
 101                         panel.useDefaultAction();
 102                 if (!inline) {
 103                     return super.keyPressed(e);
 104                 } else {
 105                 if (valid)
 106                         return true;
 107
 108                 update();
 109                 return true;
 110                 }
 111         }
 112
 113         @Override
 114         public void setWorldCoord(boolean b) {
 115
 116         }
 117
 118         @Override
 119         public boolean mouseDragged(MouseEvent e) {
 120                 //if ((e.getModifiersEx() & MouseEvent.BUTTON1_DOWN_MASK) > 0 && valid) {
 121             if (e.getButton() == MouseEvent.BUTTON1  && valid) {
 122                         if (inline) {
 123                         Vector3d translate = getTranslate(e.getX(), e.getY(), prevTranslate);
 124                         //System.out.println("translate " + translate);
 125                         if (translate == null)
 126                                 return true;
 127                         //boolean step = ((e.getModifiers() & MouseEvent.CTRL_MASK) > 0);
 128                         Vector3d pos = new Vector3d(node.getWorldPosition());
 129
 130 //                      System.out.println(pos + " " + translate);
 131                         //pos.add(translate);
 132                         pos.set(translate);
 133                         //pos = constaints(pos, step);
 134                         setWorldPos(pos);
 135                         update();
 136                         } else {
 137                             super.mouseDragged(e);
 138                         }
 139                 } else {
 140                         getDefaultAction().mouseDragged(e);
 141                         update();
 142                 }
 143                 return true;
 144         }
 145
 146         protected Vector3d getTranslate(double x, double y, Vector3d offset) {
 147
 148                 Ray ray = vtkUtil.createMouseRay(panel.getRenderer(),x, y);
 149
 150                 Vector3d p = node.getWorldPosition();
 151
 152                 if (inline) {
 153                 Vector3d i1 = new Vector3d();
 154                 Vector3d i2 = new Vector3d();
 155
 156                 double mu[] = new double[2];
 157                 MathTools.intersectStraightStraight( p, dir,ray.pos, ray.dir, i2, i1,mu);
 158
 159                 Vector3d t = MathTools.closestPointOnEdge(i1, s, e);
 160                 return t;
 161                 } else {
 162                     return super.getTranslate(x, y, offset);
 163                 }
 164
 165         }
 166
 167
 168 }