First version of fixed nozzle positions

[simantics/3d.git] / org.simantics.plant3d / src / org / simantics / plant3d / geometry / PumpGeometryProvider.java

diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/geometry/PumpGeometryProvider.java b/org.simantics.plant3d/src/org/simantics/plant3d/geometry/PumpGeometryProvider.java
index 451abe1e1da5251bb096caefca8a9efea5cc58c1..47b2d2d1b33620bd8fe60ef4867f114bb1db4724 100644 (file)
--- a/org.simantics.plant3d/src/org/simantics/plant3d/geometry/PumpGeometryProvider.java
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/geometry/PumpGeometryProvider.java
@@ -4,12 +4,16 @@ import java.util.Collection;
 import java.util.Collections;
 import java.util.Map;
 
+import javax.vecmath.AxisAngle4d;
+import javax.vecmath.Vector3d;
+
 import org.jcae.opencascade.jni.TopoDS_Shape;
 import org.simantics.db.Resource;
 import org.simantics.g3d.math.MathTools;
 import org.simantics.opencascade.OccTriangulator;
+import org.simantics.plant3d.scenegraph.Nozzle;
 
-public class PumpGeometryProvider extends BuiltinGeometryProvider  {
+public class PumpGeometryProvider extends BuiltinGeometryProvider  implements FixedNozzleProvider{
        
        public PumpGeometryProvider(Resource resource) {
                super(resource);
@@ -62,5 +66,34 @@ public class PumpGeometryProvider extends BuiltinGeometryProvider  {
                }
                
        }
+       
+       @Override
+       public int numberOfNozzles() {
+           return 2;
+       }
+       
+       @Override
+       public void updateNozzlePosition(int index, Nozzle nozzle) {
+           Double fl = nozzle.getFlowLength();
+           if (fl == null)
+               fl = 0.1;
+           if (index == 0) {
+               nozzle.setPosition(new Vector3d(-length*0.5- fl,width*0.6,0.0));
+               nozzle.setOrientation(MathTools.getQuat(new AxisAngle4d(0,1,0,Math.PI)));
+           } else if (index == 1) {
+               nozzle.setPosition(new Vector3d(-length*0.4,width*1.1+ fl,0.0));
+            nozzle.setOrientation(MathTools.getQuat(new AxisAngle4d(0,0,1,Math.PI*0.5)));
+           }
+           
+       }
+       
+       @Override
+       public String getNozzleName(int index) {
+           switch (index) {
+           case 0 : return "Input";
+           case 1: return "Output";
+           default: return null;
+           }
+       }
 
 }