package seed.minerva.optics.polarisation;

import java.text.DecimalFormat;
import java.util.List;

import binaryMatrixFile.BinaryMatrixFile;
import binaryMatrixFile.BinaryMatrixWriter;

import otherSupport.ColorMaps;
import otherSupport.RandomManager;
import otherSupport.StatusOutput;
import oneLiners.OneLiners;
import seed.minerva.MinervaOpticsSettings;
import jafama.FastMath;
import seed.minerva.optics.Util;
import seed.minerva.optics.collection.IntersectionProcessor;
import seed.minerva.optics.collection.LensDepolarisationInfoCollector;
import seed.minerva.optics.drawing.VRMLDrawer;
import seed.minerva.optics.interfaces.Absorber;
import seed.minerva.optics.interfaces.IsoIsoAntiReflective;
import seed.minerva.optics.interfaces.IsoIsoInterface;
import seed.minerva.optics.interfaces.IsoIsoStdFresnel;
import seed.minerva.optics.interfaces.IsoUniaxialInterface;
import seed.minerva.optics.interfaces.NullInterface;
import seed.minerva.optics.interfaces.Reflector;
import seed.minerva.optics.interfaces.SimplePolariser;
import seed.minerva.optics.materials.BK7;
import seed.minerva.optics.materials.Calcite;
import seed.minerva.optics.materials.IsotropicFixedIndexGlass;
import seed.minerva.optics.materials.LithiumNiobate;
import seed.minerva.optics.materials.CrystalQuartz;
import seed.minerva.optics.materials.SchottSFL6;
import seed.minerva.optics.optics.Box;
import seed.minerva.optics.optics.SimpleDoubleConvexLens;
import seed.minerva.optics.pointSpread.DualGaussianPSF;
import seed.minerva.optics.pointSpread.GaussianPSF;
import seed.minerva.optics.pointSpread.PointSpreadBuilder;
import seed.minerva.optics.pointSpread.PointSpreadFunction;
import seed.minerva.optics.surfaces.Cylinder;
import seed.minerva.optics.surfaces.Disc;
import seed.minerva.optics.surfaces.Dish;
import seed.minerva.optics.surfaces.Iris;
import seed.minerva.optics.surfaces.Square;
import seed.minerva.optics.tracer.Tracer;
import seed.minerva.optics.types.Element;
import seed.minerva.optics.types.Interface;
import seed.minerva.optics.types.Intersection;
import seed.minerva.optics.types.Material;
import seed.minerva.optics.types.Medium;
import seed.minerva.optics.types.Optic;
import seed.minerva.optics.types.Pol;
import seed.minerva.optics.types.RaySegment;
import seed.minerva.optics.types.Surface;

/**
 * Investigation of depolarision / polarisation modification effect
 * due to a lens.
 * 
 * @author oliford
 */
public class LensDepolarisation {
	final static String outPath = MinervaOpticsSettings.getAppsOutputPath() + "/rayTracing/lensDepol";
	
	final static int maxHits = 10000;
	final static int maxRays = 100000;
	final static double maxTheta = 30 * Math.PI / 180;
	final static double rt2 = Math.sqrt(2);
	
	final static double wavelen = 593e-9;
	
