import drcl.comp.*;
import drcl.inet.*;
import drcl.inet.data.RTKey;
import drcl.inet.data.RTEntry;
import drcl.inet.transport.TCP;
import drcl.inet.transport.TCPSink;
import drcl.inet.core.PktDispatcher;

public class OneBottleneck implements Runnable
{
	static boolean EXIT = true;
	static Component root_ = null;
	static int nc_ = 0;
	static double simulationTime = 0.0;
	static double interval = 100.0;

	public static void main(String[] args)
	{
		if (args == null || args.length < 2) {
			System.out.println("Usage: java OneBottleneck <#connections> "
				+ "<simultion_time> ?interval<nn>? ?event? ?radix?");
			return;
		}

		nc_ = Integer.parseInt(args[0]);
		simulationTime = Double.valueOf(args[1]).doubleValue();

		boolean event_ = false;
		boolean radix_ = false;
		boolean linkemu_ = true;
		if (args.length > 2) {
			for (int i=2; i<args.length; i++)
				if (args[i].equals("event")) event_ = true;
				else if (args[i].equals("radix")) radix_ = true;
				else if (args[i].startsWith("interval")) interval = Integer.parseInt(args[i].substring(8));
		}

		System.out.println("Constructing a network of " + (2*nc_+2) + " nodes with " + nc_ + " TCP connections");
		if (event_) System.out.println("-- Event simulation --");
		if (radix_) {
			System.out.println("-- Radix tree implementation --");
			drcl.inet.core.RT.IMPLEMENTATION = drcl.inet.core.RT.RADIX_TREE;
		}
		if (linkemu_) System.out.println("-- Link emulation --");

		long time_ = System.currentTimeMillis();

		Node[] senders_ = new Node[nc_];
		Node[] receivers_ = new Node[nc_];
		Node[] routers_ = null;

		root_ = new Component("test");
		Component.Root.addComponent(root_);

		// senders and router0 take address space (0, addrDivider-1)
		// receivers and router1 take address space (addrDivider, 2*addrDivider-1)
		long addrDivider = 1;
		while (addrDivider < (nc_+1)) addrDivider<<=1;

		System.out.println("address divider = " + addrDivider);

		try {

		System.out.print("Create topology...");

		// adjacency matrix
		int[][] adjMatrix_ = new int[2*nc_ + 2][];
		for (int i=0; i<nc_; i++)
			adjMatrix_[i+2] = new int[]{0};
		for (int i=0; i<nc_; i++)
			adjMatrix_[i+nc_+2] = new int[]{1};
		adjMatrix_[0] = new int[nc_+1];
		adjMatrix_[1] = new int[nc_+1];
		adjMatrix_[0][0] = 1;
		adjMatrix_[1][0] = 0;
		for (int i=0; i<nc_; i++) {
			adjMatrix_[0][i+1] = i+2;
			adjMatrix_[1][i+1] = i+nc_+2;
		}

		Link link_ = null;
		if (!linkemu_) {
			link_ = new Link();
			link_.setPropDelay(0.01);
		}
		InetUtil.createTopology(root_, adjMatrix_, link_, false);
		routers_ = new Node[]{(Node)root_.getComponent("n0"), (Node)root_.getComponent("n1")};
		for (int i=0; i<nc_; i++) {
			senders_[i] = (Node)root_.getComponent("h" + (i+2));
			senders_[i].addAddress(i);
			receivers_[i] = (Node)root_.getComponent("h" + (i+nc_+2));
			receivers_[i].addAddress(i + addrDivider);
		}
		routers_[0].addAddress(nc_);
		routers_[1].addAddress(addrDivider + nc_);

		System.out.print("\n\nBuilding...");

		NodeBuilder nb_ = new NodeBuilder(".nb");
		nb_.setBandwidth(10000000); // 10Mbps
		if (linkemu_) {
			nb_.setLinkEmulationEnabled(true);
			nb_.setLinkPropDelay(0.01);
		}
		nb_.build(routers_);
		nb_.build(senders_, "tcp drcl.inet.transport.TCP\nsource  -/tcp drcl.inet.application.BulkSource");
		nb_.build(receivers_, "tcpsink drcl.inet.transport.TCPSink\nsink  -/tcpsink drcl.inet.application.BulkSink");
		routers_[0].setBufferSize(0, 4096 * nc_); // 4KB per connection
		routers_[0].setBandwidth(0, 1500000); // 1.5Mbps
		routers_[1].setBandwidth(0, 1500000); // 1.5Mbps
		if (linkemu_)
			((CoreServiceLayer)routers_[0].getComponent("csl")).setLinkPropDelay(0, 0.1);
		else
			((Link)routers_[0].getPort("0").getPeers()[0].getHost()).setPropDelay(0.1);

		System.out.print("\n\nSetup routes...");

		// router0
		CoreServiceLayer csl_ = (CoreServiceLayer)routers_[0].getComponent("csl");
		csl_.addRTEntry(new RTKey(0L, 0L, addrDivider, -addrDivider, 0, 0),
			new RTEntry(new drcl.data.BitSet(new int[]{0})), Double.NaN);
		for (int i=0; i<nc_; i++)
			csl_.addRTEntry(new RTKey(0L, 0L, i, -1L, 0, 0),
				new RTEntry(new drcl.data.BitSet(new int[]{i+1})), Double.NaN);

		// router1
		csl_ = (CoreServiceLayer)routers_[1].getComponent("csl");
		csl_.addRTEntry(new RTKey(0L, 0L, 0L, -addrDivider, 0, 0),
			new RTEntry(new drcl.data.BitSet(new int[]{0})), Double.NaN);
		for (int i=0; i<nc_; i++)
			csl_.addRTEntry(new RTKey(0L, 0L, i+addrDivider, -1L, 0, 0),
				new RTEntry(new drcl.data.BitSet(new int[]{i+1})), Double.NaN);

		System.out.print("\n\nSetup TCP...");

		for (int i=0; i<nc_; i++) {
			TCP tcp_ = (TCP)senders_[i].getComponent("tcp");
			tcp_.setPeer(i+addrDivider);
			tcp_.setMSS(950);
			((TCPSink)receivers_[i].getComponent("tcpsink")).setMSS(950);
		}

		}
		catch (Exception e_) {
			e_.printStackTrace();
			System.out.println("\n" + e_);
		}
		catch (OutOfMemoryError e_) {
			System.out.println("\n" + e_);
		}

		System.out.println("Time for building scenario = " + ((System.currentTimeMillis() - time_) / 1000.0));

		System.out.println("Start simulation");
		drcl.comp.ACARuntime sim_ = null;
		if (event_)
			sim_ = new drcl.sim.event.SESimulator();
		else {
			sim_ = new drcl.sim.process.SMMTSimulator();
			((ARuntime)sim_).setMaxWorkforce(1);
		}
		sim_.takeover(root_);
		sim_.stop();
		sim_.addRunnable(.001, new OneBottleneck(sim_, time_));
		Util.operate(senders_, "start");
		sim_.resume();
	}

	OneBottleneck ()
	{}

	drcl.comp.ACARuntime sim;
	long starttime;

	OneBottleneck (drcl.comp.ACARuntime sim_, long starttime_)
	{
		sim = sim_;
		starttime = starttime_;
	}

	public void run()
	{
		double now_ = sim.getTime();
		if (now_ < simulationTime) {
			if (now_ < 10.0)
				sim.addRunnableAt(10.0, this);
			else if (now_ < interval)
				sim.addRunnableAt(interval, this);
			else
				sim.addRunnable(interval, this);
			System.out.println("     " + sim.getTime() + "\t" + (sim.getWallTimeElapsed()/1000.0));
		}
		else {
			System.out.println("     " + sim.getTime() + "\t" + (sim.getWallTimeElapsed()/1000.0));
			System.out.println("Time for the whole thing = " + ((System.currentTimeMillis() - starttime) / 1000.0));
			System.out.println("Simulation ends.\n" + sim.diag());
			System.out.println("Event Queue:\n" + ((drcl.util.queue.Queue)sim.getEventQueue()).diag(false));
			if (EXIT) System.exit(0);
			else sim.stop();
		}
	}
}