source: src/main/java/agents/anac/y2015/pnegotiator/BayesLearner.java@ 1

package agents.anac.y2015.pnegotiator;

import java.util.ArrayList;
import java.util.List;

import genius.core.AgentID;
import genius.core.Bid;
import genius.core.actions.Accept;
import genius.core.actions.Action;
import genius.core.actions.ActionWithBid;
import genius.core.actions.DefaultAction;
import genius.core.actions.Offer;
import genius.core.issue.Objective;
import genius.core.parties.AbstractNegotiationParty;
import genius.core.parties.NegotiationInfo;
import genius.core.utility.AdditiveUtilitySpace;

/**
 * This is your negotiation party.
 */
public class BayesLearner extends AbstractNegotiationParty {

        // State of the agent (am I being a hardliner or am I conceding?)
        public enum AgentState {
                HARDLINER, CONCEDER
        };

        public AgentState agentState;

        // Parameters of the negotiation
        // private UtilitySpace utilitySpace;
        // private Timeline timeline;
        private List<Objective> objectives;

        // Other private fields
        private Bid currentBid = null;
        private Bid nextBid = null;
        // private double concessionFactor = 1;
        // private int totalBids = 1;

        private BestBids bestBids;
        private BayesLogic bayesLogic;

        private double lRng = 0.9, rRng = 1.0;

        @Override
        public void init(NegotiationInfo info) {
                super.init(info);
                // Set agent's game parameters
                // this.utilitySpace = utilitySpace;
                // this.timeline = timeline;
                this.objectives = utilitySpace.getDomain().getObjectives();

                try {
                        this.bestBids = new BestBids((AdditiveUtilitySpace) this.utilitySpace);
                } catch (Exception e) {
                        e.printStackTrace();
                }

                agentState = AgentState.HARDLINER;

                // Print stuff for the benefit of our understanding
                // System.out.println("Discount factor: " +
                // utilitySpace.getDiscountFactor());
                // for(int i = 1; i < objectives.size(); ++i)
                // {
                // System.out.println("(Name, Type, Weight): (" +
                // objectives.get(i).getName() + ", " + objectives.get(i).getType() +
                // ", "+ utilitySpace.getWeight(i) + ")");
                // }
                //
                // System.out.println("Disagreement point (Discounted): " +
                // utilitySpace.getReservationValueWithDiscount(timeline));
                // System.out.println("Disagreement point (Undiscounted): " +
                // utilitySpace.getReservationValueUndiscounted());
                // System.out.println("Discount factor: " +
                // utilitySpace.getDiscountFactor());
                // System.out.println("Number of possible bids: " +
                // domain.getNumberOfPossibleBids());
        }

        /**
         * Each round this method gets called and ask you to accept or offer. The
         * first party in the first round is a bit different, it can only propose an
         * offer.
         *
         * @param validActions
         *            Either a list containing both accept and offer or only offer.
         * @return The chosen action.
         */
        public Action chooseAction(List<Class<? extends Action>> validActions) {
                // ++totalBids;
                try {
                        setState();
                        updateBayes();
                        System.out.println(currentBid);
                        if (currentBid == null) {
                                nextBid = utilitySpace.getMaxUtilityBid();
                        } else {
                                // setState();
                                lRng = getConcessionUtility();
                                Bid bestBid = bestBids.getRandomBid(generateRandomBid(), rand, lRng, rRng);
                                double CU = utilitySpace.getUtility(currentBid);
                                System.out.format("%6.4f < %6.4f < %6.4f\n", lRng, utilitySpace.getUtility(bestBid), rRng);
                                switch (agentState) {
                                case HARDLINER:
                                        // System.out.println("HARDLINER");
                                        nextBid = bestBid;
                                        break;
                                default:
                                        // System.out.println("CONCEDER");
                                        Bid bayes;
                                        double U1;
                                        do {
                                                bayes = bayesLogic.bayesBid(utilitySpace.getMaxUtilityBid());
                                                U1 = utilitySpace.getUtility(bayes);
                                                // System.out.format("V: %3d U: %6.4f\n", bayesLogic.V,
                                                // U1);
                                                bayesLogic.V++;
                                        } while (U1 < lRng);
                                        bayesLogic.V = 1;
                                        // bayes = bayesLogic.bayesBid2(rand);
                                        // U1 = utilitySpace.getUtility(bayes);
                                        // System.out.format("Bayes: %6.4f | Proposed: %6.4f\n",
                                        // U1,CU);
                                        nextBid = bayes;
                                }
                                // bayesLogic.updateOpponentFrequency(bestBid, 0);
                                if (utilitySpace.getUtility(currentBid) >= lRng)
                                        return new Accept(getPartyId(), ((ActionWithBid) getLastReceivedAction()).getBid());
                        }
                } catch (Exception e) {
                        e.printStackTrace();
                }
                // Print the total utility of the bid we are making and submit it
                System.out.println("Proposing bid with utility: " + utilitySpace.getUtilityWithDiscount(nextBid, timeline));
                return new Offer(getPartyId(), nextBid);
        }

        private void updateBayes() throws Exception {
                if (bayesLogic == null) {
                        this.bayesLogic = new BayesLogic((AdditiveUtilitySpace) this.utilitySpace, this.getNumberOfParties());
                }
                bayesLogic.T = (int) (getTimeLine().getTime() * 100);
                System.out.format("T: %d\n", bayesLogic.T);
        }

        /**
         * All offers proposed by the other parties will be received as a message.
         * You can use this information to your advantage, for example to predict
         * their utility.
         *
         * @param sender
         *            The party that did the action.
         * @param action
         *            The action that party did.
         */
        @Override
        public void receiveMessage(AgentID sender, Action action) {
                super.receiveMessage(sender, action);
                if (!players.contains(sender)) {
                        players.add(sender);
                }
                int P = players.indexOf(sender);
                if (DefaultAction.getBidFromAction(action) != null)
                        currentBid = DefaultAction.getBidFromAction(action);
                try {
                        if (currentBid != null) {
                                updateBayes();
                                bayesLogic.updateOpponentFrequency(currentBid, P);
                        }
                } catch (Exception e) {
                        e.printStackTrace();
                }
        }

        ArrayList<Object> players = new ArrayList<Object>();

        // A dumb function that just sets a scalar concession factor based on the
        // current time
        private double getConcessionUtility() throws Exception {
                double Bj = utilitySpace.getUtility(utilitySpace.getMaxUtilityBid());
                double Cj = Bj * timeline.getTime() * timeline.getTime();
                double Uj = Bj - (Cj / objectives.size());
                // System.out.println("Max : "+Bj+"Cj "+Cj+" time:
                // "+timeline.getTime()+"Bound:"+Uj);
                return Uj * Uj;
        }

        // Logic to decide the next bid
        // private void decideNextBid() throws Exception {
        // switch(agentState) {
        // case HARDLINER:
        // System.out.println("MaxRand");
        // nextBid = bestBids.getRandomBid(generateRandomBid(), rand, 0.8, 1.);
        // break;
        // case CONCEDER:
        // System.out.println("Bayes");
        // nextBid = bayesLogic.bayesBid(utilitySpace.getMaxUtilityBid());
        // }
        // }

        private void setState() {
                if (timeline.getTime() >= .2) {
                        agentState = AgentState.CONCEDER;
                }
        }

        @Override
        public String getDescription() {
                return "ANAC2015-22-PNegotiator";
        }
}