package negotiator.parties;

import java.util.List;

import genius.core.Bid;
import genius.core.Domain;
import genius.core.actions.Action;
import genius.core.actions.Offer;
import genius.core.parties.AbstractNegotiationParty;
import genius.core.parties.NegotiationInfo;
import genius.core.uncertainty.BidRanking;
import genius.core.uncertainty.ExperimentalUserModel;
import genius.core.utility.AbstractUtilitySpace;
import genius.core.utility.CustomUtilitySpace;

public class UncertaintyAgentCustomUtilitySpace extends AbstractNegotiationParty 
{
	
	@Override
	public void init(NegotiationInfo info) 
	{
		super.init(info);
		log("This is the Uncertainty Agent example with Custom UtilitySpace.");
		log("The user model is: " + userModel);
		log("The default estimated utility space is: " + getUtilitySpace());

		Bid randomBid = getUtilitySpace().getDomain().getRandomBid(rand);
		log("The default estimate of the utility of a random bid (" + randomBid
				+ ") is: " + getUtility(randomBid));

		if (userModel instanceof ExperimentalUserModel) {
			log("You have given the agent access to the real utility space for debugging purposes.");
			ExperimentalUserModel e = (ExperimentalUserModel) userModel;
			AbstractUtilitySpace realUSpace = e.getRealUtilitySpace();

			log("The real utility space is: " + realUSpace);
			log("The real utility of the random bid is: "
					+ realUSpace.getUtility(randomBid));
		}
	}

	@Override
	public Action chooseAction(List<Class<? extends Action>> possibleActions) 
	{
		// Otherwise, return a random offer
		Bid randomBid;
		do
		{
			randomBid = generateRandomBid();
		}
		while (getUtility(randomBid) < 0.9);
		return new Offer(getPartyId(), randomBid);
	}

	private void log(String s) {
		System.out.println(s);
	}

	/**
	 * We override the default estimate of the utility
	 * space by using {@link ClosestKnownBid} defined below.
	 */
	@Override
	public AbstractUtilitySpace estimateUtilitySpace() 
	{
		return new ClosestKnownBid(getDomain());
	}

	@Override
	public String getDescription() {
		return "Example agent with a custom utility space";
	}
	
	private class ClosestKnownBid extends CustomUtilitySpace
	{

		public ClosestKnownBid(Domain dom) {
			super(dom);
		}

		@Override
		public double getUtility(Bid bid) 
		{
			Bid closestRankedBid = getClosestBidRanked(bid);
			System.out.println("Closest bid: " + closestRankedBid);
			return estimateUtilityOfRankedBid(closestRankedBid);
		}
		
		public double estimateUtilityOfRankedBid(Bid b)
		{
			BidRanking bidRanking = getUserModel().getBidRanking();
			int i = bidRanking.indexOf(b);
			
			System.out.println("Index: " + i);
			// index:0 has utility 0, index n-1 has utility 1
			return i / (double) bidRanking.getSize();
		}
		
		public Bid getClosestBidRanked(Bid bid)
		{
			List<Bid> bidOrder = getUserModel().getBidRanking().getBidOrder();
			Bid closestBid = null;
			double closestDistance = Double.MAX_VALUE;
			int rank = 0;
			int closestRank = 0;
				
			for (Bid b : bidOrder)
			{
				double d = 1 / (double) b.countEqualValues(bid);
				if (d < closestDistance)
				{
					closestDistance = d;
					closestBid = b;
					closestRank = rank;
				}
				rank++;
			}
			return closestBid;
		}
		
	}

}