import java.io.File;  																													//Library for handling files
import java.io.FileWriter;																											//Library for writing to files
import java.io.IOException;																											//Library for handling exceptions
import java.io.FileOutputStream;																								//Library for handling file output streams, in order to create images
import java.util.List;																													//Library for Lists
import java.util.ArrayList; 																										//Library for ArrayLists

public class userApp {

	//Request codes that change in each 2-hour session
	public static String echoRequestCode =  "E4246";
	public static String imageRequestCode = "M7435";
	public static String imageRequestCodeError = "G8606";
	public static String gpsRequestCode = "P2698";
	public static String ackCode = "Q7959";
	public static String nackCode = "R4816";

	public static void main(String[] param) {
		(new userApp()).demo();
	}

	//Initializes the modem
	public Modem initModem(int speed, int timeout){
		Modem modem;
		modem=new Modem();
		modem.setSpeed(speed);
		modem.setTimeout(timeout);
		modem.open("ithaki");
		return modem;
	}

	//Receives what Ithaki sends without sending a request code. Prints the packet on the terminal if printRx is true. Packet comes as a string.
	public String listen(Modem modem, String delimiter, boolean printRx){
		int k;
		String rxmessage ="";
		System.out.println("Sent request for message.");

		//Endless loop for listening for message
		for (;;) {
			try {
				//Read a byte from the modem
				k=modem.read();
				if (k==-1){
					System.out.println("Connection closed.");
					break;
				}
				//System.out.println((char)k);
				//System.out.print((char)k+" <"+k+"> ");
				rxmessage = rxmessage + (char)k;
				//System.out.println(rxmessage);
				//Break endless loop by catching break sequence.
				if (rxmessage.indexOf(delimiter)>-1){
					if(printRx==true){System.out.println(rxmessage);}
					System.out.println("End of listen message.");
					break;
				}
			} catch (Exception x) {
				break;
			}
		}
		return rxmessage;
	}

	//Sends a request code and receives the answer. Prints the packet on the terminal if printRx is true.
	public String sendAndListen(Modem modem, String txmessage, String delimiter, boolean printRx){
		int k;
		String rxmessage = "";
		//We create a different variable for the message we will send to Ithaki. The reason for differentiating is because for some reason the terminal couldn't print the
		//txmessage if we sent it along with the txmessage.
		String txmessageToSend = txmessage + "\r";
		modem.write(txmessageToSend.getBytes());
		String toPrint = "Sent " + txmessage + " request code.";
		System.out.println(toPrint);
		rxmessage = listen(modem, delimiter, printRx);
		return rxmessage;
	}

	//Creates a file.
	public void createFile(String fileName){
		try {
			File myObj = new File(fileName);
      if (myObj.createNewFile()) {
      	System.out.println("File created: " + myObj.getName());
      } else {
        System.out.println("File already exists.");
      }
    } catch (IOException e) {
      System.out.println("An error occurred.");
      e.printStackTrace();
    }
	}

	//Writes a string to a file.
	public void writeToFile(String fileName, String toWrite){
		try {
			//The true argument is needed in order to append to the file and not overwrite it.
			FileWriter myWriter = new FileWriter(fileName, true);
			myWriter.write(toWrite);
			myWriter.close();
		} catch (IOException e) {
			System.out.println("An error occurred.");
			e.printStackTrace();
		}
	}

	//Appends a string to a file.
	public void appendToFile(String fileName, String toWrite){
		try{
			FileWriter myWriter = new FileWriter(fileName, true);
			myWriter.append(toWrite);
			myWriter.flush();
			myWriter.close();
		} catch(IOException e){
			System.out.println("An error occured.");
			e.printStackTrace();
		}
	}