	public static void main(String[] args) {
		
		double r = 0.5;
		double f = 1.0;
		double u = 1.0;
		double v = (u == f) ? 1.0 : 1 / (1/f - 1/u);
		
		//Lens lens = new Lens("lens", new double[]{ u, 0, 0 }, new double[]{ -1, 0, 0 }, r, new Medium(lensMat), IsoIsoStdFresnel.ideal(), f, wavelen);
		Medium lensGlass = new Medium(new SchottSFL6());
		//Medium lensGlass = new Medium(new Quartz(), new double[][]{ { rt2/2, 0, rt2/2 } }, 300);
		//IsotropicFixedIndexGlass lensGlass = new IsotropicFixedIndexGlass(1.805);
		Interface lensIFace = IsoIsoInterface.ideal();
		//Interface lensIFace = IsoIsoStdFresnel.ideal();
		//Interface lensIFace = new IsoIsoAntiReflective(1.4, wavelen/4/1.4, 0);
		//Interface lensIFace = IsoUniaxialInterface.ideal();
		
		SimpleDoubleConvexLens lens = new SimpleDoubleConvexLens("lens", new double[]{ u, 0, 0 }, new double[]{ -1, 0, 0 }, r, lensGlass, lensIFace, f, wavelen);
		Iris iris = new Iris("iris", new double[]{ u, 0, 0 }, new double[]{ -1, 0, 0 }, 1.4*r, 0.99*r, Absorber.ideal());
		Square imgPlane = new Square("fwdPlane", new double[]{ (u+v), 0, 0 }, new double[]{ -1, 0, 0 }, new double[]{ 0, 0, 1 }, 1, 1, Absorber.ideal());		
		
		Disc midLens = new Disc("midLens", new double[]{ u, 0, 0 }, new double[]{ -1, 0, 0 }, r, NullInterface.ideal());

		/*
		 * //L1 from __WRONG__ AUG tube optics
		Iris iris = new Iris("lens1Iris", new double[]{ 0.2595-0.2595+u, 0,0 }, new double[]{ -1, 0, 0 }, 0.062, 0.025, Absorber.ideal());
		Dish lens1Front = new Dish("lens1Front", new double[]{ 0.2595-0.2595+u, 0,0 }, new double[]{ 1, 0, 0 }, 0.311, 0.025,  lensGlass, null, lensIFace);
		Disc lens1Back = new Disc("lens1Back", new double[]{ 0.264-0.2595+u, 0,0 }, new double[]{ 1, 0, 0}, 0.025, null, lensGlass, lensIFace);
		Optic lens = new Optic("lens", new Element[]{ lens1Front, lens1Back });
		*/
		
		Optic all = new Optic("all", new Element[]{ lens, iris, midLens, imgPlane });
		
		VRMLDrawer vrmlOut = new VRMLDrawer(outPath + "/lensDepol.vrml", 0.01);
		vrmlOut.setDrawPolarisationFrames(true);
		vrmlOut.setSkipRays(maxHits <= 1 ? 0 : (maxHits/99));
		
		double col[][] = ColorMaps.jet(maxRays);
		
		LensDepolarisationInfoCollector polCheck = new LensDepolarisationInfoCollector(midLens, imgPlane, 50, 0, outPath + "/polInf");
		
		PointSpreadBuilder psfBuild = new PointSpreadBuilder(imgPlane);
		psfBuild.startNewPSF(new double[]{ 0, 0, 0 }, new DualGaussianPSF());
		
		psfBuild.setMaxCoherentIntegrationRadius(0.001);
	
		StatusOutput stat = new StatusOutput(MullerTest.class, maxHits);
		for(int i=0; i < maxRays; i++) {
			RaySegment ray = new RaySegment();
			
			ray.startPos = new double[]{ 0, 0, 0 };
			ray.length = Double.POSITIVE_INFINITY;
			ray.dir = Tracer.generateRandomRayTowardSurface(ray.startPos, lens);
			
			ray.up = Tracer.generateRayFanConsistentPolarisationDefinition(ray.startPos, ray.dir, lens.getBoundarySphereCentre(), new double[]{0,0,1});
			ray.up = Util.cross(Util.reNorm(Util.cross(ray.dir, new double[]{0,0,1})), ray.dir);
			
			ray.E0 = PointSpreadFunction.getInputStatesForMuellerCalc();
			
			ray.wavelength = wavelen;
			
			Tracer.trace(all, ray, 1000, 0.1, false);
			
			vrmlOut.drawRay(ray, col[i]);
			
			ray.processIntersections(imgPlane, psfBuild, polCheck);
			psfBuild.nextCoherentSet();
			
			List<Intersection> hits = ray.getIntersections(imgPlane);
			
			if(hits.size() > 0){
				Intersection hit = hits.get(0); //there should only be 1
				
				double theta = FastMath.acos(ray.dir[0]);
				double phi = FastMath.atan2(ray.dir[1], ray.dir[2]);
				
				//put final ray back in sense of 'z'
				hit.incidentRay.rotatePolRefFrame(new double[]{ 0, 0, 1 });
			}
			
			
			Pol.recoverAll();
			
			int n = psfBuild.getNPointsCollected();
			stat.doStatus(n);
			if(n >= maxHits)
				break;
		}
		stat.done();

		DualGaussianPSF psf = (DualGaussianPSF) psfBuild.psfDone(false);
		System.out.println("Muller: ");
		psf.dumpNormalisedMueller();

		vrmlOut.drawOptic(all);
		vrmlOut.destroy();
		
		polCheck.write();
	}
	
	
}