import com.comsol.model.*;
import com.comsol.model.util.*;
import java.util.Arrays;
import java.io.*;

public class export_elecStat {

    public static void main(String[] args) {
	System.out.println("Starting");
	run(args[0],args[1]);
	System.out.println("Done");
	return;
    }

    public static int run(String outpath, String fieldf) {
	Model model;
      
	model = ModelUtil.model(System.getProperty("cs.currentmodel"));
	

	//String outpath="/home/mberger/fopiroot/fopiroot_dev/padresponse_COMSOL/";
	//String fieldf="elecDrift.txt";

	File outfile=new File(outpath+fieldf);
	FileWriter writer;
	try 
	    {
		writer=new FileWriter(outfile,false);
		/*
		writer.write("# px=Particle position X\n");
		writer.write("# py=Particle position Y\n");
		writer.write("# pz=Particle position Y\n");
		writer.write("# wEx=Weighting field component X\n");
		writer.write("# wEy=Weighting field component Y\n");
		writer.write("# wEz=Weighting field component Z\n");
		writer.write("# vx=Particle velocity X\n");
		writer.write("# vy=Particle velocity Y\n");
		writer.write("# vz=Particle velocity Z\n");
		writer.write("# px py pz wEx wEy wEz vx vy vz\n");
		*/

		model.result().numerical("par1").set("timeinterp", "on");
		model.result().numerical("par2").set("timeinterp", "on");

		double[][] particleRelT;
		double[][] particleStopT;

		particleRelT=model.result().numerical("par6").getReal();
		particleStopT=model.result().numerical("par7").getReal();

		//System.out.println(Arrays.toString(model.sol("sol3").getPVals()));
		
		double timestep=model.sol("sol3").getPVals()[1]-model.sol("sol3").getPVals()[0];

		//count times
		int ipart,ipos;
		double oldtime=particleRelT[0][0];
		int numRelTimes=1;

		for(ipart=0;ipart<particleRelT.length;ipart++)
		    {
			if(particleRelT[ipart][0]>0)
			    particleRelT[ipart][0]+=timestep;
			
			if(oldtime!=particleRelT[ipart][0])
			    {
				numRelTimes++;
				oldtime=particleRelT[ipart][0];
			    }
		    }

		//fill array with the release times
		double[] reltimes=new double[numRelTimes]; 
		double maxStopTime=0;
		reltimes[0]=particleRelT[0][0];
		int counter=0;
		for(ipart=0;ipart<particleRelT.length;ipart++)
		    {
			if(reltimes[counter]!=particleRelT[ipart][0])
			    {
				counter++;
				reltimes[counter]=particleRelT[ipart][0];
			    }
			if(maxStopTime<particleStopT[ipart][0])
			    maxStopTime=particleStopT[ipart][0];
		    }

		double[] evaltimes=Arrays.copyOf(reltimes,numRelTimes+1);
		evaltimes[numRelTimes]=maxStopTime;
		System.out.println("Times to evaluate:"+Arrays.toString(evaltimes));
		
		model.result().numerical("par1").set("t", evaltimes);
		model.result().numerical("par2").set("t", evaltimes);
		

		double[][] particleX;
		double[][] particleY;
		double[][] particleZ;

		particleX=model.result().numerical("par1").getReal();
		particleY=model.result().numerical("par2").getReal();
		//System.out.println(Arrays.deepToString(particleX));

		int reltimecounter=0;
		int timeInArray=0;
		double posX,posY,posZ=0;

		for(ipart=0;ipart<particleRelT.length;ipart++)
		    {
			if (ipart%10==0)
			    System.out.println("At particle: "+Integer.toString(ipart));

			
			if(particleRelT[ipart][0]!=reltimes[timeInArray])
			    {
				System.out.println("Increasing timeInArray: "+Integer.toString(timeInArray));
				timeInArray++;
				System.out.println("max size:"+Integer.toString(particleX[ipart].length));
			    }

			writer.write(Integer.toString(ipart)+" ");

			//write out position at release
			posX=particleX[ipart][timeInArray];
			posY=0;
			posZ=particleY[ipart][timeInArray];
			writer.write(Double.toString(posX)+" ");
			writer.write(Double.toString(posY)+" ");
			writer.write(Double.toString(posZ)+" ");
			writer.write(Double.toString(particleRelT[ipart][0])+" ");

			//write out position at stop
			posX=particleX[ipart][numRelTimes];
			posY=0;
			posZ=particleY[ipart][numRelTimes];
			writer.write(Double.toString(posX)+" ");
			writer.write(Double.toString(posY)+" ");
			writer.write(Double.toString(posZ)+" ");
			writer.write(Double.toString(particleStopT[ipart][0]));
			
		
			writer.write("\n");
			writer.flush();
		    }
		System.out.println("");
		

		writer.close();
	    }
	catch (IOException e)
	    {
		e.printStackTrace();
		System.err.println("error 1");
	    }
		


	System.out.println("All Done");
	return 0;
    }

}