#include "Factory.h"
#include "Genr.h"
#include <iostream>
#include <cstdlib>
#include <map>
using namespace adevs;
using namespace std;

/**
 * The observer will keep track of individual order service times.
 */
class Observer:
	public EventListener<int>
{
	public:
		Observer():EventListener<int>()
		{ 
			max_time = avg_time = 0.0;
			count = 0; 
			min_time = DBL_MAX; // this should be 3 at the end of a run
		}
		// Track order processing statistics as orders move through the system
		void outputEvent(Event<int> x, double t)
		{
			// Ignore machine outputs, just look at factory and generator events
			if (dynamic_cast<Machine*>(x.model) != NULL) return;
			// Put new orders into the order table
			if (orders.find(x.value) == orders.end())
			{
				orders[x.value] = t;
			}
			// Compute statistics for orders that are complete
			else
			{
				// Maximum time to process any order
				max_time = max(max_time,t-orders[x.value]);
				// Min time to process any order
				min_time = min(min_time,t-orders[x.value]);
				// Update the average
				count++;
				avg_time += t-orders[x.value];
				// Clear order from the table
				orders.erase(x.value);
			}
		}
		// Get the maximum service time
		double maxServiceTime() { return max_time; }
		// Get the minimum service time
		double minServiceTime() { return min_time; }
		// Get the average service time
		double avgServiceTime() { return avg_time/count; }
	private:
		map<int,double> orders;
		double max_time, avg_time, min_time;
		int count;
};

int main(int argc, char** argv)
{
	// Create the model
	Factory* factory = new Factory();
	Genr* genr = new Genr(atoi(argv[1])); // Seed from command line argument
	SimpleDigraph<int>* model = new SimpleDigraph<int>();
	model->add(factory);
	model->add(genr);
	model->couple(genr,factory);
	// Create the simulator
	Simulator<int>* sim = new Simulator<int>(model);
	Observer* obs = new Observer();
	sim->addEventListener(obs);
	// Initial active count (should be 0)
	cout << "0 " << factory->getMachineCount() << endl;
	// Run the simulation and output active machine count at each iteration
	while (sim->nextEventTime() <= 365.0)
	{
		cout << sim->nextEventTime() << " ";
		sim->execNextEvent();
		cout << factory->getMachineCount() << endl;
	}
	// Output service time statistics
	cerr << "Avg. service time: " << obs->avgServiceTime() << " days" << endl;
	cerr << "Max. service time: " << obs->maxServiceTime() << " days" << endl;
	cerr << "Min. service time: " << obs->minServiceTime() << " days" << endl;
	// Clean up and exit
	delete sim;
	delete obs;
	delete model;
	return 0;
}