Publish Plant3D feature

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

diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/geometry/ElbowGeometryProvider.java b/org.simantics.plant3d/src/org/simantics/plant3d/geometry/ElbowGeometryProvider.java
new file mode 100644 (file)
index 0000000..46aea58
--- /dev/null
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/geometry/ElbowGeometryProvider.java
@@ -0,0 +1,47 @@
+package org.simantics.plant3d.geometry;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.Map;
+
+import org.jcae.opencascade.jni.TopoDS_Shape;
+import org.simantics.db.Resource;
+import org.simantics.g3d.math.MathTools;
+import org.simantics.opencascade.OccTriangulator;
+
+public class ElbowGeometryProvider extends BuiltinGeometryProvider  {
+       
+       public ElbowGeometryProvider(Resource resource) {
+               super(resource);
+       }
+
+       private double radius = 0.01;
+       private double turnRadius = 0.05;
+       private double turnAngle = Math.PI * 0.5;
+       
+       @Override
+       public Collection<TopoDS_Shape> getModel() throws Exception {
+               double t = Math.tan((Math.PI - turnAngle) * 0.5);
+               double R = 0.0;
+               if (t > MathTools.NEAR_ZERO)
+                       R = turnRadius / t;
+               TopoDS_Shape shape = OccTriangulator.makeTorus(new double[]{-R,0.0,-turnRadius}, new double[] { 0.0, 1.0, 0.0 }, turnRadius, radius,0.0,Math.PI*2.0,turnAngle);
+//             System.out.println("Create elbow tr:" + turnRadius + " r:" + radius + " angle:"  +turnAngle + " " + R);
+               return Collections.singletonList(shape);
+       }
+       
+       @Override
+       public void setProperties(Map<String, Object> props) {
+               if (props.containsKey("turnRadius"))
+                       turnRadius = (Double)props.get("turnRadius");
+               if (props.containsKey("turnAngle"))
+                       turnAngle = (Double)props.get("turnAngle");
+               if (props.containsKey("radius")) {
+                       radius = (Double)props.get("radius");
+               }
+               if (radius < MathTools.NEAR_ZERO)
+                       radius = MathTools.NEAR_ZERO;
+       }
+
+}
+