import uchicago.src.sim.engine.SimpleModel;
import uchicago.src.sim.analysis.Plot;
import java.util.ArrayList;
import java.io.*;

public class MarketModel extends SimpleModel {

    protected int numBuyers  = 5;
    protected int numSellers = 5;
    protected int mTick = 50;
    protected int hMacro = 10;
    protected Time time;

    protected ArrayList buyers;
    protected ArrayList sellers;

    protected String outputFile = "standard.txt";
    protected FileWriter file;

    /**
     * Constructor
     * initialize parameter and model names
     */
    public MarketModel() {
	params = new String[]{"NumBuyers", "NumSellers", "OutputFile"};
	name = "Virtual Market Model";
    }

    /**
     * Called by SimpleModel.begin() (i.e. when clicking the initialize button)
     * Creates the agents and initializes them (calls initagents() )
     */
    public void buildModel() {
	int i;
	int id=0;

	time=new Time(mTick,hMacro);
	buyers=new ArrayList();
	sellers=new ArrayList();

	for (i=0; i<numBuyers; i++, id++) {
	    BuyerAgent buyer=new BuyerAgent(id);
	    agentList.add(buyer);
	    buyers.add(buyer);
	}
	for (i=0; i<numSellers; i++, id++) {
	    SellerAgent seller=new SellerAgent(id);
	    agentList.add(seller);
	    sellers.add(seller);
	}
	initAgents();

	// write header of file, force new file:
	try {
	    file = new FileWriter(outputFile,false);	
	    file.write("time;buyerID;buyerValue;sellerID;sellerCost;price;quantity\n");
	    file.close();
	} catch (IOException e) {
	    System.out.println("Error writing file!");
	}	

    }

    /**
     * Called every tick by SimpleModel.run()
     * For each agent in the agent list, calls its step method.
     * At the begining of each macroTick and hyperTick, calls the agent's
     *  macroStep or hyperStep methods (except on tick 0)
     */
    protected void step() {
	time.inc();
	int size = agentList.size();
	for (int i=0; i< size; i++) {
	    TraderAgent agent=(TraderAgent)agentList.get(i);
	    agent.step(time);
	}
	if (time.macro())
	    for (int i=0; i< size; i++) {
		TraderAgent agent=(TraderAgent)agentList.get(i);
		agent.macroStep(time);
	    }
	if (time.hyper()) 
	    for (int i=0; i< size; i++) {
		TraderAgent agent=(TraderAgent)agentList.get(i);
		agent.hyperStep(time);
	    }

    }

    /**
     * For each agent in the list, initializes its values and calls its init() method
     */
    public void initAgents() {
	int size = agentList.size();
	for (int i=0; i< size; i++) {
	    TraderAgent agent=(TraderAgent)agentList.get(i);
	    agent.setModel(this);
	    if (agent.buys()) {
		agent.setStock(0);
		agent.setDesiredStock(1);
		agent.setMoney(30);
		agent.setValue(30);
	    } else if (agent.sells()) {
		agent.setStock(1);
		agent.setDesiredStock(0);
		agent.setMoney(0);
		agent.setValue(10);
	    }
	    agent.init();
	}
    }

    /**
     * A trade has been made between two agents
     * called only by the agent who accepts the trade
     * it is used to keep statistics
     * @param buyer The supplier in the transaction
     * @param seller The demander in the transaction
     * @param price The price at which the transaction was made
     * @param qty Number of units traded (defaults to 1)
     */
    public void transaction(TraderAgent buyer, TraderAgent seller, int price, int qty) {
	System.out.println(time+" Trade between buyer "+buyer.getId()+" (val "+buyer.getValue()+") and seller "+seller.getId()+" (val "+seller.getValue()+"). "+qty+ " units at $"+price+" each");
   	transToFile(buyer, seller, price, qty, time);
    }
    public void transaction(TraderAgent buyer, TraderAgent seller, int price) { 
	transaction(buyer, seller, price, 1);
    }


    /**
     *
     * Sends a message (It is the Post Office)
     * @param sender The Agent who sends the message
     * @param receiver The message's destination (Agent)
     * @param msg The message (what else?)
     */
    public void sendMessage(TraderAgent sender, TraderAgent receiver, Message msg) {
	msg.sender=sender;
	msg.timestamp=new Time(time);
	receiver.inboxAdd(msg);
    }

    //////////
    // 
    // Accessors
    // 
    //////////

    public ArrayList getAgentList() {
	return agentList;
    }
    public ArrayList getBuyerList() {
	return buyers;
    }
    public ArrayList getSellerList() {
	return sellers;
    }


    public void setNumBuyers(int val) {
	numBuyers=val;
    }
    public int getNumBuyers() {
	return numBuyers;
    }

    public void setNumSellers(int val) {
	numSellers=val;
    }
    public int getNumSellers() {
	return numSellers;
    }

    public void setOutputFile(String val) {
	outputFile=val;
    }
    public String getOutputFile() {
	return outputFile;
    }

    //////////
    //
    // Output
    //
    //////////
    public void transToFile(TraderAgent buyer, TraderAgent seller, int price, int qty, Time time) {
	try {
	    file = new FileWriter(outputFile,true);	
	    file.write(time + ";" + buyer.getId() + ";" + buyer.getValue() + ";" 
		       + seller.getId() + ";" + seller.getValue() + ";" + price + ";" + qty + "\n");
	    file.close();
	} catch (IOException e) {
	    System.out.println("Error writing file!");
	}
    }
  
}