package robots;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.concurrent.Semaphore;

import lejos.pc.comm.NXTComm;
import lejos.pc.comm.NXTCommException;
import lejos.pc.comm.NXTCommFactory;
import lejos.pc.comm.NXTInfo;
import lejos.robotics.navigation.Pose;

public class BluetoothRobot implements Robot {
	private String name;
	private NXTComm nxtComm;
	private RobotReturn rr;
	private Semaphore semaphore;
	private Semaphore countsema;
	private int creads = 1;
	
	public static final byte FORWARD = 0;
	public static final byte STOP = 1;
	public static final byte EXIT = 2;
	public static final byte LEFT = 3;
	public static final byte RIGHT = 4;
	public static final byte BACKWARD = 5;
	
	public static final byte STOPSCANN = 6;
	public static final byte STARTSCANN = 7;
	public static final byte MOVE = 8;
	public static final byte ROTATE = 9;
	public static final byte SETPOSE = 10;
	public static final byte SETSCANANGLE = 11;
	
	private static final int scannangle = 5;

	private DataOutputStream output;
	private DataInputStream input;
	private Receiver receivethread;
	
	private ArrayList<DataPose> reads;
		
	private class Receiver extends Thread {
		public boolean run = true;
		@Override
		public void run() {
			int bytes_valiable = -1;
			
			while(run) {
				try {
					bytes_valiable = input.available();
				} catch (IOException e1) {
					// TODO Auto-generated catch block
					e1.printStackTrace();
				}
				if (bytes_valiable >= 0) {
					try {
						if (input.readByte() != '@') continue;
						int angle = input.readByte();
						float alpha = input.readFloat();
						float x = input.readFloat();
						int distance = input.readByte();
						float y = input.readFloat();
						DataPose d = new DataPose();
						d.setDistance(distance);
						d.setSensorAngle(angle);
						d.setPose(new Pose(x, y, alpha-90));
						if (rr != null)
							rr.robotData(d);	
						
						if (reads.size() <= creads) {
							reads.add(d);
						}
						
					} catch (IOException e1) {
						continue;
					}
				}
			}
		}
	}

	private void send(int command) {
		if (semaphore.tryAcquire()) {
			try {
				output.write(command);
				output.flush();
			} catch (IOException e) {
				System.out.println("Erro ao enviar...");
			}
			semaphore.release();
		}
	}

	private void send(int command, float f) {
		if (semaphore.tryAcquire()) {
			try {
				output.write(command);
				output.writeFloat(f);
				output.flush();
			} catch (IOException e) {
				System.out.println("Erro ao enviar...");
			}
			semaphore.release();
		}
	}
	
	private void send(int command, int i) {
		if (semaphore.tryAcquire()) {
			try {
				output.write(command);
				output.write(i);
				output.flush();
			} catch (IOException e) {
				System.out.println("Erro ao enviar...");
			}
			semaphore.release();
		}
	}
	
	private void send(Pose p) {
		if (semaphore.tryAcquire()) {
			try {
				output.write(SETPOSE);
				output.writeFloat(p.getX());
				output.writeFloat(p.getY());
				output.writeFloat(p.getHeading());
				output.flush();
			} catch (IOException e) {
				System.out.println("Erro ao enviar...");
			}
			semaphore.release();
		}
	}
	
	public BluetoothRobot (String name) {
		reads = new ArrayList<DataPose>();
		semaphore = new Semaphore(1);
		countsema = new Semaphore(1);
		receivethread = new Receiver();
		this.name = name;
	}

	@Override
	public void moveForward() {
		send(FORWARD);
	}

	@Override
	public void moveLeft() {
		send(LEFT);
	}

	@Override
	public void moveRight() {
		send(RIGHT);
	}

	@Override
	public void moveBackward() {
		send(BACKWARD);
	}

	@Override
	public boolean connect () {
		try {
			nxtComm = NXTCommFactory.createNXTComm(NXTCommFactory.BLUETOOTH);
			NXTInfo[] nxtInfo = nxtComm.search(name); //find brick with NXT_ID by doing a Bluetooth inquiry
			if (nxtInfo.length == 0) { // failed to find a brick with the ID
				System.err.println("NO NXT found");
				return false;
			}
			if (!nxtComm.open(nxtInfo[0])) { // the brick was found but a connection could not be establish
				System.err.println("Failed to open NXT");
				return false;
			}
			
			input = new DataInputStream(nxtComm.getInputStream()); // open data input stream 
			output = new DataOutputStream(nxtComm.getOutputStream()); // open data output stream
			send(SETSCANANGLE, scannangle); // vai scanear em 5 em 5 graus
		} catch (NXTCommException e) {
			return false;
		}
		
		receivethread.start();
		
		return true;	
	}

	@Override
	public void stop() {
		send(STOP);
	}


	@Override
	public void move(double x) {
		send(MOVE, (float)x);
	}

	@Override
	public void rotate(double x) {
		send(ROTATE, (float)x);
	}

	@Override
	public ArrayList<DataPose> scann(int ini, int end, int interval) {
		send(SETSCANANGLE, interval);
		
		creads = 180/interval;
		reads.clear();
		send(SETSCANANGLE, scannangle);
		
		while(reads.size() <= creads) { };
		
		stopScann();
		countsema.release(180);
		creads = -1;
		return null;
	}

	@Override
	public void scann(RobotReturn r) {
		rr = r;
		send(STARTSCANN);
	}

	@Override
	public void stopScann() {
		send(STOPSCANN);
	}

	@Override
	public void disconnect() {
		send(EXIT);
		if (receivethread == null) return;
		receivethread.run = false;
		try {
			receivethread.join();
		} catch (InterruptedException e1) {
			System.out.println("Nao foi possivel finalizar as threads...");
		}
		try {
			nxtComm.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
	}

	@Override
	public void setPose(float x, float y, float a) {
		send(new Pose(x, y, a));
	}
	
	
	@Override
	public String toString() {
		return "Bluetooth Mestre/Escravo";
	}

}
