/*
 * (c) 1998-2000 The Brookings Institution, All Rights Reserved
 *
 * Permission to use this software and its documentation for non-commercial
 * purposes and without fee is hereby granted, provided this copyright statement
 * is included. Please contact us for permission for redistribution and other uses.
 * 
 * BROOKINGS MAKES NO REPRESENTATIONS OR WARRANTIES ABOUT THE SUITABILITY OF
 * THE SOFTWARE, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
 * TO THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, 
 * OR NON-INFRINGEMENT. BROOKINGS SHALL NOT BE LIABLE FOR ANY DAMAGES SUFFERED BY
 * LICENSEE AS A RESULT OF USING THIS SOFTWARE OR ITS DERIVATIVES.
 *
 * _Sugarscape_
 * See _Growing Artificial Societies: Social Science from the Ground Up_
 * Brookings Institution Press, The MIT Press
 * Joshua Epstein
 * jepstein@brook.edu
 * Robert Axtell
 * raxtell@brook.edu
 *
 * Miles Parker
 * mparker@brook.edu

 * http://www.brook.edu/es/dynamics/models/ascape
 * The Brookings Institution
 * Washington, D.C.
 */

package edu.brook.sugarscape;

import java.lang.*;
import java.util.*;
import java.awt.Color;

import edu.brook.ascape.model.*;
import edu.brook.ascape.rule.*;
import edu.brook.ascape.util.*;

/**
 * A basic sugarscape citizen. Provides basic funtionality for sugarscape agents
 * as well as all desired functionality that could be included in base class without
 * compromising good design or supporting unnecessary member variables.
 * Despite the relativly large size of this class it is actually
 * quite simple; much of the code is simply getters and setters for various 
 * initialization paramaters.
 * 
 * @author Alan Lockard
 * alockard@gmu.edu
 * @version 1.0
 */
public class SpiceAgent extends SugarAgent {

    public final static Rule DEATH_STARVATION_SPICE_RULE = new Rule("Death From Starvation") {
        public void execute(Agent agent) {
            //System.out.println("sugar = " + ((SpiceAgent) agent).getSugar().getStock() + " spice = " + ((SpiceAgent) agent).getSpice().getStock()); 
            if ((((SpiceAgent) agent).getSugar().getStock() <= 0)
                || (((SpiceAgent) agent).getSpice().getStock() <= 0)) {
                    agent.die();
            }
        }
    };

    public final static Rule DEATH_STARVATION_OLD_AGE_SPICE_RULE = new Rule("Death From Starvation and/or Old Age") {
        public void execute(Agent agent) {
            //if (((SpiceAgent) agent).getSugar().getStock() <= 0) System.out.println ("Died for Lack of SUGAR");
            //if (((SpiceAgent) agent).getSpice().getStock() <= 0) System.out.println ("Died for Lack of Spice");
            //if (((SugarAgent) agent).getAge() >= ((SugarAgent) agent).getDeathAge()) System.out.println ("Died of old aGe");
            if ((((SpiceAgent) agent).getSugar().getStock() <= 0) 
	            || (((SpiceAgent) agent).getSpice().getStock() <= 0)
	            || (((SugarAgent) agent).getAge() >= ((SugarAgent) agent).getDeathAge())) {
                        agent.die();
            }
        }
    };

    //Agent attributes
    protected int spiceMetabolism;
    public CommodityStock spice;
    protected float aBigNumber = 100000f; // not quite + infinity 
/**
 * Factor by which one would like to inflate (deflate) price 
 * when buying spice (sugar)
 */
    protected float negotiatingFactor = .25f;
    protected float firstOfferFactor = 2.0f;
    public int maxNumOffers = 6;
    
    public void initialize() {
	super.initialize();
        spice = new CommodityStock();
        spice.setName("spice");
        spice.setOwner(this);
        setSpiceMetabolism(randomInRange(((GAS_SpiceBase) getRoot()).getMinSpiceMetabolism(), ((GAS_SpiceBase) getRoot()).getMaxSpiceMetabolism()));
        setSpice(randomInRange(((GAS_SpiceBase) getRoot()).getMinInitialSpice(), ((GAS_SpiceBase) getRoot()).getMaxInitialSpice()));
    }

    public void scapeCreated() {
        //scape.addRule(new DefaultUpdate());
        StatCollector[] stats = {new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).getSugar().getStock();
            }
            public String getName() {
                return "Sugar";
            }
        },
        new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).getSpice().getStock();
            }
            public String getName() {
                return "Spice";
            }
        },
        new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).calculateMRS();
            }
            public String getName() {
                return "MRS";
            }
        },
