// MonteScatter.java
// This class is an applet that uses monte carlo methods to determine scattering
// path lengths in a number of cloud structures.

import java.awt.*;
import java.applet.*;

public class MonteScatter extends Applet implements Runnable {
	public void init() {
		Panel tmp_panel;
		int i;

		thread=null;
		cloud=new Slab1D(10.0);
		sm=new RandomScatter();
		dir=new RTVector(1);
		dir.Set(0,-1.0);

		this.setLayout(new BorderLayout());

		main_panel=new Panel();
		main_panel.setLayout(new GridLayout(3,3));

		cloud_choice=new Choice();
		cloud_choice.addItem("1D Slab");
		cloud_choice.addItem("2D Slab");
		cloud_choice.addItem("2D Sphere");
		cloud_choice.addItem("2D Fractal");
		cloud_choice.addItem("3D Slab");
		cloud_choice.addItem("3D Sphere");
		cloud_choice.addItem("3D Fractal");
		main_panel.add(cloud_choice);

		scatter_choice=new Choice();
		scatter_choice.addItem("Isotropic");
		scatter_choice.addItem("Rayleigh");
		scatter_choice.addItem("Henye-Greenstein");
		main_panel.add(scatter_choice);

		aoi_choice=new Choice();
		aoi_choice.addItem("0 degrees");
		aoi_choice.addItem("15 degrees");
		aoi_choice.addItem("30 degrees");
		aoi_choice.addItem("45 degrees");
		aoi_choice.addItem("60 degrees");
		aoi_choice.addItem("85 degrees");
		main_panel.add(aoi_choice);

		ss_button=new Button("Start");
		main_panel.add(ss_button);

		tmp_panel=new Panel();
		tmp_panel.add(new Label("Optical Depth:"));
		tau_label=new Label("10.0");
		tmp_panel.add(tau_label);
		main_panel.add(tmp_panel);

		iteration_count=0;
		tmp_panel=new Panel();
		tmp_panel.add(new Label("Num. Iteration:"));
		iteration_label=new Label("000000");
		tmp_panel.add(iteration_label);
		main_panel.add(tmp_panel);

		tmp_panel=new Panel();
		tmp_panel.add(new Label("Trans. Path Length:"));
		aplt_label=new Label("0.0");
		tmp_panel.add(aplt_label);
		main_panel.add(tmp_panel);
		aplt_sum=0.0;

		tmp_panel=new Panel();
		tmp_panel.add(new Label("Reflected Path Length:"));
		aplr_label=new Label("0.0");
		tmp_panel.add(aplr_label);
		main_panel.add(tmp_panel);
		aplr_sum=0.0;

		tmp_panel=new Panel();
		tmp_panel.add(new Label("% Reflected:"));
		albedo_label=new Label("0.0");
		tmp_panel.add(albedo_label);
		main_panel.add(tmp_panel);
		num_reflected=0;

		this.add("South",main_panel);

		tmp_panel=new Panel();
		tmp_panel.setLayout(new GridLayout(1,3));

		data_plot=new AdjustablePlot2D();
		data_plot.AddDataSet(cloud.GetData());
		data_plot.x_label="Optical Depth Density";
		data_plot.y_label="Altitude";
		tmp_panel.add("West",data_plot);

		cloud_panel=new Panel();
		tmp_panel.add("Center",cloud_panel);

		distrib_plot=new Plot2D();
		trans_distrib_data=new DataSet2D(20,10);
		trans_distrib_data.SetSortOrder(DataSet2D.SORT_Y);
		for(i=19; i>=0; i--) {
			trans_distrib_data.AddElement(new Point2D(1.0*i,0.0));
		}
		trans_distrib_data.plot_color=Color.red;
		trans_distrib_data.description="Transmitted";
		distrib_plot.AddDataSet(trans_distrib_data);
		ref_distrib_data=new DataSet2D(20,10);
		ref_distrib_data.SetSortOrder(DataSet2D.SORT_Y);
		for(i=19; i>=0; i--) {
			ref_distrib_data.AddElement(new Point2D(1.0*i,0.0));
		}
		ref_distrib_data.plot_color=Color.blue;
		ref_distrib_data.description="Reflected";
		distrib_plot.AddDataSet(ref_distrib_data);
		distrib_plot.x_label="Path Length";
		distrib_plot.y_label="Count";
		tmp_panel.add(distrib_plot);

		this.add("Center",tmp_panel);

		cloud.DrawCloud(cloud_panel);
	}