	//Receives a lot of echo packets and calculates each packet's response time in ms. Writes the results in a txt and a csv file.
	public void makeEchoPacketsList(Modem modem, int packetsNum){
		long time1;
		long timePassed;
		createFile("echopackets.txt");
		writeToFile("echopackets.txt", "Echo packets received: \n\n");
		createFile("echopackets.csv");
		appendToFile("echopackets.csv", "Response time (ms)");
		appendToFile("echopackets.csv", "\n");

		for(int i=0; i<packetsNum; i++){
			time1 = System.currentTimeMillis();
			//Echo packets don't have an \r character after PSTOP for some reason.
			String rxmessage = sendAndListen(modem, echoRequestCode, "PSTOP", false);
			timePassed = System.currentTimeMillis() - time1;
			writeToFile("echopackets.txt", rxmessage+"\t"+timePassed+" ms\r\n");
			appendToFile("echopackets.csv", String.valueOf(timePassed));
			appendToFile("echopackets.csv", "\n");
		}
	}

	//Receives an image and stores it into a jpeg file. Also stores its info in a txt file for checking purposes (commented out).
	public void receiveImage(Modem modem, String requestCode){
		//The byteList is used for writing into jpeg file. The intList is used for writing into txt file (optional).
		ArrayList<Integer> intList = new ArrayList<Integer>();
		ArrayList<Byte> byteList = new ArrayList<Byte>();
		int k;
		boolean foundEndDelimiter = false;
		int lastValue = 0;

		String jpegFileName = "";
		//String txtFileName = "";
		if(requestCode==imageRequestCode){
			jpegFileName = "imageNoError.jpeg";
			//txtFileName = "imageNoError.txt";
		}
		else if(requestCode==imageRequestCodeError){
			jpegFileName = "imageWithError.jpeg";
			//txtFileName = "imageWithError.txt";
		}
		else{
			jpegFileName = "gpsImage.jpeg";
			//txtFileName = "gpsImage.txt";
		}

		String txmessageToSend = requestCode + "\r";
		modem.write(txmessageToSend.getBytes());
		System.out.println("Sent request for image.");
		String toPrint = "Sent " + requestCode + " request code.";
		System.out.println(toPrint);

		for(;;){
			try{
				k = modem.read();
				if (k==-1){
					System.out.println("Connection closed.");
					break;
				}
				intList.add(k);
				byteList.add((byte)k);
				if(lastValue==255 && k==217){foundEndDelimiter=true;}
				if(foundEndDelimiter){
					System.out.println("End of listen message.");
					break;
				}
				else{
					lastValue = k;
				}
			} catch(Exception x){
				break;
			}
		}

		//Now intList stores the image in an ArrayList<Integer> form. We will now write that into a txt file for checking (optional).
		// createFile(txtFileName);
		// writeToFile(txtFileName, "Int array of image:/n/n");
		// for(int i=0; i<intList.size();i++){
		// 	writeToFile(txtFileName, intList.get(i)+"/r/n");
		// }

		//We will write the int array into a jpeg file using FileOutputStream.

		//Create the jpeg file.
		createFile(jpegFileName);

		//Convert byteList to byteArr
		byte byteArr[] = new byte[byteList.size()];
		for(int i=0; i<byteList.size();i++){
			byteArr[i] = byteList.get(i);
		}

		//Use FileOutputStream to copy byteArr into jpeg file.
		try{
			FileOutputStream fos = new FileOutputStream(jpegFileName);
			fos.write(byteArr);
			fos.close();
		} catch(Exception e){
			System.out.println(e);
		}
	}

	//Sends GPS + R=XPPPPLL request code and receives GPS traces in strings.
	public void receiveGPStraces(Modem modem, String rCode){
		createFile("gpstraces.txt");
		writeToFile("gpstraces.txt", "GPS traces received: \n\n");
		String rxmessage = sendAndListen(modem, gpsRequestCode+rCode, "STOP ITHAKI GPS TRACKING\r\n", false);
		writeToFile("gpstraces.txt", rxmessage);
	}

