/* RBTreeAlgThread.java */

import java.awt.*;
import java.io.*;
import java.net.*;
import java.util.*;

public class RBTreeAlgThread extends AlgThread {
	
	RBTree tree;
	IntList int_list;
	static String[] RBTreeDataSets = { "Graph 1", "Graph 2", "Graph 3"};
	private static String base_data_file_name = "graph.rb";

	
	public RBTreeAlgThread( AlgAnimFrame frame ) {
		super( frame );
		}

	private boolean loadData( int choice ) {
		String fn;
		DataInputStream inStream;
		String line;

		System.out.println("loadData: choice " + choice );
		//fn = frame.algfile+choice;
		// Can I do this?
		fn = base_data_file_name + choice;
		System.out.println("loadData: filename "+fn);
		try {
			URL dataURL = new URL(frame.getApplet().getCodeBase(), fn);
			int_list = new IntList( dataURL );
			} 
		catch (IOException e) {
			System.out.println("Data file or source file not found");
			} 
		catch (NullPointerException e) {}

		return int_list != null;
		}

	/** Overrides generateData of super class */
	public void generateData() {
		//int choice = frame.control_panel.getDataChoice();
		//added
		this.dpAfter = frame.getDrawingPanel();
		dpAfter.init();

		int choice = frame.getDataChoice();

		if ( loadData( choice ) ) {
			System.out.println("Data loaded OK");
			}
		else {
			System.out.println("Data loading error");
			}
		}

	// public void init( String fn ) {
	//	System.out.println("init: enter - fn " + fn);
	//		}

	public void run() {
		int k;
		Graphics g1, g2, g3;
		//DrawingPanel dp;
		Image beforeOffSc = null;
		Graphics befOffGraphics = null;

		System.out.println("run: enter");
		dpAfter = frame.getDrawingPanel();
		dpBefore = frame.getBeforeDp();
		if ( dpAfter == null ) {
			System.out.println( "run: null drawingPanel");
			return;
			}
		tree = new RBTree();
		//tree.setPanel( dpAfter );

		//Initialise beforeOffsc
		tree.initBeforeOffsc(frame, tree);

		generateData();
		if ( int_list != null ) {
			for(k=0;k<int_list.size();k++) {
				// Get the next one to insert
				int data = int_list.elementAt( k );

				if(k!=0) {
					ShadowLabel bforeLabel = new ShadowLabel("Before New Insertion");
					bforeLabel.move(50, (dpBefore.getPanelHeight() - y_offset));
					bforeLabel.draw(befOffGraphics);
					g2 = dpBefore.getGraphics();
					g2.drawImage(beforeOffSc, 0, 0, null);		
					}	
					
			
				// Insert into the tree
				tree.InsertNode( data, frame, tree );
				g1 = dpAfter.getGraphics();
				tree.drawRBTree(frame);
				if (tree.GetOffscreen() == null) {
					System.out.println("AlgThread: Run: offscreen is null");
					return;
					}
				else {
					g1.drawImage(tree.GetOffscreen(), 0, 0, null);
					beforeOffSc = tree.GetOffscreen();
					befOffGraphics = tree.GetOffGraphics();
					frame.waitSkip();
					}
				}

			ShadowLabel finalLabel = new ShadowLabel("Final Red-Black Tree");
			finalLabel.move(x_offset, (dpBefore.getPanelHeight() - y_offset));
			finalLabel.draw(befOffGraphics);
			g2 = dpBefore.getGraphics();
			g2.drawImage(beforeOffSc, 0, 0, null);

			// finish creating tree 
			frame.finishAlg();
			dpAfter.repaint();
			
			//g2.drawImage(beforeOffSc, 0, 0, null);
			}
		else {
			System.out.println("Data file not found\n");
			}
		}

	}