/*
        new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).getAge();
//                return -((SugarAgent) object).getAge();
            }
            public String getName() {
                return "Age";
            }
        },
        new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).getVision();
            }
            public String getName() {
                return "Vision";
            }
        },
*/
        new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).getSugarMetabolism();
            }
            public String getName() {
                return "Sugar Metabolism";
            }
        },
        new StatCollectorCSAMMVar() {
            public double getValue(Object object) {
                return ((SpiceAgent) object).getSpiceMetabolism();
            }
            public String getName() {
                return "Spice Metabolism";
            }
        },

        new StatCollectorCSAMMVar() {
	    // returning double here no longer required in most recent code
	    // Because isAutoCollect() returns false, this value of 0.0
	    // is never actually collected
            public double getValue(Object object) {
                return 0.0;
            }
            public String getName() {
                return "Trades";
            }
		    public boolean isAutoCollect() {
				return false;
		    }
        },
        new StatCollectorCSAMMVar() {
	    // returning double here no longer required in most recent code
	    // Because isAutoCollect() returns false, this value of 0.0
	    // is never actually collected
            public double getValue(Object object) {
                return 0.0;
            }
            public String getName() {
                return "Price";
            }
		    public boolean isAutoCollect() {
				return false;
		    }
        },
        new StatCollectorCSAMMVar() {
	    // returning double here no longer required in most recent code
	    // Because isAutoCollect() returns false, this value of 0.0
	    // is never actually collected
            public double getValue(Object object) {
                return 0.0;
            }
            public String getName() {
                return "Log Price";
            }
		    public boolean isAutoCollect() {
				return false;
		    }
        },
        new StatCollectorCSAMMVar() {
	    // returning double here no longer required in most recent code
	    // Because isAutoCollect() returns false, this value of 0.0
	    // is never actually collected
            public double getValue(Object object) {
                return 0.0;
            }
            public String getName() {
                return "Self Gains from Trade";
            }
		    public boolean isAutoCollect() {
				return false;
		    }
        },
        new StatCollectorCSAMMVar() {
	    // returning double here no longer required in most recent code
	    // Because isAutoCollect() returns false, this value of 0.0
	    // is never actually collected
            public double getValue(Object object) {
                return 0.0;
            }
            public String getName() {
                return "Partner Gains from Trade";
            }
		    public boolean isAutoCollect() {
				return false;
		    }
        }
     };
        scape.addStatCollectors(stats);
    }

    public CommodityStock getSpice() {
        return spice;
    }

    public void harvest() {
        putSugar(((SpiceCell) getHostCell()).takeSugar());
        putSpice(((SpiceCell) getHostCell()).takeSpice());
    }

    public void metabolism() {
        super.metabolism();
        spice.reduceStock(spiceMetabolism);
        age++;
    }
    
    public void movement() {
        float sugarStock = getSugar().getStock();
        float spiceStock = getSpice().getStock();
        int sugarMet = getSugarMetabolism();
        int spiceMet = getSpiceMetabolism();
        Cell[] visibleCells = getHostScape().getCellsNear(getHostCell(), getVision(), true);
        float bestValue = ((SpiceCell) visibleCells[0]).getPotentialValue(sugarStock, spiceStock, sugarMet, spiceMet);
        for (int i = 1; i < visibleCells.length; i++) {
            if (((SpiceCell) visibleCells[i]).getPotentialValue(sugarStock, spiceStock, sugarMet, spiceMet) > bestValue && ((HostCell) visibleCells[i]).isAvailable()) {
                bestValue = ((SpiceCell) visibleCells[i]).getPotentialValue(sugarStock, spiceStock, sugarMet, spiceMet);
            }
        }
        if (((SpiceCell) visibleCells[0]).getPotentialValue(sugarStock, spiceStock, sugarMet, spiceMet) != bestValue) {
            SpiceCell spice = null;
            for (int i = 1; i <= getVision(); i++) {
                int series = randomToLimit(24);
                for (int j = 0; j < 4; j++) {
                    SpiceCell current = (SpiceCell) visibleCells[(i * 4 - Utility.uniqueSeries[4][series][j]) + 1];
                    if ((current.getPotentialValue(sugarStock, spiceStock, sugarMet, spiceMet) == bestValue) && current.isAvailable()) {
                        moveTo(current);
                        return;
                    }
                }
            }
        }
    }            
           
    public void putSpice(float saffron) {
	    spice.putStock(saffron);
    }
    
    public float getSpiceStock() {
        return spice.getStock();
    }

    public void setSpice(float saffron) {
        spice.setStock(saffron);
    }
    
    public float takeSpice() {
        return spice.takeStock();
    }
    
    public float takeSpice(float saffron) {
        return spice.takeStock(saffron);
    }
    
    public void addSpice(float saffron) {
        spice.addStock(saffron);
    }
    
    public int getSpiceMetabolism() {
        return spiceMetabolism;
    }

    public void setSpiceMetabolism(int metabolism) {
        this.spiceMetabolism = metabolism;
    }   

    public void setNegotiatingFactor (float factor) {
	this.negotiatingFactor = factor;
    }

    public float getNegotiatingFactor () {
	return negotiatingFactor;
    }

    public float calculateMRS (float sugarStock, float spiceStock) {
        float mrs;
	if (sugarStock <= 0) { // point of starvation
            mrs = aBigNumber;  // arbitrary, pretty big number (price)
        }
	else {
            if (spiceStock <= 0) {
		mrs = 1/aBigNumber;
            }
	    else {
                mrs = (this.sugarMetabolism * spiceStock) / 
                      (this.spiceMetabolism * sugarStock);
	    }
	}
        if (mrs < 0) System.out.println("mrs = " + mrs + " = (" + this.sugarMetabolism + " * " + spiceStock + ") / (" + this.spiceMetabolism + " * " + sugarStock + ")");
//if (Float.isNaN(mrs)) System.out.println ("mrs = " + mrs + ", sugarmet = " + this.sugarMetabolism + ", spicemet = " + this.spiceMetabolism + ", sugarstock = " + sugarStock + ", spiceStock = " + spiceStock);
	return mrs;
    }
    
    public float calculateMRS () {
        float mrs;
	if (sugar.getStock() <= 0) { // point of starvation
            mrs = aBigNumber;  // arbitrary, pretty big number (price)
        }
	else {
            if (spice.getStock() < 0) {
		mrs = 1/aBigNumber;
	    }            
	    else {
                mrs = (this.sugarMetabolism * spice.getStock()) / 
                      (this.spiceMetabolism * sugar.getStock());
	    }
	}
        if (mrs < 0) System.out.println("mrs = " + mrs + " = (" + this.sugarMetabolism + " * " + spice.getStock() + ") / (" + this.spiceMetabolism + " * " + sugar.getStock() + ")");
//if (Float.isNaN(mrs)) System.out.println ("mrs = " + mrs + ", sugarmet = " + this.sugarMetabolism + ", spicemet = " + this.spiceMetabolism + ", sugarstock = " + sugar.getStock() + ", spiceStock = " + spice.getStock());
	return mrs;
    }

    public float calculateUtility (float sugarStock, float spiceStock) {
        int mt = this.sugarMetabolism + this.spiceMetabolism;
        return (float) (Math.pow(sugarStock, (float) this.sugarMetabolism/mt) 
                      * Math.pow(spiceStock, (float) this.spiceMetabolism/mt));
    }

    public float calculateUtility () {
        int mt = this.sugarMetabolism + this.spiceMetabolism;
        return (float) (Math.pow(sugar.getStock(), (float) this.sugarMetabolism/mt)
                      * Math.pow(spice.getStock(), (float) this.spiceMetabolism/mt));
    }

    public float makeOffer (SpiceAgent partner) {
	float offer;
        offer = (float) Math.sqrt(calculateMRS() * partner.calculateMRS());
	if (Float.isNaN(offer)) {
	    System.out.println ("Offer = " + offer + " = (Math.sqrt(" + calculateMRS() + " * " + partner.calculateMRS() + ")");
	}
	return offer;
    }

    public boolean acceptOffer (Exchange exchange) {
	if ((exchange.tradeIsValid())  
           && (exchange.selfGainsFromTrade > 0) // good for me
           && (exchange.partnerGainsFromTrade > 0) // good for you
	   && (exchange.selfExAnteMRS != exchange.partnerExAnteMRS) // goods not valued equally
	   && !((exchange.selfExAnteMRS > exchange.partnerExAnteMRS) ^ (exchange.selfExPostMRS > exchange.partnerExPostMRS))) { // MRSs don't flip. Otherwise could lead to cycling.
	    return true;
	}
	else { return false;}
    }
}