	//Gets a string containing all the GPS packets (lines) and breaks it up into each separate packet.
	public ArrayList<String> getGPSlines(String rxmessage){
		ArrayList<String> result = new ArrayList<String>();
		//temp string will store each packet every time.
		String temp = "";
		char c = ' ';
		int i = 0;
		while(temp != "STOP ITHAKI GPS TRACKING"){
			c = rxmessage.charAt(i);
			while(c != '\n'){
				temp += c;
				i++;
				c = rxmessage.charAt(i);
			}
			//When line seperator is found, empty temp in order to store the next packet.
			if(c == '\n'){
				result.add(temp);
				temp = "";
				i++;
			}
			if(i>=rxmessage.length()){
				break;
			}
		}
		return result;
	}

	//Parses GPS packet (single line). Produces an int array containing time, latitude and longitude.
	public int[] parseGPSpacket(String gpsPacket){
		String strResult = "";
		int result[] = {0, 0, 0, 0, 0, 0, 0};
		//index1 is where the time starts, index2 is where the latitude starts, index3 is where longitude starts.
		int index1 = 0;
		int index2 = 0;
		int index3 = 0;
		//numOfCommas will store the number of commas we've found so far.
		int numOfCommasFound = 0;

		int i = 0;
		char c = gpsPacket.charAt(i);
		while(c!= '\n'){
			if(c == ','){
				switch(numOfCommasFound){
					//If we haven't found any commas yet, then this comma is the first and denotes the start of the time information.
					case 0:
						index1 = i+1;
						break;
					//If we've only found one comma, then this comma is the second and denotes the start of the latitude information.
					case 1:
						index2 = i+1;
						break;
					//If we've found two commas, then this comma is the third but nothing starts here, so we skip it.
					case 2:
						break;
					//If we've found three commas, then this comma is the fourth and denotes the start of the longitude information.
					//However there's always a 0 in the start, so we skip that too.
					case 3:
						index3 = i+2;
						break;
				}
				numOfCommasFound++;
			}
			i++;
			if(i>=gpsPacket.length()){
				break;
			}
			c = gpsPacket.charAt(i);
		}

		//Now that we've found the indexes of the information we need, we fill the result array with that info.
		//This could be done more neatly.
		String temp = "";
		temp += gpsPacket.charAt(index1);
		temp += gpsPacket.charAt(index1+1);
		result[0] = Integer.valueOf(temp);

		temp = "";
		temp += gpsPacket.charAt(index1+2);
		temp += gpsPacket.charAt(index1+3);
		result[1] = Integer.valueOf(temp);

		temp = "";
		temp += gpsPacket.charAt(index1+4);
		temp += gpsPacket.charAt(index1+5);
		result[2] = Integer.valueOf(temp);

		temp = "";
		temp += gpsPacket.charAt(index2);
		temp += gpsPacket.charAt(index2+1);
		temp += gpsPacket.charAt(index2+2);
		temp += gpsPacket.charAt(index2+3);
		result[3] = Integer.valueOf(temp);

		temp = "";
		temp += gpsPacket.charAt(index2+5);
		temp += gpsPacket.charAt(index2+6);
		temp += gpsPacket.charAt(index2+7);
		temp += gpsPacket.charAt(index2+8);
		result[4] = Integer.valueOf(temp);

		temp = "";
		temp += gpsPacket.charAt(index3);
		temp += gpsPacket.charAt(index3+1);
		temp += gpsPacket.charAt(index3+2);
		temp += gpsPacket.charAt(index3+3);
		result[5] = Integer.valueOf(temp);

		temp = "";
		temp += gpsPacket.charAt(index3+5);
		temp += gpsPacket.charAt(index3+6);
		temp += gpsPacket.charAt(index3+7);
		temp += gpsPacket.charAt(index3+8);
		result[6] = Integer.valueOf(temp);

		return result;
	}

