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 GateValveGeometryProvider extends BuiltinGeometryProvider {
	
	public GateValveGeometryProvider(Resource resource) {
		super(resource);
	}

	private double radius = 0.01;
	
	@Override
	public Collection<TopoDS_Shape> getModel() throws Exception {
		if (radius < MathTools.NEAR_ZERO)
			return Collections.emptyList();
		TopoDS_Shape con = OccTriangulator.makeCone(new double[] {-radius, 0.0, 0.0}, new double[] { 1.0, 0.0, 0.0 }, radius, radius*0.1,radius);
		TopoDS_Shape con2 = OccTriangulator.makeCone(new double[] {radius, 0.0, 0.0}, new double[] { -1.0, 0.0, 0.0 }, radius, radius*0.1,radius);
		//TopoDS_Shape sph = OccTriangulator.makeSphere(0, 0, 0, radius*0.75);
		//TopoDS_Shape cyl = OccTriangulator.makeCylinder(new double[] {0.0, 0.0, 0.0}, new double[] { 0.0, 1.0, 0.0 }, radius*0.75, radius*2);
		TopoDS_Shape cyl = OccTriangulator.makeFillet(OccTriangulator.makeBox(-radius*0.5, -radius*0.5, -radius*0.75, radius*0.5, radius*1.25, radius*0.75), radius*0.25);
		//TopoDS_Shape cyl = OccTriangulator.makeChamfer(OccTriangulator.makeBox(-radius*0.5, -radius*0.5, -radius*0.75, radius*0.5, radius*1.25, radius*0.75), radius*0.25);
		TopoDS_Shape cyl2 = OccTriangulator.makeCylinder(new double[] {0.0, 0.0, 0.0}, new double[] { 0.0, 1.0, 0.0 }, radius*0.125, radius*2);
		TopoDS_Shape tor = OccTriangulator.makeTorus(new double[] { 0.0, radius*2, 0.0}, new double[] { 0.0, 1.0, 0.0 }, radius*0.5, radius*0.125);
		
		TopoDS_Shape shape = OccTriangulator.makeCompound(new TopoDS_Shape[] {con,con2,cyl,cyl2,tor});
		cyl.delete();
		con.delete();
		con2.delete();
		//sph.delete();
		tor.delete();
		return Collections.singletonList(shape);
	}
	
	@Override
	public void setProperties(Map<String, Object> props) {
		if (props.containsKey("radius")) {
			radius = (Double)props.get("radius");
		}
		if (radius < MathTools.NEAR_ZERO)
			radius = MathTools.NEAR_ZERO;
		
	}
	
	@Override
	public void updateCalculatedProperties(Map<String, Object> returnProps) {
		returnProps.put("length", radius*2);
		
	}


}