package wifibroadcast;

import java.io.IOException;
import java.nio.ByteBuffer;

public class WifiBroadcastTest {
	private static int DefaultPort = 8412;
	private static String MCASTADDR = "224.0.0.1";
	private static int testSize = 256;
	private static long timeout = 1000;


	int port = DefaultPort ; 
	int count = 16;
	boolean multicast = false;
	boolean mchannel = false;
	int receiver_count = 1;
	int sender_count = 1;
	WifiReceiver wbr[] = new WifiReceiver[receiver_count];
	WifiReceiver wbs[] = new WifiReceiver[sender_count];

	public static void main(String args[]) {
		new WifiBroadcastTest().test(args);
	}

	public void test(String args[]) {
		options(args);
		try {
			if (multicast) {
				for(int i=0;i<wbr.length;i++) wbr[i] = new WifiMulticast(i,MCASTADDR,port,WifiReceiver.SocketType.Receiver);
				Thread.sleep(timeout/4);
				for(int i=0;i<wbs.length;i++) wbs[i] = new WifiMulticast(i,MCASTADDR,port,WifiReceiver.SocketType.Sender);
			} else if (mchannel) {
				for(int i=0;i<wbr.length;i++) wbr[i] = new WifiMulticastChannel(i,MCASTADDR,port,WifiReceiver.SocketType.Receiver);
				Thread.sleep(timeout/4);
				for(int i=0;i<wbs.length;i++) wbs[i] = new WifiMulticastChannel(i,MCASTADDR,port,WifiReceiver.SocketType.Sender);
			} else {
				for(int i=0;i<wbr.length;i++) wbr[i] = new WifiBroadcast(i,port,WifiReceiver.SocketType.Receiver);
				Thread.sleep(timeout/4);
				for(int i=0;i<wbs.length;i++) wbs[i] = new WifiBroadcast(i,port,WifiReceiver.SocketType.Sender);
			}
		} catch (IOException e) {
			System.err.println("err "+e);
		} catch (InterruptedException e) {
			System.err.println("err "+e);
		}
		runTest(count, wbr, wbs);

	}

	public void options(String[] args) {
		for(int i=0;i<args.length;i++) {
			if (args[i].equals("-m")) multicast = true;
			else if (args[i].equals("-channel")) mchannel = true;
			else if (args[i].equals("-r"))  { i++; receiver_count = getOptInt(args, count, i); wbr = new WifiReceiver[receiver_count]; }
			else if (args[i].equals("-s"))  { i++; sender_count = getOptInt(args, count, i); wbs = new WifiReceiver[sender_count]; }
			else if (args[i].equals("-c")) { i++; count = getOptInt(args, count, i);}
			else if (args[i].equals("-t")) { i++; timeout = getOptInt(args, count, i);}
			else if (args[i].equals("-T")) { i++; testSize = getOptInt(args, count, i);}
			else if (args[i].equals("-p")) { i++; port = getOptInt(args, count, i);}
		}
	}



	public void runTest(int count, WifiReceiver[] wbr, WifiReceiver wbs[]) {
		Thread t = timeouter();
		Thread send[]  = new Thread[wbs.length];
		int id = 0;
		for(WifiReceiver s:wbs) {
			send[id]= sender(s,count);
			id ++;
		}
		id = 0;
		Thread recv[]  = new Thread[wbr.length];
		for(WifiReceiver r: wbr) {
			recv[id] = receiver(r,id, count, timeout);
			id ++;
		}
		try {
			for(int i = 0;i<wbr.length;i++) {
				recv[i].join();
			}			
			for(int i = 0;i<wbs.length;i++) {
				send[i].join();
			}
			t.join();
		} catch (InterruptedException e) {
		}
	}



	public int getOptInt(String[] args, int count, int i) {
		if (i<args.length) { count = Integer.parseInt(args[i]); }
		return count;
	}



	public Thread sender(final WifiReceiver wbs, final int count) {
		Runnable sender = new Runnable() {

			@Override
			public void run() {
				ByteBuffer testData = getTestData(testSize);
				int i = 0;
				try {
					Thread.sleep(timeout);
					for(i = 0; i<count;i++) {
						testData.putInt(0, i);
						wbs.send(testData);
						testData.flip();
					}
				} catch (Exception e) {
					System.err.println("sender error at "+i+" "+e);
				}
		}
		};
		Thread s = new Thread(sender);
		s.start();
		return s;
	}

	private  boolean running;

	public Thread timeouter() {
		running = true;
		Runnable timeouter = new Runnable() {
			@Override
			public void run() {
				try {
					Thread.sleep(30*1000);
				} catch (InterruptedException e) {
				}
				running = false;
			}
		};
		Thread t = new Thread(timeouter); t.start();
		return t;
	}
	
	public Thread receiver(final WifiReceiver wbr, final int id, final int count,  final long timeout) {

		Runnable receiver = new Runnable() {
			@Override
			public void run() {
				ByteBuffer testData = ByteBuffer.allocate(4096);
				int bad = 0, good = 0;
				try {
					for(int i = 0; running &&  i<count;i++) {
						testData.clear();
						wbr.recieve(testData,timeout);
						if (!testData.hasRemaining()) continue;
						System.out.println("receive id"+id+":"+testData.remaining()+" bytes.");
						int seq = testData.getInt();
						if (seq!=i) {
							bad++; i = seq;
						} else {
							good++;
						}
					}
				} catch (Exception e) {
					System.err.println("receiver error "+e);
				}
				System.out.println("get "+good+" packets, "+bad+" losts.");
		}
		};
		Thread r = new Thread(receiver); r.start();
		return r;
	}
	
	public ByteBuffer getTestData(int i) {
		ByteBuffer b = ByteBuffer.allocate(i);
		b.putInt(0);
		for(int j = 0; j<256; j++ ) {
			b.put((byte)j);
			if (! b.hasRemaining()) break;
			if (j == 255) j=0;
		}
		b.flip();
		return b;
	}
}