	public void run() {
		boolean flag=true;
		int i;

		try {
			while(true) {
				pp=new PhotonPath(cloud,sm,dir);
				if(pp.Follow(10000)) {
					iteration_count++;
					iteration_label.setText(Integer.toString(iteration_count));
					if(pp.WasTransmitted()) {
						aplt_sum+=pp.GetPathLength();
						aplt_label.setText(Double.toString(aplt_sum/(iteration_count-num_reflected)));
						tau_label.setText(Double.toString(cloud.OpticalDepth()));
						try {
							flag=true;
							for(i=1; (i<20) && flag; i++) {
								if(pp.GetPathLength()<trans_distrib_data.GetElement(i).x) {
									trans_distrib_data.GetElement(i).y+=1.0;
									flag=false;
								}
							}
						}
						catch(DataSet.DataException e) {}
					} else {
						num_reflected++;
						aplr_sum+=pp.GetPathLength();
						aplr_label.setText(Double.toString(aplr_sum/num_reflected));
						albedo_label.setText(Double.toString(100.0*num_reflected/iteration_count));
						try {
							flag=true;
							for(i=1; (i<20) && flag; i++) {
								if(pp.GetPathLength()<ref_distrib_data.GetElement(i).x) {
									ref_distrib_data.GetElement(i).y+=1.0;
									flag=false;
								}
							}
						}
						catch(DataSet.DataException e) {}
					}
					if(iteration_count%10==0) {
						cloud.DrawCloud(cloud_panel);
						cloud.DrawPath(cloud_panel,pp.GetPath());
						distrib_plot.repaint();
						repaint();
					}
				}
				Thread.sleep(10);
			}
		}
		catch(InterruptedException e) {
			System.out.println("Thread Interrupted.");
		}
	}

	public boolean action(Event evt,Object what) {
		if(evt.target==ss_button) {
			if(running) {
				ClickStop();
			} else {
				ClickStart();
			}
			return(true);
		} else if(evt.target==cloud_choice) {
			if(running) ClickStop();
			switch(cloud_choice.getSelectedIndex()) {
			case 0:
				cloud=new Slab1D(10.0);
				break;
			case 1:
				cloud=new Slab2D(10.0);
				break;
			case 2:
				cloud=new Sphere2D(10.0);
				break;
			case 3:
				cloud=new Fractal2D(10.0);
				break;
			case 4:
				cloud=new Slab3D(10.0);
				break;
			case 5:
				cloud=new Sphere3D(10.0);
				break;
			case 6:
				cloud=new Fractal3D(10.0);
				break;
			}
			data_plot.RemoveAllData();
			data_plot.AddDataSet(cloud.GetData());
			data_plot.repaint();
		} else if(evt.target==scatter_choice) {
			if(running) ClickStop();
			switch(scatter_choice.getSelectedIndex()) {
			case 0:
				sm=new RandomScatter();
				break;
			case 1:
				break;
			case 2:
				break;
			}
		} 
		if((evt.target==aoi_choice) || (evt.target==cloud_choice)) {
			if(running) ClickStop();
			dir=new RTVector(cloud.Dimension());
			if(cloud.Dimension()==1) {
				dir.Set(0,-1.0);
			} else {
				switch(aoi_choice.getSelectedIndex()) {
				case 0:		// 0
					dir.Set(cloud.Dimension()-1,-1.0);
					break;
				case 1:		// 15
					dir.Set(cloud.Dimension()-1,-0.9659);
					dir.Set(0,0.2588);
					break;
				case 2:		// 30
					dir.Set(cloud.Dimension()-1,-0.8660);
					dir.Set(0,0.5);
					break;
				case 3:		// 45
					dir.Set(cloud.Dimension()-1,-0.7071);
					dir.Set(0,0.7071);
					break;
				case 4:		// 60
					dir.Set(cloud.Dimension()-1,-0.5);
					dir.Set(0,0.8660);
					break;
				case 5:		// 85
					dir.Set(cloud.Dimension()-1,-0.2588);
					dir.Set(0,0.9659);
					break;
				}
			}
		}
		repaint();
		return(false);
	}

	public void ClickStart() {
		running=true;
		ss_button.setLabel("Stop");
		if(thread==null) {
			thread=new Thread(this);
			thread.start();
		}
	}

	public void ClickStop() {
		int i;

		running=false;
		ss_button.setLabel("Start");
		if(thread!=null) {
			thread.stop();
			thread=null;
			iteration_count=0;
			aplt_sum=0.0;
			aplr_sum=0.0;
			num_reflected=0;
			try {
				for(i=1; i<20; i++) {
					trans_distrib_data.GetElement(i).y=0.0;
					ref_distrib_data.GetElement(i).y=0.0;
				}
			}
			catch(DataSet.DataException e) {}
		}
	}

	public void paint(Graphics g) {
	}

	public void update(Graphics g) {
		paint(g);
	}

	private Thread thread;
	private Cloud cloud;
	private ScatteringModel sm;
	private RTVector dir;
	private PhotonPath pp;

	private Panel main_panel;
	private Choice cloud_choice;
	private Choice scatter_choice;
	private Button ss_button;
	private boolean running=false;
	private Label tau_label;
	private Label aplt_label;	// Average path length through.
	private double aplt_sum;
	private Label aplr_label;	// Average path length reflected.
	private double aplr_sum;
	private Label albedo_label;
	private int num_reflected;
	private Label iteration_label;
	private int iteration_count;
	private Choice aoi_choice;

	private Panel cloud_panel;
	private AdjustablePlot2D data_plot;
	private Plot2D distrib_plot;
	private DataSet2D trans_distrib_data;
	private DataSet2D ref_distrib_data;
}