package org.simantics.plant3d.geometry;

import java.util.Collection;
import java.util.Collections;
import java.util.Map;

import org.jcae.opencascade.jni.BRepBuilderAPI_MakeEdge;
import org.jcae.opencascade.jni.BRepBuilderAPI_MakeFace;
import org.jcae.opencascade.jni.BRepBuilderAPI_MakeWire;
import org.jcae.opencascade.jni.BRepPrimAPI_MakeRevol;
import org.jcae.opencascade.jni.GC_MakeArcOfCircle;
import org.jcae.opencascade.jni.GC_MakeSegment;
import org.jcae.opencascade.jni.TopoDS_Edge;
import org.jcae.opencascade.jni.TopoDS_Face;
import org.jcae.opencascade.jni.TopoDS_Shape;
import org.jcae.opencascade.jni.TopoDS_Wire;
import org.simantics.db.Resource;
import org.simantics.opencascade.OccTriangulator;

public class PumpGeometryProvider extends BuiltinGeometryProvider  {
	
	public PumpGeometryProvider(Resource resource) {
		super(resource);
	}

	private double length = 0.5;
	private double width = 0.25;
	
	@Override
	public Collection<TopoDS_Shape> getModel() throws Exception {
		double h = width * 0.5;
		double h2 = width * 0.1;
		double ld2 = length * 0.5;
		double wd2 = width * 0.5;
		
		double r1 = width * 0.5;
		double r2 = r1 *0.2;
		double r3 = r1 *0.5;
		
		double l1 = length * 0.2;
		double l2 = length * 0.2;
		double l2b = l2 + length * 0.1;
		double l3 = length * 0.6;
		
		double dir[] = new double[]{1.0,0.0,0.0};
		
		TopoDS_Shape foundation = OccTriangulator.makeBox(-ld2, 0.0, -wd2, ld2, h2, wd2);
		TopoDS_Shape rotator = OccTriangulator.makeCylinder(new double[]{-ld2,h+h2,0.0}, dir, r1, l1);
		TopoDS_Shape axis = OccTriangulator.makeCylinder(new double[]{-ld2+l1,h+h2,0.0}, dir, r2, l2b);
		TopoDS_Shape motor = OccTriangulator.makeCylinder(new double[]{-ld2+l1+l2,h+h2,0.0}, dir, r3, l3);
		TopoDS_Shape motorBox = OccTriangulator.makeBox(-ld2+l1+l2, h2, -r3, ld2, h2+h-r3, r3);
		
		TopoDS_Shape shape = OccTriangulator.makeCompound(new TopoDS_Shape[]{foundation,rotator,axis,motor,motorBox});
		foundation.delete();
		rotator.delete();
		axis.delete();
		motor.delete();
		motorBox.delete();
		return Collections.singletonList(shape);
	}
	
	@Override
	public void setProperties(Map<String, Object> props) {
		if (props.containsKey("length"))
			length = (Double)props.get("length");
		if (props.containsKey("width")) {
			width = (Double)props.get("width");
		}
		
	}

}