	//Creates a T=... code out of a gpsPacket string.
	public String createTcode(String gpsPacket){
		int[] intArr = parseGPSpacket(gpsPacket);
		String t1 = String.valueOf(intArr[3]);
		t1 += String.valueOf(Math.round(0.006*intArr[4]));

		String t2 = String.valueOf(intArr[5]);
		t2 += String.valueOf(Math.round(0.006*intArr[6]));

		return "T="+t2+t1;
	}

	//Creates a GPS image with traces we get.
	public void createGPSimage(Modem modem, String rCode, int numTraces){
		String rxmessage = sendAndListen(modem, gpsRequestCode+rCode, "STOP ITHAKI GPS TRACKING\r\n", false);
		ArrayList<String> strArray = getGPSlines(rxmessage);

		//Get the number of packets we want out of R code.
		String temp = "";
		temp += rCode.charAt(7);
		temp += rCode.charAt(8);
		int numOfPackets = Integer.valueOf(temp);

		//Ask for at least 13 packets in order to be able to find at least 4 that are at least 4 seconds apart from each other.
		//This was found empirically.
		if(numOfPackets < 13){
			System.out.println("Please request at least 13 GPS packets in order for them to have at least 4 seconds difference.");
			return;
		}

		//We pick some of the GPS packets. We try to split the numOfPackets we have in almost equal parts, in order for them to be as much away from each other (in seconds) as possible.
		ArrayList<String> gpsPackets = new ArrayList<String>();
		gpsPackets.add(strArray.get(1));																						//First GPS packet is always the first one (stored in line 1, since line 0 has start delimiter).
		for(int i=1; i<numTraces-1; i++){																						//First packet is already stored, so we begin from 1.
			gpsPackets.add(strArray.get(Math.round(i*numOfPackets/(numTraces-1))));
		}
		gpsPackets.add(strArray.get(numOfPackets-1));																	//Last GPS packet is always the last one.

		String finalCode = gpsRequestCode;

		for(int i=0; i<gpsPackets.size(); i++){
			finalCode += createTcode(gpsPackets.get(i));
		}

		receiveImage(modem, finalCode);
	}

	//Takes arqPacket string as argument and finds its FCS by parsing the string.
	public int fcs(String arqPacket){
		String fcsString = "";
		int startIndex = 0;
		for (int i=0; i<arqPacket.length(); i++){
			if(arqPacket.charAt(i) == '>'){
				startIndex = i+2;
				break;
			}
		}
		for(int i=0; i<3; i++){
			fcsString += arqPacket.charAt(startIndex + i);
		}
		int fcs;
		fcs = Integer.parseInt(fcsString);
		return fcs;
	}

	//Takes arqPacket string and finds the XOR result of its characters.
	public int xorResult(String arqPacket){
		String xSeq = "";
		int startIndex = 0;

		//Find the beginning of the X sequence.
		for (int i=0; i<arqPacket.length(); i++){
			if(arqPacket.charAt(i) == '<'){
				startIndex = i+1;
				break;
			}
		}

		//Save the X sequence into xSeq string.
		int j = startIndex;
		while(arqPacket.charAt(j) != '>'){
			xSeq += arqPacket.charAt(j);
			j++;
		}

		//Make each character in xSeq a number (ASCII).
		ArrayList<Integer> xArr = new ArrayList<Integer>();
		for (int i=0; i<xSeq.length(); i++){
			xArr.add((int)(xSeq.charAt(i)));
		}

		//XOR all the X characters and save result into p.
		int p = xArr.get(0) ^ xArr.get(1);
		for(int i=2; i<xSeq.length(); i++){
			p = p ^ xArr.get(i);
		}
		return p;
	}

	//Implements an ARQ mechanism, for duration milliseconds (taken as an argument). Calculates the number of ACK and NACK packets we sent, their response times and the times of repetitions.
	//Stores the results in a txt and a csv file.
	public void arqMechanism(Modem modem, long duration){
		createFile("arqpackets.txt");
		writeToFile("arqpackets.txt", "ARQ packets received: \n\n");

		createFile("ARQrepetitions.csv");
		appendToFile("ARQrepetitions.csv", "Repetitions");
		appendToFile("ARQrepetitions.csv", ", ");
		appendToFile("ARQrepetitions.csv", "Number of times");
		appendToFile("ARQrepetitions.csv", "\n");

		createFile("ARQresponseTimes.csv");
		appendToFile("ARQresponseTimes.csv", "Response times (ACK)");
		appendToFile("ARQresponseTimes.csv", "\n");

		int ackNum = 1;
		int nackNum = 0;
		String nextCode;
		long startTime = System.currentTimeMillis();
		long responseTime;
		//Send an ACK to begin.
		String rxmessage = sendAndListen(modem, ackCode, "PSTOP", false);
		long timePassed = System.currentTimeMillis() - startTime;
		responseTime = System.currentTimeMillis() - startTime;
		long lastTimeACK = startTime;
		writeToFile("arqpackets.txt", rxmessage+"\n");

		int repetitions = 0;
		//repTimes[0] stores how many times we had to repeat 0 times in order to get a correct packet and so on.
		int repTimes[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};

		//Keep sending and receiving as long as time passed < duration.
		do{
			if(fcs(rxmessage) == xorResult(rxmessage)){
				nextCode = ackCode;
				ackNum++;
				repTimes[repetitions] += 1;
				repetitions = 0;
				responseTime = System.currentTimeMillis() - lastTimeACK;
				lastTimeACK = System.currentTimeMillis();
				appendToFile("ARQresponseTimes.csv", String.valueOf(responseTime));
				appendToFile("ARQresponseTimes.csv", "\n");
			}
			else{
				nextCode = nackCode;
				nackNum++;
				repetitions++;
			}
			rxmessage = sendAndListen(modem, nextCode, "PSTOP", false);
			timePassed = System.currentTimeMillis() - startTime;
			writeToFile("arqpackets.txt", rxmessage+"  "+nextCode+"\n");
		}while(timePassed <= duration);

		//Write the results neatly in a txt file for checking purposes.
		writeToFile("arqpackets.txt", "Number of ACK packets: "+ackNum+"\n");
		writeToFile("arqpackets.txt", "Number of NACK packets: "+nackNum+"\n");
		for(int i=0; i<11; i++){
			writeToFile("arqpackets.txt", String.valueOf(i)+" repetitions: "+String.valueOf(repTimes[i])+"\n");
			appendToFile("ARQrepetitions.csv", String.valueOf(i));
			appendToFile("ARQrepetitions.csv", ", ");
			appendToFile("ARQrepetitions.csv", String.valueOf(repTimes[i]));
			appendToFile("ARQrepetitions.csv", "\n");
		}
	}

	public void demo() {
		//Initialize modem.
		Modem modem = initModem(80000, 2000);																				//For text, speed=1000. For images, speed=80000.

		//Listen for welcome message from Ithaki.
		String welcomeMessage = listen(modem, "\r\n\n\n", true);

		//Send a test message and listen for answer.
		String testMessage = sendAndListen(modem, "test", "PSTOP\r\n", true);
		
		//Comment out the section you want according to each question.

		//-------------------------question (i)-------------------------
		//Send echoRequestCodes and listen for answers.
		//Set speed to 1000 bps. To run for at least 4 minutes, the number of packets must be around 900.
		//makeEchoPacketsList(modem, 900);

		//-------------------------question (ii)-------------------------
		//Receive an image with no error and one with error. Set speed to 80000.
		//receiveImage(modem, imageRequestCode);
		//receiveImage(modem, imageRequestCodeError);

		//-------------------------question (iii)-------------------------
		//Receive GPS track packets and create image out of them. Set speed to 80000.
		//String rCode = "R=1000190";
		//createGPSimage(modem, rCode, 5);

		//-------------------------question (iv)-------------------------
		//Implement ARQ mechanism. Set speed to 1000.
		//arqMechanism(modem, 250000);

		//Close modem
		modem.close();
	}
}